Advertisement
Guest User

Untitled

a guest
Mar 24th, 2019
79
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.99 KB | None | 0 0
  1. func init() {
  2. rootCmd.PersistentFlags().StringP(config.VariableXFlag, "x", config.VariableXDefault, "Variable X")
  3. rootCmd.PersistentFlags().StringP(config.VariableYFlag, "y", config.VariableYDefault, "Variable Y")
  4.  
  5. rootCmd.AddCommand(&cobra.Command{
  6. Short: "Print injected variables",
  7. Use: "print",
  8. Run: func(cmd *cobra.Command, args []string) {
  9. fmt.Printf("VariableX: %s\n", viper.GetString(config.VariableXFlag))
  10. fmt.Printf("VariableY: %s\n", viper.GetString(config.VariableYFlag))
  11. },
  12. })
  13.  
  14. rootCmd.AddCommand(&cobra.Command{
  15. Use: "version",
  16. Short: "Print version",
  17. Run: func(cmd *cobra.Command, args []string) {
  18. fmt.Println(version)
  19. },
  20. })
  21.  
  22. viper.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
  23. viper.AutomaticEnv() // reads environment variables
  24. viper.BindPFlag(config.VariableXFlag, rootCmd.PersistentFlags().Lookup(config.VariableXFlag))
  25. viper.BindPFlag(config.VariableYFlag, rootCmd.PersistentFlags().Lookup(config.VariableYFlag))
  26. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement