package postgres

import (
	"context"
	"database/sql"
	"fmt"
	"log/slog"

	_ "github.com/jackc/pgx/v5/stdlib" // registra o driver "pgx" no database/sql
	"github.com/pressly/goose/v3"

	"botpontos/db/migrations"
	"botpontos/internal/config"
)

// Migrate executa o goose com as migrations embutidas no binário.
// Deve rodar com o usuário dono do banco, não com o usuário da aplicação.
func Migrate(ctx context.Context, cfg config.DB, command string, logger *slog.Logger) error {
	db, err := sql.Open("pgx", cfg.DSN())
	if err != nil {
		return fmt.Errorf("abrir conexão para migrations: %w", err)
	}
	defer db.Close()

	if err := waitFor(ctx, logger, db.PingContext); err != nil {
		return err
	}

	goose.SetBaseFS(migrations.FS)
	if err := goose.SetDialect("postgres"); err != nil {
		return err
	}

	switch command {
	case "up":
		err = goose.UpContext(ctx, db, ".")
	case "status":
		err = goose.StatusContext(ctx, db, ".")
	case "down":
		err = goose.DownContext(ctx, db, ".")
	default:
		return fmt.Errorf("subcomando de migrate desconhecido: %s (use up, status ou down)", command)
	}
	if err != nil {
		return fmt.Errorf("migrate %s: %w", command, err)
	}

	logger.Info("migrations concluídas", "command", command)
	return nil
}