Class: FaradayMiddleware::Gzip

Inherits:
Faraday::Middleware
  • Object
show all
Defined in:
lib/middleware/gzip.rb

Overview

TODO: Remove this file and the related line on the client.rb

it was only added because faraday_middleware has not released
a gem version including this specific middleware.

Middleware to automatically decompress response bodies. If the “Accept-Encoding” header wasn’t set in the request, this sets it to “gzip,deflate” and appropriately handles the compressed response from the server. This resembles what Ruby 1.9+ does internally in Net::HTTP#get.

This middleware is NOT necessary when these adapters are used:

  • net_http on Ruby 1.9+

  • net_http_persistent on Ruby 2.0+

  • em_http

Constant Summary collapse

ACCEPT_ENCODING = 
'Accept-Encoding'.freeze
CONTENT_ENCODING = 
'Content-Encoding'.freeze
CONTENT_LENGTH = 
'Content-Length'.freeze
SUPPORTED_ENCODINGS = 
'gzip,deflate'.freeze
RUBY_ENCODING = 
'1.9'.respond_to?(:force_encoding)

Instance Method Summary collapse

Instance Method Details

#call(env) ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/middleware/gzip.rb', line 24

def call(env)
  env[:request_headers][ACCEPT_ENCODING] ||= SUPPORTED_ENCODINGS
  @app.call(env).on_complete do |response_env|
    case response_env[:response_headers][CONTENT_ENCODING]
    when 'gzip'
      reset_body(response_env, &method(:uncompress_gzip))
    when 'deflate'
      reset_body(response_env, &method(:inflate))
    end
  end
end

#inflate(body) ⇒ Object 



52
53
54
 
# File 'lib/middleware/gzip.rb', line 52

def inflate(body)
  Zlib::Inflate.inflate(body)
end

#reset_body(env) ⇒ Object 



36
37
38
39
40
 
# File 'lib/middleware/gzip.rb', line 36

def reset_body(env)
  env[:body] = yield(env[:body])
  env[:response_headers].delete(CONTENT_ENCODING)
  env[:response_headers][CONTENT_LENGTH] = env[:body].length
end

#uncompress_gzip(body) ⇒ Object 



42
43
44
45
46
47
48
49
50
 
# File 'lib/middleware/gzip.rb', line 42

def uncompress_gzip(body)
  io = StringIO.new(body)
  gzip_reader = if RUBY_ENCODING
    Zlib::GzipReader.new(io, :encoding => 'ASCII-8BIT')
  else
    Zlib::GzipReader.new(io)
  end
  gzip_reader.read
end