@jaypie/express 1.2.8 → 1.2.10

package/dist/esm/adapter/LambdaResponseStreaming.d.ts

@@ -1,6 +1,7 @@
 import type { OutgoingHttpHeaders } from "node:http";
 import { Writable } from "node:stream";
 import type { ResponseStream } from "./types.js";
+export declare const JAYPIE_LAMBDA_STREAMING: unique symbol;
 /**
  * Mock ServerResponse that streams directly to Lambda responseStream.
  * Uses awslambda.HttpResponseStream.from() to set status and headers.

package/dist/esm/index.js

@@ -624,6 +624,9 @@ class LambdaResponseBuffered extends Writable {
 //
 // Constants
 //
+// Symbol to identify Lambda streaming responses. Uses Symbol.for() to ensure
+// the same symbol is used across bundles/realms. Survives prototype manipulation.
+const JAYPIE_LAMBDA_STREAMING = Symbol.for("@jaypie/express/LambdaStreaming");
 // Get Node's internal kOutHeaders symbol from ServerResponse prototype.
 // This is needed for compatibility with Datadog dd-trace instrumentation,
 // which patches HTTP methods and expects this internal state to exist.
@@ -652,6 +655,10 @@ class LambdaResponseStreaming extends Writable {
         this._pendingWrites = [];
         this._wrappedStream = null;
         this._responseStream = responseStream;
+        // Mark as Lambda streaming response using Symbol for reliable detection.
+        // Survives prototype manipulation from Express and dd-trace.
+        this[JAYPIE_LAMBDA_STREAMING] =
+            true;
         // Initialize Node's internal kOutHeaders for dd-trace compatibility.
         // dd-trace patches HTTP methods and expects this internal state.
         if (kOutHeaders) {
@@ -1024,13 +1031,6 @@ function createLambdaHandler(app, _options) {
             await runExpressApp(app, req, res);
             // Get Lambda response - await explicitly to ensure we have the result
             result = await res.getResult();
-            // Debug: Log the response before returning
-            console.log("[createLambdaHandler] Returning response:", JSON.stringify({
-                statusCode: result.statusCode,
-                headers: result.headers,
-                bodyLength: result.body?.length,
-                isBase64Encoded: result.isBase64Encoded,
-            }));
             return result;
         }
         catch (error) {
@@ -1551,6 +1551,14 @@ const logger$1 = log;
 function isLambdaMockResponse(res) {
     return (res[JAYPIE_LAMBDA_MOCK] === true);
 }
+/**
+ * Check if response is a Lambda streaming response.
+ * Uses Symbol marker to survive prototype chain modifications from Express and dd-trace.
+ */
+function isLambdaStreamingResponse(res) {
+    return (res[JAYPIE_LAMBDA_STREAMING] ===
+        true);
+}
 /**
  * Safely send a JSON response, avoiding dd-trace interception.
  * For Lambda mock responses, directly manipulates internal state instead of
@@ -1583,7 +1591,16 @@ function safeSendJson(res, statusCode, data) {
         return;
     }
     // Fall back to standard Express methods for real responses
-    res.status(statusCode).json(data);
+    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).
+    // Uses Symbol marker for reliable detection that survives prototype manipulation.
+    if (isLambdaStreamingResponse(res) &&
+        typeof res.flushHeaders === "function") {
+        res.flushHeaders();
+    }
+    res.json(data);
 }
 /**
  * Safely send a response body, avoiding dd-trace interception.
@@ -1611,11 +1628,20 @@ function safeSend(res, statusCode, body) {
         return;
     }
     // 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).
+    // Uses Symbol marker for reliable detection that survives prototype manipulation.
+    if (isLambdaStreamingResponse(res) &&
+        typeof res.flushHeaders === "function") {
+        res.flushHeaders();
+    }
     if (body !== undefined) {
-        res.status(statusCode).send(body);
+        res.send(body);
     }
     else {
-        res.status(statusCode).send();
+        res.send();
     }
 }
 function expressHandler(handlerOrOptions, optionsOrHandler) {