47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
|
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
|
||
|
|
|
||
|
|
|
||
|
|
class Vector3(BaseModel):
|
||
|
|
"""3D vector."""
|
||
|
|
x: float = 0.0
|
||
|
|
y: float = 0.0
|
||
|
|
z: float = 0.0
|
||
|
|
|
||
|
|
|
||
|
|
class ActiveToolData(BaseModel):
|
||
|
|
"""Active tool data fields."""
|
||
|
|
activeTool: str = ""
|
||
|
|
isCustom: bool = False
|
||
|
|
pivotMode: str = ""
|
||
|
|
pivotRotation: str = ""
|
||
|
|
handleRotation: Vector3 = Vector3()
|
||
|
|
handlePosition: Vector3 = Vector3()
|
||
|
|
|
||
|
|
|
||
|
|
class ActiveToolResponse(MCPResponse):
|
||
|
|
"""Information about the currently active editor tool."""
|
||
|
|
data: ActiveToolData = ActiveToolData()
|
||
|
|
|
||
|
|
|
||
|
|
@mcp_for_unity_resource(
|
||
|
|
uri="unity://editor/active-tool",
|
||
|
|
name="editor_active_tool",
|
||
|
|
description="Currently active editor tool (Move, Rotate, Scale, etc.) and transform handle settings."
|
||
|
|
)
|
||
|
|
async def get_active_tool(ctx: Context) -> ActiveToolResponse | MCPResponse:
|
||
|
|
"""Get active editor tool information."""
|
||
|
|
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
|