46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
from typing import Annotated, Any
|
|
|
|
from fastmcp import Context
|
|
from registry import mcp_for_unity_tool
|
|
from unity_connection import get_unity_connection_pool
|
|
from unity_instance_middleware import get_unity_instance_middleware
|
|
|
|
|
|
@mcp_for_unity_tool(
|
|
description="Set the active Unity instance for this client/session. Accepts Name@hash or hash."
|
|
)
|
|
def set_active_instance(
|
|
ctx: Context,
|
|
instance: Annotated[str, "Target instance (Name@hash or hash prefix)"]
|
|
) -> dict[str, Any]:
|
|
# Discover running instances
|
|
pool = get_unity_connection_pool()
|
|
instances = pool.discover_all_instances(force_refresh=True)
|
|
ids = {inst.id: inst for inst in instances}
|
|
hashes = {}
|
|
for inst in instances:
|
|
# exact hash and prefix map; last write wins but we'll detect ambiguity
|
|
hashes.setdefault(inst.hash, inst)
|
|
|
|
# Disallow plain names to ensure determinism
|
|
value = instance.strip()
|
|
resolved = None
|
|
if "@" in value:
|
|
resolved = ids.get(value)
|
|
if resolved is None:
|
|
return {"success": False, "error": f"Instance '{value}' not found. Check unity://instances resource."}
|
|
else:
|
|
# Treat as hash/prefix; require unique match
|
|
candidates = [inst for inst in instances if inst.hash.startswith(value)]
|
|
if len(candidates) == 1:
|
|
resolved = candidates[0]
|
|
elif len(candidates) == 0:
|
|
return {"success": False, "error": f"No instance with hash '{value}'."}
|
|
else:
|
|
return {"success": False, "error": f"Hash '{value}' matches multiple instances: {[c.id for c in candidates]}"}
|
|
|
|
# Store selection in middleware (session-scoped)
|
|
middleware = get_unity_instance_middleware()
|
|
middleware.set_active_instance(ctx, resolved.id)
|
|
return {"success": True, "message": f"Active instance set to {resolved.id}", "data": {"instance": resolved.id}}
|