Merge pull request #49 from miyucy/respondtoread

Made Brotli.inflate accept objects with a #read method, allowing for more flexible input, such as IO-like objects.

ext/brotli/brotli.c

@@ -68,12 +68,18 @@ brotli_inflate_no_gvl(void *arg)
     return arg;
 }
 
+static ID id_read;
+
 static VALUE
 brotli_inflate(VALUE self, VALUE str)
 {
     VALUE value = Qnil;
     brotli_inflate_args_t args;
 
+    if (rb_respond_to(str, id_read)) {
+      str = rb_funcall(str, id_read, 0, 0);
+    }
+
     StringValue(str);
 
     args.str = (uint8_t*)RSTRING_PTR(str);
@@ -472,6 +478,7 @@ Init_brotli(void)
     rb_define_singleton_method(rb_mBrotli, "deflate", RUBY_METHOD_FUNC(brotli_deflate), -1);
     rb_define_singleton_method(rb_mBrotli, "inflate", RUBY_METHOD_FUNC(brotli_inflate), 1);
     rb_define_singleton_method(rb_mBrotli, "version", RUBY_METHOD_FUNC(brotli_version), 0);
+    id_read = rb_intern("read");
     // Brotli::Writer
     id_write = rb_intern("write");
     id_flush = rb_intern("flush");

test/brotli_test.rb

@@ -77,6 +77,16 @@ def testdata2
       assert_equal testdata, Brotli.inflate(testdata2)
     end
 
+    test "works with StringIO" do
+      testdata3 = StringIO.new testdata2
+      assert_equal testdata, Brotli.inflate(testdata3)
+    end
+
+    test "works with File" do
+      f = File.open(File.expand_path(File.join("..", "vendor", "brotli", "tests", "testdata", "alice29.txt.compressed"), __dir__))
+      assert_equal testdata, Brotli.inflate(f)
+    end
+
     test "raise error when pass insufficient data" do
       assert_raise Brotli::Error do
         Brotli.inflate(testdata2[0, 64])

test/test_helper.rb

@@ -6,3 +6,4 @@
 require "test-unit"
 require "test/unit/rr"
 require "rantly/testunit_extensions"
+require "stringio"