switch initramfs too

cmd/kernelswitcher/list.go

@@ -46,7 +46,7 @@ $ mos kernel-switcher switch kernel/mocaccino-full
 			if err != nil {
 				log.Fatal(err)
 			}
-			kernels := allKernelsPackages.FilterByName("firmware").FilterByName("initramfs").FilterByName("minimal").FilterByName("modules").FilterByName("sources")
+			kernels := allKernelsPackages.FilterOutByName("firmware").FilterOutByName("initramfs").FilterOutByName("minimal").FilterOutByName("modules").FilterOutByName("sources")
 
 			for _, j := range kernels.Packages {
 				install := ""

cmd/kernelswitcher/switch.go

@@ -72,7 +72,30 @@ $ mos kernel-switcher switch kernel/mocaccino-lts-full
 				panic(lookErr)
 			}
 
-			cmdargs := []string{"luet", "replace", "--nodeps", "--for", argkernel, "--for", fmt.Sprintf("%smodules", strings.ReplaceAll(argkernel, "full", ""))}
+			for _, i := range installed.Packages {
+				installedRevDeps, err := kernel.RevDepsForPackage(i)
+				if err != nil {
+					log.Fatal(err)
+				}
+				installedRevDeps = installedRevDeps.FilterByInstalled(true)
+				installed = kernel.MergeSearchResults(installedRevDeps, installed)
+			}
+
+			revDeps, err := kernel.RevDeps(argkernel)
+			if err != nil {
+				log.Fatal(err)
+			}
+			revDeps = revDeps.FilterByInstalled(false)
+			revDeps = kernel.FilterByNamePart(revDeps, installed, "-lts")
+
+			cmdargs := []string{"luet", "replace", "--nodeps",
+				"--for", argkernel,
+				"--for", fmt.Sprintf("%smodules", strings.ReplaceAll(argkernel, "full", "")),
+				"--for", fmt.Sprintf("%sinitramfs", strings.ReplaceAll(argkernel, "full", "")),
+			}
+			for _, k := range revDeps.Packages {
+				cmdargs = append(cmdargs, "--for", fmt.Sprintf("%s/%s", k.Category, k.Name))
+			}
 			for _, i := range installed.Packages {
 				cmdargs = append(cmdargs, fmt.Sprintf("%s/%s", i.Category, i.Name))
 			}

pkg/kernel/kernels.go

@@ -14,6 +14,7 @@ type SearchResult struct {
 
 type Package struct {
 	Name, Category, Version string
+	Installed               bool
 }
 
 func (p Package) Equal(pp Package) bool {
@@ -65,6 +66,25 @@ func All() (searchResult SearchResult, err error) {
 	return
 }
 
+func RevDepsForPackage(p Package) (searchResult SearchResult, err error) {
+	searchResult, err = RevDeps(fmt.Sprintf("%s/%s-%s", p.Category, p.Name, p.Version))
+	return
+}
+
+func RevDeps(s string) (searchResult SearchResult, err error) {
+	var res []byte
+	res, err = utils.RunSH("search", fmt.Sprintf("luet search %s --revdeps --output json", s))
+	if err != nil {
+		return
+	}
+	var revDeps SearchResult
+	json.Unmarshal(res, &revDeps)
+	searchResult = revDeps.FilterByCategory("kernel")
+	searchResult.Packages = append(searchResult.Packages, revDeps.FilterByCategory("kernel-modules").Packages...)
+	searchResult.Packages = append(searchResult.Packages, revDeps.FilterByCategory("kernel-sources").Packages...)
+	return
+}
+
 func (s SearchResult) FilterByCategory(cat string) SearchResult {
 	new := SearchResult{Packages: []Package{}}
 
@@ -76,7 +96,7 @@ func (s SearchResult) FilterByCategory(cat string) SearchResult {
 	return new
 }
 
-func (s SearchResult) FilterByName(name string) SearchResult {
+func (s SearchResult) FilterOutByName(name string) SearchResult {
 	new := SearchResult{Packages: []Package{}}
 
 	for _, r := range s.Packages {
@@ -86,3 +106,50 @@ func (s SearchResult) FilterByName(name string) SearchResult {
 	}
 	return new
 }
+
+func (s SearchResult) FilterByInstalled(installed bool) SearchResult {
+	new := SearchResult{Packages: []Package{}}
+
+	for _, r := range s.Packages {
+		if r.Installed == installed {
+			new.Packages = append(new.Packages, r)
+		}
+	}
+	return new
+}
+
+func MergeSearchResults(source SearchResult, target SearchResult) SearchResult {
+	for _, s := range source.Packages {
+		add := true
+		for _, t := range target.Packages {
+			if s == t {
+				add = false
+				break
+			}
+		}
+		if add {
+			target.Packages = append(target.Packages, s)
+		}
+	}
+
+	return target
+}
+
+func FilterByNamePart(source SearchResult, target SearchResult, namePart string) SearchResult {
+	new := SearchResult{Packages: []Package{}}
+
+	for _, s := range source.Packages {
+		add := false
+		for _, t := range target.Packages {
+			if s.Category == t.Category && strings.ReplaceAll(s.Name, namePart, "") == strings.ReplaceAll(t.Name, namePart, "") {
+				add = true
+				break
+			}
+		}
+		if add {
+			new.Packages = append(new.Packages, s)
+		}
+	}
+
+	return new
+}