onebusaway-sdk 0.1.0.pre.alpha.205 → 0.1.0.pre.alpha.206

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 429ccc3f00c50a5b1de30b48e7f6b1b050bd1b17c397a6a752c8e9e46d469184
-  data.tar.gz: fa2d71c19e522f6baa5d9b11afdd5876707e2197791d3dff8c3e1d79c82029c2
+  metadata.gz: 75896ea053e81f2065eafd3e5dca608343251ec36a36f651f784fee694daf5a8
+  data.tar.gz: 548e74e04af38ba29427750afe087efe67582b4f25af60493a331abb656c74ac
 SHA512:
-  metadata.gz: e86dc1479c36ee96fc7ac234fbd5df45f67590f5f3854a2bf20532e6624b8e423b56ae34002f7bdfa105dddbfab46c8df07b3ab2743ecdd442518fde9d2cd422
-  data.tar.gz: a13e694db85c0bae734ca456560d1c16e107bb92caadc322154c69e14a1a8d9a8456e76ea6f1060c93192c5d2c4179ffcde3b862e047844c424b4c92dd804a2d
+  metadata.gz: 63c34916992cb26b12ec8d3ad2a067608d5d2fd912e916cc6db8f340ba89a11f5ae3d03dfa2de2dee6bb2bf8f855f6313cda9a7c7aa7a1d76351d8200609ddc6
+  data.tar.gz: 33b8ddc65ca3dd17ba252facdb4515c8f41508b9ad1f0ecbee755957bb2dd42b25d108e795ecfc098339b22ea61c47174532679545971c88787e3153dc4ab6ab

data/lib/onebusaway-sdk/base_client.rb

@@ -28,7 +28,7 @@ module OnebusawaySDK
       # @raise [ArgumentError]
       #
       def validate!(req)
-        keys = [:method, :path, :query, :headers, :body, :unwrap, :page, :model, :options]
+        keys = [:method, :path, :query, :headers, :body, :unwrap, :page, :stream, :model, :options]
         case req
         in Hash
           req.each_key do |k|
@@ -201,6 +201,8 @@ module OnebusawaySDK
     #
     #   @option req [Class, nil] :page
     #
+    #   @option req [Class, nil] :stream
+    #
     #   @option req [OnebusawaySDK::Converter, Class, nil] :model
     #
     # @param opts [Hash{Symbol=>Object}] .
@@ -318,7 +320,7 @@ module OnebusawaySDK
     # @param send_retry_header [Boolean]
     #
     # @raise [OnebusawaySDK::APIError]
-    # @return [Array(Net::HTTPResponse, Enumerable)]
+    # @return [Array(Integer, Net::HTTPResponse, Enumerable)]
     #
     private def send_request(request, redirect_count:, retry_count:, send_retry_header:)
       url, headers, max_retries, timeout = request.fetch_values(:url, :headers, :max_retries, :timeout)
@@ -341,7 +343,7 @@ module OnebusawaySDK
 
       case status
       in ..299
-        [response, stream]
+        [status, response, stream]
       in 300..399 if redirect_count >= self.class::MAX_REDIRECTS
         message = "Failed to complete the request within #{self.class::MAX_REDIRECTS} redirects."
 
@@ -359,13 +361,15 @@ module OnebusawaySDK
         )
       in OnebusawaySDK::APIConnectionError if retry_count >= max_retries
         raise status
-      in (400..) if retry_count >= max_retries || (response && !self.class.should_retry?(
-        status,
-        headers: response
-      ))
+      in (400..) if retry_count >= max_retries || !self.class.should_retry?(status, headers: response)
         decoded = OnebusawaySDK::Util.decode_content(response, stream: stream, suppress_error: true)
 
-        stream.each { srv_fault ? break : next }
+        if srv_fault
+          OnebusawaySDK::Util.close_fused!(stream)
+        else
+          stream.each { next }
+        end
+
         raise OnebusawaySDK::APIStatusError.for(
           url: url,
           status: status,
@@ -376,7 +380,11 @@ module OnebusawaySDK
       in (400..) | OnebusawaySDK::APIConnectionError
         delay = retry_delay(response, retry_count: retry_count)
 
-        stream&.each { srv_fault ? break : next }
+        if srv_fault
+          OnebusawaySDK::Util.close_fused!(stream)
+        else
+          stream&.each { next }
+        end
         sleep(delay)
 
         send_request(
@@ -388,48 +396,6 @@ module OnebusawaySDK
       end
     end
 
-    # @private
-    #
-    # @param req [Hash{Symbol=>Object}] .
-    #
-    #   @option req [Symbol] :method
-    #
-    #   @option req [String, Array<String>] :path
-    #
-    #   @option req [Hash{String=>Array<String>, String, nil}, nil] :query
-    #
-    #   @option req [Hash{String=>String, Integer, Array<String, Integer, nil>, nil}, nil] :headers
-    #
-    #   @option req [Object, nil] :body
-    #
-    #   @option req [Symbol, nil] :unwrap
-    #
-    #   @option req [Class, nil] :page
-    #
-    #   @option req [OnebusawaySDK::Converter, Class, nil] :model
-    #
-    #   @option req [OnebusawaySDK::RequestOptions, Hash{Symbol=>Object}, nil] :options
-    #
-    # @param headers [Hash{String=>String}, Net::HTTPHeader]
-    #
-    # @param stream [Enumerable]
-    #
-    # @return [Object]
-    #
-    private def parse_response(req, headers:, stream:)
-      decoded = OnebusawaySDK::Util.decode_content(headers, stream: stream)
-      unwrapped = OnebusawaySDK::Util.dig(decoded, req[:unwrap])
-
-      case [req[:page], req.fetch(:model, OnebusawaySDK::Unknown)]
-      in [Class => page, _]
-        page.new(client: self, req: req, headers: headers, unwrapped: unwrapped)
-      in [nil, Class | OnebusawaySDK::Converter => model]
-        OnebusawaySDK::Converter.coerce(model, unwrapped)
-      in [nil, nil]
-        unwrapped
-      end
-    end
-
     # Execute the request specified by `req`. This is the method that all resource
     #   methods call into.
     #
@@ -449,6 +415,8 @@ module OnebusawaySDK
     #
     #   @option req [Class, nil] :page
     #
+    #   @option req [Class, nil] :stream
+    #
     #   @option req [OnebusawaySDK::Converter, Class, nil] :model
     #
     #   @option req [OnebusawaySDK::RequestOptions, Hash{Symbol=>Object}, nil] :options
@@ -458,19 +426,31 @@ module OnebusawaySDK
     #
     def request(req)
       self.class.validate!(req)
+      model = req.fetch(:model) { OnebusawaySDK::Unknown }
       opts = req[:options].to_h
       OnebusawaySDK::RequestOptions.validate!(opts)
       request = build_request(req.except(:options), opts)
+      url = request.fetch(:url)
 
       # Don't send the current retry count in the headers if the caller modified the header defaults.
       send_retry_header = request.fetch(:headers)["x-stainless-retry-count"] == "0"
-      response, stream = send_request(
+      status, response, stream = send_request(
         request,
         redirect_count: 0,
         retry_count: 0,
         send_retry_header: send_retry_header
       )
-      parse_response(req, headers: response, stream: stream)
+
+      decoded = OnebusawaySDK::Util.decode_content(response, stream: stream)
+      case req
+      in { stream: Class => st }
+        st.new(model: model, url: url, status: status, response: response, messages: decoded)
+      in { page: Class => page }
+        page.new(client: self, req: req, headers: response, unwrapped: decoded)
+      else
+        unwrapped = OnebusawaySDK::Util.dig(decoded, req[:unwrap])
+        OnebusawaySDK::Converter.coerce(model, unwrapped)
+      end
     end
 
     # @return [String]

data/lib/onebusaway-sdk/errors.rb

@@ -99,11 +99,12 @@ module OnebusawaySDK
     # @param body [Object, nil]
     # @param request [nil]
     # @param response [nil]
+    # @param message [String, nil]
     #
     # @return [OnebusawaySDK::APIStatusError]
     #
-    def self.for(url:, status:, body:, request:, response:)
-      kwargs = {url: url, status: status, body: body, request: request, response: response}
+    def self.for(url:, status:, body:, request:, response:, message: nil)
+      kwargs = {url: url, status: status, body: body, request: request, response: response, message: message}
 
       case status
       in 400

data/lib/onebusaway-sdk/pooled_net_requester.rb

@@ -131,16 +131,23 @@ module OnebusawaySDK
       req = self.class.build_request(request)
 
       eof = false
+      finished = false
       enum = Enumerator.new do |y|
         with_pool(url) do |conn|
+          next if finished
+
           self.class.calibrate_socket_timeout(conn, deadline)
           conn.start unless conn.started?
 
           self.class.calibrate_socket_timeout(conn, deadline)
           conn.request(req) do |rsp|
             y << [conn, rsp]
+            break if finished
+
             rsp.read_body do |bytes|
               y << bytes
+              break if finished
+
               self.class.calibrate_socket_timeout(conn, deadline)
             end
             eof = true
@@ -148,16 +155,14 @@ module OnebusawaySDK
         end
       end
 
-      # need to protect the `Enumerator` against `#.rewind`
-      fused = false
       conn, response = enum.next
-      body = Enumerator.new do |y|
-        next if fused
-
-        fused = true
-        loop { y << enum.next }
-      ensure
-        conn.finish if !eof && conn.started?
+      body = OnebusawaySDK::Util.fused_enum(enum) do
+        finished = true
+        tap do
+          enum.next
+        rescue StopIteration
+        end
+        conn.finish if !eof && conn&.started?
       end
       [response, (response.body = body)]
     end

data/lib/onebusaway-sdk/util.rb

@@ -496,6 +496,9 @@ module OnebusawaySDK
       #
       def decode_content(headers, stream:, suppress_error: false)
         case headers["content-type"]
+        in %r{^text/event-stream}
+          lines = enum_lines(stream)
+          parse_sse(lines)
         in %r{^application/json}
           json = stream.to_a.join
           begin
@@ -512,6 +515,121 @@ module OnebusawaySDK
         end
       end
     end
+
+    class << self
+      # @private
+      #
+      # https://doc.rust-lang.org/std/iter/trait.FusedIterator.html
+      #
+      # @param enum [Enumerable]
+      # @param close [Proc]
+      #
+      # @return [Enumerable]
+      #
+      def fused_enum(enum, &close)
+        fused = false
+        iter = Enumerator.new do |y|
+          next if fused
+
+          fused = true
+          loop { y << enum.next }
+        ensure
+          close&.call
+          close = nil
+        end
+
+        iter.define_singleton_method(:rewind) do
+          fused = true
+          self
+        end
+        iter
+      end
+
+      # @private
+      #
+      # @param enum [Enumerable, nil]
+      #
+      def close_fused!(enum)
+        return unless enum.is_a?(Enumerator)
+
+        # rubocop:disable Lint/UnreachableLoop
+        enum.rewind.each { break }
+        # rubocop:enable Lint/UnreachableLoop
+      end
+
+      # @private
+      #
+      # @param enum [Enumerable, nil]
+      # @param blk [Proc]
+      #
+      def chain_fused(enum, &blk)
+        iter = Enumerator.new { blk.call(_1) }
+        fused_enum(iter) { close_fused!(enum) }
+      end
+    end
+
+    class << self
+      # @private
+      #
+      # @param enum [Enumerable]
+      #
+      # @return [Enumerable]
+      #
+      def enum_lines(enum)
+        chain_fused(enum) do |y|
+          buffer = String.new
+          enum.each do |row|
+            buffer << row
+            while (idx = buffer.index("\n"))
+              y << buffer.slice!(..idx)
+            end
+          end
+          y << buffer unless buffer.empty?
+        end
+      end
+
+      # @private
+      #
+      # https://html.spec.whatwg.org/multipage/server-sent-events.html#parsing-an-event-stream
+      #
+      # @param lines [Enumerable]
+      #
+      # @return [Hash{Symbol=>Object}]
+      #
+      def parse_sse(lines)
+        chain_fused(lines) do |y|
+          blank = {event: nil, data: nil, id: nil, retry: nil}
+          current = {}
+
+          lines.each do |line|
+            case line.strip
+            in ""
+              next if current.empty?
+              y << {**blank, **current}
+              current = {}
+            in /^:/
+              next
+            in /^([^:]+):\s?(.*)$/
+              _, field, value = Regexp.last_match.to_a
+              case field
+              in "event"
+                current.merge!(event: value)
+              in "data"
+                (current[:data] ||= String.new) << value << "\n"
+              in "id" unless value.include?("\0")
+                current.merge!(id: value)
+              in "retry" if /^\d+$/ =~ value
+                current.merge!(retry: Integer(value))
+              else
+              end
+            else
+            end
+          end
+
+          y << {**blank, **current} unless current.empty?
+        end
+      end
+    end
   end
 
   # rubocop:enable Metrics/ModuleLength

data/lib/onebusaway-sdk/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OnebusawaySDK
-  VERSION = "0.1.0-alpha.205"
+  VERSION = "0.1.0-alpha.206"
 end

data/rbi/lib/onebusaway-sdk/base_client.rbi

@@ -21,7 +21,8 @@ module OnebusawaySDK
         ),
         body: T.nilable(T.anything),
         unwrap: T.nilable(Symbol),
-        page: T.nilable(T::Class[OnebusawaySDK::BaseModel]),
+        page: T.nilable(T::Class[OnebusawaySDK::BasePage[OnebusawaySDK::BaseModel]]),
+        stream: T.nilable(T::Class[T.anything]),
         model: T.nilable(OnebusawaySDK::Converter::Input),
         options: T.nilable(T.any(OnebusawaySDK::RequestOptions, T::Hash[Symbol, T.anything]))
       }
@@ -122,22 +123,11 @@ module OnebusawaySDK
         retry_count: Integer,
         send_retry_header: T::Boolean
       )
-        .returns([Net::HTTPResponse, T::Enumerable[String]])
+        .returns([Integer, Net::HTTPResponse, T::Enumerable[String]])
     end
     private def send_request(request, redirect_count:, retry_count:, send_retry_header:)
     end
 
-    sig do
-      params(
-        req: OnebusawaySDK::BaseClient::RequestComponentsShape,
-        headers: T.any(T::Hash[String, String], Net::HTTPHeader),
-        stream: T::Enumerable[String]
-      )
-        .returns(T.anything)
-    end
-    private def parse_response(req, headers:, stream:)
-    end
-
     sig do
       params(
         method: Symbol,
@@ -155,7 +145,8 @@ module OnebusawaySDK
         ),
         body: T.nilable(T.anything),
         unwrap: T.nilable(Symbol),
-        page: T.nilable(T::Class[OnebusawaySDK::BaseModel]),
+        page: T.nilable(T::Class[OnebusawaySDK::BasePage[OnebusawaySDK::BaseModel]]),
+        stream: T.nilable(T::Class[T.anything]),
         model: T.nilable(OnebusawaySDK::Converter::Input),
         options: T.nilable(T.any(OnebusawaySDK::RequestOptions, T::Hash[Symbol, T.anything]))
       )
@@ -169,6 +160,7 @@ module OnebusawaySDK
       body: nil,
       unwrap: nil,
       page: nil,
+      stream: nil,
       model: OnebusawaySDK::Unknown,
       options: {}
     )

data/rbi/lib/onebusaway-sdk/errors.rbi

@@ -85,11 +85,12 @@ module OnebusawaySDK
         status: Integer,
         body: T.nilable(Object),
         request: NilClass,
-        response: NilClass
+        response: NilClass,
+        message: T.nilable(String)
       )
         .returns(T.attached_class)
     end
-    def self.for(url:, status:, body:, request:, response:)
+    def self.for(url:, status:, body:, request:, response:, message: nil)
     end
 
     sig { returns(Integer) }

data/rbi/lib/onebusaway-sdk/util.rbi

@@ -135,5 +135,34 @@ module OnebusawaySDK
     end
     def self.decode_content(headers, stream:, suppress_error: false)
     end
+
+    sig { params(enum: T::Enumerable[T.anything], close: T.proc.void).returns(T::Enumerable[T.anything]) }
+    def self.fused_enum(enum, &close)
+    end
+
+    sig { params(enum: T.nilable(T::Enumerable[T.anything])).void }
+    def self.close_fused!(enum)
+    end
+
+    sig do
+      params(
+        enum: T.nilable(T::Enumerable[T.anything]),
+        blk: T.proc.params(arg0: Enumerator::Yielder).void
+      ).void
+    end
+    def self.chain_fused(enum, &blk)
+    end
+
+    SSEMessage = T.type_alias do
+      {event: T.nilable(String), data: T.nilable(String), id: T.nilable(String), retry: T.nilable(Integer)}
+    end
+
+    sig { params(enum: T::Enumerable[String]).returns(T::Enumerable[String]) }
+    def self.enum_lines(enum)
+    end
+
+    sig { params(lines: T::Enumerable[String]).returns(OnebusawaySDK::Util::SSEMessage) }
+    def self.parse_sse(lines)
+    end
   end
 end

data/rbi/lib/onebusaway-sdk/version.rbi

@@ -1,5 +1,5 @@
 # typed: strong
 
 module OnebusawaySDK
-  VERSION = "0.1.0-alpha.205"
+  VERSION = "0.1.0-alpha.206"
 end

data/sig/onebusaway-sdk/base_client.rbs

@@ -11,6 +11,7 @@ module OnebusawaySDK
         body: top?,
         unwrap: Symbol?,
         page: Class?,
+        stream: Class?,
         model: OnebusawaySDK::Converter::input?,
         options: OnebusawaySDK::request_opts?
       }
@@ -78,13 +79,7 @@ module OnebusawaySDK
       redirect_count: Integer,
       retry_count: Integer,
       send_retry_header: bool
-    ) -> [top, Enumerable[String]]
-
-    private def parse_response: (
-      OnebusawaySDK::BaseClient::request_components req,
-      headers: ::Hash[String, String],
-      stream: Enumerable[String]
-    ) -> top
+    ) -> [Integer, top, Enumerable[String]]
 
     def request:
       (
@@ -97,6 +92,7 @@ module OnebusawaySDK
         body: top?,
         unwrap: Symbol?,
         page: Class?,
+        stream: Class?,
         model: OnebusawaySDK::Converter::input?,
         options: OnebusawaySDK::request_opts?
       ) -> top

data/sig/onebusaway-sdk/errors.rbs

@@ -51,7 +51,8 @@ module OnebusawaySDK
       status: Integer,
       body: Object?,
       request: nil,
-      response: nil
+      response: nil,
+      message: String?
     ) -> instance
 
     def initialize: (

data/sig/onebusaway-sdk/util.rbs

@@ -88,5 +88,24 @@ module OnebusawaySDK
       stream: Enumerable[String],
       suppress_error: bool
     ) -> top
+
+    def self?.fused_enum: (Enumerable[top] enum) { -> void } -> Enumerable[top]
+
+    def self?.close_fused!: (Enumerable[top]? enum) -> void
+
+    def self?.chain_fused: (
+      Enumerable[top]? enum
+    ) {
+      (Enumerator::Yielder arg0) -> void
+    } -> void
+
+    type sse_message =
+      { event: String?, data: String?, id: String?, retry: Integer? }
+
+    def self?.enum_lines: (Enumerable[String] enum) -> Enumerable[String]
+
+    def self?.parse_sse: (
+      Enumerable[String] lines
+    ) -> OnebusawaySDK::Util::sse_message
   end
 end

data/sig/onebusaway-sdk/version.rbs

@@ -1,3 +1,3 @@
 module OnebusawaySDK
-  VERSION: "0.1.0-alpha.204"
+  VERSION: "0.1.0-alpha.205"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: onebusaway-sdk
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre.alpha.205
+  version: 0.1.0.pre.alpha.206
 platform: ruby
 authors:
 - Onebusaway SDK
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-03-05 00:00:00.000000000 Z
+date: 2025-03-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: connection_pool