rebrowser-playwright-core 1.49.1 → 1.52.0

package/lib/server/recorder/chat.js

@@ -0,0 +1,161 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var chat_exports = {};
+__export(chat_exports, {
+  Chat: () => Chat,
+  asString: () => asString
+});
+module.exports = __toCommonJS(chat_exports);
+var import_transport = require("../transport");
+class Chat {
+  constructor(wsEndpoint) {
+    this._history = [];
+    this._chatSinks = /* @__PURE__ */ new Map();
+    this._wsEndpoint = wsEndpoint;
+  }
+  clearHistory() {
+    this._history = [];
+  }
+  async post(prompt) {
+    await this._append("user", prompt);
+    let text = await asString(await this._post());
+    if (text.startsWith("```json") && text.endsWith("```"))
+      text = text.substring("```json".length, text.length - "```".length);
+    for (let i = 0; i < 3; ++i) {
+      try {
+        return JSON.parse(text);
+      } catch (e) {
+        await this._append("user", String(e));
+      }
+    }
+    throw new Error("Failed to parse response: " + text);
+  }
+  async _append(user, content) {
+    this._history.push({ user, content });
+  }
+  async _connection() {
+    if (!this._connectionPromise) {
+      this._connectionPromise = import_transport.WebSocketTransport.connect(void 0, this._wsEndpoint).then((transport) => {
+        return new Connection(transport, (method, params) => this._dispatchEvent(method, params), () => {
+        });
+      });
+    }
+    return this._connectionPromise;
+  }
+  _dispatchEvent(method, params) {
+    if (method === "chatChunk") {
+      const { chatId, chunk } = params;
+      const chunkSink = this._chatSinks.get(chatId);
+      chunkSink(chunk);
+      if (!chunk)
+        this._chatSinks.delete(chatId);
+    }
+  }
+  async _post() {
+    const connection = await this._connection();
+    const result = await connection.send("chat", { history: this._history });
+    const { chatId } = result;
+    const { iterable, addChunk } = iterablePump();
+    this._chatSinks.set(chatId, addChunk);
+    return iterable;
+  }
+}
+async function asString(stream) {
+  let result = "";
+  for await (const chunk of stream)
+    result += chunk;
+  return result;
+}
+function iterablePump() {
+  let controller;
+  const stream = new ReadableStream({ start: (c) => controller = c });
+  const iterable = async function* () {
+    const reader = stream.getReader();
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done)
+        break;
+      yield value;
+    }
+  }();
+  return {
+    iterable,
+    addChunk: (chunk) => {
+      if (chunk)
+        controller.enqueue(chunk);
+      else
+        controller.close();
+    }
+  };
+}
+class Connection {
+  constructor(transport, onEvent, onClose) {
+    this._lastId = 0;
+    this._closed = false;
+    this._pending = /* @__PURE__ */ new Map();
+    this._transport = transport;
+    this._onEvent = onEvent;
+    this._onClose = onClose;
+    this._transport.onmessage = this._dispatchMessage.bind(this);
+    this._transport.onclose = this._close.bind(this);
+  }
+  send(method, params) {
+    const id = this._lastId++;
+    const message = { id, method, params };
+    this._transport.send(message);
+    return new Promise((resolve, reject) => {
+      this._pending.set(id, { resolve, reject });
+    });
+  }
+  _dispatchMessage(message) {
+    if (message.id === void 0) {
+      this._onEvent(message.method, message.params);
+      return;
+    }
+    const callback = this._pending.get(message.id);
+    this._pending.delete(message.id);
+    if (!callback)
+      return;
+    if (message.error) {
+      callback.reject(new Error(message.error.message));
+      return;
+    }
+    callback.resolve(message.result);
+  }
+  _close() {
+    this._closed = true;
+    this._transport.onmessage = void 0;
+    this._transport.onclose = void 0;
+    for (const { reject } of this._pending.values())
+      reject(new Error("Connection closed"));
+    this._onClose();
+  }
+  isClosed() {
+    return this._closed;
+  }
+  close() {
+    if (!this._closed)
+      this._transport.close();
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Chat,
+  asString
+});

package/lib/server/recorder/contextRecorder.js

@@ -1,88 +1,76 @@
 "use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var contextRecorder_exports = {};
+__export(contextRecorder_exports, {
+  ContextRecorder: () => ContextRecorder,
+  generateFrameSelector: () => generateFrameSelector
 });
-exports.ContextRecorder = void 0;
-exports.generateFrameSelector = generateFrameSelector;
-var _events = require("events");
-var recorderSource = _interopRequireWildcard(require("../../generated/pollingRecorderSource"));
-var _utils = require("../../utils");
-var _timeoutRunner = require("../../utils/timeoutRunner");
-var _browserContext = require("../browserContext");
-var _languages = require("../codegen/languages");
-var _frames = require("../frames");
-var _page = require("../page");
-var _throttledFile = require("./throttledFile");
-var _recorderCollection = require("./recorderCollection");
-var _language = require("../codegen/language");
-function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
-function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
-/**
- * Copyright (c) Microsoft Corporation.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-class ContextRecorder extends _events.EventEmitter {
-  constructor(codegenMode, context, params, delegate) {
+module.exports = __toCommonJS(contextRecorder_exports);
+var import_events = require("events");
+var import_recorderCollection = require("./recorderCollection");
+var recorderSource = __toESM(require("../../generated/pollingRecorderSource"));
+var import_utils = require("../../utils");
+var import_timeoutRunner = require("../../utils/isomorphic/timeoutRunner");
+var import_browserContext = require("../browserContext");
+var import_languages = require("../codegen/languages");
+var import_frames = require("../frames");
+var import_page = require("../page");
+var import_throttledFile = require("./throttledFile");
+var import_language = require("../codegen/language");
+class ContextRecorder extends import_events.EventEmitter {
+  constructor(context, params, delegate) {
     super();
-    this._collection = void 0;
-    this._pageAliases = new Map();
+    this._pageAliases = /* @__PURE__ */ new Map();
     this._lastPopupOrdinal = 0;
     this._lastDialogOrdinal = -1;
     this._lastDownloadOrdinal = -1;
-    this._context = void 0;
-    this._params = void 0;
-    this._delegate = void 0;
-    this._recorderSources = void 0;
     this._throttledOutputFile = null;
     this._orderedLanguages = [];
     this._listeners = [];
-    this._codegenMode = void 0;
-    this._codegenMode = codegenMode;
     this._context = context;
     this._params = params;
     this._delegate = delegate;
     this._recorderSources = [];
     const language = params.language || context.attribution.playwright.options.sdkLanguage;
     this.setOutput(language, params.outputFile);
-
-    // Make a copy of options to modify them later.
     const languageGeneratorOptions = {
       browserName: context._browser.options.name,
-      launchOptions: {
-        headless: false,
-        ...params.launchOptions,
-        tracesDir: undefined
-      },
-      contextOptions: {
-        ...params.contextOptions
-      },
+      launchOptions: { headless: false, ...params.launchOptions, tracesDir: void 0 },
+      contextOptions: { ...params.contextOptions },
       deviceName: params.device,
       saveStorage: params.saveStorage
     };
-    this._collection = new _recorderCollection.RecorderCollection(this._pageAliases);
-    this._collection.on('change', actions => {
+    this._collection = new import_recorderCollection.RecorderCollection(this._pageAliases);
+    this._collection.on("change", (actions) => {
       this._recorderSources = [];
       for (const languageGenerator of this._orderedLanguages) {
-        var _this$_throttledOutpu;
-        const {
-          header,
-          footer,
-          actionTexts,
-          text
-        } = (0, _language.generateCode)(actions, languageGenerator, languageGeneratorOptions);
+        const { header, footer, actionTexts, text } = (0, import_language.generateCode)(actions, languageGenerator, languageGeneratorOptions);
         const source = {
           isRecorded: true,
           label: languageGenerator.name,
@@ -95,88 +83,93 @@ class ContextRecorder extends _events.EventEmitter {
           language: languageGenerator.highlighter,
           highlight: []
         };
-        source.revealLine = text.split('\n').length - 1;
+        source.revealLine = text.split("\n").length - 1;
         this._recorderSources.push(source);
-        if (languageGenerator === this._orderedLanguages[0]) (_this$_throttledOutpu = this._throttledOutputFile) === null || _this$_throttledOutpu === void 0 || _this$_throttledOutpu.setContent(source.text);
+        if (languageGenerator === this._orderedLanguages[0])
+          this._throttledOutputFile?.setContent(source.text);
       }
       this.emit(ContextRecorder.Events.Change, {
         sources: this._recorderSources,
         actions
       });
     });
-    context.on(_browserContext.BrowserContext.Events.BeforeClose, () => {
-      var _this$_throttledOutpu2;
-      (_this$_throttledOutpu2 = this._throttledOutputFile) === null || _this$_throttledOutpu2 === void 0 || _this$_throttledOutpu2.flush();
+    context.on(import_browserContext.BrowserContext.Events.BeforeClose, () => {
+      this._throttledOutputFile?.flush();
     });
-    this._listeners.push(_utils.eventsHelper.addEventListener(process, 'exit', () => {
-      var _this$_throttledOutpu3;
-      (_this$_throttledOutpu3 = this._throttledOutputFile) === null || _this$_throttledOutpu3 === void 0 || _this$_throttledOutpu3.flush();
+    this._listeners.push(import_utils.eventsHelper.addEventListener(process, "exit", () => {
+      this._throttledOutputFile?.flush();
     }));
-    this.setEnabled(true);
+    this.setEnabled(params.mode === "recording");
+  }
+  static {
+    this.Events = {
+      Change: "change"
+    };
   }
   setOutput(codegenId, outputFile) {
-    var _this$_collection;
-    const languages = (0, _languages.languageSet)();
-    const primaryLanguage = [...languages].find(l => l.id === codegenId);
-    if (!primaryLanguage) throw new Error(`\n===============================\nUnsupported language: '${codegenId}'\n===============================\n`);
+    const languages = (0, import_languages.languageSet)();
+    const primaryLanguage = [...languages].find((l) => l.id === codegenId);
+    if (!primaryLanguage)
+      throw new Error(`
+===============================
+Unsupported language: '${codegenId}'
+===============================
+`);
     languages.delete(primaryLanguage);
     this._orderedLanguages = [primaryLanguage, ...languages];
-    this._throttledOutputFile = outputFile ? new _throttledFile.ThrottledFile(outputFile) : null;
-    (_this$_collection = this._collection) === null || _this$_collection === void 0 || _this$_collection.restart();
+    this._throttledOutputFile = outputFile ? new import_throttledFile.ThrottledFile(outputFile) : null;
+    this._collection?.restart();
   }
   languageName(id) {
     for (const lang of this._orderedLanguages) {
-      if (!id || lang.id === id) return lang.highlighter;
+      if (!id || lang.id === id)
+        return lang.highlighter;
     }
-    return 'javascript';
+    return "javascript";
   }
   async install() {
-    this._context.on(_browserContext.BrowserContext.Events.Page, page => this._onPage(page));
-    for (const page of this._context.pages()) this._onPage(page);
-    this._context.on(_browserContext.BrowserContext.Events.Dialog, dialog => this._onDialog(dialog.page()));
-
-    // Input actions that potentially lead to navigation are intercepted on the page and are
-    // performed by the Playwright.
-    await this._context.exposeBinding('__pw_recorderPerformAction', false, (source, action) => this._performAction(source.frame, action));
-
-    // Other non-essential actions are simply being recorded.
-    await this._context.exposeBinding('__pw_recorderRecordAction', false, (source, action) => this._recordAction(source.frame, action));
+    this._context.on(import_browserContext.BrowserContext.Events.Page, (page) => this._onPage(page));
+    for (const page of this._context.pages())
+      this._onPage(page);
+    this._context.on(import_browserContext.BrowserContext.Events.Dialog, (dialog) => this._onDialog(dialog.page()));
+    await this._context.exposeBinding(
+      "__pw_recorderPerformAction",
+      false,
+      (source, action) => this._performAction(source.frame, action)
+    );
+    await this._context.exposeBinding(
+      "__pw_recorderRecordAction",
+      false,
+      (source, action) => this._recordAction(source.frame, action)
+    );
     await this._context.extendInjectedScript(recorderSource.source);
   }
   setEnabled(enabled) {
     this._collection.setEnabled(enabled);
-    if (this._codegenMode === 'trace-events') {
-      if (enabled) this._context.tracing.startChunk({
-        name: 'trace',
-        title: 'trace'
-      }).catch(() => {});else this._context.tracing.stopChunk({
-        mode: 'discard'
-      }).catch(() => {});
-    }
   }
   dispose() {
-    _utils.eventsHelper.removeEventListeners(this._listeners);
+    import_utils.eventsHelper.removeEventListeners(this._listeners);
   }
   async _onPage(page) {
-    // First page is called page, others are called popup1, popup2, etc.
     const frame = page.mainFrame();
-    page.on('close', () => {
+    page.on("close", () => {
       this._collection.addRecordedAction({
         frame: this._describeMainFrame(page),
         action: {
-          name: 'closePage',
+          name: "closePage",
           signals: []
         },
-        startTime: (0, _utils.monotonicTime)()
+        startTime: (0, import_utils.monotonicTime)()
       });
       this._pageAliases.delete(page);
     });
-    frame.on(_frames.Frame.Events.InternalNavigation, event => {
-      if (event.isPublic) this._onFrameNavigated(frame, page);
+    frame.on(import_frames.Frame.Events.InternalNavigation, (event) => {
+      if (event.isPublic)
+        this._onFrameNavigated(frame, page);
     });
-    page.on(_page.Page.Events.Download, () => this._onDownload(page));
-    const suffix = this._pageAliases.size ? String(++this._lastPopupOrdinal) : '';
-    const pageAlias = 'page' + suffix;
+    page.on(import_page.Page.Events.Download, () => this._onDownload(page));
+    const suffix = this._pageAliases.size ? String(++this._lastPopupOrdinal) : "";
+    const pageAlias = "page" + suffix;
     this._pageAliases.set(page, pageAlias);
     if (page.opener()) {
       this._onPopup(page.opener(), page);
@@ -184,20 +177,23 @@ class ContextRecorder extends _events.EventEmitter {
       this._collection.addRecordedAction({
         frame: this._describeMainFrame(page),
         action: {
-          name: 'openPage',
+          name: "openPage",
           url: page.mainFrame().url(),
           signals: []
         },
-        startTime: (0, _utils.monotonicTime)()
+        startTime: (0, import_utils.monotonicTime)()
       });
     }
   }
   clearScript() {
     this._collection.restart();
-    if (this._params.mode === 'recording') {
-      for (const page of this._context.pages()) this._onFrameNavigated(page.mainFrame(), page);
+    if (this._params.mode === "recording") {
+      for (const page of this._context.pages())
+        this._onFrameNavigated(page.mainFrame(), page);
     }
   }
+  runTask(task) {
+  }
   _describeMainFrame(page) {
     return {
       pageAlias: this._pageAliases.get(page),
@@ -211,18 +207,17 @@ class ContextRecorder extends _events.EventEmitter {
     };
   }
   testIdAttributeName() {
-    return this._params.testIdAttributeName || this._context.selectors().testIdAttributeName() || 'data-testid';
+    return this._params.testIdAttributeName || this._context.selectors().testIdAttributeName() || "data-testid";
   }
   async _createActionInContext(frame, action) {
-    var _this$_delegate$rewri, _this$_delegate;
     const frameDescription = await this._describeFrame(frame);
     const actionInContext = {
       frame: frameDescription,
       action,
-      description: undefined,
-      startTime: (0, _utils.monotonicTime)()
+      description: void 0,
+      startTime: (0, import_utils.monotonicTime)()
     };
-    await ((_this$_delegate$rewri = (_this$_delegate = this._delegate).rewriteActionInContext) === null || _this$_delegate$rewri === void 0 ? void 0 : _this$_delegate$rewri.call(_this$_delegate, this._pageAliases, actionInContext));
+    await this._delegate.rewriteActionInContext?.(this._pageAliases, actionInContext);
     return actionInContext;
   }
   async _performAction(frame, action) {
@@ -233,45 +228,30 @@ class ContextRecorder extends _events.EventEmitter {
   }
   _onFrameNavigated(frame, page) {
     const pageAlias = this._pageAliases.get(page);
-    this._collection.signal(pageAlias, frame, {
-      name: 'navigation',
-      url: frame.url()
-    });
+    this._collection.signal(pageAlias, frame, { name: "navigation", url: frame.url() });
   }
   _onPopup(page, popup) {
     const pageAlias = this._pageAliases.get(page);
     const popupAlias = this._pageAliases.get(popup);
-    this._collection.signal(pageAlias, page.mainFrame(), {
-      name: 'popup',
-      popupAlias
-    });
+    this._collection.signal(pageAlias, page.mainFrame(), { name: "popup", popupAlias });
   }
   _onDownload(page) {
     const pageAlias = this._pageAliases.get(page);
     ++this._lastDownloadOrdinal;
-    this._collection.signal(pageAlias, page.mainFrame(), {
-      name: 'download',
-      downloadAlias: this._lastDownloadOrdinal ? String(this._lastDownloadOrdinal) : ''
-    });
+    this._collection.signal(pageAlias, page.mainFrame(), { name: "download", downloadAlias: this._lastDownloadOrdinal ? String(this._lastDownloadOrdinal) : "" });
   }
   _onDialog(page) {
     const pageAlias = this._pageAliases.get(page);
     ++this._lastDialogOrdinal;
-    this._collection.signal(pageAlias, page.mainFrame(), {
-      name: 'dialog',
-      dialogAlias: this._lastDialogOrdinal ? String(this._lastDialogOrdinal) : ''
-    });
+    this._collection.signal(pageAlias, page.mainFrame(), { name: "dialog", dialogAlias: this._lastDialogOrdinal ? String(this._lastDialogOrdinal) : "" });
   }
 }
-exports.ContextRecorder = ContextRecorder;
-ContextRecorder.Events = {
-  Change: 'change'
-};
 async function generateFrameSelector(frame) {
   const selectorPromises = [];
   while (frame) {
     const parent = frame.parentFrame();
-    if (!parent) break;
+    if (!parent)
+      break;
     selectorPromises.push(generateFrameSelectorInParent(parent, frame));
     frame = parent;
   }
@@ -279,21 +259,28 @@ async function generateFrameSelector(frame) {
   return result.reverse();
 }
 async function generateFrameSelectorInParent(parent, frame) {
-  const result = await (0, _timeoutRunner.raceAgainstDeadline)(async () => {
+  const result = await (0, import_timeoutRunner.raceAgainstDeadline)(async () => {
     try {
       const frameElement = await frame.frameElement();
-      if (!frameElement || !parent) return;
+      if (!frameElement || !parent)
+        return;
       const utility = await parent._utilityContext();
       const injected = await utility.injectedScript();
-      const selector = await injected.evaluate((injected, element) => {
-        return injected.generateSelectorSimple(element);
+      const selector = await injected.evaluate((injected2, element) => {
+        return injected2.generateSelectorSimple(element);
       }, frameElement);
       return selector;
     } catch (e) {
-      return e.toString();
     }
-  }, (0, _utils.monotonicTime)() + 2000);
-  if (!result.timedOut && result.result) return result.result;
-  if (frame.name()) return `iframe[name=${(0, _utils.quoteCSSAttributeValue)(frame.name())}]`;
-  return `iframe[src=${(0, _utils.quoteCSSAttributeValue)(frame.url())}]`;
-}
+  }, (0, import_utils.monotonicTime)() + 2e3);
+  if (!result.timedOut && result.result)
+    return result.result;
+  if (frame.name())
+    return `iframe[name=${(0, import_utils.quoteCSSAttributeValue)(frame.name())}]`;
+  return `iframe[src=${(0, import_utils.quoteCSSAttributeValue)(frame.url())}]`;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ContextRecorder,
+  generateFrameSelector
+});