system_setup_tool/search_cmd.go

84 lines
2.1 KiB
Go

package main
import (
"fmt"
"log"
"strings"
"github.com/spf13/cobra"
)
var searchCmd = &cobra.Command{
Use: "search [package_name]",
Short: "Search for a package across package managers (will search in os|flatpak|homebrew)",
Args: cobra.ExactArgs(1),
Run: func(cmd *cobra.Command, args []string) {
packageName := args[0]
managerName, _ := cmd.Flags().GetString("manager")
// var manager PackageManager
if managerName == "" {
managers := []string{"brew", "os", "flatpak"}
for _, manager := range managers {
fmt.Printf("Searching in %s:\n", manager)
results := searchPackage(packageName, manager)
displayResults(results, manager)
fmt.Println()
}
}
},
}
func displayResults(results []string, manager string) {
if len(results) == 0 {
fmt.Printf("No results found in %s\n", manager)
return
}
for _, result := range results {
fmt.Printf("- %s\n", result)
}
}
func searchPackage(packageName, manager string) []string {
switch manager {
case "brew":
cmd := execCommand("brew", "search", packageName)
packages, err := cmd.Output()
if err != nil {
log.Printf("error fetching %s packages: %v", manager, err)
}
packageList := strings.Split(strings.TrimSpace(string(packages)), "\n")
return packageList
case "os":
opSys, err := getLinuxDistribution()
if err != nil {
log.Printf("Error getting OS information: %v", err)
return []string{}
}
cmdParts := strings.Fields(opSys.SearchCommand)
if len(cmdParts) == 0 {
log.Printf("Invalid search command for OS package manager")
return []string{}
}
cmd := execCommand(cmdParts[0], append(cmdParts[1:], packageName)...)
packages, err := cmd.Output()
if err != nil {
log.Printf("Error fetching %s packages: %v", manager, err)
return []string{}
}
packageList := strings.Split(strings.TrimSpace(string(packages)), "\n")
return packageList
case "flatpak":
cmd := execCommand("flatpak", "search", packageName)
packages, err := cmd.Output()
if err != nil {
log.Printf("error fetching %s packages: %v", manager, err)
}
packageList := strings.Split(strings.TrimSpace(string(packages)), "\n")
return packageList
}
return []string{}
}