func main() { // parse global arguments globalFlagset.Parse(os.Args[1:]) var args = globalFlagset.Args() getFlagsFromEnv(cliName, globalFlagset) if globalFlags.Debug { log.SetVerbosity(1) } // no command specified - trigger help if len(args) < 1 { args = append(args, "help") } // deal specially with --version if globalFlags.Version { args[0] = "version" } var cmd *Command // determine which Command should be run for _, c := range commands { if c.Name == args[0] { cmd = c if err := c.Flags.Parse(args[1:]); err != nil { fmt.Println(err.Error()) os.Exit(2) } break } } if cmd == nil { fmt.Printf("%v: unknown subcommand: %q\n", cliName, args[0]) fmt.Printf("Run '%v help' for usage.\n", cliName) os.Exit(2) } if sharedFlags.Sign { fmt.Fprintln(os.Stderr, "WARNING: The signed/verified units feature is DEPRECATED and cannot be used.") os.Exit(2) } if cmd.Name != "help" && cmd.Name != "version" { var err error cAPI, err = getClient() if err != nil { msg := fmt.Sprintf("Unable to initialize client: %v\n", err) fmt.Fprint(os.Stderr, msg) os.Exit(1) } } os.Exit(cmd.Run(cmd.Flags.Args())) }
func getConfig(flagset *flag.FlagSet, userCfgFile string) (*config.Config, error) { opts := globalconf.Options{EnvPrefix: "FLEET_"} if userCfgFile != "" { // Fail hard if a user-provided config is not usable fi, err := os.Stat(userCfgFile) if err != nil { log.Fatalf("Unable to use config file %s: %v", userCfgFile, err) } if fi.IsDir() { log.Fatalf("Provided config %s is a directory, not a file", userCfgFile) } log.Infof("Using provided config file %s", userCfgFile) opts.Filename = userCfgFile } else if _, err := os.Stat(DefaultConfigFile); err == nil { log.Infof("Using default config file %s", DefaultConfigFile) opts.Filename = DefaultConfigFile } else { log.Infof("No provided or default config file found - proceeding without") } gconf, err := globalconf.NewWithOptions(&opts) if err != nil { return nil, err } gconf.ParseSet("", flagset) cfg := config.Config{ Verbosity: (*flagset.Lookup("verbosity")).Value.(flag.Getter).Get().(int), EtcdServers: (*flagset.Lookup("etcd_servers")).Value.(flag.Getter).Get().(stringSlice), EtcdKeyPrefix: (*flagset.Lookup("etcd_key_prefix")).Value.(flag.Getter).Get().(string), EtcdKeyFile: (*flagset.Lookup("etcd_keyfile")).Value.(flag.Getter).Get().(string), EtcdCertFile: (*flagset.Lookup("etcd_certfile")).Value.(flag.Getter).Get().(string), EtcdCAFile: (*flagset.Lookup("etcd_cafile")).Value.(flag.Getter).Get().(string), EtcdRequestTimeout: (*flagset.Lookup("etcd_request_timeout")).Value.(flag.Getter).Get().(float64), EngineReconcileInterval: (*flagset.Lookup("engine_reconcile_interval")).Value.(flag.Getter).Get().(float64), PublicIP: (*flagset.Lookup("public_ip")).Value.(flag.Getter).Get().(string), RawMetadata: (*flagset.Lookup("metadata")).Value.(flag.Getter).Get().(string), AgentTTL: (*flagset.Lookup("agent_ttl")).Value.(flag.Getter).Get().(string), VerifyUnits: (*flagset.Lookup("verify_units")).Value.(flag.Getter).Get().(bool), AuthorizedKeysFile: (*flagset.Lookup("authorized_keys_file")).Value.(flag.Getter).Get().(string), } if cfg.VerifyUnits { log.Error("Config option verify_units is no longer supported - ignoring") } if len(cfg.AuthorizedKeysFile) > 0 { log.Error("Config option authorized_keys_file is no longer supported - ignoring") } log.SetVerbosity(cfg.Verbosity) return &cfg, nil }
func main() { // parse global arguments globalFlagset.Parse(os.Args[1:]) var args = globalFlagset.Args() getFlagsFromEnv(cliName, globalFlagset) if globalFlags.Debug { log.SetVerbosity(1) } if globalFlags.Version { args = []string{"version"} } else if len(args) < 1 || globalFlags.Help { args = []string{"help"} } var cmd *Command // determine which Command should be run for _, c := range commands { if c.Name == args[0] { cmd = c if err := c.Flags.Parse(args[1:]); err != nil { stderr("%v", err) os.Exit(2) } break } } if cmd == nil { stderr("%v: unknown subcommand: %q", cliName, args[0]) stderr("Run '%v help' for usage.", cliName) os.Exit(2) } if sharedFlags.Sign { stderr("WARNING: The signed/verified units feature is DEPRECATED and cannot be used.") os.Exit(2) } if cmd.Name != "help" && cmd.Name != "version" { var err error cAPI, err = getClient() if err != nil { stderr("Unable to initialize client: %v", err) os.Exit(1) } } os.Exit(cmd.Run(cmd.Flags.Args())) }