package cmd

import (
	"context"

	"github.com/pkg/errors"
	"github.com/pressly/goose/v3"
	"github.com/samber/lo"
	log "github.com/sirupsen/logrus"
	"github.com/spf13/cobra"
	"go.ipao.vip/atomctl/v2/pkg/postgres"
)

// migrate
func CommandMigrate(root *cobra.Command) {
	cmd := &cobra.Command{
		Use:     "migrate [up|up-by-one|up-to|create|down|down-to|fix|redo|reset|status|version]",
		Aliases: []string{"m"},
		RunE:    commandMigrate,
	}
	cmd.Flags().StringP("config", "c", "config.toml", "database config file")
	cmd.Flags().String("dir", "database/migrations", "migrations directory")
	cmd.Flags().String("table", "migrations", "migrations table name")

	root.AddCommand(cmd)
}

func commandMigrate(cmd *cobra.Command, args []string) error {
	if len(args) == 0 {
		args = append(args, "up")
	}

	if args[0] == "create" {
		if !lo.Contains(args, "sql") {
			args = append(args, "sql")
		}
	}

	db, _, err := postgres.GetDB(cmd.Flag("config").Value.String())
	if err != nil {
		return errors.Wrap(err, "get db")
	}

	action, args := args[0], args[1:]
	log.Infof("migration action: %s args: %+v", action, args)

	dir := cmd.Flag("dir").Value.String()
	table := cmd.Flag("table").Value.String()

	goose.SetTableName(table)

	return goose.RunContext(context.Background(), action, db, dir, args...)
}