easyai 1.0.2 → 1.0.4

data/lib/easyai/command/update.rb

@@ -0,0 +1,210 @@
+require 'fileutils'
+require 'tmpdir'
+require 'colored2'
+
+module EasyAI
+  class Command
+    class Update < Command
+      self.summary = '从代码仓库安装最新开发版本（仅用于测试）'
+      self.description = <<-DESC
+        从 Gitee 代码仓库下载并安装最新的开发版本。
+        
+        主要功能:
+
+            * 从远程仓库拉取最新代码
+
+            * 自动构建 gem 包
+
+            * 安装到本地系统
+
+        使用示例:
+
+            $ easyai update                      # 安装 master 分支
+
+            $ easyai update --branch=dev         # 安装指定分支
+
+            $ easyai update --keep-source        # 保留源代码
+
+            $ easyai update --verbose            # 显示详细输出
+      DESC
+
+      def self.options
+        [
+          ['--branch=BRANCH', '指定要安装的分支（默认: master）'],
+          ['--keep-source', '保留下载的源代码'],
+          ['--verbose', '显示详细输出']
+        ].concat(super)
+      end
+
+      def initialize(argv)
+        @branch = argv.option('branch') || 'master'
+        @keep_source = argv.flag?('keep-source')
+        @verbose = argv.flag?('verbose')
+        super
+      end
+
+      def validate!
+        super
+        help! '未找到 git 命令。请先安装 git。' unless git_available?
+      end
+
+      def run
+        puts "\n╔#{'═' * 58}╗".cyan
+        puts "║#{'EasyAI 开发版本更新工具'.center(58)}║".cyan
+        puts "╚#{'═' * 58}╝\n".cyan
+        
+        puts "⚠️  注意：此命令仅用于测试开发版本".yellow
+        puts "📦 生产环境请使用: #{'gem update easyai'.green}\n\n"
+        
+        # 确认继续
+        unless confirm_update?
+          puts "❌ 更新已取消".red
+          return
+        end
+        
+        # 创建临时目录
+        temp_dir = Dir.mktmpdir('easyai-update-')
+        
+        begin
+          # 克隆仓库
+          clone_repository(temp_dir)
+          
+          # 执行安装脚本
+          install_from_source(temp_dir)
+          
+          # 显示成功信息
+          show_success_message
+          
+        rescue => e
+          puts "\n❌ 更新失败: #{e.message}".red
+          puts "   请检查网络连接或稍后重试".red
+          exit 1
+        ensure
+          # 清理临时文件
+          cleanup_temp_files(temp_dir) unless @keep_source
+        end
+      end
+
+      private
+
+      def git_available?
+        system('which git > /dev/null 2>&1')
+      end
+
+      def confirm_update?
+        print "确认从 Gitee 安装开发版本？[y/N]: ".yellow
+        begin
+          response = STDIN.gets
+          return false unless response
+          response = response.chomp.downcase
+          response == 'y' || response == 'yes'
+        rescue => e
+          puts "\n❌ 无法读取输入: #{e.message}".red
+          false
+        end
+      end
+
+      def clone_repository(temp_dir)
+        repo_url = 'https://gitee.com/goodtools/EasyAI.git'
+        clone_path = File.join(temp_dir, 'EasyAI')
+        
+        puts "📥 正在下载源代码...".cyan
+        puts "   仓库: #{repo_url}"
+        puts "   分支: #{@branch}"
+        
+        # 构建 git clone 命令
+        git_cmd = "git clone --depth 1 --branch #{@branch} #{repo_url} #{clone_path}"
+        git_cmd += " > /dev/null 2>&1" unless @verbose
+        
+        # 执行克隆
+        success = system(git_cmd)
+        
+        unless success
+          raise "无法克隆仓库。请检查网络连接和分支名称。"
+        end
+        
+        puts "   ✓ 源代码下载完成".green
+        
+        clone_path
+      end
+
+      def install_from_source(temp_dir)
+        source_dir = File.join(temp_dir, 'EasyAI')
+        
+        unless File.exist?(source_dir)
+          raise "源代码目录不存在: #{source_dir}"
+        end
+        
+        # 检查 test_local.sh 是否存在
+        install_script = File.join(source_dir, 'test_local.sh')
+        unless File.exist?(install_script)
+          raise "安装脚本不存在: test_local.sh"
+        end
+        
+        puts "\n🔧 正在构建并安装...".cyan
+        
+        # 切换到源代码目录并执行安装脚本
+        Dir.chdir(source_dir) do
+          # 确保脚本可执行
+          File.chmod(0755, 'test_local.sh')
+          
+          # 执行安装脚本
+          install_cmd = "./test_local.sh"
+          install_cmd += " > /dev/null 2>&1" unless @verbose
+          
+          success = system(install_cmd)
+          
+          unless success
+            raise "安装脚本执行失败"
+          end
+        end
+        
+        puts "   ✓ 安装完成".green
+      end
+
+      def cleanup_temp_files(temp_dir)
+        if @keep_source
+          puts "\n📁 源代码保留在: #{temp_dir}".cyan
+        else
+          puts "\n🧹 正在清理临时文件...".cyan
+          FileUtils.rm_rf(temp_dir)
+          puts "   ✓ 清理完成".green
+        end
+      end
+
+      def show_success_message
+        # 获取安装后的版本
+        installed_version = get_installed_version
+        
+        puts "\n" + "─" * 60
+        puts "✅ #{'开发版本安装成功！'.green}"
+        puts "─" * 60
+        
+        if installed_version
+          puts "📌 当前版本: #{installed_version}"
+        end
+        
+        puts "\n提示:".cyan
+        puts "  • 这是开发版本，可能包含未发布的功能"
+        puts "  • 如需回退到稳定版本: gem install easyai"
+        puts "  • 查看版本信息: easyai --version"
+        
+        if @branch != 'master'
+          puts "\n📝 已安装分支: #{@branch}".yellow
+        end
+      end
+
+      def get_installed_version
+        # 尝试获取已安装的版本
+        output = `easyai --version 2>&1`
+        if output =~ /EasyAI 版本:\s*(\S+)/
+          $1
+        else
+          nil
+        end
+      rescue
+        nil
+      end
+    end
+  end
+end

data/lib/easyai/command/utils/decry.rb

@@ -0,0 +1,102 @@
+require_relative '../../base/file_crypto'
+
+module EasyAI
+  class Command
+    class Utils
+      class Decry < Utils
+        self.summary = '解密指定文件或目录'
+        self.description = <<-DESC
+        对指定的加密文件或目录进行解密处理。
+        
+        支持功能:
+
+            * 单个文件解密
+
+            * 目录批量解密
+
+            * 自动恢复原文件名
+
+        使用示例:
+
+            $ easyai utils decry file.encrypted     # 解密单个文件
+
+            $ easyai utils decry /path/to/dir       # 解密整个目录
+
+            $ easyai utils decry file.enc out.txt   # 指定输出文件
+
+            $ easyai utils decry --output=out.txt   # 使用选项指定输出
+        DESC
+
+        def self.options
+          [
+            ['--output=PATH', '指定解密后文件的输出路径'],
+          ].concat(super)
+        end
+
+        def initialize(argv)
+          @file_path = argv.shift_argument
+          @output_path = argv.option('output') || argv.shift_argument
+          super
+        end
+
+        def validate!
+          super
+          help! '请指定要解密的文件或目录路径' if @file_path.nil? || @file_path.empty?
+          help! "路径不存在: #{@file_path}" unless File.exist?(@file_path)
+        end
+
+        def run
+          if File.file?(@file_path)
+            decrypt_file_mode
+          elsif File.directory?(@file_path)
+            decrypt_directory_mode
+          else
+            puts "❌ 指定路径既不是文件也不是目录: #{@file_path}"
+            exit 1
+          end
+        end
+
+        private
+
+        def decrypt_file_mode
+          puts "🔓 正在解密文件: #{@file_path}"
+          
+          begin
+            # 获取解密密码
+            password = Base::FileCrypto.get_password("请输入解密密码: ")
+            
+            # 解密文件
+            output_file = Base::FileCrypto.decrypt_file(@file_path, password, @output_path)
+            
+            puts "✅ 文件解密完成"
+            puts "📁 解密文件保存位置: #{output_file}"
+            
+          rescue => e
+            puts "❌ 解密失败: #{e.message}"
+            exit 1
+          end
+        end
+
+        def decrypt_directory_mode
+          puts "🔓 正在解密目录: #{@file_path}"
+          
+          begin
+            # 获取解密密码
+            password = Base::FileCrypto.get_password("请输入解密密码: ")
+            
+            # 解密目录
+            result = Base::FileCrypto.decrypt_directory(@file_path, password, @output_path)
+            
+            puts "✅ 目录解密完成"
+            puts "📁 解密目录保存位置: #{result[:output_dir]}"
+            puts "📋 已解密 #{result[:decrypted_files].length} 个文件"
+            
+          rescue => e
+            puts "❌ 解密失败: #{e.message}"
+            exit 1
+          end
+        end
+      end
+    end
+  end
+end

data/lib/easyai/command/utils/encry.rb

@@ -0,0 +1,102 @@
+require_relative '../../base/file_crypto'
+
+module EasyAI
+  class Command
+    class Utils
+      class Encry < Utils
+        self.summary = '加密指定文件或目录'
+        self.description = <<-DESC
+        对指定文件或目录进行加密处理。
+        
+        支持功能:
+
+            * 单个文件加密
+
+            * 目录批量加密
+
+            * 自定义输出路径
+
+        使用示例:
+
+            $ easyai utils encry file.txt           # 加密单个文件
+
+            $ easyai utils encry /path/to/dir       # 加密整个目录
+
+            $ easyai utils encry file.txt out.enc   # 指定输出文件
+
+            $ easyai utils encry --output=out.enc   # 使用选项指定输出
+        DESC
+
+        def self.options
+          [
+            ['--output=PATH', '指定加密后文件的输出路径'],
+          ].concat(super)
+        end
+
+        def initialize(argv)
+          @file_path = argv.shift_argument
+          @output_path = argv.option('output') || argv.shift_argument
+          super
+        end
+
+        def validate!
+          super
+          help! '请指定要加密的文件或目录路径' if @file_path.nil? || @file_path.empty?
+          help! "路径不存在: #{@file_path}" unless File.exist?(@file_path)
+        end
+
+        def run
+          if File.file?(@file_path)
+            encrypt_file_mode
+          elsif File.directory?(@file_path)
+            encrypt_directory_mode
+          else
+            puts "❌ 指定路径既不是文件也不是目录: #{@file_path}"
+            exit 1
+          end
+        end
+
+        private
+
+        def encrypt_file_mode
+          puts "🔐 正在加密文件: #{@file_path}"
+          
+          begin
+            # 获取加密密码
+            password = Base::FileCrypto.get_password("请输入加密密码: ")
+            
+            # 加密文件
+            output_file = Base::FileCrypto.encrypt_file(@file_path, password, @output_path)
+            
+            puts "✅ 文件加密完成"
+            puts "📁 加密文件保存位置: #{output_file}"
+            
+          rescue => e
+            puts "❌ 加密失败: #{e.message}"
+            exit 1
+          end
+        end
+
+        def encrypt_directory_mode
+          puts "🔐 正在加密目录: #{@file_path}"
+          
+          begin
+            # 获取加密密码
+            password = Base::FileCrypto.get_password("请输入加密密码: ")
+            
+            # 加密目录
+            result = Base::FileCrypto.encrypt_directory(@file_path, password, @output_path)
+            
+            puts "✅ 目录加密完成"
+            puts "📁 加密目录保存位置: #{result[:output_dir]}"
+            puts "📋 已加密 #{result[:encrypted_files].length} 个文件"
+            
+          rescue => e
+            puts "❌ 加密失败: #{e.message}"
+            exit 1
+          end
+        end
+      end
+    end
+  end
+end

data/lib/easyai/command/utils.rb

@@ -0,0 +1,32 @@
+module EasyAI
+  class Command
+    class Utils < Command
+      self.summary = '实用工具集合'
+      self.description = <<-DESC
+        提供各种实用工具。
+        
+        可用命令:
+
+            * encry - 文件加密工具
+
+            * decry - 文件解密工具
+
+        使用示例:
+
+            $ easyai utils encry file.txt       # 加密文件
+
+            $ easyai utils decry file.encrypted # 解密文件
+      DESC
+      
+      self.abstract_command = true
+      
+      def self.options
+        [].concat(super)
+      end
+    end
+  end
+end
+
+# 加载子命令
+require_relative 'utils/encry'
+require_relative 'utils/decry'

data/lib/easyai/command.rb

@@ -0,0 +1,56 @@
+require 'claide'
+require 'yaml'
+require 'fileutils'
+require 'colored2'
+require_relative 'base/system_keychain'
+require_relative 'base/version_checker'
+
+module EasyAI
+  class Command < CLAide::Command
+
+    self.abstract_command = true  
+    self.command = 'easyai'
+    self.version = VERSION
+    self.description = '简化 AI 命令行工具的 CLI 包装器，支持 Claude、Gemini 和 GPT'
+    self.plugin_prefixes = %w(easyai)
+
+    # 定义帮助标志访问器（类似 pindo）
+    attr_accessor :args_help_flag
+    alias_method :args_help_flag?, :args_help_flag
+
+    # 根命令的选项
+    DEFAULT_ROOT_OPTIONS = [
+      ['--version', '显示版本信息']
+    ]
+
+    # 所有命令通用的选项
+    DEFAULT_OPTIONS = [
+      ['--help', '显示此帮助信息']
+    ]
+
+    def self.options
+      if self == Command  # 根命令
+        DEFAULT_ROOT_OPTIONS + DEFAULT_OPTIONS
+      else  # 子命令
+        DEFAULT_OPTIONS
+      end
+    end
+    
+
+    def initialize(argv)
+      super
+      # 使用CLAide已经设置的help_arg，而不是重新获取
+      @args_help_flag = @help_arg
+    end
+
+    def validate!
+      super
+      help! if args_help_flag?
+    end
+
+    def self.run(argv)
+      super(argv)
+    end
+
+  end
+end