package spells

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"

func RegisterPromptAssets(registry *llm.AssetRegistry) error {
	promptFS, err := shared.ModulePromptFS("dnd.spells", embeddedAssets, []promptfs.ModulePromptFile{
		{Name: "dnd.spells.yaml", Path: "assets/prompts/dnd.spells.yaml"},
		{Name: "catalog.md", Path: "assets/prompts/catalog.md"},
		{Name: "task.md", Path: "assets/prompts/task.md"},
		{Name: "instructions.md", Path: "assets/prompts/instructions.md"},
	})
	if err != nil {
		return fmt.Errorf("prepare spell 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() {
		parts := append([]llm.AssetHashPart{
			{FS: embeddedAssets, Path: "assets/prompts/dnd.spells.yaml"},
			{FS: embeddedAssets, Path: "assets/prompts/catalog.md"},
			{FS: embeddedAssets, Path: "assets/prompts/task.md"},
			{FS: embeddedAssets, Path: "assets/prompts/instructions.md"},
		}, append(shared.CommonHashParts(), shared.ReferenceHashParts()...)...)
		scriptoriumPromptHash, scriptoriumPromptHashErr = llm.HashAssets(parts)
	})
	return scriptoriumPromptHash, scriptoriumPromptHashErr
}

var (
	scriptoriumPromptHashOnce sync.Once
	scriptoriumPromptHash     string
	scriptoriumPromptHashErr  error
)