easyai 1.7.0 → 2.0.0

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

@@ -0,0 +1,150 @@
+require 'json'
+require 'fileutils'
+require 'open3'
+require 'etc'
+
+module EasyAI
+  class Command
+    class Restore
+      class Claude < Restore
+        self.summary = '恢复 Claude Code 登录信息'
+        self.description = <<-DESC
+          读取 ~/.easyai/backup/.claude.json：取出 _easyai_keychain 字段写回 macOS Keychain，
+          剩余字段按字段级深度 merge 写回 ~/.claude.json。
+
+          特点:
+
+              * 深度递归 merge：嵌套对象内部字段一对一合并；备份字段覆盖/新增，目标独有保留
+
+              * Keychain 恢复：macOS 上把 _easyai_keychain.<label> 的值写回对应 Keychain 条目（如 "Claude Code-credentials"）；
+                _easyai_keychain 字段不写入 ~/.claude.json，保持纯净
+
+              * 目标独有保留：~/.claude.json 里的 projects 等运行时数据原样保留，不被清空
+
+              * 安全权限：恢复后的 ~/.claude.json 仍 chmod 600
+
+          使用示例:
+
+              $ easyai restore claude
+        DESC
+
+        BACKUP_PATH = File.expand_path('~/.easyai/backup/.claude.json').freeze
+        TARGET_PATH = File.expand_path('~/.claude.json').freeze
+        KEYCHAIN_BACKUP_KEY = '_easyai_keychain'.freeze
+
+        def validate!
+          super
+          help! "备份文件不存在: #{BACKUP_PATH}\n请先运行 easyai backup claude 创建备份。" unless File.exist?(BACKUP_PATH)
+        end
+
+        def run
+          source = read_json(BACKUP_PATH)
+
+          keychain_results = restore_keychain(source.delete(KEYCHAIN_BACKUP_KEY))
+
+          target = File.exist?(TARGET_PATH) ? read_json(TARGET_PATH) : {}
+          before_keys = target.keys
+
+          merged = deep_merge(target, source)
+
+          File.write(TARGET_PATH, JSON.pretty_generate(merged))
+          File.chmod(0o600, TARGET_PATH) unless windows?
+
+          report(before_keys, source.keys, merged.keys, keychain_results)
+        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
+
+        # 把备份中的 _easyai_keychain hash 逐项写回 macOS Keychain。
+        # 返回 { label => :ok | :fail | :skip } 用于 report；
+        # 非 macOS / 备份无该字段时直接跳过，返回空 hash。
+        def restore_keychain(keychain_data)
+          return {} unless keychain_data.is_a?(Hash) && !keychain_data.empty?
+
+          unless macos?
+            warn "  ⚠ 非 macOS 环境，跳过 Keychain 恢复（备份中含 #{keychain_data.size} 个 Keychain 条目）"
+            return keychain_data.keys.each_with_object({}) { |k, h| h[k] = :skip }
+          end
+
+          account = ENV['USER'] || (Etc.getlogin rescue nil)
+          if account.nil? || account.empty?
+            warn "  ⚠ 无法确定当前用户名，跳过 Keychain 恢复"
+            return keychain_data.keys.each_with_object({}) { |k, h| h[k] = :skip }
+          end
+
+          keychain_data.each_with_object({}) do |(label, password), results|
+            results[label] = write_keychain_credential(label, password, account)
+          end
+        end
+
+        # 写入 Keychain（如已存在则更新）。security 命令的 -w <password> 在命令行参数中可见，
+        # 但本机用户跑本机命令的场景下风险可控；macOS 上 ps 默认不展示其他用户进程的完整 cmdline。
+        def write_keychain_credential(label, password, account)
+          unless password.is_a?(String) && !password.empty?
+            warn "  ⚠ Keychain 条目「#{label}」备份值无效（非字符串或为空），跳过"
+            return :fail
+          end
+
+          args = ['security', 'add-generic-password',
+                  '-a', account,
+                  '-s', label,
+                  '-l', label,
+                  '-w', password,
+                  '-U']
+          output, status = Open3.capture2e(*args)
+          if status.success?
+            :ok
+          else
+            warn "  ⚠ 写入 Keychain 条目「#{label}」失败（exit=#{status.exitstatus}）：#{output.strip}"
+            :fail
+          end
+        rescue StandardError => e
+          warn "  ⚠ 写入 Keychain 条目「#{label}」异常：#{e.message}"
+          :fail
+        end
+
+        def report(before_keys, source_keys, after_keys, keychain_results)
+          updated = source_keys & before_keys
+          added   = source_keys - before_keys
+          kept    = before_keys - source_keys
+
+          puts "✓ Claude 登录信息已恢复"
+          puts "  备份文件: #{BACKUP_PATH}"
+          puts "  目标路径: #{TARGET_PATH}"
+          puts "  字段更新: 新增 #{added.size}，更新 #{updated.size}，保留目标独有 #{kept.size}（合计 #{after_keys.size}）"
+          if keychain_results.empty?
+            puts "  Keychain: 备份中未含 #{KEYCHAIN_BACKUP_KEY} 字段，无需恢复"
+          else
+            ok = keychain_results.count { |_, v| v == :ok }
+            puts "  Keychain: 恢复 #{ok}/#{keychain_results.size} 个条目（#{keychain_results.map { |k, v| "#{k}=#{v}" }.join(', ')}）"
+          end
+        end
+
+        def windows?
+          Base::SystemInfo.windows?
+        end
+
+        def macos?
+          Base::SystemInfo.macos?
+        end
+      end
+    end
+  end
+end

data/lib/easyai/command/restore.rb

@@ -0,0 +1,24 @@
+module EasyAI
+  class Command
+    class Restore < Command
+      self.summary = '从 ~/.easyai/backup/ 恢复 AI CLI 登录信息'
+      self.description = <<-DESC
+        把 easyai backup 备份的登录信息深度合并回各家 AI CLI 的原始配置文件。
+        合并策略：备份字段覆盖/新增到目标，目标独有字段（如 Claude 的 projects）保留。
+
+        可用命令:
+
+            * claude - 恢复 Claude Code 登录信息
+
+        使用示例:
+
+            $ easyai restore claude    # 把 ~/.easyai/backup/.claude.json 深度 merge 回 ~/.claude.json
+      DESC
+
+      self.abstract_command = true
+    end
+  end
+end
+
+# 加载子命令
+require_relative 'restore/claude'