Patch cloudflare worker placement (#6421)

Author: vimtor

platform/src/components/cloudflare/providers/worker-placement.ts

@@ -0,0 +1,91 @@
+import { CustomResourceOptions, Input, Output, dynamic } from "@pulumi/pulumi";
+import { cfFetch } from "../helpers/fetch.js";
+import { DEFAULT_ACCOUNT_ID } from "../account-id.js";
+
+interface Inputs {
+  accountId: string;
+  scriptName: string;
+  mode?: string;
+  region?: string;
+  host?: string;
+  hostname?: string;
+}
+
+export interface WorkerPlacementInputs {
+  accountId?: Input<Inputs["accountId"]>;
+  scriptName: Input<Inputs["scriptName"]>;
+  mode?: Input<Inputs["mode"]>;
+  region?: Input<Inputs["region"]>;
+  host?: Input<Inputs["host"]>;
+  hostname?: Input<Inputs["hostname"]>;
+}
+
+export interface WorkerPlacement {
+  scriptName: Output<string>;
+}
+
+function buildPlacement(inputs: Inputs) {
+  if (inputs.mode) return { mode: inputs.mode };
+  if (inputs.region) return { region: inputs.region };
+  if (inputs.host) return { host: inputs.host };
+  if (inputs.hostname) return { hostname: inputs.hostname };
+  return {};
+}
+
+class Provider implements dynamic.ResourceProvider {
+  async create(inputs: Inputs): Promise<dynamic.CreateResult> {
+    await this.patch(inputs, buildPlacement(inputs));
+    return {
+      id: inputs.scriptName,
+      outs: { scriptName: inputs.scriptName },
+    };
+  }
+
+  async update(
+    id: string,
+    olds: Inputs,
+    news: Inputs,
+  ): Promise<dynamic.UpdateResult> {
+    await this.patch(news, buildPlacement(news));
+    return {
+      outs: { scriptName: news.scriptName },
+    };
+  }
+
+  async delete(id: string, inputs: Inputs) {
+    await this.patch(inputs, {});
+  }
+
+  async patch(inputs: Inputs, placement: Partial<Record<string, string>>) {
+    const accountId = inputs.accountId || DEFAULT_ACCOUNT_ID;
+    const formData = new FormData();
+    formData.append(
+      "settings",
+      new Blob([JSON.stringify({ placement })], {
+        type: "application/json",
+      }),
+    );
+    await cfFetch(
+      `/accounts/${accountId}/workers/scripts/${inputs.scriptName}/settings`,
+      {
+        method: "PATCH",
+        body: formData,
+      },
+    );
+  }
+}
+
+export class WorkerPlacement extends dynamic.Resource {
+  constructor(
+    name: string,
+    args: WorkerPlacementInputs,
+    opts?: CustomResourceOptions,
+  ) {
+    super(
+      new Provider(),
+      `${name}.sst.cloudflare.WorkerPlacement`,
+      { ...args, scriptName: args.scriptName },
+      opts,
+    );
+  }
+}

platform/src/components/cloudflare/worker.ts

@@ -13,6 +13,7 @@ import type { Loader } from "esbuild";
 import type { EsbuildOptions } from "../esbuild.js";
 import { Component, Transform, transform } from "../component";
 import { WorkerUrl } from "./providers/worker-url.js";
+import { WorkerPlacement } from "./providers/worker-placement.js";
 import { Link } from "../link.js";
 import type { Input } from "../input.js";
 import { ZoneLookup } from "./providers/zone-lookup.js";
@@ -181,6 +182,36 @@ export interface WorkerArgs {
      */
     directory: Input<string>;
   }>;
+  /**
+   * Configure [placement](https://developers.cloudflare.com/workers/configuration/placement/)
+   * for your Worker.
+   *
+   * @example
+   *
+   * #### Smart Placement
+   * ```js
+   * {
+   *   placement: {
+   *     mode: "smart"
+   *   }
+   * }
+   * ```
+   *
+   * #### Explicit region
+   * ```js
+   * {
+   *   placement: {
+   *     region: "aws:us-east-1"
+   *   }
+   * }
+   * ```
+   */
+  placement?: Input<{
+    mode?: Input<string>;
+    region?: Input<string>;
+    host?: Input<string>;
+    hostname?: Input<string>;
+  }>;
   /**
    * [Transform](/docs/components/#transform) how this component creates its underlying
    * resources.
@@ -261,6 +292,7 @@ export interface WorkerArgs {
 export class Worker extends Component implements Link.Linkable {
   private script: cf.WorkersScript;
   private workerUrl: WorkerUrl;
+  private workerPlacement?: WorkerPlacement;
   private workerDomain?: cf.WorkerDomain;
 
   constructor(name: string, args: WorkerArgs, opts?: ComponentResourceOptions) {
@@ -290,10 +322,12 @@ export class Worker extends Component implements Link.Linkable {
     const build = buildHandler();
     const script = createScript();
     const workerUrl = createWorkersUrl();
+    const workerPlacement = createWorkerPlacement();
     const workerDomain = createWorkersDomain();
 
     this.script = script;
     this.workerUrl = workerUrl;
+    this.workerPlacement = workerPlacement;
     this.workerDomain = workerDomain;
 
     all([dev, buildInput, script.scriptName]).apply(
@@ -539,6 +573,22 @@ export class Worker extends Component implements Link.Linkable {
       );
     }
 
+    // workaround: pulumi cloudflare provider marks placement as read-only,
+    // so we use the CF API directly until upstream support lands
+    function createWorkerPlacement() {
+      if (!args.placement) return;
+
+      return new WorkerPlacement(
+        `${name}Placement`,
+        {
+          accountId: DEFAULT_ACCOUNT_ID,
+          scriptName: script.scriptName,
+          ...args.placement,
+        },
+        { parent },
+      );
+    }
+
     function createWorkersDomain() {
       if (!args.domain) return;