Merge pull request #12 from michal-kazmierczak/master

Automatically decompress the body

lib/postgrest/responses/base_response.rb

@@ -32,7 +32,12 @@ def status_text
       end
 
       def data
-        error ? [] : safe_json_parse(response.body)
+        return [] if error
+
+        body = response.body
+        body = decompress(body) if compressed_body?
+
+        safe_json_parse(body)
       end
       alias as_json data
 
@@ -50,6 +55,16 @@ def safe_json_parse(json)
       rescue TypeError, JSON::ParserError
         {}
       end
+
+      def decompress(body)
+        gz = Zlib::GzipReader.new(StringIO.new(body))
+        gz.read
+      end
+
+      def compressed_body?
+        # https://ruby-doc.org/3.2.1/stdlibs/net/Net/HTTP.html#class-Net::HTTP-label-Compression
+        response['content-encoding'] == 'gzip' && response['content-range']
+      end
     end
   end
 end

spec/postgrest/responses/base_response_spec.rb

@@ -11,6 +11,7 @@ module Postgrest
     before do
       allow(response).to receive(:is_a?).and_return(true)
       allow(response).to receive(:body).and_return(response_body)
+      allow(response).to receive(:[]).and_return({})
     end
 
     describe '#error' do
@@ -80,6 +81,26 @@ module Postgrest
 
         it { expect(subject.data).to eq({}) }
       end
+
+      context 'when response body was compressed' do
+        subject { described_class.new(request, response) }
+
+        let(:response_body) do
+          data = { success: true, foo: :bar }.to_json
+          gz = Zlib::GzipWriter.new(StringIO.new)
+          gz << data
+          gz.close.string
+        end
+
+        before do
+          allow(response).to receive(:[]).with('content-encoding').and_return('gzip')
+          allow(response).to receive(:[]).with('content-range').and_return('0-999/*')
+        end
+
+        it 'decompresses the body' do
+          expect(subject.data).to eq({ 'success' => true, 'foo' => 'bar' })
+        end
+      end
     end
 
     describe '#params' do
@@ -90,4 +111,4 @@ module Postgrest
       end
     end
   end
-end
+end