136 lines
4.3 KiB
Go
136 lines
4.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"system_setup_tool/internal/tui"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var RootCmd = &cobra.Command{
|
|
Use: "system_setup_tool",
|
|
Short: "Installs packages based on TOML configuration",
|
|
Version: "0.4.0",
|
|
Run: tui.Run,
|
|
}
|
|
|
|
func init() {
|
|
cobra.OnInitialize(initConfig)
|
|
RootCmd.PersistentFlags().StringP("config", "c", "", "Path to the configuration file")
|
|
viper.BindPFlag("config", RootCmd.PersistentFlags().Lookup("config"))
|
|
|
|
addCmd.Flags().StringP("name", "n", "", "The name of the package you want to add")
|
|
addCmd.Flags().StringP("manager", "m", "", "The package manager you want to add the package to (homebrew|cargo|flatpak|pipx|go)")
|
|
addCmd.Flags().Bool("system", false, "Add as a system package")
|
|
addCmd.Flags().Bool("headless", false, "Add as a headless system package (only used with --system)")
|
|
|
|
deleteCmd.Flags().StringP("name", "n", "", "The name of the package you want to delete")
|
|
deleteCmd.Flags().StringP("manager", "m", "", "The package manager you want to delete the package from (homebrew|cargo|flatpak|pipx|go)")
|
|
deleteCmd.Flags().Bool("system", false, "Delete from system packages")
|
|
deleteCmd.Flags().Bool("headless", false, "Delete from headless system packages (only used with --system)")
|
|
|
|
enableCmd.Flags().Bool("value", true, "Set to true to enable, false to disable")
|
|
|
|
installCmd.Flags().StringP("manager", "m", "os", "The package manager you want to install a package with. (Options: os|homebrew|pipx|flatpak|cargo)")
|
|
|
|
removeCmd.Flags().StringP("manager", "m", "os", "The package manager you want to remove a package with. (Options: os|homebrew|pipx|flatpak|cargo)")
|
|
|
|
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)")
|
|
|
|
checkCmd.Flags().StringP("manager", "m", "os", "The package manager you want to check for a packages with. (Options: os|homebrew|pipx|flatpak)")
|
|
|
|
packageCmd.AddCommand(
|
|
addCmd,
|
|
deleteCmd,
|
|
showCmd,
|
|
enableCmd,
|
|
)
|
|
RootCmd.AddCommand(
|
|
packageCmd,
|
|
searchCmd,
|
|
installCmd,
|
|
removeCmd,
|
|
updateCmd,
|
|
updateAllCmd,
|
|
historyCmd,
|
|
checkCmd,
|
|
)
|
|
}
|
|
|
|
func initConfig() {
|
|
if cfgFile := viper.GetString("config"); cfgFile != "" {
|
|
viper.SetConfigFile(cfgFile)
|
|
} else {
|
|
viper.SetConfigName("config")
|
|
viper.SetConfigType("toml")
|
|
dirname, err := os.UserConfigDir()
|
|
if err != nil {
|
|
log.Println("cant obtain config dir")
|
|
} else {
|
|
viper.AddConfigPath(filepath.Join(dirname, "sst"))
|
|
}
|
|
}
|
|
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
if _, ok := err.(viper.ConfigFileNotFoundError); ok {
|
|
fmt.Println("no config file found, creating a new one...")
|
|
if err := createDefaultConfig(); err != nil {
|
|
log.Printf("error creating default config: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
} else {
|
|
log.Printf("error reading config file: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
}
|
|
|
|
func createDefaultConfig() error {
|
|
v := viper.New()
|
|
|
|
v.SetDefault("config", "")
|
|
v.SetDefault("headless", true)
|
|
|
|
v.SetDefault("dotfiles.enable", false)
|
|
v.SetDefault("dotfiles.git_repo", "")
|
|
|
|
v.SetDefault("package_managers.cargo.enable", false)
|
|
v.SetDefault("package_managers.cargo.packages", []string{})
|
|
|
|
v.SetDefault("package_managers.flatpak.enable", false)
|
|
v.SetDefault("package_managers.flatpak.packages", []string{})
|
|
|
|
v.SetDefault("package_managers.go.enable", false)
|
|
v.SetDefault("package_managers.go.packages", []string{})
|
|
|
|
v.SetDefault("package_managers.homebrew.enable", false)
|
|
v.SetDefault("package_managers.homebrew.packages", []string{})
|
|
|
|
v.SetDefault("package_managers.pipx.enable", false)
|
|
v.SetDefault("package_managers.pipx.packages", []string{})
|
|
|
|
v.SetDefault("package_managers.os.enable", true)
|
|
v.SetDefault("package_managers.os.packages", []string{})
|
|
|
|
configDir := filepath.Join(os.Getenv("HOME"), ".config", "sst")
|
|
if err := os.MkdirAll(configDir, 0755); err != nil {
|
|
return fmt.Errorf("error creating config dir: %w", err)
|
|
}
|
|
|
|
configPath := filepath.Join(configDir, "config.toml")
|
|
v.SetConfigFile(configPath)
|
|
v.SetConfigType("toml")
|
|
|
|
if err := v.WriteConfig(); err != nil {
|
|
return fmt.Errorf("error writing default config: %w", err)
|
|
}
|
|
|
|
fmt.Println("default config created:", configPath)
|
|
return nil
|
|
}
|