from __future__ import annotations

import subprocess
from pathlib import Path

from agents.coder import Coder
from agents.providers import DeterministicCodingProvider
from model_router.router import ModelRouter
from tools.capabilities import Capability
from tools.runtime import CapabilityError, WorktreeTools


def test_worktree_investigation_tools_are_path_scoped(tmp_path: Path) -> None:
    repo = tmp_path / "repo"
    repo.mkdir()
    (repo / "app").mkdir()
    (repo / "app" / "urls.py").write_text("urlpatterns = []\n", encoding="utf-8")
    (repo / "items.py").write_text("class Item:\n    pass\n", encoding="utf-8")
    subprocess.run(["git", "init", "-b", "main"], cwd=repo, check=True, capture_output=True, text=True)
    tools = WorktreeTools(repo, {Capability.READ_REPOSITORY, Capability.INVESTIGATE_WORKTREE})

    assert "app/" in tools.list_directory(".")
    assert "urlpatterns" in tools.read_file("app/urls.py")
    assert tools.search_code("urlpatterns")[0]["path"] == "app/urls.py"
    assert tools.find_symbol("Item")[0]["path"] == "items.py"
    assert tools.git_status() == "?? app/\n?? items.py\n"

    try:
        tools.read_file("../outside.txt")
    except CapabilityError as exc:
        assert "escapes worktree" in str(exc)
    else:
        raise AssertionError("path escape should be blocked")


def test_coder_requires_inspection_for_architecture_sensitive_tasks(tmp_path: Path) -> None:
    repo = tmp_path / "repo"
    repo.mkdir()
    (repo / "app").mkdir()
    (repo / "app" / "urls.py").write_text("urlpatterns = []\n", encoding="utf-8")
    (repo / "tests").mkdir()
    subprocess.run(["git", "init", "-b", "main"], cwd=repo, check=True, capture_output=True, text=True)
    tools = WorktreeTools(
        repo,
        {Capability.READ_REPOSITORY, Capability.INVESTIGATE_WORKTREE, Capability.WRITE_WORKTREE, Capability.RUN_TESTS},
    )
    context = {
        "task": {
            "goal": 'Add a /health endpoint returning JSON {"status": "ok"} and add tests.',
            "acceptance_criteria": ["tests pass"],
        }
    }

    result = Coder(ModelRouter({"qwen": DeterministicCodingProvider()})).execute(context, tools)

    assert result.status == "COMPLETE"
    assert result.metadata["inspection_results"]
    assert (repo / "app" / "urls.py").read_text(encoding="utf-8")