mirror of
https://github.com/packwiz/packwiz.git
synced 2025-11-18 17:14:32 +01:00
Switch to cobra/viper
This commit is contained in:
comp500
309
main.go
309
main.go
@@ -1,316 +1,11 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"strings"
|
||||
"bufio"
|
||||
|
||||
"github.com/comp500/packwiz/core"
|
||||
"github.com/urfave/cli"
|
||||
|
||||
// Modules of packwiz
|
||||
"github.com/comp500/packwiz/cmd"
|
||||
_ "github.com/comp500/packwiz/curseforge"
|
||||
)
|
||||
|
||||
func init() {
|
||||
core.Commands = append(core.Commands, cli.Command{
|
||||
Name: "remove",
|
||||
Aliases: []string{"delete", "uninstall"},
|
||||
Usage: "Remove a mod from the modpack",
|
||||
Action: func(c *cli.Context) error {
|
||||
return cmdDelete(core.FlagsFromContext(c), c.Args().Get(0))
|
||||
},
|
||||
}, cli.Command{
|
||||
Name: "update",
|
||||
Aliases: []string{"upgrade"},
|
||||
Usage: "Update a mod (or all mods) in the modpack",
|
||||
Action: func(c *cli.Context) error {
|
||||
return cmdUpdate(core.FlagsFromContext(c), c.Args().Get(0))
|
||||
},
|
||||
}, cli.Command{
|
||||
Name: "refresh",
|
||||
Usage: "Refresh the index file",
|
||||
Action: func(c *cli.Context) error {
|
||||
return cmdRefresh(core.FlagsFromContext(c))
|
||||
},
|
||||
})
|
||||
}
|
||||
func main() {
|
||||
app := cli.NewApp()
|
||||
app.Commands = core.Commands
|
||||
app.Flags = core.CLIFlags[:]
|
||||
app.HideVersion = true
|
||||
app.Usage = "A command line tool for creating Minecraft modpacks."
|
||||
|
||||
err := app.Run(os.Args)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func cmdDelete(flags core.Flags, mod string) error {
|
||||
if len(mod) == 0 {
|
||||
return cli.NewExitError("You must specify a mod.", 1)
|
||||
}
|
||||
fmt.Println("Loading modpack...")
|
||||
pack, err := core.LoadPack(flags)
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
index, err := pack.LoadIndex()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
resolvedMod, ok := index.FindMod(mod)
|
||||
if !ok {
|
||||
return cli.NewExitError("You don't have this mod installed.", 1)
|
||||
}
|
||||
err = os.Remove(resolvedMod)
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
fmt.Println("Removing mod from index...")
|
||||
err = index.RemoveFile(resolvedMod)
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = index.Write()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = pack.UpdateIndexHash()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = pack.Write()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
|
||||
fmt.Printf("Mod %s removed successfully!", mod)
|
||||
return nil
|
||||
}
|
||||
|
||||
func cmdRefresh(flags core.Flags) error {
|
||||
fmt.Println("Loading modpack...")
|
||||
pack, err := core.LoadPack(flags)
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
index, err := pack.LoadIndex()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = index.Refresh()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = index.Write()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = pack.UpdateIndexHash()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = pack.Write()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
fmt.Println("Index refreshed!")
|
||||
return nil
|
||||
}
|
||||
|
||||
func cmdUpdate(flags core.Flags, mod string) error {
|
||||
// TODO: --check flag?
|
||||
// TODO: specify multiple mods to update at once?
|
||||
|
||||
fmt.Println("Loading modpack...")
|
||||
pack, err := core.LoadPack(flags)
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
index, err := pack.LoadIndex()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
mcVersion, err := pack.GetMCVersion()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
|
||||
multiple := false
|
||||
var singleUpdatedName string
|
||||
if len(mod) == 0 || mod == "*" {
|
||||
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 nil
|
||||
}
|
||||
|
||||
fmt.Print("Do you want to update? [Y/n]: ")
|
||||
answer, err := bufio.NewReader(os.Stdin).ReadString('\n')
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
|
||||
ansNormal := strings.ToLower(strings.TrimSpace(answer))
|
||||
if len(ansNormal) > 0 && ansNormal[0] == 'n' {
|
||||
fmt.Println("Cancelled!")
|
||||
return nil
|
||||
}
|
||||
|
||||
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(mod)
|
||||
if !ok {
|
||||
return cli.NewExitError("You don't have this mod installed.", 1)
|
||||
}
|
||||
modData, err := core.LoadMod(modPath)
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 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 {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
if len(check) != 1 {
|
||||
return cli.NewExitError("Invalid update check response", 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 {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
|
||||
format, hash, err := modData.Write()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = index.RefreshFileWithHash(modPath, format, hash, true)
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
} else {
|
||||
fmt.Printf("\"%s\" is already up to date!\n", modData.Name)
|
||||
return nil
|
||||
}
|
||||
|
||||
break
|
||||
}
|
||||
if !updaterFound {
|
||||
// TODO: use file name instead of Name when len(Name) == 0 in all places?
|
||||
return cli.NewExitError("A supported update system for \""+modData.Name+"\" cannot be found.", 1)
|
||||
}
|
||||
}
|
||||
|
||||
err = index.Write()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = pack.UpdateIndexHash()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
err = pack.Write()
|
||||
if err != nil {
|
||||
return cli.NewExitError(err, 1)
|
||||
}
|
||||
if multiple {
|
||||
fmt.Println("Mods updated!")
|
||||
} else {
|
||||
fmt.Printf("\"%s\" updated!\n", singleUpdatedName)
|
||||
}
|
||||
return nil
|
||||
cmd.Execute()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user