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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
// Normal Imports
import type { SSRManifest } from 'astro';
import { App } from 'astro/app';
// @ts-expect-error
import { fromFileUrl, serveFile, Server } from '@astrojs/deno/__deno_imports.js';
interface Options {
port?: number;
hostname?: string;
start?: boolean;
}
let _server: Server | undefined = undefined;
let _startPromise: Promise<void> | undefined = undefined;
async function* getPrerenderedFiles(clientRoot: URL): AsyncGenerator<URL> {
// @ts-expect-error
for await (const ent of Deno.readDir(clientRoot)) {
if (ent.isDirectory) {
yield* getPrerenderedFiles(new URL(`./${ent.name}/`, clientRoot));
} else if (ent.name.endsWith('.html')) {
yield new URL(`./${ent.name}`, clientRoot);
}
}
}
export function start(manifest: SSRManifest, options: Options) {
if (options.start === false) {
return;
}
const clientRoot = new URL('../client/', import.meta.url);
const app = new App(manifest);
const handler = async (request: Request, connInfo: any) => {
if (app.match(request)) {
let ip = connInfo?.remoteAddr?.hostname;
Reflect.set(request, Symbol.for('astro.clientAddress'), ip);
const response = await app.render(request);
if (app.setCookieHeaders) {
for (const setCookieHeader of app.setCookieHeaders(response)) {
response.headers.append('Set-Cookie', setCookieHeader);
}
}
return response;
}
// If the request path wasn't found in astro,
// try to fetch a static file instead
const url = new URL(request.url);
const localPath = new URL('./' + app.removeBase(url.pathname), clientRoot);
let fileResp = await serveFile(request, fromFileUrl(localPath));
// Attempt to serve `index.html` if 404
if (fileResp.status == 404) {
let fallback;
for await (const file of getPrerenderedFiles(clientRoot)) {
const pathname = file.pathname.replace(/\/(index)?\.html$/, '');
if (localPath.pathname.endsWith(pathname)) {
fallback = file;
break;
}
}
if (fallback) {
fileResp = await serveFile(request, fromFileUrl(fallback));
}
}
// If the static file can't be found
if (fileResp.status == 404) {
// Render the astro custom 404 page
const response = await app.render(request);
if (app.setCookieHeaders) {
for (const setCookieHeader of app.setCookieHeaders(response)) {
response.headers.append('Set-Cookie', setCookieHeader);
}
}
return response;
// If the static file is found
} else {
return fileResp;
}
};
const port = options.port ?? 8085;
_server = new Server({
port,
hostname: options.hostname ?? '0.0.0.0',
handler,
});
_startPromise = Promise.resolve(_server.listenAndServe());
console.error(`Server running on port ${port}`);
}
export function createExports(manifest: SSRManifest, options: Options) {
const app = new App(manifest);
return {
async stop() {
if (_server) {
_server.close();
_server = undefined;
}
await Promise.resolve(_startPromise);
},
running() {
return _server !== undefined;
},
async start() {
return start(manifest, options);
},
async handle(request: Request) {
return app.render(request);
},
};
}
|
