技術選択編 - #golang で CLI 作るときにいつもつかうやつ

// cmd/foobar/main.go
//----------------------------------------------------------------

func main() {
    if err := run(); err != nil {
        fmt.Fprintln(os.Stderr, err)
        os.Exit(1)
    }
}

func run() {
    cmd := cmd.NewFoobarCommand()
    return cmd.Execute()
}


// pkg/foobar/cmd/cmd.go
//----------------------------------------------------------------

func NewFoobarCommand() *cobra.Command {
    var (
        flagVerbose bool
    )

    cmd := &cobra.Command{
        Use: "foobar",
    }

    cmd.AddCommand(
        newInitCommand(),
    )

    // `Persistent` ってつくやつはサブコマンドまで影響が及ぶやつ
    cmd.PersistentFlags().BoolVarP(&flagVerbose, "verbose", "v", false, "enable verbose log")

    return cmd
}

// pkg/foobar/cmd/init.go
//----------------------------------------------------------------

func newInitCommand() *cobra.Command {
    cmd := &cobra.Command{
        Use: "init",
        Args: cobra.ExactArgs(1), // 引数のバリデーションもできる
        RunE: func(_ *cobra.Command, args []string) error {
            // snip.
            return nil
        },
    }

    return cmd
}

v := viper.New()

// 環境変数
v.EnvPrefix("app")
v.BindEnv("host") // APP_HOST

// 設定ファイル
v.SetConfigName(".app") // .app.toml, .app.yaml, ...
v.AddConfigPath(".")    // ./.app.toml, ./.app.yaml, ...
err = v.ReadInConfig()  // ファイル読み込み
if err != nil {
    // ...
}

// flag
viper.BindPFlag("author", cmd.Flags().Lookup("author"))

// 環境変数・設定ファイル・flag からいい感じに値を取り出す
v.Get("author")

// struct に unmarshal もできる
type Config struct {
    Author string
}
err = v.Unmarshal(&cfg)

func GenerateFile(fs afero.Fs, params interface{}) error {
    buf := new(bytes.Buffer)
    err := tmpl.Execute(buf, params)
    if err != nil {
        // ...
    }

    err = afero.WriteFile(fs, "awesome.go", buf.Bytes(), 0644)
    if err != nil {
        // ...
    }

    return nil
}

// ふつうにファイルを生成
fs := afero.NewOsFs()
GenerateFile(fs, params)

// テスト用のモック filesystem にファイルを生成
fs := afero.NewMemMapFs()
GenerateFile(fs, params)

exec := exec.New()

cmd := exec.Command("echo", "Hello world!")
buf := new(bytes.Buffer)
cmd.SetStdout(buf)
err := cmd.Run()
if err != nil {
    // ...
}
fmt.Println(buf.String()) // Hello world!