Excel MCP Server

"""
配置管理模块
处理配置文件加载和验证
"""

import json
import os
import logging
from pathlib import Path
from typing import Dict, Any, List


class ConfigManager:
    """配置管理器"""
    
    def __init__(self, config_path: str = None):
        self.config_path = config_path or os.getenv('EXCEL_MCP_CONFIG', 'config.json')
        self.config = self._load_config()
        
    def _load_config(self) -> Dict[str, Any]:
        """加载配置文件"""
        try:
            with open(self.config_path, 'r', encoding='utf-8') as f:
                config = json.load(f)
            return config
        except FileNotFoundError:
            logging.warning(f"配置文件 {self.config_path} 不存在，使用默认配置")
            return self._get_default_config()
        except json.JSONDecodeError as e:
            logging.error(f"配置文件解析错误: {e}")
            return self._get_default_config()
    
    def _get_default_config(self) -> Dict[str, Any]:
        """获取默认配置"""
        return {
            "server": {
                "name": "excel-mcp",
                "version": "1.0.0",
                "transport": "stdio"
            },
            "excel": {
                "max_file_size": "100MB",
                "supported_formats": [".xlsx", ".xls", ".csv"],
                "backup_enabled": True,
                "temp_dir": "./temp"
            },
            "security": {
                "allowed_paths": ["./data", "~/Documents", "~/Desktop"],
                "forbidden_extensions": [".exe", ".bat", ".cmd"],
                "max_memory_usage": "512MB"
            },
            "logging": {
                "level": "INFO",
                "file": "./logs/excel_mcp.log",
                "max_size": "10MB",
                "backup_count": 5
            }
        }
    
    def get(self, key: str, default: Any = None) -> Any:
        """获取配置值"""
        keys = key.split('.')
        value = self.config
        
        for k in keys:
            if isinstance(value, dict) and k in value:
                value = value[k]
            else:
                return default
        
        return value
    
    def get_allowed_paths(self) -> List[str]:
        """获取允许访问的路径列表"""
        paths = self.get('security.allowed_paths', [])
        # 展开用户目录
        expanded_paths = []
        for path in paths:
            expanded_path = Path(path).expanduser().resolve()
            expanded_paths.append(str(expanded_path))
        return expanded_paths
    
    def get_supported_formats(self) -> List[str]:
        """获取支持的文件格式"""
        return self.get('excel.supported_formats', ['.xlsx', '.xls'])
    
    def is_backup_enabled(self) -> bool:
        """是否启用备份"""
        return self.get('excel.backup_enabled', True)
    
    def get_temp_dir(self) -> Path:
        """获取临时目录"""
        temp_dir = Path(self.get('excel.temp_dir', './temp'))
        temp_dir.mkdir(exist_ok=True)
        return temp_dir