wrangler 2.0.21 → 2.0.24

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (65) hide show
  1. package/README.md +20 -2
  2. package/bin/wrangler.js +1 -1
  3. package/miniflare-dist/index.mjs +527 -5
  4. package/package.json +18 -5
  5. package/src/__tests__/configuration.test.ts +88 -16
  6. package/src/__tests__/dev.test.tsx +95 -4
  7. package/src/__tests__/generate.test.ts +93 -0
  8. package/src/__tests__/helpers/mock-cfetch.ts +54 -2
  9. package/src/__tests__/index.test.ts +10 -27
  10. package/src/__tests__/jest.setup.ts +31 -1
  11. package/src/__tests__/kv.test.ts +82 -61
  12. package/src/__tests__/metrics.test.ts +35 -0
  13. package/src/__tests__/publish.test.ts +573 -254
  14. package/src/__tests__/r2.test.ts +155 -71
  15. package/src/__tests__/user.test.ts +1 -0
  16. package/src/__tests__/validate-dev-props.test.ts +56 -0
  17. package/src/__tests__/version.test.ts +35 -0
  18. package/src/__tests__/whoami.test.tsx +60 -1
  19. package/src/api/dev.ts +43 -9
  20. package/src/bundle.ts +297 -37
  21. package/src/cfetch/internal.ts +34 -2
  22. package/src/config/config.ts +14 -2
  23. package/src/config/environment.ts +40 -8
  24. package/src/config/index.ts +13 -0
  25. package/src/config/validation.ts +110 -8
  26. package/src/create-worker-preview.ts +3 -1
  27. package/src/create-worker-upload-form.ts +25 -0
  28. package/src/dev/dev.tsx +135 -31
  29. package/src/dev/local.tsx +48 -20
  30. package/src/dev/remote.tsx +39 -12
  31. package/src/dev/use-esbuild.ts +25 -0
  32. package/src/dev/validate-dev-props.ts +31 -0
  33. package/src/dev-registry.tsx +157 -0
  34. package/src/dev.tsx +137 -65
  35. package/src/generate.ts +112 -14
  36. package/src/index.tsx +222 -7
  37. package/src/inspect.ts +93 -5
  38. package/src/metrics/index.ts +1 -0
  39. package/src/metrics/is-ci.ts +14 -0
  40. package/src/metrics/metrics-config.ts +19 -2
  41. package/src/metrics/metrics-dispatcher.ts +1 -0
  42. package/src/metrics/metrics-usage-headers.ts +24 -0
  43. package/src/metrics/send-event.ts +2 -2
  44. package/src/miniflare-cli/assets.ts +543 -0
  45. package/src/miniflare-cli/index.ts +36 -4
  46. package/src/module-collection.ts +3 -3
  47. package/src/pages/constants.ts +1 -0
  48. package/src/pages/deployments.tsx +1 -1
  49. package/src/pages/dev.tsx +85 -639
  50. package/src/pages/publish.tsx +1 -1
  51. package/src/pages/upload.tsx +32 -13
  52. package/src/publish.ts +139 -112
  53. package/src/r2.ts +68 -0
  54. package/src/user/choose-account.tsx +20 -11
  55. package/src/user/user.tsx +20 -2
  56. package/src/whoami.tsx +79 -1
  57. package/src/worker.ts +12 -0
  58. package/templates/first-party-worker-module-facade.ts +18 -0
  59. package/templates/format-dev-errors.ts +32 -0
  60. package/templates/pages-shim.ts +9 -0
  61. package/templates/{static-asset-facade.js → serve-static-assets.ts} +21 -7
  62. package/templates/service-bindings-module-facade.js +51 -0
  63. package/templates/service-bindings-sw-facade.js +39 -0
  64. package/wrangler-dist/cli.d.ts +32 -3
  65. package/wrangler-dist/cli.js +45257 -25209
package/src/pages/dev.tsx CHANGED
@@ -1,26 +1,16 @@
1
1
  import { execSync, spawn } from "node:child_process";
2
- import { existsSync, lstatSync, readFileSync } from "node:fs";
2
+ import { existsSync } from "node:fs";
3
3
  import { tmpdir } from "node:os";
4
- import { join, resolve as resolvePath } from "node:path";
5
- import { URL } from "node:url";
4
+ import { join, resolve } from "node:path";
6
5
  import { watch } from "chokidar";
7
- import { getType } from "mime";
8
- import { getVarsForDev } from "../dev/dev-vars";
6
+ import { unstable_dev } from "../api";
9
7
  import { FatalError } from "../errors";
10
8
  import { logger } from "../logger";
11
9
  import * as metrics from "../metrics";
12
- import { getRequestContextCheckOptions } from "../miniflare-cli/request-context";
13
- import openInBrowser from "../open-in-browser";
14
10
  import { buildFunctions } from "./build";
15
11
  import { SECONDS_TO_WAIT_FOR_PROXY } from "./constants";
16
12
  import { CLEANUP, CLEANUP_CALLBACKS, pagesBetaWarning } from "./utils";
17
- import type { Config } from "../config";
18
- import type {
19
- fetch as miniflareFetch,
20
- Headers as MiniflareHeaders,
21
- } from "@miniflare/core";
22
- import type { MiniflareOptions, Request as MiniflareRequest } from "miniflare";
23
- import type { Argv, ArgumentsCamelCase } from "yargs";
13
+ import type { ArgumentsCamelCase, Argv } from "yargs";
24
14
 
25
15
  type PagesDevArgs = {
26
16
  directory?: string;
@@ -33,6 +23,8 @@ type PagesDevArgs = {
33
23
  kv?: (string | number)[];
34
24
  do?: (string | number)[];
35
25
  "live-reload": boolean;
26
+ "local-protocol"?: "https" | "http";
27
+ "experimental-enable-local-persistence": boolean;
36
28
  "node-compat": boolean;
37
29
  };
38
30
 
@@ -89,6 +81,15 @@ export function Options(yargs: Argv): Argv<PagesDevArgs> {
89
81
  default: false,
90
82
  description: "Auto reload HTML pages when change is detected",
91
83
  },
84
+ "local-protocol": {
85
+ describe: "Protocol to listen to requests on, defaults to http.",
86
+ choices: ["http", "https"] as const,
87
+ },
88
+ "experimental-enable-local-persistence": {
89
+ type: "boolean",
90
+ default: false,
91
+ describe: "Enable persistence for this session (only for local mode)",
92
+ },
92
93
  "node-compat": {
93
94
  describe: "Enable node.js compatibility",
94
95
  default: false,
@@ -100,7 +101,6 @@ export function Options(yargs: Argv): Argv<PagesDevArgs> {
100
101
  type: "string",
101
102
  hidden: true,
102
103
  },
103
- // // TODO: Miniflare user options
104
104
  })
105
105
  .epilogue(pagesBetaWarning);
106
106
  }
@@ -115,6 +115,8 @@ export const Handler = async ({
115
115
  kv: kvs = [],
116
116
  do: durableObjects = [],
117
117
  "live-reload": liveReload,
118
+ "local-protocol": localProtocol,
119
+ "experimental-enable-local-persistence": experimentalEnableLocalPersistence,
118
120
  "node-compat": nodeCompat,
119
121
  config: config,
120
122
  _: [_pages, _dev, ...remaining],
@@ -133,27 +135,35 @@ export const Handler = async ({
133
135
  const functionsDirectory = "./functions";
134
136
  const usingFunctions = existsSync(functionsDirectory);
135
137
 
136
- const command = remaining as (string | number)[];
138
+ const command = remaining;
137
139
 
138
- let proxyPort: number | void;
140
+ let proxyPort: number | undefined;
139
141
 
140
- if (directory === undefined) {
142
+ if (directory !== undefined && command.length > 0) {
143
+ throw new FatalError(
144
+ "Specify either a directory OR a proxy command, not both.",
145
+ 1
146
+ );
147
+ } else if (directory === undefined) {
141
148
  proxyPort = await spawnProxyProcess({
142
149
  port: requestedProxyPort,
143
150
  command,
144
151
  });
145
152
  if (proxyPort === undefined) return undefined;
153
+ } else {
154
+ directory = resolve(directory);
146
155
  }
147
156
 
148
- let miniflareArgs: MiniflareOptions = {};
149
-
150
157
  let scriptReadyResolve: () => void;
151
158
  const scriptReadyPromise = new Promise<void>(
152
- (resolve) => (scriptReadyResolve = resolve)
159
+ (promiseResolve) => (scriptReadyResolve = promiseResolve)
153
160
  );
154
161
 
162
+ let scriptPath: string;
163
+
155
164
  if (usingFunctions) {
156
165
  const outfile = join(tmpdir(), `./functionsWorker-${Math.random()}.js`);
166
+ scriptPath = outfile;
157
167
 
158
168
  if (nodeCompat) {
159
169
  console.warn(
@@ -191,160 +201,83 @@ export const Handler = async ({
191
201
  });
192
202
  await metrics.sendMetricsEvent("build pages functions");
193
203
  });
194
-
195
- miniflareArgs = {
196
- scriptPath: outfile,
197
- };
198
204
  } else {
199
205
  // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
200
206
  scriptReadyResolve!();
201
207
 
202
- const scriptPath =
208
+ scriptPath =
203
209
  directory !== undefined
204
210
  ? join(directory, singleWorkerScriptPath)
205
211
  : singleWorkerScriptPath;
206
212
 
207
- if (existsSync(scriptPath)) {
208
- miniflareArgs = {
209
- scriptPath,
210
- };
211
- } else {
213
+ if (!existsSync(scriptPath)) {
212
214
  logger.log("No functions. Shimming...");
213
- miniflareArgs = {
214
- // cfFetch sets the `cf` object that a function could expect
215
- // If there are no functions, there's no reason to set this up (and not make that network call)
216
- cfFetch: false,
217
- // TODO: The fact that these request/response hacks are necessary is ridiculous.
218
- // We need to eliminate them from env.ASSETS.fetch (not sure if just local or prod as well)
219
- script: `
220
- export default {
221
- async fetch(request, env, context) {
222
- const response = await env.ASSETS.fetch(request.url, request)
223
- return new Response(response.body, response)
224
- }
225
- }`,
226
- };
215
+ scriptPath = resolve(__dirname, "../templates/pages-shim.ts");
227
216
  }
228
217
  }
229
218
 
230
- // Defer importing miniflare until we really need it
231
- const { Miniflare, Log, LogLevel } = await import("miniflare");
232
- const { Response, fetch } = await import("@miniflare/core");
233
-
234
- // Wait for esbuild to finish building before starting Miniflare.
235
- // This must be before the call to `new Miniflare`, as that will
236
- // asynchronously start loading the script. `await startServer()`
237
- // internally just waits for that promise to resolve.
238
219
  await scriptReadyPromise;
239
220
 
240
- // `assetsFetch()` will only be called if there is `proxyPort` defined.
241
- // We only define `proxyPort`, above, when there is no `directory` defined.
242
- const assetsFetch =
243
- directory !== undefined
244
- ? await generateAssetsFetch(directory)
245
- : invalidAssetsFetch;
246
-
247
- const requestContextCheckOptions = await getRequestContextCheckOptions();
248
-
249
- const vars = getVarsForDev({
250
- configPath: resolvePath(".dev.vars"),
251
- } as Config);
252
-
253
- const miniflare = new Miniflare({
254
- port,
255
- watch: true,
256
- modules: true,
257
-
258
- log: new Log(LogLevel.ERROR, { prefix: "pages" }),
259
- logUnhandledRejections: true,
260
- sourceMap: true,
261
-
262
- kvNamespaces: kvs.map((kv) => kv.toString()),
263
-
264
- durableObjects: Object.fromEntries(
265
- durableObjects.map((durableObject) => durableObject.toString().split("="))
266
- ),
267
-
268
- // User bindings
269
- bindings: {
270
- ...vars,
271
- ...Object.fromEntries(
221
+ const { stop, waitUntilExit } = await unstable_dev(
222
+ scriptPath,
223
+ {
224
+ port,
225
+ watch: true,
226
+ localProtocol,
227
+ liveReload,
228
+
229
+ compatibilityDate: "2021-11-02",
230
+ nodeCompat,
231
+ vars: Object.fromEntries(
272
232
  bindings
273
233
  .map((binding) => binding.toString().split("="))
274
234
  .map(([key, ...values]) => [key, values.join("=")])
275
235
  ),
276
- },
236
+ kv: kvs.map((val) => ({
237
+ binding: val.toString(),
238
+ id: "",
239
+ })),
240
+ durableObjects: durableObjects.map((durableObject) => {
241
+ const [name, class_name] = durableObject.toString().split("=");
242
+ return {
243
+ name,
244
+ class_name,
245
+ };
246
+ }),
277
247
 
278
- // env.ASSETS.fetch
279
- serviceBindings: {
280
- async ASSETS(request: MiniflareRequest) {
281
- if (proxyPort) {
282
- try {
283
- const url = new URL(request.url);
284
- url.host = `localhost:${proxyPort}`;
285
- return await fetch(url, request);
286
- } catch (thrown) {
287
- logger.error(`Could not proxy request: ${thrown}`);
288
-
289
- // TODO: Pretty error page
290
- return new Response(
291
- `[wrangler] Could not proxy request: ${thrown}`,
292
- { status: 502 }
293
- );
294
- }
295
- } else {
296
- try {
297
- return await assetsFetch(request);
298
- } catch (thrown) {
299
- logger.error(`Could not serve static asset: ${thrown}`);
300
-
301
- // TODO: Pretty error page
302
- return new Response(
303
- `[wrangler] Could not serve static asset: ${thrown}`,
304
- { status: 502 }
305
- );
306
- }
307
- }
248
+ enablePagesAssetsServiceBinding: {
249
+ proxyPort,
250
+ directory,
308
251
  },
252
+ forceLocal: true,
253
+ experimentalEnableLocalPersistence,
254
+ showInteractiveDevSession: undefined,
255
+ inspect: true,
256
+ logLevel: "error",
257
+ logPrefix: "pages",
309
258
  },
259
+ true
260
+ );
261
+ await metrics.sendMetricsEvent("run pages dev");
310
262
 
311
- kvPersist: true,
312
- durableObjectsPersist: true,
313
- cachePersist: true,
314
- liveReload,
315
-
316
- ...requestContextCheckOptions,
317
- ...miniflareArgs,
263
+ waitUntilExit().then(() => {
264
+ CLEANUP();
265
+ stop();
266
+ process.exit(0);
318
267
  });
319
268
 
320
- try {
321
- // `startServer` might throw if user code contains errors
322
- const server = await miniflare.startServer();
323
- logger.log(`Serving at http://localhost:${port}/`);
324
- await metrics.sendMetricsEvent("run pages dev");
325
-
326
- if (process.env.BROWSER !== "none") {
327
- await openInBrowser(`http://localhost:${port}/`);
328
- }
329
-
330
- if (directory !== undefined && liveReload) {
331
- watch([directory], {
332
- persistent: true,
333
- ignoreInitial: true,
334
- }).on("all", async () => {
335
- await miniflare.reload();
336
- });
337
- }
338
-
339
- CLEANUP_CALLBACKS.push(() => {
340
- server.close();
341
- miniflare.dispose().catch((err) => miniflare.log.error(err));
342
- });
343
- } catch (e) {
344
- miniflare.log.error(e as Error);
269
+ process.on("exit", () => {
345
270
  CLEANUP();
346
- throw new FatalError("Could not start Miniflare.", 1);
347
- }
271
+ stop();
272
+ });
273
+ process.on("SIGINT", () => {
274
+ CLEANUP();
275
+ stop();
276
+ });
277
+ process.on("SIGTERM", () => {
278
+ CLEANUP();
279
+ stop();
280
+ });
348
281
  };
349
282
 
350
283
  function isWindows() {
@@ -352,7 +285,7 @@ function isWindows() {
352
285
  }
353
286
 
354
287
  async function sleep(ms: number) {
355
- await new Promise((resolve) => setTimeout(resolve, ms));
288
+ await new Promise((promiseResolve) => setTimeout(promiseResolve, ms));
356
289
  }
357
290
 
358
291
  function getPids(pid: number) {
@@ -415,7 +348,7 @@ async function spawnProxyProcess({
415
348
  }: {
416
349
  port?: number;
417
350
  command: (string | number)[];
418
- }): Promise<void | number> {
351
+ }): Promise<undefined | number> {
419
352
  if (command.length === 0) {
420
353
  CLEANUP();
421
354
  throw new FatalError(
@@ -479,490 +412,3 @@ async function spawnProxyProcess({
479
412
 
480
413
  return port;
481
414
  }
482
-
483
- function escapeRegex(str: string) {
484
- return str.replace(/[-/\\^$*+?.()|[]{}]/g, "\\$&");
485
- }
486
-
487
- type Replacements = Record<string, string>;
488
-
489
- function replacer(str: string, replacements: Replacements) {
490
- for (const [replacement, value] of Object.entries(replacements)) {
491
- str = str.replace(`:${replacement}`, value);
492
- }
493
- return str;
494
- }
495
-
496
- function generateRulesMatcher<T>(
497
- rules?: Record<string, T>,
498
- replacerFn: (match: T, replacements: Replacements) => T = (match) => match
499
- ) {
500
- // TODO: How can you test cross-host rules?
501
- if (!rules) return () => [];
502
-
503
- const compiledRules = Object.entries(rules)
504
- .map(([rule, match]) => {
505
- const crossHost = rule.startsWith("https://");
506
-
507
- rule = rule.split("*").map(escapeRegex).join("(?<splat>.*)");
508
-
509
- const host_matches = rule.matchAll(
510
- /(?<=^https:\\\/\\\/[^/]*?):([^\\]+)(?=\\)/g
511
- );
512
- for (const hostMatch of host_matches) {
513
- rule = rule.split(hostMatch[0]).join(`(?<${hostMatch[1]}>[^/.]+)`);
514
- }
515
-
516
- const path_matches = rule.matchAll(/:(\w+)/g);
517
- for (const pathMatch of path_matches) {
518
- rule = rule.split(pathMatch[0]).join(`(?<${pathMatch[1]}>[^/]+)`);
519
- }
520
-
521
- rule = "^" + rule + "$";
522
-
523
- try {
524
- const regExp = new RegExp(rule);
525
- return [{ crossHost, regExp }, match];
526
- } catch {}
527
- })
528
- .filter((value) => value !== undefined) as [
529
- { crossHost: boolean; regExp: RegExp },
530
- T
531
- ][];
532
-
533
- return ({ request }: { request: MiniflareRequest }) => {
534
- const { pathname, host } = new URL(request.url);
535
-
536
- return compiledRules
537
- .map(([{ crossHost, regExp }, match]) => {
538
- const test = crossHost ? `https://${host}${pathname}` : pathname;
539
- const result = regExp.exec(test);
540
- if (result) {
541
- return replacerFn(match, result.groups || {});
542
- }
543
- })
544
- .filter((value) => value !== undefined) as T[];
545
- };
546
- }
547
-
548
- function generateHeadersMatcher(headersFile: string) {
549
- if (existsSync(headersFile)) {
550
- const contents = readFileSync(headersFile).toString();
551
-
552
- // TODO: Log errors
553
- const lines = contents
554
- .split("\n")
555
- .map((line) => line.trim())
556
- .filter((line) => !line.startsWith("#") && line !== "");
557
-
558
- const rules: Record<string, Record<string, string>> = {};
559
- let rule: { path: string; headers: Record<string, string> } | undefined =
560
- undefined;
561
-
562
- for (const line of lines) {
563
- if (/^([^\s]+:\/\/|^\/)/.test(line)) {
564
- if (rule && Object.keys(rule.headers).length > 0) {
565
- rules[rule.path] = rule.headers;
566
- }
567
-
568
- const path = validateURL(line);
569
- if (path) {
570
- rule = {
571
- path,
572
- headers: {},
573
- };
574
- continue;
575
- }
576
- }
577
-
578
- if (!line.includes(":")) continue;
579
-
580
- const [rawName, ...rawValue] = line.split(":");
581
- const name = rawName.trim().toLowerCase();
582
- const value = rawValue.join(":").trim();
583
-
584
- if (name === "") continue;
585
- if (!rule) continue;
586
-
587
- const existingValues = rule.headers[name];
588
- rule.headers[name] = existingValues
589
- ? `${existingValues}, ${value}`
590
- : value;
591
- }
592
-
593
- if (rule && Object.keys(rule.headers).length > 0) {
594
- rules[rule.path] = rule.headers;
595
- }
596
-
597
- const rulesMatcher = generateRulesMatcher(rules, (match, replacements) =>
598
- Object.fromEntries(
599
- Object.entries(match).map(([name, value]) => [
600
- name,
601
- replacer(value, replacements),
602
- ])
603
- )
604
- );
605
-
606
- return (request: MiniflareRequest) => {
607
- const matches = rulesMatcher({
608
- request,
609
- });
610
- if (matches) return matches;
611
- };
612
- } else {
613
- return () => undefined;
614
- }
615
- }
616
-
617
- function generateRedirectsMatcher(redirectsFile: string) {
618
- if (existsSync(redirectsFile)) {
619
- const contents = readFileSync(redirectsFile).toString();
620
-
621
- // TODO: Log errors
622
- const lines = contents
623
- .split("\n")
624
- .map((line) => line.trim())
625
- .filter((line) => !line.startsWith("#") && line !== "");
626
-
627
- const rules = Object.fromEntries(
628
- lines
629
- .map((line) => line.split(" "))
630
- .filter((tokens) => tokens.length === 2 || tokens.length === 3)
631
- .map((tokens) => {
632
- const from = validateURL(tokens[0], true, false, false);
633
- const to = validateURL(tokens[1], false, true, true);
634
- let status: number | undefined = parseInt(tokens[2]) || 302;
635
- status = [301, 302, 303, 307, 308].includes(status)
636
- ? status
637
- : undefined;
638
-
639
- return from && to && status ? [from, { to, status }] : undefined;
640
- })
641
- .filter((rule) => rule !== undefined) as [
642
- string,
643
- { to: string; status?: number }
644
- ][]
645
- );
646
-
647
- const rulesMatcher = generateRulesMatcher(
648
- rules,
649
- ({ status, to }, replacements) => ({
650
- status,
651
- to: replacer(to, replacements),
652
- })
653
- );
654
-
655
- return (request: MiniflareRequest) => {
656
- const match = rulesMatcher({
657
- request,
658
- })[0];
659
- if (match) return match;
660
- };
661
- } else {
662
- return () => undefined;
663
- }
664
- }
665
-
666
- function extractPathname(
667
- path = "/",
668
- includeSearch: boolean,
669
- includeHash: boolean
670
- ) {
671
- if (!path.startsWith("/")) path = `/${path}`;
672
- const url = new URL(`//${path}`, "relative://");
673
- return `${url.pathname}${includeSearch ? url.search : ""}${
674
- includeHash ? url.hash : ""
675
- }`;
676
- }
677
-
678
- function validateURL(
679
- token: string,
680
- onlyRelative = false,
681
- includeSearch = false,
682
- includeHash = false
683
- ) {
684
- const host = /^https:\/\/+(?<host>[^/]+)\/?(?<path>.*)/.exec(token);
685
- if (host && host.groups && host.groups.host) {
686
- if (onlyRelative) return;
687
-
688
- return `https://${host.groups.host}${extractPathname(
689
- host.groups.path,
690
- includeSearch,
691
- includeHash
692
- )}`;
693
- } else {
694
- if (!token.startsWith("/") && onlyRelative) token = `/${token}`;
695
-
696
- const path = /^\//.exec(token);
697
- if (path) {
698
- try {
699
- return extractPathname(token, includeSearch, includeHash);
700
- } catch {}
701
- }
702
- }
703
- return "";
704
- }
705
-
706
- function hasFileExtension(pathname: string) {
707
- return /\/.+\.[a-z0-9]+$/i.test(pathname);
708
- }
709
-
710
- async function generateAssetsFetch(
711
- directory: string
712
- ): Promise<typeof miniflareFetch> {
713
- // Defer importing miniflare until we really need it
714
- const { Headers, Request, Response } = await import("@miniflare/core");
715
-
716
- const headersFile = join(directory, "_headers");
717
- const redirectsFile = join(directory, "_redirects");
718
- const workerFile = join(directory, "_worker.js");
719
-
720
- const ignoredFiles = [headersFile, redirectsFile, workerFile];
721
-
722
- const assetExists = (path: string) => {
723
- path = join(directory, path);
724
- return (
725
- existsSync(path) &&
726
- lstatSync(path).isFile() &&
727
- !ignoredFiles.includes(path)
728
- );
729
- };
730
-
731
- const getAsset = (path: string) => {
732
- if (assetExists(path)) {
733
- return join(directory, path);
734
- }
735
- };
736
-
737
- let redirectsMatcher = generateRedirectsMatcher(redirectsFile);
738
- let headersMatcher = generateHeadersMatcher(headersFile);
739
-
740
- watch([headersFile, redirectsFile], {
741
- persistent: true,
742
- }).on("change", (path) => {
743
- switch (path) {
744
- case headersFile: {
745
- logger.log("_headers modified. Re-evaluating...");
746
- headersMatcher = generateHeadersMatcher(headersFile);
747
- break;
748
- }
749
- case redirectsFile: {
750
- logger.log("_redirects modified. Re-evaluating...");
751
- redirectsMatcher = generateRedirectsMatcher(redirectsFile);
752
- break;
753
- }
754
- }
755
- });
756
-
757
- const serveAsset = (file: string) => {
758
- return readFileSync(file);
759
- };
760
-
761
- const generateResponse = (request: MiniflareRequest) => {
762
- const url = new URL(request.url);
763
-
764
- const deconstructedResponse: {
765
- status: number;
766
- headers: MiniflareHeaders;
767
- body?: Buffer;
768
- } = {
769
- status: 200,
770
- headers: new Headers(),
771
- body: undefined,
772
- };
773
-
774
- const match = redirectsMatcher(request);
775
- if (match) {
776
- const { status, to } = match;
777
-
778
- let location = to;
779
- let search;
780
-
781
- if (to.startsWith("/")) {
782
- search = new URL(location, "http://fakehost").search;
783
- } else {
784
- search = new URL(location).search;
785
- }
786
-
787
- location = `${location}${search ? "" : url.search}`;
788
-
789
- if (status && [301, 302, 303, 307, 308].includes(status)) {
790
- deconstructedResponse.status = status;
791
- } else {
792
- deconstructedResponse.status = 302;
793
- }
794
-
795
- deconstructedResponse.headers.set("Location", location);
796
- return deconstructedResponse;
797
- }
798
-
799
- if (!request.method?.match(/^(get|head)$/i)) {
800
- deconstructedResponse.status = 405;
801
- return deconstructedResponse;
802
- }
803
-
804
- const notFound = () => {
805
- let cwd = url.pathname;
806
- while (cwd) {
807
- cwd = cwd.slice(0, cwd.lastIndexOf("/"));
808
-
809
- if ((asset = getAsset(`${cwd}/404.html`))) {
810
- deconstructedResponse.status = 404;
811
- deconstructedResponse.body = serveAsset(asset);
812
- deconstructedResponse.headers.set(
813
- "Content-Type",
814
- getType(asset) || "application/octet-stream"
815
- );
816
- return deconstructedResponse;
817
- }
818
- }
819
-
820
- if ((asset = getAsset(`/index.html`))) {
821
- deconstructedResponse.body = serveAsset(asset);
822
- deconstructedResponse.headers.set(
823
- "Content-Type",
824
- getType(asset) || "application/octet-stream"
825
- );
826
- return deconstructedResponse;
827
- }
828
-
829
- deconstructedResponse.status = 404;
830
- return deconstructedResponse;
831
- };
832
-
833
- let asset;
834
-
835
- if (url.pathname.endsWith("/")) {
836
- if ((asset = getAsset(`${url.pathname}/index.html`))) {
837
- deconstructedResponse.body = serveAsset(asset);
838
- deconstructedResponse.headers.set(
839
- "Content-Type",
840
- getType(asset) || "application/octet-stream"
841
- );
842
- return deconstructedResponse;
843
- } else if (
844
- (asset = getAsset(`${url.pathname.replace(/\/$/, ".html")}`))
845
- ) {
846
- deconstructedResponse.status = 301;
847
- deconstructedResponse.headers.set(
848
- "Location",
849
- `${url.pathname.slice(0, -1)}${url.search}`
850
- );
851
- return deconstructedResponse;
852
- }
853
- }
854
-
855
- if (url.pathname.endsWith("/index")) {
856
- deconstructedResponse.status = 301;
857
- deconstructedResponse.headers.set(
858
- "Location",
859
- `${url.pathname.slice(0, -"index".length)}${url.search}`
860
- );
861
- return deconstructedResponse;
862
- }
863
-
864
- if ((asset = getAsset(url.pathname))) {
865
- if (url.pathname.endsWith(".html")) {
866
- const extensionlessPath = url.pathname.slice(0, -".html".length);
867
- if (getAsset(extensionlessPath) || extensionlessPath === "/") {
868
- deconstructedResponse.body = serveAsset(asset);
869
- deconstructedResponse.headers.set(
870
- "Content-Type",
871
- getType(asset) || "application/octet-stream"
872
- );
873
- return deconstructedResponse;
874
- } else {
875
- deconstructedResponse.status = 301;
876
- deconstructedResponse.headers.set(
877
- "Location",
878
- `${extensionlessPath}${url.search}`
879
- );
880
- return deconstructedResponse;
881
- }
882
- } else {
883
- deconstructedResponse.body = serveAsset(asset);
884
- deconstructedResponse.headers.set(
885
- "Content-Type",
886
- getType(asset) || "application/octet-stream"
887
- );
888
- return deconstructedResponse;
889
- }
890
- } else if (hasFileExtension(url.pathname)) {
891
- notFound();
892
- return deconstructedResponse;
893
- }
894
-
895
- if ((asset = getAsset(`${url.pathname}.html`))) {
896
- deconstructedResponse.body = serveAsset(asset);
897
- deconstructedResponse.headers.set(
898
- "Content-Type",
899
- getType(asset) || "application/octet-stream"
900
- );
901
- return deconstructedResponse;
902
- }
903
-
904
- if ((asset = getAsset(`${url.pathname}/index.html`))) {
905
- deconstructedResponse.status = 301;
906
- deconstructedResponse.headers.set(
907
- "Location",
908
- `${url.pathname}/${url.search}`
909
- );
910
- return deconstructedResponse;
911
- } else {
912
- notFound();
913
- return deconstructedResponse;
914
- }
915
- };
916
-
917
- const attachHeaders = (
918
- request: MiniflareRequest,
919
- deconstructedResponse: {
920
- status: number;
921
- headers: MiniflareHeaders;
922
- body?: Buffer;
923
- }
924
- ) => {
925
- const headers = deconstructedResponse.headers;
926
- const newHeaders = new Headers({});
927
- const matches = headersMatcher(request) || [];
928
-
929
- matches.forEach((match) => {
930
- Object.entries(match).forEach(([name, value]) => {
931
- newHeaders.append(name, `${value}`);
932
- });
933
- });
934
-
935
- const combinedHeaders = {
936
- ...Object.fromEntries(headers.entries()),
937
- ...Object.fromEntries(newHeaders.entries()),
938
- };
939
-
940
- deconstructedResponse.headers = new Headers({});
941
- Object.entries(combinedHeaders).forEach(([name, value]) => {
942
- if (value) deconstructedResponse.headers.set(name, value);
943
- });
944
- };
945
-
946
- return async (input, init) => {
947
- const request = new Request(input, init);
948
- const deconstructedResponse = generateResponse(request);
949
- attachHeaders(request, deconstructedResponse);
950
-
951
- const headers = new Headers();
952
-
953
- [...deconstructedResponse.headers.entries()].forEach(([name, value]) => {
954
- if (value) headers.set(name, value);
955
- });
956
-
957
- return new Response(deconstructedResponse.body, {
958
- headers,
959
- status: deconstructedResponse.status,
960
- });
961
- };
962
- }
963
-
964
- const invalidAssetsFetch: typeof miniflareFetch = () => {
965
- throw new Error(
966
- "Trying to fetch assets directly when there is no `directory` option specified, and not in `local` mode."
967
- );
968
- };