easyai 1.2.1 → 1.4.0

data/lib/easyai/config/easyai_config.rb

@@ -0,0 +1,258 @@
+#!/usr/bin/env ruby
+
+require 'fileutils'
+require 'open3'
+require 'json'
+require 'tmpdir'
+require_relative '../base/file_crypto'
+require_relative '../base/system_keychain'
+
+module EasyAI
+  # EasyAIConfig 类负责配置仓库的下载、更新和文件解密
+  class EasyAIConfig
+    REPO_URL = "https://gitee.com/goodtools/EasyAISetting.git"
+    EASYAI_DIR = File.expand_path('~/.easyai')
+    CONFIG_REPO_DIR = File.join(EASYAI_DIR, 'EasyAISetting')
+
+    class << self
+      # 初始化配置仓库
+      def initialize(options = {})
+        verbose = options[:verbose] || false
+        ensure_repo_ready(verbose)
+        # 确保临时目录存在
+        @temp_dir ||= File.join(Dir.tmpdir, "easyai_#{Process.pid}")
+        FileUtils.mkdir_p(@temp_dir) unless Dir.exist?(@temp_dir)
+      end
+
+      # 核心接口：获取解密后的配置内容
+      # @param filename [String] 配置文件名
+      # @param options [Hash] 选项
+      # @return [Hash, nil] 解密后的配置内容
+      def get_config(filename, options = {})
+        return nil unless filename
+
+        ensure_repo_ready(options[:verbose])
+        filename = normalize_filename(filename)
+
+        # 查找加密文件的实际路径（支持子目录）
+        encrypted_path = find_encrypted_file(filename)
+        return nil unless encrypted_path
+
+        # 解密到临时目录
+        temp_decrypted_path = get_temp_path(filename)
+
+        # 解密文件
+        return nil unless decrypt_file(encrypted_path, temp_decrypted_path, options[:verbose])
+
+        # 读取并解析配置
+        result = read_json_file(temp_decrypted_path, options[:verbose])
+
+        # 立即清理临时文件
+        FileUtils.rm_f(temp_decrypted_path) if File.exist?(temp_decrypted_path)
+
+        result
+      end
+
+      # 获取配置文件路径（用于需要文件路径的场景）
+      def get_config_path(filename, options = {})
+        return nil unless filename
+
+        ensure_repo_ready(options[:verbose])
+        filename = normalize_filename(filename)
+
+        # 查找加密文件的实际路径（支持子目录）
+        encrypted_path = find_encrypted_file(filename)
+        return nil unless encrypted_path
+
+        # 解密到临时目录
+        temp_decrypted_path = get_temp_path(filename)
+
+        # 解密文件
+        return nil unless decrypt_file(encrypted_path, temp_decrypted_path, options[:verbose])
+
+        File.exist?(temp_decrypted_path) ? temp_decrypted_path : nil
+      end
+
+      # 清理所有临时文件
+      def cleanup(verbose = false)
+        if @temp_dir && Dir.exist?(@temp_dir)
+          FileUtils.rm_rf(@temp_dir)
+          puts "✓ 已清理临时目录" if verbose
+        end
+        @temp_dir = nil
+      end
+
+      private
+
+      # 确保配置仓库就绪
+      def ensure_repo_ready(verbose = false)
+        FileUtils.mkdir_p(EASYAI_DIR) unless Dir.exist?(EASYAI_DIR)
+
+        if Dir.exist?(CONFIG_REPO_DIR)
+          update_repo(verbose)
+        else
+          download_repo(verbose)
+        end
+
+        # 确保临时目录存在
+        @temp_dir ||= File.join(Dir.tmpdir, "easyai_#{Process.pid}")
+        FileUtils.mkdir_p(@temp_dir) unless Dir.exist?(@temp_dir)
+      end
+
+      # 标准化文件名
+      def normalize_filename(filename)
+        filename.end_with?('.json') ? filename : "#{filename}.json"
+      end
+
+      # 查找加密文件（支持子目录）
+      def find_encrypted_file(filename)
+        # 直接在根目录查找
+        root_path = File.join(CONFIG_REPO_DIR, "#{filename}.encrypted")
+        return root_path if File.exist?(root_path)
+
+        # 在子目录中查找（保持原有路径）
+        if filename.include?('/')
+          sub_path = File.join(CONFIG_REPO_DIR, "#{filename}.encrypted")
+          return sub_path if File.exist?(sub_path)
+        end
+
+        nil
+      end
+
+      # 获取临时文件路径
+      def get_temp_path(filename)
+        # 保持文件名结构，替换 / 为 _
+        safe_filename = filename.gsub('/', '_')
+        File.join(@temp_dir, safe_filename)
+      end
+
+      # 读取 JSON 文件
+      def read_json_file(file_path, verbose = false)
+        return nil unless File.exist?(file_path)
+
+        begin
+          content = File.read(file_path)
+          JSON.parse(content)
+        rescue JSON::ParserError => e
+          puts "✗ 解析配置文件失败: #{e.message}" if verbose
+          nil
+        rescue => e
+          puts "✗ 读取配置文件失败: #{e.message}" if verbose
+          nil
+        end
+      end
+
+      # 下载配置仓库
+      def download_repo(verbose = false)
+        # 根据测试环境变量选择分支
+        branch = get_target_branch
+        puts "正在下载配置仓库 (#{branch} 分支)..." if verbose
+
+        cmd = "git clone --depth 1 --branch #{branch} #{REPO_URL} #{CONFIG_REPO_DIR} 2>&1"
+        _, _, status = Open3.capture3(cmd)
+
+        if status.success?
+          puts "✓ 配置仓库下载成功 (#{branch} 分支)" if verbose
+          true
+        else
+          puts "✗ 配置仓库下载失败，请检查网络连接" if verbose
+          false
+        end
+      rescue => e
+        puts "✗ 下载配置仓库失败: #{e.message}" if verbose
+        false
+      end
+
+      # 获取目标分支
+      def get_target_branch
+        # 检查是否有测试环境变量
+        if ENV['EASYAI_TEST_PASSWORD'] || ENV['EASYAI_DEV_MODE']
+          'dev'
+        else
+          'master'
+        end
+      end
+
+      # 更新配置仓库
+      def update_repo(verbose = false)
+        return false unless Dir.exist?(CONFIG_REPO_DIR)
+
+        target_branch = get_target_branch
+
+        Dir.chdir(CONFIG_REPO_DIR) do
+          # 重置本地更改
+          system("git reset --hard HEAD > /dev/null 2>&1")
+
+          # 获取当前分支
+          current_branch = `git branch --show-current`.chomp
+
+          # 如果需要切换分支
+          if current_branch != target_branch
+            puts "切换到 #{target_branch} 分支..." if verbose
+            system("git fetch origin #{target_branch} > /dev/null 2>&1")
+            system("git checkout #{target_branch} > /dev/null 2>&1")
+          end
+
+          # 拉取最新更新
+          _, _, status = Open3.capture3("git pull --force 2>&1")
+
+          if status.success?
+            puts "✓ 配置仓库已更新 (#{target_branch} 分支)" if verbose
+          else
+            puts "⚠️ 配置更新失败，使用现有配置" if verbose
+          end
+        end
+        true
+      rescue => e
+        puts "⚠️ 更新配置时出错: #{e.message}" if verbose
+        true
+      end
+
+      # 解密文件
+      def decrypt_file(encrypted_path, decrypted_path, verbose = false)
+        puts "正在解密: #{File.basename(encrypted_path)}..." if verbose
+
+        # 智能密码获取：环境变量 > 系统钥匙串 > 用户输入
+        password = get_decryption_password(verbose)
+        return false unless password
+
+        # 使用静态方法解密文件
+        Base::FileCrypto.decrypt_file(encrypted_path, password[:value], decrypted_path)
+        File.chmod(0600, decrypted_path)
+
+        # 如果密码来自用户输入且解密成功，保存到钥匙串
+        if password[:from_user_input]
+          Base::SystemKeychain.store_password(password[:value])
+        end
+
+        true
+      rescue => e
+        puts "✗ 解密失败: #{e.message}" if verbose
+        false
+      end
+
+      # 智能密码获取
+      def get_decryption_password(verbose = false)
+        # 1. 优先使用环境变量（用于测试和自动化）
+        if ENV['EASYAI_TEST_PASSWORD']
+          puts "使用环境变量密码" if verbose
+          return { value: ENV['EASYAI_TEST_PASSWORD'], from_user_input: false }
+        end
+
+        # 2. 尝试从系统钥匙串获取
+        stored_password = Base::SystemKeychain.get_stored_password
+        if stored_password && !stored_password.empty?
+          puts "使用系统存储的密码" if verbose
+          return { value: stored_password, from_user_input: false }
+        end
+
+        # 3. 提示用户输入
+        puts "请输入解密密码（成功后将自动保存）" if verbose
+        password = Base::FileCrypto.get_password("请输入解密密码: ")
+        return nil unless password && !password.empty?
+
+        { value: password, from_user_input: true }
+      end
+    end
+  end
+end

data/lib/easyai/version.rb

@@ -1,3 +1,3 @@
 module EasyAI
-  VERSION = '1.2.1'
+  VERSION = '1.4.0'
 end

data/lib/easyai.rb

@@ -1,5 +1,6 @@
 require 'easyai/version'
 require 'easyai/config/config'
+require 'easyai/config/easyai_config'
 require 'easyai/command'
 require 'easyai/command/claude'
 require 'easyai/command/gemini'
@@ -7,8 +8,10 @@ require 'easyai/command/gpt'
 require 'easyai/command/utils'
 require 'easyai/command/clean'
 require 'easyai/command/update'
+require 'easyai/command/setup'
 require 'easyai/base/version_checker'
 require 'easyai/base/update_notifier'
+require 'easyai/base/cross_platform'
 
 module EasyAI
   # EasyAI 应用程序主类
@@ -18,17 +21,23 @@ module EasyAI
       require 'claide'
       require 'colored2'
       coerced_argv = CLAide::ARGV.coerce(argv)
-      
+
       # 如果不是帮助或版本查询，显示启动标志
       unless coerced_argv.flag?('help') || coerced_argv.flag?('version')
         show_banner
       end
-      
+
       # 在运行命令前进行版本检查
       check_version_before_run(argv)
-      
+
+      # 设置退出时清理
+      setup_cleanup_handler
+
       # 运行命令
       EasyAI::Command.run(argv)
+    ensure
+      # 确保清理临时文件
+      cleanup_temp_files
     end
     
     private
@@ -87,5 +96,25 @@ module EasyAI
       puts "⚠️  版本检查出错: #{e.message}".yellow
       puts e.backtrace.join("\n") if ENV['EASYAI_DEBUG']
     end
+
+    # 设置清理处理器
+    def setup_cleanup_handler
+      # 注册信号处理器，确保 Ctrl+C 等信号时也能清理
+      %w[INT TERM].each do |signal|
+        Signal.trap(signal) do
+          cleanup_temp_files
+          exit(130) # 标准的 Ctrl+C 退出码
+        end
+      end
+    end
+
+    # 清理临时文件
+    def cleanup_temp_files
+      verbose = ENV['EASYAI_DEBUG'] || ENV['EASYAI_VERBOSE']
+      EasyAIConfig.cleanup(verbose)
+    rescue => e
+      # 清理失败不影响退出
+      puts "⚠️ 清理临时文件失败: #{e.message}" if ENV['EASYAI_DEBUG']
+    end
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: easyai
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.4.0
 platform: ruby
 authors:
 - Wade
 bindir: bin
 cert_chain: []
-date: 1980-01-02 00:00:00.000000000 Z
+date: 2025-09-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: claide
@@ -51,6 +51,26 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.9'
+- !ruby/object:Gem::Dependency
+  name: jpsclient
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.0
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -108,7 +128,8 @@ files:
 - easyai.gemspec
 - lib/easyai.rb
 - lib/easyai/auth/authclaude.rb
-- lib/easyai/auth/jpslogin.rb
+- lib/easyai/auth/jpsloginhelper.rb
+- lib/easyai/base/cross_platform.rb
 - lib/easyai/base/file_crypto.rb
 - lib/easyai/base/system_keychain.rb
 - lib/easyai/base/update_notifier.rb
@@ -118,12 +139,14 @@ files:
 - lib/easyai/command/clean.rb
 - lib/easyai/command/gemini.rb
 - lib/easyai/command/gpt.rb
+- lib/easyai/command/setup.rb
 - lib/easyai/command/update.rb
 - lib/easyai/command/utils.rb
 - lib/easyai/command/utils/decry.rb
 - lib/easyai/command/utils/encry.rb
 - lib/easyai/command/utils/export.rb
 - lib/easyai/config/config.rb
+- lib/easyai/config/easyai_config.rb
 - lib/easyai/version.rb
 homepage: https://github.com/wade/easyai
 licenses:
@@ -143,7 +166,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.9
+rubygems_version: 3.6.3
 specification_version: 4
 summary: Easy AI CLI tool wrapper
 test_files: []