wrangler 0.0.0-e043479 → 0.0.0-e09abbc

package/pages/functions/template-worker.ts

@@ -1,130 +0,0 @@
-import { match } from "path-to-regexp";
-import type { HTTPMethod } from "./routes";
-
-/* TODO: Grab these from @cloudflare/workers-types instead */
-type Params<P extends string = string> = Record<P, string | string[]>;
-
-type EventContext<Env, P extends string, Data> = {
-  request: Request;
-  waitUntil: (promise: Promise<unknown>) => void;
-  next: (input?: Request | string, init?: RequestInit) => Promise<Response>;
-  env: Env & { ASSETS: { fetch: typeof fetch } };
-  params: Params<P>;
-  data: Data;
-};
-
-declare type PagesFunction<
-  Env = unknown,
-  P extends string = string,
-  Data extends Record<string, unknown> = Record<string, unknown>
-> = (context: EventContext<Env, P, Data>) => Response | Promise<Response>;
-/* end @cloudflare/workers-types */
-
-type RouteHandler = {
-  routePath: string;
-  method?: HTTPMethod;
-  modules: PagesFunction[];
-  middlewares: PagesFunction[];
-};
-
-// inject `routes` via ESBuild
-declare const routes: RouteHandler[];
-// define `__FALLBACK_SERVICE__` via ESBuild
-declare const __FALLBACK_SERVICE__: string;
-
-// expect an ASSETS fetcher binding pointing to the asset-server stage
-type FetchEnv = {
-  [name: string]: { fetch: typeof fetch };
-  ASSETS: { fetch: typeof fetch };
-};
-
-type WorkerContext = {
-  waitUntil: (promise: Promise<unknown>) => void;
-};
-
-function* executeRequest(request: Request, _env: FetchEnv) {
-  const requestPath = new URL(request.url).pathname;
-
-  // First, iterate through the routes (backwards) and execute "middlewares" on partial route matches
-  for (const route of [...routes].reverse()) {
-    if (route.method && route.method !== request.method) {
-      continue;
-    }
-
-    const routeMatcher = match(route.routePath, { end: false });
-    const matchResult = routeMatcher(requestPath);
-    if (matchResult) {
-      for (const handler of route.middlewares.flat()) {
-        yield {
-          handler,
-          params: matchResult.params as Params,
-        };
-      }
-    }
-  }
-
-  // Then look for the first exact route match and execute its "modules"
-  for (const route of routes) {
-    if (route.method && route.method !== request.method) {
-      continue;
-    }
-
-    const routeMatcher = match(route.routePath, { end: true });
-    const matchResult = routeMatcher(requestPath);
-    if (matchResult && route.modules.length) {
-      for (const handler of route.modules.flat()) {
-        yield {
-          handler,
-          params: matchResult.params as Params,
-        };
-      }
-      break;
-    }
-  }
-}
-
-export default {
-  async fetch(request: Request, env: FetchEnv, workerContext: WorkerContext) {
-    const handlerIterator = executeRequest(request, env);
-    const data = {}; // arbitrary data the user can set between functions
-    const next = async (input?: RequestInfo, init?: RequestInit) => {
-      if (input !== undefined) {
-        request = new Request(input, init);
-      }
-
-      const result = handlerIterator.next();
-      // Note we can't use `!result.done` because this doesn't narrow to the correct type
-      if (result.done == false) {
-        const { handler, params } = result.value;
-        const context = {
-          request: new Request(request.clone()),
-          next,
-          params,
-          data,
-          env,
-          waitUntil: workerContext.waitUntil.bind(workerContext),
-        };
-
-        const response = await handler(context);
-
-        // https://fetch.spec.whatwg.org/#null-body-status
-        return new Response(
-          [101, 204, 205, 304].includes(response.status) ? null : response.body,
-          response
-        );
-      } else if (__FALLBACK_SERVICE__) {
-        // There are no more handlers so finish with the fallback service (`env.ASSETS.fetch` in Pages' case)
-        return env[__FALLBACK_SERVICE__].fetch(request);
-      } else {
-        // There was not fallback service so actually make the request to the origin.
-        return fetch(request);
-      }
-    };
-
-    try {
-      return next();
-    } catch (err) {
-      return new Response("Internal Error", { status: 500 });
-    }
-  },
-};

package/src/__tests__/helpers/faye-websocket.d.ts

@@ -1,6 +0,0 @@
-module "faye-websocket" {
-  /**
-   * Standards-compliant WebSocket client and server.
-   */
-  export default WebSocket;
-}

package/src/__tests__/helpers/mock-user.ts

@@ -1,27 +0,0 @@
-import { mkdirSync, writeFileSync } from "node:fs";
-import os from "node:os";
-import path from "node:path";
-
-export function writeUserConfig(
-  oauth_token?: string,
-  refresh_token?: string,
-  expiration_time?: string
-) {
-  const lines: string[] = [];
-  if (oauth_token) {
-    lines.push(`oauth_token = "${oauth_token}"`);
-  }
-  if (refresh_token) {
-    lines.push(`refresh_token = "${refresh_token}"`);
-  }
-  if (expiration_time) {
-    lines.push(`expiration_time = "${expiration_time}"`);
-  }
-  const configPath = path.join(os.homedir(), ".wrangler/config");
-  mkdirSync(configPath, { recursive: true });
-  writeFileSync(
-    path.join(configPath, "default.toml"),
-    lines.join("\n"),
-    "utf-8"
-  );
-}

package/src/__tests__/logout.test.ts

@@ -1,50 +0,0 @@
-import { existsSync } from "node:fs";
-import os from "node:os";
-import path from "node:path";
-import fetchMock from "jest-fetch-mock";
-import { initialise } from "../user";
-import { mockConsoleMethods } from "./helpers/mock-console";
-import { writeUserConfig } from "./helpers/mock-user";
-import { runInTempDir } from "./helpers/run-in-tmp";
-import { runWrangler } from "./helpers/run-wrangler";
-
-describe("wrangler", () => {
-  runInTempDir({ homedir: "./home" });
-  const std = mockConsoleMethods();
-
-  describe("logout", () => {
-    it("should exit with a message stating the user is not logged in", async () => {
-      await initialise();
-      await runWrangler("logout");
-      expect(std.out).toMatchInlineSnapshot(`"Not logged in, exiting..."`);
-    });
-
-    it("should logout user that has been properly logged in", async () => {
-      writeUserConfig("some-oauth-tok", "some-refresh-tok");
-
-      // Mock out the response for a request to revoke the auth tokens,
-      // checking the form of the request is as expected.
-      fetchMock.mockResponseOnce(async (req) => {
-        expect(req.url).toEqual("https://dash.cloudflare.com/oauth2/revoke");
-        expect(req.method).toEqual("POST");
-        return "";
-      });
-
-      await initialise();
-      await runWrangler("logout");
-
-      expect(std.out).toMatchInlineSnapshot(`
-        "💁  Wrangler is configured with an OAuth token. The token has been successfully revoked
-        Removing ./home/.wrangler/config/default.toml.. success!"
-      `);
-
-      // Make sure that we made the request to logout.
-      expect(fetchMock).toHaveBeenCalledTimes(1);
-
-      // Make sure that logout removed the config file containing the auth tokens.
-      expect(
-        existsSync(path.join(os.homedir(), ".wrangler/config/default.toml"))
-      ).toBe(false);
-    });
-  });
-});

package/src/__tests__/sentry.test.ts

@@ -1,146 +0,0 @@
-import * as fs from "node:fs";
-import * as fsp from "node:fs/promises";
-import os from "node:os";
-import path from "node:path";
-import * as TOML from "@iarna/toml";
-import * as Sentry from "@sentry/node";
-import prompts from "prompts";
-
-import { mockConsoleMethods } from "./helpers/mock-console";
-import { runInTempDir } from "./helpers/run-in-tmp";
-import { runWrangler } from "./helpers/run-wrangler";
-const { reportError } = jest.requireActual("../reporting");
-
-describe("Error Reporting", () => {
-  runInTempDir({ homedir: "./home" });
-  mockConsoleMethods();
-  const reportingTOMLPath = ".wrangler/config/reporting.toml";
-
-  const originalTTY = process.stdout.isTTY;
-  beforeEach(() => {
-    jest.mock("@sentry/node");
-    jest.spyOn(Sentry, "captureException");
-    process.stdout.isTTY = true;
-  });
-
-  afterEach(() => {
-    jest.unmock("@sentry/node");
-    jest.clearAllMocks();
-    process.stdout.isTTY = originalTTY;
-  });
-
-  it("should confirm user will allow error reporting usage", async () => {
-    jest.spyOn(prompts, "prompt").mockResolvedValue({ sentryDecision: true });
-
-    await reportError(new Error("test error"), "testFalse");
-
-    const { error_tracking_opt, error_tracking_opt_date } = TOML.parse(
-      await fsp.readFile(path.join(os.homedir(), reportingTOMLPath), "utf-8")
-    );
-
-    expect(error_tracking_opt).toBe(true);
-    expect(error_tracking_opt_date).toBeTruthy();
-
-    expect(Sentry.captureException).toHaveBeenCalledWith(
-      new Error("test error")
-    );
-  });
-
-  it("should confirm user will disallow error reporting usage", async () => {
-    jest.spyOn(prompts, "prompt").mockResolvedValue({ sentryDecision: false });
-    await reportError(new Error("test error"), "testFalse");
-
-    const { error_tracking_opt, error_tracking_opt_date } = TOML.parse(
-      await fsp.readFile(path.join(os.homedir(), reportingTOMLPath), "utf-8")
-    );
-
-    expect(error_tracking_opt).toBe(false);
-    expect(error_tracking_opt_date).toBeTruthy();
-
-    expect(Sentry.captureException).not.toHaveBeenCalledWith(
-      new Error("test error"),
-      "testFalse"
-    );
-  });
-
-  it("should confirm user will allow 'once' error reporting usage", async () => {
-    jest.spyOn(prompts, "prompt").mockResolvedValue({ sentryDecision: "once" });
-    await runWrangler();
-    await reportError(new Error("test error"), "testFalse");
-
-    expect(fs.existsSync(path.join(os.homedir(), reportingTOMLPath))).toBe(
-      false
-    );
-
-    expect(Sentry.captureException).not.toHaveBeenCalledWith(
-      new Error("test error"),
-      "testFalse"
-    );
-  });
-
-  it("should not prompt w/ subsequent Errors after user disallows error reporting", async () => {
-    jest.spyOn(prompts, "prompt").mockResolvedValue({ sentryDecision: false });
-    await reportError(new Error("test error"), "testFalse");
-
-    expect(Sentry.captureException).not.toHaveBeenCalledWith(
-      new Error("test error"),
-      "testFalse"
-    );
-
-    await reportError(new Error("second test error"), "testFalse");
-    expect(Sentry.captureException).not.toHaveBeenCalledWith(
-      new Error("second test error"),
-      "testFalse"
-    );
-  });
-
-  it("should not prompt w/ subsequent Errors after user Always allows error reporting", async () => {
-    jest.spyOn(prompts, "prompt").mockResolvedValue({ sentryDecision: true });
-    await reportError(new Error("test error"), "testFalse");
-
-    expect(Sentry.captureException).toHaveBeenCalledWith(
-      new Error("test error")
-    );
-
-    await reportError(new Error("second test error"), "testFalse");
-    expect(Sentry.captureException).toHaveBeenCalledWith(
-      new Error("second test error")
-    );
-  });
-
-  it("should prompt w/ subsequent Errors after user allows error reporting once", async () => {
-    const promptSpy = jest
-      .spyOn(prompts, "prompt")
-      .mockResolvedValue({ sentryDecision: "once" });
-    await reportError(new Error("test error"), "testFalse");
-
-    expect(Sentry.captureException).toHaveBeenCalledWith(
-      new Error("test error")
-    );
-
-    await reportError(new Error("second test error"), "testFalse");
-    expect(Sentry.captureException).toHaveBeenCalledWith(
-      new Error("second test error")
-    );
-
-    expect(promptSpy).toBeCalledTimes(2);
-  });
-
-  it("should not prompt in non-TTY environment", async () => {
-    process.stdout.isTTY = false;
-
-    await reportError(new Error("test error"), "testFalse");
-
-    const { error_tracking_opt, error_tracking_opt_date } = TOML.parse(
-      await fsp.readFile(path.join(os.homedir(), reportingTOMLPath), "utf-8")
-    );
-
-    expect(error_tracking_opt).toBe(false);
-    expect(error_tracking_opt_date).toBeTruthy();
-
-    expect(Sentry.captureException).not.toHaveBeenCalledWith(
-      new Error("test error")
-    );
-    process.stdout.isTTY = originalTTY;
-  });
-});

package/src/api/form_data.ts

@@ -1,217 +0,0 @@
-import { readFileSync } from "node:fs";
-import { FormData, File } from "undici";
-import type {
-  CfWorkerInit,
-  CfModuleType,
-  CfDurableObjectMigrations,
-} from "./worker.js";
-
-export function toMimeType(type: CfModuleType): string {
-  switch (type) {
-    case "esm":
-      return "application/javascript+module";
-    case "commonjs":
-      return "application/javascript";
-    case "compiled-wasm":
-      return "application/wasm";
-    case "buffer":
-      return "application/octet-stream";
-    case "text":
-      return "text/plain";
-    default:
-      throw new TypeError("Unsupported module: " + type);
-  }
-}
-
-export interface WorkerMetadata {
-  /** The name of the entry point module. Only exists when the worker is in the ES module format */
-  main_module?: string;
-  /** The name of the entry point module. Only exists when the worker is in the Service Worker format */
-  body_part?: string;
-  compatibility_date?: string;
-  compatibility_flags?: string[];
-  usage_model?: "bundled" | "unbound";
-  migrations?: CfDurableObjectMigrations;
-  bindings: (
-    | { type: "kv_namespace"; name: string; namespace_id: string }
-    | { type: "plain_text"; name: string; text: string }
-    | { type: "json"; name: string; json: unknown }
-    | { type: "wasm_module"; name: string; part: string }
-    | { type: "text_blob"; name: string; part: string }
-    | {
-        type: "durable_object_namespace";
-        name: string;
-        class_name: string;
-        script_name?: string;
-      }
-    | { type: "r2_bucket"; name: string; bucket_name: string }
-  )[];
-}
-
-/**
- * Creates a `FormData` upload from a `CfWorkerInit`.
- */
-export function toFormData(worker: CfWorkerInit): FormData {
-  const formData = new FormData();
-  const {
-    main,
-    bindings,
-    migrations,
-    usage_model,
-    compatibility_date,
-    compatibility_flags,
-  } = worker;
-
-  let { modules } = worker;
-
-  const metadataBindings: WorkerMetadata["bindings"] = [];
-
-  bindings.kv_namespaces?.forEach(({ id, binding }) => {
-    metadataBindings.push({
-      name: binding,
-      type: "kv_namespace",
-      namespace_id: id,
-    });
-  });
-
-  bindings.durable_objects?.bindings.forEach(
-    ({ name, class_name, script_name }) => {
-      metadataBindings.push({
-        name,
-        type: "durable_object_namespace",
-        class_name: class_name,
-        ...(script_name && { script_name }),
-      });
-    }
-  );
-
-  bindings.r2_buckets?.forEach(({ binding, bucket_name }) => {
-    metadataBindings.push({
-      name: binding,
-      type: "r2_bucket",
-      bucket_name,
-    });
-  });
-
-  Object.entries(bindings.vars || {})?.forEach(([key, value]) => {
-    if (typeof value === "string") {
-      metadataBindings.push({ name: key, type: "plain_text", text: value });
-    } else {
-      metadataBindings.push({ name: key, type: "json", json: value });
-    }
-  });
-
-  for (const [name, filePath] of Object.entries(bindings.wasm_modules || {})) {
-    metadataBindings.push({
-      name,
-      type: "wasm_module",
-      part: name,
-    });
-
-    formData.set(
-      name,
-      new File([readFileSync(filePath)], filePath, {
-        type: "application/wasm",
-      })
-    );
-  }
-
-  for (const [name, filePath] of Object.entries(bindings.text_blobs || {})) {
-    metadataBindings.push({
-      name,
-      type: "text_blob",
-      part: name,
-    });
-
-    if (name !== "__STATIC_CONTENT_MANIFEST") {
-      formData.set(
-        name,
-        new File([readFileSync(filePath)], filePath, {
-          type: "text/plain",
-        })
-      );
-    }
-  }
-
-  if (main.type === "commonjs") {
-    // This is a service-worker format worker.
-    for (const module of Object.values([...(modules || [])])) {
-      if (module.name === "__STATIC_CONTENT_MANIFEST") {
-        // Add the manifest to the form data.
-        formData.set(
-          module.name,
-          new File([module.content], module.name, {
-            type: "text/plain",
-          })
-        );
-        // And then remove it from the modules collection
-        modules = modules?.filter((m) => m !== module);
-      } else if (module.type === "compiled-wasm" || module.type === "text") {
-        // Convert all wasm/text modules into `wasm_module`/`text_blob` bindings.
-        // The "name" of the module is a file path. We use it
-        // to instead be a "part" of the body, and a reference
-        // that we can use inside our source. This identifier has to be a valid
-        // JS identifier, so we replace all non alphanumeric characters
-        // with an underscore.
-        const name = module.name.replace(/[^a-zA-Z0-9_$]/g, "_");
-        metadataBindings.push({
-          name,
-          type: module.type === "compiled-wasm" ? "wasm_module" : "text_blob",
-          part: name,
-        });
-
-        // Add the module to the form data.
-        formData.set(
-          name,
-          new File([module.content], module.name, {
-            type:
-              module.type === "compiled-wasm"
-                ? "application/wasm"
-                : "text/plain",
-          })
-        );
-        // And then remove it from the modules collection
-        modules = modules?.filter((m) => m !== module);
-      } else if (module.type === "buffer") {
-        throw new Error(
-          'The "buffer" module type is not yet supported in service worker format workers'
-        );
-      }
-    }
-  }
-
-  if (bindings.unsafe) {
-    // @ts-expect-error unsafe bindings don't need to match a specific type here
-    metadataBindings.push(...bindings.unsafe);
-  }
-
-  const metadata: WorkerMetadata = {
-    ...(main.type !== "commonjs"
-      ? { main_module: main.name }
-      : { body_part: main.name }),
-    bindings: metadataBindings,
-    ...(compatibility_date && { compatibility_date }),
-    ...(compatibility_flags && { compatibility_flags }),
-    ...(usage_model && { usage_model }),
-    ...(migrations && { migrations }),
-  };
-
-  formData.set("metadata", JSON.stringify(metadata));
-
-  if (main.type === "commonjs" && modules && modules.length > 0) {
-    throw new TypeError(
-      "More than one module can only be specified when type = 'esm'"
-    );
-  }
-
-  for (const module of [main].concat(modules || [])) {
-    formData.set(
-      module.name,
-      new File([module.content], module.name, {
-        type: toMimeType(module.type ?? main.type ?? "esm"),
-      })
-    );
-  }
-
-  return formData;
-}