easyai 1.7.0 → 2.0.0

data/lib/easyai/config/easyai_config.rb

@@ -1,258 +0,0 @@
-#!/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