@stablyai/internal-playwright-core 0.1.0

package/lib/server/pipeTransport.js

@@ -0,0 +1,89 @@
+"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 pipeTransport_exports = {};
+__export(pipeTransport_exports, {
+  PipeTransport: () => PipeTransport
+});
+module.exports = __toCommonJS(pipeTransport_exports);
+var import_utils = require("../utils");
+var import_debugLogger = require("./utils/debugLogger");
+class PipeTransport {
+  constructor(pipeWrite, pipeRead) {
+    this._pendingBuffers = [];
+    this._waitForNextTask = (0, import_utils.makeWaitForNextTask)();
+    this._closed = false;
+    this._pipeRead = pipeRead;
+    this._pipeWrite = pipeWrite;
+    pipeRead.on("data", (buffer) => this._dispatch(buffer));
+    pipeRead.on("close", () => {
+      this._closed = true;
+      if (this._onclose)
+        this._onclose.call(null);
+    });
+    pipeRead.on("error", (e) => import_debugLogger.debugLogger.log("error", e));
+    pipeWrite.on("error", (e) => import_debugLogger.debugLogger.log("error", e));
+    this.onmessage = void 0;
+  }
+  get onclose() {
+    return this._onclose;
+  }
+  set onclose(onclose) {
+    this._onclose = onclose;
+    if (onclose && !this._pipeRead.readable)
+      onclose();
+  }
+  send(message) {
+    if (this._closed)
+      throw new Error("Pipe has been closed");
+    this._pipeWrite.write(JSON.stringify(message));
+    this._pipeWrite.write("\0");
+  }
+  close() {
+    throw new Error("unimplemented");
+  }
+  _dispatch(buffer) {
+    let end = buffer.indexOf("\0");
+    if (end === -1) {
+      this._pendingBuffers.push(buffer);
+      return;
+    }
+    this._pendingBuffers.push(buffer.slice(0, end));
+    const message = Buffer.concat(this._pendingBuffers).toString();
+    this._waitForNextTask(() => {
+      if (this.onmessage)
+        this.onmessage.call(null, JSON.parse(message));
+    });
+    let start = end + 1;
+    end = buffer.indexOf("\0", start);
+    while (end !== -1) {
+      const message2 = buffer.toString(void 0, start, end);
+      this._waitForNextTask(() => {
+        if (this.onmessage)
+          this.onmessage.call(null, JSON.parse(message2));
+      });
+      start = end + 1;
+      end = buffer.indexOf("\0", start);
+    }
+    this._pendingBuffers = [buffer.slice(start)];
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  PipeTransport
+});

package/lib/server/playwright.js

@@ -0,0 +1,69 @@
+"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 playwright_exports = {};
+__export(playwright_exports, {
+  Playwright: () => Playwright,
+  createPlaywright: () => createPlaywright
+});
+module.exports = __toCommonJS(playwright_exports);
+var import_android = require("./android/android");
+var import_backendAdb = require("./android/backendAdb");
+var import_bidiChromium = require("./bidi/bidiChromium");
+var import_bidiFirefox = require("./bidi/bidiFirefox");
+var import_chromium = require("./chromium/chromium");
+var import_debugController = require("./debugController");
+var import_electron = require("./electron/electron");
+var import_firefox = require("./firefox/firefox");
+var import_instrumentation = require("./instrumentation");
+var import_webkit = require("./webkit/webkit");
+class Playwright extends import_instrumentation.SdkObject {
+  constructor(options) {
+    super((0, import_instrumentation.createRootSdkObject)(), void 0, "Playwright");
+    this._allPages = /* @__PURE__ */ new Set();
+    this._allBrowsers = /* @__PURE__ */ new Set();
+    this.options = options;
+    this.attribution.playwright = this;
+    this.instrumentation.addListener({
+      onBrowserOpen: (browser) => this._allBrowsers.add(browser),
+      onBrowserClose: (browser) => this._allBrowsers.delete(browser),
+      onPageOpen: (page) => this._allPages.add(page),
+      onPageClose: (page) => this._allPages.delete(page)
+    }, null);
+    this.chromium = new import_chromium.Chromium(this, new import_bidiChromium.BidiChromium(this));
+    this.firefox = new import_firefox.Firefox(this, new import_bidiFirefox.BidiFirefox(this));
+    this.webkit = new import_webkit.WebKit(this);
+    this.electron = new import_electron.Electron(this);
+    this.android = new import_android.Android(this, new import_backendAdb.AdbBackend());
+    this.debugController = new import_debugController.DebugController(this);
+  }
+  allBrowsers() {
+    return [...this._allBrowsers];
+  }
+  allPages() {
+    return [...this._allPages];
+  }
+}
+function createPlaywright(options) {
+  return new Playwright(options);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Playwright,
+  createPlaywright
+});

package/lib/server/progress.js

@@ -0,0 +1,112 @@
+"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 progress_exports = {};
+__export(progress_exports, {
+  ProgressController: () => ProgressController,
+  isAbortError: () => isAbortError,
+  raceUncancellableOperationWithCleanup: () => raceUncancellableOperationWithCleanup
+});
+module.exports = __toCommonJS(progress_exports);
+var import_errors = require("./errors");
+var import_utils = require("../utils");
+var import_manualPromise = require("../utils/isomorphic/manualPromise");
+class ProgressController {
+  constructor(metadata, onCallLog) {
+    this._forceAbortPromise = new import_manualPromise.ManualPromise();
+    this._donePromise = new import_manualPromise.ManualPromise();
+    this._state = "before";
+    this.metadata = metadata || { id: "", startTime: 0, endTime: 0, type: "Internal", method: "", params: {}, log: [], internal: true };
+    this._onCallLog = onCallLog;
+    this._forceAbortPromise.catch((e) => null);
+  }
+  async abort(error) {
+    if (this._state === "running") {
+      error[kAbortErrorSymbol] = true;
+      this._state = { error };
+      this._forceAbortPromise.reject(error);
+    }
+    await this._donePromise;
+  }
+  async run(task, timeout) {
+    (0, import_utils.assert)(this._state === "before");
+    this._state = "running";
+    const progress = {
+      log: (message) => {
+        if (this._state === "running")
+          this.metadata.log.push(message);
+        this._onCallLog?.(message);
+      },
+      metadata: this.metadata,
+      race: (promise) => {
+        const promises = Array.isArray(promise) ? promise : [promise];
+        return Promise.race([...promises, this._forceAbortPromise]);
+      },
+      wait: async (timeout2) => {
+        let timer2;
+        const promise = new Promise((f) => timer2 = setTimeout(f, timeout2));
+        return progress.race(promise).finally(() => clearTimeout(timer2));
+      }
+    };
+    let timer;
+    if (timeout) {
+      const timeoutError = new import_errors.TimeoutError(`Timeout ${timeout}ms exceeded.`);
+      timer = setTimeout(() => {
+        if (this._state === "running") {
+          timeoutError[kAbortErrorSymbol] = true;
+          this._state = { error: timeoutError };
+          this._forceAbortPromise.reject(timeoutError);
+        }
+      }, timeout);
+    }
+    try {
+      const result = await task(progress);
+      this._state = "finished";
+      return result;
+    } catch (error) {
+      this._state = { error };
+      throw error;
+    } finally {
+      clearTimeout(timer);
+      this._donePromise.resolve();
+    }
+  }
+}
+const kAbortErrorSymbol = Symbol("kAbortError");
+function isAbortError(error) {
+  return !!error[kAbortErrorSymbol];
+}
+async function raceUncancellableOperationWithCleanup(progress, run, cleanup) {
+  let aborted = false;
+  try {
+    return await progress.race(run().then(async (t) => {
+      if (aborted)
+        await cleanup(t);
+      return t;
+    }));
+  } catch (error) {
+    aborted = true;
+    throw error;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ProgressController,
+  isAbortError,
+  raceUncancellableOperationWithCleanup
+});

package/lib/server/protocolError.js

@@ -0,0 +1,52 @@
+"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 protocolError_exports = {};
+__export(protocolError_exports, {
+  ProtocolError: () => ProtocolError,
+  isProtocolError: () => isProtocolError,
+  isSessionClosedError: () => isSessionClosedError
+});
+module.exports = __toCommonJS(protocolError_exports);
+var import_stackTrace = require("../utils/isomorphic/stackTrace");
+class ProtocolError extends Error {
+  constructor(type, method, logs) {
+    super();
+    this.type = type;
+    this.method = method;
+    this.logs = logs;
+  }
+  setMessage(message) {
+    (0, import_stackTrace.rewriteErrorMessage)(this, `Protocol error (${this.method}): ${message}`);
+  }
+  browserLogMessage() {
+    return this.logs ? "\nBrowser logs:\n" + this.logs : "";
+  }
+}
+function isProtocolError(e) {
+  return e instanceof ProtocolError;
+}
+function isSessionClosedError(e) {
+  return e instanceof ProtocolError && (e.type === "closed" || e.type === "crashed");
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ProtocolError,
+  isProtocolError,
+  isSessionClosedError
+});

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/codeGenerator.js

@@ -0,0 +1,153 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.CodeGenerator = void 0;
+var _events = require("events");
+/**
+ * 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 CodeGenerator extends _events.EventEmitter {
+  constructor(browserName, enabled, launchOptions, contextOptions, deviceName, saveStorage) {
+    super();
+
+    // Make a copy of options to modify them later.
+    this._currentAction = null;
+    this._lastAction = null;
+    this._actions = [];
+    this._enabled = void 0;
+    this._options = void 0;
+    launchOptions = {
+      headless: false,
+      ...launchOptions
+    };
+    contextOptions = {
+      ...contextOptions
+    };
+    this._enabled = enabled;
+    this._options = {
+      browserName,
+      launchOptions,
+      contextOptions,
+      deviceName,
+      saveStorage
+    };
+    this.restart();
+  }
+  restart() {
+    this._currentAction = null;
+    this._lastAction = null;
+    this._actions = [];
+    this.emit('change');
+  }
+  setEnabled(enabled) {
+    this._enabled = enabled;
+  }
+  addAction(action) {
+    if (!this._enabled) return;
+    this.willPerformAction(action);
+    this.didPerformAction(action);
+  }
+  willPerformAction(action) {
+    if (!this._enabled) return;
+    this._currentAction = action;
+  }
+  performedActionFailed(action) {
+    if (!this._enabled) return;
+    if (this._currentAction === action) this._currentAction = null;
+  }
+  didPerformAction(actionInContext) {
+    if (!this._enabled) return;
+    const action = actionInContext.action;
+    let eraseLastAction = false;
+    if (this._lastAction && this._lastAction.frame.pageAlias === actionInContext.frame.pageAlias) {
+      const lastAction = this._lastAction.action;
+      // We augment last action based on the type.
+      if (this._lastAction && action.name === 'fill' && lastAction.name === 'fill') {
+        if (action.selector === lastAction.selector) eraseLastAction = true;
+      }
+      if (lastAction && action.name === 'click' && lastAction.name === 'click') {
+        if (action.selector === lastAction.selector && action.clickCount > lastAction.clickCount) eraseLastAction = true;
+      }
+      if (lastAction && action.name === 'navigate' && lastAction.name === 'navigate') {
+        if (action.url === lastAction.url) {
+          // Already at a target URL.
+          this._currentAction = null;
+          return;
+        }
+      }
+      // Check and uncheck erase click.
+      if (lastAction && (action.name === 'check' || action.name === 'uncheck') && lastAction.name === 'click') {
+        if (action.selector === lastAction.selector) eraseLastAction = true;
+      }
+    }
+    this._lastAction = actionInContext;
+    this._currentAction = null;
+    if (eraseLastAction) this._actions.pop();
+    this._actions.push(actionInContext);
+    this.emit('change');
+  }
+  commitLastAction() {
+    if (!this._enabled) return;
+    const action = this._lastAction;
+    if (action) action.committed = true;
+  }
+  signal(pageAlias, frame, signal) {
+    if (!this._enabled) return;
+
+    // Signal either arrives while action is being performed or shortly after.
+    if (this._currentAction) {
+      this._currentAction.action.signals.push(signal);
+      return;
+    }
+    if (this._lastAction && (!this._lastAction.committed || signal.name !== 'navigation')) {
+      const signals = this._lastAction.action.signals;
+      if (signal.name === 'navigation' && signals.length && signals[signals.length - 1].name === 'download') return;
+      if (signal.name === 'download' && signals.length && signals[signals.length - 1].name === 'navigation') signals.length = signals.length - 1;
+      this._lastAction.action.signals.push(signal);
+      this.emit('change');
+      return;
+    }
+    if (signal.name === 'navigation' && frame._page.mainFrame() === frame) {
+      this.addAction({
+        frame: {
+          pageAlias,
+          isMainFrame: true
+        },
+        committed: true,
+        action: {
+          name: 'navigate',
+          url: frame.url(),
+          signals: []
+        }
+      });
+    }
+  }
+  generateStructure(languageGenerator) {
+    const header = languageGenerator.generateHeader(this._options);
+    const footer = languageGenerator.generateFooter(this._options.saveStorage);
+    const actions = this._actions.map(a => languageGenerator.generateAction(a)).filter(Boolean);
+    const text = [header, ...actions, footer].join('\n');
+    return {
+      header,
+      footer,
+      actions,
+      text
+    };
+  }
+}
+exports.CodeGenerator = CodeGenerator;