easyai 1.1.1 → 1.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 60e6aec7d6d0d2475a75e26f61f21f1f5354b75ba828006293cdcdd3a6961f8a
-  data.tar.gz: bd0c31dccfcb0918707972a72415382b6ca778015ae67932626e4ad40933b747
+  metadata.gz: 92482873797bd9a6936d1b493799244501b94440c155fe49e15b5aa889291ada
+  data.tar.gz: e3e619b6bd230d469dbb21fc2372e714813a504162e392680907f8fb3eec39aa
 SHA512:
-  metadata.gz: 31725ddd0f5c1ce5e3acda536b8c502addefc67271abb0b719521349fd3f3d840f2c41151982fbaca3044d34e6e110ef5e663155ee5c71ff636a6eb5af6b749a
-  data.tar.gz: 3045e64aa95c96a0ca58934f080dc6e82512121348290db4b4d8859b0af016f343893a8208f281f6be33af75afd4e2160fbc04451adf8f89ff9d62a21b202719
+  metadata.gz: 1bb5fef07b98087e297e3db2a37f29abe34966771a58fca4c6297f7c29cb2e0136ef97cd7467d2bf8eff730bb2e693334776eeb241d7148ea94b12fdaa7f148e
+  data.tar.gz: 5a46f13a20d253320c6a039802cbd4caf93780bd5e1570680b51442722c8a42f9eb1b8fa607d4f8eb11ccb2fcf936764d2ad8ba70653a97af9eec2d010bd4b90

data/lib/easyai/auth/jpslogin.rb

@@ -14,7 +14,7 @@ module EasyAI
     class JPSLogin
       attr_reader :access_token, :username, :expires_at
 
-      def initialize
+      def initialize(options = {})
         @client_id = "cli_a7bc7fe9b3d1d00b"
         @server_port = 8898
         @state = "client_login"
@@ -32,7 +32,10 @@ module EasyAI
         @access_token = nil
         @username = nil
         @expires_at = nil
-        
+
+        # 从选项中获取 verbose 标志
+        @verbose = options[:verbose] || false
+
         # token 存储路径
         @token_dir = File.expand_path('~/.easyai')
         @token_file = File.join(@token_dir, '.jpstoken')
@@ -44,9 +47,14 @@ module EasyAI
         if load_stored_token && validate_token
           return true
         end
-        
+
         puts "🔐 需要登录 JPS..."
-        return authorize_and_login
+        result = authorize_and_login
+
+        # 如果用户主动取消，返回特殊标识
+        return :user_cancelled if result == :user_cancelled
+
+        return result
       end
 
       # 获取用户名（登录后可用）
@@ -93,8 +101,12 @@ module EasyAI
         # 构建授权 URL
         authorization_uri = build_authorization_uri
         puts "正在打开浏览器进行飞书 OAuth 授权..."
-        puts "授权 URL: #{authorization_uri}"
-        
+        puts "\n授权 URL（如自动打开失败，请手动复制下面的链接到浏览器）:"
+        puts "=" * 80
+        puts authorization_uri
+        puts "=" * 80
+        puts ""
+
         # 在浏览器中打开授权 URL
         open_browser(authorization_uri)
         
@@ -103,33 +115,88 @@ module EasyAI
         
         # 如果自动获取失败，提示用户手动输入
         if code.nil?
-          puts "自动获取授权码失败，您可以手动输入："
-          puts "1. 授权码 (直接复制 'code=' 后面的内容)"
-          puts "2. 完整回调 URL"
-          print "> "
-          input = STDIN.gets.chomp
-          
-          if input.start_with?("http")
-            # 尝试从 URL 中提取 code
+          loop do
+            puts "\n自动获取授权码失败，请选择："
+            puts "1. 输入授权码 (直接复制 'code=' 后面的内容)"
+            puts "2. 输入完整回调 URL"
+            puts "3. 重新打开授权网页"
+            puts "4. 退出"
+            print "> "
+
             begin
-              uri = URI(input)
-              query_params = URI.decode_www_form(uri.query || '').to_h
-              code = query_params['code']
-              puts "从 URL 中成功提取授权码" if code
-            rescue => e
-              puts "无法从 URL 中提取授权码: #{e.message}"
+              choice = STDIN.gets&.chomp
+
+              # 处理 Ctrl+C 中断
+              if choice.nil?
+                puts "\n用户中断操作"
+                return :user_cancelled
+              end
+
+              case choice
+              when "1"
+                puts "请输入授权码："
+                print "> "
+                code_input = STDIN.gets&.chomp
+                if code_input.nil?
+                  puts "用户中断操作"
+                  return :user_cancelled
+                elsif !code_input.empty?
+                  code = code_input
+                  break
+                else
+                  puts "授权码不能为空，请重新选择"
+                end
+              when "2"
+                puts "请输入完整回调 URL："
+                print "> "
+                url_input = STDIN.gets&.chomp
+                if url_input.nil?
+                  puts "用户中断操作"
+                  return :user_cancelled
+                elsif url_input.start_with?("http")
+                  # 尝试从 URL 中提取 code
+                  begin
+                    uri = URI(url_input)
+                    query_params = URI.decode_www_form(uri.query || '').to_h
+                    code = query_params['code']
+                    if code
+                      puts "✓ 从 URL 中成功提取授权码"
+                      break
+                    else
+                      puts "✗ URL 中没有找到授权码，请重新选择"
+                    end
+                  rescue => e
+                    puts "✗ 无法从 URL 中提取授权码: #{e.message}"
+                  end
+                else
+                  puts "✗ 无效的 URL 格式，请重新选择"
+                end
+              when "3"
+                # 重新打开授权网页
+                puts "正在重新打开授权网页..."
+                open_browser(authorization_uri)
+                # 重新启动服务器尝试获取授权码
+                code = start_callback_server
+                if code
+                  break
+                end
+                # 如果还是失败，继续循环让用户选择
+              when "4"
+                puts "已退出授权流程"
+                return :user_cancelled
+              else
+                puts "无效的选择，请输入 1-4 之间的数字"
+              end
+            rescue Interrupt
+              puts "\n\n用户中断操作"
+              return :user_cancelled
             end
-          else
-            # 将输入直接作为 code
-            code = input unless input.empty?
           end
         end
         
         if code
-          puts "成功获取授权码！正在使用飞书身份登录 JPS..."
           if exchange_code_for_token(code)
-            puts "✓ JPS 登录成功！"
-            puts "  用户名: #{@username}"
+            puts "✓ JPS 登录成功！用户名: #{@username}"
             store_token
             return true
           end
@@ -181,8 +248,10 @@ module EasyAI
           return nil
         end
         
-        puts "启动本地服务器，监听端口 #{@server_port}..."
-        
+        puts "启动本地服务器，监听端口 #{@server_port}..." if @verbose
+        puts "提示：按 Ctrl+C 可以中断并获得更多选择"
+        puts "🔄 正在使用飞书身份登录 JPS..."
+
         begin
           server = WEBrick::HTTPServer.new(
             Port: @server_port,
@@ -209,19 +278,19 @@ module EasyAI
           begin
             # 解析请求参数
             query_params = URI.decode_www_form(req.query_string || '').to_h
-            puts "接收到回调请求，参数: #{query_params.inspect}"
+            puts "接收到回调请求，参数: #{query_params.inspect}" if @verbose
             
             if query_params['error']
-              puts "授权错误: #{query_params['error']}"
+              puts "授权错误: #{query_params['error']}" if @verbose
               res.content_type = "text/html; charset=UTF-8"
               res.body = build_error_page(query_params['error'])
             elsif query_params['code']
               code = query_params['code']
-              puts "成功获取授权码"
+              puts "成功获取授权码" if @verbose
               res.content_type = "text/html; charset=UTF-8"
               res.body = build_success_page
             else
-              puts "未获取到授权码"
+              puts "未获取到授权码" if @verbose
               res.content_type = "text/html; charset=UTF-8"
               res.body = build_error_page("未获取到授权码")
             end
@@ -275,14 +344,14 @@ module EasyAI
           request['Content-Type'] = 'application/json'
           request.body = request_data.to_json
           
-          puts "正在请求 JPS API: #{@api_endpoint}"
+          puts "正在请求 JPS API: #{@api_endpoint}" if @verbose
           response = http.request(request)
-          
-          puts "API 响应状态码: #{response.code}"
+
+          puts "API 响应状态码: #{response.code}" if @verbose
           
           if response.body
             result = JSON.parse(response.body)
-            puts "API 响应: #{result.inspect}"
+            puts "API 响应: #{result.inspect}" if @verbose
             
             if result['meta'] && result['meta']['code'] == 200 && result['data']
               data = result['data']
@@ -322,7 +391,7 @@ module EasyAI
         }
         
         File.write(@token_file, token_data.to_json)
-        puts "✓ JPS Token 和用户名已存储到 #{@token_file}"
+        puts "✓ JPS Token 和用户名已存储到 #{@token_file}" if @verbose
       rescue => e
         puts "⚠ 存储 JPS Token 失败: #{e.message}"
       end

data/lib/easyai/command/claude.rb

@@ -72,8 +72,15 @@ module EasyAI
         else
           # 从远程下载配置，传递选项
           print_status("🔄 获取远程配置", "默认用户")
-          options = { no_keychain: @no_keychain }
+          options = { no_keychain: @no_keychain, verbose: @verbose_mode }
           remote_config = ConfigManager.download_user_config(nil, options)
+
+          # 处理用户取消授权的情况
+          if remote_config == :user_cancelled
+            print_error("用户取消了授权登录")
+            exit 0
+          end
+
           print_success("配置加载成功") if remote_config
         end
 
@@ -81,7 +88,7 @@ module EasyAI
         if remote_config.nil?
           print_warning("使用本地配置")
           remote_config = load_local_yaml_config
-          
+
           # 如果本地配置也为空，提示用户先进行设置
           if remote_config.empty?
             print_error("未找到有效配置")

data/lib/easyai/config/config.rb

@@ -14,6 +14,7 @@ module EasyAI
     
     # 类变量初始化
     @@no_keychain = false
+    @@verbose = false
     
     # 管理配置仓库：如果存在则更新，不存在则下载
     def self.manage_config_repo
@@ -65,7 +66,7 @@ module EasyAI
     
     # 下载配置仓库到固定位置
     def self.download_config_repo
-      puts "正在下载配置仓库到 #{CONFIG_REPO_DIR}..."
+      puts "正在下载配置仓库到 #{CONFIG_REPO_DIR}..." if @@verbose
       
       begin
         # 克隆仓库到固定位置，捕获错误信息
@@ -94,8 +95,9 @@ module EasyAI
       
       # 设置全局选项
       @@no_keychain = options[:no_keychain] || false
+      @@verbose = options[:verbose] || false
       
-      puts "正在从本地配置仓库加载配置..."
+      puts "正在从本地配置仓库加载配置..." if @@verbose
       
       begin
         # 检查 index.json 文件，支持加密版本
@@ -134,8 +136,13 @@ module EasyAI
           selected_user = user_name
         else
           # 统一使用 JPS 登录获取用户名
-          selected_user = get_username_from_jps
-          if selected_user.nil? || selected_user.strip.empty?
+          selected_user = get_username_from_jps(@@verbose)
+
+          # 处理用户取消的情况
+          if selected_user == :user_cancelled
+            puts "✗ 用户取消了授权登录"
+            return :user_cancelled
+          elsif selected_user.nil? || selected_user.strip.empty?
             puts "✗ JPS 登录失败，无法获取用户名"
             return nil
           end
@@ -176,8 +183,14 @@ module EasyAI
     def self.download_user_config(user_name = nil, options = {})
       # 优先尝试使用本地配置仓库
       config = load_from_local_repo(user_name, options)
+
+      # 如果用户取消了授权，返回特殊标识
+      if config == :user_cancelled
+        return :user_cancelled
+      end
+
       return config if config
-      
+
       # 如果本地仓库失败，回退到临时下载方式
       puts "本地配置仓库不可用，使用临时下载..."
       download_user_config_temp(user_name, options)
@@ -189,6 +202,7 @@ module EasyAI
       
       # 设置全局选项
       @@no_keychain = options[:no_keychain] || false
+      @@verbose = options[:verbose] || false
       
       begin
         # 创建临时目录
@@ -248,8 +262,14 @@ module EasyAI
           selected_user = user_name
         else
           # 统一使用 JPS 登录获取用户名
-          selected_user = get_username_from_jps
-          if selected_user.nil? || selected_user.strip.empty?
+          selected_user = get_username_from_jps(@@verbose)
+
+          # 处理用户取消的情况
+          if selected_user == :user_cancelled
+            puts "✗ 用户取消了授权登录"
+            cleanup_temp_dir(temp_dir)
+            return nil
+          elsif selected_user.nil? || selected_user.strip.empty?
             puts "✗ JPS 登录失败，无法获取用户名"
             cleanup_temp_dir(temp_dir)
             return nil
@@ -315,11 +335,17 @@ module EasyAI
     end
     
     # 通过 JPS 登录获取用户名
-    def self.get_username_from_jps      
+    def self.get_username_from_jps(verbose = false)
       begin
-        jps_login = Auth::JPSLogin.new
-        
-        if jps_login.login
+        jps_login = Auth::JPSLogin.new(verbose: verbose)
+        login_result = jps_login.login
+
+        # 处理用户主动取消的情况
+        if login_result == :user_cancelled
+          return :user_cancelled
+        end
+
+        if login_result
           username = jps_login.get_username
           if username && !username.empty?
             puts "👤 用户: #{username}"
@@ -329,7 +355,7 @@ module EasyAI
       rescue => e
         puts "❌ JPS 登录失败: #{e.message}"
       end
-      
+
       return nil
     end
     
@@ -356,7 +382,7 @@ module EasyAI
     end
     
     def self.load_encrypted_config_file(encrypted_config_file)
-      puts "正在解密配置文件..."
+      puts "正在解密配置文件..." if @@verbose
       
       # 使用新的密码验证机制
       result = get_and_validate_password(encrypted_config_file, "请输入解密密码: ")
@@ -373,7 +399,7 @@ module EasyAI
     end
     
     def self.parse_encrypted_index_file(encrypted_index_file)
-      puts "正在解密 index.json..."
+      puts "正在解密 index.json..." if @@verbose
       
       # 使用新的密码验证机制
       result = get_and_validate_password(encrypted_index_file, "请输入解密密码: ")
@@ -472,7 +498,7 @@ module EasyAI
       stored_password = Base::SystemKeychain.get_stored_password
       
       if stored_password && !stored_password.empty?
-        puts "使用系统存储的密码进行解密..."
+        puts "使用系统存储的密码进行解密..." if @@verbose
         return { password: stored_password, from_system: true }
       end
       

data/lib/easyai/version.rb

@@ -1,3 +1,3 @@
 module EasyAI
-  VERSION = '1.1.1'
+  VERSION = '1.1.3'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: easyai
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.1.3
 platform: ruby
 authors:
 - Wade