// Copyright 2016-2017 Amazon.com, Inc. or its affiliates. All Rights Reserved. // // Licensed under the Apache License, Version 2.0 (the "License"). You may // not use this file except in compliance with the License. A copy of the // License is located at // // http://aws.amazon.com/apache2.0/ // // or in the "license" file accompanying this file. This file is distributed // on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either // express or implied. See the License for the specific language governing // permissions and limitations under the License. package cmd import ( "github.com/spf13/cobra" "github.com/spf13/viper" ) // ClusterStateServiceEndpoint represents the css endpoint to connect to. var ClusterStateServiceEndpoint string // RootCmd represents the base command when called without any subcommands var RootCmd = createRootCommand() // Init the CLI. func init() { cobra.OnInitialize(initConfig) } // initConfig reads in config file and ENV variables if set. func initConfig() { viper.AutomaticEnv() // read in environment variables that match } func createRootCommand() *cobra.Command { rootCmd := &cobra.Command{ Use: "cluster-state-service-canary", Short: "cluster-state-service-canary", Run: func(cmd *cobra.Command, args []string) { }, } rootCmd.PersistentFlags().StringVar(&ClusterStateServiceEndpoint, "css-endpoint", "", "Cluster state service address") return rootCmd }