package config

import (
	"code.icb4dc0.de/prskr/searcherside/core/ports"
	"code.icb4dc0.de/prskr/searcherside/infrastructure/db/migrate"
	"code.icb4dc0.de/prskr/searcherside/internal/ent"
	"code.icb4dc0.de/prskr/searcherside/scripts/migrations"
)

type DB struct {
	Driver ports.Driver `env:"DB_DRIVER" name:"driver" default:"sqlite" help:"DB driver, either postgres or sqlite"`
	URL    string       `env:"DB_URL" name:"url" default:"sqlite://file::memory:?cache=shared" help:"Connection URL"`
}

func (d DB) Client() (*ent.Client, error) {
	return ent.Open(d.Driver.String(), d.URL)
}

func (d DB) Migrator() (ports.Migrator, *ent.Client, error) {
	client, err := d.Client()
	if err != nil {
		return nil, nil, err
	}

	return migrate.AtlasMigrator{
		MigrationsFS: migrations.Fs,
		RevisionRW:   migrate.NewDBRevisionRW(client),
	}, client, nil
}