migrator

var CleanCmd = &cobra.Command{
	Use:   "clean",
	Short: "remove all unapplied migration files.",
	Args:  cobra.NoArgs,
	Run: func(cmd *cobra.Command, args []string) {

		dryRun := cmd.Flag("dry").Value.String() == "true"

		workDir, dbType, logFile := sqlmigrator.DefaultContextConfig()
		ctx := sqlmigrator.NewContext(workDir, dbType, logFile, dryRun)
		ctx.LoadMigrationEntries()

		log.Println("cleaning unapplied migrations...")

		generator := &sqlmigrator.Generator{
			DryRun: dryRun,
		}

		removedFiles, err := generator.Clean(ctx)
		if err != nil {
			log.Fatal(err)
			return
		}

		displayCleanedFiles(removedFiles)
		err = ctx.WriteMigrationEntries()
		if err != nil {
			log.Println("error writing migration entries:", err)
			return
		}
		log.Println("cleaned migrations successfully.")

	},
}

var DownCmd = &cobra.Command{
	Use:   "down",
	Short: "perform backward migration from the files in the migrations folder",
	Args:  cobra.MaximumNArgs(1),
	Run: func(cmd *cobra.Command, args []string) {
		dryRun := cmd.Flag("dry").Value.String() == "true"
		num := getNumberFromArgs(args, 0)

		workDir, dbType, logFile := sqlmigrator.DefaultContextConfig()
		ctx := sqlmigrator.NewContext(workDir, dbType, logFile, dryRun)
		ctx.LoadMigrationEntries()

		dbRepo := dbrepo.SelectDBRepo(dbType)
		migrator := sqlmigrator.NewMigrator(dbRepo)

		rolledBackMigrations, err := migrator.MigrateDown(ctx, num)
		if err != nil {
			log.Fatal(err)
			return
		}

		err = ctx.WriteMigrationEntries()
		if err != nil {
			log.Println("error writing migration entries:", err)
			return
		}

		displayRolledBack(rolledBackMigrations)
		log.Println("migrated to database successfully.")

	},
}

var GenerateCmd = &cobra.Command{
	Use:   "generate",
	Short: "generate migration files.",
	Args:  cobra.MaximumNArgs(1),
	Run: func(cmd *cobra.Command, args []string) {

		dryRun := cmd.Flag("dry").Value.String() == "true"
		fill := cmd.Flag("fill").Value.String() == "true"

		numToGenerate := getNumberFromArgs(args, 1)

		workDir, dbType, logFile := sqlmigrator.DefaultContextConfig()
		ctx := sqlmigrator.NewContext(workDir, dbType, logFile, dryRun)
		ctx.LoadMigrationEntries()
		generator := sqlmigrator.NewGenerator(migrationName, numToGenerate, fill)
		displayContextAndConfig(ctx, generator)

		log.Println("generating migrations...")
		generatedFiles, err := generator.Generate(ctx)
		if err != nil {
			log.Println("error generating migrations:", err)
			return
		}
		displayGeneratedFiles(generatedFiles)

		err = ctx.WriteMigrationEntries()
		if err != nil {
			log.Println("error writing migration entries:", err)
			return
		}
		log.Println("generated migrations successfully.")

	},
}

var HistoryCmd = &cobra.Command{
	Use:   "history",
	Short: "view the migration history of the database.",
	Args:  cobra.NoArgs,
	Run: func(cmd *cobra.Command, args []string) {

		workDir, dbType, logFile := sqlmigrator.DefaultContextConfig()
		ctx := sqlmigrator.NewContext(workDir, dbType, logFile, false)

		dbRepo := dbrepo.SelectDBRepo(dbType)
		migrator := sqlmigrator.NewMigrator(dbRepo)

		history, err := migrator.MigrationHistory(ctx)
		if err != nil {
			log.Fatal(err)
			return
		}

		displayMigrationHistory(history)

	},
}

var PurgeCmd = &cobra.Command{
	Use:   "purge",
	Short: "remove all migration files and the migration table from the database",
	Args:  cobra.NoArgs,
	Run: func(cmd *cobra.Command, args []string) {
		log.Println("purging migrations...")
		rootDirectory := viper.GetString("migrator.workdir")
		err := os.RemoveAll(rootDirectory)
		if err != nil {
			log.Fatal(err)
			return
		}

		dbType := sqlmigrator.SelectDatabase(viper.GetString("migrator.dbtype"))
		dbrepo := dbrepo.SelectDBRepo(dbType)
		err = dbrepo.DeleteMigrationTable()
		if err != nil {
			log.Fatal(err)
			return
		}

		log.Println("purged migrations successfully.")
	},
}

var UpCmd = &cobra.Command{
	Use:   "up",
	Short: "perform forward migration from the files in the migrations folder",
	Args:  cobra.MaximumNArgs(1),
	Run: func(cmd *cobra.Command, args []string) {

		dryRun := cmd.Flag("dry").Value.String() == "true"
		num := getNumberFromArgs(args, 0)

		workDir, dbType, logFile := sqlmigrator.DefaultContextConfig()
		ctx := sqlmigrator.NewContext(workDir, dbType, logFile, dryRun)
		ctx.LoadMigrationEntries()

		dbRepo := dbrepo.SelectDBRepo(dbType)
		migrator := sqlmigrator.NewMigrator(dbRepo)

		committedMigration, err := migrator.MigrateUp(ctx, num)
		if err != nil {
			log.Fatal(err)
			return
		}

		err = ctx.WriteMigrationEntries()
		if err != nil {
			log.Println("error writing migration entries:", err)
			return
		}

		displayCommitted(committedMigration)
		log.Println("migrated to database successfully.")

	},
}