"""CodeParser 测试（Impact Agent MVP）：Java 项目解析 → CodeStructure。"""
from pathlib import Path

import pytest

from genesis.impact.code_parser import CodeParseError, CodeParser

FIXTURE_JAVA = {
    "trade-order/src/main/java/com/trade/order/controller/OrderController.java": (
        "package com.trade.order.controller;\n"
        "import org.springframework.web.bind.annotation.*;\n"
        "@RestController\n"
        "@RequestMapping(\"/api/order\")\n"
        "public class OrderController {\n"
        "    @GetMapping(\"/{id}\")\n"
        "    public String getOrder(@PathVariable Long id) { return \"ok\"; }\n"
        "    @PostMapping(\"/create\")\n"
        "    public String createOrder() { return \"ok\"; }\n"
        "}\n"
    ),
    "trade-order/src/main/java/com/trade/order/service/OrderQueryService.java": (
        "package com.trade.order.service;\n"
        "import org.springframework.stereotype.Service;\n"
        "@Service\n"
        "public class OrderQueryService {\n"
        "    public void query() {}\n"
        "    public void update() {}\n"
        "}\n"
    ),
    "trade-order/src/main/java/com/trade/order/entity/OrderEntity.java": (
        "package com.trade.order.entity;\n"
        "import javax.persistence.Entity;\n"
        "import javax.persistence.Table;\n"
        "@Entity\n"
        "@Table(name = \"t_order\")\n"
        "public class OrderEntity {\n"
        "    private Long id;\n"
        "    private String orderNo;\n"
        "}\n"
    ),
}


def _make_project(tmp_path, files: dict[str, str]) -> Path:
    for rel, content in files.items():
        p = tmp_path / rel
        p.parent.mkdir(parents=True, exist_ok=True)
        p.write_text(content, encoding="utf-8")
    return tmp_path


def test_parse_controller_detected(tmp_path):
    root = _make_project(tmp_path, FIXTURE_JAVA)
    result = CodeParser().parse(root)
    ctrls = {c.class_name: c for c in result.controllers}
    oc = ctrls["OrderController"]
    assert oc.base_path == "/api/order"
    assert "/api/order/{id}" in oc.endpoints
    assert "/api/order/create" in oc.endpoints
    assert oc.source_uri.endswith("OrderController.java")


def test_parse_service_detected(tmp_path):
    root = _make_project(tmp_path, FIXTURE_JAVA)
    result = CodeParser().parse(root)
    svcs = {s.class_name: s for s in result.services}
    oq = svcs["OrderQueryService"]
    assert "query" in oq.methods
    assert "update" in oq.methods


def test_parse_entity_detected(tmp_path):
    root = _make_project(tmp_path, FIXTURE_JAVA)
    result = CodeParser().parse(root)
    ents = {e.class_name: e for e in result.entities}
    oe = ents["OrderEntity"]
    assert oe.table_name == "t_order"
    assert "id" in oe.fields
    assert "orderNo" in oe.fields


def test_parse_endpoints_detected(tmp_path):
    root = _make_project(tmp_path, FIXTURE_JAVA)
    result = CodeParser().parse(root)
    paths = {(e.method, e.path) for e in result.endpoints}
    assert ("GET", "/api/order/{id}") in paths
    assert ("POST", "/api/order/create") in paths
    assert all(e.controller == "OrderController" for e in result.endpoints)


def test_parse_modules_top_level_dirs(tmp_path):
    root = _make_project(tmp_path, FIXTURE_JAVA)
    result = CodeParser().parse(root)
    assert "trade-order" in result.modules
    assert result.language == "java"


def test_parse_no_java_raises(tmp_path):
    (tmp_path / "readme.txt").write_text("hello", encoding="utf-8")
    with pytest.raises(CodeParseError):
        CodeParser().parse(tmp_path)


def test_parse_missing_root_raises(tmp_path):
    with pytest.raises(CodeParseError):
        CodeParser().parse(tmp_path / "nope")


def test_parse_real_sample_has_order_controller():
    root = Path(__file__).resolve().parents[1] / "samples" / "existing-system"
    if not root.exists():
        pytest.skip("样本缺失")
    result = CodeParser().parse(root)
    names = {c.class_name for c in result.controllers}
    assert "OrderController" in names
    assert any(c.base_path for c in result.controllers)


def test_parse_real_sample_has_entities_and_endpoints():
    root = Path(__file__).resolve().parents[1] / "samples" / "existing-system"
    if not root.exists():
        pytest.skip("样本缺失")
    result = CodeParser().parse(root)
    assert result.entities, "既有系统样本应至少有一个实体类"
    assert result.endpoints, "既有系统样本应至少有一个端点"
    assert result.services, "既有系统样本应至少有一个 @Service 类"