65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/mitchellh/cli"
|
|
"github.com/sean-/seed"
|
|
)
|
|
|
|
func init() {
|
|
seed.Init()
|
|
}
|
|
|
|
func main() {
|
|
os.Exit(Run(os.Args[1:]))
|
|
}
|
|
|
|
func Run(args []string) int {
|
|
return RunCustom(args, Commands(nil))
|
|
}
|
|
|
|
func RunCustom(args []string, commands map[string]cli.CommandFactory) int {
|
|
// Get the command line args. We shortcut "--version" and "-v" to
|
|
// just show the version.
|
|
for _, arg := range args {
|
|
if arg == "-v" || arg == "-version" || arg == "--version" {
|
|
newArgs := make([]string, len(args)+1)
|
|
newArgs[0] = "version"
|
|
copy(newArgs[1:], args)
|
|
args = newArgs
|
|
break
|
|
}
|
|
}
|
|
|
|
// Build the commands to include in the help now.
|
|
commandsInclude := make([]string, 0, len(commands))
|
|
for k, _ := range commands {
|
|
switch k {
|
|
case "executor":
|
|
case "syslog":
|
|
case "operator raft", "operator raft list-peers", "operator raft remove-peer":
|
|
case "job dispatch":
|
|
case "fs ls", "fs cat", "fs stat":
|
|
case "check":
|
|
default:
|
|
commandsInclude = append(commandsInclude, k)
|
|
}
|
|
}
|
|
|
|
cli := &cli.CLI{
|
|
Args: args,
|
|
Commands: commands,
|
|
HelpFunc: cli.FilteredHelpFunc(commandsInclude, cli.BasicHelpFunc("nomad")),
|
|
}
|
|
|
|
exitCode, err := cli.Run()
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error executing CLI: %s\n", err.Error())
|
|
return 1
|
|
}
|
|
|
|
return exitCode
|
|
}
|