easyai 1.0.2 → 1.0.4

data/lib/easyai/command/clean.rb

@@ -0,0 +1,453 @@
+require 'json'
+require 'yaml'
+require 'fileutils'
+require 'etc'
+require 'open3'
+
+module EasyAI
+  class Command
+    class Clean < Command
+      self.summary = '清理AI工具配置信息'
+      self.description = <<-DESC
+        清理指定AI工具的所有配置信息。
+        
+        支持清理:
+
+            * Claude 相关配置
+
+            * Gemini 相关配置
+
+            * OpenAI 相关配置
+
+        使用示例:
+
+            $ easyai clean                       # 清理Claude配置（默认）
+
+            $ easyai clean gemini                # 清理Gemini配置
+
+            $ easyai clean openai                # 清理OpenAI配置
+
+            $ easyai clean all                   # 清理所有配置
+      DESC
+
+      def self.options
+        [
+          ['--force', '强制清理，不询问确认'],
+          ['--dry-run', '预览将要清理的项目，不执行实际清理'],
+        ].concat(super)
+      end
+
+      def initialize(argv)
+        @tool_name = argv.shift_argument || 'claude'
+        @force = argv.flag?('force')
+        @dry_run = argv.flag?('dry-run')
+        super
+      end
+
+      def validate!
+        super
+        valid_tools = %w[claude gemini openai all]
+        unless valid_tools.include?(@tool_name.downcase)
+          help! "不支持的工具名称: #{@tool_name}。支持的工具: #{valid_tools.join(', ')}"
+        end
+        @tool_name = @tool_name.downcase
+      end
+
+      def run
+        puts "🧹 准备清理 #{@tool_name == 'all' ? '所有AI工具' : @tool_name.upcase} 的配置信息"
+        puts
+        
+        # 预览模式或确认
+        if @dry_run
+          puts "📋 预览模式 - 以下项目将被清理："
+          preview_cleanup
+          return
+        end
+        
+        unless @force
+          puts "⚠️  警告：此操作将删除以下配置信息："
+          preview_cleanup
+          puts
+          print "确认继续清理？(y/N): "
+          confirmation = STDIN.gets.chomp.downcase
+          unless confirmation == 'y' || confirmation == 'yes'
+            puts "❌ 清理操作已取消"
+            return
+          end
+        end
+        
+        puts "🧹 开始清理配置..."
+        perform_cleanup
+        puts "✅ 清理完成"
+      end
+
+      private
+
+      def preview_cleanup
+        case @tool_name
+        when 'claude'
+          preview_claude_cleanup
+        when 'gemini'
+          preview_gemini_cleanup  
+        when 'openai'
+          preview_openai_cleanup
+        when 'all'
+          preview_claude_cleanup
+          preview_gemini_cleanup
+          preview_openai_cleanup
+        end
+      end
+
+      def perform_cleanup
+        case @tool_name
+        when 'claude'
+          clean_claude
+        when 'gemini'
+          clean_gemini
+        when 'openai'
+          clean_openai
+        when 'all'
+          clean_claude
+          clean_gemini
+          clean_openai
+        end
+      end
+
+      def preview_claude_cleanup
+        puts "📌 Claude 配置（由 easyai claude 命令设置的项目）："
+        
+        # 本地配置文件 - easyai 自己的配置，不是 authclaude.rb 设置的
+        config_file = File.expand_path('~/.easyai/config.yml')
+        if File.exist?(config_file)
+          config = YAML.load_file(config_file) rescue {}
+          if config && config['claude_token']
+            puts "  • 本地配置文件: #{config_file} (claude_token)"
+          end
+        end
+        
+        # Claude JSON 配置文件 - authclaude.rb 中 update_claude_json 设置的
+        claude_file = File.expand_path('~/.claude.json')
+        if File.exist?(claude_file) && has_claude_json_config?(claude_file)
+          puts "  • Claude配置文件: #{claude_file} (由远程配置合并的内容)"
+        end
+        
+        # Shell 配置文件中的代理别名 - authclaude.rb 中 configure_proxy 设置的
+        shell_files = get_shell_files
+        shell_files.each do |shell_file|
+          if File.exist?(shell_file) && has_claude_proxy_config?(shell_file)
+            puts "  • Shell代理配置: #{shell_file} (claude_proxy, unclaude_proxy 别名)"
+          end
+        end
+        
+        # Keychain 认证信息 - authclaude.rb 中 configure_keychain 设置的
+        if RUBY_PLATFORM.include?('darwin')
+          keychain_entries = get_claude_keychain_entries
+          keychain_entries.each do |service_name|
+            puts "  • macOS Keychain: #{service_name} (Claude认证信息)"
+          end
+        end
+        
+        puts
+      end
+
+      def preview_gemini_cleanup
+        puts "📌 Gemini 配置："
+        
+        # 本地配置文件
+        config_file = File.expand_path('~/.easyai/config.yml')
+        if File.exist?(config_file)
+          puts "  • 本地配置文件: #{config_file} (gemini_token)"
+        end
+        
+        puts
+      end
+
+      def preview_openai_cleanup
+        puts "📌 OpenAI 配置："
+        
+        # 本地配置文件
+        config_file = File.expand_path('~/.easyai/config.yml')
+        if File.exist?(config_file)
+          puts "  • 本地配置文件: #{config_file} (openai_token)"
+        end
+        
+        puts
+      end
+
+      def clean_claude
+        puts "🧹 清理Claude配置..."
+        
+        # 清理本地配置文件中的Claude令牌
+        clean_local_config('claude_token')
+        
+        # 清理 ~/.claude.json
+        clean_claude_json
+        
+        # 清理Shell配置文件
+        clean_shell_config_claude
+        
+        # 清理Keychain
+        clean_keychain_claude
+      end
+
+      def clean_gemini
+        puts "🧹 清理Gemini配置..."
+        clean_local_config('gemini_token')
+      end
+
+      def clean_openai
+        puts "🧹 清理OpenAI配置..."
+        clean_local_config('openai_token')
+      end
+
+      def clean_local_config(key)
+        config_file = File.expand_path('~/.easyai/config.yml')
+        return unless File.exist?(config_file)
+        
+        begin
+          config = YAML.load_file(config_file) || {}
+          if config.key?(key)
+            config.delete(key)
+            File.write(config_file, config.to_yaml)
+            puts "  ✓ 已清理本地配置文件中的 #{key}"
+            
+            # 如果配置文件为空，删除整个配置目录
+            if config.empty?
+              config_dir = File.dirname(config_file)
+              FileUtils.rm_rf(config_dir)
+              puts "  ✓ 已删除空配置目录: #{config_dir}"
+            end
+          end
+        rescue => e
+          puts "  ✗ 清理本地配置失败: #{e.message}"
+        end
+      end
+
+      def clean_claude_json
+        claude_file = File.expand_path('~/.claude.json')
+        return unless File.exist?(claude_file)
+        
+        begin
+          # 不是完全删除文件，而是恢复到只包含默认字段的状态
+          # 因为 Claude Code 会自动重新创建基本配置
+          content = JSON.parse(File.read(claude_file))
+          
+          # 保留默认字段，删除由 authclaude.rb update_claude_json 方法添加的配置
+          default_fields = %w[installMethod autoUpdates userID fallbackAvailableWarningThreshold projects tipsHistory]
+          cleaned_content = content.select { |key, _| default_fields.include?(key) }
+          
+          if cleaned_content != content
+            File.write(claude_file, JSON.pretty_generate(cleaned_content))
+            puts "  ✓ 已清理Claude配置文件中的远程配置内容: #{claude_file}"
+          end
+        rescue JSON::ParserError => e
+          puts "  ✗ 解析Claude配置文件失败: #{e.message}"
+        rescue => e
+          puts "  ✗ 清理Claude配置文件失败: #{e.message}"
+        end
+      end
+
+      def clean_shell_config_claude
+        shell_files = get_shell_files
+        
+        shell_files.each do |shell_file|
+          next unless File.exist?(shell_file)
+          
+          begin
+            content = File.read(shell_file)
+            original_content = content.dup
+            lines = content.split("\n", -1)
+            lines.pop if lines.last == ""
+            
+            # 只清理由 authclaude.rb 添加的代理配置（claude_proxy 和 unclaude_proxy）
+            # 不清理 CLAUDE_CODE_OAUTH_TOKEN，因为那不是 authclaude.rb 设置的
+            lines = remove_claude_proxy_config(lines, shell_file)
+            
+            if lines.join("\n") + "\n" != original_content
+              File.write(shell_file, lines.join("\n") + "\n")
+              puts "  ✓ 已清理Shell代理配置: #{shell_file}"
+            end
+          rescue => e
+            puts "  ✗ 清理Shell配置失败 #{shell_file}: #{e.message}"
+          end
+        end
+      end
+
+      def clean_keychain_claude
+        return unless RUBY_PLATFORM.include?('darwin')
+        
+        account_name = Etc.getlogin || ENV['USER'] || ENV['LOGNAME']
+        keychain_entries = get_claude_keychain_entries
+        
+        if keychain_entries.empty?
+          # 检查默认的 Claude Code-credentials
+          service_name = "Claude Code-credentials"
+          cmd = ["security", "delete-generic-password", "-a", account_name, "-s", service_name]
+          stdout, stderr, status = Open3.capture3(*cmd)
+          
+          if status.success?
+            puts "  ✓ 已清理Keychain: #{service_name}"
+          elsif !stderr.include?("could not be found")
+            puts "  ✗ 清理Keychain失败: #{stderr}"
+          end
+        else
+          # 清理所有找到的 Claude 相关条目
+          keychain_entries.each do |service_name|
+            cmd = ["security", "delete-generic-password", "-a", account_name, "-s", service_name]
+            stdout, stderr, status = Open3.capture3(*cmd)
+            
+            if status.success?
+              puts "  ✓ 已清理Keychain: #{service_name}"
+            elsif !stderr.include?("could not be found")
+              puts "  ✗ 清理Keychain失败 #{service_name}: #{stderr}"
+            end
+          end
+        end
+      end
+
+      def has_claude_json_config?(claude_file)
+        return false unless File.exist?(claude_file)
+        
+        begin
+          content = JSON.parse(File.read(claude_file))
+          # 检查是否有非默认的配置项（排除 Claude Code 自动生成的默认字段）
+          default_fields = %w[installMethod autoUpdates userID fallbackAvailableWarningThreshold projects tipsHistory]
+          has_custom_config = content.keys.any? { |key| !default_fields.include?(key) }
+          return has_custom_config
+        rescue JSON::ParserError, StandardError
+          return false
+        end
+      end
+
+      def has_claude_proxy_config?(shell_file)
+        return false unless File.exist?(shell_file)
+        
+        content = File.read(shell_file)
+        content.include?('claude_proxy') || content.include?('unclaude_proxy')
+      end
+
+      def get_claude_keychain_entries
+        return [] unless RUBY_PLATFORM.include?('darwin')
+        
+        entries = []
+        
+        # 检查各种可能的 Claude 相关 keychain 条目
+        possible_services = [
+          "Claude Code-credentials",
+          # 从 authclaude.rb 中的 configure_keychain 方法可以看到，service_name 来自配置
+          # 但我们不知道具体的服务名，所以需要通过搜索来查找
+        ]
+        
+        account_name = Etc.getlogin || ENV['USER'] || ENV['LOGNAME']
+        
+        # 搜索所有包含 "claude" 的 keychain 条目
+        search_cmd = ["security", "dump-keychain", "-d"]
+        
+        begin
+          stdout, stderr, status = Open3.capture3(*search_cmd)
+          if status.success?
+            # 解析输出，查找 Claude 相关条目
+            stdout.lines.each do |line|
+              if line.include?("claude") && line.include?("svce")
+                # 提取服务名称
+                match = line.match(/"([^"]*claude[^"]*)"/)
+                entries << match[1] if match
+              end
+            end
+          end
+        rescue => e
+          # 如果搜索失败，回退到检查已知的服务名
+          possible_services.each do |service_name|
+            cmd = ["security", "find-generic-password", "-a", account_name, "-s", service_name]
+            if system(*cmd, out: File::NULL, err: File::NULL)
+              entries << service_name
+            end
+          end
+        end
+        
+        entries.uniq
+      end
+
+      def get_shell_files
+        shell_files = []
+        current_shell = ENV['SHELL']&.split('/')&.last || 'bash'
+        
+        case current_shell
+        when 'zsh'
+          shell_files << File.expand_path("~/.zshrc")
+        when 'bash'
+          bash_profile = File.expand_path("~/.bash_profile")
+          bashrc = File.expand_path("~/.bashrc")
+          shell_files << (File.exist?(bash_profile) ? bash_profile : bashrc)
+        when 'fish'
+          fish_config = File.expand_path("~/.config/fish/config.fish")
+          shell_files << fish_config
+        else
+          shell_files << File.expand_path("~/.profile")
+        end
+        
+        # 总是添加 ~/.profile 作为备份
+        profile_file = File.expand_path("~/.profile")
+        shell_files << profile_file unless shell_files.include?(profile_file)
+        
+        shell_files
+      end
+
+      def remove_claude_proxy_config(lines, shell_file)
+        # 只清理 claude_proxy 和 unclaude_proxy 相关配置
+        # 这是 authclaude.rb 中 configure_proxy 方法添加的内容
+        if shell_file.include?("config.fish")
+          # Fish shell - 移除代理相关的别名/函数
+          lines.reject! { |line|
+            line.match(/^alias\s+claude_proxy=/) ||
+            line.match(/^alias\s+unclaude_proxy=/) ||
+            line.match(/^function\s+(claude_proxy|unclaude_proxy)/)
+          }
+          
+          # 移除函数块
+          in_function = false
+          lines.select do |line|
+            if line.match(/^function\s+(claude_proxy|unclaude_proxy)/)
+              in_function = true
+              false
+            elsif in_function && line.strip == "end"
+              in_function = false
+              false
+            elsif in_function
+              false
+            else
+              true
+            end
+          end
+        else
+          # Bash/zsh - 移除代理相关的别名/函数
+          lines.reject! { |line|
+            line.match(/^alias\s+(claude_proxy|unclaude_proxy)=/) ||
+            line.match(/^(claude_proxy|unclaude_proxy)\(\)/)
+          }
+          
+          # 移除函数块
+          in_function = false
+          function_depth = 0
+          lines.select do |line|
+            if line.match(/^(claude_proxy|unclaude_proxy)\(\)/)
+              in_function = true
+              function_depth = 0
+              false
+            elsif in_function
+              function_depth += 1 if line.include?("{")
+              if line.include?("}")
+                function_depth -= 1
+                in_function = false if function_depth <= 0
+              end
+              false
+            else
+              true
+            end
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/easyai/command/gemini.rb

@@ -0,0 +1,58 @@
+module EasyAI
+  class Command
+    class Gemini < Command
+      self.summary = '运行 Gemini CLI'
+      self.description = <<-DESC
+        启动 Google Gemini CLI 工具。
+        
+        主要功能:
+
+            * 自动配置 API 密钥
+
+            * 支持交互式对话
+
+            * 透传所有参数
+
+        使用示例:
+
+            $ easyai gemini                      # 启动交互式 Gemini
+
+            $ easyai gemini chat                 # 开始聊天会话
+
+            $ easyai gemini --help               # 查看 Gemini 帮助
+      DESC
+
+      def initialize(argv)
+        super
+        @gemini_args = @argv.remainder!
+      end
+
+      def validate!
+        super
+        help! '未找到 Gemini CLI。请安装：npm install -g @google/gemini-cli' unless gemini_available?
+      end
+
+      def run
+        # 直接使用环境变量，不再依赖配置文件
+        env = ENV.to_h
+        
+        # 如果环境变量中已经设置了 API KEY，直接使用
+        # 用户可以通过 export GOOGLE_API_KEY=xxx 来设置
+        
+        puts "正在运行: gemini #{@gemini_args.join(' ')}".blue if verbose?
+        exec(env, 'gemini', *@gemini_args)
+      end
+
+      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
+      end
+    end
+  end
+end

data/lib/easyai/command/gpt.rb

@@ -0,0 +1,58 @@
+module EasyAI
+  class Command
+    class GPT < Command
+      self.summary = '运行 OpenAI CLI'
+      self.description = <<-DESC
+        启动 OpenAI GPT CLI 工具。
+        
+        主要功能:
+
+            * 自动配置 API 密钥
+
+            * 支持 API 调用
+
+            * 透传所有参数
+
+        使用示例:
+
+            $ easyai gpt                         # 启动 OpenAI CLI
+
+            $ easyai gpt api chat.completions    # 调用 Chat API
+
+            $ easyai gpt --help                  # 查看 OpenAI 帮助
+      DESC
+
+      def initialize(argv)
+        super
+        @gpt_args = @argv.remainder!
+      end
+
+      def validate!
+        super
+        help! '未找到 OpenAI CLI。请安装：pip install openai' unless gpt_available?
+      end
+
+      def run
+        # 直接使用环境变量，不再依赖配置文件
+        env = ENV.to_h
+        
+        # 如果环境变量中已经设置了 API KEY，直接使用
+        # 用户可以通过 export OPENAI_API_KEY=xxx 来设置
+        
+        puts "正在运行: openai #{@gpt_args.join(' ')}".blue if verbose?
+        exec(env, 'openai', *@gpt_args)
+      end
+
+      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
+      end
+    end
+  end
+end