Skip to content

Commit

Permalink
Merge pull request sinatra#517 from jormon/no_length_head
Browse files Browse the repository at this point in the history
HEAD requests on static files should have proper Content-Length
  • Loading branch information
rkh committed May 23, 2012
2 parents 7614d3b + 00b63ec commit f948c23
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 3 deletions.
8 changes: 5 additions & 3 deletions lib/sinatra/base.rb
Expand Up @@ -77,7 +77,9 @@ def finish
headers.delete "Content-Length"
headers.delete "Content-Type"
elsif Array === body and not [204, 304].include?(status.to_i)
headers["Content-Length"] = body.inject(0) { |l, p| l + Rack::Utils.bytesize(p) }.to_s
# if some other code has already set Content-Length, don't muck with it
# currently, this would be the static file-handler
headers["Content-Length"] ||= body.inject(0) { |l, p| l + Rack::Utils.bytesize(p) }.to_s
end

# Rack::Response#finish sometimes returns self as response body. We don't want that.
Expand Down Expand Up @@ -986,7 +988,7 @@ def handle_exception!(boom)
def error_block!(key, *block_params)
base = settings
while base.respond_to?(:errors)
next base = base.superclass unless args_array = base.errors[key]
next base = base.superclass unless args_array = base.errors[key]
args_array.reverse_each do |args|
first = args == args_array.first
args += [block_params]
Expand Down Expand Up @@ -1092,7 +1094,7 @@ def disable(*opts)
# Define a custom error handler. Optionally takes either an Exception
# class, or an HTTP status code to specify which errors should be
# handled.
def error(*codes, &block)
def error(*codes, &block)
args = compile! "ERROR", //, block
codes = codes.map { |c| Array(c) }.flatten
codes << Exception if codes.empty?
Expand Down
1 change: 1 addition & 0 deletions test/static_test.rb
Expand Up @@ -37,6 +37,7 @@ class StaticTest < Test::Unit::TestCase
assert ok?
assert_equal '', body
assert response.headers.include?('Last-Modified')
assert_equal File.size(__FILE__).to_s, response['Content-Length']
end

%w[POST PUT DELETE].each do |verb|
Expand Down

0 comments on commit f948c23

Please sign in to comment.