mirror of
https://github.com/jmagar/unraid-mcp.git
synced 2026-03-23 12:39:24 -07:00
4b43c47091
settings.py: drop update_temperature, update_time, update_api, connect_sign_in, connect_sign_out, setup_remote_access, enable_dynamic_remote_access, update_ssh — all 8 reference mutations confirmed absent from Unraid API v4.29.2. Keep update + configure_ups. info.py: drop update_server (updateServerIdentity not in Mutation type) and update_ssh (duplicate of removed settings action). MUTATIONS is now empty; DESTRUCTIVE_ACTIONS is now an empty set. notifications.py: drop create_unique (notifyIfUnique not in Mutation type). Tests: remove corresponding test classes, add parametrized regression tests asserting removed actions are not in each tool's Literal type, update KNOWN_DESTRUCTIVE and _DESTRUCTIVE_TEST_CASES in safety audit, update schema coverage assertions. 858 tests passing, 0 failures.
94 lines
3.2 KiB
Python
94 lines
3.2 KiB
Python
"""System settings and UPS mutations.
|
|
|
|
Provides the `unraid_settings` tool with 2 actions for updating system
|
|
configuration and UPS monitoring.
|
|
"""
|
|
|
|
from typing import Any, Literal, get_args
|
|
|
|
from fastmcp import FastMCP
|
|
|
|
from ..config.logging import logger
|
|
from ..core.client import make_graphql_request
|
|
from ..core.exceptions import ToolError, tool_error_handler
|
|
|
|
|
|
MUTATIONS: dict[str, str] = {
|
|
"update": """
|
|
mutation UpdateSettings($input: JSON!) {
|
|
updateSettings(input: $input) { restartRequired values warnings }
|
|
}
|
|
""",
|
|
"configure_ups": """
|
|
mutation ConfigureUps($config: UPSConfigInput!) {
|
|
configureUps(config: $config)
|
|
}
|
|
""",
|
|
}
|
|
|
|
DESTRUCTIVE_ACTIONS = {
|
|
"configure_ups",
|
|
}
|
|
ALL_ACTIONS = set(MUTATIONS)
|
|
|
|
SETTINGS_ACTIONS = Literal[
|
|
"configure_ups",
|
|
"update",
|
|
]
|
|
|
|
if set(get_args(SETTINGS_ACTIONS)) != ALL_ACTIONS:
|
|
_missing = ALL_ACTIONS - set(get_args(SETTINGS_ACTIONS))
|
|
_extra = set(get_args(SETTINGS_ACTIONS)) - ALL_ACTIONS
|
|
raise RuntimeError(
|
|
f"SETTINGS_ACTIONS and ALL_ACTIONS are out of sync. "
|
|
f"Missing from Literal: {_missing or 'none'}. Extra in Literal: {_extra or 'none'}"
|
|
)
|
|
|
|
|
|
def register_settings_tool(mcp: FastMCP) -> None:
|
|
"""Register the unraid_settings tool with the FastMCP instance."""
|
|
|
|
@mcp.tool()
|
|
async def unraid_settings(
|
|
action: SETTINGS_ACTIONS,
|
|
confirm: bool = False,
|
|
settings_input: dict[str, Any] | None = None,
|
|
ups_config: dict[str, Any] | None = None,
|
|
) -> dict[str, Any]:
|
|
"""Update Unraid system settings and UPS configuration.
|
|
|
|
Actions:
|
|
update - Update system settings (requires settings_input dict)
|
|
configure_ups - Configure UPS monitoring (requires ups_config dict, confirm=True)
|
|
"""
|
|
if action not in ALL_ACTIONS:
|
|
raise ToolError(f"Invalid action '{action}'. Must be one of: {sorted(ALL_ACTIONS)}")
|
|
|
|
if action in DESTRUCTIVE_ACTIONS and not confirm:
|
|
raise ToolError(f"Action '{action}' is destructive. Set confirm=True to proceed.")
|
|
|
|
with tool_error_handler("settings", action, logger):
|
|
logger.info(f"Executing unraid_settings action={action}")
|
|
|
|
if action == "update":
|
|
if settings_input is None:
|
|
raise ToolError("settings_input is required for 'update' action")
|
|
data = await make_graphql_request(MUTATIONS["update"], {"input": settings_input})
|
|
return {"success": True, "action": "update", "data": data.get("updateSettings")}
|
|
|
|
if action == "configure_ups":
|
|
if ups_config is None:
|
|
raise ToolError("ups_config is required for 'configure_ups' action")
|
|
data = await make_graphql_request(
|
|
MUTATIONS["configure_ups"], {"config": ups_config}
|
|
)
|
|
return {
|
|
"success": True,
|
|
"action": "configure_ups",
|
|
"result": data.get("configureUps"),
|
|
}
|
|
|
|
raise ToolError(f"Unhandled action '{action}' — this is a bug")
|
|
|
|
logger.info("Settings tool registered successfully")
|