/
router.ts
190 lines (167 loc) · 5.14 KB
/
router.ts
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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
import type { Ipc, StructuredError } from "@vercel/turbopack-next/ipc/index";
import type { IncomingMessage, ServerResponse } from "node:http";
import { Buffer } from "node:buffer";
import { createServer, makeRequest } from "@vercel/turbopack-next/ipc/server";
import { toPairs } from "@vercel/turbopack-next/internal/headers";
import { makeResolver, RouteResult } from "next/dist/server/lib/route-resolver";
import loadConfig from "next/dist/server/config";
import { PHASE_DEVELOPMENT_SERVER } from "next/dist/shared/lib/constants";
import "next/dist/server/node-polyfill-fetch.js";
import middlewareChunkGroup from "MIDDLEWARE_CHUNK_GROUP";
import middlewareConfig from "MIDDLEWARE_CONFIG";
type RouterRequest = {
method: string;
pathname: string;
rawHeaders: [string, string][];
rawQuery: string;
};
type IpcOutgoingMessage = {
type: "value";
data: string | Buffer;
};
type MessageData =
| { type: "middleware-headers"; data: MiddlewareHeadersResponse }
| { type: "middleware-body"; data: Uint8Array }
| {
type: "rewrite";
data: RewriteResponse;
}
| {
type: "error";
error: StructuredError;
}
| { type: "none" };
type RewriteResponse = {
url: string;
headers: [string, string][];
};
type MiddlewareHeadersResponse = {
statusCode: number;
headers: [string, string][];
};
let resolveRouteMemo: Promise<
(req: IncomingMessage, res: ServerResponse) => Promise<void>
>;
async function getResolveRoute(
dir: string
): ReturnType<
typeof import("next/dist/server/lib/route-resolver").makeResolver
> {
const nextConfig = await loadConfig(
PHASE_DEVELOPMENT_SERVER,
process.cwd(),
undefined,
undefined,
true
);
return await makeResolver(dir, nextConfig, {
files: middlewareChunkGroup.filter((f) => /\.[mc]?js$/.test(f)),
matcher: middlewareConfig.matcher,
});
}
export default async function route(
ipc: Ipc<RouterRequest, IpcOutgoingMessage>,
routerRequest: RouterRequest,
dir: string
) {
const [resolveRoute, server] = await Promise.all([
(resolveRouteMemo ??= getResolveRoute(dir)),
createServer(),
]);
try {
const {
clientRequest,
clientResponsePromise,
serverRequest,
serverResponse,
} = await makeRequest(
server,
routerRequest.method,
routerRequest.pathname,
routerRequest.rawQuery,
routerRequest.rawHeaders
);
// Send the clientRequest, so the server parses everything. We can then pass
// the serverRequest to Next.js to handle.
clientRequest.end();
// The route promise must not block us from starting the client response
// handling, so we cannot await it yet. By making the call, we allow
// Next.js to start writing to the response whenever it's ready.
const routePromise = resolveRoute(serverRequest, serverResponse);
// Now that the Next.js has started processing the route, the
// clientResponsePromise will resolve once they write data and then we can
// begin streaming.
// We again cannot block on the clientResponsePromise, because an error may
// occur in the routePromise while we're waiting.
const responsePromise = clientResponsePromise.then((c) =>
handleClientResponse(ipc, c)
);
// Now that both promises are in progress, we await both so that a
// rejection in either will end the routing.
const [response] = await Promise.all([responsePromise, routePromise]);
server.close();
return response;
} catch (e) {
// Server doesn't need to be closed, because the sendError will terminate
// the process.
ipc.sendError(e as Error);
}
}
async function handleClientResponse(
ipc: Ipc<RouterRequest, IpcOutgoingMessage>,
clientResponse: IncomingMessage
): Promise<MessageData | void> {
if (clientResponse.headers["x-nextjs-route-result"] === "1") {
clientResponse.setEncoding("utf8");
// We're either a redirect or a rewrite
let buffer = "";
for await (const chunk of clientResponse) {
buffer += chunk;
}
const data = JSON.parse(buffer) as RouteResult;
switch (data.type) {
case "none":
return {
type: "none",
};
case "error":
return {
type: "error",
error: data.error,
};
case "rewrite":
return {
type: "rewrite",
data: {
url: data.url,
headers: Object.entries(data.headers)
.filter(([, val]) => val != null)
.map(([name, value]) => [name, value!.toString()]),
},
};
default:
// @ts-expect-error data.type is never
throw new Error(`unknown route result type: ${data.type}`);
}
}
const responseHeaders: MiddlewareHeadersResponse = {
statusCode: clientResponse.statusCode!,
headers: toPairs(clientResponse.rawHeaders),
};
await ipc.send({
type: "value",
data: JSON.stringify({
type: "middleware-headers",
data: responseHeaders,
}),
});
for await (const chunk of clientResponse) {
await ipc.send({
type: "value",
data: JSON.stringify({
type: "middleware-body",
data: (chunk as Buffer).toJSON().data,
}),
});
}
}