Проверка флагов с помощью Cobra

Приведенный ниже скетч представляет собой приложение командной строки, написанное с использованием Cobra and Go. Я хотел бы выдать ошибку, если значение flag1 не соответствует регулярному выражению ^\s+\/\s+. Как я могу это сделать?

package cmd

import (
        "fmt"
        "os"
        "github.com/spf13/cobra"

        homedir "github.com/mitchellh/go-homedir"
        "github.com/spf13/viper"
)

var flag1 string
var cfgFile string

// rootCmd represents the base command when called without any subcommands
var rootCmd = &cobra.Command{
        Use:   "cobra-sketch",
        Short: "Sketch for Cobra flags",
  Long: "Sketch for Cobra flags",
        Run: func(cmd *cobra.Command, args []string) { fmt.Printf("Flag1 is %s\n", flag1)},
}

// 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/.cobra-sketch.yaml)")
  rootCmd.PersistentFlags().StringVar(&flag1, "flag1", "", "Value of Flag 1")
}

// initConfig reads in config file and ENV variables if set.
func initConfig() {
        if cfgFile != "" {
                // Use config file from the flag.
                viper.SetConfigFile(cfgFile)
        } else {
                // Find home directory.
                home, err := homedir.Dir()
                cobra.CheckErr(err)

                // Search config in home directory with name ".cobra-sketch" (without extension).
                viper.AddConfigPath(home)
                viper.SetConfigName(".cobra-sketch")
        }

        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())
        }
}

go go-cobra
person Kit Peters    schedule 22.04.2021    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Допустим, пользователь запускает такую ​​команду: cobra-sketch --flag1 "hello". hello будет сохранен в переменной var flag1 string, которую вы назначили флагу, чтобы проверить, соответствует ли ввод какому-либо регулярному выражению, вы можете сделать:

var rootCmd = &cobra.Command{
    Use:   "cobra-sketch",
        ...
    RunE: func(cmd *cobra.Command, args []string) error {
        // You can also use MustCompile if you are sure the regular expression 
        // is valid, it panics instead of returning an error
        re, err := regexp.Compile(`^\s+\/\s+`)
        if err != nil {
            return err // Handle error
        }

        if !regexp.MatchString(flag1) {
            return fmt.Errorf("invalid value: %q", flag1)
        }

        fmt.Printf("Flag1 is %s\n", flag1)
        return nil
    },
}
person Gastón Palomeque    schedule 22.04.2021
comment
Это определенно работает и отвечает на вопрос, который я задал. Однако я задал неправильный вопрос, потому что не смог добавить, что хочу, чтобы проверка применялась ко всем подкомандам. - person Kit Peters; 22.04.2021
comment
Хорошая новость заключается в том, что ваш ответ привел меня к ответу, который я на самом деле искал, а именно к использованию PersistentPreRunE. И вы показали мне материал с регулярными выражениями, который тоже оказался полезным. :) - person Kit Peters; 22.04.2021