package.go

package migratorCli

import (
	"fmt"
	"log"
	"strings"

	"github.com/uptrace/bun/migrate"
	"github.com/urfave/cli/v2"
)

// InitCli initializes the cli with the given migrator and os args
// extendCli can be used to add additional commands to the cli
func InitCli(migrator *migrate.Migrator, osArgs []string, extendCli ...*cli.Command) {
	app := &cli.App{
		Name:      "migrator",
		Usage:     "Bun Migration CLI Tool",
		UsageText: "migrator [command]",
		Commands:  getCommands(migrator, extendCli...),
	}

	if err := app.Run(osArgs); err != nil {
		log.Fatal(err)
	}
}

func getCommands(migrator *migrate.Migrator, extendCli ...*cli.Command) []*cli.Command {
	commands := []*cli.Command{
		{
			Name:  "init",
			Usage: "create migration tables",
			Action: func(c *cli.Context) error {
				return migrator.Init(c.Context)
			},
		},
		{
			Name:  "migrate",
			Usage: "migrate database",
			Action: func(c *cli.Context) error {
				if err := migrator.Lock(c.Context); err != nil {
					return err
				}
				defer migrator.Unlock(c.Context) //nolint:errcheck

				group, err := migrator.Migrate(c.Context)
				if err != nil {
					return err
				}
				if group.IsZero() {
					fmt.Printf("there are no new migrations to run (database is up to date)\n")
					return nil
				}
				fmt.Printf("migrated to %s\n", group)
				return nil
			},
		},
		{
			Name:  "rollback",
			Usage: "rollback the last migration group",
			Action: func(c *cli.Context) error {
				if err := migrator.Lock(c.Context); err != nil {
					return err
				}
				defer migrator.Unlock(c.Context) //nolint:errcheck

				group, err := migrator.Rollback(c.Context)
				if err != nil {
					return err
				}
				if group.IsZero() {
					fmt.Printf("there are no groups to roll back\n")
					return nil
				}
				fmt.Printf("rolled back %s\n", group)
				return nil
			},
		},
		{
			Name:  "lock",
			Usage: "lock migrations",
			Action: func(c *cli.Context) error {
				return migrator.Lock(c.Context)
			},
		},
		{
			Name:  "unlock",
			Usage: "unlock migrations",
			Action: func(c *cli.Context) error {
				return migrator.Unlock(c.Context)
			},
		},
		{
			Name:  "create_go",
			Usage: "create Go migration",
			Action: func(c *cli.Context) error {
				name := strings.Join(c.Args().Slice(), "_")
				mf, err := migrator.CreateGoMigration(c.Context, name)
				if err != nil {
					return err
				}
				fmt.Printf("created migration %s (%s)\n", mf.Name, mf.Path)
				return nil
			},
		},
		{
			Name:  "create_sql",
			Usage: "create up and down SQL migrations",
			Action: func(c *cli.Context) error {
				name := strings.Join(c.Args().Slice(), "_")
				files, err := migrator.CreateSQLMigrations(c.Context, name)
				if err != nil {
					return err
				}

				for _, mf := range files {
					fmt.Printf("created migration %s (%s)\n", mf.Name, mf.Path)
				}

				return nil
			},
		},
		{
			Name:  "create_tx_sql",
			Usage: "create up and down transactional SQL migrations",
			Action: func(c *cli.Context) error {
				name := strings.Join(c.Args().Slice(), "_")
				files, err := migrator.CreateTxSQLMigrations(c.Context, name)
				if err != nil {
					return err
				}

				for _, mf := range files {
					fmt.Printf("created transaction migration %s (%s)\n", mf.Name, mf.Path)
				}

				return nil
			},
		},
		{
			Name:  "status",
			Usage: "print migrations status",
			Action: func(c *cli.Context) error {
				ms, err := migrator.MigrationsWithStatus(c.Context)
				if err != nil {
					return err
				}
				fmt.Printf("migrations: %s\n", ms)
				fmt.Printf("unapplied migrations: %s\n", ms.Unapplied())
				fmt.Printf("last migration group: %s\n", ms.LastGroup())
				return nil
			},
		},
		{
			Name:  "mark_applied",
			Usage: "mark migrations as applied without actually running them",
			Action: func(c *cli.Context) error {
				group, err := migrator.Migrate(c.Context, migrate.WithNopMigration())
				if err != nil {
					return err
				}
				if group.IsZero() {
					fmt.Printf("there are no new migrations to mark as applied\n")
					return nil
				}
				fmt.Printf("marked as applied %s\n", group)
				return nil
			},
		},
	}

	return append(commands, extendCli...)
}