@checkly/playwright-core 1.48.22 → 1.51.11-beta.2

package/lib/server/recorder/chat.js

@@ -0,0 +1,177 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.Chat = void 0;
+exports.asString = asString;
+var _transport = require("../transport");
+/**
+ * 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 Chat {
+  constructor(wsEndpoint) {
+    this._history = [];
+    this._connectionPromise = void 0;
+    this._chatSinks = new Map();
+    this._wsEndpoint = void 0;
+    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 = _transport.WebSocketTransport.connect(undefined, 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;
+  }
+}
+exports.Chat = Chat;
+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._transport = void 0;
+    this._lastId = 0;
+    this._closed = false;
+    this._pending = new Map();
+    this._onEvent = void 0;
+    this._onClose = void 0;
+    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 === undefined) {
+      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 = undefined;
+    this._transport.onclose = undefined;
+    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();
+  }
+}

package/lib/server/recorder/codeGenerator.js

@@ -0,0 +1,154 @@
+"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) {
+      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;
+//# sourceMappingURL=codeGenerator.js.map

package/lib/server/recorder/contextRecorder.js

@@ -6,18 +6,18 @@ Object.defineProperty(exports, "__esModule", {
 exports.ContextRecorder = void 0;
 exports.generateFrameSelector = generateFrameSelector;
 var _events = require("events");
+var _recorderCollection = require("./recorderCollection");
 var recorderSource = _interopRequireWildcard(require("../../generated/pollingRecorderSource"));
 var _utils = require("../../utils");
-var _timeoutRunner = require("../../utils/timeoutRunner");
+var _timeoutRunner = require("../../utils/isomorphic/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; }
+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 && {}.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.
  *
@@ -35,7 +35,7 @@ function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e;
  */
 
 class ContextRecorder extends _events.EventEmitter {
-  constructor(codegenMode, context, params, delegate) {
+  constructor(context, params, delegate) {
     super();
     this._collection = void 0;
     this._pageAliases = new Map();
@@ -49,8 +49,6 @@ class ContextRecorder extends _events.EventEmitter {
     this._throttledOutputFile = null;
     this._orderedLanguages = [];
     this._listeners = [];
-    this._codegenMode = void 0;
-    this._codegenMode = codegenMode;
     this._context = context;
     this._params = params;
     this._delegate = delegate;
@@ -112,7 +110,7 @@ class ContextRecorder extends _events.EventEmitter {
       var _this$_throttledOutpu3;
       (_this$_throttledOutpu3 = this._throttledOutputFile) === null || _this$_throttledOutpu3 === void 0 || _this$_throttledOutpu3.flush();
     }));
-    this.setEnabled(true);
+    this.setEnabled(params.mode === 'recording');
   }
   setOutput(codegenId, outputFile) {
     var _this$_collection;
@@ -145,14 +143,6 @@ class ContextRecorder extends _events.EventEmitter {
   }
   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);
@@ -198,6 +188,9 @@ class ContextRecorder extends _events.EventEmitter {
       for (const page of this._context.pages()) this._onFrameNavigated(page.mainFrame(), page);
     }
   }
+  runTask(task) {
+    // TODO: implement
+  }
   _describeMainFrame(page) {
     return {
       pageAlias: this._pageAliases.get(page),
@@ -289,9 +282,7 @@ async function generateFrameSelectorInParent(parent, frame) {
         return injected.generateSelectorSimple(element);
       }, frameElement);
       return selector;
-    } catch (e) {
-      return e.toString();
-    }
+    } catch (e) {}
   }, (0, _utils.monotonicTime)() + 2000);
   if (!result.timedOut && result.result) return result.result;
   if (frame.name()) return `iframe[name=${(0, _utils.quoteCSSAttributeValue)(frame.name())}]`;