add pupeteer resapwn

Author: softmarshmallow

testing/testing-screenshot/index.ts

@@ -8,14 +8,6 @@ interface ScreenshotOptions {
   };
 }
 
-export async function screenshot({ htmlcss, viewport }: ScreenshotOptions) {
-  const worker = new Worker({});
-  await worker.launch();
-  const buffer = worker.screenshot({ htmlcss, viewport });
-  await worker.terminate();
-  return buffer;
-}
-
 export class Worker {
   private browser: Browser;
   private page: Page;
@@ -46,30 +38,47 @@ export class Worker {
   }
 
   async screenshot({ htmlcss, viewport }: ScreenshotOptions) {
-    // check if page is available (try to avoid TargetClosedError)
-    if (!this.page || this.page.isClosed()) {
+    try {
+      if (!this.browser || !this.page || this.page.isClosed()) {
+        await this.relaunch();
+      }
+      await this.page.setViewport(viewport);
+      await this.page.setContent(htmlcss, { waitUntil: "networkidle0" });
+      const buffer = await this.page.screenshot({
+        type: "png",
+        // support transparency
+        omitBackground: true,
+      });
+      return buffer;
+    } catch (error) {
+      console.log(`Failed to take screenshot: ${error.message}`);
       await this.relaunch();
+      // After relaunch, retry taking screenshot or rethrow the error
+      return this.screenshot({ htmlcss, viewport });
     }
-
-    this.page.setViewport(viewport);
-    await this.page.setContent(htmlcss, { waitUntil: "networkidle0" });
-    const buffer = await this.page.screenshot({
-      type: "png",
-      // support transparency
-      omitBackground: true,
-    });
-    return buffer;
   }
 
   async close() {
-    try {
-      await this.browser.close();
-    } catch (e) {}
-    this.browser = null;
-    this.page = null;
+    if (this.browser) {
+      try {
+        await this.browser.close();
+      } catch (e) {
+        console.log(`Failed to close browser: ${e.message}`);
+      }
+      this.browser = null;
+      this.page = null;
+    }
   }
 
   terminate() {
     this.close();
   }
 }
+
+export async function screenshot(options: ScreenshotOptions) {
+  const worker = new Worker({});
+  await worker.launch();
+  const buffer = await worker.screenshot(options);
+  await worker.terminate();
+  return buffer;
+}