@midscene/web 0.3.0 → 0.3.1

package/dist/lib/index.js

@@ -346,7 +346,8 @@ var require_dayjs_min = __commonJS({
 var src_exports = {};
 __export(src_exports, {
   PlaywrightAiFixture: () => PlaywrightAiFixture,
-  PuppeteerAgent: () => PageAgent
+  PuppeteerAgent: () => PageAgent,
+  generateExtractData: () => generateExtractData
 });
 module.exports = __toCommonJS(src_exports);
 
@@ -486,17 +487,16 @@ async function alignElements(screenshotBuffer, elements, page) {
     return item.rect.height >= sizeThreshold && item.rect.width >= sizeThreshold;
   });
   for (const item of validElements) {
-    const { rect } = item;
-    const aligned = await (0, import_image.alignCoordByTrim)(screenshotBuffer, rect);
-    item.rect = aligned;
-    item.center = [
-      Math.round(aligned.left + aligned.width / 2),
-      Math.round(aligned.top + aligned.height / 2)
-    ];
+    const { rect, id, content, attributes, locator } = item;
     textsAligned.push(
-      new WebElementInfo(__spreadProps(__spreadValues({}, item), {
+      new WebElementInfo({
+        rect,
+        locator,
+        id,
+        content,
+        attributes,
         page
-      }))
+      })
     );
   }
   return textsAligned;
@@ -620,7 +620,7 @@ var PageTaskExecutor = class {
         };
         return taskFind;
       }
-      if (plan2.type === "Assert") {
+      if (plan2.type === "Assert" || plan2.type === "AssertWithoutThrow") {
         const assertPlan = plan2;
         const taskAssert = {
           type: "Insight",
@@ -637,13 +637,16 @@ var PageTaskExecutor = class {
               assertPlan.param.assertion
             );
             if (!assertion.pass) {
-              task.output = assertion;
-              task.log = {
-                dump: insightDump
-              };
-              throw new Error(
-                assertion.thought || "Assertion failed without reason"
-              );
+              if (plan2.type === "Assert") {
+                task.output = assertion;
+                task.log = {
+                  dump: insightDump
+                };
+                throw new Error(
+                  assertion.thought || "Assertion failed without reason"
+                );
+              }
+              task.error = assertion.thought;
             }
             return {
               output: assertion,
@@ -759,7 +762,19 @@ var PageTaskExecutor = class {
         return taskActionSleep;
       }
       if (plan2.type === "Error") {
-        throw new Error(`Got a task plan with type Error: ${plan2.thought}`);
+        const taskActionError = {
+          type: "Action",
+          subType: "Error",
+          param: plan2.param,
+          executor: async (taskParam) => {
+            (0, import_node_assert2.default)(
+              taskParam.thought,
+              "An error occurred, but no thought provided"
+            );
+            throw new Error(taskParam.thought);
+          }
+        };
+        return taskActionError;
       }
       throw new Error(`Unknown or Unsupported task type: ${plan2.type}`);
     }).map((task) => {
@@ -769,7 +784,6 @@ var PageTaskExecutor = class {
   }
   async action(userPrompt) {
     const taskExecutor = new import_core.Executor(userPrompt);
-    taskExecutor.description = userPrompt;
     let plans = [];
     const planningTask = {
       type: "Planning",
@@ -829,7 +843,6 @@ var PageTaskExecutor = class {
   async query(demand) {
     const description = typeof demand === "string" ? demand : JSON.stringify(demand);
     const taskExecutor = new import_core.Executor(description);
-    taskExecutor.description = description;
     const queryTask = {
       type: "Insight",
       subType: "Query",
@@ -857,9 +870,8 @@ var PageTaskExecutor = class {
     };
   }
   async assert(assertion) {
-    const description = assertion;
+    const description = `assert: ${assertion}`;
     const taskExecutor = new import_core.Executor(description);
-    taskExecutor.description = description;
     const assertionPlan = {
       type: "Assert",
       param: {
@@ -874,6 +886,64 @@ var PageTaskExecutor = class {
       executor: taskExecutor
     };
   }
+  async waitFor(assertion, opt) {
+    const description = `waitFor: ${assertion}`;
+    const taskExecutor = new import_core.Executor(description);
+    const { timeoutMs, checkIntervalMs } = opt;
+    (0, import_node_assert2.default)(assertion, "No assertion for waitFor");
+    (0, import_node_assert2.default)(timeoutMs, "No timeoutMs for waitFor");
+    (0, import_node_assert2.default)(checkIntervalMs, "No checkIntervalMs for waitFor");
+    const overallStartTime = Date.now();
+    let startTime = Date.now();
+    let errorThought = "";
+    while (Date.now() - overallStartTime < timeoutMs) {
+      startTime = Date.now();
+      const assertPlan = {
+        type: "AssertWithoutThrow",
+        param: {
+          assertion
+        }
+      };
+      const assertTask = await this.convertPlanToExecutable([assertPlan]);
+      await taskExecutor.append(this.wrapExecutorWithScreenshot(assertTask[0]));
+      const output = await taskExecutor.flush();
+      if (output.pass) {
+        return {
+          output: void 0,
+          executor: taskExecutor
+        };
+      }
+      errorThought = output.thought;
+      const now = Date.now();
+      if (now - startTime < checkIntervalMs) {
+        const timeRemaining = checkIntervalMs - (now - startTime);
+        const sleepPlan = {
+          type: "Sleep",
+          param: {
+            timeMs: timeRemaining
+          }
+        };
+        const sleepTask = await this.convertPlanToExecutable([sleepPlan]);
+        await taskExecutor.append(
+          this.wrapExecutorWithScreenshot(sleepTask[0])
+        );
+        await taskExecutor.flush();
+      }
+    }
+    const errorPlan = {
+      type: "Error",
+      param: {
+        thought: `waitFor timeout: ${errorThought}`
+      }
+    };
+    const errorTask = await this.convertPlanToExecutable([errorPlan]);
+    await taskExecutor.append(errorTask[0]);
+    await taskExecutor.flush();
+    return {
+      output: void 0,
+      executor: taskExecutor
+    };
+  }
 };
 
 // src/common/agent.ts
@@ -883,6 +953,7 @@ var PageAgent = class {
     this.opts = Object.assign(
       {
         generateReport: true,
+        autoPrintReportMsg: true,
         groupName: "Midscene Report",
         groupDescription: ""
       },
@@ -908,7 +979,7 @@ var PageAgent = class {
     return (0, import_utils4.stringifyDumpData)(this.dump);
   }
   writeOutActionDumps() {
-    const generateReport = this.opts.generateReport;
+    const { generateReport, autoPrintReportMsg } = this.opts;
     this.reportFile = (0, import_utils4.writeLogFile)({
       fileName: this.reportFileName,
       fileExt: import_utils4.groupedActionDumpFileExt,
@@ -916,7 +987,7 @@ var PageAgent = class {
       type: "dump",
       generateReport
     });
-    if (generateReport) {
+    if (generateReport && autoPrintReportMsg) {
       printReportMsg(this.reportFile);
     }
   }
@@ -952,6 +1023,20 @@ ${errorTask == null ? void 0 : errorTask.errorStack}`);
 ${reasonMsg}`);
     }
   }
+  async aiWaitFor(assertion, opt) {
+    const { executor } = await this.taskExecutor.waitFor(assertion, {
+      timeoutMs: (opt == null ? void 0 : opt.timeoutMs) || 15 * 1e3,
+      checkIntervalMs: (opt == null ? void 0 : opt.checkIntervalMs) || 3 * 1e3,
+      assertion
+    });
+    this.appendExecutionDump(executor.dump());
+    this.writeOutActionDumps();
+    if (executor.isInErrorState()) {
+      const errorTask = executor.latestErrorTask();
+      throw new Error(`${errorTask == null ? void 0 : errorTask.error}
+${errorTask == null ? void 0 : errorTask.errorStack}`);
+    }
+  }
   async ai(taskPrompt, type = "action") {
     if (type === "action") {
       return this.aiAction(taskPrompt);
@@ -968,6 +1053,9 @@ ${reasonMsg}`);
   }
 };
 
+// src/playwright/index.ts
+var import_test = require("@playwright/test");
+
 // src/playwright/cache.ts
 var import_node_fs2 = __toESM(require("fs"));
 var import_node_path2 = __toESM(require("path"));
@@ -1069,14 +1157,14 @@ var PlaywrightAiFixture = () => {
     }
     return pageAgentMap[idForPage];
   };
-  const updateDumpAnnotation = (test, dump) => {
-    const currentAnnotation = test.annotations.find((item) => {
+  const updateDumpAnnotation = (test2, dump) => {
+    const currentAnnotation = test2.annotations.find((item) => {
       return item.type === midsceneDumpAnnotationId;
     });
     if (currentAnnotation) {
       currentAnnotation.description = dump;
     } else {
-      test.annotations.push({
+      test2.annotations.push({
         type: midsceneDumpAnnotationId,
         description: dump
       });
@@ -1088,10 +1176,14 @@ var PlaywrightAiFixture = () => {
       const agent = agentForPage(page, testInfo);
       await use(
         async (taskPrompt, opts) => {
-          await page.waitForLoadState("networkidle");
-          const actionType = (opts == null ? void 0 : opts.type) || "action";
-          const result = await agent.ai(taskPrompt, actionType);
-          return result;
+          return new Promise((resolve, reject) => {
+            import_test.test.step(`ai - ${taskPrompt}`, async () => {
+              await page.waitForLoadState("networkidle");
+              const actionType = (opts == null ? void 0 : opts.type) || "action";
+              const result = await agent.ai(taskPrompt, actionType);
+              resolve(result);
+            });
+          });
         }
       );
       const taskCacheJson = agent.taskExecutor.taskCache.generateTaskCache();
@@ -1102,32 +1194,244 @@ var PlaywrightAiFixture = () => {
       const { taskFile, taskTitle } = groupAndCaseForTest(testInfo);
       const agent = agentForPage(page, testInfo);
       await use(async (taskPrompt) => {
-        await page.waitForLoadState("networkidle");
-        await agent.aiAction(taskPrompt);
+        import_test.test.step(`aiAction - ${taskPrompt}`, async () => {
+          await page.waitForLoadState("networkidle");
+          await agent.aiAction(taskPrompt);
+        });
       });
       updateDumpAnnotation(testInfo, agent.dumpDataString());
     },
     aiQuery: async ({ page }, use, testInfo) => {
       const agent = agentForPage(page, testInfo);
       await use(async (demand) => {
-        await page.waitForLoadState("networkidle");
-        const result = await agent.aiQuery(demand);
-        return result;
+        return new Promise((resolve, reject) => {
+          import_test.test.step(`aiQuery - ${JSON.stringify(demand)}`, async () => {
+            await page.waitForLoadState("networkidle");
+            const result = await agent.aiQuery(demand);
+            resolve(result);
+          });
+        });
       });
       updateDumpAnnotation(testInfo, agent.dumpDataString());
     },
     aiAssert: async ({ page }, use, testInfo) => {
       const agent = agentForPage(page, testInfo);
       await use(async (assertion, errorMsg) => {
-        await page.waitForLoadState("networkidle");
-        await agent.aiAssert(assertion, errorMsg);
+        return new Promise((resolve, reject) => {
+          import_test.test.step(`aiAssert - ${assertion}`, async () => {
+            await page.waitForLoadState("networkidle");
+            await agent.aiAssert(assertion, errorMsg);
+            resolve(null);
+          });
+        });
+      });
+      updateDumpAnnotation(testInfo, agent.dumpDataString());
+    },
+    aiWaitFor: async ({ page }, use, testInfo) => {
+      const agent = agentForPage(page, testInfo);
+      await use(async (assertion, opt) => {
+        return new Promise((resolve, reject) => {
+          import_test.test.step(`aiWaitFor - ${assertion}`, async () => {
+            await agent.aiWaitFor(assertion, opt);
+            resolve(null);
+          });
+        });
       });
       updateDumpAnnotation(testInfo, agent.dumpDataString());
     }
   };
 };
+
+// src/debug/index.ts
+var import_node_fs3 = require("fs");
+var import_node_path3 = __toESM(require("path"));
+
+// src/img/img.ts
+var import_node_assert3 = __toESM(require("assert"));
+var import_node_buffer = require("buffer");
+var import_sharp = __toESM(require("sharp"));
+var createSvgOverlay = (elements, imageWidth, imageHeight) => {
+  let svgContent = `<svg width="${imageWidth}" height="${imageHeight}" xmlns="http://www.w3.org/2000/svg">`;
+  const colors = [
+    { rect: "blue", text: "white" },
+    { rect: "green", text: "white" }
+  ];
+  svgContent += "<defs>";
+  elements.forEach((element, index) => {
+    svgContent += `
+      <clipPath id="clip${index}">
+        <rect x="${element.x}" y="${element.y}" width="${element.width}" height="${element.height}" />
+      </clipPath>
+    `;
+  });
+  svgContent += "</defs>";
+  elements.forEach((element, index) => {
+    const textWidth = element.label.length * 8;
+    const textHeight = 12;
+    const rectWidth = textWidth + 5;
+    const rectHeight = textHeight + 4;
+    let rectX = element.x - rectWidth;
+    let rectY = element.y + element.height / 2 - textHeight / 2 - 2;
+    let textX = rectX + rectWidth / 2;
+    let textY = rectY + rectHeight / 2 + 6;
+    if (rectX < 0) {
+      rectX = element.x;
+      rectY = element.y - rectHeight;
+      textX = rectX + rectWidth / 2;
+      textY = rectY + rectHeight / 2 + 6;
+    }
+    const color = colors[index % colors.length];
+    svgContent += `
+      <rect x="${element.x}" y="${element.y}" width="${element.width}" height="${element.height}" 
+            style="fill:none;stroke:${color.rect};stroke-width:4" clip-path="url(#clip${index})" />
+      <rect x="${rectX}" y="${rectY}" width="${rectWidth}" height="${rectHeight}" style="fill:${color.rect};" />
+      <text x="${textX}" y="${textY}" 
+            text-anchor="middle" dominant-baseline="middle" style="fill:${color.text};font-size:12px;font-weight:bold;">
+        ${element.label}
+      </text>
+    `;
+  });
+  svgContent += "</svg>";
+  return import_node_buffer.Buffer.from(svgContent);
+};
+var processImageElementInfo = async (options) => {
+  const base64Image = options.inputImgBase64.split(";base64,").pop();
+  (0, import_node_assert3.default)(base64Image, "base64Image is undefined");
+  const imageBuffer = import_node_buffer.Buffer.from(base64Image, "base64");
+  const metadata = await (0, import_sharp.default)(imageBuffer).metadata();
+  const { width, height } = metadata;
+  if (width && height) {
+    const svgOverlay = createSvgOverlay(
+      options.elementsPositionInfo,
+      width,
+      height
+    );
+    const svgOverlayWithoutText = createSvgOverlay(
+      options.elementsPositionInfoWithoutText,
+      width,
+      height
+    );
+    const compositeElementInfoImgBase64 = await (0, import_sharp.default)(imageBuffer).composite([{ input: svgOverlay, blend: "over" }]).toBuffer().then((data) => {
+      return data.toString("base64");
+    }).catch((err) => {
+      throw err;
+    });
+    const compositeElementInfoImgWithoutTextBase64 = await (0, import_sharp.default)(imageBuffer).composite([{ input: svgOverlayWithoutText, blend: "over" }]).toBuffer().then((data) => {
+      return data.toString("base64");
+    }).catch((err) => {
+      throw err;
+    });
+    return {
+      compositeElementInfoImgBase64,
+      compositeElementInfoImgWithoutTextBase64
+    };
+  }
+  throw Error("Image processing failed because width or height is undefined");
+};
+
+// src/img/util.ts
+async function getElementInfos(page) {
+  const captureElementSnapshot = await getElementInfosFromPage(page);
+  const elementsPositionInfo = captureElementSnapshot.map((elementInfo) => {
+    return {
+      label: elementInfo.indexId.toString(),
+      x: elementInfo.rect.left,
+      y: elementInfo.rect.top,
+      width: elementInfo.rect.width,
+      height: elementInfo.rect.height,
+      attributes: elementInfo.attributes
+    };
+  });
+  const elementsPositionInfoWithoutText = elementsPositionInfo.filter(
+    (elementInfo) => {
+      if (elementInfo.attributes.nodeType === "TEXT Node" /* TEXT */) {
+        return false;
+      }
+      return true;
+    }
+  );
+  return {
+    elementsPositionInfo,
+    captureElementSnapshot,
+    elementsPositionInfoWithoutText
+  };
+}
+
+// src/debug/index.ts
+var import_image3 = require("@midscene/core/image");
+async function generateExtractData(page, targetDir, saveImgType) {
+  const buffer = await page.screenshot({
+    encoding: "base64"
+  });
+  const inputImgBase64 = buffer.toString("base64");
+  const {
+    elementsPositionInfo,
+    captureElementSnapshot,
+    elementsPositionInfoWithoutText
+  } = await getElementInfos(page);
+  const inputImagePath = import_node_path3.default.join(targetDir, "input.png");
+  const outputImagePath = import_node_path3.default.join(targetDir, "output.png");
+  const outputWithoutTextImgPath = import_node_path3.default.join(
+    targetDir,
+    "output_without_text.png"
+  );
+  const resizeOutputImgPath = import_node_path3.default.join(targetDir, "resize-output.png");
+  const snapshotJsonPath = import_node_path3.default.join(targetDir, "element-snapshot.json");
+  const {
+    compositeElementInfoImgBase64,
+    compositeElementInfoImgWithoutTextBase64
+  } = await processImageElementInfo({
+    elementsPositionInfo,
+    elementsPositionInfoWithoutText,
+    inputImgBase64
+  });
+  const resizeImgBase64 = await (0, import_image3.resizeImg)(inputImgBase64);
+  if (!(saveImgType == null ? void 0 : saveImgType.disableSnapshot)) {
+    writeFileSyncWithDir(
+      snapshotJsonPath,
+      JSON.stringify(captureElementSnapshot, null, 2)
+    );
+  }
+  if (!(saveImgType == null ? void 0 : saveImgType.disableInputImage)) {
+    await (0, import_image3.saveBase64Image)({
+      base64Data: inputImgBase64,
+      outputPath: inputImagePath
+    });
+  }
+  if (!(saveImgType == null ? void 0 : saveImgType.disableOutputImage)) {
+    await (0, import_image3.saveBase64Image)({
+      base64Data: compositeElementInfoImgBase64,
+      outputPath: outputImagePath
+    });
+  }
+  if (!(saveImgType == null ? void 0 : saveImgType.disableOutputWithoutTextImg)) {
+    await (0, import_image3.saveBase64Image)({
+      base64Data: compositeElementInfoImgWithoutTextBase64,
+      outputPath: outputWithoutTextImgPath
+    });
+  }
+  if (!(saveImgType == null ? void 0 : saveImgType.disableResizeOutputImg)) {
+    await (0, import_image3.saveBase64Image)({
+      base64Data: resizeImgBase64,
+      outputPath: resizeOutputImgPath
+    });
+  }
+}
+function ensureDirectoryExistence(filePath) {
+  const dirname2 = import_node_path3.default.dirname(filePath);
+  if ((0, import_node_fs3.existsSync)(dirname2)) {
+    return;
+  }
+  ensureDirectoryExistence(dirname2);
+  (0, import_node_fs3.mkdirSync)(dirname2);
+}
+function writeFileSyncWithDir(filePath, content, options = {}) {
+  ensureDirectoryExistence(filePath);
+  (0, import_node_fs3.writeFileSync)(filePath, content, options);
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   PlaywrightAiFixture,
-  PuppeteerAgent
+  PuppeteerAgent,
+  generateExtractData
 });