package test

import (
	"context"
	"database/sql"
	migrate "github.com/rubenv/sql-migrate"
	postgrescontainer "github.com/testcontainers/testcontainers-go/modules/postgres"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"time"

	"github.com/testcontainers/testcontainers-go"
	"github.com/testcontainers/testcontainers-go/wait"
)

const (
	databaseName     = "anthrove"
	databaseUser     = "anthrove"
	databasePassword = "anthrove"
	migrationSource  = "../../pkg/database/migrations"
)

func StartPostgresContainer(ctx context.Context) (*postgrescontainer.PostgresContainer, *gorm.DB, error) {

	pgContainer, err := postgrescontainer.RunContainer(ctx,
		testcontainers.WithImage("postgres:alpine"),
		postgrescontainer.WithDatabase(databaseName),
		postgrescontainer.WithUsername(databaseUser),
		postgrescontainer.WithPassword(databasePassword),
		testcontainers.WithWaitStrategy(
			wait.ForLog("database system is ready to accept connections").
				WithOccurrence(2).WithStartupTimeout(5*time.Second)),
	)
	if err != nil {
		return nil, nil, err
	}

	connectionString, err := pgContainer.ConnectionString(ctx, "sslmode=disable")
	if err != nil {
		return nil, nil, err
	}

	err = migrateDatabase(connectionString)
	if err != nil {
		return nil, nil, err

	}

	gormDB, err := getGormDB(connectionString)
	if err != nil {
		return nil, nil, err
	}
	return pgContainer, gormDB, nil
}

func migrateDatabase(connectionString string) error {
	db, err := sql.Open("postgres", connectionString)
	if err != nil {
		return err
	}

	migrations := &migrate.FileMigrationSource{
		Dir: migrationSource,
	}

	_, err = migrate.Exec(db, "postgres", migrations, migrate.Up)
	if err != nil {
		return err
	}

	return nil

}

func getGormDB(connectionString string) (*gorm.DB, error) {
	return gorm.Open(postgres.Open(connectionString), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info),
	})
}