package template

import (
	"os"
	"path/filepath"
	"strings"
	"testing"

	"git.warky.dev/wdevs/relspecgo/pkg/models"
	"git.warky.dev/wdevs/relspecgo/pkg/writers"
)

func TestWriterTableIndexValuesDeterministic(t *testing.T) {
	dir := t.TempDir()
	templatePath := filepath.Join(dir, "indexes.tmpl")
	outputDir := filepath.Join(dir, "out")
	outputPath := filepath.Join(outputDir, "accounts.txt")
	templateBody := "{{range values .Table.Indexes}}{{.Name}}:{{join .Columns \",\"}}\n{{end}}"

	if err := os.MkdirAll(outputDir, 0755); err != nil {
		t.Fatalf("create output dir: %v", err)
	}
	if err := os.WriteFile(templatePath, []byte(templateBody), 0644); err != nil {
		t.Fatalf("write template: %v", err)
	}

	db := databaseWithMultipleIndexes()
	var first []byte
	const runs = 100
	for i := 0; i < runs; i++ {
		writer, err := NewWriter(&writers.WriterOptions{
			OutputPath: outputDir,
			Metadata: map[string]interface{}{
				"template_path": templatePath,
				"mode":          string(TableMode),
			},
		})
		if err != nil {
			t.Fatalf("new writer: %v", err)
		}

		if err := writer.WriteDatabase(db); err != nil {
			t.Fatalf("write database run %d: %v", i, err)
		}

		got, err := os.ReadFile(outputPath)
		if err != nil {
			t.Fatalf("read output run %d: %v", i, err)
		}

		if i == 0 {
			first = got
			continue
		}
		if string(got) != string(first) {
			t.Fatalf("run %d output differed from first run\nfirst:\n%s\nrun %d:\n%s", i, first, i, got)
		}
	}

	want := strings.Join([]string{
		"idx_accounts_email:email",
		"idx_accounts_last_login:last_login",
		"idx_accounts_name:name",
		"idx_accounts_status:status",
		"idx_accounts_tenant:tenant_id",
		"",
	}, "\n")
	if string(first) != want {
		t.Fatalf("unexpected index order\nwant:\n%s\ngot:\n%s", want, first)
	}
}

func databaseWithMultipleIndexes() *models.Database {
	db := models.InitDatabase("test")
	schema := models.InitSchema("public")
	table := models.InitTable("accounts", "public")

	table.Indexes["idx_accounts_status"] = &models.Index{Name: "idx_accounts_status", Table: table.Name, Schema: schema.Name, Columns: []string{"status"}}
	table.Indexes["idx_accounts_email"] = &models.Index{Name: "idx_accounts_email", Table: table.Name, Schema: schema.Name, Columns: []string{"email"}}
	table.Indexes["idx_accounts_tenant"] = &models.Index{Name: "idx_accounts_tenant", Table: table.Name, Schema: schema.Name, Columns: []string{"tenant_id"}}
	table.Indexes["idx_accounts_name"] = &models.Index{Name: "idx_accounts_name", Table: table.Name, Schema: schema.Name, Columns: []string{"name"}}
	table.Indexes["idx_accounts_last_login"] = &models.Index{Name: "idx_accounts_last_login", Table: table.Name, Schema: schema.Name, Columns: []string{"last_login"}}

	schema.Tables = append(schema.Tables, table)
	db.Schemas = append(db.Schemas, schema)
	return db
}