easyai 1.7.0 → 2.1.0

data/lib/easyai/command/ai_tool_base.rb

@@ -0,0 +1,228 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'colored2'
+require_relative '../command'
+require_relative '../config/local_config'
+require_relative '../base/system_info'
+require_relative '../base/secret_masker'
+
+module EasyAI
+  class Command
+    # AI 工具命令的公共基类：claude / gemini / codex 共享的运行流程
+    #
+    # 子类必须实现：
+    #   - tool_name      返回 config.json 顶层键，如 'claude'
+    #   - exec_command   返回子进程命令名，如 'claude'
+    #   - install_hint   返回未安装时的中文安装提示
+    class AIToolBase < Command
+      self.ignore_in_command_lookup = true
+
+      # 系统关键变量保护清单：永远不允许被 config 中的 env / proxy 覆盖
+      PROTECTED_ENV_KEYS = %w[PATH HOME USER SHELL].freeze
+
+      def self.options
+        [
+          ['--platform=PLATFORM', '指定平台（如 claude_official / kimi / deepseek）'],
+          ['--verbose', '显示详细信息']
+        ].concat(super)
+      end
+
+      def initialize(argv)
+        @platform = argv.option('platform')
+        @verbose = argv.flag?('verbose')
+
+        super
+
+        remaining = @argv.remainder!
+        @local_config_file = extract_local_config_file!(remaining)
+        @passthrough_args = remaining
+      end
+
+      def validate!
+        super
+        return if Base::SystemInfo.which_command(exec_command)
+
+        help! "未找到 #{exec_command} CLI。#{install_hint}"
+      end
+
+      def run
+        cfg = resolve_config
+        @resolved_config = cfg # 供子类 pre_exec 读取当前生效的平台配置
+        print_platform_env(cfg)
+        @env = build_environment(cfg) # 存 ivar：pre_exec 可按需追加（如 codex 注入 CODEX_HOME）
+        export_environment_variables(@env)
+        pre_exec
+        exec(@env, exec_command, *@passthrough_args)
+      rescue Config::LocalConfig::NotFoundError => e
+        print_error(e.message)
+        puts "  请运行: #{'easyai setup'.yellow}"
+        exit 1
+      rescue Config::LocalConfig::PlatformNotFoundError => e
+        print_error(e.message)
+        available = Config::LocalConfig.available_platforms(tool_name)
+        puts "  当前可用平台：#{available.join(', ')}" unless available.empty?
+        exit 1
+      rescue Config::LocalConfig::ToolNotConfiguredError => e
+        print_error(e.message)
+        puts "  请运行: #{"easyai setup --tool=#{tool_name}".yellow}"
+        exit 1
+      rescue Config::LocalConfig::ParseError, Config::LocalConfig::IncompatibleVersionError => e
+        print_error(e.message)
+        puts "  可运行: #{'easyai setup --reset'.yellow} 重新生成配置"
+        exit 1
+      rescue Interrupt
+        puts
+        print_error('用户取消操作')
+        exit 130
+      end
+
+      # 子类必须实现的抽象方法
+      def tool_name
+        raise NotImplementedError, "#{self.class} 必须实现 #tool_name"
+      end
+
+      def exec_command
+        raise NotImplementedError, "#{self.class} 必须实现 #exec_command"
+      end
+
+      def install_hint
+        raise NotImplementedError, "#{self.class} 必须实现 #install_hint"
+      end
+
+      # 子类可选重写：返回该工具固定要注入的环境变量。
+      # 优先级低于 cfg.env（用户在 config.json 中显式声明的同名 key 会覆盖默认值）。
+      def default_env
+        {}
+      end
+
+      # 子类可选重写：在 exec 启动子进程之前执行的钩子（如：写入子进程依赖的状态文件、
+      # 追加只对本次子进程生效的环境变量到 @env，如 codex 注入 CODEX_HOME）。
+      # 默认 no-op；不应阻塞主流程，异常应降级为 warning。
+      def pre_exec
+        # no-op
+      end
+
+      # 子类可选重写：向平台选择列表注入的虚拟平台 { key => data }（如 codex 的 ChatGPT Subscribe）。
+      # 默认空；与 config.json 真实平台合并参与交互选择。
+      def extra_platforms
+        {}
+      end
+
+      private
+
+      # 从 remainder 中识别第一个以 .json 结尾且文件存在的位置参数作为本地覆盖入口
+      def extract_local_config_file!(remaining)
+        idx = remaining.find_index { |arg| arg.is_a?(String) && arg.end_with?('.json') && File.file?(arg) }
+        return nil if idx.nil?
+
+        remaining.delete_at(idx)
+      end
+
+      # 解析最终配置（hash with env/proxy 字段）。
+      # 选定平台 / 加载本地覆盖文件后，统一打印一行绿色 ✓ 提示，让用户明确看到当前生效的来源。
+      def resolve_config
+        if @local_config_file
+          cfg = load_local_override(@local_config_file)
+          print_success("使用本地覆盖配置 #{@local_config_file}（工具：#{tool_name}）")
+          cfg
+        else
+          Config::LocalConfig.resolve_platform(
+            tool: tool_name, platform: @platform, verbose: @verbose, extra: extra_platforms
+          ) do |name|
+            print_success("使用平台 #{name}（工具：#{tool_name}）")
+          end
+        end
+      end
+
+      def load_local_override(path)
+        raw = File.read(path)
+        data = JSON.parse(raw)
+        unless data.is_a?(Hash)
+          raise Config::LocalConfig::ParseError, "本地覆盖配置必须是 JSON 对象：#{path}"
+        end
+
+        data
+      rescue JSON::ParserError => e
+        raise Config::LocalConfig::ParseError, "解析本地覆盖配置失败 #{path}: #{e.message}"
+      end
+
+      # 构造注入子进程的 env：当前 ENV + default_env（子类钩子）+ config.env + proxy（同时大小写），
+      # 保护系统关键变量。优先级从低到高：当前 ENV → default_env → cfg.env → proxy。
+      def build_environment(cfg)
+        env = ENV.to_h
+
+        default_env.each do |key, value|
+          next if PROTECTED_ENV_KEYS.include?(key)
+
+          env[key.to_s] = value.to_s
+        end
+
+        cfg_env = cfg.is_a?(Hash) ? cfg['env'] : nil
+        if cfg_env.is_a?(Hash)
+          cfg_env.each do |key, value|
+            next if PROTECTED_ENV_KEYS.include?(key)
+
+            env[key.to_s] = value.to_s
+          end
+        end
+
+        proxy = cfg.is_a?(Hash) ? cfg['proxy'] : nil
+        if proxy.is_a?(Hash)
+          if (http_proxy = proxy['HTTP_PROXY'] || proxy['http_proxy'])
+            env['HTTP_PROXY'] = http_proxy.to_s
+            env['http_proxy'] = http_proxy.to_s
+          end
+          if (https_proxy = proxy['HTTPS_PROXY'] || proxy['https_proxy'])
+            env['HTTPS_PROXY'] = https_proxy.to_s
+            env['https_proxy'] = https_proxy.to_s
+          end
+        end
+
+        env
+      end
+
+      # 把 env 写入当前进程 ENV，但保护 PATH/HOME/USER/SHELL 不被覆盖
+      def export_environment_variables(env)
+        env.each do |key, value|
+          next if PROTECTED_ENV_KEYS.include?(key)
+
+          ENV[key] = value
+        end
+      end
+
+      # 输出辅助方法（与 Setup / Clean 等命令保持一致）
+      def print_success(message)
+        puts "  ✓ #{message.green}"
+      end
+
+      def print_error(message)
+        puts "  ✗ #{message.red}"
+      end
+
+      # 打印当前平台的 env / proxy（敏感字段走 Base::SecretMasker 脱敏），
+      # 让用户在 exec 前肉眼确认实际生效的配置，又不暴露 API Key 明文。
+      def print_platform_env(cfg)
+        return unless cfg.is_a?(Hash)
+
+        env = cfg['env']
+        proxy = cfg['proxy']
+        return if (env.nil? || env.empty?) && (proxy.nil? || proxy.empty?)
+
+        if env.is_a?(Hash) && !env.empty?
+          puts '    env:'.cyan
+          env.keys.sort.each do |key|
+            puts "      #{key} = #{Base::SecretMasker.format_value(key, env[key])}"
+          end
+        end
+
+        if proxy.is_a?(Hash) && !proxy.empty?
+          puts '    proxy:'.cyan
+          proxy.keys.sort.each do |key|
+            puts "      #{key} = #{proxy[key]}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/easyai/command/backup/claude.rb

@@ -0,0 +1,140 @@
+require 'json'
+require 'fileutils'
+require 'open3'
+
+module EasyAI
+  class Command
+    class Backup
+      class Claude < Backup
+        self.summary = '备份 Claude Code 登录信息'
+        self.description = <<-DESC
+          读取 ~/.claude.json（去除顶层 projects）+ macOS Keychain 中 "Claude Code-credentials" 条目，
+          按字段级深度 merge 更新到 ~/.easyai/backup/.claude.json。
+
+          特点:
+
+              * 字段级深度 merge：嵌套对象内部字段一对一合并，源字段覆盖/新增到目标，目标独有字段保留
+
+              * 排除 projects：避免会话历史等大数据进入备份
+
+              * 包含 Keychain token：macOS 上额外读取 "Claude Code-credentials" 条目（含 OAuth access/refresh token），
+                存放在备份文件的 _easyai_keychain 字段下；非 macOS 环境跳过此步
+
+              * 安全权限：备份文件 chmod 600（仅当前用户可读写）
+
+          使用示例:
+
+              $ easyai backup claude
+        DESC
+
+        EXCLUDED_KEYS = %w[projects].freeze
+        KEYCHAIN_LABEL = 'Claude Code-credentials'.freeze
+        KEYCHAIN_BACKUP_KEY = '_easyai_keychain'.freeze
+
+        # 注意：路径用方法动态 expand_path，避免常量在加载期冻结真实 HOME，
+        # 让 spec 切换 ENV['HOME'] 后定位失败（同 LocalConfig.config_path 的处理）
+        def source_path = File.expand_path('~/.claude.json')
+        def backup_dir  = File.expand_path('~/.easyai/backup')
+        def backup_path = File.join(backup_dir, '.claude.json')
+
+        def validate!
+          super
+          help! "源文件不存在: #{source_path}\n请先登录 Claude Code 后再备份。" unless File.exist?(source_path)
+        end
+
+        def run
+          source = read_json(source_path)
+          EXCLUDED_KEYS.each { |key| source.delete(key) }
+
+          keychain_value = read_keychain_credential(KEYCHAIN_LABEL)
+          if keychain_value
+            source[KEYCHAIN_BACKUP_KEY] = { KEYCHAIN_LABEL => keychain_value }
+          end
+
+          target = File.exist?(backup_path) ? read_json(backup_path) : {}
+          before_keys = target.keys
+
+          merged = deep_merge(target, source)
+
+          FileUtils.mkdir_p(backup_dir)
+          File.write(backup_path, JSON.pretty_generate(merged))
+          File.chmod(0o600, backup_path) unless windows?
+
+          report(before_keys, source.keys, merged.keys, keychain_value)
+        end
+
+        # 供 easyai claude 运行前调用：仅当备份不存在且源文件存在时自动备份一次。
+        # 返回是否执行了备份；任何异常降级为 warning，绝不阻塞启动。
+        def backup_if_absent
+          return false if File.exist?(backup_path)
+          return false unless File.exist?(source_path)
+
+          run
+          true
+        rescue StandardError => e
+          warn "  ⚠ 自动备份 Claude 登录信息失败（#{e.message}），继续启动"
+          false
+        end
+
+        private
+
+        def read_json(path)
+          JSON.parse(File.read(path))
+        rescue JSON::ParserError => e
+          raise "解析 JSON 失败 (#{path}): #{e.message}"
+        end
+
+        # 深度递归 merge：嵌套 hash 逐层合并，叶子值由源覆盖目标
+        def deep_merge(target, source)
+          target.merge(source) do |_key, old_val, new_val|
+            if old_val.is_a?(Hash) && new_val.is_a?(Hash)
+              deep_merge(old_val, new_val)
+            else
+              new_val
+            end
+          end
+        end
+
+        # 读取 macOS Keychain 中指定 label 的 generic password 值。
+        # 非 macOS 平台返回 nil；找不到条目或无权限访问也返回 nil（仅 warning，不阻塞备份）。
+        def read_keychain_credential(label)
+          return nil unless macos?
+
+          output, status = Open3.capture2('security', 'find-generic-password', '-l', label, '-w')
+          return output.strip if status.success? && !output.strip.empty?
+
+          warn "  ⚠ 未读到 Keychain 条目「#{label}」（exit=#{status.exitstatus}）；本次备份不含 Keychain token"
+          nil
+        rescue StandardError => e
+          warn "  ⚠ 读取 Keychain 异常：#{e.message}；本次备份不含 Keychain token"
+          nil
+        end
+
+        def report(before_keys, source_keys, after_keys, keychain_value)
+          updated = source_keys & before_keys
+          added   = source_keys - before_keys
+          kept    = before_keys - source_keys
+
+          puts "✓ Claude 登录信息已备份"
+          puts "  源文件:   #{source_path}"
+          puts "  备份路径: #{backup_path}"
+          puts "  排除字段: #{EXCLUDED_KEYS.join(', ')}"
+          puts "  字段更新: 新增 #{added.size}，更新 #{updated.size}，保留目标独有 #{kept.size}（合计 #{after_keys.size}）"
+          if keychain_value
+            puts "  Keychain: 已读取「#{KEYCHAIN_LABEL}」（#{keychain_value.bytesize} 字节），写入 #{KEYCHAIN_BACKUP_KEY} 字段"
+          else
+            puts "  Keychain: 未包含（非 macOS 或未找到该条目）"
+          end
+        end
+
+        def windows?
+          Base::SystemInfo.windows?
+        end
+
+        def macos?
+          Base::SystemInfo.macos?
+        end
+      end
+    end
+  end
+end

data/lib/easyai/command/backup/codex.rb

@@ -0,0 +1,206 @@
+require 'json'
+require 'fileutils'
+require_relative '../../base/toml_sections'
+require_relative '../../base/system_info'
+
+module EasyAI
+  class Command
+    class Backup
+      class Codex < Backup
+        self.summary = '备份 Codex 登录信息'
+        self.description = <<-DESC
+          把 ~/.codex/auth.json（登录凭证）+ ~/.codex/config.toml（去除 [projects.*] 授信段）
+          合并存进单个文件 ~/.easyai/backup/.codex.json，便于跨机器迁移或灾难恢复。
+
+          特点:
+
+              * 单文件存储：auth 与 config.toml 文本一并放进 .codex.json 的 auth / config_toml 字段
+
+              * auth 字段级深度 merge：源字段覆盖/新增到备份，备份独有字段保留
+
+              * config.toml 排除 [projects.*]：本地按项目授信的段不进备份，其余段段级 merge（源覆盖）
+
+              * 安全权限：备份文件 chmod 600（仅当前用户可读写）
+
+          使用示例:
+
+              $ easyai backup codex
+        DESC
+
+        AUTH_KEY   = 'auth'.freeze
+        CONFIG_KEY = 'config_toml'.freeze
+
+        # 注意：路径用方法动态 expand_path，避免常量在加载期冻结真实 HOME，
+        # 让 spec 切换 ENV['HOME'] 后定位失败（同 LocalConfig.config_path 的处理）
+        def auth_source   = File.expand_path('~/.codex/auth.json')
+        def config_source = File.expand_path('~/.codex/config.toml')
+        def backup_dir    = File.expand_path('~/.easyai/backup')
+        def backup_path   = File.join(backup_dir, '.codex.json')
+
+        def validate!
+          super
+          ensure_file_credentials_supported!
+          help! "源文件不存在: #{auth_source}\n请先登录 Codex 后再备份。" unless File.exist?(auth_source)
+        end
+
+        def run
+          ensure_file_credentials_supported!
+          unless official_token_present?
+            puts '⚠ 未在 ~/.codex/auth.json 检测到 ChatGPT 官方登录 token（OAuth），跳过备份。'
+            puts '  如需备份，请先运行 codex 完成 ChatGPT 订阅账户登录后重试。'
+            return
+          end
+
+          perform_backup
+        end
+
+        # 供 easyai setup 选择 ChatGPT Subscribe 时复用：仅当存在官方 token 才更新备份，
+        # 效果等同 easyai backup codex。返回是否实际执行了备份。
+        def backup_if_official
+          return false unless official_token_present?
+
+          perform_backup
+          true
+        end
+
+        # 供 easyai codex 运行前调用：仅当备份不存在、源文件存在且含官方 token 时自动备份一次。
+        # 返回是否执行了备份；任何异常降级为 warning，绝不阻塞启动。
+        def backup_if_absent
+          return false if File.exist?(backup_path)
+          return false unless File.exist?(auth_source)
+          # keyring 模式下凭证不在文件里，自动备份会一路触发 help! → 每次启动刷屏；此处静默跳过
+          return false if File.exist?(config_source) && keyring_credentials_store?(File.read(config_source))
+          # 只备份真正的 ChatGPT 官方登录态；EasyAI 给 longcat/deepseek 写的纯 API Key auth.json 不在此列
+          return false unless official_token_present?
+
+          perform_backup
+          true
+        rescue StandardError => e
+          warn "  ⚠ 自动备份 Codex 登录信息失败（#{e.message}），继续启动"
+          false
+        end
+
+        # 备份文件 .codex.json 中是否含 ChatGPT 官方 OAuth token，
+        # 供 easyai codex 运行时决定是否提供 ChatGPT Subscribe 选项。
+        def backup_has_official_token?
+          return false unless File.exist?(backup_path)
+
+          oauth_token?(read_json(backup_path)[AUTH_KEY])
+        rescue StandardError
+          false
+        end
+
+        private
+
+        def perform_backup
+          FileUtils.mkdir_p(backup_dir)
+          existing = File.exist?(backup_path) ? read_json(backup_path) : {}
+
+          auth_summary = merge_auth(existing)
+          config_summary = merge_config(existing)
+
+          File.write(backup_path, JSON.pretty_generate(existing))
+          File.chmod(0o600, backup_path) unless windows?
+
+          report(auth_summary, config_summary)
+        end
+
+        # 检测 ~/.codex/auth.json 是否为 ChatGPT 官方登录（OAuth）。
+        def official_token_present?
+          return false unless File.exist?(auth_source)
+
+          oauth_token?(read_json(auth_source))
+        rescue StandardError
+          false
+        end
+
+        # auth hash 是否带 ChatGPT OAuth 凭证：含非空 tokens 段，或 auth_mode == "chatgpt"。
+        # 仅有第三方 OPENAI_API_KEY 不算。
+        def oauth_token?(auth)
+          return false unless auth.is_a?(Hash)
+
+          tokens = auth['tokens']
+          has_oauth = tokens.is_a?(Hash) && tokens.values.any? { |v| !v.to_s.strip.empty? }
+          has_oauth || auth['auth_mode'].to_s == 'chatgpt'
+        end
+
+        def ensure_file_credentials_supported!
+          return unless File.exist?(config_source)
+
+          return unless keyring_credentials_store?(File.read(config_source))
+
+          help! <<~MSG
+            当前 Codex 配置 cli_auth_credentials_store = "keyring"，凭证存储在系统钥匙串中。
+            easyai backup codex 目前只支持 file/auth.json 凭证备份，请先改用 file 存储或手动备份系统钥匙串凭证。
+          MSG
+        end
+
+        def keyring_credentials_store?(toml_text)
+          toml_text.to_s.each_line.any? do |line|
+            line.match?(/^\s*cli_auth_credentials_store\s*=\s*["']keyring["']\s*(?:#.*)?$/)
+          end
+        end
+
+        # 把 ~/.codex/auth.json 深度 merge 进 existing[AUTH_KEY]，返回统计
+        def merge_auth(existing)
+          source = read_json(auth_source)
+          target = existing[AUTH_KEY].is_a?(Hash) ? existing[AUTH_KEY] : {}
+          before = target.keys
+          existing[AUTH_KEY] = deep_merge(target, source)
+
+          {
+            added: (source.keys - before).size,
+            updated: (source.keys & before).size,
+            kept: (before - source.keys).size,
+            total: existing[AUTH_KEY].keys.size
+          }
+        end
+
+        # 把 config.toml（去 projects）段级 merge 进 existing[CONFIG_KEY]，返回统计
+        def merge_config(existing)
+          return nil unless File.exist?(config_source)
+
+          raw = File.read(config_source)
+          excluded = Base::TomlSections.split_sections(raw)
+                                       .count { |s| Base::TomlSections.project_header?(s.header) }
+          filtered = Base::TomlSections.reject_projects(raw)
+          base = existing[CONFIG_KEY].is_a?(String) ? existing[CONFIG_KEY] : ''
+          existing[CONFIG_KEY] = Base::TomlSections.merge(base, filtered)
+
+          { excluded_projects: excluded }
+        end
+
+        def read_json(path)
+          JSON.parse(File.read(path))
+        rescue JSON::ParserError => e
+          raise "解析 JSON 失败 (#{path}): #{e.message}"
+        end
+
+        def deep_merge(target, source)
+          target.merge(source) do |_key, old_val, new_val|
+            if old_val.is_a?(Hash) && new_val.is_a?(Hash)
+              deep_merge(old_val, new_val)
+            else
+              new_val
+            end
+          end
+        end
+
+        def report(auth, config)
+          puts "✓ Codex 登录信息已备份"
+          puts "  备份文件: #{backup_path}"
+          puts "  auth:        新增 #{auth[:added]}，更新 #{auth[:updated]}，保留备份独有 #{auth[:kept]}（合计 #{auth[:total]}）"
+          if config
+            puts "  config.toml: 已备份（排除 #{config[:excluded_projects]} 个 [projects.*] 授信段）"
+          else
+            puts "  config.toml: 源文件不存在，跳过"
+          end
+        end
+
+        def windows?
+          Base::SystemInfo.windows?
+        end
+      end
+    end
+  end
+end

data/lib/easyai/command/backup.rb

@@ -0,0 +1,26 @@
+module EasyAI
+  class Command
+    class Backup < Command
+      self.summary = '备份 AI CLI 登录信息'
+      self.description = <<-DESC
+        备份各家 AI CLI 工具的登录信息到 ~/.easyai/backup/，便于跨机器迁移或灾难恢复。
+
+        可用命令:
+
+            * claude - 备份 Claude Code 登录信息
+            * codex  - 备份 Codex 登录信息
+
+        使用示例:
+
+            $ easyai backup claude    # 备份 ~/.claude.json（去除 projects）到 ~/.easyai/backup/.claude.json
+            $ easyai backup codex     # 备份 ~/.codex/auth.json + config.toml（去除 [projects.*]）到 ~/.easyai/backup/.codex.json
+      DESC
+
+      self.abstract_command = true
+    end
+  end
+end
+
+# 加载子命令
+require_relative 'backup/claude'
+require_relative 'backup/codex'