gohub/app/cmd/cmd.go

package cmd

import (
	"os"

	"github.com/runningwater/gohub/pkg/helpers"
	"github.com/spf13/cobra"
)

// Env 全局选项 --env 配置的值
var Env string

// RegisterGlobalFlags 注册全局选项(flag)
func RegisterGlobalFlags(rootCmd *cobra.Command) {
	rootCmd.PersistentFlags().StringVarP(&Env, "env", "e", "", "加载.env文件，如 --env=local, 加载的是.env.local 文件")
}

// RegisterDefaultCmd 注册默认命令
func RegisterDefaultCmd(rootCmd *cobra.Command, subCmd *cobra.Command) {
	cmd, _, err := rootCmd.Find(os.Args[1:])
	firstArg := helpers.FirstElement(os.Args[1:])

	if err == nil && cmd.Use == rootCmd.Use && firstArg != "-h" && firstArg != "--help" {
		args := append([]string{subCmd.Use}, os.Args[1:]...)
		rootCmd.SetArgs(args)
	}
}