feat: add edge support to aws-lambda presets

package.json

@@ -144,6 +144,7 @@
     "get-port-please": "^3.0.1",
     "miniflare": "^2.14.1",
     "prettier": "^3.0.2",
+    "sst": "^2.24.16",
     "typescript": "^5.2.2",
     "unbuild": "^2.0.0",
     "undici": "^5.23.0",
@@ -163,4 +164,4 @@
       ]
     }
   }
-}
+}

src/build.ts

@@ -405,6 +405,7 @@ async function _build(nitro: Nitro, rollupConfig: RollupConfig) {
   const buildInfo = {
     date: new Date(),
     preset: nitro.options.preset,
+    entry: nitro.options.entry.split("/").pop(),
     commands: {
       preview: nitro.options.commands.preview,
       deploy: nitro.options.commands.deploy,

src/presets/aws-lambda.ts

@@ -1,5 +1,34 @@
 import { defineNitroPreset } from "../preset";
+import { generateCdkApp } from "../utils/cdk";
+import { generateSST } from "../utils/sst";
 
 export const awsLambda = defineNitroPreset({
-  entry: "#internal/nitro/entries/aws-lambda",
+  entry: `#internal/nitro/entries/aws-lambda-{{ awsLambda.target }}`,
+  awsLambda: {
+    // we need this defined here so it's picked up by the template in lambda entries
+    target: (process.env.NITRO_AWS_LAMBDA_TARGET || "default") as any,
+  },
+  hooks: {
+    "rollup:before": (nitro) => {
+      const target = nitro.options.awsLambda?.target as unknown;
+      if (!target || target === "default") {
+        nitro.logger.warn(
+          "Neither `awsLambda.target` or `NITRO_AWS_LAMBDA_TARGET` is set. Set the target to remove this warning. See https://nitro.unjs.io/deploy/providers/aws-lambda for more information."
+        );
+        // Default to single region lambda
+        nitro.options.awsLambda = { target: "single" };
+      }
+    },
+    async compiled(nitro) {
+      if (nitro.options.awsLambda.sst) {
+        await generateSST(nitro);
+      }
+      if (
+        nitro.options.awsLambda.target === "edge" &&
+        nitro.options.awsLambda.cdk === true
+      ) {
+        await generateCdkApp(nitro);
+      }
+    },
+  },
 });

src/runtime/entries/aws-lambda-default.ts

@@ -0,0 +1,2 @@
+// we need this file to detect if the user is passing a `target` property
+export { handler } from "./aws-lambda-single";

src/runtime/entries/aws-lambda-edge.ts

@@ -0,0 +1,43 @@
+import type {
+  CloudFrontRequestEvent,
+  CloudFrontResultResponse,
+  Context,
+} from "aws-lambda";
+import "#internal/nitro/virtual/polyfill";
+import { nitroApp } from "../app";
+import {
+  normalizeCloudfrontBody,
+  normalizeCloudfrontIncomingHeaders,
+  normalizeCloudfrontOutgoingHeaders,
+  normalizeLambdaOutgoingBody,
+} from "../utils.lambda";
+
+export const handler = async function handler(
+  event: CloudFrontRequestEvent,
+  context: Context
+): Promise<CloudFrontResultResponse> {
+  const request = event.Records[0].cf.request;
+
+  const url = getFullUrl(request.uri, request.querystring);
+  const headers = normalizeCloudfrontIncomingHeaders(request.headers);
+  const body = normalizeCloudfrontBody(request.body);
+
+  const r = await nitroApp.localCall({
+    event,
+    url,
+    context,
+    headers,
+    method: request.method,
+    body,
+  });
+
+  return {
+    status: r.status.toString(),
+    headers: normalizeCloudfrontOutgoingHeaders(r.headers),
+    body: normalizeLambdaOutgoingBody(r.body, r.headers),
+  };
+};
+
+function getFullUrl(uri: string, queryString?: string) {
+  return queryString ? `${uri}?${queryString}` : uri;
+}

src/runtime/utils.lambda.ts

@@ -1,5 +1,9 @@
 import type { Readable } from "node:stream";
-import type { APIGatewayProxyEventHeaders } from "aws-lambda";
+import type {
+  APIGatewayProxyEventHeaders,
+  CloudFrontHeaders,
+  CloudFrontRequest,
+} from "aws-lambda";
 
 export function normalizeLambdaIncomingHeaders(
   headers?: APIGatewayProxyEventHeaders
@@ -89,3 +93,37 @@ const TEXT_TYPE_RE = /^text\/|\/(json|xml)|utf-?8/;
 function isTextType(contentType = "") {
   return TEXT_TYPE_RE.test(contentType);
 }
+
+export function normalizeCloudfrontOutgoingHeaders(
+  headers: Record<string, string | number | string[] | undefined>
+): CloudFrontHeaders {
+  return Object.fromEntries(
+    Object.entries(headers)
+      .filter(([key]) => !["content-length"].includes(key))
+      .map(([key, v]) => [
+        key,
+        Array.isArray(v)
+          ? v.map((value) => ({ key, value }))
+          : [{ key, value: v.toString() }],
+      ])
+  );
+}
+
+export function normalizeCloudfrontIncomingHeaders(headers: CloudFrontHeaders) {
+  return Object.fromEntries(
+    Object.entries(headers).map(([key, keyValues]) => [
+      key,
+      keyValues.map(({ value }) => value),
+    ])
+  );
+}
+
+export function normalizeCloudfrontBody(body?: CloudFrontRequest["body"]) {
+  if (body === undefined) {
+    return undefined;
+  }
+
+  return body.encoding === "base64"
+    ? decodeURIComponent(Buffer.from(body.data, "base64").toString("utf8"))
+    : body.data;
+}

src/types/presets.ts

@@ -1,7 +1,7 @@
+import type { SSTConfig } from "sst";
 import type { HttpsOptions } from "firebase-functions/v2/https";
 import type { RuntimeOptions, region } from "firebase-functions";
 import type { CloudflarePagesRoutes } from "../presets/cloudflare-pages";
-
 /**
  * Vercel Build Output Configuration
  * @see https://vercel.com/docs/build-output-api/v3
@@ -95,6 +95,22 @@ interface FirebaseOptionsGen2 extends FirebaseOptionsBase {
 
 type FirebaseOptions = FirebaseOptionsGen1 | FirebaseOptionsGen2;
 
+interface AWSLambdaOptionsBase {
+  target: "single" | "edge";
+  sst?: boolean;
+  sstOptions?: Awaited<ReturnType<SSTConfig["config"]>>;
+}
+
+interface AwsLambdaOptionsSingleRegion extends AWSLambdaOptionsBase {
+  target: "single";
+}
+
+interface AwsLambdaOptionsEdge extends AWSLambdaOptionsBase {
+  target: "edge";
+  cdk?: boolean;
+}
+
+type AWSLambdaOptions = AwsLambdaOptionsSingleRegion | AwsLambdaOptionsEdge;
 /**
  * https://vercel.com/docs/build-output-api/v3/primitives#serverless-function-configuration
  */
@@ -149,6 +165,7 @@ export interface PresetOptions {
     functions?: VercelServerlessFunctionConfig;
   };
   firebase: FirebaseOptions;
+  awsLambda: AWSLambdaOptions;
   cloudflare: {
     pages: {
       /**