easyai 1.0.2 → 1.0.3

data/lib/easyai/command/claude.rb

@@ -0,0 +1,278 @@
+require_relative '../config/config'
+require_relative '../auth/authclaude'
+require_relative '../base/update_notifier'
+
+module EasyAI
+  class Command
+    class Claude < Command
+      self.summary = '运行 Claude CLI'
+      self.description = <<-DESC
+        启动 Anthropic Claude CLI 工具。
+        
+        主要功能:
+
+            * 支持远程配置下载
+
+            * 自动配置环境变量
+
+            * 支持本地配置文件
+
+        使用示例:
+
+            $ easyai claude                      # 使用默认配置启动
+
+            $ easyai claude ./config.json        # 使用本地配置文件
+
+            $ easyai claude --verbose            # 显示详细信息
+
+            $ easyai claude --no-keychain        # 禁用密码存储
+      DESC
+
+      def self.options
+        [
+          ['--no-keychain', '禁用自动密码存储'],
+          ['--verbose', '显示详细信息（包括完整的代理和令牌）']
+        ].concat(super)
+      end
+
+      def initialize(argv)
+        @no_keychain = argv.flag?('no-keychain')
+        @verbose_mode = argv.flag?('verbose')
+        
+        super
+        
+        # 获取剩余参数
+        remaining_args = @argv.remainder!
+        
+        # 检查第一个参数是否是配置文件
+        if remaining_args.first && File.exist?(remaining_args.first) && remaining_args.first.end_with?('.json')
+          @config_file = remaining_args.shift
+        end
+        
+        # 剩余的参数传递给 claude
+        @claude_args = remaining_args
+      end
+
+      def validate!
+        super
+        help! '未找到 Claude CLI。请安装：npm install -g @anthropic-ai/claude-code' unless claude_available?
+      end
+
+      def run
+        begin
+          # 在非verbose模式下显示简洁的更新提醒
+          Base::UpdateNotifier.maybe_show_notification unless @verbose_mode
+          
+          # 美化的启动信息
+          print_header
+
+          # 首先尝试从远程下载配置
+          remote_config = nil
+
+        if @config_file
+          # 使用指定的本地配置文件
+          print_status("📁 使用本地配置", File.basename(@config_file))
+          remote_config = load_local_config(@config_file)
+          print_success("配置加载成功") if remote_config
+        else
+          # 从远程下载配置，传递选项
+          print_status("🔄 获取远程配置", "默认用户")
+          options = { no_keychain: @no_keychain }
+          remote_config = ConfigManager.download_user_config(nil, options)
+          print_success("配置加载成功") if remote_config
+        end
+
+        # 如果远程配置获取失败，回退到本地配置
+        if remote_config.nil?
+          print_warning("使用本地配置")
+          remote_config = load_local_yaml_config
+          
+          # 如果本地配置也为空，提示用户先进行设置
+          if remote_config.empty?
+            print_error("未找到有效配置")
+            puts "  请先运行: #{'easyai --setup'.yellow}"
+            exit 1
+          end
+        end
+
+        # 使用 Auth::AuthClaude 模块配置认证
+        print_status("🔐 配置认证", "Claude OAuth")
+        unless Auth::AuthClaude.configure(remote_config, nil)
+          print_error("配置认证失败")
+          exit 1
+        end
+        print_success("认证配置完成")
+
+        # 构建环境变量
+        env = build_environment(remote_config)
+        
+        # 导出环境变量到当前进程（不包括 CLAUDE_CODE_OAUTH_TOKEN）
+        export_environment_variables(env)
+        
+        # 打印分隔线
+        puts "─" * 60
+        puts "🚀 #{'Claude CLI 已启动'.green}"
+        puts "─" * 60
+        puts
+        
+        # 启动 Claude
+        exec(env, 'claude', *@claude_args)
+        
+        rescue => e
+          print_error("Claude 命令执行失败: #{e.message}")
+          puts "  请检查配置文件和网络连接"
+          puts "  错误详情: #{e.class.name}" if @verbose_mode
+          exit 1
+        end
+      end
+
+      private
+
+      def print_header
+        puts "\n╔#{'═' * 58}╗"
+        puts "║#{'EasyAI Claude CLI 启动器'.center(58)}║"
+        puts "╚#{'═' * 58}╝\n"
+      end
+
+      def print_status(icon_text, detail = nil)
+        if detail
+          puts "#{icon_text.ljust(20)} #{detail.cyan}"
+        else
+          puts icon_text
+        end
+      end
+
+      def print_success(message)
+        puts "  ✓ #{message.green}"
+      end
+
+      def print_warning(message)
+        puts "  ⚠ #{message.yellow}"
+      end
+
+      def print_error(message)
+        puts "  ✗ #{message.red}"
+      end
+
+      def claude_available?
+        # 跨平台命令检测
+        if RUBY_PLATFORM =~ /mswin|mingw|cygwin/
+          system('where claude >nul 2>&1')
+        else
+          system('which claude > /dev/null 2>&1')
+        end
+      end
+
+      def load_local_config(config_path)
+        return nil unless File.exist?(config_path)
+
+        begin
+          config_content = File.read(config_path)
+          JSON.parse(config_content)
+        rescue JSON::ParserError => e
+          puts "解析本地配置文件失败: #{e.message}".red
+          nil
+        rescue => e
+          puts "读取本地配置文件失败: #{e.message}".red
+          nil
+        end
+      end
+
+      def load_local_yaml_config
+        # 不再使用本地YAML配置文件，返回空配置
+        # claude命令主要依赖远程配置或用户指定的JSON配置文件
+        {}
+      end
+
+      def build_environment(config)
+        env = ENV.to_h
+
+        puts "\n📋 环境配置:" if @verbose_mode
+        puts "─" * 60 if @verbose_mode
+
+        # 从配置中提取环境变量 - 只设置 CLAUDE_CODE_OAUTH_TOKEN
+        if config && config['env'] && config['env']['CLAUDE_CODE_OAUTH_TOKEN']
+          env['CLAUDE_CODE_OAUTH_TOKEN'] = config['env']['CLAUDE_CODE_OAUTH_TOKEN']
+          
+          if @verbose_mode
+            # verbose 模式：显示完整的令牌和代理
+            puts "\n🔑 CLAUDE_CODE_OAUTH_TOKEN:"
+            puts "   #{config['env']['CLAUDE_CODE_OAUTH_TOKEN'].green}"
+          else
+            # 普通模式：只显示令牌预览
+            token_preview = config['env']['CLAUDE_CODE_OAUTH_TOKEN'].length > 15 ? 
+                           "#{config['env']['CLAUDE_CODE_OAUTH_TOKEN'][0..15]}..." : 
+                           config['env']['CLAUDE_CODE_OAUTH_TOKEN']
+            print_status("🔑 令牌已配置", token_preview)
+          end
+        end
+
+        # 从代理配置中提取代理设置
+        if config && config['claude_proxy']
+          proxy_urls = []
+          if config['claude_proxy']['HTTP_PROXY']
+            env['HTTP_PROXY'] = config['claude_proxy']['HTTP_PROXY']
+            env['http_proxy'] = config['claude_proxy']['HTTP_PROXY']
+            proxy_urls << config['claude_proxy']['HTTP_PROXY']
+          end
+          if config['claude_proxy']['HTTPS_PROXY'] && config['claude_proxy']['HTTPS_PROXY'] != config['claude_proxy']['HTTP_PROXY']
+            env['HTTPS_PROXY'] = config['claude_proxy']['HTTPS_PROXY']
+            env['https_proxy'] = config['claude_proxy']['HTTPS_PROXY']
+            proxy_urls << config['claude_proxy']['HTTPS_PROXY']
+          elsif config['claude_proxy']['HTTPS_PROXY']
+            env['HTTPS_PROXY'] = config['claude_proxy']['HTTPS_PROXY']
+            env['https_proxy'] = config['claude_proxy']['HTTPS_PROXY']
+          end
+          
+          if proxy_urls.any?
+            if @verbose_mode
+              # verbose 模式：显示完整的代理地址
+              puts "\n🔗 代理配置:"
+              proxy_urls.uniq.each do |url|
+                puts "   #{url.green}"
+              end
+            else
+              # 普通模式：简化代理显示，隐藏密码
+              simplified_proxies = proxy_urls.uniq.map do |url|
+                uri = URI(url) rescue nil
+                if uri && uri.password
+                  "#{uri.scheme}://***@#{uri.host}:#{uri.port}"
+                else
+                  url
+                end
+              end
+              print_status("🌐 代理已配置", simplified_proxies.join(', '))
+            end
+          end
+        end
+
+        puts "─" * 60 if @verbose_mode
+        puts if @verbose_mode
+
+        env
+      end
+
+
+      def export_environment_variables(env)
+        # 导出 CLAUDE_CODE_OAUTH_TOKEN（仅在内存中，不写入文件）
+        if env['CLAUDE_CODE_OAUTH_TOKEN']
+          ENV['CLAUDE_CODE_OAUTH_TOKEN'] = env['CLAUDE_CODE_OAUTH_TOKEN']
+          print_success("令牌已设置到环境变量") if @verbose_mode
+        end
+        
+        # 导出代理相关环境变量
+        proxy_vars = ['HTTP_PROXY', 'HTTPS_PROXY', 'http_proxy', 'https_proxy']
+        proxy_exported = false
+        proxy_vars.each do |var|
+          if env[var]
+            ENV[var] = env[var]
+            proxy_exported = true
+          end
+        end
+        print_success("代理已设置到环境变量") if proxy_exported && @verbose_mode
+      end
+
+
+    end
+  end
+end