plum 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4194e9fa59a660fc839ea356770a05a627dba684
-  data.tar.gz: 5b4f22bd734f7d7df5dca01667716ecc3dc4e51e
+  metadata.gz: 7edaa74a52403fc469e096512924e275a1e0b2dc
+  data.tar.gz: b0e40798462d83baff094cbb4664401afe711694
 SHA512:
-  metadata.gz: 987380963b7702bd5a6aa0f4ab817ae515f34cbd1b74ba48ea2b46f5e27169173c80821af49f353bc61363a634107290bc4a43abe599a21707cbaadbc6446f0f
-  data.tar.gz: bd645808edbcf763c896e00085ab3d69d89639f44b1de09c3ac39f7ed2c79d6c5f313f2791dad57c43d31d68eeb9b59e745ba3e4040835e5670543f9284c905e
+  metadata.gz: 1fda3ee328f9de55f22d575cbdde788cd388fb1df15ead696cc4f86f8ce02532c458b50dc24866d3e9032345b752dccedbb6d58c5b1bfe70606c85adeeb529ff
+  data.tar.gz: fa4810540a9d8f81a4611cd91fb24d5de3d38543b2c366474388959e6006f99bce46846abe4834b8566a6e6470d4ec2ab3215c1ee3020401d76b79af6665b819

data/lib/plum.rb

@@ -2,6 +2,7 @@ require "openssl"
 require "socket"
 require "base64"
 require "set"
+require "zlib"
 require "plum/version"
 require "plum/errors"
 require "plum/binary_string"
@@ -24,6 +25,7 @@ require "plum/server/https_connection"
 require "plum/server/http_connection"
 require "plum/client"
 require "plum/client/response"
+require "plum/client/decoders"
 require "plum/client/connection"
 require "plum/client/client_session"
 require "plum/client/legacy_client_session"

data/lib/plum/client.rb

@@ -9,6 +9,7 @@ module Plum
       ssl_context: nil,
       http2_settings: {},
       user_agent: "plum/#{Plum::VERSION}",
+      auto_decode: true,
     }.freeze
 
     attr_reader :host, :port, :config
@@ -79,7 +80,7 @@ module Plum
     # @param block [Proc] if passed, it will be called when received response headers.
     def request(headers, body, options = {}, &block)
       raise ArgumentError, ":method and :path headers are required" unless headers[":method"] && headers[":path"]
-      @session.request(headers, body, options, &block)
+      @session.request(headers, body, @config.merge(options), &block)
     end
 
     # @!method get!

data/lib/plum/client/client_session.rb

@@ -42,7 +42,7 @@ module Plum
                   ":scheme" => @config[:scheme]
       }.merge(headers)
 
-      response = Response.new
+      response = Response.new(**options)
       @responses << response
       stream = @plum.open_stream
       stream.send_headers(headers, end_stream: !body)

data/lib/plum/client/decoders.rb

@@ -0,0 +1,51 @@
+module Plum
+  module Decoders
+    class Base
+      def decode(chunk)
+        chunk
+      end
+
+      def finish
+      end
+    end
+
+    # `deflate` is not just deflate, wrapped by zlib format (RFC 1950)
+    class Deflate < Base
+      def initialize
+        @inflate = Zlib::Inflate.new(Zlib::MAX_WBITS)
+      end
+
+      def decode(chunk)
+        @inflate.inflate(chunk)
+      rescue Zlib::Error => e
+        raise DecoderError.new("failed to decode chunk", e)
+      end
+
+      def finish
+        @inflate.finish
+      rescue Zlib::Error => e
+        raise DecoderError.new("failed to finalize", e)
+      end
+    end
+
+    class GZip < Base
+      def initialize
+        @stream = Zlib::Inflate.new(Zlib::MAX_WBITS + 16)
+      end
+
+      def decode(chunk)
+        @stream.inflate(chunk)
+      rescue Zlib::Error => e
+        raise DecoderError.new("failed to decode chunk", e)
+      end
+
+      def finish
+        @stream.finish
+      rescue Zlib::Error => e
+        raise DecoderError.new("failed to finalize", e)
+      end
+    end
+
+    DECODERS = { "gzip" => GZip, "deflate" => Deflate }.freeze
+  end
+end

data/lib/plum/client/legacy_client_session.rb

@@ -40,7 +40,7 @@ module Plum
         end
       end
 
-      response = Response.new
+      response = Response.new(**options)
       @requests << [response, headers, body, chunked, headers_cb]
       consume_queue
       response

data/lib/plum/client/response.rb

@@ -6,11 +6,12 @@ module Plum
     attr_reader :headers
 
     # @api private
-    def initialize
+    def initialize(auto_decode: true, **options)
       @body = Queue.new
       @finished = false
       @failed = false
       @body = []
+      @auto_decode = auto_decode
     end
 
     # Returns the HTTP status code.
@@ -54,7 +55,7 @@ module Plum
     def on_finish(&block)
       raise ArgumentError, "block must be given" unless block_given?
       if finished?
-        block.call
+        yield
       else
         @on_finish = block
       end
@@ -64,21 +65,20 @@ module Plum
     # @return [String] the whole response body
     def body
       raise "Body already read" if @on_chunk
-      if finished?
-        @body.join
-      else
-        raise "Response body is not complete"
-      end
+      raise "Response body is not complete" unless finished?
+      @body.join
     end
 
     # @api private
     def _headers(raw_headers)
       # response headers should not have duplicates
       @headers = raw_headers.to_h.freeze
+      @decoder = setup_decoder
     end
 
     # @api private
-    def _chunk(chunk)
+    def _chunk(encoded)
+      chunk = @decoder.decode(encoded)
       if @on_chunk
         @on_chunk.call(chunk)
       else
@@ -89,6 +89,7 @@ module Plum
     # @api private
     def _finish
       @finished = true
+      @decoder.finish
       @on_finish.call if @on_finish
     end
 
@@ -96,5 +97,14 @@ module Plum
     def _fail
       @failed = true
     end
+
+    private
+    def setup_decoder
+      if @auto_decode
+        klass = Decoders::DECODERS[@headers["content-encoding"]]
+      end
+      klass ||= Decoders::Base
+      klass.new
+    end
   end
 end

data/lib/plum/errors.rb

@@ -31,6 +31,14 @@ module Plum
       ERROR_CODES[@http2_error_type]
     end
   end
+
+  class RemoteHTTPError < HTTPError; end
+  class RemoteConnectionError < RemoteHTTPError; end
+  class RemoteStreamError < RemoteHTTPError; end
+  class LocalHTTPError < HTTPError; end
+  class LocalConnectionError < LocalHTTPError; end
+  class LocalStreamError < LocalHTTPError; end
+
   class LegacyHTTPError < Error
     attr_reader :headers, :data, :parser
 
@@ -41,10 +49,12 @@ module Plum
     end
   end
 
-  class RemoteHTTPError < HTTPError; end
-  class RemoteConnectionError < RemoteHTTPError; end
-  class RemoteStreamError < RemoteHTTPError; end
-  class LocalHTTPError < HTTPError; end
-  class LocalConnectionError < LocalHTTPError; end
-  class LocalStreamError < LocalHTTPError; end
+  class DecoderError < Error
+    attr_reader :inner_error
+
+    def initialize(message, inner_error = nil)
+      super(message)
+      @inner_error = inner_error
+    end
+  end
 end

data/lib/plum/version.rb

@@ -1,4 +1,4 @@
 # -*- frozen-string-literal: true -*-
 module Plum
-  VERSION = "0.2.0"
+  VERSION = "0.2.1"
 end

data/test/plum/client/test_decoders.rb

@@ -0,0 +1,54 @@
+require "test_helper"
+
+using Plum::BinaryString
+class DecodersTest < Minitest::Test
+  def test_base_decode
+    decoder = Decoders::Base.new
+    assert_equal("abc", decoder.decode("abc"))
+  end
+  
+  def test_base_finish
+    decoder = Decoders::Base.new
+    decoder.finish
+  end
+
+  def test_deflate_decode
+    decoder = Decoders::Deflate.new
+    assert_equal("hello", decoder.decode("\x78\x9c\xcb\x48\xcd\xc9\xc9\x07\x00\x06\x2c\x02\x15"))
+  end
+
+  def test_deflate_decode_error
+    decoder = Decoders::Deflate.new
+    assert_raises(DecoderError) {
+      decoder.decode("\x79\x9c\xcb\x48\xcd\xc9\xc9\x07\x00\x06\x2c\x02\x15")
+    }
+  end
+
+  def test_deflate_finish_error
+    decoder = Decoders::Deflate.new
+    decoder.decode("\x78\x9c\xcb\x48\xcd\xc9\xc9\x07\x00\x06\x2c\x02")
+    assert_raises(DecoderError) {
+      decoder.finish
+    }
+  end
+
+  def test_gzip_decode
+    decoder = Decoders::GZip.new
+    assert_equal("hello", decoder.decode("\x1f\x8b\x08\x00\x1a\x96\xe0\x4c\x00\x03\xcb\x48\xcd\xc9\xc9\x07\x00\x86\xa6\x10\x36\x05\x00\x00\x00"))
+  end
+
+  def test_gzip_decode_error
+    decoder = Decoders::GZip.new
+    assert_raises(DecoderError) {
+      decoder.decode("\x2f\x8b\x08\x00\x1a\x96\xe0\x4c\x00\x03\xcb\x48\xcd\xc9\xc9\x07\x00\x86\xa6\x10\x36\x05\x00\x00\x00")
+    }
+  end
+
+  def test_gzip_finish_error
+    decoder = Decoders::GZip.new
+    decoder.decode("\x1f\x8b\x08\x00\x1a\x96")
+    assert_raises(DecoderError) {
+      decoder.finish
+    }
+  end
+end

data/test/plum/client/test_response.rb

@@ -4,6 +4,7 @@ using Plum::BinaryString
 class ResponseTest < Minitest::Test
   def test_finished
     resp = Response.new
+    resp._headers({})
     assert_equal(false, resp.finished?)
     resp._finish
     assert_equal(true, resp.finished?)
@@ -34,6 +35,7 @@ class ResponseTest < Minitest::Test
 
   def test_body
     resp = Response.new
+    resp._headers({})
     resp._chunk("a")
     resp._chunk("b")
     resp._finish
@@ -42,6 +44,7 @@ class ResponseTest < Minitest::Test
 
   def test_body_not_finished
     resp = Response.new
+    resp._headers({})
     resp._chunk("a")
     resp._chunk("b")
     assert_raises { # TODO
@@ -51,6 +54,7 @@ class ResponseTest < Minitest::Test
 
   def test_on_chunk
     resp = Response.new
+    resp._headers({})
     res = []
     resp._chunk("a")
     resp._chunk("b")
@@ -63,6 +67,7 @@ class ResponseTest < Minitest::Test
 
   def test_on_finish
     resp = Response.new
+    resp._headers({})
     ran = false
     resp.on_finish { ran = true }
     resp._finish

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: plum
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - rhenium
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-11-09 00:00:00.000000000 Z
+date: 2015-11-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -178,6 +178,7 @@ files:
 - lib/plum/client.rb
 - lib/plum/client/client_session.rb
 - lib/plum/client/connection.rb
+- lib/plum/client/decoders.rb
 - lib/plum/client/legacy_client_session.rb
 - lib/plum/client/response.rb
 - lib/plum/client/upgrade_client_session.rb
@@ -211,6 +212,7 @@ files:
 - plum.gemspec
 - test/plum/client/test_client.rb
 - test/plum/client/test_connection.rb
+- test/plum/client/test_decoders.rb
 - test/plum/client/test_legacy_client_session.rb
 - test/plum/client/test_response.rb
 - test/plum/client/test_upgrade_client_session.rb
@@ -268,6 +270,7 @@ summary: An HTTP/2 Library for Ruby
 test_files:
 - test/plum/client/test_client.rb
 - test/plum/client/test_connection.rb
+- test/plum/client/test_decoders.rb
 - test/plum/client/test_legacy_client_session.rb
 - test/plum/client/test_response.rb
 - test/plum/client/test_upgrade_client_session.rb