forked from HomeLab/unraid-mcp
b00d78f408
- Remove array_status, system_info, notifications_overview, and parity_status resources - Keep only logs_stream resource (unraid://logs/stream) which is working properly - Update README.md with current resource documentation and modern docker compose syntax - Fix import path issues that were causing subscription errors - Update environment configuration examples - Clean up subscription manager to only include working log streaming 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
141 lines
4.9 KiB
Python
141 lines
4.9 KiB
Python
"""Modular Unraid MCP Server.
|
|
|
|
This is the main server implementation using the modular architecture with
|
|
separate modules for configuration, core functionality, subscriptions, and tools.
|
|
"""
|
|
|
|
import sys
|
|
|
|
from fastmcp import FastMCP
|
|
|
|
from .config.logging import logger
|
|
from .config.settings import (
|
|
UNRAID_API_KEY,
|
|
UNRAID_API_URL,
|
|
UNRAID_MCP_HOST,
|
|
UNRAID_MCP_PORT,
|
|
UNRAID_MCP_TRANSPORT,
|
|
)
|
|
from .subscriptions.diagnostics import register_diagnostic_tools
|
|
from .subscriptions.manager import SubscriptionManager
|
|
from .subscriptions.resources import register_subscription_resources
|
|
from .tools.docker import register_docker_tools
|
|
from .tools.health import register_health_tools
|
|
from .tools.rclone import register_rclone_tools
|
|
from .tools.storage import register_storage_tools
|
|
from .tools.system import register_system_tools
|
|
from .tools.virtualization import register_vm_tools
|
|
|
|
# Initialize FastMCP instance
|
|
mcp = FastMCP(
|
|
name="Unraid MCP Server",
|
|
instructions="Provides tools to interact with an Unraid server's GraphQL API.",
|
|
version="0.1.0",
|
|
)
|
|
|
|
# Initialize subscription manager
|
|
subscription_manager = SubscriptionManager()
|
|
|
|
|
|
async def autostart_subscriptions():
|
|
"""Auto-start all subscriptions marked for auto-start in SubscriptionManager"""
|
|
logger.info("[AUTOSTART] Initiating subscription auto-start process...")
|
|
|
|
try:
|
|
# Use the SubscriptionManager auto-start method
|
|
await subscription_manager.auto_start_all_subscriptions()
|
|
logger.info("[AUTOSTART] Auto-start process completed successfully")
|
|
except Exception as e:
|
|
logger.error(f"[AUTOSTART] Failed during auto-start process: {e}", exc_info=True)
|
|
|
|
|
|
def register_all_modules():
|
|
"""Register all tools and resources with the MCP instance."""
|
|
try:
|
|
# Register subscription resources first
|
|
register_subscription_resources(mcp)
|
|
logger.info("📊 Subscription resources registered")
|
|
|
|
# Register diagnostic tools
|
|
register_diagnostic_tools(mcp)
|
|
logger.info("🔧 Diagnostic tools registered")
|
|
|
|
# Register all tool categories
|
|
register_system_tools(mcp)
|
|
logger.info("🖥️ System tools registered")
|
|
|
|
register_docker_tools(mcp)
|
|
logger.info("🐳 Docker tools registered")
|
|
|
|
register_vm_tools(mcp)
|
|
logger.info("💻 Virtualization tools registered")
|
|
|
|
register_storage_tools(mcp)
|
|
logger.info("💾 Storage tools registered")
|
|
|
|
register_health_tools(mcp)
|
|
logger.info("🏥 Health tools registered")
|
|
|
|
register_rclone_tools(mcp)
|
|
logger.info("☁️ RClone tools registered")
|
|
|
|
logger.info("🎯 All modules registered successfully - Server ready!")
|
|
|
|
except Exception as e:
|
|
logger.error(f"❌ Failed to register modules: {e}", exc_info=True)
|
|
raise
|
|
|
|
|
|
def run_server():
|
|
"""Run the MCP server with the configured transport."""
|
|
# Log configuration
|
|
if UNRAID_API_URL:
|
|
logger.info(f"UNRAID_API_URL loaded: {UNRAID_API_URL[:20]}...")
|
|
else:
|
|
logger.warning("UNRAID_API_URL not found in environment or .env file.")
|
|
|
|
if UNRAID_API_KEY:
|
|
logger.info("UNRAID_API_KEY loaded: ****")
|
|
else:
|
|
logger.warning("UNRAID_API_KEY not found in environment or .env file.")
|
|
|
|
logger.info(f"UNRAID_MCP_PORT set to: {UNRAID_MCP_PORT}")
|
|
logger.info(f"UNRAID_MCP_HOST set to: {UNRAID_MCP_HOST}")
|
|
logger.info(f"UNRAID_MCP_TRANSPORT set to: {UNRAID_MCP_TRANSPORT}")
|
|
|
|
# Register all modules
|
|
register_all_modules()
|
|
|
|
logger.info(f"🚀 Starting Unraid MCP Server on {UNRAID_MCP_HOST}:{UNRAID_MCP_PORT} using {UNRAID_MCP_TRANSPORT} transport...")
|
|
|
|
try:
|
|
# Auto-start subscriptions on first async operation
|
|
if UNRAID_MCP_TRANSPORT == "streamable-http":
|
|
# Use the recommended Streamable HTTP transport
|
|
mcp.run(
|
|
transport="streamable-http",
|
|
host=UNRAID_MCP_HOST,
|
|
port=UNRAID_MCP_PORT,
|
|
path="/mcp" # Standard path for MCP
|
|
)
|
|
elif UNRAID_MCP_TRANSPORT == "sse":
|
|
# Deprecated SSE transport - log warning
|
|
logger.warning("SSE transport is deprecated and may be removed in a future version. Consider switching to 'streamable-http'.")
|
|
mcp.run(
|
|
transport="sse",
|
|
host=UNRAID_MCP_HOST,
|
|
port=UNRAID_MCP_PORT,
|
|
path="/mcp" # Keep custom path for SSE
|
|
)
|
|
elif UNRAID_MCP_TRANSPORT == "stdio":
|
|
mcp.run() # Defaults to stdio
|
|
else:
|
|
logger.error(f"Unsupported MCP_TRANSPORT: {UNRAID_MCP_TRANSPORT}. Choose 'streamable-http' (recommended), 'sse' (deprecated), or 'stdio'.")
|
|
sys.exit(1)
|
|
except Exception as e:
|
|
logger.critical(f"❌ Failed to start Unraid MCP server: {e}", exc_info=True)
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_server() |