Feat: Extension puppeteer (#1323)

* feat: add puppeteer extension

* chore: update package and config

* feat: add puppeteer task

* Create little-donkeys-protect.md

---------

Co-authored-by: Eric Allam <[email protected]>

Author: ThibautCuchet

.changeset/little-donkeys-protect.md

@@ -0,0 +1,5 @@
+---
+"@trigger.dev/build": patch
+---
+
+Feat: puppeteer build extension

packages/build/package.json

@@ -27,7 +27,8 @@
       "./extensions/core": "./src/extensions/core.ts",
       "./extensions/prisma": "./src/extensions/prisma.ts",
       "./extensions/audioWaveform": "./src/extensions/audioWaveform.ts",
-      "./extensions/typescript": "./src/extensions/typescript.ts"
+      "./extensions/typescript": "./src/extensions/typescript.ts",
+      "./extensions/puppeteer": "./src/extensions/puppeteer.ts"
     },
     "sourceDialects": [
       "@triggerdotdev/source"
@@ -49,6 +50,9 @@
       ],
       "extensions/typescript": [
         "dist/commonjs/extensions/typescript.d.ts"
+      ],
+      "extensions/puppeteer": [
+        "dist/commonjs/extensions/puppeteer.d.ts"
       ]
     }
   },
@@ -145,6 +149,17 @@
         "types": "./dist/commonjs/extensions/typescript.d.ts",
         "default": "./dist/commonjs/extensions/typescript.js"
       }
+    },
+    "./extensions/puppeteer": {
+      "import": {
+        "@triggerdotdev/source": "./src/extensions/puppeteer.ts",
+        "types": "./dist/esm/extensions/puppeteer.d.ts",
+        "default": "./dist/esm/extensions/puppeteer.js"
+      },
+      "require": {
+        "types": "./dist/commonjs/extensions/puppeteer.d.ts",
+        "default": "./dist/commonjs/extensions/puppeteer.js"
+      }
     }
   },
   "main": "./dist/commonjs/index.js",

packages/build/src/extensions/puppeteer.ts

@@ -0,0 +1,34 @@
+import { BuildManifest } from "@trigger.dev/core/v3";
+import { BuildContext, BuildExtension } from "@trigger.dev/core/v3/build";
+
+export function puppeteer() {
+  return new PuppeteerExtension();
+}
+
+class PuppeteerExtension implements BuildExtension {
+  public readonly name = "PuppeteerExtension";
+
+  async onBuildComplete(context: BuildContext, manifest: BuildManifest) {
+    if (context.target === "dev") {
+      return;
+    }
+
+    context.logger.debug(`Adding ${this.name} to the build`);
+
+    const instructions = [
+      `RUN apt-get update && apt-get install curl gnupg -y \
+  && curl --location --silent https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add - \
+  && sh -c 'echo "deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main" >> /etc/apt/sources.list.d/google.list' \
+  && apt-get update \
+  && apt-get install google-chrome-stable -y --no-install-recommends \
+  && rm -rf /var/lib/apt/lists/*`,
+    ];
+
+    context.addLayer({
+      id: "puppeteer",
+      image: {
+        instructions,
+      },
+    });
+  }
+}