patchright 1.48.0__py3-none-win32.whl

patchright/driver/package/lib/utils/wsServer.js

@@ -0,0 +1,127 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.perMessageDeflate = exports.WSServer = void 0;
+var _utils = require("../utils");
+var _utilsBundle = require("../utilsBundle");
+var _debugLogger = require("./debugLogger");
+/**
+ * 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.
+ */
+
+let lastConnectionId = 0;
+const kConnectionSymbol = Symbol('kConnection');
+const perMessageDeflate = exports.perMessageDeflate = {
+  zlibDeflateOptions: {
+    level: 3
+  },
+  zlibInflateOptions: {
+    chunkSize: 10 * 1024
+  },
+  threshold: 10 * 1024
+};
+class WSServer {
+  constructor(delegate) {
+    this._wsServer = void 0;
+    this.server = void 0;
+    this._delegate = void 0;
+    this._delegate = delegate;
+  }
+  async listen(port = 0, hostname, path) {
+    _debugLogger.debugLogger.log('server', `Server started at ${new Date()}`);
+    const server = (0, _utils.createHttpServer)((request, response) => {
+      if (request.method === 'GET' && request.url === '/json') {
+        response.setHeader('Content-Type', 'application/json');
+        response.end(JSON.stringify({
+          wsEndpointPath: path
+        }));
+        return;
+      }
+      response.end('Running');
+    });
+    server.on('error', error => _debugLogger.debugLogger.log('server', String(error)));
+    this.server = server;
+    const wsEndpoint = await new Promise((resolve, reject) => {
+      server.listen(port, hostname, () => {
+        const address = server.address();
+        if (!address) {
+          reject(new Error('Could not bind server socket'));
+          return;
+        }
+        const wsEndpoint = typeof address === 'string' ? `${address}${path}` : `ws://${hostname || 'localhost'}:${address.port}${path}`;
+        resolve(wsEndpoint);
+      }).on('error', reject);
+    });
+    _debugLogger.debugLogger.log('server', 'Listening at ' + wsEndpoint);
+    this._wsServer = new _utilsBundle.wsServer({
+      noServer: true,
+      perMessageDeflate
+    });
+    if (this._delegate.onHeaders) this._wsServer.on('headers', headers => this._delegate.onHeaders(headers));
+    server.on('upgrade', (request, socket, head) => {
+      var _this$_delegate$onUpg, _this$_delegate, _this$_wsServer;
+      const pathname = new URL('http://localhost' + request.url).pathname;
+      if (pathname !== path) {
+        socket.write(`HTTP/${request.httpVersion} 400 Bad Request\r\n\r\n`);
+        socket.destroy();
+        return;
+      }
+      const upgradeResult = (_this$_delegate$onUpg = (_this$_delegate = this._delegate).onUpgrade) === null || _this$_delegate$onUpg === void 0 ? void 0 : _this$_delegate$onUpg.call(_this$_delegate, request, socket);
+      if (upgradeResult) {
+        socket.write(upgradeResult.error);
+        socket.destroy();
+        return;
+      }
+      (_this$_wsServer = this._wsServer) === null || _this$_wsServer === void 0 || _this$_wsServer.handleUpgrade(request, socket, head, ws => {
+        var _this$_wsServer2;
+        return (_this$_wsServer2 = this._wsServer) === null || _this$_wsServer2 === void 0 ? void 0 : _this$_wsServer2.emit('connection', ws, request);
+      });
+    });
+    this._wsServer.on('connection', (ws, request) => {
+      _debugLogger.debugLogger.log('server', 'Connected client ws.extension=' + ws.extensions);
+      const url = new URL('http://localhost' + (request.url || ''));
+      const id = String(++lastConnectionId);
+      _debugLogger.debugLogger.log('server', `[${id}] serving connection: ${request.url}`);
+      const connection = this._delegate.onConnection(request, url, ws, id);
+      ws[kConnectionSymbol] = connection;
+    });
+    return wsEndpoint;
+  }
+  async close() {
+    var _this$_delegate$onClo, _this$_delegate2;
+    const server = this._wsServer;
+    if (!server) return;
+    _debugLogger.debugLogger.log('server', 'closing websocket server');
+    const waitForClose = new Promise(f => server.close(f));
+    // First disconnect all remaining clients.
+    await Promise.all(Array.from(server.clients).map(async ws => {
+      const connection = ws[kConnectionSymbol];
+      if (connection) await connection.close();
+      try {
+        ws.terminate();
+      } catch (e) {}
+    }));
+    await waitForClose;
+    _debugLogger.debugLogger.log('server', 'closing http server');
+    if (this.server) await new Promise(f => this.server.close(f));
+    this._wsServer = undefined;
+    this.server = undefined;
+    _debugLogger.debugLogger.log('server', 'closed server');
+    await ((_this$_delegate$onClo = (_this$_delegate2 = this._delegate).onClose) === null || _this$_delegate$onClo === void 0 ? void 0 : _this$_delegate$onClo.call(_this$_delegate2));
+  }
+}
+exports.WSServer = WSServer;

patchright/driver/package/lib/utils/zipFile.js

@@ -0,0 +1,75 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.ZipFile = void 0;
+var _zipBundle = require("../zipBundle");
+/**
+ * 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 ZipFile {
+  constructor(fileName) {
+    this._fileName = void 0;
+    this._zipFile = void 0;
+    this._entries = new Map();
+    this._openedPromise = void 0;
+    this._fileName = fileName;
+    this._openedPromise = this._open();
+  }
+  async _open() {
+    await new Promise((fulfill, reject) => {
+      _zipBundle.yauzl.open(this._fileName, {
+        autoClose: false
+      }, (e, z) => {
+        if (e) {
+          reject(e);
+          return;
+        }
+        this._zipFile = z;
+        this._zipFile.on('entry', entry => {
+          this._entries.set(entry.fileName, entry);
+        });
+        this._zipFile.on('end', fulfill);
+      });
+    });
+  }
+  async entries() {
+    await this._openedPromise;
+    return [...this._entries.keys()];
+  }
+  async read(entryPath) {
+    await this._openedPromise;
+    const entry = this._entries.get(entryPath);
+    if (!entry) throw new Error(`${entryPath} not found in file ${this._fileName}`);
+    return new Promise((resolve, reject) => {
+      this._zipFile.openReadStream(entry, (error, readStream) => {
+        if (error || !readStream) {
+          reject(error || 'Entry not found');
+          return;
+        }
+        const buffers = [];
+        readStream.on('data', data => buffers.push(data));
+        readStream.on('end', () => resolve(Buffer.concat(buffers)));
+      });
+    });
+  }
+  close() {
+    var _this$_zipFile;
+    (_this$_zipFile = this._zipFile) === null || _this$_zipFile === void 0 || _this$_zipFile.close();
+  }
+}
+exports.ZipFile = ZipFile;

patchright/driver/package/lib/utils/zones.js

@@ -0,0 +1,62 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.zones = void 0;
+var _async_hooks = require("async_hooks");
+/**
+ * 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 ZoneManager {
+  constructor() {
+    this._asyncLocalStorage = new _async_hooks.AsyncLocalStorage();
+  }
+  run(type, data, func) {
+    const previous = this._asyncLocalStorage.getStore();
+    const zone = new Zone(previous, type, data);
+    return this._asyncLocalStorage.run(zone, func);
+  }
+  zoneData(type) {
+    for (let zone = this._asyncLocalStorage.getStore(); zone; zone = zone.previous) {
+      if (zone.type === type) return zone.data;
+    }
+    return undefined;
+  }
+  exitZones(func) {
+    return this._asyncLocalStorage.run(undefined, func);
+  }
+  printZones() {
+    const zones = [];
+    for (let zone = this._asyncLocalStorage.getStore(); zone; zone = zone.previous) {
+      let str = zone.type;
+      if (zone.type === 'apiZone') str += `(${zone.data.apiName})`;
+      zones.push(str);
+    }
+    console.log('zones: ', zones.join(' -> '));
+  }
+}
+class Zone {
+  constructor(previous, type, data) {
+    this.type = void 0;
+    this.data = void 0;
+    this.previous = void 0;
+    this.type = type;
+    this.data = data;
+    this.previous = previous;
+  }
+}
+const zones = exports.zones = new ZoneManager();

patchright/driver/package/lib/utilsBundle.js

@@ -0,0 +1,82 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.minimatch = exports.mime = exports.lockfile = exports.jpegjs = exports.getProxyForUrl = exports.dotenv = exports.debug = exports.colors = exports.SocksProxyAgent = exports.PNG = exports.HttpsProxyAgent = void 0;
+exports.ms = ms;
+exports.open = void 0;
+exports.parseStackTraceLine = parseStackTraceLine;
+exports.wsServer = exports.wsSender = exports.wsReceiver = exports.ws = exports.progress = exports.program = void 0;
+var _url = _interopRequireDefault(require("url"));
+var _path = _interopRequireDefault(require("path"));
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+/**
+ * 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.
+ */
+
+const colors = exports.colors = require('./utilsBundleImpl').colors;
+const debug = exports.debug = require('./utilsBundleImpl').debug;
+const dotenv = exports.dotenv = require('./utilsBundleImpl').dotenv;
+const getProxyForUrl = exports.getProxyForUrl = require('./utilsBundleImpl').getProxyForUrl;
+const HttpsProxyAgent = exports.HttpsProxyAgent = require('./utilsBundleImpl').HttpsProxyAgent;
+const jpegjs = exports.jpegjs = require('./utilsBundleImpl').jpegjs;
+const lockfile = exports.lockfile = require('./utilsBundleImpl').lockfile;
+const mime = exports.mime = require('./utilsBundleImpl').mime;
+const minimatch = exports.minimatch = require('./utilsBundleImpl').minimatch;
+const open = exports.open = require('./utilsBundleImpl').open;
+const PNG = exports.PNG = require('./utilsBundleImpl').PNG;
+const program = exports.program = require('./utilsBundleImpl').program;
+const progress = exports.progress = require('./utilsBundleImpl').progress;
+const SocksProxyAgent = exports.SocksProxyAgent = require('./utilsBundleImpl').SocksProxyAgent;
+const ws = exports.ws = require('./utilsBundleImpl').ws;
+const wsServer = exports.wsServer = require('./utilsBundleImpl').wsServer;
+const wsReceiver = exports.wsReceiver = require('./utilsBundleImpl').wsReceiver;
+const wsSender = exports.wsSender = require('./utilsBundleImpl').wsSender;
+const StackUtils = require('./utilsBundleImpl').StackUtils;
+const stackUtils = new StackUtils({
+  internals: StackUtils.nodeInternals()
+});
+const nodeInternals = StackUtils.nodeInternals();
+const nodeMajorVersion = +process.versions.node.split('.')[0];
+function parseStackTraceLine(line) {
+  var _frame$file, _frame$file2;
+  if (!process.env.PWDEBUGIMPL && nodeMajorVersion < 16 && nodeInternals.some(internal => internal.test(line))) return null;
+  const frame = stackUtils.parseLine(line);
+  if (!frame) return null;
+  if (!process.env.PWDEBUGIMPL && ((_frame$file = frame.file) !== null && _frame$file !== void 0 && _frame$file.startsWith('internal') || (_frame$file2 = frame.file) !== null && _frame$file2 !== void 0 && _frame$file2.startsWith('node:'))) return null;
+  if (!frame.file) return null;
+  // ESM files return file:// URLs, see here: https://github.com/tapjs/stack-utils/issues/60
+  const file = frame.file.startsWith('file://') ? _url.default.fileURLToPath(frame.file) : _path.default.resolve(process.cwd(), frame.file);
+  return {
+    file,
+    line: frame.line || 0,
+    column: frame.column || 0,
+    function: frame.function
+  };
+}
+function ms(ms) {
+  if (!isFinite(ms)) return '-';
+  if (ms === 0) return '0ms';
+  if (ms < 1000) return ms.toFixed(0) + 'ms';
+  const seconds = ms / 1000;
+  if (seconds < 60) return seconds.toFixed(1) + 's';
+  const minutes = seconds / 60;
+  if (minutes < 60) return minutes.toFixed(1) + 'm';
+  const hours = minutes / 60;
+  if (hours < 24) return hours.toFixed(1) + 'h';
+  const days = hours / 24;
+  return days.toFixed(1) + 'd';
+}