wait.js 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. "use strict";
  2. var __defProp = Object.defineProperty;
  3. var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
  4. var __getOwnPropNames = Object.getOwnPropertyNames;
  5. var __hasOwnProp = Object.prototype.hasOwnProperty;
  6. var __export = (target, all) => {
  7. for (var name in all)
  8. __defProp(target, name, { get: all[name], enumerable: true });
  9. };
  10. var __copyProps = (to, from, except, desc) => {
  11. if (from && typeof from === "object" || typeof from === "function") {
  12. for (let key of __getOwnPropNames(from))
  13. if (!__hasOwnProp.call(to, key) && key !== except)
  14. __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  15. }
  16. return to;
  17. };
  18. var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
  19. var wait_exports = {};
  20. __export(wait_exports, {
  21. default: () => wait_default
  22. });
  23. module.exports = __toCommonJS(wait_exports);
  24. var import_bundle = require("../../sdk/bundle");
  25. var import_tool = require("./tool");
  26. const wait = (0, import_tool.defineTool)({
  27. capability: "core",
  28. schema: {
  29. name: "browser_wait_for",
  30. title: "Wait for",
  31. description: "Wait for text to appear or disappear or a specified time to pass",
  32. inputSchema: import_bundle.z.object({
  33. time: import_bundle.z.number().optional().describe("The time to wait in seconds"),
  34. text: import_bundle.z.string().optional().describe("The text to wait for"),
  35. textGone: import_bundle.z.string().optional().describe("The text to wait for to disappear")
  36. }),
  37. type: "assertion"
  38. },
  39. handle: async (context, params, response) => {
  40. if (!params.text && !params.textGone && !params.time)
  41. throw new Error("Either time, text or textGone must be provided");
  42. if (params.time) {
  43. response.addCode(`await new Promise(f => setTimeout(f, ${params.time} * 1000));`);
  44. await new Promise((f) => setTimeout(f, Math.min(3e4, params.time * 1e3)));
  45. }
  46. const tab = context.currentTabOrDie();
  47. const locator = params.text ? tab.page.getByText(params.text).first() : void 0;
  48. const goneLocator = params.textGone ? tab.page.getByText(params.textGone).first() : void 0;
  49. if (goneLocator) {
  50. response.addCode(`await page.getByText(${JSON.stringify(params.textGone)}).first().waitFor({ state: 'hidden' });`);
  51. await goneLocator.waitFor({ state: "hidden" });
  52. }
  53. if (locator) {
  54. response.addCode(`await page.getByText(${JSON.stringify(params.text)}).first().waitFor({ state: 'visible' });`);
  55. await locator.waitFor({ state: "visible" });
  56. }
  57. response.addResult(`Waited for ${params.text || params.textGone || params.time}`);
  58. response.setIncludeSnapshot();
  59. }
  60. });
  61. var wait_default = [
  62. wait
  63. ];