31 lines
730 B
Go
31 lines
730 B
Go
package cmd
|
|
|
|
import (
|
|
"github.com/rogeecn/swag/format"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func CommandSwagFmt(root *cobra.Command) {
|
|
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 {
|
|
dir := cmd.Flag("dir").Value.String()
|
|
main := cmd.Flag("main").Value.String()
|
|
return format.New().Build(&format.Config{
|
|
SearchDir: dir,
|
|
Excludes: "",
|
|
MainFile: main,
|
|
})
|
|
}
|