unity-mcp/UnityMcpBridge/UnityMcpServer~/src/tools/manage_scene.py

from typing import Annotated, Literal, Any

from mcp.server.fastmcp import FastMCP, Context
from telemetry_decorator import telemetry_tool

from unity_connection import send_command_with_retry


def register_manage_scene_tools(mcp: FastMCP):
    """Register all scene management tools with the MCP server."""

    @mcp.tool(name="manage_scene", description="Manage Unity scenes")
    @telemetry_tool("manage_scene")
    def manage_scene(
        ctx: Context,
        action: Annotated[Literal["create", "load", "save", "get_hierarchy", "get_active", "get_build_settings"], "Perform CRUD operations on Unity scenes."],
        name: Annotated[str,
                        "Scene name. Not required get_active/get_build_settings"] | None = None,
        path: Annotated[str,
                        "Asset path for scene operations (default: 'Assets/')"] | None = None,
        build_index: Annotated[int,
                               "Build index for load/build settings actions"] | None = None,
    ) -> dict[str, Any]:
        ctx.info(f"Processing manage_scene: {action}")
        try:
            # Coerce numeric inputs defensively
            def _coerce_int(value, default=None):
                if value is None:
                    return default
                try:
                    if isinstance(value, bool):
                        return default
                    if isinstance(value, int):
                        return int(value)
                    s = str(value).strip()
                    if s.lower() in ("", "none", "null"):
                        return default
                    return int(float(s))
                except Exception:
                    return default

            coerced_build_index = _coerce_int(build_index, default=None)

            params = {"action": action}
            if name:
                params["name"] = name
            if path:
                params["path"] = path
            if coerced_build_index is not None:
                params["buildIndex"] = coerced_build_index

            # Use centralized retry helper
            response = send_command_with_retry("manage_scene", params)

            # Preserve structured failure data; unwrap success into a friendlier shape
            if isinstance(response, dict) and response.get("success"):
                return {"success": True, "message": response.get("message", "Scene operation successful."), "data": response.get("data")}
            return response if isinstance(response, dict) else {"success": False, "message": str(response)}

        except Exception as e:
            return {"success": False, "message": f"Python error managing scene: {str(e)}"}