lib/rainbows/http_response.rb

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