don't transform payloads

Author: cnuss

src/proxy.ts

@@ -1,4 +1,4 @@
-import { APIGatewayProxyResultV2 } from "aws-lambda";
+import { APIGatewayProxyResult } from "aws-lambda";
 import axios, { AxiosResponseHeaders, RawAxiosResponseHeaders } from "axios";
 import net from "net";
 import { EndpointRequest, EndpointResponse } from "./types";
@@ -102,22 +102,16 @@ export const endpointProxy = async ({
       headers: rawHeaders,
       data: decodedBody,
       timeout: deadline,
+      responseType: "arraybuffer",
     })
     .then((response) => {
       const { data: rawData, headers: rawHeaders } = response;
 
-      const body =
-        rawData && typeof rawData === "string"
-          ? rawData
-          : Buffer.from(rawData).toString("base64");
-
-      const isBase64Encoded = rawData && typeof rawData !== "string";
-
-      const payload: APIGatewayProxyResultV2 = {
+      const payload: APIGatewayProxyResult = {
         statusCode: response.status,
         headers: convertHeaders(rawHeaders),
-        body,
-        isBase64Encoded,
+        body: Buffer.from(rawData).toString("base64"),
+        isBase64Encoded: true,
       };
 
       return {

src/routing.ts

@@ -1,4 +1,3 @@
-import { APIGatewayProxyEventV2, APIGatewayProxyResultV2 } from "aws-lambda";
 import axios from "axios";
 import { EndpointRequest } from "./types";
 import { endpointProxy } from "./proxy";
@@ -13,34 +12,37 @@ export const routeEvents = async (
     {
       // block indefinitely until a response is received
       timeout: 0,
+      responseType: "text",
     }
   );
 
-  const requestId = headers["Lambda-Runtime-Aws-Request-Id"] as string;
+  const requestId = headers["lambda-runtime-aws-request-id"];
+
+  if (!requestId) {
+    throw new Error("No request ID found in response headers");
+  }
 
   console.log("Received request from Lambda Runtime API", { requestId });
 
   const initialDeadline = Number.parseInt(
-    headers["Lambda-Runtime-Deadline-Ms"]
+    headers["lambda-runtime-deadline-ms"]
   );
 
-  // TODO handle V1 payloads
-  const event = JSON.parse(data) as APIGatewayProxyEventV2;
-  let payload: APIGatewayProxyResultV2 | undefined = undefined;
+  let payload: any | undefined = undefined;
 
   if (!endpoint) {
     const { execa } = await import("execa");
     // no endpoint, just exec the bin
     const { stdout } = await execa({
       stderr: ["inherit"],
-    })`${bin} ${JSON.stringify(event)}`;
+    })`${bin} ${data}`;
 
-    payload = JSON.parse(stdout) as APIGatewayProxyResultV2;
+    payload = JSON.parse(stdout);
   } else {
     const request: EndpointRequest = {
       requestId,
       endpoint,
-      event,
+      event: JSON.parse(data),
       initialDeadline,
     };
 

src/types.ts

@@ -1,13 +1,11 @@
-import { APIGatewayProxyEventV2, APIGatewayProxyResultV2 } from "aws-lambda";
-
 export type EndpointRequest = {
   requestId: string;
   endpoint: string;
-  event: APIGatewayProxyEventV2;
+  event: any;
   initialDeadline: number;
 };
 
 export type EndpointResponse = {
   requestId: string;
-  payload: APIGatewayProxyResultV2;
+  payload: any;
 };