Skip to content

Instantly share code, notes, and snippets.

@danielwestendorf
Forked from relistan/compress_requests.rb
Created March 28, 2017 23:28
Show Gist options
  • Save danielwestendorf/6a37ffb4720929659979b974ce45a541 to your computer and use it in GitHub Desktop.
Save danielwestendorf/6a37ffb4720929659979b974ce45a541 to your computer and use it in GitHub Desktop.
Rack Middleware to automatically unzip gzipped/deflated POST data
class CompressedRequests
def initialize(app)
@app = app
end
def method_handled?(env)
!!(env['REQUEST_METHOD'] =~ /(POST|PUT)/)
end
def encoding_handled?(env)
['gzip', 'deflate'].include? env['HTTP_CONTENT_ENCODING']
end
def call(env)
if method_handled?(env) && encoding_handled?(env)
extracted = decode(env['rack.input'], env['HTTP_CONTENT_ENCODING'])
env.delete('HTTP_CONTENT_ENCODING')
env['CONTENT_LENGTH'] = extracted.bytesize
env['rack.input'] = StringIO.new(extracted)
end
status, headers, response = @app.call(env)
return [status, headers, response]
end
def decode(input, content_encoding)
case content_encoding
when 'gzip' then Zlib::GzipReader.new(input).read
when 'deflate' then Zlib::Inflate.inflate(input.read)
end
end
end
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment