easyai 1.2.1 → 1.3.0

data/lib/easyai/command/gemini.rb

@@ -46,12 +46,8 @@ module EasyAI
       private
 
       def gemini_available?
-        # 跨平台命令检测
-        if RUBY_PLATFORM =~ /mswin|mingw|cygwin/
-          system('where gemini >nul 2>&1')
-        else
-          system('which gemini > /dev/null 2>&1')
-        end
+        # 使用统一的跨平台命令检测
+        EasyAI::CrossPlatform.command_exists?('gemini')
       end
     end
   end

data/lib/easyai/command/gpt.rb

@@ -127,12 +127,8 @@ module EasyAI
       private
 
       def gpt_available?
-        # 跨平台命令检测
-        if RUBY_PLATFORM =~ /mswin|mingw|cygwin/
-          system('where openai >nul 2>&1')
-        else
-          system('which openai > /dev/null 2>&1')
-        end
+        # 使用统一的跨平台命令检测
+        EasyAI::CrossPlatform.command_exists?('openai')
       end
 
       def load_local_config(config_file)
@@ -218,11 +214,21 @@ module EasyAI
       def mask_url(url)
         return url unless url
 
-        if url =~ /^(https?:\/\/)([^:]+):([^@]+)@(.+)$/
-          protocol, user, pass, rest = $1, $2, $3, $4
-          masked_pass = '*' * [pass.length, 8].min
-          "#{protocol}#{user}:#{masked_pass}@#{rest}"
-        else
+        begin
+          uri = URI(url)
+
+          # 隐藏IP地址的前两段，显示后两段
+          masked_host = uri.host
+          if uri.host =~ /^(\d+)\.(\d+)\.(\d+)\.(\d+)$/
+            masked_host = "*.*.#{$3}.#{$4}"
+          end
+
+          if uri.password
+            "#{uri.scheme}://***@#{masked_host}:#{uri.port}"
+          else
+            "#{uri.scheme}://#{masked_host}:#{uri.port}"
+          end
+        rescue URI::InvalidURIError
           url
         end
       end

data/lib/easyai/command/setup.rb

@@ -0,0 +1,494 @@
+require_relative '../config/easyai_config'
+require_relative '../base/system_keychain'
+require 'colored2'
+require 'json'
+require 'open3'
+
+module EasyAI
+  class Command
+    class Setup < Command
+      self.summary = '初始化配置环境'
+      self.description = <<-DESC
+      初始化 EasyAI 配置环境，自动下载并设置配置仓库。
+
+      主要功能：
+          * 自动下载配置仓库
+          * 验证配置完整性
+          * 解密配置文件（如需要）
+          * 显示可用用户列表
+          * 管理密码存储
+
+      使用示例：
+          $ easyai setup                    # 标准初始化
+          $ easyai setup --force            # 强制重新下载配置
+          $ easyai setup --branch dev       # 使用开发分支
+          $ easyai setup --verify           # 仅验证配置状态
+          $ easyai setup --list-users       # 列出可用用户
+      DESC
+
+      def self.options
+        [
+          ['--force', '强制重新下载配置仓库'],
+          ['--branch=BRANCH', '指定配置仓库分支 (默认: master)'],
+          ['--verify', '仅验证现有配置，不进行修改'],
+          ['--list-users', '列出所有可用用户'],
+          ['--clean', '清理所有临时解密文件'],
+          ['--no-password', '跳过密码设置'],
+        ].concat(super)
+      end
+
+      def initialize(argv)
+        @force = argv.flag?('force')
+        @branch = argv.option('branch') || 'master'
+        @verify_only = argv.flag?('verify')
+        @list_users = argv.flag?('list-users')
+        @clean = argv.flag?('clean')
+        @skip_password = argv.flag?('no-password')
+        super
+
+        @config_dir = File.expand_path('~/.easyai')
+        @repo_dir = File.join(@config_dir, 'EasyAISetting')
+      end
+
+      def run
+        show_welcome_banner
+
+        if @verify_only
+          verify_configuration
+          return
+        end
+
+        if @clean
+          cleanup_files
+          return
+        end
+
+        if @list_users
+          show_available_users
+          return
+        end
+
+        # 执行主要设置流程
+        perform_setup
+
+        # 显示完成信息
+        show_completion_info
+      end
+
+      private
+
+      def show_welcome_banner
+        puts
+        puts "🚀 " + "EasyAI 配置环境初始化".green.bold
+        puts "=" * 60
+        puts
+      end
+
+      def perform_setup
+        step_counter = 0
+
+        # 步骤1: 检查并创建目录
+        step_counter += 1
+        print_step(step_counter, "检查配置目录")
+        ensure_directories
+
+        # 步骤2: 下载或更新配置仓库
+        step_counter += 1
+        print_step(step_counter, "下载配置仓库")
+        if @force || !repo_exists?
+          download_repo
+        else
+          update_repo
+        end
+
+        # 步骤3: 处理加密文件
+        step_counter += 1
+        print_step(step_counter, "处理配置文件")
+        handle_encrypted_files unless @skip_password
+
+        # 步骤4: 验证配置
+        step_counter += 1
+        print_step(step_counter, "验证配置完整性")
+        verify_basic_config
+
+        # 步骤5: 显示可用用户
+        step_counter += 1
+        print_step(step_counter, "加载用户列表")
+        show_available_users(compact: true)
+      end
+
+      def print_step(number, description)
+        puts "\n" + "步骤 #{number}:".cyan + " #{description}"
+        puts "-" * 40
+      end
+
+      def ensure_directories
+        unless Dir.exist?(@config_dir)
+          FileUtils.mkdir_p(@config_dir)
+          puts "  ✓ 创建配置目录: #{@config_dir}"
+        else
+          puts "  ✓ 配置目录已存在: #{@config_dir}"
+        end
+      end
+
+      def repo_exists?
+        Dir.exist?(@repo_dir) && Dir.exist?(File.join(@repo_dir, '.git'))
+      end
+
+      def download_repo
+        if repo_exists? && @force
+          puts "  正在删除现有仓库..."
+          FileUtils.rm_rf(@repo_dir)
+        end
+
+        puts "  正在从 Gitee 下载配置仓库 (#{@branch} 分支)..."
+
+        # Gitee 不需要代理，临时清除代理环境变量
+        output = nil
+        success = false
+
+        if Gem.win_platform?
+          # Windows: 保存并清除环境变量
+          old_env = {}
+          ['HTTP_PROXY', 'HTTPS_PROXY', 'http_proxy', 'https_proxy'].each do |key|
+            old_env[key] = ENV[key]
+            ENV.delete(key)
+          end
+
+          begin
+            cmd = "git clone --depth 1 --branch #{@branch} #{EasyAIConfig::REPO_URL} #{@repo_dir} 2>&1"
+            puts "  执行命令: #{cmd}" if ENV['EASYAI_DEBUG']
+            output = `#{cmd}`
+            success = $?.success?
+          ensure
+            # 恢复环境变量
+            old_env.each { |k, v| ENV[k] = v if v }
+          end
+        else
+          # Unix/Linux/macOS: 使用 env -u
+          cmd = "env -u HTTP_PROXY -u HTTPS_PROXY -u http_proxy -u https_proxy git clone --depth 1 --branch #{@branch} #{EasyAIConfig::REPO_URL} #{@repo_dir} 2>&1"
+          puts "  执行命令: #{cmd}" if ENV['EASYAI_DEBUG']
+          output = `#{cmd}`
+          success = $?.success?
+        end
+
+        if success
+          puts "  ✓ 配置仓库下载成功"
+        else
+          puts "  ✗ 下载失败: #{output}".red
+          exit 1
+        end
+      end
+
+      def update_repo
+        puts "  检查配置仓库更新..."
+
+        Dir.chdir(@repo_dir) do
+          # 获取当前分支
+          current_branch = `git branch --show-current`.chomp
+
+          if current_branch != @branch
+            puts "  切换到 #{@branch} 分支..."
+            # 不使用代理
+            if Gem.win_platform?
+              # Windows: 临时清除环境变量
+              old_env = {}
+              ['HTTP_PROXY', 'HTTPS_PROXY', 'http_proxy', 'https_proxy'].each do |key|
+                old_env[key] = ENV[key]
+                ENV.delete(key)
+              end
+
+              begin
+                system("git fetch origin #{@branch} > nul 2>&1")
+              ensure
+                old_env.each { |k, v| ENV[k] = v if v }
+              end
+            else
+              system("env -u HTTP_PROXY -u HTTPS_PROXY -u http_proxy -u https_proxy git fetch origin #{@branch} > /dev/null 2>&1")
+            end
+            system("git checkout #{@branch} > /dev/null 2>&1")
+          end
+
+          # 拉取更新（不使用代理）
+          output = nil
+          if Gem.win_platform?
+            # Windows: 临时清除环境变量
+            old_env = {}
+            ['HTTP_PROXY', 'HTTPS_PROXY', 'http_proxy', 'https_proxy'].each do |key|
+              old_env[key] = ENV[key]
+              ENV.delete(key)
+            end
+
+            begin
+              output = `git pull --force 2>&1`
+            ensure
+              old_env.each { |k, v| ENV[k] = v if v }
+            end
+          else
+            output = `env -u HTTP_PROXY -u HTTPS_PROXY -u http_proxy -u https_proxy git pull --force 2>&1`
+          end
+
+          if $?.success?
+            if output.include?("Already up to date")
+              puts "  ✓ 配置已是最新版本"
+            else
+              puts "  ✓ 配置已更新到最新版本"
+            end
+          else
+            puts "  ⚠️ 更新失败，使用现有配置"
+          end
+        end
+      end
+
+      def handle_encrypted_files
+        encrypted_files = Dir.glob(File.join(@repo_dir, '**/*.encrypted'))
+
+        if encrypted_files.empty?
+          puts "  ✓ 未发现加密文件"
+          return
+        end
+
+        puts "  发现 #{encrypted_files.length} 个加密文件"
+
+        # 获取密码
+        password = get_setup_password
+        return unless password
+
+        # 测试密码
+        if test_password(password, encrypted_files.first)
+          puts "  ✓ 密码验证成功"
+
+          # 保存到钥匙串
+          if should_save_password?
+            if Base::SystemKeychain.store_password(password)
+              puts "  ✓ 密码已保存到系统钥匙串"
+            end
+          end
+        else
+          puts "  ✗ 密码验证失败".red
+          puts "  提示: 配置文件将在使用时解密"
+        end
+      end
+
+      def get_setup_password
+        # 优先使用环境变量
+        if ENV['EASYAI_TEST_PASSWORD']
+          puts "  使用环境变量中的密码"
+          return ENV['EASYAI_TEST_PASSWORD']
+        end
+
+        # 检查钥匙串
+        stored_password = Base::SystemKeychain.get_stored_password
+        if stored_password && !stored_password.empty?
+          puts "  使用系统钥匙串中的密码"
+          return stored_password
+        end
+
+        # 提示用户输入
+        puts "  配置文件已加密，请输入解密密码"
+        print "  密码: "
+
+        begin
+          require 'io/console'
+          password = STDIN.noecho(&:gets)&.chomp
+          puts
+          password
+        rescue
+          nil
+        end
+      end
+
+      def test_password(password, test_file)
+        begin
+          require_relative '../base/file_crypto'
+          key = Base::FileCrypto.generate_key(password)
+          content = File.read(test_file)
+          Base::FileCrypto.aes_128_ecb_decrypt(key, content)
+          true
+        rescue
+          false
+        end
+      end
+
+      def should_save_password?
+        print "  是否保存密码到系统钥匙串？(y/n): "
+        response = STDIN.gets&.chomp&.downcase
+        response == 'y' || response == 'yes'
+      end
+
+      def verify_basic_config
+        required_files = [
+          'index.json.encrypted',
+          'jps_client_config.json.encrypted'
+        ]
+
+        missing_files = []
+        required_files.each do |file|
+          path = File.join(@repo_dir, file)
+          if File.exist?(path)
+            puts "  ✓ #{file}"
+          else
+            missing_files << file
+            puts "  ✗ #{file} (缺失)".red
+          end
+        end
+
+        if missing_files.empty?
+          puts "  ✓ 核心配置文件完整"
+        else
+          puts "  ⚠️ 缺少 #{missing_files.length} 个配置文件".yellow
+        end
+      end
+
+      def show_available_users(compact: false)
+        # 初始化配置
+        EasyAIConfig.initialize(verbose: false)
+
+        # 获取 index 配置
+        index_config = EasyAIConfig.get_config('index', verbose: false)
+
+        unless index_config
+          puts "  ✗ 无法加载用户配置".red
+          return
+        end
+
+        if compact
+          user_count = count_users(index_config)
+          puts "  ✓ 发现 #{user_count} 个可用用户配置"
+        else
+          display_users_table(index_config)
+        end
+      rescue => e
+        puts "  ✗ 加载用户列表失败: #{e.message}".red
+      end
+
+      def count_users(config)
+        count = 0
+
+        if config.is_a?(Hash)
+          %w[claude gemini gpt].each do |tool|
+            next unless config[tool]
+
+            if tool == "claude" && config[tool].is_a?(Hash)
+              config[tool].each do |_auth_type, users|
+                count += users.keys.length if users.is_a?(Hash)
+              end
+            elsif config[tool].is_a?(Hash)
+              count += config[tool].keys.length
+            end
+          end
+        end
+
+        count
+      end
+
+      def display_users_table(config)
+        puts "\n可用用户列表:"
+        puts "=" * 60
+
+        %w[claude gemini gpt].each do |tool|
+          next unless config[tool]
+
+          puts "\n#{tool.upcase}:"
+
+          if tool == "claude" && config[tool].is_a?(Hash)
+            config[tool].each do |auth_type, users|
+              next unless users.is_a?(Hash)
+              puts "  #{auth_type}:"
+              users.each do |name, _file|
+                puts "    - #{name}"
+              end
+            end
+          elsif config[tool].is_a?(Hash)
+            config[tool].each do |name, _file|
+              puts "  - #{name}"
+            end
+          end
+        end
+      end
+
+      def verify_configuration
+        puts "\n验证配置状态..."
+        puts "=" * 60
+
+        # 检查目录
+        puts "\n目录检查:"
+        puts "  配置目录: #{@config_dir} - " + (Dir.exist?(@config_dir) ? "✓".green : "✗".red)
+        puts "  仓库目录: #{@repo_dir} - " + (repo_exists? ? "✓".green : "✗".red)
+
+        # 检查 Git 状态
+        if repo_exists?
+          puts "\nGit 状态:"
+          Dir.chdir(@repo_dir) do
+            branch = `git branch --show-current`.chomp
+            puts "  当前分支: #{branch}"
+
+            status = `git status --short`
+            if status.empty?
+              puts "  工作区: 干净 ✓".green
+            else
+              puts "  工作区: 有修改 ⚠️".yellow
+            end
+          end
+        end
+
+        # 检查配置文件
+        puts "\n配置文件:"
+        verify_basic_config
+
+        # 检查密码
+        puts "\n密码状态:"
+        if ENV['EASYAI_TEST_PASSWORD']
+          puts "  环境变量密码: ✓".green
+        elsif Base::SystemKeychain.get_stored_password
+          puts "  系统钥匙串密码: ✓".green
+        else
+          puts "  未设置密码 ⚠️".yellow
+        end
+
+        # 尝试加载用户
+        show_available_users(compact: true)
+      end
+
+      def cleanup_files
+        puts "\n清理临时文件..."
+        puts "=" * 60
+
+        # 清理解密的 JSON 文件
+        json_files = Dir.glob(File.join(@repo_dir, '**/*.json'))
+        cleanup_count = 0
+
+        json_files.each do |file|
+          encrypted_file = "#{file}.encrypted"
+          if File.exist?(encrypted_file)
+            FileUtils.rm_f(file)
+            cleanup_count += 1
+            puts "  删除: #{File.basename(file)}"
+          end
+        end
+
+        if cleanup_count > 0
+          puts "\n  ✓ 清理了 #{cleanup_count} 个临时文件".green
+        else
+          puts "  ✓ 没有需要清理的文件".green
+        end
+      end
+
+      def show_completion_info
+        puts "\n" + "=" * 60
+        puts "✅ " + "配置初始化完成！".green.bold
+        puts
+        puts "您可以开始使用以下命令："
+        puts "  • easyai claude".cyan + "    - 启动 Claude"
+        puts "  • easyai gemini".cyan + "    - 启动 Gemini"
+        puts "  • easyai gpt".cyan + "       - 启动 GPT"
+        puts
+        puts "其他有用的命令："
+        puts "  • easyai setup --verify".cyan + "      - 验证配置状态"
+        puts "  • easyai setup --list-users".cyan + "  - 查看可用用户"
+        puts "  • easyai setup --clean".cyan + "       - 清理临时文件"
+        puts
+      end
+    end
+  end
+end