package inventory

import (
	"context"
	"fmt"
	"os"
	"strings"
	"testing"
	"time"

	"github.com/jackc/pgx/v5"

	"github.com/novox/mesh-control/internal/store"
)

// ForTest is a fresh inventory in a database of its own, dropped when the test ends.
//
// Exported because the check that this mesh reads what a node sends lives beside the link, and a
// second copy of this would be a second thing to keep true. It takes a *testing.T, so it cannot
// be called from anything that is not a test.
func ForTest(t *testing.T) *Inventory {
	t.Helper()
	admin := os.Getenv("MESH_TEST_POSTGRES")
	if admin == "" {
		t.Skip("no MESH_TEST_POSTGRES; run `make check` to raise one")
	}
	name := fmt.Sprintf("inv_%d_%s", time.Now().UnixNano()%1_000_000,
		strings.ToLower(strings.NewReplacer("/", "", "-", "").Replace(t.Name())))
	if len(name) > 60 {
		name = name[:60]
	}

	conn, err := pgx.Connect(t.Context(), admin)
	if err != nil {
		t.Fatalf("cannot reach the test PostgreSQL: %v", err)
	}
	if _, err := conn.Exec(t.Context(), "create database "+name); err != nil {
		t.Fatalf("cannot create %s: %v", name, err)
	}
	conn.Close(t.Context())

	cut := strings.LastIndex(admin, "/")
	t.Setenv(store.Variable(Name), admin[:cut]+"/"+name+"?sslmode=disable")

	inv, err := Open(t.Context())
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() {
		inv.Close()
		c, err := pgx.Connect(context.Background(), admin)
		if err != nil {
			return
		}
		defer c.Close(context.Background())
		_, _ = c.Exec(context.Background(), "drop database if exists "+name+" with (force)")
	})
	if err := inv.Ready(t.Context(), 20*time.Second); err != nil {
		t.Fatal(err)
	}

	migrations, err := Migrations()
	if err != nil {
		t.Fatal(err)
	}
	if _, err := inv.store.Migrate(t.Context(), migrations); err != nil {
		t.Fatal(err)
	}
	return inv
}