/
server-template.js
110 lines (89 loc) · 3.09 KB
/
server-template.js
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
// This is a custom CloudFront Lambda@Edge handler which we will utilise to wrap
// the Remix server build with.
//
// It additionally exposes an environment variable token which is used by our
// Lambda Replacer code to inject the appropriate environment variables. This
// strategy is required as Lambda@Edge doesn't natively support environment
// variables - they need to be inlined.
import {
Headers as NodeHeaders,
Request as NodeRequest,
} from "@remix-run/node";
import { installGlobals } from "@remix-run/node";
import { createRequestHandler as createRemixRequestHandler } from "@remix-run/server-runtime";
import { URL } from "url";
import * as remixServerBuild from "./index.js";
installGlobals();
function createNodeRequestHeaders(requestHeaders) {
const headers = new NodeHeaders();
for (const [key, values] of Object.entries(requestHeaders)) {
for (const { value } of values) {
if (value) {
headers.append(key, value);
}
}
}
return headers;
}
function createNodeRequest(event) {
const request = event.Records[0].cf.request;
const host = request.headers["host"]
? request.headers["host"][0].value
: undefined;
const search = request.querystring.length ? `?${request.querystring}` : "";
const url = new URL(request.uri + search, `https://${host}`);
return new NodeRequest(url.toString(), {
method: request.method,
headers: createNodeRequestHeaders(request.headers),
body: request.body?.data
? request.body.encoding === "base64"
? Buffer.from(request.body.data, "base64").toString()
: request.body.data
: undefined,
});
}
function createCloudFrontResponseHeaders(responseHeaders) {
const headers = {};
const rawHeaders = responseHeaders.raw();
for (const key in rawHeaders) {
const value = rawHeaders[key];
for (const v of value) {
headers[key] = [...(headers[key] || []), { key, value: v }];
}
}
return headers;
}
function createCloudFrontRequestHandler({
remixServerBuild,
getLoadContext,
mode = process.env.NODE_ENV,
}) {
const remixRequestHandler = createRemixRequestHandler(remixServerBuild, mode);
return async (event, _context) => {
const request = createNodeRequest(event);
const loadContext =
typeof getLoadContext === "function" ? getLoadContext(event) : undefined;
const response = await remixRequestHandler(request, loadContext);
return {
status: String(response.status),
headers: createCloudFrontResponseHeaders(response.headers),
bodyEncoding: "text",
body: await response.text(),
};
};
}
const cloudFrontRequestHandler = createCloudFrontRequestHandler({
remixServerBuild,
});
export async function handler(...args) {
try {
// The right hand side of this assignment will get replaced during
// deployment with an object of environment key-value pairs;
const environment = "{{ _SST_REMIX_SITE_ENVIRONMENT_ }}";
process.env = { ...process.env, ...environment };
} catch (e) {
console.log("Failed to set SST RemixSite environment.");
console.log(e);
}
return await cloudFrontRequestHandler(...args);
}