@jaypie/express 1.2.11-dev.6 → 1.2.12

package/dist/cjs/index.cjs

@@ -856,8 +856,6 @@ class LambdaResponseStreaming extends node_stream.Writable {
         for (const [key, value] of this._headers) {
             headers[key] = Array.isArray(value) ? value.join(", ") : String(value);
         }
-        // Add version header for deployment verification
-        headers["x-jaypie-streaming"] = "1.2.11-dev.6";
         // Lambda streaming requires body content for metadata to be transmitted.
         // Convert 204 No Content to 200 OK with empty JSON body as workaround.
         // See: https://github.com/finlaysonstudio/jaypie/issues/178
@@ -1615,15 +1613,22 @@ function safeSendJson(res, statusCode, data) {
     }
     // Fall back to standard Express methods for real responses
     res.status(statusCode);
-    // CRITICAL: For Lambda streaming responses, flush headers before send to
-    // initialize the stream wrapper. This ensures the status code is captured
-    // before any writes occur (which would auto-flush with default 200).
+    // CRITICAL: For Lambda streaming responses, set content-type BEFORE flushing
+    // headers, then flush to initialize the stream wrapper with all headers.
     // Uses Symbol marker for reliable detection that survives prototype manipulation.
-    if (isLambdaStreamingResponse(res) &&
-        typeof res.flushHeaders === "function") {
-        res.flushHeaders();
+    if (isLambdaStreamingResponse(res)) {
+        // Set content-type before flushing so it's included in the metadata
+        res.setHeader("content-type", "application/json");
+        if (typeof res.flushHeaders === "function") {
+            res.flushHeaders();
+        }
+        // Write JSON directly and end the stream (don't call res.json which would
+        // try to set content-type again after headers are already sent)
+        res.end(JSON.stringify(data));
+    }
+    else {
+        res.json(data);
     }
-    res.json(data);
 }
 /**
  * Safely send a response body, avoiding dd-trace interception.
@@ -1655,15 +1660,25 @@ function safeSend(res, statusCode, body) {
     // initialize the stream wrapper. This ensures the status code is captured
     // before any writes occur (which would auto-flush with default 200).
     // Uses Symbol marker for reliable detection that survives prototype manipulation.
-    if (isLambdaStreamingResponse(res) &&
-        typeof res.flushHeaders === "function") {
-        res.flushHeaders();
-    }
-    if (body !== undefined) {
-        res.send(body);
+    if (isLambdaStreamingResponse(res)) {
+        if (typeof res.flushHeaders === "function") {
+            res.flushHeaders();
+        }
+        // Write directly to stream and end (bypass Express send middleware)
+        if (body !== undefined) {
+            res.end(body);
+        }
+        else {
+            res.end();
+        }
     }
     else {
-        res.send();
+        if (body !== undefined) {
+            res.send(body);
+        }
+        else {
+            res.send();
+        }
     }
 }
 function expressHandler(handlerOrOptions, optionsOrHandler) {