package database

import (
	"context"
	"fmt"
	"log"
	"os"
	"time"

	"github.com/pressly/goose/v3"
)

const (
	defaultGooseMigrationsDir = "migrations_goose"
	migrationAdvisoryLockKey  = int64(637266846588921720)
)

var migrationLockRetryInterval = 250 * time.Millisecond

// MigrateAll runs the goose-managed migrations. goose is the only migration
// path; the legacy filename-tracked system was consolidated into the baseline.
func (db *DB) MigrateAll(gooseDir string) error {
	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
	defer cancel()
	return db.MigrateAllWithLock(ctx, gooseDir)
}

// MigrateAllWithLock runs goose migrations while holding a PostgreSQL
// advisory lock to prevent concurrent migrators from racing.
func (db *DB) MigrateAllWithLock(ctx context.Context, gooseDir string) error {
	if gooseDir == "" {
		gooseDir = defaultGooseMigrationsDir
	}

	releaseLock, err := db.acquireMigrationLock(ctx)
	if err != nil {
		return err
	}
	defer func() {
		if unlockErr := releaseLock(); unlockErr != nil {
			log.Printf("Warning: failed to release migration advisory lock: %v", unlockErr)
		}
	}()

	return db.MigrateGoose(gooseDir)
}

func (db *DB) acquireMigrationLock(ctx context.Context) (func() error, error) {
	if ctx == nil {
		ctx = context.Background()
	}

	conn, err := db.Conn(ctx)
	if err != nil {
		return nil, fmt.Errorf("failed to open connection for migration lock: %w", err)
	}

	for {
		var acquired bool
		if err := conn.QueryRowContext(ctx, "SELECT pg_try_advisory_lock($1)", migrationAdvisoryLockKey).Scan(&acquired); err != nil {
			conn.Close()
			return nil, fmt.Errorf("failed to acquire migration advisory lock: %w", err)
		}
		if acquired {
			break
		}

		select {
		case <-ctx.Done():
			conn.Close()
			return nil, fmt.Errorf("timed out waiting for migration advisory lock: %w", ctx.Err())
		case <-time.After(migrationLockRetryInterval):
		}
	}

	log.Println("Acquired migration advisory lock")

	return func() error {
		defer conn.Close()

		var released bool
		if err := conn.QueryRowContext(context.Background(), "SELECT pg_advisory_unlock($1)", migrationAdvisoryLockKey).Scan(&released); err != nil {
			return fmt.Errorf("failed to release migration advisory lock: %w", err)
		}
		if !released {
			return fmt.Errorf("migration advisory lock was not held by current session")
		}

		log.Println("Released migration advisory lock")
		return nil
	}, nil
}

func (db *DB) MigrateGoose(migrationsDir string) error {
	if _, err := os.Stat(migrationsDir); err != nil {
		if os.IsNotExist(err) {
			log.Printf("Goose migrations directory %q not found, skipping goose migrations", migrationsDir)
			return nil
		}
		return fmt.Errorf("failed to access goose migrations directory %q: %w", migrationsDir, err)
	}

	if err := db.ensureRequiredExtensions(); err != nil {
		return err
	}

	if err := goose.SetDialect("postgres"); err != nil {
		return fmt.Errorf("failed to set goose dialect: %w", err)
	}

	if err := goose.Up(db.DB, migrationsDir); err != nil {
		return fmt.Errorf("failed to run goose migrations from %q: %w", migrationsDir, err)
	}

	log.Printf("Goose migrations completed successfully from %q", migrationsDir)
	return nil
}

func (db *DB) GooseStatus(migrationsDir string) error {
	if err := goose.SetDialect("postgres"); err != nil {
		return fmt.Errorf("failed to set goose dialect: %w", err)
	}
	if err := goose.Status(db.DB, migrationsDir); err != nil {
		return fmt.Errorf("failed to get goose migration status: %w", err)
	}
	return nil
}