package database

import (
	"context"

	"github.com/jackc/pgx/v5/pgxpool"

	"git.unkin.net/unkin/golib/pg"

	"git.unkin.net/unkin/artifactapi/migrations"
)

// migrationLockName names the cluster-wide advisory lock the migration run
// contends for; golib derives the key as FNV-1a/64 of it. Replicas starting
// together queue on it instead of racing each other through the DDL.
const migrationLockName = "artifactapi-migrations"

type DB struct {
	Pool *pgxpool.Pool
}

func New(dsn string) (*DB, error) {
	ctx := context.Background()
	pool, err := pg.NewMigrated(ctx, dsn, migrations.FS, pg.MigrateOptions{
		LockName: migrationLockName,
	})
	if err != nil {
		return nil, err
	}
	return &DB{Pool: pool}, nil
}

func (db *DB) Close() {
	db.Pool.Close()
}