70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
from mcp.server.fastmcp import FastMCP, Context, Image
|
|
import logging
|
|
from dataclasses import dataclass
|
|
from contextlib import asynccontextmanager
|
|
from typing import AsyncIterator, Dict, Any, List
|
|
from config import config
|
|
from tools import register_all_tools
|
|
from unity_connection import get_unity_connection, UnityConnection
|
|
|
|
# Configure logging using settings from config
|
|
logging.basicConfig(
|
|
level=getattr(logging, config.log_level),
|
|
format=config.log_format
|
|
)
|
|
logger = logging.getLogger("UnityMCP")
|
|
|
|
# Global connection state
|
|
_unity_connection: UnityConnection = None
|
|
|
|
@asynccontextmanager
|
|
async def server_lifespan(server: FastMCP) -> AsyncIterator[Dict[str, Any]]:
|
|
"""Handle server startup and shutdown."""
|
|
global _unity_connection
|
|
logger.info("UnityMCP server starting up")
|
|
try:
|
|
_unity_connection = get_unity_connection()
|
|
logger.info("Connected to Unity on startup")
|
|
except Exception as e:
|
|
logger.warning(f"Could not connect to Unity on startup: {str(e)}")
|
|
_unity_connection = None
|
|
try:
|
|
# Yield the connection object so it can be attached to the context
|
|
# The key 'bridge' matches how tools like read_console expect to access it (ctx.bridge)
|
|
yield {"bridge": _unity_connection}
|
|
finally:
|
|
if _unity_connection:
|
|
_unity_connection.disconnect()
|
|
_unity_connection = None
|
|
logger.info("UnityMCP server shut down")
|
|
|
|
# Initialize MCP server
|
|
mcp = FastMCP(
|
|
"UnityMCP",
|
|
description="Unity Editor integration via Model Context Protocol",
|
|
lifespan=server_lifespan
|
|
)
|
|
|
|
# Register all tools
|
|
register_all_tools(mcp)
|
|
|
|
# Asset Creation Strategy
|
|
|
|
@mcp.prompt()
|
|
def asset_creation_strategy() -> str:
|
|
"""Guide for discovering and using Unity MCP tools effectively."""
|
|
return (
|
|
"Available Unity MCP Server Tools:\\n\\n"
|
|
"For detailed usage, please refer to the specific tool's documentation.\\n\\n"
|
|
"- `manage_editor`: Controls editor state (play/pause/stop) and queries info (state, selection).\\n"
|
|
"- `execute_menu_item`: Executes Unity Editor menu items by path (e.g., 'File/Save Project').\\n"
|
|
"- `read_console`: Reads or clears Unity console messages, with filtering options.\\n"
|
|
"- `manage_scene`: Manages scenes (load, save, create, get hierarchy).\\n"
|
|
"- `manage_gameobject`: Manages GameObjects in the scene (CRUD, find, components, assign properties).\\n"
|
|
"- `manage_script`: Manages C# script files (CRUD).\\n"
|
|
"- `manage_asset`: Manages project assets (import, create, modify, delete, search).\\n\\n"
|
|
)
|
|
|
|
# Run the server
|
|
if __name__ == "__main__":
|
|
mcp.run(transport='stdio') |