refactor: add context support to resource handlers

- Updated all resource handlers to accept Context parameter for Unity instance routing
- Replaced direct async_send_command_with_retry calls with async_send_with_unity_instance wrapper
- Added imports for get_unity_instance_from_context and async_send_with_unity_instance helpers

Author: msanatan

MCPForUnity/UnityMcpServer~/src/resources/active_tool.py

@@ -1,6 +1,9 @@
 from pydantic import BaseModel
+from fastmcp import Context
+
 from models import MCPResponse
 from registry import mcp_for_unity_resource
+from tools import get_unity_instance_from_context, async_send_with_unity_instance
 from unity_connection import async_send_command_with_retry
 
 
@@ -31,7 +34,13 @@ class ActiveToolResponse(MCPResponse):
     name="editor_active_tool",
     description="Currently active editor tool (Move, Rotate, Scale, etc.) and transform handle settings."
 )
-async def get_active_tool() -> ActiveToolResponse | MCPResponse:
+async def get_active_tool(ctx: Context) -> ActiveToolResponse | MCPResponse:
     """Get active editor tool information."""
-    response = await async_send_command_with_retry("get_active_tool", {})
+    unity_instance = get_unity_instance_from_context(ctx)
+    response = await async_send_with_unity_instance(
+        async_send_command_with_retry,
+        unity_instance,
+        "get_active_tool",
+        {}
+    )
     return ActiveToolResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/editor_state.py

@@ -1,6 +1,9 @@
 from pydantic import BaseModel
+from fastmcp import Context
+
 from models import MCPResponse
 from registry import mcp_for_unity_resource
+from tools import get_unity_instance_from_context, async_send_with_unity_instance
 from unity_connection import async_send_command_with_retry
 
 
@@ -26,7 +29,13 @@ class EditorStateResponse(MCPResponse):
     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() -> EditorStateResponse | MCPResponse:
+async def get_editor_state(ctx: Context) -> EditorStateResponse | MCPResponse:
     """Get current editor runtime state."""
-    response = await async_send_command_with_retry("get_editor_state", {})
+    unity_instance = get_unity_instance_from_context(ctx)
+    response = await async_send_with_unity_instance(
+        async_send_command_with_retry,
+        unity_instance,
+        "get_editor_state",
+        {}
+    )
     return EditorStateResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/layers.py

@@ -1,5 +1,8 @@
+from fastmcp import Context
+
 from models import MCPResponse
 from registry import mcp_for_unity_resource
+from tools import get_unity_instance_from_context, async_send_with_unity_instance
 from unity_connection import async_send_command_with_retry
 
 
@@ -13,7 +16,13 @@ class LayersResponse(MCPResponse):
     name="project_layers",
     description="All layers defined in the project's TagManager with their indices (0-31). Read this before using add_layer or remove_layer tools."
 )
-async def get_layers() -> LayersResponse | MCPResponse:
+async def get_layers(ctx: Context) -> LayersResponse | MCPResponse:
     """Get all project layers with their indices."""
-    response = await async_send_command_with_retry("get_layers", {})
+    unity_instance = get_unity_instance_from_context(ctx)
+    response = await async_send_with_unity_instance(
+        async_send_command_with_retry,
+        unity_instance,
+        "get_layers",
+        {}
+    )
     return LayersResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/prefab_stage.py

@@ -1,6 +1,9 @@
 from pydantic import BaseModel
+from fastmcp import Context
+
 from models import MCPResponse
 from registry import mcp_for_unity_resource
+from tools import get_unity_instance_from_context, async_send_with_unity_instance
 from unity_connection import async_send_command_with_retry
 
 
@@ -23,7 +26,13 @@ class PrefabStageResponse(MCPResponse):
     name="editor_prefab_stage",
     description="Current prefab editing context if a prefab is open in isolation mode. Returns isOpen=false if no prefab is being edited."
 )
-async def get_prefab_stage() -> PrefabStageResponse | MCPResponse:
+async def get_prefab_stage(ctx: Context) -> PrefabStageResponse | MCPResponse:
     """Get current prefab stage information."""
-    response = await async_send_command_with_retry("get_prefab_stage", {})
+    unity_instance = get_unity_instance_from_context(ctx)
+    response = await async_send_with_unity_instance(
+        async_send_command_with_retry,
+        unity_instance,
+        "get_prefab_stage",
+        {}
+    )
     return PrefabStageResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/project_info.py

@@ -1,6 +1,9 @@
 from pydantic import BaseModel
+from fastmcp import Context
+
 from models import MCPResponse
 from registry import mcp_for_unity_resource
+from tools import get_unity_instance_from_context, async_send_with_unity_instance
 from unity_connection import async_send_command_with_retry
 
 
@@ -23,7 +26,13 @@ class ProjectInfoResponse(MCPResponse):
     name="project_info",
     description="Static project information including root path, Unity version, and platform. This data rarely changes."
 )
-async def get_project_info() -> ProjectInfoResponse | MCPResponse:
+async def get_project_info(ctx: Context) -> ProjectInfoResponse | MCPResponse:
     """Get static project configuration information."""
-    response = await async_send_command_with_retry("get_project_info", {})
+    unity_instance = get_unity_instance_from_context(ctx)
+    response = await async_send_with_unity_instance(
+        async_send_command_with_retry,
+        unity_instance,
+        "get_project_info",
+        {}
+    )
     return ProjectInfoResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/selection.py

@@ -1,6 +1,9 @@
 from pydantic import BaseModel
+from fastmcp import Context
+
 from models import MCPResponse
 from registry import mcp_for_unity_resource
+from tools import get_unity_instance_from_context, async_send_with_unity_instance
 from unity_connection import async_send_command_with_retry
 
 
@@ -39,7 +42,13 @@ class SelectionResponse(MCPResponse):
     name="editor_selection",
     description="Detailed information about currently selected objects in the editor, including GameObjects, assets, and their properties."
 )
-async def get_selection() -> SelectionResponse | MCPResponse:
+async def get_selection(ctx: Context) -> SelectionResponse | MCPResponse:
     """Get detailed editor selection information."""
-    response = await async_send_command_with_retry("get_selection", {})
+    unity_instance = get_unity_instance_from_context(ctx)
+    response = await async_send_with_unity_instance(
+        async_send_command_with_retry,
+        unity_instance,
+        "get_selection",
+        {}
+    )
     return SelectionResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/tags.py

@@ -1,6 +1,9 @@
 from pydantic import Field
+from fastmcp import Context
+
 from models import MCPResponse
 from registry import mcp_for_unity_resource
+from tools import get_unity_instance_from_context, async_send_with_unity_instance
 from unity_connection import async_send_command_with_retry
 
 
@@ -14,7 +17,13 @@ class TagsResponse(MCPResponse):
     name="project_tags",
     description="All tags defined in the project's TagManager. Read this before using add_tag or remove_tag tools."
 )
-async def get_tags() -> TagsResponse | MCPResponse:
+async def get_tags(ctx: Context) -> TagsResponse | MCPResponse:
     """Get all project tags."""
-    response = await async_send_command_with_retry("get_tags", {})
+    unity_instance = get_unity_instance_from_context(ctx)
+    response = await async_send_with_unity_instance(
+        async_send_command_with_retry,
+        unity_instance,
+        "get_tags",
+        {}
+    )
     return TagsResponse(**response) if isinstance(response, dict) else response

MCPForUnity/UnityMcpServer~/src/resources/windows.py

@@ -1,6 +1,9 @@
 from pydantic import BaseModel
+from fastmcp import Context
+
 from models import MCPResponse
 from registry import mcp_for_unity_resource
+from tools import get_unity_instance_from_context, async_send_with_unity_instance
 from unity_connection import async_send_command_with_retry
 
 
@@ -31,7 +34,13 @@ class WindowsResponse(MCPResponse):
     name="editor_windows",
     description="All currently open editor windows with their titles, types, positions, and focus state."
 )
-async def get_windows() -> WindowsResponse | MCPResponse:
+async def get_windows(ctx: Context) -> WindowsResponse | MCPResponse:
     """Get all open editor windows."""
-    response = await async_send_command_with_retry("get_windows", {})
+    unity_instance = get_unity_instance_from_context(ctx)
+    response = await async_send_with_unity_instance(
+        async_send_command_with_retry,
+        unity_instance,
+        "get_windows",
+        {}
+    )
     return WindowsResponse(**response) if isinstance(response, dict) else response