TrendRadar/mcp_server/tools/config_mgmt.py

"""
配置管理工具

实现配置查询和管理功能。
"""

from typing import Dict, Optional, Any, TypedDict

from ..services.data_service import DataService
from ..utils.validators import validate_config_section
from ..utils.errors import MCPError


class ErrorInfo(TypedDict, total=False):
    """错误信息结构"""
    code: str
    message: str
    suggestion: str


class ConfigResult(TypedDict):
    """配置查询结果 - success 字段必需，其他字段可选"""
    success: bool
    config: Optional[Dict[str, Any]]
    section: Optional[str]
    error: Optional[ErrorInfo]


class ConfigManagementTools:
    """配置管理工具类"""

    def __init__(self, project_root: str = None):
        """
        初始化配置管理工具

        Args:
            project_root: 项目根目录
        """
        self.data_service = DataService(project_root)

    def get_current_config(self, section: Optional[str] = None) -> ConfigResult:
        """
        获取当前系统配置

        Args:
            section: 配置节 - all/crawler/push/keywords/weights，默认all

        Returns:
            配置字典

        Example:
            >>> tools = ConfigManagementTools()
            >>> result = tools.get_current_config(section="crawler")
            >>> print(result['crawler']['platforms'])
        """
        try:
            # 参数验证
            section = validate_config_section(section)

            # 获取配置
            config = self.data_service.get_current_config(section=section)

            return ConfigResult(
                success=True,
                config=config,
                section=section,
                error=None
            )

        except MCPError as e:
            return ConfigResult(
                success=False,
                config=None,
                section=None,
                error=e.to_dict()
            )
        except Exception as e:
            return ConfigResult(
                success=False,
                config=None,
                section=None,
                error={"code": "INTERNAL_ERROR", "message": str(e), "suggestion": "请查看服务日志获取详细信息"}
            )