package main import ( "context" "fmt" "isle/daemon" "isle/daemon/jsonrpc2" "os" "strings" "dev.mediocregopher.com/mediocre-go-lib.git/mlog" "github.com/spf13/pflag" ) type flagSet struct { *pflag.FlagSet } func (fs flagSet) Parse(args []string) error { fs.VisitAll(func(f *pflag.Flag) { if f.Shorthand == "h" { panic(fmt.Sprintf("flag %+v has reserved shorthand `-h`", f)) } if f.Name == "help" { panic(fmt.Sprintf("flag %+v has reserved name `--help`", f)) } }) return fs.FlagSet.Parse(args) } // subCmdCtx contains all information available to a subCmd's do method. type subCmdCtx struct { context.Context subCmd subCmd // the subCmd itself args []string // command-line arguments, excluding the subCmd itself. subCmdNames []string // names of subCmds so far, including this one logger *mlog.Logger daemonRPC daemon.RPC } type subCmd struct { name string descr string do func(subCmdCtx) error // If set then the name will be allowed to be suffixed with this string. plural string } func (ctx subCmdCtx) usagePrefix() string { subCmdNamesStr := strings.Join(ctx.subCmdNames, " ") if subCmdNamesStr != "" { subCmdNamesStr += " " } return fmt.Sprintf("\nUSAGE: %s %s", os.Args[0], subCmdNamesStr) } func (ctx subCmdCtx) flagSet(withPassthrough bool) flagSet { flags := pflag.NewFlagSet(ctx.subCmd.name, pflag.ExitOnError) flags.Usage = func() { var passthroughStr string if withPassthrough { passthroughStr = " [--] [args...]" } fmt.Fprintf( os.Stderr, "%s[-h|--help] [%s flags...]%s\n\n", ctx.usagePrefix(), ctx.subCmd.name, passthroughStr, ) fmt.Fprintf(os.Stderr, "%s FLAGS:\n\n", strings.ToUpper(ctx.subCmd.name)) fmt.Fprintln(os.Stderr, flags.FlagUsages()) os.Stderr.Sync() os.Exit(2) } return flagSet{flags} } func (ctx subCmdCtx) doSubCmd(subCmds ...subCmd) error { printUsageExit := func(subCmdName string) { fmt.Fprintf(os.Stderr, "unknown sub-command %q\n", subCmdName) fmt.Fprintf( os.Stderr, "%s [-h|--help] [sub-command flags...]\n", ctx.usagePrefix(), ) fmt.Fprintf(os.Stderr, "\nSUB-COMMANDS:\n\n") for _, subCmd := range subCmds { name := subCmd.name if subCmd.plural != "" { name += "(" + subCmd.plural + ")" } fmt.Fprintf(os.Stderr, " %s\t%s\n", name, subCmd.descr) } fmt.Fprintf(os.Stderr, "\n") os.Stderr.Sync() os.Exit(2) } args := ctx.args if len(args) == 0 { printUsageExit("") } subCmdsMap := map[string]subCmd{} for _, subCmd := range subCmds { subCmdsMap[subCmd.name] = subCmd if subCmd.plural != "" { subCmdsMap[subCmd.name+subCmd.plural] = subCmd } } subCmdName, args := args[0], args[1:] subCmd, ok := subCmdsMap[subCmdName] if !ok { printUsageExit(subCmdName) } daemonRPC := daemon.RPCFromClient( jsonrpc2.NewUnixHTTPClient(daemon.HTTPSocketPath(), daemonHTTPRPCPath), ) err := subCmd.do(subCmdCtx{ Context: ctx.Context, subCmd: subCmd, args: args, subCmdNames: append(ctx.subCmdNames, subCmdName), logger: ctx.logger, daemonRPC: daemonRPC, }) if err != nil { return err } return nil }