SAGE-MCP

"""
Central configuration for SAGE MCP
All settings in one place
"""

import os
from pathlib import Path


class Config:
    """Central configuration manager"""

    # Project paths
    PROJECT_ROOT = Path(__file__).parent
    LOGS_DIR = PROJECT_ROOT / "logs"

    # Ensure directories exist
    LOGS_DIR.mkdir(exist_ok=True)

    # Model configurations
    DEFAULT_MODEL = os.getenv("DEFAULT_MODEL", "auto")

    # Temperature settings per mode
    TEMPERATURES = {
        "chat": 0.7,  # Creative discussion
        "analyze": 0.3,  # Analytical precision
        "review": 0.3,  # Methodical review
        "debug": 0.2,  # Precise debugging
        "plan": 0.5,  # Balanced planning
        "test": 0.4,  # Test generation
        "refactor": 0.4,  # Code improvement
        "think": 0.8,  # Creative thinking
    }

    # Token limits (model-specific context windows)
    MAX_TOKENS = {
        "gemini-2.5-pro": 2_000_000,
        "gemini-2.5-flash": 1_000_000,
        "gemini-1.5-pro": 2_000_000,
        "gemini-1.5-flash": 1_000_000,
        "gpt-4o": 128_000,
        "gpt-4o-mini": 128_000,
        "o1": 200_000,
        "o1-mini": 128_000,
        "o3-mini": 200_000,
        "claude-3.5-sonnet": 200_000,
        "default": 100_000,
    }

    # MCP protocol limits
    MCP_PROMPT_SIZE_LIMIT = int(os.getenv("MCP_PROMPT_SIZE_LIMIT", "50000"))  # 50K chars for MCP transport

    # Conversation settings
    MAX_CONVERSATION_TURNS = int(os.getenv("MAX_CONVERSATION_TURNS", "20"))
    CONVERSATION_TIMEOUT_HOURS = int(os.getenv("CONVERSATION_TIMEOUT_HOURS", "3"))

    # Model restrictions for cost/security control
    @classmethod
    def get_model_restrictions(cls) -> dict:
        """Get model restriction patterns"""
        return {
            "openai_allowed": (
                os.getenv("OPENAI_ALLOWED_MODELS", "").split(",") if os.getenv("OPENAI_ALLOWED_MODELS") else []
            ),
            "google_allowed": (
                os.getenv("GOOGLE_ALLOWED_MODELS", "").split(",") if os.getenv("GOOGLE_ALLOWED_MODELS") else []
            ),
            "anthropic_allowed": (
                os.getenv("ANTHROPIC_ALLOWED_MODELS", "").split(",") if os.getenv("ANTHROPIC_ALLOWED_MODELS") else []
            ),
            "blocked_models": os.getenv("BLOCKED_MODELS", "").split(",") if os.getenv("BLOCKED_MODELS") else [],
            "disabled_patterns": (
                os.getenv("DISABLED_MODEL_PATTERNS", "").split(",") if os.getenv("DISABLED_MODEL_PATTERNS") else []
            ),
        }

    # Mode descriptions for help
    MODE_DESCRIPTIONS = {
        "chat": "General development discussion and Q&A",
        "analyze": "Code and architecture analysis",
        "review": "Code review for bugs, security, and quality",
        "debug": "Debug issues and find root causes",
        "plan": "Project planning and task breakdown",
        "test": "Generate comprehensive tests",
        "refactor": "Suggest code improvements",
        "think": "Deep reasoning for complex problems",
    }

    # Provider API keys (from environment)
    @classmethod
    def get_api_keys(cls) -> dict:
        """Get all configured API keys"""
        return {
            "gemini": os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY"),
            "openai": os.getenv("OPENAI_API_KEY"),
            "anthropic": os.getenv("ANTHROPIC_API_KEY"),
            "openrouter": os.getenv("OPENROUTER_API_KEY"),
            "deepseek": os.getenv("DEEPSEEK_API_KEY"),
            "custom_url": os.getenv("CUSTOM_API_URL"),
            "custom_key": os.getenv("CUSTOM_API_KEY"),
            "xai": os.getenv("XAI_API_KEY"),
        }

    # File handling settings
    MAX_FILE_SIZE = int(os.getenv("MAX_FILE_SIZE", "10000000"))  # 10MB default

    # Security settings
    ALLOWED_FILE_EXTENSIONS = {
        ".py",
        ".js",
        ".ts",
        ".jsx",
        ".tsx",
        ".java",
        ".c",
        ".cpp",
        ".h",
        ".cs",
        ".rb",
        ".go",
        ".rs",
        ".swift",
        ".kt",
        ".php",
        ".sql",
        ".html",
        ".css",
        ".scss",
        ".json",
        ".xml",
        ".yaml",
        ".yml",
        ".md",
        ".txt",
        ".log",
        ".conf",
        ".ini",
        ".toml",
        ".env",
        ".csv",
        ".sh",
        ".bash",
        ".zsh",
        ".fish",
        ".ps1",
        ".bat",
        ".cmd",
        ".png",
        ".jpg",
        ".jpeg",
        ".gif",
        ".webp",
        ".svg",  # Image support
    }

    EXCLUDED_DIRS = {
        "__pycache__",
        ".git",
        ".svn",
        ".hg",
        "node_modules",
        ".venv",
        "venv",
        ".env",
        "dist",
        "build",
        ".pytest_cache",
        ".mypy_cache",
        ".tox",
        "coverage_html_report",
    }

    # Redis configuration for conversation memory
    REDIS_URL = os.getenv("REDIS_URL", "redis://localhost:6379")
    REDIS_DB = int(os.getenv("REDIS_DB", "0"))