optimized

action/action.go

@@ -10,6 +10,7 @@ import (
 
 func Generator(ctx *cli.Context) error {
 	pkg := ctx.String("package")
+	fileName := ctx.String("filename")
 	std, err := ioutil.ReadAll(os.Stdin)
 	if err != nil {
 		return err
@@ -17,7 +18,11 @@ func Generator(ctx *cli.Context) error {
 
 	var plugin generate.Plugin
 	plugin.ParentPackage = pkg
+	plugin.FileName = fileName
 
+	if len(plugin.FileName) == 0 {
+		plugin.FileName = "rest.swagger.json"
+	}
 	err = json.Unmarshal(std, &plugin)
 	if err != nil {
 		return err