Allow custom headers in responses from HttpServer

Author: rastating

lib/wpxf/net/http_server.rb

@@ -92,21 +92,33 @@ def http_server_thread
 
   private
 
+  def send_headers(socket, body, content_type, custom_headers)
+    headers = []
+    headers.push 'HTTP/1.1 200 OK'
+    headers.push "Content-Type: #{content_type}"
+    headers.push "Content-Length: #{body.bytesize}"
+    headers += custom_headers unless custom_headers.nil?
+    headers.push 'Connection: close'
+
+    headers.each do |header|
+      socket.print "#{header}\r\n"
+    end
+  end
+
   def send_response(response, socket)
     content_type = 'text/plain'
     body = ''
+    custom_headers = nil
 
     if response.is_a? String
       body = response
     else
       body = response[:body]
       content_type = response[:type]
+      custom_headers = response[:headers]
     end
 
-    socket.print "HTTP/1.1 200 OK\r\n"\
-                 "Content-Type: #{content_type}\r\n"\
-                 "Content-Length: #{body.bytesize}\r\n"\
-                 "Connection: close\r\n"
+    send_headers(socket, body, content_type, custom_headers)
     socket.print "\r\n"
     socket.print body
   end