1 files changed, 91 insertions, 0 deletions
diff --git a/packages/bun-lambda/scripts/publish-layer.ts b/packages/bun-lambda/scripts/publish-layer.ts
new file mode 100644
index 000000000..b7129fc50
--- /dev/null
+++ b/packages/bun-lambda/scripts/publish-layer.ts
@@ -0,0 +1,91 @@
+import { spawnSync } from "node:child_process";
+import { BuildCommand } from "./build-layer";
+
+export class PublishCommand extends BuildCommand {
+  static summary = "Publish a custom Lambda layer for Bun.";
+
+  #aws(args: string[]): string {
+    this.debug("$", "aws", ...args);
+    const { status, stdout, stderr } = spawnSync("aws", args, {
+      stdio: "pipe",
+    });
+    const result = stdout.toString("utf-8").trim();
+    if (status === 0) {
+      return result;
+    }
+    const reason = stderr.toString("utf-8").trim() || result;
+    throw new Error(`aws ${args.join(" ")} exited with ${status}: ${reason}`);
+  }
+
+  async run() {
+    const { flags } = await this.parse(PublishCommand);
+    this.debug("Options:", flags);
+    try {
+      const version = this.#aws(["--version"]);
+      this.debug("AWS CLI:", version);
+    } catch (error) {
+      this.debug(error);
+      this.error(
+        "Install the `aws` CLI to continue: https://docs.aws.amazon.com/cli/latest/userguide/getting-started-install.html",
+        { exit: 1 },
+      );
+    }
+    const { layer, region, arch, output, public: isPublic } = flags;
+    if (region.includes("*")) {
+      // prettier-ignore
+      const result = this.#aws([
+        "ec2",
+        "describe-regions",
+        "--query", "Regions[].RegionName",
+        "--output", "json"
+      ]);
+      region.length = 0;
+      for (const name of JSON.parse(result)) {
+        region.push(name);
+      }
+    } else if (!region.length) {
+      // prettier-ignore
+      region.push(this.#aws([
+        "configure",
+        "get",
+        "region"
+      ]));
+    }
+    this.log("Publishing...");
+    for (const regionName of region) {
+      for (const layerName of layer) {
+        // prettier-ignore
+        const result = this.#aws([
+          "lambda",
+          "publish-layer-version",
+          "--layer-name", layerName,
+          "--region", regionName,
+          "--description", "Bun is an incredibly fast JavaScript runtime, bundler, transpiler, and package manager.",
+          "--license-info", "MIT",
+          "--compatible-architectures", arch === "x64" ? "x86_64" : "arm64",
+          "--compatible-runtimes", "provided.al2", "provided",
+          "--zip-file", `fileb://${output}`,
+          "--output", "json",
+        ]);
+        const { LayerVersionArn } = JSON.parse(result);
+        this.log("Published", LayerVersionArn);
+        if (isPublic) {
+          // prettier-ignore
+          this.#aws([
+            "lambda",
+            "add-layer-version-permission",
+            "--layer-name", layerName,
+            "--region", regionName,
+            "--version-number", LayerVersionArn.split(":").pop(),
+            "--statement-id", `${layerName}-public`,
+            "--action", "lambda:GetLayerVersion",
+            "--principal", "*",
+          ]);
+        }
+      }
+    }
+    this.log("Done");
+  }
+}
+
+await PublishCommand.run(process.argv.slice(2));