| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License 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 ( |
| "fmt" |
| "os" |
| ) |
| |
| import ( |
| "github.com/spf13/cobra" |
| |
| "github.com/spf13/viper" |
| ) |
| |
| var cfgFile string |
| |
| // rootCmd represents the base command when called without any subcommands |
| var rootCmd = &cobra.Command{ |
| Use: "dubbogo-cli", |
| Short: "", |
| Long: ``, |
| Run: func(cmd *cobra.Command, args []string) { |
| fmt.Println("hello") |
| }, |
| } |
| |
| // Execute adds all child commands to the root command and sets flags appropriately. |
| // This is called by main.main(). It only needs to happen once to the rootCmd. |
| func Execute() { |
| cobra.CheckErr(rootCmd.Execute()) |
| } |
| |
| func init() { |
| cobra.OnInitialize(initConfig) |
| rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.dubbogo-cli-v2.yaml)") |
| rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle") |
| } |
| |
| // initConfig reads in config file and ENV variables if set. |
| func initConfig() { |
| if cfgFile != "" { |
| // Use config file from the flag. |
| viper.SetConfigFile(cfgFile) |
| } else { |
| home, err := os.UserHomeDir() |
| cobra.CheckErr(err) |
| |
| viper.AddConfigPath(home) |
| viper.SetConfigType("yaml") |
| viper.SetConfigName(".dubbogo-cli-v2") |
| } |
| |
| viper.AutomaticEnv() // read in environment variables that match |
| |
| // If a config file is found, read it in. |
| if err := viper.ReadInConfig(); err == nil { |
| fmt.Fprintln(os.Stderr, "Using config file:", viper.ConfigFileUsed()) |
| } |
| } |