151 lines
3.1 KiB
Go
151 lines
3.1 KiB
Go
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<subCmd> [-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
|
|
}
|