Merge pull request #38 from cloudflare/add_keep_alive_option

Add keep alive option

.github/workflows/ci.yml

@@ -59,7 +59,7 @@ jobs:
             machine: ubuntu-latest
             node: 18.18
           - name: macOS
-            machine: macos-latest
+            machine: macos-13
             node: 14
           - name: Windows
             machine: windows-latest

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cloudflare/puppeteer",
-  "version": "0.0.6",
+  "version": "0.0.7",
   "description": "A high-level API to control headless Chrome over the DevTools Protocol for use in Workers",
   "keywords": [
     "puppeteer",

src/puppeteer-core.ts

@@ -87,6 +87,10 @@ export interface LimitsResponse {
   timeUntilNextAllowedBrowserAcquisition: number;
 }
 
+export interface LaunchOptions {
+  keep_alive?: number; // milliseconds to keep browser alive even if it has no activity (from 10_000ms to 600_000ms, default is 60_000)
+}
+
 class PuppeteerWorkers extends Puppeteer {
   public constructor() {
     super({isPuppeteerCore: true});
@@ -103,8 +107,15 @@ class PuppeteerWorkers extends Puppeteer {
    * @param endpoint - Cloudflare worker binding
    * @returns a browser session or throws
    */
-  public async launch(endpoint: BrowserWorker): Promise<Browser> {
-    const res = await endpoint.fetch('/v1/acquire');
+  public async launch(
+    endpoint: BrowserWorker,
+    options?: LaunchOptions
+  ): Promise<Browser> {
+    let acquireUrl = '/v1/acquire';
+    if (options?.keep_alive) {
+      acquireUrl = `${acquireUrl}?keep_alive=${options.keep_alive}`;
+    }
+    const res = await endpoint.fetch(acquireUrl);
     const status = res.status;
     const text = await res.text();
     if (status !== 200) {