"""
AI配置文件
用于配置智谱AI API相关设置
已重构为使用配置文件而非环境变量
"""

from typing import Optional
from server.config.server_config import get_server_config


class AIConfig:
    """AI配置类（兼容性包装器）"""

    def __init__(self):
        """初始化AI配置"""
        # 获取服务器配置
        self.server_config = get_server_config()
        # 加载配置属性
        self._load_config()

    def _load_config(self):
        """从配置文件加载AI设置"""
        ai_config = self.server_config.get_ai_config()

        # 设置属性（保持向后兼容）
        self.api_key = ai_config.get("api_key", "")
        self.model = ai_config.get("model", "glm-4-flash")
        self.max_tokens = ai_config.get("max_tokens", 1024)
        self.temperature = ai_config.get("temperature", 0.7)
        self.top_p = ai_config.get("top_p", 0.9)
        self.max_context_length = ai_config.get("max_context_length", 10)
        self.context_timeout = ai_config.get("context_timeout", 3600)
        self.enable_group_chat = ai_config.get("enable_group_chat", True)
        self.enable_private_chat = ai_config.get("enable_private_chat", True)
        self.auto_reply = ai_config.get("auto_reply", True)
        self.trigger_keywords = ai_config.get("trigger_keywords",
            ["ai", "人工智能", "助手", "机器人", "智能", "问答"])
        self.require_at_mention = ai_config.get("require_at_mention", False)

    def is_enabled(self) -> bool:
        """检查AI功能是否可用"""
        return self.server_config.is_ai_enabled()

    def get_api_key(self) -> Optional[str]:
        """获取API密钥"""
        return self.server_config.get_ai_api_key()

    def set_api_key(self, api_key: str):
        """设置API密钥"""
        success = self.server_config.set_ai_api_key(api_key)
        if success:
            self.api_key = api_key
        return success
    
    def set_model(self, model_name: str) -> bool:
        """
        设置AI模型

        Args:
            model_name: 模型名称

        Returns:
            设置是否成功
        """
        success = self.server_config.set_ai_model(model_name)
        if success:
            self.model = model_name
        return success

    def get_available_models(self) -> list:
        """获取可用模型列表"""
        ai_config = self.server_config.get_ai_config()
        return ai_config.get("available_models", [
            "glm-4-flash", "glm-4", "glm-4-plus",
            "glm-4-air", "glm-4-airx", "glm-4-long"
        ])

    def to_dict(self) -> dict:
        """转换为字典"""
        return {
            "api_key_set": bool(self.api_key),
            "model": self.model,
            "max_tokens": self.max_tokens,
            "temperature": self.temperature,
            "top_p": self.top_p,
            "max_context_length": self.max_context_length,
            "context_timeout": self.context_timeout,
            "enable_group_chat": self.enable_group_chat,
            "enable_private_chat": self.enable_private_chat,
            "auto_reply": self.auto_reply,
            "trigger_keywords": self.trigger_keywords,
            "require_at_mention": self.require_at_mention,
            "available_models": self.get_available_models()
        }

    def reload_from_config(self):
        """从配置文件重新加载设置"""
        self._load_config()


# 全局配置实例
ai_config = AIConfig()


def get_ai_config() -> AIConfig:
    """获取AI配置实例"""
    return ai_config


def print_ai_setup_guide():
    """打印AI设置指南"""
    print("\n" + "="*60)
    print("🤖 智谱AI设置指南")
    print("="*60)
    print("1. 访问智谱AI开放平台: https://bigmodel.cn/")
    print("2. 注册账号并登录")
    print("3. 在控制台创建API密钥")
    print("4. 设置环境变量:")
    print("   export ZHIPU_API_KEY='your_api_key_here'")
    print("5. 或者在代码中直接设置:")
    print("   ai_config.set_api_key('your_api_key_here')")
    print("\n💡 推荐模型:")
    print("   • glm-4-flash: 免费模型，速度快，适合聊天")
    print("   • glm-4: 标准模型，平衡性能和质量")
    print("   • glm-4-plus: 增强模型，质量更高")
    print("   • glm-4-long: 长文本模型，支持更长上下文")
    print("\n🔧 安装官方SDK (推荐):")
    print("   pip install zhipuai")
    print("="*60)


def setup_ai_from_env():
    """从环境变量设置AI配置"""
    config = get_ai_config()
    
    # 检查环境变量
    if not config.api_key:
        print("⚠️  未设置智谱AI API密钥")
        print("💡 请设置环境变量 ZHIPU_API_KEY 来启用AI功能")
        print("   例如: export ZHIPU_API_KEY='your-api-key-here'")
        return False
    
    print(f"✅ 智谱AI配置已加载，模型: {config.model}")
    return True


def print_ai_setup_guide():
    """打印AI设置指南"""
    print("""
🤖 AI功能设置指南:

1. 获取智谱AI API密钥:
   - 访问 https://open.bigmodel.cn/
   - 注册账号并获取API密钥

2. 配置API密钥:
   - 编辑配置文件 config/server_config.yaml
   - 在 ai.api_key 字段中填入您的API密钥
   - 或使用命令行工具设置: python -m server.config.setup_ai

3. 重启服务器以应用配置

4. 使用AI功能:
   - 群聊中@AI或使用AI关键词
   - 私聊中直接发送消息
   - 使用 /ai status 查看AI状态
   - 使用 /ai help 获取帮助

💡 配置文件位置: config/server_config.yaml
📝 配置模板: config/server_config.template.yaml
""")


if __name__ == "__main__":
    # 测试配置
    config = get_ai_config()
    print("AI配置信息:")
    for key, value in config.to_dict().items():
        print(f"  {key}: {value}")
    
    if not config.is_enabled():
        print_ai_setup_guide()