feat: 增加命令行工具的参数支持,包括路径、干运行模式和输出目录选项
This commit is contained in:
@@ -6,20 +6,25 @@ import (
|
||||
)
|
||||
|
||||
func CommandSwagFmt(root *cobra.Command) {
|
||||
cmd := &cobra.Command{
|
||||
Use: "fmt",
|
||||
Aliases: []string{"f"},
|
||||
Short: "swag format",
|
||||
RunE: commandSwagFmtE,
|
||||
}
|
||||
cmd := &cobra.Command{
|
||||
Use: "fmt",
|
||||
Aliases: []string{"f"},
|
||||
Short: "swag format",
|
||||
RunE: commandSwagFmtE,
|
||||
}
|
||||
|
||||
cmd.Flags().String("dir", "./app/http", "SearchDir for swag format")
|
||||
cmd.Flags().String("main", "main.go", "MainFile for swag format")
|
||||
|
||||
root.AddCommand(cmd)
|
||||
}
|
||||
|
||||
func commandSwagFmtE(cmd *cobra.Command, args []string) error {
|
||||
return format.New().Build(&format.Config{
|
||||
SearchDir: "./app/http",
|
||||
Excludes: "",
|
||||
MainFile: "main.go",
|
||||
})
|
||||
dir := cmd.Flag("dir").Value.String()
|
||||
main := cmd.Flag("main").Value.String()
|
||||
return format.New().Build(&format.Config{
|
||||
SearchDir: dir,
|
||||
Excludes: "",
|
||||
MainFile: main,
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user