You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.2 KiB
61 lines
1.2 KiB
package main |
|
|
|
import ( |
|
"flag" |
|
"fmt" |
|
"os" |
|
) |
|
|
|
var ( |
|
err error |
|
_mode, _func string |
|
files []string |
|
parses []*parse |
|
) |
|
|
|
func main() { |
|
flag.StringVar(&_mode, "m", "test", "Generating code by Working mode. [test|interface|mock|upgrade...]") |
|
flag.StringVar(&_func, "func", "", "Generating code by function.") |
|
flag.Parse() |
|
if len(os.Args) == 1 { |
|
println("Creater is a tool for generating code.\n\nUsage: creater [-m]") |
|
flag.PrintDefaults() |
|
return |
|
} |
|
if err = parseArgs(os.Args[1:], &files, 0); err != nil { |
|
panic(err) |
|
} |
|
switch _mode { |
|
case "monkey": |
|
if parses, err = parseFile(files...); err != nil { |
|
panic(err) |
|
} |
|
if err = genMonkey(parses); err != nil { |
|
panic(err) |
|
} |
|
case "test": |
|
if parses, err = parseFile(files...); err != nil { |
|
panic(err) |
|
} |
|
if err = genTest(parses); err != nil { |
|
panic(err) |
|
} |
|
case "interface": |
|
if parses, err = parseFile(files...); err != nil { |
|
panic(err) |
|
} |
|
if err = genInterface(parses); err != nil { |
|
panic(err) |
|
} |
|
case "mock": |
|
if err = genMock(files...); err != nil { |
|
panic(err) |
|
} |
|
case "upgrade": |
|
if err = upBladeMaster(files); err != nil { |
|
panic(err) |
|
} |
|
default: |
|
} |
|
fmt.Println(moha) |
|
}
|
|
|