ice-jade 0.1.0 → 0.5.0

data/lib/http_poster.rb

@@ -0,0 +1,225 @@
+# http_poster.rb
+# Ruby 纯标准库 HTTP POST 工具模块
+# 零外部依赖，支持 JSON / Form / Multipart
+#
+# =============================================================================
+# 说明：本模块是独立工具脚本，不依赖 ice-jade gem。
+#
+#   如果你正在使用 ice-jade 项目，推荐优先使用 IceJade::Poster::Client
+#   （位于 lib/ice_jade/poster/client.rb），原因：
+#
+#   - 面向对象：初始化时配置 base_url/headers/timeout，后续复用
+#   - 统一响应：返回 IceJade::Response（ok, code, message, data, raw）
+#   - 异常友好：超时/HTTP 错误均包装为 Response，不抛异常
+#   - 与 Quantum 分支共享同一套响应体系
+#
+#   HttpPoster 适合：
+#   - 不想引入任何 gem 依赖的独立脚本
+#   - 快速测试、一次性请求
+#   - 学习/教学用途，代码更扁平直观
+#
+#   对比示例见：examples/comparison_poster.rb
+# =============================================================================
+
+require 'net/http'
+require 'json'
+require 'uri'
+require 'securerandom'
+
+module HttpPoster
+  class Error < StandardError; end
+  class TimeoutError < Error; end
+  class HttpError < Error; end
+
+  # ============ 核心 POST 方法 ============
+  #
+  # @param url      [String]  完整 URL 或路径
+  # @param params   [Hash]    提交参数
+  # @param headers  [Hash]    自定义请求头（自动覆盖默认头）
+  # @param opts     [Hash]    选项: :timeout, :open_timeout, :max_retries
+  #
+  # @return [Hash/String] 解析后的 JSON 或原始响应体
+  #
+  def self.post(url, params = {}, headers = {}, opts = {})
+    uri = URI.parse(url)
+    content_type = headers['Content-Type'] || headers[:content_type] || 'application/json'
+
+    req = Net::HTTP::Post.new(uri)
+    req['User-Agent'] = headers['User-Agent'] || 'ruby-http-poster/1.0'
+    req['Accept'] = headers['Accept'] || 'application/json'
+
+    # 认证头（支持 Bearer Token）
+    if headers['Authorization'] || headers[:authorization]
+      req['Authorization'] = headers['Authorization'] || headers[:authorization]
+    end
+
+    # 根据 Content-Type 序列化参数
+    case content_type.to_s
+    when /application\/json/
+      req['Content-Type'] = 'application/json'
+      req.body = params.to_json
+    when /application\/x-www-form-urlencoded/
+      req['Content-Type'] = 'application/x-www-form-urlencoded'
+      req.body = URI.encode_www_form(params)
+    when /multipart\/form-data/
+      body, boundary = build_multipart(params)
+      req['Content-Type'] = "multipart/form-data; boundary=#{boundary}"
+      req.body = body
+    else
+      req['Content-Type'] = content_type
+      req.body = params.is_a?(String) ? params : params.to_json
+    end
+
+    # 追加其余自定义头
+    headers.each do |k, v|
+      next if %w[Content-Type Authorization User-Agent Accept].include?(k.to_s)
+      req[k.to_s] = v.to_s
+    end
+
+    execute(uri, req, opts)
+  end
+
+  # ============ 快捷方法 ============
+
+  # POST JSON（API 调用最常用）
+  def self.post_json(url, params = {}, headers = {}, opts = {})
+    headers = headers.merge('Content-Type' => 'application/json')
+    post(url, params, headers, opts)
+  end
+
+  # POST Form（传统表单提交）
+  def self.post_form(url, params = {}, headers = {}, opts = {})
+    headers = headers.merge('Content-Type' => 'application/x-www-form-urlencoded')
+    post(url, params, headers, opts)
+  end
+
+  # POST Multipart（文件上传）
+  # params 示例: { file: '/path/to/file.png', purpose: 'image' }
+  def self.post_multipart(url, params = {}, headers = {}, opts = {})
+    headers = headers.merge('Content-Type' => 'multipart/form-data')
+    post(url, params, headers, opts)
+  end
+
+  private
+
+  def self.execute(uri, request, opts = {})
+    timeout = opts[:timeout] || 60
+    open_timeout = opts[:open_timeout] || 10
+    max_retries = opts[:max_retries] || 2
+
+    http = Net::HTTP.new(uri.host, uri.port)
+    http.use_ssl = uri.scheme == 'https'
+    http.open_timeout = open_timeout
+    http.read_timeout = timeout
+
+    retries = 0
+    begin
+      response = http.request(request)
+      parse(response)
+    rescue Net::OpenTimeout, Net::ReadTimeout => e
+      retries += 1
+      if retries <= max_retries
+        sleep(2 ** retries)
+        retry
+      end
+      raise TimeoutError, "请求超时（已重试 #{max_retries} 次）: #{e.message}"
+    ensure
+      http.finish if http.started?
+    end
+  end
+
+  def self.parse(response)
+    body = response.body.to_s
+    code = response.code.to_i
+
+    if code >= 400
+      begin
+        err = JSON.parse(body)
+        msg = err.dig('error', 'message') || body
+      rescue
+        msg = body
+      end
+      raise HttpError, "HTTP #{code}: #{msg}"
+    end
+
+    JSON.parse(body) rescue body
+  end
+
+  def self.build_multipart(params)
+    boundary = "----RubyFormBoundary#{SecureRandom.hex(16)}"
+    parts = []
+
+    params.each do |key, value|
+      if value.is_a?(String) && File.exist?(value)
+        # 文件字段
+        filename = File.basename(value)
+        data = File.binread(value)
+        mime = mime_type(filename)
+        parts << "--#{boundary}\r\n" \
+                  "Content-Disposition: form-data; name=\"#{key}\"; filename=\"#{filename}\"\r\n" \
+                  "Content-Type: #{mime}\r\n\r\n" \
+                  "#{data}\r\n"
+      else
+        # 普通字段
+        parts << "--#{boundary}\r\n" \
+                  "Content-Disposition: form-data; name=\"#{key}\"\r\n\r\n" \
+                  "#{value}\r\n"
+      end
+    end
+
+    body = parts.join + "--#{boundary}--\r\n"
+    [body, boundary]
+  end
+
+  def self.mime_type(filename)
+    ext = File.extname(filename).downcase
+    case ext
+    when '.png'  then 'image/png'
+    when '.jpg', '.jpeg' then 'image/jpeg'
+    when '.gif'  then 'image/gif'
+    when '.webp' then 'image/webp'
+    when '.pdf'  then 'application/pdf'
+    when '.txt'  then 'text/plain'
+    when '.json' then 'application/json'
+    else 'application/octet-stream'
+    end
+  end
+end
+
+# ============ 使用示例 ============
+if __FILE__ == $0
+  # 示例 1: POST JSON（调用 Kimi Code API）
+  begin
+    res = HttpPoster.post_json(
+      'https://api.kimi.com/coding/v1/chat/completions',
+      {
+        model: 'kimi-for-coding',
+        messages: [
+          { role: 'user', content: '用一句话解释BGP Flowspec' }
+        ],
+        temperature: 0.3
+      },
+      {
+        'Authorization' => 'Bearer sk-kimi-your-key-here',
+        'User-Agent' => 'my-bot/1.0'
+      },
+      timeout: 120
+    )
+    puts res.dig('choices', 0, 'message', 'content')
+  rescue HttpPoster::HttpError => e
+    puts "请求失败: #{e.message}"
+  end
+
+  # 示例 2: POST Form（传统表单）
+  # res = HttpPoster.post_form(
+  #   'https://httpbin.org/post',
+  #   { username: 'admin', password: 'secret' }
+  # )
+
+  # 示例 3: POST Multipart（上传文件）
+  # res = HttpPoster.post_multipart(
+  #   'https://api.kimi.com/coding/v1/files',
+  #   { file: './screenshot.png', purpose: 'image' },
+  #   { 'Authorization' => 'Bearer sk-kimi-xxx' }
+  # )
+end

data/lib/ice_jade/cradle/server.rb

@@ -0,0 +1,402 @@
+# frozen_string_literal: true
+
+require 'socket'
+require 'json'
+require 'uri'
+
+module IceJade
+  module Cradle
+    # 简易 HTTP 测试服务器
+    #
+    # 基于 Ruby 标准库 socket + thread，零外部依赖。
+    # 内置常用测试路由，支持 JSON / Form / Multipart，
+    # 用于本地测试 ice-jade 各分支（Quantum / Poster）的 HTTP 调用。
+    #
+    # @example 启动服务器
+    #   server = IceJade::Cradle::Server.new(port: 8765)
+    #   server.start
+    #
+    class Server
+      attr_reader :port, :host, :routes
+
+      DEFAULT_PORT = 8765
+      DEFAULT_HOST = '0.0.0.0'
+
+      # @param port [Integer] 监听端口
+      # @param host [String]  监听地址
+      # @param silent [Boolean] 是否静默输出（测试用）
+      def initialize(port: DEFAULT_PORT, host: DEFAULT_HOST, silent: false)
+        @port   = port
+        @host   = host
+        @silent = silent
+        @routes = []
+        @server = nil
+        @running = false
+        setup_default_routes
+      end
+
+      # 注册路由
+      #
+      # @param method [String, Symbol, :all] HTTP 方法，:all 表示匹配所有
+      # @param path   [String] 路径，支持 `:param` 占位符
+      # @yieldparam req [HTTPRequest]
+      # @yieldreturn [Array(Integer, Hash, String)] [status, headers, body]
+      def route(method, path, &block)
+        @routes << Route.new(method.to_s.upcase, path, block)
+      end
+
+      # 启动服务器（阻塞）
+      def start
+        @server = TCPServer.new(@host, @port)
+        @running = true
+        log "Cradle HTTP Server listening on http://#{@host}:#{@port}"
+        log "Press Ctrl+C to stop"
+
+        while @running
+          begin
+            socket = @server.accept
+            Thread.new { handle_connection(socket) }
+          rescue IOError
+            break unless @running
+          end
+        end
+      end
+
+      # 停止服务器
+      def stop
+        @running = false
+        @server&.close
+        log "Server stopped."
+      end
+
+      private
+
+      def handle_connection(socket)
+        req = HTTPRequest.parse(socket)
+        return socket.close unless req
+
+        matched = match_route(req)
+        status, headers, body = if matched
+                                  begin
+                                    matched.call(req)
+                                  rescue StandardError => e
+                                    [500, { 'Content-Type' => 'application/json' },
+                                     json_body(error: e.class.name, message: e.message)]
+                                  end
+                                else
+                                  [404, { 'Content-Type' => 'application/json' },
+                                   json_body(error: 'Not Found', path: req.path, method: req.method)]
+                                end
+
+        send_response(socket, status, headers, body)
+        socket.close
+      rescue StandardError => e
+        socket.close
+      end
+
+      def match_route(req)
+        route = @routes.find { |r| r.match?(req.method, req.path) }
+        return nil unless route
+
+        lambda do |request|
+          params = route.extract_params(request.path)
+          request.route_params = params
+          route.handler.call(request)
+        end
+      end
+
+      def send_response(socket, status, headers, body)
+        status_text = HTTP_STATUS[status] || 'Unknown'
+        socket.write "HTTP/1.1 #{status} #{status_text}\r\n"
+        headers.each { |k, v| socket.write "#{k}: #{v}\r\n" }
+        socket.write "Content-Length: #{body.bytesize}\r\n" unless body.nil?
+        socket.write "Connection: close\r\n"
+        socket.write "\r\n"
+        socket.write body.to_s
+      end
+
+      def json_body(hash)
+        JSON.generate(hash)
+      end
+
+      def log(msg)
+        return if @silent
+        puts "[#{Time.now.strftime('%Y-%m-%d %H:%M:%S')}] #{msg}"
+      end
+
+      # ------------------------------------------------------------------
+      # 内置测试路由
+      # ------------------------------------------------------------------
+      def setup_default_routes
+        # 健康检查
+        route(:all, '/health') do |req|
+          [200, { 'Content-Type' => 'application/json' },
+           json_body(status: 'ok', method: req.method, time: Time.now.to_i)]
+        end
+
+        # GET 回显查询参数
+        route(:get, '/echo') do |req|
+          [200, { 'Content-Type' => 'application/json' },
+           json_body(method: req.method, path: req.path, query: req.query, headers: req.headers)]
+        end
+
+        # POST 回显请求体（JSON / Form）
+        route(:post, '/echo') do |req|
+          [200, { 'Content-Type' => 'application/json' },
+           json_body(method: req.method, path: req.path,
+                     content_type: req.content_type, body: req.parsed_body)]
+        end
+
+        # POST 接收 multipart 上传
+        route(:post, '/upload') do |req|
+          files = req.files.transform_values do |v|
+            { filename: v[:filename], size: v[:data].bytesize,
+              content_type: v[:content_type] }
+          end
+          fields = req.form_data.reject { |_, v| v.is_a?(Hash) && v[:filename] }
+          [200, { 'Content-Type' => 'application/json' },
+           json_body(files: files, fields: fields, content_type: req.content_type)]
+        end
+
+        # GET /status/:code —— 返回指定状态码
+        route(:get, '/status/:code') do |req|
+          code = req.route_params['code'].to_i
+          status_text = HTTP_STATUS[code] || 'Unknown'
+          [code, { 'Content-Type' => 'application/json' },
+           json_body(code: code, status: status_text)]
+        end
+
+        # POST /delay/:seconds —— 延迟响应
+        route(:post, '/delay/:seconds') do |req|
+          seconds = req.route_params['seconds'].to_f
+          sleep(seconds)
+          [200, { 'Content-Type' => 'application/json' },
+           json_body(delayed: seconds, message: 'Delayed response')]
+        end
+
+        # GET /headers —— 回显请求头
+        route(:get, '/headers') do |req|
+          [200, { 'Content-Type' => 'application/json' },
+           json_body(headers: req.headers)]
+        end
+
+        # POST /form —— 接收 form-urlencoded
+        route(:post, '/form') do |req|
+          [200, { 'Content-Type' => 'application/json' },
+           json_body(form: req.parsed_body)]
+        end
+
+        # GET / —— 欢迎页
+        route(:get, '/') do |req|
+          body = <<~HTML
+            <!DOCTYPE html>
+            <html><head><title>Cradle</title></head>
+            <body>
+              <h1>🍼 Cradle HTTP Test Server</h1>
+              <p>ice-jade 内置测试服务器，零外部依赖。</p>
+              <h2>内置路由</h2>
+              <ul>
+                <li><code>GET  /health</code> — 健康检查</li>
+                <li><code>GET  /echo?foo=bar</code> — 回显查询参数</li>
+                <li><code>POST /echo</code> — 回显请求体（JSON / Form）</li>
+                <li><code>POST /upload</code> — 接收 multipart 文件上传</li>
+                <li><code>GET  /status/:code</code> — 返回指定 HTTP 状态码</li>
+                <li><code>POST /delay/:seconds</code> — 延迟响应</li>
+                <li><code>GET  /headers</code> — 回显请求头</li>
+                <li><code>POST /form</code> — 接收 form-urlencoded</li>
+              </ul>
+              <p>监听端口: #{@port}</p>
+            </body></html>
+          HTML
+          [200, { 'Content-Type' => 'text/html; charset=utf-8' }, body]
+        end
+      end
+
+      # =================================================================
+      # Route 定义
+      # =================================================================
+      class Route
+        attr_reader :method, :path_pattern, :handler
+
+        def initialize(method, path_pattern, handler)
+          @method = method == 'ALL' ? nil : method
+          @path_pattern = path_pattern
+          @handler = handler
+          @regex, @param_names = compile(path_pattern)
+        end
+
+        def match?(http_method, path)
+          return false if @method && @method != http_method
+          path_without_query = path.split('?').first
+          @regex.match?(path_without_query)
+        end
+
+        def extract_params(path)
+          path_without_query = path.split('?').first
+          match = @regex.match(path_without_query)
+          return {} unless match
+
+          @param_names.each_with_object({}).with_index do |(name, hash), i|
+            hash[name] = match[i + 1]
+          end
+        end
+
+        private
+
+        def compile(pattern)
+          names = []
+          regex_str = pattern.split('/').map do |segment|
+            if segment.start_with?(':')
+              names << segment[1..-1]
+              '([^/]+)'
+            else
+              Regexp.escape(segment)
+            end
+          end.join('/')
+          [Regexp.new("^#{regex_str}$"), names]
+        end
+      end
+
+      # =================================================================
+      # HTTP 请求解析
+      # =================================================================
+      class HTTPRequest
+        attr_reader :method, :path, :query, :headers, :body, :content_type
+        attr_accessor :route_params
+
+        def initialize(method, path, query, headers, body)
+          @method = method
+          @path = path
+          @query = query
+          @headers = headers
+          @body = body
+          @content_type = headers['Content-Type']&.split(';')&.first&.strip
+          @route_params = {}
+        end
+
+        def self.parse(socket)
+          line = read_line(socket)
+          return nil unless line
+
+          method, full_path, _protocol = line.split(' ', 3)
+          return nil unless method && full_path
+
+          uri = URI.parse(full_path)
+          query = URI.decode_www_form(uri.query || '').to_h
+
+          headers = {}
+          while (header_line = read_line(socket))
+            break if header_line == "\r\n" || header_line == "\n" || header_line.empty?
+            key, value = header_line.split(':', 2)
+            headers[key.strip] = value.strip if key && value
+          end
+
+          body = ''
+          if headers['Content-Length']
+            length = headers['Content-Length'].to_i
+            body = socket.read(length) if length > 0
+          end
+
+          new(method, uri.path, query, headers, body)
+        rescue StandardError
+          nil
+        end
+
+        def parsed_body
+          case content_type
+          when 'application/json'
+            JSON.parse(body) rescue body
+          when 'application/x-www-form-urlencoded'
+            URI.decode_www_form(body).to_h rescue body
+          else
+            body
+          end
+        end
+
+        # multipart 解析结果
+        def files
+          return {} unless multipart?
+          parse_multipart[:files]
+        end
+
+        def form_data
+          return {} unless multipart?
+          parse_multipart[:fields]
+        end
+
+        def multipart?
+          content_type == 'multipart/form-data' ||
+            headers['Content-Type']&.include?('multipart/form-data')
+        end
+
+        private
+
+        def self.read_line(socket)
+          line = +""
+          while (char = socket.getc)
+            line << char
+            break if line.end_with?("\r\n")
+          end
+          line
+        rescue StandardError
+          nil
+        end
+
+        def parse_multipart
+          ct = headers['Content-Type'] || ''
+          boundary = ct[/boundary=([^\s;]+)/, 1]
+          boundary = boundary[1..-2] if boundary&.start_with?('"') && boundary&.end_with?('"')
+
+          files = {}
+          fields = {}
+          return { files: files, fields: fields } unless boundary
+
+          delimiter = "--#{boundary}"
+          parts = @body.split("#{delimiter}\r\n")
+
+          parts.each do |part|
+            part = part.sub(/\r\n--#{Regexp.escape(boundary)}--\r\n?\z/, '')
+            next if part.strip.empty?
+
+            header_end = part.index("\r\n\r\n")
+            next unless header_end
+
+            header_section = part[0...header_end]
+            data = part[(header_end + 4)..-1]
+            data = data.chomp("\r\n") if data
+
+            disp = header_section[/Content-Disposition: (.+)/i, 1]
+            next unless disp
+
+            name = disp[/name="([^"]+)"/, 1]
+            filename = disp[/filename="([^"]+)"/, 1]
+            mime = header_section[/Content-Type: (.+)/i, 1]&.strip
+
+            if filename
+              files[name] = { filename: filename, content_type: mime || 'application/octet-stream', data: data || '' }
+            else
+              fields[name] = data || ''
+            end
+          end
+
+          { files: files, fields: fields }
+        end
+      end
+
+      # =================================================================
+      # HTTP 状态码表
+      # =================================================================
+      HTTP_STATUS = {
+        100 => 'Continue', 101 => 'Switching Protocols',
+        200 => 'OK', 201 => 'Created', 202 => 'Accepted',
+        204 => 'No Content',
+        301 => 'Moved Permanently', 302 => 'Found', 304 => 'Not Modified',
+        400 => 'Bad Request', 401 => 'Unauthorized', 403 => 'Forbidden',
+        404 => 'Not Found', 405 => 'Method Not Allowed',
+        429 => 'Too Many Requests',
+        500 => 'Internal Server Error', 502 => 'Bad Gateway',
+        503 => 'Service Unavailable'
+      }.freeze
+    end
+  end
+end