easyai 1.0.2 → 1.0.4

data/lib/easyai/config/config.rb

@@ -0,0 +1,550 @@
+require 'json'
+require 'tmpdir'
+require 'fileutils'
+require 'open3'
+require_relative '../base/file_crypto'
+require_relative '../base/system_keychain'
+require_relative '../auth/jpslogin'
+
+module EasyAI
+  class ConfigManager
+    REPO_URL = "https://gitee.com/goodtools/EasyAISetting.git"
+    EASYAI_DIR = File.expand_path('~/.easyai')
+    CONFIG_REPO_DIR = File.join(EASYAI_DIR, 'EasyAISetting')
+    
+    # 类变量初始化
+    @@no_keychain = false
+    
+    # 管理配置仓库：如果存在则更新，不存在则下载
+    def self.manage_config_repo
+      # 确保 ~/.easyai 目录存在
+      FileUtils.mkdir_p(EASYAI_DIR) unless Dir.exist?(EASYAI_DIR)
+      
+      if Dir.exist?(CONFIG_REPO_DIR)
+        update_config_repo
+      else
+        download_config_repo
+      end
+    end
+    
+    # 更新现有的配置仓库
+    def self.update_config_repo
+      begin
+        # 检查是否是有效的 git 仓库
+        unless Dir.exist?(File.join(CONFIG_REPO_DIR, '.git'))
+          FileUtils.rm_rf(CONFIG_REPO_DIR)
+          return download_config_repo
+        end
+        
+        # 切换到仓库目录并拉取更新
+        Dir.chdir(CONFIG_REPO_DIR) do
+          # 重置本地更改
+          reset_success = system('git reset --hard HEAD', out: File::NULL, err: File::NULL)
+          unless reset_success
+            FileUtils.rm_rf(CONFIG_REPO_DIR)
+            return download_config_repo
+          end
+          
+          # 拉取最新更新，捕获错误信息
+          stdout, stderr, status = Open3.capture3('git pull origin master')
+          
+          unless status.success?
+            puts "⚠️  配置更新失败，使用缓存配置" if stderr && !stderr.empty?
+            FileUtils.rm_rf(CONFIG_REPO_DIR)
+            return download_config_repo
+          end
+        end
+        
+        return true
+        
+      rescue => e
+        FileUtils.rm_rf(CONFIG_REPO_DIR) if Dir.exist?(CONFIG_REPO_DIR)
+        return download_config_repo
+      end
+    end
+    
+    # 下载配置仓库到固定位置
+    def self.download_config_repo
+      puts "正在下载配置仓库到 #{CONFIG_REPO_DIR}..."
+      
+      begin
+        # 克隆仓库到固定位置，捕获错误信息
+        stdout, stderr, status = Open3.capture3("git clone --quiet #{REPO_URL} #{CONFIG_REPO_DIR}")
+        
+        unless status.success?
+          puts "✗ 配置仓库下载失败"
+          puts "  错误信息: #{stderr.chomp}" if stderr && !stderr.empty?
+          puts "  请检查网络连接、git 安装状态"
+          return false
+        end
+        
+        puts "✓ 配置仓库下载成功"
+        return true
+        
+      rescue => e
+        puts "✗ 下载配置仓库失败: #{e.message}"
+        return false
+      end
+    end
+    
+    # 从本地配置仓库加载配置
+    def self.load_from_local_repo(user_name = nil, options = {})
+      # 首先管理配置仓库
+      return nil unless manage_config_repo
+      
+      # 设置全局选项
+      @@no_keychain = options[:no_keychain] || false
+      
+      puts "正在从本地配置仓库加载配置..."
+      
+      begin
+        # 检查 index.json 文件，支持加密版本
+        index_file = File.join(CONFIG_REPO_DIR, "index.json")
+        encrypted_index_file = File.join(CONFIG_REPO_DIR, "index.json.encrypted")
+        
+        # 优先尝试加密的 index 文件
+        if File.exist?(encrypted_index_file)
+          users = parse_encrypted_index_file(encrypted_index_file)
+        elsif File.exist?(index_file)
+          users = parse_index_file(index_file)
+        else
+          puts "未找到 index.json 或 index.json.encrypted，使用默认配置"
+          config_file = File.join(CONFIG_REPO_DIR, "claude_setting.json")
+          encrypted_config_file = File.join(CONFIG_REPO_DIR, "claude_setting.json.encrypted")
+          
+          if File.exist?(encrypted_config_file)
+            return load_encrypted_config_file(encrypted_config_file)
+          elsif File.exist?(config_file)
+            return load_config_file(config_file)
+          else
+            puts "✗ 未找到配置文件"
+            return nil
+          end
+        end
+        
+        return nil if users.nil?
+        
+        if users.empty?
+          puts "✗ index.json 中未找到用户"
+          return nil
+        end
+        
+        # 获取用户名：优先使用提供的用户名，否则通过 JPS 登录获取
+        if user_name
+          selected_user = user_name
+        else
+          # 统一使用 JPS 登录获取用户名
+          selected_user = get_username_from_jps
+          if selected_user.nil? || selected_user.strip.empty?
+            puts "✗ JPS 登录失败，无法获取用户名"
+            return nil
+          end
+        end
+        
+        # 查找用户配置文件
+        config_filename = find_user_config(users, selected_user.strip)
+        
+        if config_filename.nil?
+          puts "✗ 用户 '#{selected_user}' 未找到"
+          puts "  可用用户: #{users.keys.join(', ')}"
+          return nil
+        end
+        
+        # 加载配置文件，支持加密版本
+        config_filename += '.json' unless config_filename.end_with?('.json')
+        config_file = File.join(CONFIG_REPO_DIR, config_filename)
+        encrypted_config_file = File.join(CONFIG_REPO_DIR, "#{config_filename}.encrypted")
+        
+        puts "✓ 正在为 #{selected_user} 加载配置"
+        
+        # 优先尝试加密文件
+        if File.exist?(encrypted_config_file)
+          load_encrypted_config_file(encrypted_config_file)
+        elsif File.exist?(config_file)
+          load_config_file(config_file)
+        else
+          puts "✗ 配置文件未找到: #{config_filename}"
+          return nil
+        end
+        
+      rescue => e
+        puts "✗ 加载本地配置失败: #{e.message}"
+        nil
+      end
+    end
+    
+    def self.download_user_config(user_name = nil, options = {})
+      # 优先尝试使用本地配置仓库
+      config = load_from_local_repo(user_name, options)
+      return config if config
+      
+      # 如果本地仓库失败，回退到临时下载方式
+      puts "本地配置仓库不可用，使用临时下载..."
+      download_user_config_temp(user_name, options)
+    end
+    
+    # 原有的临时下载方式（作为备用方案）
+    def self.download_user_config_temp(user_name = nil, options = {})
+      puts "正在获取配置文件..."
+      
+      # 设置全局选项
+      @@no_keychain = options[:no_keychain] || false
+      
+      begin
+        # 创建临时目录
+        temp_dir = Dir.mktmpdir("EasyAISetting")
+        
+        # 克隆仓库
+        clone_cmd = "git clone --depth 1 --quiet #{REPO_URL} #{temp_dir}"
+        success = system(clone_cmd, out: File::NULL, err: File::NULL)
+        
+        unless success
+          puts "✗ 配置文件获取失败"
+          puts "  请检查网络连接、git 安装，或使用本地配置文件"
+          cleanup_temp_dir(temp_dir)
+          return nil
+        end
+        
+        puts "✓ 配置文件获取成功"
+        
+        # 检查 index.json 文件，支持加密版本
+        index_file = File.join(temp_dir, "index.json")
+        encrypted_index_file = File.join(temp_dir, "index.json.encrypted")
+        
+        # 优先尝试加密的 index 文件
+        if File.exist?(encrypted_index_file)
+          users = parse_encrypted_index_file(encrypted_index_file)
+        elsif File.exist?(index_file)
+          users = parse_index_file(index_file)
+        else
+          puts "未找到 index.json 或 index.json.encrypted，使用默认配置"
+          config_file = File.join(temp_dir, "claude_setting.json")
+          encrypted_config_file = File.join(temp_dir, "claude_setting.json.encrypted")
+          
+          if File.exist?(encrypted_config_file)
+            config_content = load_encrypted_config_file(encrypted_config_file)
+          elsif File.exist?(config_file)
+            config_content = load_config_file(config_file)
+          else
+            puts "✗ 未找到配置文件"
+            cleanup_temp_dir(temp_dir)
+            return nil
+          end
+          
+          cleanup_temp_dir(temp_dir)
+          return config_content
+        end
+        
+        return nil if users.nil?
+        
+        if users.empty?
+          puts "✗ index.json 中未找到用户"
+          cleanup_temp_dir(temp_dir)
+          return nil
+        end
+        
+        # 获取用户名：优先使用提供的用户名，否则通过 JPS 登录获取
+        if user_name
+          selected_user = user_name
+        else
+          # 统一使用 JPS 登录获取用户名
+          selected_user = get_username_from_jps
+          if selected_user.nil? || selected_user.strip.empty?
+            puts "✗ JPS 登录失败，无法获取用户名"
+            cleanup_temp_dir(temp_dir)
+            return nil
+          end
+        end
+        
+        # 查找用户配置文件
+        config_filename = find_user_config(users, selected_user.strip)
+        
+        if config_filename.nil?
+          puts "✗ 用户 '#{selected_user}' 未找到"
+          puts "  可用用户: #{users.keys.join(', ')}"
+          cleanup_temp_dir(temp_dir)
+          return nil
+        end
+        
+        # 加载配置文件，支持加密版本
+        config_filename += '.json' unless config_filename.end_with?('.json')
+        config_file = File.join(temp_dir, config_filename)
+        encrypted_config_file = File.join(temp_dir, "#{config_filename}.encrypted")
+        
+        puts "✓ 正在为 #{selected_user} 加载配置"
+        
+        # 优先尝试加密文件
+        if File.exist?(encrypted_config_file)
+          config_content = load_encrypted_config_file(encrypted_config_file)
+        elsif File.exist?(config_file)
+          config_content = load_config_file(config_file)
+        else
+          puts "✗ 配置文件未找到: #{config_filename}"
+          cleanup_temp_dir(temp_dir)
+          return nil
+        end
+        
+        cleanup_temp_dir(temp_dir)
+        config_content
+        
+      rescue => e
+        puts "✗ 配置获取失败: #{e.message}"
+        cleanup_temp_dir(temp_dir) if temp_dir
+        nil
+      end
+    end
+    
+    private
+    
+    def self.cleanup_temp_dir(temp_dir)
+      return unless temp_dir && Dir.exist?(temp_dir)
+      FileUtils.rm_rf(temp_dir)
+    end
+    
+    def self.parse_index_file(index_file)
+      begin
+        index_content = File.read(index_file)
+        JSON.parse(index_content)
+      rescue JSON::ParserError => e
+        puts "✗ 解析 index.json 失败: #{e.message}"
+        nil
+      rescue => e
+        puts "✗ 读取 index.json 失败: #{e.message}"
+        nil
+      end
+    end
+    
+    # 通过 JPS 登录获取用户名
+    def self.get_username_from_jps      
+      begin
+        jps_login = Auth::JPSLogin.new
+        
+        if jps_login.login
+          username = jps_login.get_username
+          if username && !username.empty?
+            puts "👤 用户: #{username}"
+            return username
+          end
+        end
+      rescue => e
+        puts "❌ JPS 登录失败: #{e.message}"
+      end
+      
+      return nil
+    end
+    
+    def self.find_user_config(users, user_input_clean)
+      users.each do |name, filename|
+        if name.downcase == user_input_clean.downcase
+          return filename
+        end
+      end
+      nil
+    end
+    
+    def self.load_config_file(config_file)
+      begin
+        config_content = File.read(config_file)
+        JSON.parse(config_content)
+      rescue JSON::ParserError => e
+        puts "✗ 解析配置文件失败: #{e.message}"
+        nil
+      rescue => e
+        puts "✗ 读取配置文件失败: #{e.message}"
+        nil
+      end
+    end
+    
+    def self.load_encrypted_config_file(encrypted_config_file)
+      puts "正在解密配置文件..."
+      
+      # 使用新的密码验证机制
+      result = get_and_validate_password(encrypted_config_file, "请输入解密密码: ")
+      return nil unless result
+      
+      begin
+        # 解析 JSON
+        JSON.parse(result[:content])
+      rescue JSON::ParserError => e
+        puts "✗ 解析解密后的配置文件失败: #{e.message}"
+        puts "文件可能已损坏或不是有效的JSON格式"
+        nil
+      end
+    end
+    
+    def self.parse_encrypted_index_file(encrypted_index_file)
+      puts "正在解密 index.json..."
+      
+      # 使用新的密码验证机制
+      result = get_and_validate_password(encrypted_index_file, "请输入解密密码: ")
+      return nil unless result
+      
+      begin
+        # 解析 JSON
+        JSON.parse(result[:content])
+      rescue JSON::ParserError => e
+        puts "✗ 解析解密后的 index.json 失败: #{e.message}"
+        puts "文件可能已损坏或不是有效的JSON格式"
+        nil
+      end
+    end
+    
+    # 智能密码获取和验证：包含错误处理和重试机制
+    def self.get_and_validate_password(encrypted_file_path, prompt, max_retries = 3)
+      retries = 0
+      
+      loop do
+        # 获取密码（系统存储或用户输入）
+        password_info = get_decryption_password_basic(prompt, retries > 0)
+        return nil unless password_info
+        
+        # 尝试解密验证密码
+        begin
+          decrypted_content = decrypt_file_content(encrypted_file_path, password_info[:password])
+          
+          # 解密成功，保存密码（如果不是从系统获取且未禁用存储）
+          unless password_info[:from_system] || @@no_keychain
+            if Base::SystemKeychain.store_password(password_info[:password])
+              puts "✓ 密码已安全存储，下次使用将自动获取"
+            end
+          end
+          
+          return { content: decrypted_content, password_info: password_info }
+          
+        rescue => e
+          retries += 1
+          
+          # 分析错误类型并提供针对性提示
+          error_type = analyze_decryption_error(e)
+          
+          # 处理系统密码错误
+          if password_info[:from_system]
+            puts "✗ 系统存储的密码无效"
+            puts "  错误详情：#{e.message}" if error_type[:show_details]
+            puts "  错误原因：#{error_type[:reason]}"
+            puts "正在清除无效的系统存储密码..."
+            Base::SystemKeychain.delete_stored_password
+            puts "✓ 已清除系统存储的无效密码"
+            puts "请重新输入正确的解密密码"
+          else
+            puts "✗ 解密失败"
+            puts "  错误详情：#{e.message}" if error_type[:show_details]
+            puts "  错误原因：#{error_type[:reason]}"
+            puts "  建议：#{error_type[:suggestion]}"
+          end
+          
+          # 检查是否达到最大重试次数
+          if retries >= max_retries
+            puts "\n已达到最大重试次数 (#{max_retries})"
+            puts "可能的解决方案："
+            puts "  1. 确认密码是否正确"
+            puts "  2. 检查文件是否完整下载"
+            puts "  3. 联系管理员确认密码是否已更改"
+            
+            if ask_user_continue?
+              retries = 0  # 重置重试次数，继续尝试
+              puts "\n继续尝试解密..."
+            else
+              puts "用户选择退出解密"
+              return nil
+            end
+          else
+            remaining = max_retries - retries
+            puts "还可以重试 #{remaining} 次 (#{retries}/#{max_retries})"
+            puts "提示：按 Ctrl+C 可随时退出\n"
+          end
+        end
+      end
+    end
+    
+    # 基础密码获取：先从系统获取，失败则提示用户输入
+    def self.get_decryption_password_basic(prompt, is_retry = false)
+      # 如果是重试，或者禁用了密码存储，直接提示用户输入
+      if is_retry || @@no_keychain
+        if @@no_keychain
+          puts "已禁用自动密码存储，请手动输入密码"
+        end
+        password = Base::FileCrypto.get_password(prompt)
+        return password ? { password: password, from_system: false } : nil
+      end
+      
+      # 首先尝试从系统获取存储的密码
+      stored_password = Base::SystemKeychain.get_stored_password
+      
+      if stored_password && !stored_password.empty?
+        puts "使用系统存储的密码进行解密..."
+        return { password: stored_password, from_system: true }
+      end
+      
+      # 如果没有存储的密码，提示用户输入
+      password = Base::FileCrypto.get_password(prompt)
+      return password ? { password: password, from_system: false } : nil
+    end
+    
+    # 分析解密错误类型，提供针对性建议
+    def self.analyze_decryption_error(error)
+      case error.message
+      when /bad decrypt/i, /wrong final block length/i, /invalid padding/i
+        {
+          reason: "密码不正确或文件已损坏",
+          suggestion: "请确认密码是否正确，或检查文件完整性",
+          show_details: false
+        }
+      when /not supported by device/i
+        {
+          reason: "系统不支持此加密算法或文件格式错误", 
+          suggestion: "请检查文件是否为正确的加密文件",
+          show_details: true
+        }
+      when /no such file/i, /cannot open/i
+        {
+          reason: "文件不存在或无法访问",
+          suggestion: "请检查文件路径和权限",
+          show_details: false
+        }
+      when /invalid base64/i
+        {
+          reason: "文件格式错误或已损坏",
+          suggestion: "请重新下载配置文件",
+          show_details: false
+        }
+      else
+        {
+          reason: "未知的解密错误",
+          suggestion: "请尝试重新下载文件或联系技术支持",
+          show_details: true
+        }
+      end
+    end
+    
+    # 询问用户是否继续尝试
+    def self.ask_user_continue?
+      puts "\n是否继续尝试解密？"
+      puts "  y/yes - 继续尝试"
+      puts "  n/no  - 退出解密 (默认)"
+      print "> "
+      
+      begin
+        response = STDIN.gets&.chomp&.downcase
+        return ['y', 'yes'].include?(response)
+      rescue Interrupt
+        puts "\n\n用户中断操作"
+        return false
+      rescue => e
+        puts "无法读取用户输入: #{e.message}"
+        return false
+      end
+    end
+    
+    def self.decrypt_file_content(encrypted_file_path, password)
+      # 生成密钥
+      key = Base::FileCrypto.generate_key(password)
+      
+      # 读取加密文件内容
+      encrypted_content = File.read(encrypted_file_path)
+      
+      # 解密内容
+      Base::FileCrypto.aes_128_ecb_decrypt(key, encrypted_content)
+    end
+  end
+end

data/lib/easyai/version.rb

@@ -1,3 +1,3 @@
 module EasyAI
-  VERSION = '1.0.2'
+  VERSION = '1.0.4'
 end

data/lib/easyai.rb

@@ -1,75 +1,87 @@
-require 'claide'
-require 'yaml'
-require 'fileutils'
-require 'colored2'
 require 'easyai/version'
+require 'easyai/config/config'
+require 'easyai/command'
+require 'easyai/command/claude'
+require 'easyai/command/gemini'
+require 'easyai/command/gpt'
+require 'easyai/command/utils'
+require 'easyai/command/clean'
+require 'easyai/command/update'
+require 'easyai/base/version_checker'
 
 module EasyAI
-  
-  class Command < CLAide::Command
-    self.abstract_command = true
-    self.command = 'easyai'
-    self.version = VERSION
-    self.description = '简化 AI 命令行工具的 CLI 包装器，支持 Claude、Gemini 和 GPT'
-    
-    def self.options
-      [
-        ['--config', '显示配置文件路径'],
-        ['--setup', '交互式设置配置']
-      ].concat(super)
-    end
-    
-    def initialize(argv)
-      @config_flag = argv.flag?('config')
-      @setup_flag = argv.flag?('setup')
-      super
-    end
-    
-    def run
-      if @config_flag
-        puts config_file_path
-        return
-      end
+  # EasyAI 应用程序主类
+  class EasyAIApp
+    def run(argv)
+      # 解析参数以确定是否显示启动标志
+      require 'claide'
+      require 'colored2'
+      coerced_argv = CLAide::ARGV.coerce(argv)
       
-      if @setup_flag
-        setup_config
-        return
+      # 如果不是帮助或版本查询，显示启动标志
+      unless coerced_argv.flag?('help') || coerced_argv.flag?('version')
+        show_banner
       end
       
-      super
+      # 在运行命令前进行版本检查
+      check_version_before_run(argv)
+      
+      # 运行命令
+      EasyAI::Command.run(argv)
     end
     
     private
     
-    def config_file_path
-      @config_file_path ||= File.expand_path('~/.easyai/config.yml')
+    # 显示启动标志
+    def show_banner
+      require 'colored2'
+      
+      # 简化的标题
+      title = "EasyAI v#{VERSION}"
+      box_width = 60
+      
+      # 美化的启动标志
+      puts
+      puts "╔#{'═' * box_width}╗".cyan
+      puts "║#{title.center(box_width).green}║".cyan
+      puts "╚#{'═' * box_width}╝".cyan
+      puts
     end
     
-    def setup_config
-      puts "正在设置 EasyAI 配置...".green
-      FileUtils.mkdir_p(File.dirname(config_file_path))
+    # 检查版本和更新
+    def check_version_before_run(argv)
+      # 解析参数以确定是否需要跳过版本检查
+      require 'claide'
+      require 'colored2'
+      coerced_argv = CLAide::ARGV.coerce(argv)
       
-      config = {}
+      # 如果是以下情况，跳过版本检查：
+      # 1. 请求帮助信息 (--help)
+      # 2. 显示版本信息 (--version)
+      # 3. 主动检查更新 (--check-update)
+      skip_check = coerced_argv.flag?('help') || 
+                   coerced_argv.flag?('version') || 
+                   coerced_argv.flag?('check-update')
       
-      print "Claude API 令牌 (ANTHROPIC_API_KEY): "
-      config['claude_token'] = STDIN.gets.chomp
+      if ENV['EASYAI_DEBUG']
+        puts "📊 版本检查状态：".cyan
+        puts "  • 当前版本: v#{EasyAI::VERSION}".green
+        puts "  • 跳过检查: #{skip_check ? '是（--help/--version）'.yellow : '否'.cyan}"
+      end
       
-      print "Gemini API 令牌 (GOOGLE_API_KEY): "
-      config['gemini_token'] = STDIN.gets.chomp
+      return if skip_check
       
-      print "OpenAI API 令牌 (OPENAI_API_KEY): "
-      config['openai_token'] = STDIN.gets.chomp
+      # 强制更新检查（会阻塞执行）
+      Base::VersionChecker.check_force_update!
       
-      print "HTTP 代理 (可选): "
-      proxy = STDIN.gets.chomp
-      config['http_proxy'] = proxy unless proxy.empty?
+      # 异步更新检查（不阻塞执行）
+      Base::VersionChecker.check_async
       
-      File.write(config_file_path, config.to_yaml)
-      puts "配置已保存到 #{config_file_path}".green
+      puts "✅ #{'版本检查完成'.green}" if ENV['EASYAI_DEBUG']
+    rescue => e
+      # 版本检查失败不应该影响命令执行
+      puts "⚠️  版本检查出错: #{e.message}".yellow
+      puts e.backtrace.join("\n") if ENV['EASYAI_DEBUG']
     end
   end
-end
-
-require 'easyai/claude'
-require 'easyai/gemini'
-require 'easyai/gpt'
+end