wrangler 0.0.0-e41c7e41 → 0.0.0-e44afaf79

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 (275) hide show
  1. package/README.md +35 -45
  2. package/bin/wrangler.js +27 -102
  3. package/config-schema.json +2487 -0
  4. package/package.json +179 -192
  5. package/templates/__tests__/tsconfig-sanity.ts +12 -0
  6. package/templates/__tests__/tsconfig.json +8 -0
  7. package/templates/checked-fetch.js +19 -6
  8. package/templates/facade.d.ts +19 -0
  9. package/templates/gitignore +2 -3
  10. package/templates/init-tests/test-jest-new-worker.js +3 -5
  11. package/templates/init-tests/test-vitest-new-worker.js +4 -6
  12. package/templates/init-tests/test-vitest-new-worker.ts +25 -0
  13. package/templates/middleware/common.ts +9 -4
  14. package/templates/middleware/loader-modules.ts +94 -44
  15. package/templates/middleware/loader-sw.ts +6 -4
  16. package/templates/middleware/middleware-ensure-req-body-drained.ts +18 -0
  17. package/templates/middleware/middleware-miniflare3-json-error.ts +17 -5
  18. package/templates/middleware/middleware-patch-console-prefix.d.ts +3 -0
  19. package/templates/middleware/middleware-patch-console-prefix.ts +21 -0
  20. package/templates/middleware/middleware-scheduled.ts +15 -1
  21. package/templates/mixedMode/proxyServerWorker/index.ts +40 -0
  22. package/templates/mixedMode/proxyServerWorker/wrangler.jsonc +4 -0
  23. package/templates/modules-watch-stub.js +4 -0
  24. package/templates/new-worker-scheduled.ts +1 -1
  25. package/templates/new-worker.ts +4 -1
  26. package/templates/pages-dev-pipeline.ts +8 -3
  27. package/templates/pages-dev-util.ts +4 -1
  28. package/templates/pages-shim.ts +1 -1
  29. package/templates/pages-template-plugin.ts +33 -14
  30. package/templates/pages-template-worker.ts +12 -3
  31. package/templates/startDevWorker/InspectorProxyWorker.ts +679 -0
  32. package/templates/startDevWorker/ProxyWorker.ts +336 -0
  33. package/templates/strip-cf-connecting-ip-header.js +13 -0
  34. package/templates/tsconfig-sanity.ts +11 -0
  35. package/templates/tsconfig.init.json +22 -0
  36. package/templates/tsconfig.json +12 -103
  37. package/templates/tsconfig.tsbuildinfo +1 -0
  38. package/wrangler-dist/InspectorProxyWorker.js +471 -0
  39. package/wrangler-dist/InspectorProxyWorker.js.map +6 -0
  40. package/wrangler-dist/ProxyWorker.js +238 -0
  41. package/wrangler-dist/ProxyWorker.js.map +6 -0
  42. package/wrangler-dist/cli.d.ts +2054 -512
  43. package/wrangler-dist/cli.js +172903 -159717
  44. package/wrangler-dist/cli.js.map +1 -0
  45. package/wrangler-dist/metafile-cjs.json +1 -0
  46. package/Cloudflare_CA.pem +0 -18
  47. package/import_meta_url.js +0 -3
  48. package/miniflare-config-stubs/.env.empty +0 -0
  49. package/miniflare-config-stubs/package.empty.json +0 -1
  50. package/miniflare-config-stubs/wrangler.empty.toml +0 -0
  51. package/miniflare-dist/index.mjs +0 -6316
  52. package/src/__tests__/access.test.ts +0 -25
  53. package/src/__tests__/api-dev.test.ts +0 -209
  54. package/src/__tests__/api-devregistry.test.js +0 -56
  55. package/src/__tests__/api.test.ts +0 -80
  56. package/src/__tests__/config-cache-without-cache-dir.test.ts +0 -38
  57. package/src/__tests__/config-cache.test.ts +0 -42
  58. package/src/__tests__/configuration.test.ts +0 -3981
  59. package/src/__tests__/d1.test.ts +0 -80
  60. package/src/__tests__/delete.test.ts +0 -230
  61. package/src/__tests__/deployments.test.ts +0 -91
  62. package/src/__tests__/dev.test.tsx +0 -1536
  63. package/src/__tests__/dialogs.test.tsx +0 -40
  64. package/src/__tests__/generate.test.ts +0 -235
  65. package/src/__tests__/get-host-from-url.test.ts +0 -16
  66. package/src/__tests__/guess-worker-format.test.ts +0 -85
  67. package/src/__tests__/helpers/clipboardy-mock.js +0 -4
  68. package/src/__tests__/helpers/cmd-shim.d.ts +0 -11
  69. package/src/__tests__/helpers/mock-account-id.ts +0 -48
  70. package/src/__tests__/helpers/mock-bin.ts +0 -36
  71. package/src/__tests__/helpers/mock-cfetch.ts +0 -333
  72. package/src/__tests__/helpers/mock-console.ts +0 -106
  73. package/src/__tests__/helpers/mock-dialogs.ts +0 -153
  74. package/src/__tests__/helpers/mock-get-zone-from-host.ts +0 -8
  75. package/src/__tests__/helpers/mock-http-server.ts +0 -46
  76. package/src/__tests__/helpers/mock-istty.ts +0 -74
  77. package/src/__tests__/helpers/mock-known-routes.ts +0 -7
  78. package/src/__tests__/helpers/mock-kv.ts +0 -33
  79. package/src/__tests__/helpers/mock-oauth-flow.ts +0 -252
  80. package/src/__tests__/helpers/mock-process.ts +0 -39
  81. package/src/__tests__/helpers/mock-stdin.ts +0 -108
  82. package/src/__tests__/helpers/mock-web-socket.ts +0 -29
  83. package/src/__tests__/helpers/msw/handlers/access.ts +0 -13
  84. package/src/__tests__/helpers/msw/handlers/deployments.ts +0 -59
  85. package/src/__tests__/helpers/msw/handlers/namespaces.ts +0 -104
  86. package/src/__tests__/helpers/msw/handlers/oauth.ts +0 -36
  87. package/src/__tests__/helpers/msw/handlers/r2.ts +0 -80
  88. package/src/__tests__/helpers/msw/handlers/script.ts +0 -75
  89. package/src/__tests__/helpers/msw/handlers/user.ts +0 -63
  90. package/src/__tests__/helpers/msw/handlers/zones.ts +0 -22
  91. package/src/__tests__/helpers/msw/index.ts +0 -21
  92. package/src/__tests__/helpers/run-in-tmp.ts +0 -38
  93. package/src/__tests__/helpers/run-wrangler.ts +0 -16
  94. package/src/__tests__/helpers/worker-scripts/child-wrangler.toml +0 -1
  95. package/src/__tests__/helpers/worker-scripts/hello-world-worker.js +0 -5
  96. package/src/__tests__/helpers/worker-scripts/hello-world-wrangler.toml +0 -1
  97. package/src/__tests__/helpers/worker-scripts/parent-worker.js +0 -8
  98. package/src/__tests__/helpers/worker-scripts/parent-wrangler.toml +0 -5
  99. package/src/__tests__/helpers/write-worker-source.ts +0 -31
  100. package/src/__tests__/helpers/write-wrangler-toml.ts +0 -17
  101. package/src/__tests__/https-options.test.ts +0 -163
  102. package/src/__tests__/index.test.ts +0 -264
  103. package/src/__tests__/init.test.ts +0 -2900
  104. package/src/__tests__/jest.setup.ts +0 -202
  105. package/src/__tests__/kv.test.ts +0 -1675
  106. package/src/__tests__/logger.test.ts +0 -141
  107. package/src/__tests__/metrics.test.ts +0 -450
  108. package/src/__tests__/middleware.scheduled.test.ts +0 -135
  109. package/src/__tests__/middleware.test.ts +0 -726
  110. package/src/__tests__/package-manager.test.ts +0 -353
  111. package/src/__tests__/pages-deployment-tail.test.ts +0 -820
  112. package/src/__tests__/pages.test.ts +0 -2442
  113. package/src/__tests__/parse.test.ts +0 -330
  114. package/src/__tests__/paths.test.ts +0 -39
  115. package/src/__tests__/publish.test.ts +0 -7633
  116. package/src/__tests__/pubsub.test.ts +0 -370
  117. package/src/__tests__/queues.test.ts +0 -371
  118. package/src/__tests__/r2.test.ts +0 -393
  119. package/src/__tests__/route.test.ts +0 -45
  120. package/src/__tests__/secret.test.ts +0 -644
  121. package/src/__tests__/tail.test.ts +0 -800
  122. package/src/__tests__/test-old-node-version.js +0 -31
  123. package/src/__tests__/type-generation.test.ts +0 -226
  124. package/src/__tests__/user.test.ts +0 -155
  125. package/src/__tests__/utils-collectKeyValues.test.ts +0 -47
  126. package/src/__tests__/validate-dev-props.test.ts +0 -56
  127. package/src/__tests__/version.test.ts +0 -35
  128. package/src/__tests__/whoami.test.tsx +0 -278
  129. package/src/__tests__/worker-namespace.test.ts +0 -326
  130. package/src/abort.d.ts +0 -3
  131. package/src/api/dev.ts +0 -210
  132. package/src/api/index.ts +0 -2
  133. package/src/bundle-reporter.tsx +0 -68
  134. package/src/bundle.ts +0 -823
  135. package/src/cfetch/index.ts +0 -160
  136. package/src/cfetch/internal.ts +0 -263
  137. package/src/cli.ts +0 -28
  138. package/src/config/README.md +0 -107
  139. package/src/config/config.ts +0 -275
  140. package/src/config/diagnostics.ts +0 -80
  141. package/src/config/environment.ts +0 -561
  142. package/src/config/index.ts +0 -334
  143. package/src/config/validation-helpers.ts +0 -597
  144. package/src/config/validation.ts +0 -2103
  145. package/src/config-cache.ts +0 -85
  146. package/src/create-worker-preview.ts +0 -282
  147. package/src/create-worker-upload-form.ts +0 -316
  148. package/src/d1/backups.tsx +0 -218
  149. package/src/d1/constants.ts +0 -2
  150. package/src/d1/create.tsx +0 -63
  151. package/src/d1/delete.tsx +0 -56
  152. package/src/d1/execute.tsx +0 -352
  153. package/src/d1/formatTimeAgo.ts +0 -14
  154. package/src/d1/index.ts +0 -99
  155. package/src/d1/list.tsx +0 -48
  156. package/src/d1/migrations.tsx +0 -446
  157. package/src/d1/options.ts +0 -22
  158. package/src/d1/types.tsx +0 -24
  159. package/src/d1/utils.ts +0 -48
  160. package/src/delete.ts +0 -108
  161. package/src/deployments.ts +0 -77
  162. package/src/deprecated/index.ts +0 -438
  163. package/src/dev/dev-vars.ts +0 -39
  164. package/src/dev/dev.tsx +0 -590
  165. package/src/dev/get-local-persistence-path.tsx +0 -31
  166. package/src/dev/local.tsx +0 -886
  167. package/src/dev/remote.tsx +0 -603
  168. package/src/dev/start-server.ts +0 -538
  169. package/src/dev/use-esbuild.ts +0 -204
  170. package/src/dev/validate-dev-props.ts +0 -31
  171. package/src/dev-registry.tsx +0 -201
  172. package/src/dev.tsx +0 -905
  173. package/src/dialogs.tsx +0 -168
  174. package/src/dispatch-namespace.ts +0 -207
  175. package/src/durable.ts +0 -102
  176. package/src/entry.ts +0 -329
  177. package/src/environment-variables.ts +0 -50
  178. package/src/errors.ts +0 -11
  179. package/src/git-client.ts +0 -124
  180. package/src/global-wrangler-config-path.ts +0 -26
  181. package/src/https-options.ts +0 -127
  182. package/src/index.tsx +0 -682
  183. package/src/init.ts +0 -970
  184. package/src/inspect.ts +0 -767
  185. package/src/intl-polyfill.d.ts +0 -139
  186. package/src/is-ci.ts +0 -14
  187. package/src/is-interactive.ts +0 -16
  188. package/src/kv/helpers.ts +0 -430
  189. package/src/kv/index.ts +0 -596
  190. package/src/logger.ts +0 -92
  191. package/src/metrics/index.ts +0 -5
  192. package/src/metrics/metrics-config.ts +0 -239
  193. package/src/metrics/metrics-dispatcher.ts +0 -96
  194. package/src/metrics/metrics-usage-headers.ts +0 -24
  195. package/src/metrics/send-event.ts +0 -96
  196. package/src/miniflare-cli/README.md +0 -30
  197. package/src/miniflare-cli/assets.ts +0 -215
  198. package/src/miniflare-cli/index.ts +0 -213
  199. package/src/miniflare-cli/request-context.ts +0 -40
  200. package/src/module-collection.ts +0 -267
  201. package/src/open-in-browser.ts +0 -17
  202. package/src/package-manager.ts +0 -219
  203. package/src/pages/build.tsx +0 -270
  204. package/src/pages/constants.ts +0 -18
  205. package/src/pages/deployment-tails.tsx +0 -284
  206. package/src/pages/deployments.tsx +0 -84
  207. package/src/pages/dev.tsx +0 -708
  208. package/src/pages/errors.ts +0 -66
  209. package/src/pages/functions/buildPlugin.ts +0 -107
  210. package/src/pages/functions/buildWorker.ts +0 -166
  211. package/src/pages/functions/filepath-routing.test.ts +0 -234
  212. package/src/pages/functions/filepath-routing.ts +0 -189
  213. package/src/pages/functions/identifiers.ts +0 -78
  214. package/src/pages/functions/routes-consolidation.test.ts +0 -250
  215. package/src/pages/functions/routes-consolidation.ts +0 -73
  216. package/src/pages/functions/routes-transformation.test.ts +0 -282
  217. package/src/pages/functions/routes-transformation.ts +0 -115
  218. package/src/pages/functions/routes-validation.test.ts +0 -403
  219. package/src/pages/functions/routes-validation.ts +0 -202
  220. package/src/pages/functions/routes.ts +0 -151
  221. package/src/pages/functions.tsx +0 -82
  222. package/src/pages/hash.tsx +0 -13
  223. package/src/pages/index.tsx +0 -102
  224. package/src/pages/projects.tsx +0 -169
  225. package/src/pages/prompt-select-project.tsx +0 -31
  226. package/src/pages/publish.tsx +0 -451
  227. package/src/pages/types.ts +0 -41
  228. package/src/pages/upload.tsx +0 -426
  229. package/src/pages/utils.ts +0 -23
  230. package/src/parse.ts +0 -286
  231. package/src/paths.ts +0 -71
  232. package/src/proxy.ts +0 -689
  233. package/src/publish/index.ts +0 -270
  234. package/src/publish/publish.ts +0 -1035
  235. package/src/pubsub/index.ts +0 -286
  236. package/src/pubsub/pubsub-commands.tsx +0 -624
  237. package/src/queues/cli/commands/consumer/add.ts +0 -71
  238. package/src/queues/cli/commands/consumer/index.ts +0 -22
  239. package/src/queues/cli/commands/consumer/remove.ts +0 -38
  240. package/src/queues/cli/commands/create.ts +0 -25
  241. package/src/queues/cli/commands/delete.ts +0 -26
  242. package/src/queues/cli/commands/index.ts +0 -33
  243. package/src/queues/cli/commands/list.ts +0 -25
  244. package/src/queues/client.ts +0 -135
  245. package/src/r2/constants.ts +0 -4
  246. package/src/r2/helpers.ts +0 -131
  247. package/src/r2/index.ts +0 -289
  248. package/src/routes.ts +0 -140
  249. package/src/secret/index.ts +0 -378
  250. package/src/selfsigned.d.ts +0 -29
  251. package/src/sites.tsx +0 -372
  252. package/src/tail/createTail.ts +0 -379
  253. package/src/tail/filters.ts +0 -277
  254. package/src/tail/index.ts +0 -214
  255. package/src/tail/printing.ts +0 -107
  256. package/src/type-generation.ts +0 -173
  257. package/src/update-check.ts +0 -19
  258. package/src/user/access.ts +0 -69
  259. package/src/user/choose-account.tsx +0 -69
  260. package/src/user/env-vars.ts +0 -46
  261. package/src/user/generate-auth-url.ts +0 -33
  262. package/src/user/generate-random-state.ts +0 -16
  263. package/src/user/index.ts +0 -3
  264. package/src/user/user.tsx +0 -1192
  265. package/src/utils/collectKeyValues.ts +0 -14
  266. package/src/whoami.tsx +0 -169
  267. package/src/worker.ts +0 -245
  268. package/src/yargs-types.ts +0 -24
  269. package/src/zones.ts +0 -191
  270. package/templates/d1-beta-facade.js +0 -205
  271. package/templates/first-party-worker-module-facade.ts +0 -18
  272. package/templates/format-dev-errors.ts +0 -32
  273. package/templates/serve-static-assets.ts +0 -63
  274. package/templates/service-bindings-module-facade.js +0 -54
  275. package/templates/service-bindings-sw-facade.js +0 -42
@@ -1,204 +0,0 @@
1
- import assert from "node:assert";
2
- import { watch } from "chokidar";
3
- import { useApp } from "ink";
4
- import { useState, useEffect } from "react";
5
- import { bundleWorker, rewriteNodeCompatBuildFailure } from "../bundle";
6
- import { logBuildFailure, logger } from "../logger";
7
- import type { Config } from "../config";
8
- import type { WorkerRegistry } from "../dev-registry";
9
- import type { Entry } from "../entry";
10
- import type { CfModule } from "../worker";
11
- import type { WatchMode, Metafile } from "esbuild";
12
-
13
- export type EsbuildBundle = {
14
- id: number;
15
- path: string;
16
- entry: Entry;
17
- type: "esm" | "commonjs";
18
- modules: CfModule[];
19
- dependencies: Metafile["outputs"][string]["inputs"];
20
- sourceMapPath: string | undefined;
21
- };
22
-
23
- export function useEsbuild({
24
- entry,
25
- destination,
26
- jsxFactory,
27
- jsxFragment,
28
- rules,
29
- assets,
30
- serveAssetsFromWorker,
31
- tsconfig,
32
- minify,
33
- nodeCompat,
34
- betaD1Shims,
35
- define,
36
- noBundle,
37
- workerDefinitions,
38
- services,
39
- durableObjects,
40
- firstPartyWorkerDevFacade,
41
- local,
42
- targetConsumer,
43
- testScheduled,
44
- experimentalLocal,
45
- }: {
46
- entry: Entry;
47
- destination: string | undefined;
48
- jsxFactory: string | undefined;
49
- jsxFragment: string | undefined;
50
- rules: Config["rules"];
51
- assets: Config["assets"];
52
- define: Config["define"];
53
- services: Config["services"];
54
- serveAssetsFromWorker: boolean;
55
- tsconfig: string | undefined;
56
- minify: boolean | undefined;
57
- nodeCompat: boolean | undefined;
58
- betaD1Shims?: string[];
59
- noBundle: boolean;
60
- workerDefinitions: WorkerRegistry;
61
- durableObjects: Config["durable_objects"];
62
- firstPartyWorkerDevFacade: boolean | undefined;
63
- local: boolean;
64
- targetConsumer: "dev" | "publish";
65
- testScheduled: boolean;
66
- experimentalLocal: boolean | undefined;
67
- }): EsbuildBundle | undefined {
68
- const [bundle, setBundle] = useState<EsbuildBundle>();
69
- const { exit } = useApp();
70
- useEffect(() => {
71
- let stopWatching: (() => void) | undefined = undefined;
72
-
73
- function updateBundle() {
74
- // nothing really changes here, so let's increment the id
75
- // to change the return object's identity
76
- setBundle((previousBundle) => {
77
- assert(
78
- previousBundle,
79
- "Rebuild triggered with no previous build available"
80
- );
81
- return { ...previousBundle, id: previousBundle.id + 1 };
82
- });
83
- }
84
-
85
- const watchMode: WatchMode = {
86
- async onRebuild(error) {
87
- if (error !== null) {
88
- if (!nodeCompat) rewriteNodeCompatBuildFailure(error);
89
- logBuildFailure(error);
90
- logger.error("Watch build failed:", error.message);
91
- } else {
92
- updateBundle();
93
- }
94
- },
95
- };
96
-
97
- async function build() {
98
- if (!destination) return;
99
-
100
- const {
101
- resolvedEntryPointPath,
102
- bundleType,
103
- modules,
104
- dependencies,
105
- stop,
106
- sourceMapPath,
107
- }: Awaited<ReturnType<typeof bundleWorker>> = noBundle
108
- ? {
109
- modules: [],
110
- dependencies: {},
111
- resolvedEntryPointPath: entry.file,
112
- bundleType: entry.format === "modules" ? "esm" : "commonjs",
113
- stop: undefined,
114
- sourceMapPath: undefined,
115
- }
116
- : await bundleWorker(entry, destination, {
117
- serveAssetsFromWorker,
118
- jsxFactory,
119
- jsxFragment,
120
- rules,
121
- watch: watchMode,
122
- tsconfig,
123
- minify,
124
- nodeCompat,
125
- betaD1Shims,
126
- define,
127
- checkFetch: true,
128
- assets: assets && {
129
- ...assets,
130
- // disable the cache in dev
131
- bypassCache: true,
132
- },
133
- workerDefinitions,
134
- services,
135
- firstPartyWorkerDevFacade,
136
- local,
137
- targetConsumer,
138
- testScheduled,
139
- experimentalLocal,
140
- });
141
-
142
- // Capture the `stop()` method to use as the `useEffect()` destructor.
143
- stopWatching = stop;
144
-
145
- // if "noBundle" is true, then we need to manually watch the entry point and
146
- // trigger "builds" when it changes
147
- if (noBundle) {
148
- const watcher = watch(entry.file, {
149
- persistent: true,
150
- }).on("change", async (_event) => {
151
- updateBundle();
152
- });
153
-
154
- stopWatching = () => {
155
- watcher.close();
156
- };
157
- }
158
- setBundle({
159
- id: 0,
160
- entry,
161
- path: resolvedEntryPointPath,
162
- type: bundleType,
163
- modules,
164
- dependencies,
165
- sourceMapPath,
166
- });
167
- }
168
-
169
- build().catch((err) => {
170
- // If esbuild fails on first run, we want to quit the process
171
- // since we can't recover from here
172
- // related: https://github.com/evanw/esbuild/issues/1037
173
- exit(err);
174
- });
175
-
176
- return () => {
177
- stopWatching?.();
178
- };
179
- }, [
180
- entry,
181
- destination,
182
- jsxFactory,
183
- jsxFragment,
184
- serveAssetsFromWorker,
185
- rules,
186
- tsconfig,
187
- exit,
188
- noBundle,
189
- minify,
190
- nodeCompat,
191
- define,
192
- assets,
193
- services,
194
- durableObjects,
195
- workerDefinitions,
196
- firstPartyWorkerDevFacade,
197
- betaD1Shims,
198
- local,
199
- targetConsumer,
200
- testScheduled,
201
- experimentalLocal,
202
- ]);
203
- return bundle;
204
- }
@@ -1,31 +0,0 @@
1
- import type { DevProps } from "./dev";
2
-
3
- export function validateDevProps(props: DevProps) {
4
- if (
5
- !props.isWorkersSite &&
6
- props.assetPaths &&
7
- props.entry.format === "service-worker"
8
- ) {
9
- throw new Error(
10
- "You cannot use the service-worker format with an `assets` directory yet. For information on how to migrate to the module-worker format, see: https://developers.cloudflare.com/workers/learning/migrating-to-module-workers/"
11
- );
12
- }
13
-
14
- if (props.bindings.wasm_modules && props.entry.format === "modules") {
15
- throw new Error(
16
- "You cannot configure [wasm_modules] with an ES module worker. Instead, import the .wasm module directly in your code"
17
- );
18
- }
19
-
20
- if (props.bindings.text_blobs && props.entry.format === "modules") {
21
- throw new Error(
22
- "You cannot configure [text_blobs] with an ES module worker. Instead, import the file directly in your code, and optionally configure `[rules]` in your wrangler.toml"
23
- );
24
- }
25
-
26
- if (props.bindings.data_blobs && props.entry.format === "modules") {
27
- throw new Error(
28
- "You cannot configure [data_blobs] with an ES module worker. Instead, import the file directly in your code, and optionally configure `[rules]` in your wrangler.toml"
29
- );
30
- }
31
- }
@@ -1,201 +0,0 @@
1
- import http from "http";
2
- import net from "net";
3
- import bodyParser from "body-parser";
4
- import express from "express";
5
- import { createHttpTerminator } from "http-terminator";
6
- import { fetch } from "undici";
7
- import { logger } from "./logger";
8
-
9
- import type { Config } from "./config";
10
- import type { Server } from "http";
11
- import type { HttpTerminator } from "http-terminator";
12
-
13
- const DEV_REGISTRY_PORT = "6284";
14
- const DEV_REGISTRY_HOST = `http://localhost:${DEV_REGISTRY_PORT}`;
15
-
16
- let server: Server;
17
- let terminator: HttpTerminator;
18
-
19
- export type WorkerRegistry = Record<string, WorkerDefinition>;
20
-
21
- type WorkerDefinition = {
22
- port: number | undefined;
23
- protocol: "http" | "https" | undefined;
24
- host: string | undefined;
25
- mode: "local" | "remote";
26
- headers?: Record<string, string>;
27
- durableObjects: { name: string; className: string }[];
28
- durableObjectsHost?: string;
29
- durableObjectsPort?: number;
30
- };
31
-
32
- /**
33
- * A helper function to check whether our service registry is already running
34
- */
35
- async function isPortAvailable() {
36
- return new Promise((resolve, reject) => {
37
- const netServer = net
38
- .createServer()
39
- .once("error", (err) => {
40
- netServer.close();
41
- if ((err as unknown as { code: string }).code === "EADDRINUSE") {
42
- resolve(false);
43
- } else {
44
- reject(err);
45
- }
46
- })
47
- .once("listening", () => {
48
- netServer.close();
49
- resolve(true);
50
- });
51
- netServer.listen(DEV_REGISTRY_PORT);
52
- });
53
- }
54
-
55
- const jsonBodyParser = bodyParser.json();
56
-
57
- /**
58
- * Start the service registry. It's a simple server
59
- * that exposes endpoints for registering and unregistering
60
- * services, as well as getting the state of the registry.
61
- */
62
- export async function startWorkerRegistry() {
63
- if ((await isPortAvailable()) && !server) {
64
- const app = express();
65
-
66
- let workers: WorkerRegistry = {};
67
- app
68
- .get("/workers", async (req, res) => {
69
- res.json(workers);
70
- })
71
- .post("/workers/:workerId", jsonBodyParser, async (req, res) => {
72
- workers[req.params.workerId] = req.body;
73
- res.json(null);
74
- })
75
- .delete(`/workers/:workerId`, async (req, res) => {
76
- delete workers[req.params.workerId];
77
- res.json(null);
78
- })
79
- .delete("/workers", async (req, res) => {
80
- workers = {};
81
- res.json(null);
82
- });
83
- server = http.createServer(app);
84
- terminator = createHttpTerminator({ server });
85
- server.listen(DEV_REGISTRY_PORT);
86
-
87
- /**
88
- * The registry server may have already been started by another wrangler process.
89
- * If wrangler processes are run in parallel, isPortAvailable() can return true
90
- * while another process spins up the server
91
- */
92
- server.once("error", (err) => {
93
- if ((err as unknown as { code: string }).code !== "EADDRINUSE") {
94
- throw err;
95
- }
96
- });
97
- }
98
- }
99
-
100
- /**
101
- * Stop the service registry.
102
- */
103
- export async function stopWorkerRegistry() {
104
- await terminator?.terminate();
105
- }
106
-
107
- /**
108
- * Register a worker in the registry.
109
- */
110
- export async function registerWorker(
111
- name: string,
112
- definition: WorkerDefinition
113
- ) {
114
- try {
115
- return await fetch(`${DEV_REGISTRY_HOST}/workers/${name}`, {
116
- method: "POST",
117
- headers: {
118
- "Content-Type": "application/json",
119
- },
120
- body: JSON.stringify(definition),
121
- });
122
- } catch (e) {
123
- if (
124
- !["ECONNRESET", "ECONNREFUSED"].includes(
125
- (e as unknown as { cause?: { code?: string } }).cause?.code || "___"
126
- )
127
- ) {
128
- logger.error("Failed to register worker in local service registry", e);
129
- } else {
130
- logger.debug("Failed to register worker in local service registry", e);
131
- }
132
- }
133
- }
134
-
135
- /**
136
- * Unregister a worker from the registry.
137
- */
138
- export async function unregisterWorker(name: string) {
139
- try {
140
- await fetch(`${DEV_REGISTRY_HOST}/workers/${name}`, {
141
- method: "DELETE",
142
- });
143
- } catch (e) {
144
- if (
145
- !["ECONNRESET", "ECONNREFUSED"].includes(
146
- (e as unknown as { cause?: { code?: string } }).cause?.code || "___"
147
- )
148
- ) {
149
- throw e;
150
- // logger.error("failed to unregister worker", e);
151
- }
152
- }
153
- }
154
-
155
- /**
156
- * Get the state of the service registry.
157
- */
158
- export async function getRegisteredWorkers(): Promise<
159
- WorkerRegistry | undefined
160
- > {
161
- try {
162
- const response = await fetch(`${DEV_REGISTRY_HOST}/workers`);
163
- return (await response.json()) as WorkerRegistry;
164
- } catch (e) {
165
- if (
166
- !["ECONNRESET", "ECONNREFUSED"].includes(
167
- (e as unknown as { cause?: { code?: string } }).cause?.code || "___"
168
- )
169
- ) {
170
- throw e;
171
- }
172
- }
173
- }
174
-
175
- /**
176
- * a function that takes your serviceNames and durableObjectNames and returns a
177
- * list of the running workers that we're bound to
178
- */
179
- export async function getBoundRegisteredWorkers({
180
- services,
181
- durableObjects,
182
- }: {
183
- services: Config["services"] | undefined;
184
- durableObjects: Config["durable_objects"] | undefined;
185
- }) {
186
- const serviceNames = (services || []).map(
187
- (serviceBinding) => serviceBinding.service
188
- );
189
- const durableObjectServices = (
190
- durableObjects || { bindings: [] }
191
- ).bindings.map((durableObjectBinding) => durableObjectBinding.script_name);
192
-
193
- const workerDefinitions = await getRegisteredWorkers();
194
- const filteredWorkers = Object.fromEntries(
195
- Object.entries(workerDefinitions || {}).filter(
196
- ([key, _value]) =>
197
- serviceNames.includes(key) || durableObjectServices.includes(key)
198
- )
199
- );
200
- return filteredWorkers;
201
- }