121 lines
4.5 KiB
Python
121 lines
4.5 KiB
Python
"""
|
||
Windows MCP 服务
|
||
封装Windows MCP服务的调用，供后端 API 和 Agent 使用
|
||
提供Windows系统自动化功能
|
||
"""
|
||
|
||
import asyncio
|
||
import json
|
||
import logging
|
||
import os
|
||
from pathlib import Path
|
||
from typing import Optional, Dict, Any, List, Literal
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class WindowsMCPService:
|
||
"""Windows MCP 服务类"""
|
||
|
||
def __init__(self):
|
||
"""初始化Windows MCP 服务"""
|
||
current_dir = Path(__file__).parent
|
||
backend_dir = current_dir.parent
|
||
app_dir = backend_dir.parent
|
||
project_root = app_dir.parent
|
||
self.mcp_path = project_root / "mcp" / "Windows-MCP" / "main.py"
|
||
|
||
logger.info(f"Windows MCP 服务路径: {self.mcp_path}")
|
||
|
||
if not self.mcp_path.exists():
|
||
logger.warning(f"Windows MCP 服务文件不存在: {self.mcp_path}")
|
||
|
||
self.mcp_available = self._check_mcp_available()
|
||
|
||
def _check_mcp_available(self) -> bool:
|
||
"""检查 MCP 是否可用"""
|
||
if not self.mcp_path.exists():
|
||
logger.error(f"❌ Windows MCP 服务文件不存在: {self.mcp_path}")
|
||
return False
|
||
|
||
try:
|
||
from mcp import ClientSession, StdioServerParameters
|
||
from mcp.client.stdio import stdio_client
|
||
logger.info("✅ Windows MCP 依赖检查通过")
|
||
return True
|
||
except ImportError as e:
|
||
logger.error(f"❌ Windows MCP 依赖未安装: {e}")
|
||
return False
|
||
|
||
async def _call_mcp_tool(
|
||
self, tool_name: str, arguments: Dict[str, Any]
|
||
) -> Optional[Dict[str, Any]]:
|
||
"""调用Windows MCP 工具"""
|
||
if not self.mcp_available:
|
||
return {"success": False, "message": "Windows MCP服务不可用"}
|
||
|
||
try:
|
||
from mcp import ClientSession, StdioServerParameters
|
||
from mcp.client.stdio import stdio_client
|
||
|
||
logger.info(f"✅ 调用Windows MCP 工具: {tool_name}, 参数: {arguments}")
|
||
|
||
mcp_dir = self.mcp_path.parent
|
||
|
||
server_params = StdioServerParameters(
|
||
command="uv",
|
||
args=["--directory", str(mcp_dir), "run", "main.py"],
|
||
env=os.environ.copy(),
|
||
)
|
||
|
||
async with stdio_client(server_params) as (read, write):
|
||
async with ClientSession(read, write) as session:
|
||
await session.initialize()
|
||
result = await session.call_tool(tool_name, arguments=arguments)
|
||
|
||
if hasattr(result, "content") and len(result.content) > 0:
|
||
result_text = result.content[0].text
|
||
return {"success": True, "data": result_text}
|
||
else:
|
||
return {"success": True, "data": str(result)}
|
||
|
||
except Exception as e:
|
||
logger.error(f"Windows MCP 服务调用失败: {e}", exc_info=True)
|
||
return {"success": False, "message": f"MCP服务调用失败: {str(e)}"}
|
||
|
||
# ==================== 应用管理 ====================
|
||
async def launch_app(self, app_name: str) -> Optional[Dict[str, Any]]:
|
||
"""启动应用程序"""
|
||
return await self._call_mcp_tool("App-Tool", {"mode": "launch", "name": app_name})
|
||
|
||
async def switch_app(self, app_name: str) -> Optional[Dict[str, Any]]:
|
||
"""切换到指定应用"""
|
||
return await self._call_mcp_tool("App-Tool", {"mode": "switch", "name": app_name})
|
||
|
||
# ==================== PowerShell ====================
|
||
async def execute_powershell(self, command: str) -> Optional[Dict[str, Any]]:
|
||
"""执行PowerShell命令"""
|
||
return await self._call_mcp_tool("Powershell-Tool", {"command": command})
|
||
|
||
# ==================== 桌面状态 ====================
|
||
async def get_desktop_state(self, use_vision: bool = False) -> Optional[Dict[str, Any]]:
|
||
"""获取桌面状态"""
|
||
return await self._call_mcp_tool("State-Tool", {"use_vision": use_vision})
|
||
|
||
# ==================== 快捷键 ====================
|
||
async def shortcut(self, shortcut: str) -> Optional[Dict[str, Any]]:
|
||
"""执行快捷键"""
|
||
return await self._call_mcp_tool("Shortcut-Tool", {"shortcut": shortcut})
|
||
|
||
|
||
# 全局实例
|
||
_windows_mcp_service = None
|
||
|
||
def get_windows_mcp_service() -> WindowsMCPService:
|
||
"""获取Windows MCP 服务实例（单例模式）"""
|
||
global _windows_mcp_service
|
||
if _windows_mcp_service is None:
|
||
_windows_mcp_service = WindowsMCPService()
|
||
return _windows_mcp_service
|
||
|