utils.js 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  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 utils_exports = {};
  20. __export(utils_exports, {
  21. callOnPageNoTrace: () => callOnPageNoTrace,
  22. dateAsFileName: () => dateAsFileName,
  23. waitForCompletion: () => waitForCompletion
  24. });
  25. module.exports = __toCommonJS(utils_exports);
  26. async function waitForCompletion(tab, callback) {
  27. const requests = /* @__PURE__ */ new Set();
  28. let frameNavigated = false;
  29. let waitCallback = () => {
  30. };
  31. const waitBarrier = new Promise((f) => {
  32. waitCallback = f;
  33. });
  34. const responseListener = (request) => {
  35. requests.delete(request);
  36. if (!requests.size)
  37. waitCallback();
  38. };
  39. const requestListener = (request) => {
  40. requests.add(request);
  41. void request.response().then(() => responseListener(request)).catch(() => {
  42. });
  43. };
  44. const frameNavigateListener = (frame) => {
  45. if (frame.parentFrame())
  46. return;
  47. frameNavigated = true;
  48. dispose();
  49. clearTimeout(timeout);
  50. void tab.waitForLoadState("load").then(waitCallback);
  51. };
  52. const onTimeout = () => {
  53. dispose();
  54. waitCallback();
  55. };
  56. tab.page.on("request", requestListener);
  57. tab.page.on("requestfailed", responseListener);
  58. tab.page.on("framenavigated", frameNavigateListener);
  59. const timeout = setTimeout(onTimeout, 1e4);
  60. const dispose = () => {
  61. tab.page.off("request", requestListener);
  62. tab.page.off("requestfailed", responseListener);
  63. tab.page.off("framenavigated", frameNavigateListener);
  64. clearTimeout(timeout);
  65. };
  66. try {
  67. const result = await callback();
  68. if (!requests.size && !frameNavigated)
  69. waitCallback();
  70. await waitBarrier;
  71. await tab.waitForTimeout(1e3);
  72. return result;
  73. } finally {
  74. dispose();
  75. }
  76. }
  77. async function callOnPageNoTrace(page, callback) {
  78. return await page._wrapApiCall(() => callback(page), { internal: true });
  79. }
  80. function dateAsFileName(extension) {
  81. const date = /* @__PURE__ */ new Date();
  82. return `page-${date.toISOString().replace(/[:.]/g, "-")}.${extension}`;
  83. }
  84. // Annotate the CommonJS export names for ESM import in node:
  85. 0 && (module.exports = {
  86. callOnPageNoTrace,
  87. dateAsFileName,
  88. waitForCompletion
  89. });