easyai 1.7.0 → 2.1.0

data/lib/easyai/command/clean.rb

@@ -1,451 +1,144 @@
-require 'json'
-require 'yaml'
+# frozen_string_literal: true
+
 require 'fileutils'
-require 'etc'
-require 'open3'
-require_relative '../base/system_info'
+require 'colored2'
+require_relative '../command'
 
 module EasyAI
   class Command
+    # 清理 AI CLI 自身在本地产生的缓存与配置目录。
+    #
+    # 设计要点（详见 docs/设计文档.md §2.6）：
+    #   - 仅清理两家 AI CLI（claude / codex）的缓存路径
+    #   - 不会触碰 ~/.easyai/config.json（重置 EasyAI 自身配置请用 easyai setup --reset）
+    #   - --force 跳过确认；--dry-run 仅打印
+    #   - 删除时遇到 Errno::EACCES 继续删余下项；最终如有失败项以退出码 1 退出
     class Clean < Command
-      self.summary = '清理AI工具配置信息'
+      self.summary = '清理 AI CLI 自身缓存（不影响 EasyAI 配置）'
       self.description = <<-DESC
-        清理指定AI工具的所有配置信息。
-        
-        支持清理:
+清理 claude / codex 各自在本地产生的缓存目录与配置文件。
 
-            * Claude 相关配置
+支持范围：
+    * claude → ~/.claude、~/.claude.json、~/.config/claude
+    * codex  → ~/.codex、~/.config/codex
+    * all    → 上述全部
 
-            * Gemini 相关配置
+使用示例：
 
-            * OpenAI 相关配置
+    $ easyai clean                  # 默认清理 claude
+    $ easyai clean codex            # 清理 codex
+    $ easyai clean all              # 清理三家 AI CLI 缓存
+    $ easyai clean all --dry-run    # 仅打印将要删除的路径
+    $ easyai clean all --force      # 跳过确认提示
 
-        使用示例:
+注意：本命令不会动 ~/.easyai/config.json；如需重置 EasyAI 自身配置请使用：
 
-            $ easyai clean                       # 清理Claude配置（默认）
+    $ easyai setup --reset
+DESC
 
-            $ easyai clean gemini                # 清理Gemini配置
+      VALID_TOOLS = %w[claude codex all].freeze
 
-            $ easyai clean openai                # 清理OpenAI配置
+      # AI CLI 自身的缓存路径表（不含 ~/.easyai/config.json）
+      CACHE_PATHS = {
+        'claude' => %w[~/.claude ~/.claude.json ~/.config/claude],
+        'codex'  => %w[~/.codex ~/.config/codex]
+      }.freeze
 
-            $ easyai clean all                   # 清理所有配置
-      DESC
+      self.arguments = [CLAide::Argument.new('TOOL', false)]
 
       def self.options
         [
-          ['--force', '强制清理，不询问确认'],
-          ['--dry-run', '预览将要清理的项目，不执行实际清理'],
+          ['--force',   '跳过确认提示，直接删除'],
+          ['--dry-run', '仅打印将要删除的路径，不真正删除']
         ].concat(super)
       end
 
       def initialize(argv)
-        @tool_name = argv.shift_argument || 'claude'
-        @force = argv.flag?('force')
+        @tool    = (argv.shift_argument || 'claude').to_s.downcase
+        @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(', ')}"
+        unless VALID_TOOLS.include?(@tool)
+          help! "不支持的工具：#{@tool}。可选值：#{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
+        targets  = collect_targets(@tool)
+        existing = targets.select { |path| File.exist?(path) || File.symlink?(path) }
 
-      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 Base::SystemInfo.macos?
-          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)"
+        if existing.empty?
+          puts "未找到 #{display_label(@tool)} 的缓存路径，无需清理".cyan
+          return
         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 "将清理以下路径（#{display_label(@tool)}）：".cyan
+        existing.each { |path| puts "  • #{path}" }
         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}"
+        if @dry_run
+          puts "已启用 --dry-run，未执行实际删除。".yellow
+          return
         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}"
+        unless @force
+          print "确认继续删除以上路径？(y/N): "
+          answer = ($stdin.gets || '').strip.downcase
+          unless %w[y yes].include?(answer)
+            puts "已取消".yellow
+            return
           end
         end
-      end
 
-      def clean_keychain_claude
-        return unless Base::SystemInfo.macos?
+        failures = perform_delete(existing)
 
-        account_name = Base::SystemInfo.current_user
-        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
+        if failures.empty?
+          puts "✓ 清理完成".green
         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
+          puts "⚠ 清理完成，但有 #{failures.size} 项失败".yellow
+          exit 1
         end
+      rescue Interrupt
+        puts
+        puts "已取消".yellow
+        exit 130
       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
+      private
 
-      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')
+      # @param tool [String]
+      # @return [Array<String>] expand 后的绝对路径列表
+      def collect_targets(tool)
+        keys = tool == 'all' ? CACHE_PATHS.keys : [tool]
+        keys.flat_map { |key| CACHE_PATHS.fetch(key, []) }.map { |p| File.expand_path(p) }
       end
 
-      def get_claude_keychain_entries
-        return [] unless Base::SystemInfo.macos?
-
-        entries = []
-        account_name = Base::SystemInfo.current_user
-
-        # 定义所有可能的 Claude 相关服务名
-        # 基于 authclaude.rb 和常见的 Claude 应用服务名
-        possible_services = [
-          "Claude Code-credentials",
-          "Claude Code",
-          "Claude-API",
-          "claude-api",
-          "claude-credentials",
-          "claude",
-          "anthropic-claude",
-          "anthropic",
-        ]
-
-        # 逐个检查已知的服务名，避免使用 dump-keychain
-        possible_services.each do |service_name|
-          # 使用 find-generic-password 检查条目是否存在
-          # 不使用 -w 选项，只检查是否存在，不获取密码内容
-          cmd = ["security", "find-generic-password", "-a", account_name, "-s", service_name]
-          stdout, stderr, status = Open3.capture3(*cmd)
-
-          # 如果命令成功（返回码0），说明条目存在
-          if status.success?
-            entries << service_name
+      # @param paths [Array<String>]
+      # @return [Array<String>] 删除失败的路径列表
+      def perform_delete(paths)
+        failures = []
+        paths.each do |path|
+          begin
+            FileUtils.rm_rf(path, secure: true)
+            puts "  ✓ #{path}".green
+          rescue Errno::EACCES => e
+            failures << path
+            puts "  ✗ #{path}: 权限不足（#{e.message}）".red
+          rescue StandardError => e
+            failures << path
+            puts "  ✗ #{path}: #{e.message}".red
           end
         end
-
-        # 额外检查：使用 list-keychains 安全地获取钥匙串列表
-        # 但不解密内容，只是查看是否有相关条目
-        # 注意：这个方法更安全，不会触发密码提示
-
-        entries.uniq
-      end
-
-      def get_shell_files
-        shell_files = []
-        current_shell = Base::SystemInfo.current_shell
-        
-        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
+        failures
       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
+      def display_label(tool)
+        tool == 'all' ? '全部 AI CLI' : tool
       end
-
     end
   end
-end
+end