Skip to content

Commit

Permalink
Use flag.CommandLine for flags.
Browse files Browse the repository at this point in the history
The custom flagset adds unnecessary complexity and doesn't mesh with the
simpler `flag.Type` functions.
  • Loading branch information
mcsaucy committed Jan 31, 2025
1 parent 620f75e commit 7db985f
Showing 1 changed file with 15 additions and 11 deletions.
26 changes: 15 additions & 11 deletions googet.go
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,9 @@ var (
proxyServer string
allowUnsafeURL bool
lockFile string

// Optional function to handle flag parsing. If unset, we use flag.Parse.
flagParse func()
)

type packageMap map[string]string
Expand Down Expand Up @@ -471,23 +474,24 @@ func obtainLock(lockFile string) error {
}

func main() {
ggFlags := flag.NewFlagSet(filepath.Base(os.Args[0]), flag.ContinueOnError)
ggFlags.StringVar(&rootDir, "root", os.Getenv(envVar), "googet root directory")
ggFlags.BoolVar(&noConfirm, "noconfirm", false, "skip confirmation")
ggFlags.BoolVar(&verbose, "verbose", false, "print info level logs to stdout")
ggFlags.BoolVar(&systemLog, "system_log", true, "log to Linux Syslog or Windows Event Log")
ggFlags.BoolVar(&showVer, "version", false, "display GooGet version and exit")

if err := ggFlags.Parse(os.Args[1:]); err != nil && err != flag.ErrHelp {
logger.Fatal(err)
flag.StringVar(&rootDir, "root", os.Getenv(envVar), "googet root directory")
flag.BoolVar(&noConfirm, "noconfirm", false, "skip confirmation")
flag.BoolVar(&verbose, "verbose", false, "print info level logs to stdout")
flag.BoolVar(&systemLog, "system_log", true, "log to Linux Syslog or Windows Event Log")
flag.BoolVar(&showVer, "version", false, "display GooGet version and exit")

if flagParse != nil {
flagParse()
} else {
flag.Parse()
}

if showVer {
fmt.Println("GooGet version:", version)
os.Exit(0)
}

cmdr := subcommands.NewCommander(ggFlags, "googet")
cmdr := subcommands.NewCommander(flag.CommandLine, "googet")
cmdr.Register(cmdr.FlagsCommand(), "")
cmdr.Register(cmdr.CommandsCommand(), "")
cmdr.Register(cmdr.HelpCommand(), "")
Expand All @@ -508,7 +512,7 @@ func main() {
cmdr.ImportantFlag("noconfirm")

nonLockingCommands := []string{"help", "commands", "flags"}
if ggFlags.NArg() == 0 || goolib.ContainsString(ggFlags.Args()[0], nonLockingCommands) {
if flag.NArg() == 0 || goolib.ContainsString(flag.Args()[0], nonLockingCommands) {
os.Exit(int(cmdr.Execute(context.Background())))
}

Expand Down

0 comments on commit 7db985f

Please sign in to comment.