Correctly handle title

Author: MarcelRobitaille

src/internals/StreamHttpReply.cpp

@@ -18,20 +18,25 @@ ArduinoHttpServer::AbstractStreamHttpReply::AbstractStreamHttpReply(Stream& stre
 
 }
 
-void ArduinoHttpServer::AbstractStreamHttpReply::sendHeader(size_t size)
-{
+void ArduinoHttpServer::AbstractStreamHttpReply::sendHeader(
+    size_t size, const String& title) {
    // Read away remaining bytes.
-   while(getStream().read()>=0);
+   while (getStream().read() >= 0) {
+   }
 
-   getStream().print( AHS_F("HTTP/1.1 ") );
-   getStream().print( getCode() + " ");
-   getStream().print( AHS_F("Connection: close\r\n") );
-   if (size > 0)
-   {
-      getStream().print( AHS_F("Content-Length: ") ); getStream().print(size); getStream().print( AHS_F("\r\n") );
+   getStream().print(AHS_F("HTTP/1.1 "));
+   getStream().print(getCode() + " ");
+   getStream().print(title + "\r\n");
+   getStream().print(AHS_F("Connection: keep-alive\r\n"));
+   if (size > 0) {
+      getStream().print(AHS_F("Content-Length: "));
+      getStream().print(size);
+      getStream().print(AHS_F("\r\n"));
    }
-   getStream().print( AHS_F("Content-Type: ") ); getStream().print( m_contentType ); getStream().print( AHS_F("\r\n") );
-   getStream().print( AHS_F("\r\n") );
+   getStream().print(AHS_F("Content-Type: "));
+   getStream().print(m_contentType);
+   getStream().print(AHS_F("\r\n"));
+   getStream().print(AHS_F("\r\n"));
 }
 
 //------------------------------------------------------------------------------
@@ -40,14 +45,16 @@ void ArduinoHttpServer::AbstractStreamHttpReply::sendHeader(size_t size)
 void ArduinoHttpServer::AbstractStreamHttpReply::send(const String& data, const String& title)
 {
    DEBUG_ARDUINO_HTTP_SERVER_PRINT("Printing Reply ... ");
-   AbstractStreamHttpReply::sendHeader(data.length());
+   AbstractStreamHttpReply::sendHeader(data.length(), title);
    getStream().print( data ); getStream().print( AHS_F("\r\n") );
    DEBUG_ARDUINO_HTTP_SERVER_PRINTLN("done.");
 }
 
-void ArduinoHttpServer::AbstractStreamHttpReply::send(const uint8_t* buf, const size_t size) {
+void ArduinoHttpServer::AbstractStreamHttpReply::send(const uint8_t* buf,
+                                                      const size_t size,
+                                                      const String& title) {
    DEBUG_ARDUINO_HTTP_SERVER_PRINT("Printing Reply ... ");
-   AbstractStreamHttpReply::sendHeader(size);
+   AbstractStreamHttpReply::sendHeader(size, title);
    getStream().write(buf, size);
    DEBUG_ARDUINO_HTTP_SERVER_PRINTLN("done.");
 }

src/internals/StreamHttpReply.hpp

@@ -25,9 +25,9 @@ class AbstractStreamHttpReply
 {
 
 public:
-    virtual void sendHeader(size_t size);
+    virtual void sendHeader(size_t size, const String& title);
     virtual void send(const String& data, const String& title);
-    virtual void send(const uint8_t* buf, const size_t size);
+    virtual void send(const uint8_t* buf, const size_t size, const String& title);
 
 protected:
    AbstractStreamHttpReply(Stream& stream, const String& contentType, const String& code);
@@ -83,7 +83,8 @@ class StreamHttpReply: public AbstractStreamHttpReply
 public:
     StreamHttpReply(Stream& stream, const String& contentType);
     virtual void send(const String& data, const bool gzipencoded=false) { AbstractStreamHttpReply::send(data, "OK"); };
-    virtual void send(const uint8_t *buf, const size_t size) { AbstractStreamHttpReply::send(buf, size); };
+    virtual void send(const uint8_t* buf, const size_t size, const String& title="OK") { AbstractStreamHttpReply::send(buf, size, title); };
+    virtual void sendHeader(size_t size, const String& title="OK") { AbstractStreamHttpReply::sendHeader(size, title); }
 };