Add chunkedResponseModeStart, sendChunk, chunkedResponseFinalize

to WebServer library for chunked HTTP responses

Author: ritesh006

libraries/WebServer/src/WebServer.cpp

@@ -551,6 +551,78 @@ void WebServer::enableETag(bool enable, ETagFunction fn) {
   _eTagFunction = fn;
 }
 
+void WebServer::chunkResponseBegin(const char* contentType) {
+  if (_chunkedResponseActive) {
+    log_e("Already in chunked response mode");
+    return;
+  }
+
+  if (strchr(contentType, '\r') || strchr(contentType, '\n')) {
+    log_e("Invalid character in content type");
+    return;
+  }
+
+  _chunkedResponseActive = true;
+  _chunkedClient = _currentClient;
+
+  _contentLength = CONTENT_LENGTH_UNKNOWN;
+
+  String header;
+  _prepareHeader(header, 200, contentType, 0);
+  _currentClientWrite(header.c_str(), header.length());
+
+  _chunkedResponseActive = true;
+  _chunkedClient = _currentClient;
+}
+
+void WebServer::chunkWrite(const char* data, size_t length) {
+  if (!_chunkedResponseActive)
+  {
+    log_e("Chunked response has not been started");
+    return;
+  } 
+
+  char chunkSize[11];
+  snprintf(chunkSize, sizeof(chunkSize), "%zx\r\n", length);
+
+  if (_chunkedClient.write(chunkSize) != strlen(chunkSize)) {
+    log_e("Failed to write chunk size");
+    _chunkedResponseActive = false;
+    return;
+  }
+
+  if (_chunkedClient.write((const uint8_t*)data, length) != length) {
+    log_e("Failed to write chunk data");
+    _chunkedResponseActive = false;
+    return;
+  }
+
+  if (_chunkedClient.write("\r\n") != 2) {
+    log_e("Failed to write chunk terminator");
+    _chunkedResponseActive = false;
+    return;
+  }
+}
+
+void WebServer::chunkResponseEnd() {
+  if (!_chunkedResponseActive)
+  {
+    log_e("Chunked response has not been started");
+    return;
+  }
+
+  if (_chunkedClient.write("0\r\n\r\n", 5) != 5) {
+    log_e("Failed to write terminating chunk");
+  }
+
+  _chunkedClient.flush();
+  _chunkedResponseActive = false;
+  _chunked = false;
+  _chunkedClient = NetworkClient();
+
+  _clearResponseHeaders();
+}
+
 void WebServer::_prepareHeader(String &response, int code, const char *content_type, size_t contentLength) {
   _responseCode = code;
 

libraries/WebServer/src/WebServer.h

@@ -115,6 +115,10 @@ class WebServer {
   const String AuthTypeDigest = F("Digest");
   const String AuthTypeBasic = F("Basic");
 
+  void chunkResponseBegin(const char* contentType = "text/plain");
+  void chunkWrite(const char* data, size_t length);
+  void chunkResponseEnd();
+
   /* Callbackhandler for authentication. The extra parameters depend on the
    * HTTPAuthMethod mode:
    *
@@ -241,6 +245,10 @@ class WebServer {
 
   static String responseCodeToString(int code);
 
+private:
+  bool _chunkedResponseActive = false;
+  NetworkClient _chunkedClient; // Store by value, no dangling pointer
+
 protected:
   virtual size_t _currentClientWrite(const char *b, size_t l) {
     return _currentClient.write(b, l);