[ADD] create new repeatable migration

Author: JohnBra

cmd/migration.go

@@ -39,12 +39,14 @@ var (
 		},
 	}
 
+	repeatable bool
+
 	migrationNewCmd = &cobra.Command{
 		Use:   "new <migration name>",
 		Short: "Create an empty migration script",
 		Args:  cobra.ExactArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			return new.Run(args[0], os.Stdin, afero.NewOsFs())
+			return new.Run(repeatable, args[0], os.Stdin, afero.NewOsFs())
 		},
 	}
 
@@ -149,6 +151,8 @@ func init() {
 	migrationFetchCmd.MarkFlagsMutuallyExclusive("db-url", "linked", "local")
 	migrationCmd.AddCommand(migrationFetchCmd)
 	// Build new command
+	newFlags := migrationNewCmd.Flags()
+	newFlags.BoolVarP(&repeatable, "repeatable", "r", false, "Creates a repeatable migration instead of a common migration.")
 	migrationCmd.AddCommand(migrationNewCmd)
 	rootCmd.AddCommand(migrationCmd)
 }

internal/migration/new/new.go

@@ -11,8 +11,16 @@ import (
 	"github.com/supabase/cli/internal/utils"
 )
 
-func Run(migrationName string, stdin afero.File, fsys afero.Fs) error {
-	path := GetMigrationPath(utils.GetCurrentTimestamp(), migrationName)
+func Run(repeatable bool, migrationName string, stdin afero.File, fsys afero.Fs) error {
+	path := ""
+
+	if repeatable {
+		// if migration name already exists, repeatable migration will be overwritten
+		path = GetRepeatableMigrationPath(migrationName)
+	} else {
+		path = GetMigrationPath(utils.GetCurrentTimestamp(), migrationName)
+	}
+
 	if err := utils.MkdirIfNotExistFS(fsys, filepath.Dir(path)); err != nil {
 		return err
 	}
@@ -33,6 +41,11 @@ func GetMigrationPath(timestamp, name string) string {
 	return filepath.Join(utils.MigrationsDir, fullName)
 }
 
+func GetRepeatableMigrationPath(name string) string {
+	fullName := fmt.Sprintf("r_%s.sql", name)
+	return filepath.Join(utils.MigrationsDir, fullName)
+}
+
 func CopyStdinIfExists(stdin afero.File, dst io.Writer) error {
 	if fi, err := stdin.Stat(); err != nil {
 		return errors.Errorf("failed to initialise stdin: %w", err)