import importlib.util
import subprocess
import sys
import tempfile
import unittest
from pathlib import Path


ROOT = Path(__file__).resolve().parents[1]
SCRIPT = ROOT / "skills" / "tsl-syntax-reference" / "scripts" / "lookup.py"

spec = importlib.util.spec_from_file_location("tsl_syntax_lookup", SCRIPT)
lookup = importlib.util.module_from_spec(spec)
spec.loader.exec_module(lookup)


class TslSyntaxLookupTests(unittest.TestCase):
    def test_parser_creates_unique_h2_h3_section_ids(self):
        sections = lookup.load_sections(lookup.DEFAULT_REFERENCES_DIR)
        ids = [section.id for section in sections]
        self.assertEqual(len(ids), len(set(ids)))
        self.assertTrue(any("05_functions_and_calls" in value for value in ids))

    def test_write_mode_includes_file_model_and_direct_example(self):
        result = lookup.query_sections("写函数 命名参数", "write", limit=4)
        rendered = lookup.render_result(result)
        self.assertIn("文件模型", rendered)
        self.assertIn("代码块身份：可直接照写示例", rendered)

    def test_diagnose_mode_prioritizes_invalid_statement_pitfall(self):
        result = lookup.query_sections("invalid statement 声明区", "diagnose", limit=4)
        self.assertIn("11_pitfalls.md", result.matches[0].section.page.as_posix())

    def test_explain_mode_does_not_force_write_prelude(self):
        result = lookup.query_sections("数组下标", "explain", limit=3)
        self.assertEqual(result.prelude, [])

    def test_no_match_returns_exit_code_two_in_every_mode(self):
        for mode in ("write", "diagnose", "explain"):
            with self.subTest(mode=mode):
                completed = subprocess.run(
                    [
                        sys.executable,
                        str(SCRIPT),
                        "--query",
                        "不存在的孤立语法词xyz",
                        "--mode",
                        mode,
                    ],
                    capture_output=True,
                    text=True,
                    encoding="utf-8",
                )
                self.assertEqual(completed.returncode, 2)

    def test_write_identifier_query_keeps_required_context(self):
        result = lookup.query_sections("varByRef 命名参数", "write", limit=5)
        prelude_pages = {section.page.name for section in result.prelude}
        self.assertEqual(prelude_pages, {"01_quickstart.md", "02_core_model.md"})

    def test_check_rejects_unknown_code_block_identity(self):
        with tempfile.TemporaryDirectory() as tmp_dir:
            references = Path(tmp_dir)
            (references / "sample.md").write_text(
                "# Sample\n\n"
                "## Example\n\n"
                "代码块身份：未知身份\n\n"
                "```tsl\n"
                "return 1;\n"
                "```\n",
                encoding="utf-8",
                newline="\n",
            )

            problems = lookup.validate_references(references)

        self.assertTrue(
            any("未知身份" in problem.message for problem in problems), problems
        )

    def test_check_accumulates_fence_identity_and_link_problems(self):
        with tempfile.TemporaryDirectory() as tmp_dir:
            references = Path(tmp_dir)
            (references / "sample.md").write_text(
                "# Sample\n\n"
                "## Example\n\n"
                "[missing](missing.md)\n\n"
                "代码块身份：输出片段\n"
                "代码块身份：可直接照写示例\n\n"
                "```tsl\n"
                "return 1;\n",
                encoding="utf-8",
                newline="\n",
            )

            problems = lookup.validate_references(references)

        messages = "\n".join(problem.message for problem in problems)
        self.assertIn("本地链接不存在", messages)
        self.assertIn("代码围栏未闭合", messages)
        self.assertIn("恰好一个代码块身份", messages)

    def test_check_ignores_markdown_link_shapes_inside_inline_code(self):
        with tempfile.TemporaryDirectory() as tmp_dir:
            references = Path(tmp_dir)
            (references / "sample.md").write_text(
                "# Sample\n\n"
                "## Operators\n\n"
                "Use `function operator[](index);`.\n",
                encoding="utf-8",
                newline="\n",
            )

            problems = lookup.validate_references(references)

        self.assertEqual(problems, [])

    def test_natural_chinese_query_finds_object_creation(self):
        result = lookup.query_sections("写一个类并创建对象", "write", limit=5)
        sources = "\n".join(match.section.id for match in result.matches)
        self.assertIn("08_objects_and_classes", sources)

    def test_tsl_identifier_is_preserved(self):
        result = lookup.query_sections("varByRef 命名参数", "explain", limit=5)
        rendered = lookup.render_result(result)
        self.assertRegex(rendered, r"(?i)varByRef")


if __name__ == "__main__":
    unittest.main()