package register

import (
	"gitea.maximumdirect.net/eric/notarius/internal/framework/llm"
	"gitea.maximumdirect.net/eric/notarius/internal/framework/pipeline"
	"gitea.maximumdirect.net/eric/notarius/internal/modules/dnd"
	"gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/chunk/scenes"
	combatcodec "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/codec/combatturns"
	npccodec "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/codec/npcs"
	spellcodec "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/codec/spells"
	combatextract "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/extract/combatturns"
	npcextract "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/extract/npcs"
	spellextract "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/extract/spells"
	combatnormalize "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/normalize/combatturns"
	npcnormalize "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/normalize/npcs"
	spellnormalize "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/normalize/spells"
	"gitea.maximumdirect.net/eric/notarius/internal/modules/generic/merge/appendorder"
	"gitea.maximumdirect.net/eric/notarius/internal/modules/generic/normalize/noop"
)

func registerModules(registries pipeline.Registries) error {
	codec := spellcodec.New()
	return runRegistrations([]registration{
		{name: "spells codec", register: func() error { return pipeline.RegisterArtifactCodec(registries.ArtifactCodecs, codec) }},
		{name: "npcs codec", register: func() error { return pipeline.RegisterArtifactCodec(registries.ArtifactCodecs, npccodec.New()) }},
		{name: "combat turns codec", register: func() error { return pipeline.RegisterArtifactCodec(registries.ArtifactCodecs, combatcodec.New()) }},
		{name: "scenes chunker", register: func() error { return scenes.Register(registries.Chunkers) }},
		{name: "spells extractor", register: func() error { return spellextract.Register(registries.Extractors) }},
		{name: "npcs extractor", register: func() error { return npcextract.Register(registries.Extractors) }},
		{name: "combat turns extractor", register: func() error { return combatextract.Register(registries.Extractors) }},
		{name: "spell-list appendorder merger", register: func() error {
			return appendorder.RegisterTyped(registries.Mergers, dnd.SpellListKind, appendSpellLists)
		}},
		{name: "npc-list appendorder merger", register: func() error {
			return appendorder.RegisterTyped(registries.Mergers, dnd.NPCListKind, appendNPCLists)
		}},
		{name: "combat-turn-list appendorder merger", register: func() error {
			return appendorder.RegisterTyped(registries.Mergers, dnd.CombatTurnListKind, appendCombatTurnLists)
		}},
		{name: "spells normalizer", register: func() error { return spellnormalize.Register(registries.Normalizers) }},
		{name: "npcs normalizer", register: func() error { return npcnormalize.Register(registries.Normalizers) }},
		{name: "combat turns normalizer", register: func() error { return combatnormalize.Register(registries.Normalizers) }},
		{name: "spell-list noop normalizer", register: func() error {
			return noop.RegisterTyped[dnd.SpellList](registries.Normalizers, dnd.SpellListKind)
		}},
		{name: "npc-list noop normalizer", register: func() error {
			return noop.RegisterTyped[dnd.NPCList](registries.Normalizers, dnd.NPCListKind)
		}},
		{name: "combat-turn-list noop normalizer", register: func() error {
			return noop.RegisterTyped[dnd.CombatTurnList](registries.Normalizers, dnd.CombatTurnListKind)
		}},
	})
}

func registerPromptAssets(assets *llm.AssetRegistry) error {
	return runRegistrations([]registration{
		{name: "scenes prompt assets", register: func() error { return scenes.RegisterPromptAssets(assets) }},
		{name: "spells prompt assets", register: func() error { return spellextract.RegisterPromptAssets(assets) }},
		{name: "npcs prompt assets", register: func() error { return npcextract.RegisterPromptAssets(assets) }},
		{name: "combat turns prompt assets", register: func() error { return combatextract.RegisterPromptAssets(assets) }},
	})
}