lib/pitchfork/http_response.rb in pitchfork-0.1.1 vs lib/pitchfork/http_response.rb in pitchfork-0.1.2

@@ -18,10 +18,22 @@
     def err_response(code, response_start_sent)
       "#{response_start_sent ? '' : 'HTTP/1.1 '}" \
         "#{code} #{STATUS_CODES[code]}\r\n\r\n"
     end
 
+    def append_header(buf, key, value)
+      case value
+      when Array # Rack 3
+        value.each { |v| buf << "#{key}: #{v}\r\n" }
+      when /\n/ # Rack 2
+        # avoiding blank, key-only cookies with /\n+/
+        value.split(/\n+/).each { |v| buf << "#{key}: #{v}\r\n" }
+      else
+        buf << "#{key}: #{value}\r\n"
+      end
+    end
+
     # writes the rack_response to socket as an HTTP response
     def http_response_write(socket, status, headers, body,
                             req = Pitchfork::HttpParser.new)
       hijack = nil
 
@@ -39,15 +51,10 @@
           when "rack.hijack"
             # This should only be hit under Rack >= 1.5, as this was an illegal
             # key in Rack < 1.5
             hijack = value
           else
-            if value =~ /\n/
-              # avoiding blank, key-only cookies with /\n+/
-              value.split(/\n+/).each { |v| buf << "#{key}: #{v}\r\n" }
-            else
-              buf << "#{key}: #{value}\r\n"
-            end
+            append_header(buf, key, value)
           end
         end
         socket.write(buf << "\r\n".freeze)
       end