@@ -53,6 +53,10 @@ var CmdGenerate = &commands.Command{
5353
5454 $ bee generate docs
5555
56+ ▶ {{"To generate swagger doc file:"|bold}}
57+
58+ $ bee generate routers [-ctrlDir=/path/to/controller/directory] [-routersFile=/path/to/routers/file.go] [-routersPkg=myPackage]
59+
5660 ▶ {{"To generate a test case:"|bold}}
5761
5862 $ bee generate test [routerfile]
@@ -72,6 +76,15 @@ func init() {
7276 CmdGenerate .Flag .Var (& generate .Level , "level" , "Either 1, 2 or 3. i.e. 1=models; 2=models and controllers; 3=models, controllers and routers." )
7377 CmdGenerate .Flag .Var (& generate .Fields , "fields" , "List of table Fields." )
7478 CmdGenerate .Flag .Var (& generate .DDL , "ddl" , "Generate DDL Migration" )
79+
80+ // bee generate routers
81+ CmdGenerate .Flag .Var (& generate .ControllerDirectory , "ctrlDir" ,
82+ "Controller directory. Bee scans this directory and its sub directory to generate routers" )
83+ CmdGenerate .Flag .Var (& generate .RoutersFile , "routersFile" ,
84+ "Routers file. If not found, Bee create a new one. Bee will truncates this file and output routers info into this file" )
85+ CmdGenerate .Flag .Var (& generate .RouterPkg , "routersPkg" ,
86+ `router's package. Default is routers, it means that "package routers" in the generated file` )
87+
7588 commands .AvailableCommands = append (commands .AvailableCommands , CmdGenerate )
7689}
7790
@@ -97,13 +110,25 @@ func GenerateCode(cmd *commands.Command, args []string) int {
97110 model (cmd , args , currpath )
98111 case "view" :
99112 view (args , currpath )
113+ case "routers" :
114+ genRouters (cmd , args )
100115 default :
101116 beeLogger .Log .Fatal ("Command is missing" )
102117 }
103118 beeLogger .Log .Successf ("%s successfully generated!" , strings .Title (gcmd ))
104119 return 0
105120}
106121
122+ func genRouters (cmd * commands.Command , args []string ) {
123+ err := cmd .Flag .Parse (args [1 :])
124+ beeLogger .Log .Infof ("input parameter: %v" , args )
125+ if err != nil {
126+ beeLogger .Log .Errorf ("could not parse input parameter: %+v" , err )
127+ return
128+ }
129+ generate .GenRouters ()
130+ }
131+
107132func scaffold (cmd * commands.Command , args []string , currpath string ) {
108133 if len (args ) < 2 {
109134 beeLogger .Log .Fatal ("Wrong number of arguments. Run: bee help generate" )
0 commit comments