diff --git a/main.go b/main.go index 4d10cf8..e8a95ab 100644 --- a/main.go +++ b/main.go @@ -25,6 +25,7 @@ func main() { cfgPath := flag.String("c", "lolly.yaml", "配置文件路径") cfgPathLong := flag.String("config", "", "配置文件路径(长参数)") genConfig := flag.Bool("generate-config", false, "生成默认配置") + genConfigShort := flag.Bool("g", false, "生成默认配置(短参数)") outputPath := flag.String("o", "", "输出文件路径(配合 --generate-config)") showVersion := flag.Bool("v", false, "显示版本") @@ -35,6 +36,7 @@ func main() { if *cfgPathLong != "" { configPath = *cfgPathLong } + generate := *genConfig || *genConfigShort - os.Exit(app.Run(configPath, *genConfig, *outputPath, *showVersion)) + os.Exit(app.Run(configPath, generate, *outputPath, *showVersion)) }