summaryrefslogtreecommitdiff
path: root/packages/integrations/node/src/server.ts
blob: 12fcf04484a4a7daf808905477d7a14351215323 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import { polyfill } from '@astrojs/webapi';
import type { SSRManifest } from 'astro';
import { NodeApp } from 'astro/app/node';
import type { IncomingMessage, ServerResponse } from 'http';
import type { Readable } from 'stream';

polyfill(globalThis, {
	exclude: 'window document',
});

export function createExports(manifest: SSRManifest) {
	const app = new NodeApp(manifest);
	return {
		async handler(req: IncomingMessage, res: ServerResponse, next?: (err?: unknown) => void) {
			try {
				const route = app.match(req);

				if (route) {
					try {
						const response = await app.render(req);
						await writeWebResponse(res, response);
					} catch (err: unknown) {
						if (next) {
							next(err);
						} else {
							throw err;
						}
					}
				} else if (next) {
					return next();
				}
			} catch (err: unknown) {
				if (!res.headersSent) {
					res.writeHead(500, `Server error`);
					res.end();
				}
			}
		},
	};
}

async function writeWebResponse(res: ServerResponse, webResponse: Response) {
	const { status, headers, body } = webResponse;
	res.writeHead(status, Object.fromEntries(headers.entries()));
	if (body) {
		for await (const chunk of body as unknown as Readable) {
			res.write(chunk);
		}
	}
	res.end();
}