unity-mcp/Server/src/services/resources/editor_state.py

from pydantic import BaseModel
from fastmcp import Context

from models import MCPResponse
from services.registry import mcp_for_unity_resource
from services.tools import get_unity_instance_from_context
from transport.unity_transport import send_with_unity_instance
from transport.legacy.unity_connection import async_send_command_with_retry


class EditorStateData(BaseModel):
    """Editor state data fields."""
    isPlaying: bool = False
    isPaused: bool = False
    isCompiling: bool = False
    isUpdating: bool = False
    timeSinceStartup: float = 0.0
    activeSceneName: str = ""
    selectionCount: int = 0
    activeObjectName: str | None = None


class EditorStateResponse(MCPResponse):
    """Dynamic editor state information that changes frequently."""
    data: EditorStateData = EditorStateData()


@mcp_for_unity_resource(
    uri="unity://editor/state",
    name="editor_state",
    description="Current editor runtime state including play mode, compilation status, active scene, and selection summary. Refresh frequently for up-to-date information."
)
async def get_editor_state(ctx: Context) -> EditorStateResponse | MCPResponse:
    """Get current editor runtime state."""
    unity_instance = get_unity_instance_from_context(ctx)
    response = await send_with_unity_instance(
        async_send_command_with_retry,
        unity_instance,
        "get_editor_state",
        {}
    )
    # When Unity is reloading/unresponsive (often when unfocused), transports may return
    # a retryable MCPResponse payload with success=false and no data. Do not attempt to
    # coerce that into EditorStateResponse (it would fail validation); return it as-is.
    if isinstance(response, dict):
        if not response.get("success", True):
            return MCPResponse(**response)
        if response.get("data") is None:
            return MCPResponse(success=False, error="Editor state missing 'data' payload", data=response)
        return EditorStateResponse(**response)
    return response