httpx 0.9.0 → 0.10.0

data/lib/httpx/errors.rb

@@ -28,6 +28,7 @@ module HTTPX
 
   NativeResolveError = Class.new(ResolveError) do
     attr_reader :connection, :host
+
     def initialize(connection, host, message = "Can't resolve #{host}")
       @connection = connection
       @host = host

data/lib/httpx/extensions.rb

@@ -54,11 +54,31 @@ module HTTPX
     Numeric.__send__(:include, NegMethods)
   end
 
+  module RegexpExtensions
+    # If you wonder why this is there: the oauth feature uses a refinement to enhance the
+    # Regexp class locally with #match? , but this is never tested, because ActiveSupport
+    # monkey-patches the same method... Please ActiveSupport, stop being so intrusive!
+    # :nocov:
+    refine(Regexp) do
+      def match?(*args)
+        !match(*args).nil?
+      end
+    end
+  end
+
   module URIExtensions
     refine URI::Generic do
+      def non_ascii_hostname
+        @non_ascii_hostname
+      end
+
+      def non_ascii_hostname=(hostname)
+        @non_ascii_hostname = hostname
+      end
+
       def authority
         port_string = port == default_port ? nil : ":#{port}"
-        "#{host}#{port_string}"
+        "#{@non_ascii_hostname || host}#{port_string}"
       end
 
       def origin

data/lib/httpx/io/ssl.rb

@@ -42,7 +42,6 @@ module HTTPX
       # allow reconnections
       # connect only works if initial @io is a socket
       @io = @io.io if @io.respond_to?(:io)
-      @negotiated = false
     end
 
     def connected?

data/lib/httpx/io/tcp.rb

@@ -7,11 +7,7 @@ module HTTPX
   class TCP
     include Loggable
 
-    attr_reader :ip, :port
-
-    attr_reader :addresses
-
-    attr_reader :state
+    attr_reader :ip, :port, :addresses, :state
 
     alias_method :host, :ip
 
@@ -86,6 +82,7 @@ module HTTPX
       # :nocov:
       def read(size, buffer)
         @io.read_nonblock(size, buffer)
+        log { "READ: #{buffer.bytesize} bytes..." }
         buffer.bytesize
       rescue ::IO::WaitReadable
         buffer.clear
@@ -96,6 +93,7 @@ module HTTPX
 
       def write(buffer)
         siz = @io.write_nonblock(buffer)
+        log { "WRITE: #{siz} bytes..." }
         buffer.shift!(siz)
         siz
       rescue ::IO::WaitWritable
@@ -113,6 +111,7 @@ module HTTPX
         end
         return if ret.nil?
 
+        log { "READ: #{buffer.bytesize} bytes..." }
         buffer.bytesize
       end
 
@@ -121,6 +120,8 @@ module HTTPX
         return 0 if siz == :wait_writable
         return if siz.nil?
 
+        log { "WRITE: #{siz} bytes..." }
+
         buffer.shift!(siz)
         siz
       end

data/lib/httpx/io/udp.rb

@@ -7,11 +7,12 @@ module HTTPX
   class UDP
     include Loggable
 
-    def initialize(uri, _, _)
+    def initialize(uri, _, options)
       ip = IPAddr.new(uri.host)
       @host = ip.to_s
       @port = uri.port
       @io = UDPSocket.new(ip.family)
+      @options = options
     end
 
     def to_io
@@ -40,6 +41,7 @@ module HTTPX
 
     def write(buffer)
       siz = @io.send(buffer, 0, @host, @port)
+      log { "WRITE: #{siz} bytes..." }
       buffer.shift!(siz)
       siz
     end
@@ -49,6 +51,7 @@ module HTTPX
       def read(size, buffer)
         data, _ = @io.recvfrom_nonblock(size)
         buffer.replace(data)
+        log { "READ: #{buffer.bytesize} bytes..." }
         buffer.bytesize
       rescue ::IO::WaitReadable
         0

data/lib/httpx/options.rb

@@ -76,6 +76,8 @@ module HTTPX
 
       defaults.merge!(options)
       defaults.each do |(k, v)|
+        next if v.nil?
+
         __send__(:"#{k}=", v)
       end
     end

data/lib/httpx/parser/http1.rb

@@ -9,10 +9,9 @@ module HTTPX
 
       attr_reader :status_code, :http_version, :headers
 
-      def initialize(observer, header_separator: ":")
+      def initialize(observer)
         @observer = observer
         @state = :idle
-        @header_separator = header_separator
         @buffer = "".b
         @headers = {}
       end
@@ -40,25 +39,25 @@ module HTTPX
       private
 
       def parse
-        state = @state
-        case @state
-        when :idle
-          parse_headline
-        when :headers
-          parse_headers
-        when :trailers
-          parse_headers
-        when :data
-          parse_data
+        loop do
+          state = @state
+          case @state
+          when :idle
+            parse_headline
+          when :headers, :trailers
+            parse_headers
+          when :data
+            parse_data
+          end
+          return if @buffer.empty? || state == @state
         end
-        parse if !@buffer.empty? && state != @state
       end
 
       def parse_headline
         idx = @buffer.index("\n")
         return unless idx
 
-        (m = %r{\AHTTP(?:\/(\d+\.\d+))?\s+(\d\d\d)(?:\s+(.*))?}in.match(@buffer)) ||
+        (m = %r{\AHTTP(?:/(\d+\.\d+))?\s+(\d\d\d)(?:\s+(.*))?}in.match(@buffer)) ||
           raise(Error, "wrong head line format")
         version, code, _ = m.captures
         raise(Error, "unsupported HTTP version (HTTP/#{version})") unless VERSIONS.include?(version)
@@ -91,12 +90,10 @@ module HTTPX
               @observer.on_trailers(headers)
               headers.clear
               nextstate(:complete)
-            else
-              raise Error, "wrong header format"
             end
             return
           end
-          separator_index = line.index(@header_separator)
+          separator_index = line.index(":")
           raise Error, "wrong header format" unless separator_index
 
           key = line[0..separator_index - 1]

data/lib/httpx/plugins/compression.rb

@@ -46,16 +46,13 @@ module HTTPX
           super
           return if @body.nil?
 
-          if (threshold = options.compression_threshold_size)
-            unless unbounded_body?
-              return if @body.bytesize < threshold
-            end
-          end
+          threshold = options.compression_threshold_size
+          return if threshold && !unbounded_body? && @body.bytesize < threshold
 
           @headers.get("content-encoding").each do |encoding|
             next if encoding == "identity"
 
-            @body = Encoder.new(@body, Compression.registry(encoding).encoder)
+            @body = Encoder.new(@body, Compression.registry(encoding).deflater)
           end
           @headers["content-length"] = @body.bytesize unless chunked?
         end
@@ -71,57 +68,53 @@ module HTTPX
 
           return unless @headers.key?("content-encoding")
 
-          @_decoders = @headers.get("content-encoding").map do |encoding|
+          # remove encodings that we are able to decode
+          @headers["content-encoding"] = @headers.get("content-encoding") - @encodings
+
+          compressed_length = if @headers.key?("content-length")
+            @headers["content-length"].to_i
+          else
+            Float::INFINITY
+          end
+
+          @_inflaters = @headers.get("content-encoding").map do |encoding|
             next if encoding == "identity"
 
-            decoder = Compression.registry(encoding).decoder
+            inflater = Compression.registry(encoding).inflater(compressed_length)
             # do not uncompress if there is no decoder available. In fact, we can't reliably
             # continue decompressing beyond that, so ignore.
-            break unless decoder
+            break unless inflater
 
             @encodings << encoding
-            decoder
+            inflater
           end.compact
 
-          # remove encodings that we are able to decode
-          @headers["content-encoding"] = @headers.get("content-encoding") - @encodings
-
-          @_compressed_length = if @headers.key?("content-length")
-            @headers["content-length"].to_i
-          else
-            Float::INFINITY
-          end
+          # this can happen if the only declared encoding is "identity"
+          remove_instance_variable(:@_inflaters) if @_inflaters.empty?
         end
 
         def write(chunk)
-          return super unless defined?(@_compressed_length)
+          return super unless defined?(@_inflaters)
 
-          @_compressed_length -= chunk.bytesize
           chunk = decompress(chunk)
           super(chunk)
         end
 
-        def close
-          super
-
-          return unless defined?(@_decoders)
-
-          @_decoders.each(&:close)
-        end
-
         private
 
         def decompress(buffer)
-          @_decoders.reverse_each do |decoder|
-            buffer = decoder.decode(buffer)
-            buffer << decoder.finish if @_compressed_length <= 0
+          @_inflaters.reverse_each do |inflater|
+            buffer = inflater.inflate(buffer)
           end
           buffer
         end
       end
 
       class Encoder
+        attr_reader :content_type
+
         def initialize(body, deflater)
+          @content_type = body.content_type
           @body = body.respond_to?(:read) ? body : StringIO.new(body.to_s)
           @buffer = StringIO.new("".b, File::RDWR)
           @deflater = deflater
@@ -130,11 +123,10 @@ module HTTPX
         def each(&blk)
           return enum_for(__method__) unless block_given?
 
-          unless @buffer.size.zero?
-            @buffer.rewind
-            return @buffer.each(&blk)
-          end
-          deflate(&blk)
+          return deflate(&blk) if @buffer.size.zero?
+
+          @buffer.rewind
+          @buffer.each(&blk)
         end
 
         def bytesize
@@ -142,17 +134,6 @@ module HTTPX
           @buffer.size
         end
 
-        def to_s
-          deflate
-          @buffer.rewind
-          @buffer.read
-        end
-
-        def close
-          @buffer.close
-          @body.close
-        end
-
         private
 
         def deflate(&blk)
@@ -162,22 +143,6 @@ module HTTPX
           @deflater.deflate(@body, @buffer, chunk_size: 16_384, &blk)
         end
       end
-
-      class Decoder
-        extend Forwardable
-
-        def_delegator :@inflater, :finish
-
-        def_delegator :@inflater, :close
-
-        def initialize(inflater)
-          @inflater = inflater
-        end
-
-        def decode(chunk)
-          @inflater.inflate(chunk)
-        end
-      end
     end
     register_plugin :compression, Compression
   end

data/lib/httpx/plugins/compression/brotli.rb

@@ -13,7 +13,7 @@ module HTTPX
           Compression.register "br", self
         end
 
-        module Encoder
+        module Deflater
           module_function
 
           def deflate(raw, buffer, chunk_size:)
@@ -25,28 +25,24 @@ module HTTPX
           end
         end
 
-        module BrotliWrapper
-          module_function
-
-          def inflate(text)
-            ::Brotli.inflate(text)
+        class Inflater
+          def initialize(bytesize)
+            @bytesize = bytesize
           end
 
-          def close; end
-
-          def finish
-            ""
+          def inflate(chunk)
+            ::Brotli.inflate(chunk)
           end
         end
 
         module_function
 
-        def encoder
-          Encoder
+        def deflater
+          Deflater
         end
 
-        def decoder
-          Decoder.new(BrotliWrapper)
+        def inflater(bytesize)
+          Inflater.new(bytesize)
         end
       end
     end

data/lib/httpx/plugins/compression/deflate.rb

@@ -4,16 +4,17 @@ module HTTPX
   module Plugins
     module Compression
       module Deflate
-        def self.load_dependencies(*)
+        def self.load_dependencies(klass)
           require "stringio"
           require "zlib"
+          klass.plugin(:"compression/gzip")
         end
 
         def self.configure(*)
           Compression.register "deflate", self
         end
 
-        module Encoder
+        module Deflater
           module_function
 
           def deflate(raw, buffer, chunk_size:)
@@ -36,12 +37,12 @@ module HTTPX
 
         module_function
 
-        def encoder
-          Encoder
+        def deflater
+          Deflater
         end
 
-        def decoder
-          Decoder.new(Zlib::Inflate.new(32 + Zlib::MAX_WBITS))
+        def inflater(bytesize)
+          GZIP::Inflater.new(bytesize)
         end
       end
     end

data/lib/httpx/plugins/compression/gzip.rb

@@ -14,7 +14,7 @@ module HTTPX
           Compression.register "gzip", self
         end
 
-        class Encoder
+        class Deflater
           def initialize
             @compressed_chunk = "".b
           end
@@ -53,14 +53,32 @@ module HTTPX
           end
         end
 
+        class Inflater
+          def initialize(bytesize)
+            @inflater = Zlib::Inflate.new(32 + Zlib::MAX_WBITS)
+            @bytesize = bytesize
+            @buffer = nil
+          end
+
+          def inflate(chunk)
+            buffer = @inflater.inflate(chunk)
+            @bytesize -= chunk.bytesize
+            if @bytesize <= 0
+              buffer << @inflater.finish
+              @inflater.close
+            end
+            buffer
+          end
+        end
+
         module_function
 
-        def encoder
-          Encoder.new
+        def deflater
+          Deflater.new
         end
 
-        def decoder
-          Decoder.new(Zlib::Inflate.new(32 + Zlib::MAX_WBITS))
+        def inflater(bytesize)
+          Inflater.new(bytesize)
         end
       end
     end