diff options
Diffstat (limited to 'packages/integrations/vercel/src/serverless/entrypoint.ts')
-rw-r--r-- | packages/integrations/vercel/src/serverless/entrypoint.ts | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/packages/integrations/vercel/src/serverless/entrypoint.ts b/packages/integrations/vercel/src/serverless/entrypoint.ts new file mode 100644 index 000000000..a881d701a --- /dev/null +++ b/packages/integrations/vercel/src/serverless/entrypoint.ts @@ -0,0 +1,62 @@ +import type { IncomingMessage, ServerResponse } from 'node:http'; +import type { SSRManifest } from 'astro'; +import { NodeApp, applyPolyfills } from 'astro/app/node'; +import { + ASTRO_LOCALS_HEADER, + ASTRO_MIDDLEWARE_SECRET_HEADER, + ASTRO_PATH_HEADER, + ASTRO_PATH_PARAM, +} from './adapter.js'; + +// Run polyfills immediately so any dependent code can use the globals +applyPolyfills(); + +// Won't throw if the virtual module is not available because it's not supported in +// the users's astro version or if astro:env is not enabled in the project +await import('astro/env/setup') + .then((mod) => mod.setGetEnv((key) => process.env[key])) + .catch(() => {}); + +export const createExports = ( + manifest: SSRManifest, + { middlewareSecret, skewProtection }: { middlewareSecret: string; skewProtection: boolean }, +) => { + const app = new NodeApp(manifest); + const handler = async (req: IncomingMessage, res: ServerResponse) => { + const url = new URL(`https://example.com${req.url}`); + const clientAddress = req.headers['x-forwarded-for'] as string | undefined; + const localsHeader = req.headers[ASTRO_LOCALS_HEADER]; + const middlewareSecretHeader = req.headers[ASTRO_MIDDLEWARE_SECRET_HEADER]; + const realPath = req.headers[ASTRO_PATH_HEADER] ?? url.searchParams.get(ASTRO_PATH_PARAM); + if (typeof realPath === 'string') { + req.url = realPath; + } + + let locals = {}; + if (localsHeader) { + if (middlewareSecretHeader !== middlewareSecret) { + res.statusCode = 403; + res.end('Forbidden'); + return; + } + locals = + typeof localsHeader === 'string' ? JSON.parse(localsHeader) : JSON.parse(localsHeader[0]); + } + // hide the secret from the rest of user code + delete req.headers[ASTRO_MIDDLEWARE_SECRET_HEADER]; + + // https://vercel.com/docs/deployments/skew-protection#supported-frameworks + if (skewProtection && process.env.VERCEL_SKEW_PROTECTION_ENABLED === '1') { + req.headers['x-deployment-id'] = process.env.VERCEL_DEPLOYMENT_ID; + } + + const webResponse = await app.render(req, { addCookieHeader: true, clientAddress, locals }); + await NodeApp.writeResponse(webResponse, res); + }; + + return { default: handler }; +}; + +// HACK: prevent warning +// @astrojs-ssr-virtual-entry (22:23) "start" is not exported by "dist/serverless/entrypoint.js", imported by "@astrojs-ssr-virtual-entry". +export function start() {} |