Merge branch 'dev/add-update-command' into development

* dev/add-update-command:
  feat(cli): add update and update-all command
This commit is contained in:
Patryk Hegenberg 2025-01-19 22:23:09 +01:00
commit 43eb3ae4cb
12 changed files with 192 additions and 8 deletions

View file

@ -38,8 +38,10 @@ func init() {
searchCmd.Flags().StringP("manager", "m", "", "The package manager you want to search a package with. (Options: os|homebrew|flatpak)") searchCmd.Flags().StringP("manager", "m", "", "The package manager you want to search a package with. (Options: os|homebrew|flatpak)")
updateCmd.Flags().StringP("manager", "m", "os", "The package manager you want to update packages with. (Options: os|homebrew|pipx|flatpak)")
packageCmd.AddCommand(addCmd, deleteCmd, showCmd, enableCmd) packageCmd.AddCommand(addCmd, deleteCmd, showCmd, enableCmd)
RootCmd.AddCommand(packageCmd, searchCmd, installCmd, removeCmd) RootCmd.AddCommand(packageCmd, searchCmd, installCmd, removeCmd, updateCmd, updateAllCmd)
} }
func initConfig() { func initConfig() {

View file

@ -18,6 +18,11 @@ var installCmd = &cobra.Command{
packageName := args[0] packageName := args[0]
managerName, _ := cmd.Flags().GetString("manager") managerName, _ := cmd.Flags().GetString("manager")
var manager pm.PackageManager var manager pm.PackageManager
if managerName == "os" {
managerName = "OS Package Manager"
} else if managerName == "brew" {
managerName = "homebrew"
}
switch managerName { switch managerName {
case "os": case "os":
sudoPassword, err := utils.GetSudoPassword() sudoPassword, err := utils.GetSudoPassword()
@ -28,7 +33,7 @@ var installCmd = &cobra.Command{
if err := osManager.Install([]string{packageName}); err != nil { if err := osManager.Install([]string{packageName}); err != nil {
log.Printf("error: %v\n", err) log.Printf("error: %v\n", err)
} }
case "homebrew", "brew": case "homebrew":
manager = &pm.HomebrewManager{} manager = &pm.HomebrewManager{}
case "cargo": case "cargo":
manager = &pm.CargoManager{} manager = &pm.CargoManager{}

View file

@ -18,6 +18,11 @@ var removeCmd = &cobra.Command{
packageName := args[0] packageName := args[0]
managerName, _ := cmd.Flags().GetString("manager") managerName, _ := cmd.Flags().GetString("manager")
var manager pm.PackageManager var manager pm.PackageManager
if managerName == "os" {
managerName = "OS Package Manager"
} else if managerName == "brew" {
managerName = "homebrew"
}
switch managerName { switch managerName {
case "os": case "os":
sudoPassword, err := utils.GetSudoPassword() sudoPassword, err := utils.GetSudoPassword()
@ -28,7 +33,7 @@ var removeCmd = &cobra.Command{
if err := osManager.Install([]string{packageName}); err != nil { if err := osManager.Install([]string{packageName}); err != nil {
log.Printf("error: %v\n", err) log.Printf("error: %v\n", err)
} }
case "homebrew", "brew": case "homebrew":
manager = &pm.HomebrewManager{} manager = &pm.HomebrewManager{}
case "cargo": case "cargo":
manager = &pm.CargoManager{} manager = &pm.CargoManager{}

View file

@ -18,6 +18,8 @@ var searchCmd = &cobra.Command{
managerName, _ := cmd.Flags().GetString("manager") managerName, _ := cmd.Flags().GetString("manager")
if managerName == "os" { if managerName == "os" {
managerName = "OS Package Manager" managerName = "OS Package Manager"
} else if managerName == "brew" {
managerName = "homebrew"
} }
managers := []pm.PackageManager{ managers := []pm.PackageManager{
@ -27,9 +29,6 @@ var searchCmd = &cobra.Command{
} }
if managerName != "" { if managerName != "" {
if managerName == "brew" {
managerName = "homebrew"
}
for _, m := range managers { for _, m := range managers {
if strings.EqualFold(m.Name(), managerName) { if strings.EqualFold(m.Name(), managerName) {
searchAndDisplayResults(m, packageName) searchAndDisplayResults(m, packageName)

81
cmd/update_command.go Normal file
View file

@ -0,0 +1,81 @@
package cmd
import (
"fmt"
"log"
pm "system_setup_tool/packagemanager"
"system_setup_tool/utils"
"github.com/spf13/cobra"
)
var updateCmd = &cobra.Command{
Use: "update [package_name]",
Short: "update a package with the specified package manager.",
Long: "update a package with the specified package manager.\nIf no package specified, all packages for the package manager will be updated",
Args: cobra.ExactArgs(1),
Run: func(cmd *cobra.Command, args []string) {
packageName := args[0]
managerName, _ := cmd.Flags().GetString("manager")
var manager pm.PackageManager
if managerName == "os" {
managerName = "OS Package Manager"
} else if managerName == "brew" {
managerName = "homebrew"
}
switch managerName {
case "os":
sudoPassword, err := utils.GetSudoPassword()
if err != nil {
log.Fatal(err)
}
osManager := pm.NewOSManager(sudoPassword)
if err := osManager.Install([]string{packageName}); err != nil {
log.Printf("error: %v\n", err)
}
case "homebrew":
manager = &pm.HomebrewManager{}
case "cargo":
manager = &pm.CargoManager{}
case "pipx":
manager = &pm.PipxManager{}
case "flatpak":
manager = &pm.FlatpakManager{}
default:
fmt.Println("No PackageManager found")
return
}
if packageName != "" {
if err := manager.UpdatePackage(packageName); err != nil {
log.Printf("error: %v\n", err)
}
} else {
if err := manager.UpdateAllPackages(); err != nil {
log.Printf("error: %v\n", err)
}
}
},
}
var updateAllCmd = &cobra.Command{
Use: "update-all",
Short: "update all packages with all package managers.",
Run: func(cmd *cobra.Command, args []string) {
sudoPassword, err := utils.GetSudoPassword()
if err != nil {
log.Fatal(err)
}
managers := []pm.PackageManager{
pm.NewOSManager(sudoPassword),
&pm.HomebrewManager{},
&pm.FlatpakManager{},
&pm.PipxManager{},
}
for _, m := range managers {
fmt.Printf("Updating %s-Packages\n", m.Name())
m.UpdateAllPackages()
}
},
}

View file

@ -1,6 +1,9 @@
package packagemanager package packagemanager
import "system_setup_tool/internal/shell" import (
"fmt"
"system_setup_tool/internal/shell"
)
type CargoManager struct{} type CargoManager struct{}
@ -45,3 +48,11 @@ func (c *CargoManager) RemovePackage(pkg string) error {
func (c *CargoManager) SearchPackage(pkg string) []string { func (c *CargoManager) SearchPackage(pkg string) []string {
return []string{} return []string{}
} }
func (c *CargoManager) UpdatePackage(pkg string) error {
return fmt.Errorf("update not supported")
}
func (c *CargoManager) UpdateAllPackages() error {
return fmt.Errorf("update not supported")
}

View file

@ -116,3 +116,13 @@ func (f *FlatpakManager) SearchPackage(pkg string) []string {
packageList := strings.Split(strings.TrimSpace(string(packages)), "\n") packageList := strings.Split(strings.TrimSpace(string(packages)), "\n")
return packageList return packageList
} }
func (f *FlatpakManager) UpdatePackage(pkg string) error {
cmd := shell.ExecCommand("flatpak", "update", "-y", pkg)
return cmd.Run()
}
func (f *FlatpakManager) UpdateAllPackages() error {
cmd := shell.ExecCommand("flatpak", "update", "-y")
return cmd.Run()
}

View file

@ -1,6 +1,9 @@
package packagemanager package packagemanager
import "system_setup_tool/internal/shell" import (
"fmt"
"system_setup_tool/internal/shell"
)
type GolangManager struct{} type GolangManager struct{}
@ -45,3 +48,11 @@ func (g *GolangManager) RemovePackage(pkg string) error {
func (g *GolangManager) SearchPackage(pkg string) []string { func (g *GolangManager) SearchPackage(pkg string) []string {
return []string{} return []string{}
} }
func (g *GolangManager) UpdatePackage(pkg string) error {
return fmt.Errorf("update not supported")
}
func (g *GolangManager) UpdateAllPackages() error {
return fmt.Errorf("update not supported")
}

View file

@ -59,3 +59,13 @@ func (h *HomebrewManager) SearchPackage(pkg string) []string {
packageList := strings.Split(strings.TrimSpace(string(packages)), "\n") packageList := strings.Split(strings.TrimSpace(string(packages)), "\n")
return packageList return packageList
} }
func (h *HomebrewManager) UpdatePackage(pkg string) error {
cmd := shell.ExecCommand("brew", "upgrade", pkg)
return cmd.Run()
}
func (h *HomebrewManager) UpdateAllPackages() error {
cmd := shell.ExecCommand("brew", "update")
return cmd.Run()
}

View file

@ -74,6 +74,20 @@ func (o *OSManager) SearchPackage(pkg string) []string {
return packageList return packageList
} }
func (o *OSManager) UpdatePackage(pkg string) error {
fullCmd := fmt.Sprintf("%s %s", o.OS.UpdateCommand, pkg)
command := shell.ExecCommand("sudo", "-S", "sh", "-c", fullCmd)
command.Stdin = strings.NewReader(o.SudoPassword + "\n")
return command.Run()
}
func (o *OSManager) UpdateAllPackages() error {
fullCmd := fmt.Sprintf("%s", o.OS.UpdateCommand)
command := shell.ExecCommand("sudo", "-S", "sh", "-c", fullCmd)
command.Stdin = strings.NewReader(o.SudoPassword + "\n")
return command.Run()
}
type OS struct { type OS struct {
ID string ID string
Name string Name string
@ -82,6 +96,7 @@ type OS struct {
InstallCommand string InstallCommand string
SearchCommand string SearchCommand string
RemoveCommand string RemoveCommand string
UpdateCommand string
} }
func parseOsRelease(osRelease string) *OS { func parseOsRelease(osRelease string) *OS {
@ -91,6 +106,9 @@ func parseOsRelease(osRelease string) *OS {
result.Version = "Unknown" result.Version = "Unknown"
result.PackageManager = "Unkown" result.PackageManager = "Unkown"
result.InstallCommand = "Unkown" result.InstallCommand = "Unkown"
result.SearchCommand = "Unkown"
result.RemoveCommand = "Unkown"
result.UpdateCommand = "Unkown"
lines := strings.Split(osRelease, "\n") lines := strings.Split(osRelease, "\n")
@ -120,6 +138,10 @@ func parseOsRelease(osRelease string) *OS {
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
err = result.getUpdateCommand()
if err != nil {
log.Fatal(err)
}
return &result return &result
} }
@ -209,6 +231,22 @@ func (os *OS) getDeleteCommand() error {
} }
} }
func (os *OS) getUpdateCommand() error {
switch os.PackageManager {
case "apt":
os.UpdateCommand = "apt update"
return nil
case "pacman":
os.UpdateCommand = "pacman -S"
return nil
case "dnf":
os.UpdateCommand = "dnf update"
return nil
default:
return fmt.Errorf("no install command found for package manager: %s", os.ID)
}
}
func InstallBuildEssentials(os *OS, sudoPassword string) error { func InstallBuildEssentials(os *OS, sudoPassword string) error {
var command string var command string
switch os.PackageManager { switch os.PackageManager {

View file

@ -6,5 +6,7 @@ type PackageManager interface {
InstallManager() error InstallManager() error
RemovePackage(pkg string) error RemovePackage(pkg string) error
SearchPackage(pkg string) []string SearchPackage(pkg string) []string
UpdatePackage(pkg string) error
UpdateAllPackages() error
Name() string Name() string
} }

View file

@ -46,3 +46,13 @@ func (p *PipxManager) RemovePackage(pkg string) error {
func (p *PipxManager) SearchPackage(pkg string) []string { func (p *PipxManager) SearchPackage(pkg string) []string {
return []string{} return []string{}
} }
func (p *PipxManager) UpdatePackage(pkg string) error {
cmd := shell.ExecCommand("pipx", "upgrade", pkg)
return cmd.Run()
}
func (p *PipxManager) UpdateAllPackages() error {
cmd := shell.ExecCommand("pipx", "upgrade-all")
return cmd.Run()
}