patchright 1.48.0__py3-none-win32.whl

patchright/driver/package/lib/server/dispatchers/webSocketRouteDispatcher.js

@@ -0,0 +1,189 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.WebSocketRouteDispatcher = void 0;
+var _page = require("../page");
+var _dispatcher = require("./dispatcher");
+var _utils = require("../../utils");
+var _pageDispatcher = require("./pageDispatcher");
+var webSocketMockSource = _interopRequireWildcard(require("../../generated/webSocketMockSource"));
+var _eventsHelper = require("../../utils/eventsHelper");
+var _class;
+/**
+ * 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.
+ */
+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; }
+const kBindingInstalledSymbol = Symbol('webSocketRouteBindingInstalled');
+const kInitScriptInstalledSymbol = Symbol('webSocketRouteInitScriptInstalled');
+class WebSocketRouteDispatcher extends _dispatcher.Dispatcher {
+  constructor(scope, id, url, frame) {
+    super(scope, {
+      guid: 'webSocketRoute@' + (0, _utils.createGuid)()
+    }, 'WebSocketRoute', {
+      url
+    });
+    this._type_WebSocketRoute = true;
+    this._id = void 0;
+    this._frame = void 0;
+    this._id = id;
+    this._frame = frame;
+    this._eventListeners.push(
+    // When the frame navigates or detaches, there will be no more communication
+    // from the mock websocket, so pretend like it was closed.
+    _eventsHelper.eventsHelper.addEventListener(frame._page, _page.Page.Events.InternalFrameNavigatedToNewDocument, frame => {
+      if (frame === this._frame) this._executionContextGone();
+    }), _eventsHelper.eventsHelper.addEventListener(frame._page, _page.Page.Events.FrameDetached, frame => {
+      if (frame === this._frame) this._executionContextGone();
+    }), _eventsHelper.eventsHelper.addEventListener(frame._page, _page.Page.Events.Close, () => this._executionContextGone()), _eventsHelper.eventsHelper.addEventListener(frame._page, _page.Page.Events.Crash, () => this._executionContextGone()));
+    WebSocketRouteDispatcher._idToDispatcher.set(this._id, this);
+    scope._dispatchEvent('webSocketRoute', {
+      webSocketRoute: this
+    });
+  }
+  static async installIfNeeded(contextDispatcher, target) {
+    const context = target instanceof _page.Page ? target.context() : target;
+    if (!context[kBindingInstalledSymbol]) {
+      context[kBindingInstalledSymbol] = true;
+      await context.exposeBinding('__pwWebSocketBinding', false, (source, payload) => {
+        if (payload.type === 'onCreate') {
+          const pageDispatcher = _pageDispatcher.PageDispatcher.fromNullable(contextDispatcher, source.page);
+          let scope;
+          if (pageDispatcher && matchesPattern(pageDispatcher, context._options.baseURL, payload.url)) scope = pageDispatcher;else if (matchesPattern(contextDispatcher, context._options.baseURL, payload.url)) scope = contextDispatcher;
+          if (scope) {
+            new WebSocketRouteDispatcher(scope, payload.id, payload.url, source.frame);
+          } else {
+            const request = {
+              id: payload.id,
+              type: 'passthrough'
+            };
+            source.frame.evaluateExpression(`globalThis.__pwWebSocketDispatch(${JSON.stringify(request)})`).catch(() => {});
+          }
+          return;
+        }
+        const dispatcher = WebSocketRouteDispatcher._idToDispatcher.get(payload.id);
+        if (payload.type === 'onMessageFromPage') dispatcher === null || dispatcher === void 0 || dispatcher._dispatchEvent('messageFromPage', {
+          message: payload.data.data,
+          isBase64: payload.data.isBase64
+        });
+        if (payload.type === 'onMessageFromServer') dispatcher === null || dispatcher === void 0 || dispatcher._dispatchEvent('messageFromServer', {
+          message: payload.data.data,
+          isBase64: payload.data.isBase64
+        });
+        if (payload.type === 'onClosePage') dispatcher === null || dispatcher === void 0 || dispatcher._dispatchEvent('closePage', {
+          code: payload.code,
+          reason: payload.reason,
+          wasClean: payload.wasClean
+        });
+        if (payload.type === 'onCloseServer') dispatcher === null || dispatcher === void 0 || dispatcher._dispatchEvent('closeServer', {
+          code: payload.code,
+          reason: payload.reason,
+          wasClean: payload.wasClean
+        });
+      });
+    }
+    if (!target[kInitScriptInstalledSymbol]) {
+      target[kInitScriptInstalledSymbol] = true;
+      await target.addInitScript(`
+        (() => {
+          const module = {};
+          ${webSocketMockSource.source}
+          (module.exports.inject())(globalThis);
+        })();
+      `);
+    }
+  }
+  async connect(params) {
+    await this._evaluateAPIRequest({
+      id: this._id,
+      type: 'connect'
+    });
+  }
+  async ensureOpened(params) {
+    await this._evaluateAPIRequest({
+      id: this._id,
+      type: 'ensureOpened'
+    });
+  }
+  async sendToPage(params) {
+    await this._evaluateAPIRequest({
+      id: this._id,
+      type: 'sendToPage',
+      data: {
+        data: params.message,
+        isBase64: params.isBase64
+      }
+    });
+  }
+  async sendToServer(params) {
+    await this._evaluateAPIRequest({
+      id: this._id,
+      type: 'sendToServer',
+      data: {
+        data: params.message,
+        isBase64: params.isBase64
+      }
+    });
+  }
+  async closePage(params) {
+    await this._evaluateAPIRequest({
+      id: this._id,
+      type: 'closePage',
+      code: params.code,
+      reason: params.reason,
+      wasClean: params.wasClean
+    });
+  }
+  async closeServer(params) {
+    await this._evaluateAPIRequest({
+      id: this._id,
+      type: 'closeServer',
+      code: params.code,
+      reason: params.reason,
+      wasClean: params.wasClean
+    });
+  }
+  async _evaluateAPIRequest(request) {
+    await this._frame.evaluateExpression(`globalThis.__pwWebSocketDispatch(${JSON.stringify(request)})`).catch(() => {});
+  }
+  _onDispose() {
+    WebSocketRouteDispatcher._idToDispatcher.delete(this._id);
+  }
+  _executionContextGone() {
+    // We could enter here after being disposed upon page closure:
+    // - first from the recursive dispose inintiated by PageDispatcher;
+    // - then from our own page.on('close') listener.
+    if (!this._disposed) {
+      this._dispatchEvent('closePage', {
+        wasClean: true
+      });
+      this._dispatchEvent('closeServer', {
+        wasClean: true
+      });
+    }
+  }
+}
+exports.WebSocketRouteDispatcher = WebSocketRouteDispatcher;
+_class = WebSocketRouteDispatcher;
+WebSocketRouteDispatcher._idToDispatcher = new Map();
+function matchesPattern(dispatcher, baseURL, url) {
+  for (const pattern of dispatcher._webSocketInterceptionPatterns || []) {
+    const urlMatch = pattern.regexSource ? new RegExp(pattern.regexSource, pattern.regexFlags) : pattern.glob;
+    if ((0, _utils.urlMatches)(baseURL, url, urlMatch)) return true;
+  }
+  return false;
+}

patchright/driver/package/lib/server/dispatchers/writableStreamDispatcher.js

@@ -0,0 +1,58 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.WritableStreamDispatcher = void 0;
+var _dispatcher = require("./dispatcher");
+var fs = _interopRequireWildcard(require("fs"));
+var _utils = require("../../utils");
+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 WritableStreamDispatcher extends _dispatcher.Dispatcher {
+  constructor(scope, streamOrDirectory, lastModifiedMs) {
+    super(scope, {
+      guid: 'writableStream@' + (0, _utils.createGuid)(),
+      streamOrDirectory
+    }, 'WritableStream', {});
+    this._type_WritableStream = true;
+    this._lastModifiedMs = void 0;
+    this._lastModifiedMs = lastModifiedMs;
+  }
+  async write(params) {
+    if (typeof this._object.streamOrDirectory === 'string') throw new Error('Cannot write to a directory');
+    const stream = this._object.streamOrDirectory;
+    await new Promise((fulfill, reject) => {
+      stream.write(params.binary, error => {
+        if (error) reject(error);else fulfill();
+      });
+    });
+  }
+  async close() {
+    if (typeof this._object.streamOrDirectory === 'string') throw new Error('Cannot close a directory');
+    const stream = this._object.streamOrDirectory;
+    await new Promise(fulfill => stream.end(fulfill));
+    if (this._lastModifiedMs) await fs.promises.utimes(this.path(), new Date(this._lastModifiedMs), new Date(this._lastModifiedMs));
+  }
+  path() {
+    if (typeof this._object.streamOrDirectory === 'string') return this._object.streamOrDirectory;
+    return this._object.streamOrDirectory.path;
+  }
+}
+exports.WritableStreamDispatcher = WritableStreamDispatcher;