AutoTerminal/autoterminal/config/manager.py

import os
import json
from typing import Dict, Any

class ConfigManager:
    """配置管理器，支持配置的保存和验证"""
    
    def __init__(self, config_file: str = None):
        if config_file is None:
            # 将配置文件存储在用户主目录下的.autoterminal目录中
            home_dir = os.path.expanduser("~")
            config_dir = os.path.join(home_dir, ".autoterminal")
            # 确保目录存在
            os.makedirs(config_dir, exist_ok=True)
            self.config_file = os.path.join(config_dir, "config.json")
        else:
            self.config_file = config_file
            
        self.required_keys = ['api_key', 'base_url', 'model']
        self.default_config = {
            'base_url': 'https://api.openai.com/v1',
            'model': 'gpt-4o',
            'default_prompt': '你现在是一个终端助手,用户输入想要生成的命令,你来输出一个命令,不要任何多余的文本!',
            'max_history': 10
        }
    
    def save_config(self, config: Dict[str, Any]) -> bool:
        """保存配置到文件"""
        try:
            # 确保目录存在
            os.makedirs(os.path.dirname(self.config_file) if os.path.dirname(self.config_file) else '.', exist_ok=True)
            
            with open(self.config_file, 'w', encoding='utf-8') as f:
                json.dump(config, f, indent=2, ensure_ascii=False)
            return True
        except Exception as e:
            print(f"错误: 无法保存配置文件 {self.config_file}: {e}")
            return False
    
    def validate_config(self, config: Dict[str, Any]) -> bool:
        """验证配置是否完整"""
        for key in self.required_keys:
            if not config.get(key):
                return False
        return True
    
    def initialize_config(self) -> Dict[str, Any]:
        """初始化配置向导"""
        print("欢迎使用AutoTerminal配置向导！")
        print("请提供以下信息以完成配置：")
        
        config = self.default_config.copy()
        
        # 获取API密钥
        try:
            api_key = input("请输入您的API密钥: ").strip()
            if not api_key:
                print("错误: API密钥不能为空")
                return {}
            config['api_key'] = api_key
        except EOFError:
            print("\n配置向导已取消。")
            return {}
        except Exception as e:
            print(f"错误: 无法读取API密钥输入: {e}")
            return {}
        
        # 获取Base URL
        try:
            base_url = input(f"请输入Base URL (默认: {self.default_config['base_url']}): ").strip()
            if base_url:
                config['base_url'] = base_url
        except EOFError:
            print("\n配置向导已取消。")
            return {}
        except Exception as e:
            print(f"警告: 无法读取Base URL输入: {e}")
        
        # 获取模型名称
        try:
            model = input(f"请输入模型名称 (默认: {self.default_config['model']}): ").strip()
            if model:
                config['model'] = model
        except EOFError:
            print("\n配置向导已取消。")
            return {}
        except Exception as e:
            print(f"警告: 无法读取模型名称输入: {e}")
        
        # 保存配置
        if self.save_config(config):
            print(f"配置已保存到 {self.config_file}")
            return config
        else:
            print("配置保存失败")
            return {}
    
    def get_or_create_config(self) -> Dict[str, Any]:
        """获取现有配置或创建新配置"""
        # 尝试从文件加载配置
        if os.path.exists(self.config_file):
            try:
                with open(self.config_file, 'r', encoding='utf-8') as f:
                    config = json.load(f)
                # 验证配置
                if self.validate_config(config):
                    print("已加载现有配置")
                    return config
                else:
                    print("现有配置不完整")
            except Exception as e:
                print(f"警告: 无法读取配置文件 {self.config_file}: {e}")
        
        # 如果配置不存在或不完整，启动初始化向导
        return self.initialize_config()