@smoothdeploy/playwright-core 1.57.1 → 1.58.1-beta-1770452953000

package/lib/server/recorder.js

@@ -35,6 +35,8 @@ module.exports = __toCommonJS(recorder_exports);
 var import_events = __toESM(require("events"));
 var import_fs = __toESM(require("fs"));
 var import_utils = require("../utils");
+var import_utilsBundle = require("../utilsBundle");
+var import_screenshotCompositor = require("./screenshotCompositor");
 var import_browserContext = require("./browserContext");
 var import_debugger = require("./debugger");
 var import_recorderUtils = require("./recorder/recorderUtils");
@@ -75,6 +77,7 @@ class Recorder extends import_events.default {
     this._listeners = [];
     this._enabled = false;
     this._callLogs = [];
+    this._screenshotCache = /* @__PURE__ */ new Map();
     this._context = context;
     this._params = params;
     this._mode = params.mode || "none";
@@ -449,8 +452,59 @@ class Recorder extends import_events.default {
     };
     return actionInContext;
   }
+  async _captureScreenshot(frame, actionInContext) {
+    const action = actionInContext.action;
+    const selector = "selector" in action ? action.selector : null;
+    const page = frame._page;
+    const timestamp = Date.now();
+    if (!selector || !("selector" in actionInContext.action))
+      return;
+    const cached = this._screenshotCache.get(selector);
+    if (cached) {
+      actionInContext.action.screenshot = cached.screenshot;
+      actionInContext.action.screenshotElement = cached.screenshotElement;
+      return;
+    }
+    try {
+      const elementHandle = await frame.selectors.query(selector, { strict: false });
+      if (!elementHandle)
+        return;
+      const box = await elementHandle.boundingBox();
+      if (!box)
+        return;
+      const cssViewport = await page.mainFrame().evaluateExpression(
+        "({width: window.innerWidth, height: window.innerHeight})"
+      ).catch(() => null);
+      const controller = new import_progress.ProgressController();
+      await controller.run(async (progress) => {
+        const rawBuffer = await page.screenshot(progress, {
+          animations: "disabled",
+          type: "png",
+          style: "x-pw-glass { display: none !important; }"
+        });
+        const { full, cropped } = (0, import_screenshotCompositor.compositeHighlight)(rawBuffer, box, cssViewport);
+        const fullJpeg = import_utilsBundle.jpegjs.encode(full, 50);
+        const croppedJpeg = import_utilsBundle.jpegjs.encode(cropped, 80);
+        const screenshot = {
+          buffer: Buffer.from(fullJpeg.data).toString("base64"),
+          mimeType: "image/jpeg",
+          timestamp
+        };
+        const screenshotElement = {
+          buffer: Buffer.from(croppedJpeg.data).toString("base64"),
+          mimeType: "image/jpeg",
+          timestamp
+        };
+        actionInContext.action.screenshot = screenshot;
+        actionInContext.action.screenshotElement = screenshotElement;
+        this._screenshotCache.set(selector, { screenshot, screenshotElement });
+      });
+    } catch (e) {
+    }
+  }
   async _performAction(frame, action) {
     const actionInContext = await this._createActionInContext(frame, action);
+    await this._captureScreenshot(frame, actionInContext);
     this._signalProcessor.addAction(actionInContext);
     if (actionInContext.action.name !== "openPage" && actionInContext.action.name !== "closePage")
       await (0, import_recorderRunner.performAction)(this._pageAliases, actionInContext);
@@ -458,6 +512,7 @@ class Recorder extends import_events.default {
   }
   async _recordAction(frame, action) {
     const actionInContext = await this._createActionInContext(frame, action);
+    await this._captureScreenshot(frame, actionInContext);
     this._signalProcessor.addAction(actionInContext);
   }
   _onFrameNavigated(frame, page) {

package/lib/server/registry/browserFetcher.js

@@ -42,10 +42,13 @@ var import_userAgent = require("../utils/userAgent");
 var import_utilsBundle = require("../../utilsBundle");
 var import_fileUtils = require("../utils/fileUtils");
 var import__ = require(".");
-async function downloadBrowserWithProgressBar(title, browserDirectory, executablePath, downloadURLs, downloadFileName, downloadSocketTimeout) {
+async function downloadBrowserWithProgressBar(title, browserDirectory, executablePath, downloadURLs, downloadFileName, downloadSocketTimeout, force) {
   if (await (0, import_fileUtils.existsAsync)((0, import__.browserDirectoryToMarkerFilePath)(browserDirectory))) {
     import_debugLogger.debugLogger.log("install", `${title} is already downloaded.`);
-    return false;
+    if (force)
+      import_debugLogger.debugLogger.log("install", `force-downloading ${title}.`);
+    else
+      return;
   }
   const uniqueTempDir = await import_fs.default.promises.mkdtemp(import_path.default.join(import_os.default.tmpdir(), "playwright-download-"));
   const zipPath = import_path.default.join(uniqueTempDir, downloadFileName);
@@ -63,7 +66,7 @@ async function downloadBrowserWithProgressBar(title, browserDirectory, executabl
       if (await (0, import_fileUtils.existsAsync)(zipPath))
         await import_fs.default.promises.unlink(zipPath);
       if (await (0, import_fileUtils.existsAsync)(browserDirectory))
-        await import_fs.default.promises.rmdir(browserDirectory, { recursive: true });
+        await (0, import_fileUtils.removeFolders)([browserDirectory]);
       const errorMessage = error?.message || "";
       import_debugLogger.debugLogger.log("install", `attempt #${attempt} - ERROR: ${errorMessage}`);
       if (attempt >= retryCount)
@@ -77,7 +80,6 @@ async function downloadBrowserWithProgressBar(title, browserDirectory, executabl
     await (0, import_fileUtils.removeFolders)([uniqueTempDir]);
   }
   logPolitely(`${title} downloaded to ${browserDirectory}`);
-  return true;
 }
 function downloadBrowserWithProgressBarOutOfProcess(title, browserDirectory, url, zipPath, executablePath, socketTimeout) {
   const cp = childProcess.fork(import_path.default.join(__dirname, "oopDownloadBrowserMain.js"));