unity-mcp/Server/tools/resource_tools.py

"""
Resource wrapper tools so clients that do not expose MCP resources primitives
can still list and read files via normal tools. These call into the same
safe path logic (re-implemented here to avoid importing server.py).
"""
import fnmatch
import hashlib
import os
from pathlib import Path
import re
from typing import Annotated, Any
from urllib.parse import urlparse, unquote

from fastmcp import Context

from registry import mcp_for_unity_tool
from tools import get_unity_instance_from_context, send_with_unity_instance, async_send_with_unity_instance
from unity_connection import send_command_with_retry


def _coerce_int(value: Any, default: int | None = None, minimum: int | None = None) -> int | None:
    """Safely coerce various inputs (str/float/etc.) to an int.
    Returns default on failure; clamps to minimum when provided.
    """
    if value is None:
        return default
    try:
        # Avoid treating booleans as ints implicitly
        if isinstance(value, bool):
            return default
        if isinstance(value, int):
            result = int(value)
        else:
            s = str(value).strip()
            if s.lower() in ("", "none", "null"):
                return default
            # Allow "10.0" or similar inputs
            result = int(float(s))
        if minimum is not None and result < minimum:
            return minimum
        return result
    except Exception:
        return default


def _resolve_project_root(ctx: Context, override: str | None) -> Path:
    unity_instance = get_unity_instance_from_context(ctx)
    # 1) Explicit override
    if override:
        pr = Path(override).expanduser().resolve()
        if (pr / "Assets").exists():
            return pr
    # 2) Environment
    env = os.environ.get("UNITY_PROJECT_ROOT")
    if env:
        env_path = Path(env).expanduser()
        # If UNITY_PROJECT_ROOT is relative, resolve against repo root (cwd's repo) instead of src dir
        pr = (Path.cwd(
        ) / env_path).resolve() if not env_path.is_absolute() else env_path.resolve()
        if (pr / "Assets").exists():
            return pr
    # 3) Ask Unity via manage_editor.get_project_root
    try:
        response = send_with_unity_instance(
            send_command_with_retry,
            unity_instance,
            "manage_editor",
            {"action": "get_project_root"},
        )
        if isinstance(response, dict) and response.get("success"):
            pr = Path(response.get("data", {}).get(
                "projectRoot", "")).expanduser().resolve()
            if pr and (pr / "Assets").exists():
                return pr
    except Exception:
        pass

    # 4) Walk up from CWD to find a Unity project (Assets + ProjectSettings)
    cur = Path.cwd().resolve()
    for _ in range(6):
        if (cur / "Assets").exists() and (cur / "ProjectSettings").exists():
            return cur
        if cur.parent == cur:
            break
        cur = cur.parent
    # 5) Search downwards (shallow) from repo root for first folder with Assets + ProjectSettings
    try:
        import os as _os
        root = Path.cwd().resolve()
        max_depth = 3
        for dirpath, dirnames, _ in _os.walk(root):
            rel = Path(dirpath).resolve()
            try:
                depth = len(rel.relative_to(root).parts)
            except Exception:
                # Unrelated mount/permission edge; skip deeper traversal
                dirnames[:] = []
                continue
            if depth > max_depth:
                # Prune deeper traversal
                dirnames[:] = []
                continue
            if (rel / "Assets").exists() and (rel / "ProjectSettings").exists():
                return rel
    except Exception:
        pass
    # 6) Fallback: CWD
    return Path.cwd().resolve()


def _resolve_safe_path_from_uri(uri: str, project: Path) -> Path | None:
    raw: str | None = None
    if uri.startswith("unity://path/"):
        raw = uri[len("unity://path/"):]
    elif uri.startswith("file://"):
        parsed = urlparse(uri)
        raw = unquote(parsed.path or "")
        # On Windows, urlparse('file:///C:/x') -> path='/C:/x'. Strip the leading slash for drive letters.
        try:
            import os as _os
            if _os.name == "nt" and raw.startswith("/") and re.match(r"^/[A-Za-z]:/", raw):
                raw = raw[1:]
            # UNC paths: file://server/share -> netloc='server', path='/share'. Treat as \\\\server/share
            if _os.name == "nt" and parsed.netloc:
                raw = f"//{parsed.netloc}{raw}"
        except Exception:
            pass
    elif uri.startswith("Assets/"):
        raw = uri
    if raw is None:
        return None
    # Normalize separators early
    raw = raw.replace("\\", "/")
    p = (project / raw).resolve()
    try:
        p.relative_to(project)
    except ValueError:
        return None
    return p


@mcp_for_unity_tool(description=("List project URIs (unity://path/...) under a folder (default: Assets). Only .cs files are returned by default; always appends unity://spec/script-edits.\n"))
async def list_resources(
    ctx: Context,
    pattern: Annotated[str, "Glob, default is *.cs"] | None = "*.cs",
    under: Annotated[str,
                     "Folder under project root, default is Assets"] = "Assets",
    limit: Annotated[int, "Page limit"] = 200,
    project_root: Annotated[str, "Project path"] | None = None,
) -> dict[str, Any]:
    unity_instance = get_unity_instance_from_context(ctx)
    ctx.info(f"Processing list_resources: {pattern} (unity_instance={unity_instance or 'default'})")
    try:
        project = _resolve_project_root(ctx, project_root)
        base = (project / under).resolve()
        try:
            base.relative_to(project)
        except ValueError:
            return {"success": False, "error": "Base path must be under project root"}
        # Enforce listing only under Assets
        try:
            base.relative_to(project / "Assets")
        except ValueError:
            return {"success": False, "error": "Listing is restricted to Assets/"}

        matches: list[str] = []
        limit_int = _coerce_int(limit, default=200, minimum=1)
        for p in base.rglob("*"):
            if not p.is_file():
                continue
            # Resolve symlinks and ensure the real path stays under project/Assets
            try:
                rp = p.resolve()
                rp.relative_to(project / "Assets")
            except Exception:
                continue
            # Enforce .cs extension regardless of provided pattern
            if p.suffix.lower() != ".cs":
                continue
            if pattern and not fnmatch.fnmatch(p.name, pattern):
                continue
            rel = p.relative_to(project).as_posix()
            matches.append(f"unity://path/{rel}")
            if len(matches) >= max(1, limit_int):
                break

        # Always include the canonical spec resource so NL clients can discover it
        if "unity://spec/script-edits" not in matches:
            matches.append("unity://spec/script-edits")

        return {"success": True, "data": {"uris": matches, "count": len(matches)}}
    except Exception as e:
        return {"success": False, "error": str(e)}


@mcp_for_unity_tool(description=("Reads a resource by unity://path/... URI with optional slicing."))
async def read_resource(
    ctx: Context,
    uri: Annotated[str, "The resource URI to read under Assets/"],
    start_line: Annotated[int | float | str,
                          "The starting line number (0-based)"] | None = None,
    line_count: Annotated[int | float | str,
                          "The number of lines to read"] | None = None,
    head_bytes: Annotated[int | float | str,
                          "The number of bytes to read from the start of the file"] | None = None,
    tail_lines: Annotated[int | float | str,
                          "The number of lines to read from the end of the file"] | None = None,
    project_root: Annotated[str,
                            "The project root directory"] | None = None,
    request: Annotated[str, "The request ID"] | None = None,
) -> dict[str, Any]:
    unity_instance = get_unity_instance_from_context(ctx)
    ctx.info(f"Processing read_resource: {uri} (unity_instance={unity_instance or 'default'})")
    try:
        # Serve the canonical spec directly when requested (allow bare or with scheme)
        if uri in ("unity://spec/script-edits", "spec/script-edits", "script-edits"):
            spec_json = (
                '{\n'
                '  "name": "MCP for Unity - Script Edits v1",\n'
                '  "target_tool": "script_apply_edits",\n'
                '  "canonical_rules": {\n'
                '    "always_use": ["op","className","methodName","replacement","afterMethodName","beforeMethodName"],\n'
                '    "never_use": ["new_method","anchor_method","content","newText"],\n'
                '    "defaults": {\n'
                '      "className": "\u2190 server will default to \'name\' when omitted",\n'
                '      "position": "end"\n'
                '    }\n'
                '  },\n'
                '  "ops": [\n'
                '    {"op":"replace_method","required":["className","methodName","replacement"],"optional":["returnType","parametersSignature","attributesContains"],"examples":[{"note":"match overload by signature","parametersSignature":"(int a, string b)"},{"note":"ensure attributes retained","attributesContains":"ContextMenu"}]},\n'
                '    {"op":"insert_method","required":["className","replacement"],"position":{"enum":["start","end","after","before"],"after_requires":"afterMethodName","before_requires":"beforeMethodName"}},\n'
                '    {"op":"delete_method","required":["className","methodName"]},\n'
                '    {"op":"anchor_insert","required":["anchor","text"],"notes":"regex; position=before|after"}\n'
                '  ],\n'
                '  "apply_text_edits_recipe": {\n'
                '    "step1_read": { "tool": "resources/read", "args": {"uri": "unity://path/Assets/Scripts/Interaction/SmartReach.cs"} },\n'
                '    "step2_apply": {\n'
                '      "tool": "manage_script",\n'
                '      "args": {\n'
                '        "action": "apply_text_edits",\n'
                '        "name": "SmartReach", "path": "Assets/Scripts/Interaction",\n'
                '        "edits": [{"startLine": 42, "startCol": 1, "endLine": 42, "endCol": 1, "newText": "[MyAttr]\\n"}],\n'
                '        "precondition_sha256": "<sha-from-step1>",\n'
                '        "options": {"refresh": "immediate", "validate": "standard"}\n'
                '      }\n'
                '    },\n'
                '    "note": "newText is for apply_text_edits ranges only; use replacement in script_apply_edits ops."\n'
                '  },\n'
                '  "examples": [\n'
                '    {\n'
                '      "title": "Replace a method",\n'
                '      "args": {\n'
                '        "name": "SmartReach",\n'
                '        "path": "Assets/Scripts/Interaction",\n'
                '        "edits": [\n'
                '          {"op":"replace_method","className":"SmartReach","methodName":"HasTarget","replacement":"public bool HasTarget() { return currentTarget != null; }"}\n'
                '        ],\n'
                '        "options": { "validate": "standard", "refresh": "immediate" }\n'
                '      }\n'
                '    },\n'
                '    {\n'
                '      "title": "Insert a method after another",\n'
                '      "args": {\n'
                '        "name": "SmartReach",\n'
                '        "path": "Assets/Scripts/Interaction",\n'
                '        "edits": [\n'
                '          {"op":"insert_method","className":"SmartReach","replacement":"public void PrintSeries() { Debug.Log(seriesName); }","position":"after","afterMethodName":"GetCurrentTarget"}\n'
                '        ]\n'
                '      }\n'
                '    }\n'
                '  ]\n'
                '}\n'
            )
            sha = hashlib.sha256(spec_json.encode("utf-8")).hexdigest()
            return {"success": True, "data": {"text": spec_json, "metadata": {"sha256": sha}}}

        project = _resolve_project_root(ctx, project_root)
        p = _resolve_safe_path_from_uri(uri, project)
        if not p or not p.exists() or not p.is_file():
            return {"success": False, "error": f"Resource not found: {uri}"}
        try:
            p.relative_to(project / "Assets")
        except ValueError:
            return {"success": False, "error": "Read restricted to Assets/"}
        # Natural-language convenience: request like "last 120 lines", "first 200 lines",
        # "show 40 lines around MethodName", etc.
        if request:
            req = request.strip().lower()
            m = re.search(r"last\s+(\d+)\s+lines", req)
            if m:
                tail_lines = int(m.group(1))
            m = re.search(r"first\s+(\d+)\s+lines", req)
            if m:
                start_line = 1
                line_count = int(m.group(1))
            m = re.search(r"first\s+(\d+)\s*bytes", req)
            if m:
                head_bytes = int(m.group(1))
            m = re.search(
                r"show\s+(\d+)\s+lines\s+around\s+([A-Za-z_][A-Za-z0-9_]*)", req)
            if m:
                window = int(m.group(1))
                method = m.group(2)
                # naive search for method header to get a line number
                text_all = p.read_text(encoding="utf-8")
                lines_all = text_all.splitlines()
                pat = re.compile(
                    rf"^\s*(?:\[[^\]]+\]\s*)*(?:public|private|protected|internal|static|virtual|override|sealed|async|extern|unsafe|new|partial).*?\b{re.escape(method)}\s*\(", re.MULTILINE)
                hit_line = None
                for i, line in enumerate(lines_all, start=1):
                    if pat.search(line):
                        hit_line = i
                        break
                if hit_line:
                    half = max(1, window // 2)
                    start_line = max(1, hit_line - half)
                    line_count = window

        # Coerce numeric inputs defensively (string/float -> int)
        start_line = _coerce_int(start_line)
        line_count = _coerce_int(line_count)
        head_bytes = _coerce_int(head_bytes, minimum=1)
        tail_lines = _coerce_int(tail_lines, minimum=1)

        # Compute SHA over full file contents (metadata-only default)
        full_bytes = p.read_bytes()
        full_sha = hashlib.sha256(full_bytes).hexdigest()

        # Selection only when explicitly requested via windowing args or request text hints
        selection_requested = bool(head_bytes or tail_lines or (
            start_line is not None and line_count is not None) or request)
        if selection_requested:
            # Mutually exclusive windowing options precedence:
            # 1) head_bytes, 2) tail_lines, 3) start_line+line_count, else full text
            if head_bytes and head_bytes > 0:
                raw = full_bytes[: head_bytes]
                text = raw.decode("utf-8", errors="replace")
            else:
                text = full_bytes.decode("utf-8", errors="replace")
                if tail_lines is not None and tail_lines > 0:
                    lines = text.splitlines()
                    n = max(0, tail_lines)
                    text = "\n".join(lines[-n:])
                elif start_line is not None and line_count is not None and line_count >= 0:
                    lines = text.splitlines()
                    s = max(0, start_line - 1)
                    e = min(len(lines), s + line_count)
                    text = "\n".join(lines[s:e])
            return {"success": True, "data": {"text": text, "metadata": {"sha256": full_sha, "lengthBytes": len(full_bytes)}}}
        else:
            # Default: metadata only
            return {"success": True, "data": {"metadata": {"sha256": full_sha, "lengthBytes": len(full_bytes)}}}
    except Exception as e:
        return {"success": False, "error": str(e)}


@mcp_for_unity_tool(description="Searches a file with a regex pattern and returns line numbers and excerpts.")
async def find_in_file(
    ctx: Context,
    uri: Annotated[str, "The resource URI to search under Assets/ or file path form supported by read_resource"],
    pattern: Annotated[str, "The regex pattern to search for"],
    ignore_case: Annotated[bool | str, "Case-insensitive search (accepts true/false or 'true'/'false')"] | None = True,
    project_root: Annotated[str,
                            "The project root directory"] | None = None,
    max_results: Annotated[int,
                           "Cap results to avoid huge payloads"] = 200,
) -> dict[str, Any]:
    unity_instance = get_unity_instance_from_context(ctx)
    ctx.info(f"Processing find_in_file: {uri} (unity_instance={unity_instance or 'default'})")
    try:
        project = _resolve_project_root(ctx, project_root)
        p = _resolve_safe_path_from_uri(uri, project)
        if not p or not p.exists() or not p.is_file():
            return {"success": False, "error": f"Resource not found: {uri}"}

        text = p.read_text(encoding="utf-8")
        # Tolerant boolean coercion for clients that stringify booleans
        def _coerce_bool(val, default=None):
            if val is None:
                return default
            if isinstance(val, bool):
                return val
            if isinstance(val, str):
                v = val.strip().lower()
                if v in ("true", "1", "yes", "on"):
                    return True
                if v in ("false", "0", "no", "off"):
                    return False
            return bool(val)
        ignore_case = _coerce_bool(ignore_case, default=True)
        flags = re.MULTILINE
        if ignore_case:
            flags |= re.IGNORECASE
        rx = re.compile(pattern, flags)

        results = []
        max_results_int = _coerce_int(max_results, default=200, minimum=1)
        lines = text.splitlines()
        for i, line in enumerate(lines, start=1):
            m = rx.search(line)
            if m:
                start_col = m.start() + 1  # 1-based
                end_col = m.end() + 1      # 1-based, end exclusive
                results.append({
                    "startLine": i,
                    "startCol": start_col,
                    "endLine": i,
                    "endCol": end_col,
                })
                if max_results_int and len(results) >= max_results_int:
                    break

        return {"success": True, "data": {"matches": results, "count": len(results)}}
    except Exception as e:
        return {"success": False, "error": str(e)}