49 lines
1.1 KiB
Ruby
49 lines
1.1 KiB
Ruby
require 'rack/utils'
|
|
|
|
module Rack
|
|
|
|
# Middleware that applies chunked transfer encoding to response bodies
|
|
# when the response does not include a Content-Length header.
|
|
class Chunked
|
|
include Rack::Utils
|
|
|
|
def initialize(app)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
status, headers, body = @app.call(env)
|
|
headers = HeaderHash.new(headers)
|
|
|
|
if env['HTTP_VERSION'] == 'HTTP/1.0' ||
|
|
STATUS_WITH_NO_ENTITY_BODY.include?(status) ||
|
|
headers['Content-Length'] ||
|
|
headers['Transfer-Encoding']
|
|
[status, headers, body]
|
|
else
|
|
dup.chunk(status, headers, body)
|
|
end
|
|
end
|
|
|
|
def chunk(status, headers, body)
|
|
@body = body
|
|
headers.delete('Content-Length')
|
|
headers['Transfer-Encoding'] = 'chunked'
|
|
[status, headers, self]
|
|
end
|
|
|
|
def each
|
|
term = "\r\n"
|
|
@body.each do |chunk|
|
|
size = bytesize(chunk)
|
|
next if size == 0
|
|
yield [size.to_s(16), term, chunk, term].join
|
|
end
|
|
yield ["0", term, "", term].join
|
|
end
|
|
|
|
def close
|
|
@body.close if @body.respond_to?(:close)
|
|
end
|
|
end
|
|
end
|