package npcs

import (
	"fmt"
	"sync"

	"gitea.maximumdirect.net/eric/notarius/internal/framework/llm"
	"gitea.maximumdirect.net/eric/notarius/internal/framework/promptfs"
	"gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/shared"
)

const scriptoriumPromptRoot = "assets/prompts"

var promptAssetManifest = shared.PromptAssetManifest{
	ModuleDir: PromptID,
	ModuleFiles: []promptfs.ModulePromptFile{
		{Name: "dnd.npcs.normalize.yaml", Path: "assets/prompts/dnd.npcs.normalize.yaml"},
		{Name: "task.md", Path: "assets/prompts/task.md"},
		{Name: "instructions.md", Path: "assets/prompts/instructions.md"},
		{Name: "candidates.md", Path: "assets/prompts/candidates.md"},
	},
	SharedFiles: []string{
		"common-dnd-system.md",
		"common-dnd-identity.md",
		"common-dnd-transcript.md",
	},
}

func RegisterPromptAssets(registry *llm.AssetRegistry) error {
	promptFS, err := promptAssetManifest.PromptFS(embeddedAssets)
	if err != nil {
		return fmt.Errorf("prepare NPC normalization prompt assets: %w", err)
	}
	if err := registry.RegisterPromptFS(promptFS, scriptoriumPromptRoot); err != nil {
		return err
	}
	return registry.RegisterSchemaFS(embeddedAssets, "assets/schemas")
}

func scriptoriumPromptMetadata() (string, error) {
	scriptoriumPromptHashOnce.Do(func() {
		scriptoriumPromptHash, scriptoriumPromptHashErr = promptAssetManifest.Hash(embeddedAssets)
	})
	return scriptoriumPromptHash, scriptoriumPromptHashErr
}

var (
	scriptoriumPromptHashOnce sync.Once
	scriptoriumPromptHash     string
	scriptoriumPromptHashErr  error
)