package dbsetup

import (
	"io/fs"

	"github.com/jmoiron/sqlx"
	"github.com/pressly/goose/v3"
	"github.com/uptrace/opentelemetry-go-extra/otelsqlx"
)

type DatabaseConfig struct {
	DatabaseURL        string `name:"postgres-url" env:"POSTGRES_URL" help:"URL to use to connect to Postgres" default:"postgres://postgres:postgres@:5432/postgres?sslmode=disable"`
	DatabaseDriverName string `name:"db-driver" env:"DB_DRIVER" help:"Driver to use to connect to db" default:"postgres"`
	MaxOpenConnections int    `name:"max-open-connections" env:"DB_MAX_CONNECTIONS" help:"Max open connections to db" default:"50"`
}

type Database struct {
	cfg DatabaseConfig
}

func SetupDB(cfg DatabaseConfig) (*Database, error) {
	db, err := sqlx.Connect(cfg.DatabaseDriverName, cfg.DatabaseURL)
	if err != nil {
		return nil, err
	}
	if err := db.Close(); err != nil {
		return nil, err
	}
	return &Database{
		cfg: cfg,
	}, nil
}

func (d *Database) Connect() (*sqlx.DB, error) {
	db, err := otelsqlx.Connect(d.cfg.DatabaseDriverName, d.cfg.DatabaseURL)
	if err != nil {
		return nil, err
	}
	db.SetMaxOpenConns(d.cfg.MaxOpenConnections)
	return db, nil
}

func (d *Database) RunMigrations(tableName, dir string, fs fs.FS) error {
	db, err := connect(d.cfg)
	if err != nil {
		return err
	}

	goose.SetTableName(tableName)
	goose.SetBaseFS(fs)
	if err := goose.SetDialect(d.gooseDriverName()); err != nil {
		return err
	}
	return goose.Up(db.DB, dir)
}

func (d *Database) gooseDriverName() string {
	if d.cfg.DatabaseDriverName == "sqlmock" {
		return "postgres"
	}
	return d.cfg.DatabaseDriverName
}

func connect(cfg DatabaseConfig) (*sqlx.DB, error) {
	return sqlx.Connect(cfg.DatabaseDriverName, cfg.DatabaseURL)
}