wrangler 2.0.8 → 2.0.12

package/src/index.tsx

@@ -4,6 +4,7 @@ import path from "node:path";
 import { setTimeout } from "node:timers/promises";
 import TOML from "@iarna/toml";
 import chalk from "chalk";
+import { watch } from "chokidar";
 import { execa } from "execa";
 import { findUp } from "find-up";
 import getPort from "get-port";
@@ -47,6 +48,7 @@ import {
   parsePackageJSON,
   parseTOML,
   readFileSync,
+  readFileSyncToBuffer,
 } from "./parse";
 import publish from "./publish";
 import { createR2Bucket, deleteR2Bucket, listR2Buckets } from "./r2";
@@ -66,6 +68,7 @@ import {
   requireAuth,
 } from "./user";
 import { whoami } from "./whoami";
+import { getZoneIdFromHost, getZoneForRoute } from "./zones";
 
 import type { Config } from "./config";
 import type { TailCLIFilters } from "./tail";
@@ -444,7 +447,10 @@ function createCLIParser(argv: string[]) {
           yesFlag ||
           (await confirm("Would you like to use git to manage this Worker?"));
         if (shouldInitGit) {
-          await execa("git", ["init", "--initial-branch=main"], {
+          await execa("git", ["init"], {
+            cwd: creationDirectory,
+          });
+          await execa("git", ["branch", "-m", "main"], {
             cwd: creationDirectory,
           });
           await writeFile(
@@ -1031,263 +1037,235 @@ function createCLIParser(argv: string[]) {
         });
     },
     async (args) => {
-      await printWranglerBanner();
-      const configPath =
-        (args.config as ConfigPath) ||
-        (args.script && findWranglerToml(path.dirname(args.script)));
-      const config = readConfig(configPath, args);
-      const entry = await getEntry(args, config, "dev");
-
-      if (config.services && config.services.length > 0) {
-        logger.warn(
-          `This worker is bound to live services: ${config.services
-            .map(
-              (service) =>
-                `${service.binding} (${service.service}${
-                  service.environment ? `@${service.environment}` : ""
-                })`
-            )
-            .join(", ")}`
-        );
-      }
-
-      if (args.inspect) {
-        logger.warn(
-          "Passing --inspect is unnecessary, now you can always connect to devtools."
-        );
-      }
-
-      if (args["experimental-public"]) {
-        logger.warn(
-          "The --experimental-public field is experimental and will change in the future."
-        );
-      }
-
-      if (args.public) {
-        throw new Error(
-          "The --public field has been renamed to --experimental-public, and will change behaviour in the future."
-        );
-      }
-
-      const upstreamProtocol =
-        args["upstream-protocol"] || config.dev.upstream_protocol;
-      if (upstreamProtocol === "http") {
-        logger.warn(
-          "Setting upstream-protocol to http is not currently implemented.\n" +
-            "If this is required in your project, please add your use case to the following issue:\n" +
-            "https://github.com/cloudflare/wrangler2/issues/583."
-        );
-      }
+      let watcher: ReturnType<typeof watch> | undefined;
+      try {
+        await printWranglerBanner();
+        const configPath =
+          (args.config as ConfigPath) ||
+          ((args.script &&
+            findWranglerToml(path.dirname(args.script))) as ConfigPath);
+        let config = readConfig(configPath, args);
+
+        if (config.configPath) {
+          watcher = watch(config.configPath, {
+            persistent: true,
+          }).on("change", async (_event) => {
+            // TODO: Do we need to handle different `_event` types differently?
+            //       e.g. what if the file is deleted, or added?
+            config = readConfig(configPath, args);
+            if (config.configPath) {
+              logger.log(`${path.basename(config.configPath)} changed...`);
+              rerender(await getDevReactElement(config));
+            }
+          });
+        }
 
-      const accountId = !args.local ? await requireAuth(config) : undefined;
+        const entry = await getEntry(args, config, "dev");
 
-      // TODO: if worker_dev = false and no routes, then error (only for dev)
+        if (config.services && config.services.length > 0) {
+          logger.warn(
+            `This worker is bound to live services: ${config.services
+              .map(
+                (service) =>
+                  `${service.binding} (${service.service}${
+                    service.environment ? `@${service.environment}` : ""
+                  })`
+              )
+              .join(", ")}`
+          );
+        }
 
-      /**
-       * Given something that resembles a URL,
-       * try to extract a host from it
-       */
-      function getHost(urlLike: string): string | undefined {
-        // strip leading * / *.
-        urlLike = urlLike.replace(/^\*(\.)?/g, "");
+        if (args.inspect) {
+          logger.warn(
+            "Passing --inspect is unnecessary, now you can always connect to devtools."
+          );
+        }
 
-        if (
-          !(urlLike.startsWith("http://") || urlLike.startsWith("https://"))
-        ) {
-          urlLike = "http://" + urlLike;
+        if (args["experimental-public"]) {
+          logger.warn(
+            "The --experimental-public field is experimental and will change in the future."
+          );
         }
-        return new URL(urlLike).host;
-      }
 
-      /**
-       * Given something that resembles a host,
-       * try to infer a zone id from it
-       */
-      async function getZoneId(host: string): Promise<string | undefined> {
-        const zones = await fetchResult<{ id: string }[]>(
-          `/zones`,
-          {},
-          new URLSearchParams({ name: host })
-        );
-        return zones[0]?.id;
-      }
+        if (args["experimental-public"] && (args.site || config.site)) {
+          throw new Error(
+            "Cannot use --experimental-public and a Site configuration together."
+          );
+        }
 
-      // When we're given a host (in one of the above ways), we do 2 things:
-      // - We try to extract a host from it
-      // - We try to get a zone id from the host
-      //
-      // So it turns out it's particularly hard to get a 'valid' domain
-      // from a string, so we don't even try to validate TLDs, etc.
-      // Once we get something that looks like w.x.y.z-ish, we then try to
-      // get a zone id for it, by lopping off subdomains until we get a hit
-      // from the API. That's it!
-
-      let zone: { host: string; id: string } | undefined;
-
-      if (!args.local) {
-        const hostLike =
-          args.host ||
-          config.dev.host ||
-          (args.routes && args.routes[0]) ||
-          config.route ||
-          (config.routes && config.routes[0]);
-
-        let zoneId: string | undefined =
-          typeof hostLike === "object" && "zone_id" in hostLike
-            ? hostLike.zone_id
-            : undefined;
-
-        const host =
-          typeof hostLike === "string"
-            ? getHost(hostLike)
-            : typeof hostLike === "object"
-            ? "zone_name" in hostLike
-              ? getHost(hostLike.zone_name)
-              : getHost(hostLike.pattern)
-            : undefined;
-
-        const hostPieces =
-          typeof host === "string" ? host.split(".") : undefined;
-
-        while (!zoneId && hostPieces && hostPieces.length > 1) {
-          zoneId = await getZoneId(hostPieces.join("."));
-          hostPieces.shift();
+        if (args.public) {
+          throw new Error(
+            "The --public field has been renamed to --experimental-public, and will change behaviour in the future."
+          );
         }
 
-        if (host && !zoneId) {
-          throw new Error(`Could not find zone for ${hostLike}`);
+        const upstreamProtocol =
+          args["upstream-protocol"] || config.dev.upstream_protocol;
+        if (upstreamProtocol === "http") {
+          logger.warn(
+            "Setting upstream-protocol to http is not currently implemented.\n" +
+              "If this is required in your project, please add your use case to the following issue:\n" +
+              "https://github.com/cloudflare/wrangler2/issues/583."
+          );
         }
 
-        zone =
-          typeof zoneId === "string" && typeof host === "string"
-            ? {
-                host,
-                id: zoneId,
-              }
-            : undefined;
-      }
+        // TODO: if worker_dev = false and no routes, then error (only for dev)
 
-      const nodeCompat = args.nodeCompat ?? config.node_compat;
-      if (nodeCompat) {
-        logger.warn(
-          "Enabling node.js compatibility mode for built-ins and globals. This is experimental and has serious tradeoffs. Please see https://github.com/ionic-team/rollup-plugin-node-polyfills/ for more details."
-        );
-      }
+        // Compute zone info from the `host` and `route` args and config;
+        let host = args.host || config.dev.host;
+        let zoneId: string | undefined;
 
-      const bindings = {
-        kv_namespaces: config.kv_namespaces?.map(
-          ({ binding, preview_id, id: _id }) => {
-            // In `dev`, we make folks use a separate kv namespace called
-            // `preview_id` instead of `id` so that they don't
-            // break production data. So here we check that a `preview_id`
-            // has actually been configured.
-            // This whole block of code will be obsoleted in the future
-            // when we have copy-on-write for previews on edge workers.
-            if (!preview_id) {
-              // TODO: This error has to be a _lot_ better, ideally just asking
-              // to create a preview namespace for the user automatically
-              throw new Error(
-                `In development, you should use a separate kv namespace than the one you'd use in production. Please create a new kv namespace with "wrangler kv:namespace create <name> --preview" and add its id as preview_id to the kv_namespace "${binding}" in your wrangler.toml`
-              ); // Ugh, I really don't like this message very much
-            }
-            return {
-              binding,
-              id: preview_id,
-            };
+        if (!args.local) {
+          if (host) {
+            zoneId = await getZoneIdFromHost(host);
           }
-        ),
-        // Use a copy of combinedVars since we're modifying it later
-        vars: getVarsForDev(config),
-        wasm_modules: config.wasm_modules,
-        text_blobs: config.text_blobs,
-        data_blobs: config.data_blobs,
-        durable_objects: config.durable_objects,
-        r2_buckets: config.r2_buckets?.map(
-          ({ binding, preview_bucket_name, bucket_name: _bucket_name }) => {
-            // same idea as kv namespace preview id,
-            // same copy-on-write TODO
-            if (!preview_bucket_name) {
-              throw new Error(
-                `In development, you should use a separate r2 bucket than the one you'd use in production. Please create a new r2 bucket with "wrangler r2 bucket create <name>" and add its name as preview_bucket_name to the r2_buckets "${binding}" in your wrangler.toml`
-              );
+          const routes = args.routes || config.route || config.routes;
+          if (!zoneId && routes) {
+            const firstRoute = Array.isArray(routes) ? routes[0] : routes;
+            const zone = await getZoneForRoute(firstRoute);
+            if (zone) {
+              zoneId = zone.id;
+              host = zone.host;
             }
-            return {
-              binding,
-              bucket_name: preview_bucket_name,
-            };
           }
-        ),
-        services: config.services,
-        unsafe: config.unsafe?.bindings,
-      };
+        }
 
-      // mask anything that was overridden in .dev.vars
-      // so that we don't log potential secrets into the terminal
-      const maskedVars = { ...bindings.vars };
-      for (const key of Object.keys(maskedVars)) {
-        if (maskedVars[key] !== config.vars[key]) {
-          // This means it was overridden in .dev.vars
-          // so let's mask it
-          maskedVars[key] = "(hidden)";
+        const nodeCompat = args.nodeCompat ?? config.node_compat;
+        if (nodeCompat) {
+          logger.warn(
+            "Enabling node.js compatibility mode for built-ins and globals. This is experimental and has serious tradeoffs. Please see https://github.com/ionic-team/rollup-plugin-node-polyfills/ for more details."
+          );
         }
-      }
 
-      // now log all available bindings into the terminal
-      printBindings({
-        ...bindings,
-        vars: maskedVars,
-      });
+        // eslint-disable-next-line no-inner-declarations
+        async function getBindings(configParam: Config) {
+          return {
+            kv_namespaces: configParam.kv_namespaces?.map(
+              ({ binding, preview_id, id: _id }) => {
+                // In `dev`, we make folks use a separate kv namespace called
+                // `preview_id` instead of `id` so that they don't
+                // break production data. So here we check that a `preview_id`
+                // has actually been configured.
+                // This whole block of code will be obsoleted in the future
+                // when we have copy-on-write for previews on edge workers.
+                if (!preview_id) {
+                  // TODO: This error has to be a _lot_ better, ideally just asking
+                  // to create a preview namespace for the user automatically
+                  throw new Error(
+                    `In development, you should use a separate kv namespace than the one you'd use in production. Please create a new kv namespace with "wrangler kv:namespace create <name> --preview" and add its id as preview_id to the kv_namespace "${binding}" in your wrangler.toml`
+                  ); // Ugh, I really don't like this message very much
+                }
+                return {
+                  binding,
+                  id: preview_id,
+                };
+              }
+            ),
+            // Use a copy of combinedVars since we're modifying it later
+            vars: getVarsForDev(configParam),
+            wasm_modules: configParam.wasm_modules,
+            text_blobs: configParam.text_blobs,
+            data_blobs: configParam.data_blobs,
+            durable_objects: configParam.durable_objects,
+            r2_buckets: configParam.r2_buckets?.map(
+              ({ binding, preview_bucket_name, bucket_name: _bucket_name }) => {
+                // same idea as kv namespace preview id,
+                // same copy-on-write TODO
+                if (!preview_bucket_name) {
+                  throw new Error(
+                    `In development, you should use a separate r2 bucket than the one you'd use in production. Please create a new r2 bucket with "wrangler r2 bucket create <name>" and add its name as preview_bucket_name to the r2_buckets "${binding}" in your wrangler.toml`
+                  );
+                }
+                return {
+                  binding,
+                  bucket_name: preview_bucket_name,
+                };
+              }
+            ),
+            services: configParam.services,
+            unsafe: configParam.unsafe?.bindings,
+          };
+        }
 
-      const { waitUntilExit } = render(
-        <Dev
-          name={getScriptName(args, config)}
-          entry={entry}
-          env={args.env}
-          zone={zone}
-          rules={getRules(config)}
-          legacyEnv={isLegacyEnv(config)}
-          minify={args.minify ?? config.minify}
-          nodeCompat={nodeCompat}
-          build={config.build || {}}
-          initialMode={args.local ? "local" : "remote"}
-          jsxFactory={args["jsx-factory"] || config.jsx_factory}
-          jsxFragment={args["jsx-fragment"] || config.jsx_fragment}
-          tsconfig={args.tsconfig ?? config.tsconfig}
-          upstreamProtocol={upstreamProtocol}
-          localProtocol={args["local-protocol"] || config.dev.local_protocol}
-          enableLocalPersistence={
-            args["experimental-enable-local-persistence"] || false
-          }
-          accountId={accountId}
-          assetPaths={getAssetPaths(
-            config,
-            args.site,
-            args.siteInclude,
-            args.siteExclude
-          )}
-          port={
-            args.port ||
-            config.dev.port ||
-            (await getPort({ port: DEFAULT_LOCAL_PORT }))
-          }
-          ip={args.ip || config.dev.ip}
-          inspectorPort={
-            args["inspector-port"] ?? (await getPort({ port: 9229 }))
-          }
-          public={args["experimental-public"]}
-          compatibilityDate={getDevCompatibilityDate(
-            config,
-            args["compatibility-date"]
-          )}
-          compatibilityFlags={
-            args["compatibility-flags"] || config.compatibility_flags
+        const getLocalPort = memoizeGetPort(DEFAULT_LOCAL_PORT);
+        const getInspectorPort = memoizeGetPort(9229);
+
+        // eslint-disable-next-line no-inner-declarations
+        async function getDevReactElement(configParam: Config) {
+          // now log all available bindings into the terminal
+          const bindings = await getBindings(configParam);
+          // mask anything that was overridden in .dev.vars
+          // so that we don't log potential secrets into the terminal
+          const maskedVars = { ...bindings.vars };
+          for (const key of Object.keys(maskedVars)) {
+            if (maskedVars[key] !== configParam.vars[key]) {
+              // This means it was overridden in .dev.vars
+              // so let's mask it
+              maskedVars[key] = "(hidden)";
+            }
           }
-          usageModel={config.usage_model}
-          bindings={bindings}
-          crons={config.triggers.crons}
-        />
-      );
-      await waitUntilExit();
+
+          printBindings({
+            ...bindings,
+            vars: maskedVars,
+          });
+
+          return (
+            <Dev
+              name={getScriptName(args, config)}
+              entry={entry}
+              env={args.env}
+              zone={zoneId}
+              host={host}
+              rules={getRules(config)}
+              legacyEnv={isLegacyEnv(config)}
+              minify={args.minify ?? config.minify}
+              nodeCompat={nodeCompat}
+              build={config.build || {}}
+              initialMode={args.local ? "local" : "remote"}
+              jsxFactory={args["jsx-factory"] || config.jsx_factory}
+              jsxFragment={args["jsx-fragment"] || config.jsx_fragment}
+              tsconfig={args.tsconfig ?? config.tsconfig}
+              upstreamProtocol={upstreamProtocol}
+              localProtocol={
+                args["local-protocol"] || config.dev.local_protocol
+              }
+              enableLocalPersistence={
+                args["experimental-enable-local-persistence"] || false
+              }
+              accountId={config.account_id}
+              assetPaths={getAssetPaths(
+                config,
+                args["experimental-public"] || args.site,
+                args.siteInclude,
+                args.siteExclude
+              )}
+              port={args.port || config.dev.port || (await getLocalPort())}
+              ip={args.ip || config.dev.ip}
+              inspectorPort={
+                args["inspector-port"] ?? (await getInspectorPort())
+              }
+              public={args["experimental-public"]}
+              compatibilityDate={getDevCompatibilityDate(
+                config,
+                args["compatibility-date"]
+              )}
+              compatibilityFlags={
+                args["compatibility-flags"] || config.compatibility_flags
+              }
+              usageModel={config.usage_model}
+              bindings={bindings}
+              crons={config.triggers.crons}
+            />
+          );
+        }
+        const { waitUntilExit, rerender } = render(
+          await getDevReactElement(config)
+        );
+        await waitUntilExit();
+      } finally {
+        await watcher?.close();
+      }
     }
   );
 
@@ -1413,23 +1391,29 @@ function createCLIParser(argv: string[]) {
     },
     async (args) => {
       await printWranglerBanner();
+
+      const configPath =
+        (args.config as ConfigPath) ||
+        (args.script && findWranglerToml(path.dirname(args.script)));
+      const config = readConfig(configPath, args);
+      const entry = await getEntry(args, config, "publish");
+
       if (args["experimental-public"]) {
         logger.warn(
           "The --experimental-public field is experimental and will change in the future."
         );
       }
+      if (args["experimental-public"] && (args.site || config.site)) {
+        throw new Error(
+          "Cannot use --experimental-public and a Site configuration together."
+        );
+      }
       if (args.public) {
         throw new Error(
           "The --public field has been renamed to --experimental-public, and will change behaviour in the future."
         );
       }
 
-      const configPath =
-        (args.config as ConfigPath) ||
-        (args.script && findWranglerToml(path.dirname(args.script)));
-      const config = readConfig(configPath, args);
-      const entry = await getEntry(args, config, "publish");
-
       if (args.latest) {
         logger.warn(
           "Using the latest version of the Workers runtime. To silence this warning, please choose a specific version of the runtime with --compatibility-date, or add a compatibility_date to your wrangler.toml.\n"
@@ -1672,6 +1656,7 @@ function createCLIParser(argv: string[]) {
           rules={getRules(config)}
           env={args.env}
           zone={undefined}
+          host={undefined}
           legacyEnv={isLegacyEnv(config)}
           build={config.build || {}}
           minify={undefined}
@@ -2288,7 +2273,7 @@ function createCLIParser(argv: string[]) {
             const namespaceId = getKVNamespaceId(args, config);
             // One of `args.path` and `args.value` must be defined
             const value = args.path
-              ? readFileSync(args.path)
+              ? readFileSyncToBuffer(args.path)
               : // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
                 args.value!;
 
@@ -2862,3 +2847,13 @@ function getDevCompatibilityDate(
   }
   return compatibilityDate ?? currentDate;
 }
+
+/**
+ * Avoiding calling `getPort()` multiple times by memoizing the first result.
+ */
+function memoizeGetPort(defaultPort: number) {
+  let portValue: number;
+  return async () => {
+    return portValue || (portValue = await getPort({ port: defaultPort }));
+  };
+}

package/src/kv.ts

@@ -111,7 +111,7 @@ export async function deleteKVNamespace(
  */
 export interface KeyValue {
   key: string;
-  value: string;
+  value: string | Buffer;
   expiration?: number;
   expiration_ttl?: number;
   metadata?: object;