73 lines
2.2 KiB
JavaScript
73 lines
2.2 KiB
JavaScript
|
import "#nitro-internal-pollyfills";
|
||
|
|
import { useNitroApp } from "nitropack/runtime";
|
||
|
|
import {
|
||
|
|
normalizeCookieHeader,
|
||
|
|
normalizeLambdaIncomingHeaders,
|
||
|
|
normalizeLambdaOutgoingHeaders
|
||
|
|
} from "nitropack/runtime/internal";
|
||
|
|
import { withQuery } from "ufo";
|
||
|
|
const nitroApp = useNitroApp();
|
||
|
|
export const handler = awslambda.streamifyResponse(
|
||
|
|
async (event, responseStream, context) => {
|
||
|
|
const query = {
|
||
|
|
...event.queryStringParameters
|
||
|
|
};
|
||
|
|
const url = withQuery(event.rawPath, query);
|
||
|
|
const method = event.requestContext?.http?.method || "get";
|
||
|
|
if ("cookies" in event && event.cookies) {
|
||
|
|
event.headers.cookie = event.cookies.join(";");
|
||
|
|
}
|
||
|
|
const r = await nitroApp.localCall({
|
||
|
|
event,
|
||
|
|
url,
|
||
|
|
context,
|
||
|
|
headers: normalizeLambdaIncomingHeaders(event.headers),
|
||
|
|
method,
|
||
|
|
query,
|
||
|
|
body: event.isBase64Encoded ? Buffer.from(event.body || "", "base64").toString("utf8") : event.body
|
||
|
|
});
|
||
|
|
const isApiGwV2 = "cookies" in event || "rawPath" in event;
|
||
|
|
const cookies = normalizeCookieHeader(r.headers["set-cookie"]);
|
||
|
|
const httpResponseMetadata = {
|
||
|
|
statusCode: r.status,
|
||
|
|
...cookies.length > 0 && {
|
||
|
|
...isApiGwV2 ? { cookies } : { multiValueHeaders: { "set-cookie": cookies } }
|
||
|
|
},
|
||
|
|
headers: {
|
||
|
|
...normalizeLambdaOutgoingHeaders(r.headers, true),
|
||
|
|
"Transfer-Encoding": "chunked"
|
||
|
|
}
|
||
|
|
};
|
||
|
|
const body = r.body ?? new ReadableStream({
|
||
|
|
start(controller) {
|
||
|
|
controller.enqueue("");
|
||
|
|
controller.close();
|
||
|
|
}
|
||
|
|
});
|
||
|
|
const writer = awslambda.HttpResponseStream.from(
|
||
|
|
// @ts-expect-error TODO: IMPORTANT! It should be a Writable according to the aws-lambda types
|
||
|
|
responseStream,
|
||
|
|
httpResponseMetadata
|
||
|
|
);
|
||
|
|
if (!body.getReader) {
|
||
|
|
writer.write(
|
||
|
|
r.body
|
||
|
|
/* TODO */
|
||
|
|
);
|
||
|
|
writer.end();
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const reader = body.getReader();
|
||
|
|
await streamToNodeStream(reader, responseStream);
|
||
|
|
writer.end();
|
||
|
|
}
|
||
|
|
);
|
||
|
|
async function streamToNodeStream(reader, writer) {
|
||
|
|
let readResult = await reader.read();
|
||
|
|
while (!readResult.done) {
|
||
|
|
writer.write(readResult.value);
|
||
|
|
readResult = await reader.read();
|
||
|
|
}
|
||
|
|
writer.end();
|
||
|
|
}
|