Sha256: 1e7b5fa779e65a440d30688b3c67f96f9e5b60c91d5bc56e1f0642e826ab11e1

Contents?: true

Size: 1.43 KB

Versions: 30

Compression:

Stored size: 1.43 KB

Contents

require "zlib"
require "stringio"

module Rack

class Deflater
  def initialize(app)
    @app = app
  end

  def call(env)
    status, headers, body = @app.call(env)

    request = Request.new(env)

    encoding = Utils.select_best_encoding(%w(gzip deflate identity), request.accept_encoding)

    case encoding
    when "gzip"
      mtime = headers["Last-Modified"] || Time.now
      [status, headers.merge("Content-Encoding" => "gzip"), self.class.gzip(body, mtime)]
    when "deflate"
      [status, headers.merge("Content-Encoding" => "deflate"), self.class.deflate(body)]
    when "identity"
      [status, headers, body]
    when nil
      message = "An acceptable encoding for the requested resource #{request.fullpath} could not be found."
      [406, {"Content-Type" => "text/plain"}, message]
    end
  end

  def self.gzip(body, mtime)
    io = StringIO.new
    gzip = Zlib::GzipWriter.new(io)
    gzip.mtime = mtime

    # TODO: Add streaming
    body.each { |part| gzip << part }

    gzip.close
    return io.string
  end

  DEFLATE_ARGS = [
    Zlib::DEFAULT_COMPRESSION,
    # drop the zlib header which causes both Safari and IE to choke
   -Zlib::MAX_WBITS,
    Zlib::DEF_MEM_LEVEL,
    Zlib::DEFAULT_STRATEGY
  ]

  # Loosely based on Mongrel's Deflate handler
  def self.deflate(body)
    deflater = Zlib::Deflate.new(*DEFLATE_ARGS)

    # TODO: Add streaming
    body.each { |part| deflater << part }

    return deflater.finish
  end
end

end

Version data entries

30 entries across 30 versions & 7 rubygems

Version Path
3mix-castronaut-0.5.0.2 vendor/rack/lib/rack/deflater.rb
kastner-rack-0.3.171 lib/rack/deflater.rb
kastner-rack-0.3.186 lib/rack/deflater.rb
masover-castronaut-0.4.4.4 vendor/rack/lib/rack/deflater.rb
masover-castronaut-0.4.4.5 vendor/rack/lib/rack/deflater.rb
masover-castronaut-0.5.0.1 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.2.0 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.2.5 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.3.0 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.3.5 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.3.6 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.4.1 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.4.2 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.4.3 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.4.4 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.4.5 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.4.6 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.5.0 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.5.1 vendor/rack/lib/rack/deflater.rb
relevance-castronaut-0.5.2 vendor/rack/lib/rack/deflater.rb