Return optional ext arg for create command

chapsuk · May 3, 2018 · 6774054 · 6774054
1 parent 2939c97
commit 6774054
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 12 deletions.
diff --git a/commands/create.go b/commands/create.go
@@ -14,13 +14,13 @@ func Create(ctx *cli.Context, d driver.Interface) error {
 		return errors.New("NAME required")
 	}
 
-	// ext := ctx.Args().Get(1)
-	// switch ext {
-	// case "sql":
-	// case "go":
-	// default:
-	// 	ext = "sql"
-	// }
+	ext := ctx.Args().Get(1)
+	switch ext {
+	case "sql":
+	case "go":
+	default:
+		ext = "sql"
+	}
 
-	return d.Create(name, "sql")
+	return d.Create(name, ext)
 }
diff --git a/commands/migrate/command.go b/commands/migrate/command.go
@@ -29,8 +29,8 @@ func Command() *cli.Command {
 			},
 			&cli.Command{
 				Name:      "create",
-				Usage:     "Creates new migration sql file",
-				ArgsUsage: "NAME",
+				Usage:     "Creates new migration sql|go file",
+				ArgsUsage: "NAME EXT",
 				Action: func(ctx *cli.Context) error {
 					return commands.Create(ctx, migrator)
 				},

diff --git a/commands/seed/command.go b/commands/seed/command.go
@@ -29,8 +29,8 @@ func Command() *cli.Command {
 			},
 			&cli.Command{
 				Name:      "create",
-				Usage:     "Creates new seed sql file with next version",
-				ArgsUsage: "NAME",
+				Usage:     "Creates new seed sql|go file with next version",
+				ArgsUsage: "NAME EXT",
 				Action: func(ctx *cli.Context) error {
 					return commands.Create(ctx, seeder)
 				},