import tempfile
import unittest
from pathlib import Path

from agent_runtime.prompts import PromptBuilder
from agent_runtime.session_index import SessionIndex
from agent_runtime.tools import build_builtin_registry


class PromptBuilderTests(unittest.TestCase):
    def test_prompt_injects_context_and_tool_manifest(self):
        with tempfile.TemporaryDirectory() as tmp:
            root = Path(tmp)
            (root / "prompts").mkdir()
            (root / "prompts" / "agent.md").write_text("agent rules", encoding="utf-8")
            (root / "prompts" / "workflow.md").write_text("workflow rules", encoding="utf-8")
            index = SessionIndex(root)
            index.create(idea="cat")
            registry = build_builtin_registry(root, index)
            builder = PromptBuilder(root / "prompts", index, registry)
            messages = builder.build_messages("start")
            self.assertIn("Available tools", messages[0]["content"])
            self.assertIn("当前 working_dir 尚未完成结构化文本文件", messages[0]["content"])
            self.assertIn("read_file", messages[0]["content"])
            trace = builder.trace(builder.build_parts("start"))
            self.assertGreater(trace["total_estimated_tokens"], 0)


    def test_prompt_injects_compacted_session_summary_as_reference(self):
        with tempfile.TemporaryDirectory() as tmp:
            root = Path(tmp)
            (root / "prompts").mkdir()
            (root / "prompts" / "agent.md").write_text("agent rules", encoding="utf-8")
            (root / "prompts" / "workflow.md").write_text("workflow rules", encoding="utf-8")
            index = SessionIndex(root)
            record = index.create(idea="cat")
            index.update_compaction(record["session_id"], {"summary": "## Reference Context Only\n- user wants moon cat", "compacted_message_count": 4, "preserved_message_count": 2})
            registry = build_builtin_registry(root, index)
            builder = PromptBuilder(root / "prompts", index, registry)
            message = builder.build_messages("continue")[0]["content"]
            self.assertIn("Session context summary", message)
            self.assertIn("reference context only", message)
            self.assertIn("user wants moon cat", message)
            trace = builder.trace(builder.build_parts("continue"))
            self.assertGreater(trace["totals"]["dynamic_tokens"], 0)

    def test_prompt_treats_novel_text_artifacts_as_text_stage_complete(self):
        with tempfile.TemporaryDirectory() as tmp:
            root = Path(tmp)
            (root / "prompts").mkdir()
            (root / "prompts" / "agent.md").write_text("agent rules", encoding="utf-8")
            (root / "prompts" / "workflow.md").write_text("workflow rules", encoding="utf-8")
            index = SessionIndex(root)
            record = index.create(idea="novel")
            session_root = root / record["working_dir"] / "novel2video"
            (session_root / "novel").mkdir(parents=True)
            (session_root / "novel" / "novel_compressed.txt").write_text("compressed", encoding="utf-8")
            (session_root / "events").mkdir()
            (session_root / "events" / "event_0.json").write_text("{}", encoding="utf-8")
            (session_root / "relevant_chunks" / "event_0").mkdir(parents=True)
            (session_root / "relevant_chunks" / "event_0" / "chunk.txt").write_text("chunk", encoding="utf-8")
            (session_root / "scenes" / "event_0").mkdir(parents=True)
            (session_root / "scenes" / "event_0" / "scene_0.json").write_text("{}", encoding="utf-8")
            (session_root / "global_information" / "characters" / "novel_level").mkdir(parents=True)
            (session_root / "global_information" / "characters" / "novel_level" / "novel_characters_after_event_0.json").write_text("[]", encoding="utf-8")
            registry = build_builtin_registry(root, index)
            builder = PromptBuilder(root / "prompts", index, registry)
            messages = builder.build_messages("continue")
            self.assertIn("文本规划阶段已完成", messages[0]["content"])
            self.assertIn("novel2video/events/event_*.json: present", messages[0]["content"])