package consumers_test

import (
	"errors"
	"fmt"
	"path/filepath"
	"testing"

	configkit "gitea.maximumdirect.net/eric/ConfigKit"
	"gitea.maximumdirect.net/eric/ConfigKit/examples/consumers"
)

func Example_identifiedPromptCatalog() {
	result, err := consumers.LoadPromptCatalog(fixture("prompt-catalog", "config.yml"))
	if err != nil {
		fmt.Println(err)
		return
	}
	ids := make([]string, 0, len(result.Config.Prompts))
	for _, prompt := range result.Config.Prompts {
		ids = append(ids, prompt.ID)
	}
	fmt.Println(ids, err)
	// Output: [main curated weather welcome] <nil>
}

func TestPromptCatalogImportsRegisteredItemsAndReplacesDefaults(t *testing.T) {
	result, err := consumers.LoadPromptCatalog(fixture("prompt-catalog", "config.yml"))
	if err != nil {
		var loadError *configkit.LoadError
		if errors.As(err, &loadError) {
			t.Fatalf("load prompt catalog: %v (%#v)", err, loadError.Diagnostics())
		}
		t.Fatal(err)
	}
	if len(result.Config.Prompts) != 4 {
		t.Fatalf("prompt records = %#v", result.Config.Prompts)
	}
	ids := []string{"main", "curated", "weather", "welcome"}
	for index, id := range ids {
		if result.Config.Prompts[index].ID != id {
			t.Fatalf("prompt %d ID = %q, want %q", index, result.Config.Prompts[index].ID, id)
		}
	}
	for _, prompt := range result.Config.Prompts {
		if prompt.ID == "default" {
			t.Fatalf("lower-priority default survived main sequence replacement: %#v", result.Config.Prompts)
		}
	}

	var itemImports, helperImports int
	for _, source := range result.Sources() {
		switch filepath.Base(source.Path) {
		case "curated.yml", "generated-weather.yml", "generated-welcome.yml":
			if source.PlacementMode != configkit.PlacementModeSequenceItem || source.Destination.String() != "/prompts" {
				t.Fatalf("item source placement = %#v", source)
			}
			itemImports++
		case "curated-id.yml":
			if source.PlacementMode != configkit.PlacementModeMapping || source.ImportedFrom == nil {
				t.Fatalf("helper source = %#v", source)
			}
			helperImports++
		}
	}
	if itemImports != 3 || helperImports != 1 {
		t.Fatalf("item imports = %d, helper imports = %d, sources = %#v", itemImports, helperImports, result.Sources())
	}
	if result.Config.Prompts[1].ID != "curated" {
		t.Fatalf("helper did not provide prompt ID: %#v", result.Config.Prompts[1])
	}
}

func TestPromptCatalogLeavesDomainValidationWithConsumer(t *testing.T) {
	_, err := consumers.LoadPromptCatalog(fixture("prompt-catalog", "invalid.yml"))
	var loadError *configkit.LoadError
	if !errors.As(err, &loadError) || len(loadError.Diagnostics()) != 1 ||
		loadError.Diagnostics()[0].Code != configkit.CodeValidateError ||
		loadError.Diagnostics()[0].Path.String() != "/prompts/0/messages" {
		t.Fatalf("consumer domain validation error = %v", err)
	}
}