Switch to cobra/viper

cmd/refresh.go

@@ -0,0 +1,54 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/comp500/packwiz/core"
+	"github.com/spf13/cobra"
+)
+
+// refreshCmd represents the refresh command
+var refreshCmd = &cobra.Command{
+	Use:   "refresh",
+	Short: "Refresh the index file",
+	Args:  cobra.NoArgs,
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Println("Loading modpack...")
+		pack, err := core.LoadPack()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		index, err := pack.LoadIndex()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		err = index.Refresh()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		err = index.Write()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		err = pack.UpdateIndexHash()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		err = pack.Write()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		fmt.Println("Index refreshed!")
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(refreshCmd)
+}

cmd/remove.go

@@ -0,0 +1,71 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/comp500/packwiz/core"
+	"github.com/spf13/cobra"
+)
+
+// removeCmd represents the remove command
+var removeCmd = &cobra.Command{
+	Use:     "remove",
+	Short:   "Remove a mod from the modpack",
+	Aliases: []string{"delete", "uninstall"},
+	Args:    cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		if len(args[0]) == 0 {
+			fmt.Println("You must specify a mod.")
+			os.Exit(1)
+		}
+		fmt.Println("Loading modpack...")
+		pack, err := core.LoadPack()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		index, err := pack.LoadIndex()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		resolvedMod, ok := index.FindMod(args[0])
+		if !ok {
+			fmt.Println("You don't have this mod installed.")
+			os.Exit(1)
+		}
+		err = os.Remove(resolvedMod)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		fmt.Println("Removing mod from index...")
+		err = index.RemoveFile(resolvedMod)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		err = index.Write()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		err = pack.UpdateIndexHash()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		err = pack.Write()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+
+		fmt.Printf("Mod %s removed successfully!", args[0])
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(removeCmd)
+}

cmd/root.go

@@ -0,0 +1,72 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+
+	homedir "github.com/mitchellh/go-homedir"
+	"github.com/spf13/viper"
+)
+
+var packFile string
+var modsFolder string
+var cfgFile string
+
+// rootCmd represents the base command when called without any subcommands
+var rootCmd = &cobra.Command{
+	Use:   "packwiz",
+	Short: "A command line tool for creating Minecraft modpacks",
+}
+
+// Execute starts the root command for packwiz
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+}
+
+// Add adds a new command as a subcommand to packwiz
+func Add(newCommand *cobra.Command) {
+	rootCmd.AddCommand(newCommand)
+}
+
+func init() {
+	cobra.OnInitialize(initConfig)
+
+	rootCmd.PersistentFlags().StringVar(&packFile, "pack-file", "pack.toml", "The modpack metadata file to use")
+	viper.BindPFlag("pack-file", rootCmd.PersistentFlags().Lookup("pack-file"))
+
+	rootCmd.PersistentFlags().StringVar(&modsFolder, "mods-folder", "mods", "The default folder to store mod metadata files in")
+	viper.BindPFlag("mods-folder", rootCmd.PersistentFlags().Lookup("mods-folder"))
+
+	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.packwiz.toml)")
+}
+
+// initConfig reads in config file and ENV variables if set.
+func initConfig() {
+	if cfgFile != "" {
+		// Use config file from the flag.
+		viper.SetConfigFile(cfgFile)
+	} else {
+		// Find home directory.
+		home, err := homedir.Dir()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+
+		// Search config in home directory with name ".packwiz" (without extension).
+		viper.AddConfigPath(home)
+		viper.SetConfigName(".packwiz")
+	}
+
+	viper.AutomaticEnv() // read in environment variables that match
+
+	// If a config file is found, read it in.
+	if err := viper.ReadInConfig(); err == nil {
+		fmt.Println("Using config file:", viper.ConfigFileUsed())
+	}
+}

cmd/update.go

@@ -0,0 +1,228 @@
+package cmd
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"strings"
+
+	"github.com/comp500/packwiz/core"
+	"github.com/spf13/cobra"
+)
+
+// updateCmd represents the update command
+var updateCmd = &cobra.Command{
+	Use:     "update",
+	Short:   "Update a mod (or all mods) in the modpack",
+	Aliases: []string{"upgrade"},
+	Args:    cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		// TODO: --check flag?
+		// TODO: specify multiple mods to update at once?
+
+		fmt.Println("Loading modpack...")
+		pack, err := core.LoadPack()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		index, err := pack.LoadIndex()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		mcVersion, err := pack.GetMCVersion()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+
+		multiple := false
+		var singleUpdatedName string
+		if len(args[0]) == 0 || args[0] == "*" {
+			multiple = true
+
+			updaterMap := make(map[string][]core.Mod)
+			fmt.Println("Reading mod files...")
+			for _, v := range index.GetAllMods() {
+				modData, err := core.LoadMod(v)
+				if err != nil {
+					fmt.Printf("Error reading mod file: %s", err.Error())
+					continue
+				}
+
+				updaterFound := false
+				for k := range modData.Update {
+					slice, ok := updaterMap[k]
+					if !ok {
+						_, ok = core.Updaters[k]
+						if !ok {
+							continue
+						}
+						slice = []core.Mod{}
+					}
+					updaterFound = true
+					updaterMap[k] = append(slice, modData)
+				}
+				if !updaterFound {
+					fmt.Printf("A supported update system for \"%s\" cannot be found.", modData.Name)
+				}
+			}
+
+			fmt.Println("Checking for updates...")
+			updatesFound := false
+			updaterPointerMap := make(map[string][]*core.Mod)
+			updaterCachedStateMap := make(map[string][]interface{})
+			for k, v := range updaterMap {
+				checks, err := core.Updaters[k].CheckUpdate(v, mcVersion)
+				if err != nil {
+					// TODO: do we return err code 1?
+					fmt.Println(err.Error())
+					continue
+				}
+				for i, check := range checks {
+					if check.Error != nil {
+						// TODO: do we return err code 1?
+						// TODO: better error message?
+						fmt.Println(check.Error.Error())
+						continue
+					}
+					if check.UpdateAvailable {
+						if !updatesFound {
+							fmt.Println("Updates found:")
+							updatesFound = true
+						}
+						fmt.Printf("%s: %s\n", v[i].Name, check.UpdateString)
+						updaterPointerMap[k] = append(updaterPointerMap[k], &v[i])
+						updaterCachedStateMap[k] = append(updaterCachedStateMap[k], check.CachedState)
+					}
+				}
+			}
+
+			if !updatesFound {
+				fmt.Println("All mods are up to date!")
+				return
+			}
+
+			fmt.Print("Do you want to update? [Y/n]: ")
+			answer, err := bufio.NewReader(os.Stdin).ReadString('\n')
+			if err != nil {
+				fmt.Println(err)
+				os.Exit(1)
+			}
+
+			ansNormal := strings.ToLower(strings.TrimSpace(answer))
+			if len(ansNormal) > 0 && ansNormal[0] == 'n' {
+				fmt.Println("Cancelled!")
+				return
+			}
+
+			for k, v := range updaterPointerMap {
+				err := core.Updaters[k].DoUpdate(v, updaterCachedStateMap[k])
+				if err != nil {
+					// TODO: do we return err code 1?
+					fmt.Println(err.Error())
+					continue
+				}
+				for _, modData := range v {
+					format, hash, err := modData.Write()
+					if err != nil {
+						fmt.Println(err.Error())
+						continue
+					}
+					err = index.RefreshFileWithHash(modData.GetFilePath(), format, hash, true)
+					if err != nil {
+						fmt.Println(err.Error())
+						continue
+					}
+				}
+			}
+		} else {
+			modPath, ok := index.FindMod(args[0])
+			if !ok {
+				fmt.Println("You don't have this mod installed.")
+				os.Exit(1)
+			}
+			modData, err := core.LoadMod(modPath)
+			if err != nil {
+				fmt.Println(err)
+				os.Exit(1)
+			}
+			singleUpdatedName = modData.Name
+			updaterFound := false
+			for k := range modData.Update {
+				updater, ok := core.Updaters[k]
+				if !ok {
+					continue
+				}
+				updaterFound = true
+
+				check, err := updater.CheckUpdate([]core.Mod{modData}, mcVersion)
+				if err != nil {
+					fmt.Println(err)
+					os.Exit(1)
+				}
+				if len(check) != 1 {
+					fmt.Println("Invalid update check response")
+					os.Exit(1)
+				}
+
+				if check[0].UpdateAvailable {
+					fmt.Printf("Update available: %s\n", check[0].UpdateString)
+
+					err = updater.DoUpdate([]*core.Mod{&modData}, []interface{}{check[0].CachedState})
+					if err != nil {
+						fmt.Println(err)
+						os.Exit(1)
+					}
+
+					format, hash, err := modData.Write()
+					if err != nil {
+						fmt.Println(err)
+						os.Exit(1)
+					}
+					err = index.RefreshFileWithHash(modPath, format, hash, true)
+					if err != nil {
+						fmt.Println(err)
+						os.Exit(1)
+					}
+				} else {
+					fmt.Printf("\"%s\" is already up to date!\n", modData.Name)
+					return
+				}
+
+				break
+			}
+			if !updaterFound {
+				// TODO: use file name instead of Name when len(Name) == 0 in all places?
+				fmt.Println("A supported update system for \"" + modData.Name + "\" cannot be found.")
+				os.Exit(1)
+			}
+		}
+
+		err = index.Write()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		err = pack.UpdateIndexHash()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		err = pack.Write()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		if multiple {
+			fmt.Println("Mods updated!")
+		} else {
+			fmt.Printf("\"%s\" updated!\n", singleUpdatedName)
+		}
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(updateCmd)
+}