From f85a34b0d87d68e1fc2eae93ef230a8a30e59c73 Mon Sep 17 00:00:00 2001
From: Rizky Kurniawan <55426692+kykurniawan@users.noreply.github.com>
Date: Fri, 21 Nov 2025 23:45:51 +0800
Subject: [PATCH] flags for create migration command

---
 cli.go | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/cli.go b/cli.go
index e655e0a..54e2177 100644
--- a/cli.go
+++ b/cli.go
@@ -155,10 +155,11 @@ func (c *Cli) CreateCommand(ctx context.Context) *cobra.Command {
 	var createCmd = &cobra.Command{
 		Use:   "create",
 		Short: "Create a new migration",
-		Args:  cobra.ExactArgs(1),
 		Run: func(cmd *cobra.Command, args []string) {
-			migrationName := args[0]
-			err := c.migration.Create(migrationName)
+			dir, _ := cmd.Flags().GetString("dir")
+			name, _ := cmd.Flags().GetString("name")
+
+			err := c.migration.SetMigrationFilesDir(dir).Create(name)
 			if err != nil {
 				log.Println("Error creating migration:", err)
 				return
@@ -166,6 +167,11 @@ func (c *Cli) CreateCommand(ctx context.Context) *cobra.Command {
 		},
 	}
 
+	createCmd.Flags().StringP("name", "n", "", "name of the migration")
+	createCmd.Flags().StringP("dir", "d", "", "directory of the migration")
+	createCmd.MarkFlagRequired("name")
+	createCmd.MarkFlagRequired("dir")
+
 	return createCmd
 }