1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
| | # -*- encoding: binary -*-
require 'time'
require 'rainbows'
module Rainbows
class HttpResponse < ::Unicorn::HttpResponse
def self.write(socket, rack_response, out = [])
status, headers, body = rack_response
if Array === out
status = CODES[status.to_i] || status
headers.each do |key, value|
next if %r{\AX-Rainbows-}i =~ key
next if SKIP.include?(key.downcase)
if value =~ /\n/
out.concat(value.split(/\n/).map! { |v| "#{key}: #{v}\r\n" })
else
out << "#{key}: #{value}\r\n"
end
end
socket.write("HTTP/1.1 #{status}\r\n" \
"Date: #{Time.now.httpdate}\r\n" \
"Status: #{status}\r\n" \
"#{out.join('')}\r\n")
end
body.each { |chunk| socket.write(chunk) }
ensure
body.respond_to?(:close) and body.close rescue nil
end
end
end
|