148 lines
5.6 KiB
JavaScript
148 lines
5.6 KiB
JavaScript
|
"use strict";
|
||
|
|
||
|
Object.defineProperty(exports, "__esModule", {
|
||
|
value: true
|
||
|
});
|
||
|
exports.Browser = void 0;
|
||
|
var _artifact = require("./artifact");
|
||
|
var _browserContext = require("./browserContext");
|
||
|
var _download = require("./download");
|
||
|
var _instrumentation = require("./instrumentation");
|
||
|
var _page = require("./page");
|
||
|
var _socksClientCertificatesInterceptor = require("./socksClientCertificatesInterceptor");
|
||
|
/**
|
||
|
* 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 Browser extends _instrumentation.SdkObject {
|
||
|
constructor(parent, options) {
|
||
|
super(parent, 'browser');
|
||
|
this.options = void 0;
|
||
|
this._downloads = new Map();
|
||
|
this._defaultContext = null;
|
||
|
this._startedClosing = false;
|
||
|
this._idToVideo = new Map();
|
||
|
this._contextForReuse = void 0;
|
||
|
this._closeReason = void 0;
|
||
|
this._isCollocatedWithServer = true;
|
||
|
this.attribution.browser = this;
|
||
|
this.options = options;
|
||
|
this.instrumentation.onBrowserOpen(this);
|
||
|
}
|
||
|
async newContext(metadata, options) {
|
||
|
var _options$clientCertif;
|
||
|
(0, _browserContext.validateBrowserContextOptions)(options, this.options);
|
||
|
let clientCertificatesProxy;
|
||
|
if ((_options$clientCertif = options.clientCertificates) !== null && _options$clientCertif !== void 0 && _options$clientCertif.length) {
|
||
|
clientCertificatesProxy = new _socksClientCertificatesInterceptor.ClientCertificatesProxy(options);
|
||
|
options = {
|
||
|
...options
|
||
|
};
|
||
|
options.proxyOverride = await clientCertificatesProxy.listen();
|
||
|
options.internalIgnoreHTTPSErrors = true;
|
||
|
}
|
||
|
let context;
|
||
|
try {
|
||
|
context = await this.doCreateNewContext(options);
|
||
|
} catch (error) {
|
||
|
var _clientCertificatesPr;
|
||
|
await ((_clientCertificatesPr = clientCertificatesProxy) === null || _clientCertificatesPr === void 0 ? void 0 : _clientCertificatesPr.close());
|
||
|
throw error;
|
||
|
}
|
||
|
context._clientCertificatesProxy = clientCertificatesProxy;
|
||
|
if (options.storageState) await context.setStorageState(metadata, options.storageState);
|
||
|
return context;
|
||
|
}
|
||
|
async newContextForReuse(params, metadata) {
|
||
|
const hash = _browserContext.BrowserContext.reusableContextHash(params);
|
||
|
if (!this._contextForReuse || hash !== this._contextForReuse.hash || !this._contextForReuse.context.canResetForReuse()) {
|
||
|
if (this._contextForReuse) await this._contextForReuse.context.close({
|
||
|
reason: 'Context reused'
|
||
|
});
|
||
|
this._contextForReuse = {
|
||
|
context: await this.newContext(metadata, params),
|
||
|
hash
|
||
|
};
|
||
|
return {
|
||
|
context: this._contextForReuse.context,
|
||
|
needsReset: false
|
||
|
};
|
||
|
}
|
||
|
await this._contextForReuse.context.stopPendingOperations('Context recreated');
|
||
|
return {
|
||
|
context: this._contextForReuse.context,
|
||
|
needsReset: true
|
||
|
};
|
||
|
}
|
||
|
async stopPendingOperations(reason) {
|
||
|
var _this$_contextForReus;
|
||
|
await ((_this$_contextForReus = this._contextForReuse) === null || _this$_contextForReus === void 0 || (_this$_contextForReus = _this$_contextForReus.context) === null || _this$_contextForReus === void 0 ? void 0 : _this$_contextForReus.stopPendingOperations(reason));
|
||
|
}
|
||
|
_downloadCreated(page, uuid, url, suggestedFilename) {
|
||
|
const download = new _download.Download(page, this.options.downloadsPath || '', uuid, url, suggestedFilename);
|
||
|
this._downloads.set(uuid, download);
|
||
|
}
|
||
|
_downloadFilenameSuggested(uuid, suggestedFilename) {
|
||
|
const download = this._downloads.get(uuid);
|
||
|
if (!download) return;
|
||
|
download._filenameSuggested(suggestedFilename);
|
||
|
}
|
||
|
_downloadFinished(uuid, error) {
|
||
|
const download = this._downloads.get(uuid);
|
||
|
if (!download) return;
|
||
|
download.artifact.reportFinished(error ? new Error(error) : undefined);
|
||
|
this._downloads.delete(uuid);
|
||
|
}
|
||
|
_videoStarted(context, videoId, path, pageOrError) {
|
||
|
const artifact = new _artifact.Artifact(context, path);
|
||
|
this._idToVideo.set(videoId, {
|
||
|
context,
|
||
|
artifact
|
||
|
});
|
||
|
pageOrError.then(page => {
|
||
|
if (page instanceof _page.Page) {
|
||
|
page._video = artifact;
|
||
|
page.emitOnContext(_browserContext.BrowserContext.Events.VideoStarted, artifact);
|
||
|
page.emit(_page.Page.Events.Video, artifact);
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
_takeVideo(videoId) {
|
||
|
const video = this._idToVideo.get(videoId);
|
||
|
this._idToVideo.delete(videoId);
|
||
|
return video === null || video === void 0 ? void 0 : video.artifact;
|
||
|
}
|
||
|
_didClose() {
|
||
|
for (const context of this.contexts()) context._browserClosed();
|
||
|
if (this._defaultContext) this._defaultContext._browserClosed();
|
||
|
this.emit(Browser.Events.Disconnected);
|
||
|
this.instrumentation.onBrowserClose(this);
|
||
|
}
|
||
|
async close(options) {
|
||
|
if (!this._startedClosing) {
|
||
|
if (options.reason) this._closeReason = options.reason;
|
||
|
this._startedClosing = true;
|
||
|
await this.options.browserProcess.close();
|
||
|
}
|
||
|
if (this.isConnected()) await new Promise(x => this.once(Browser.Events.Disconnected, x));
|
||
|
}
|
||
|
async killForTests() {
|
||
|
await this.options.browserProcess.kill();
|
||
|
}
|
||
|
}
|
||
|
exports.Browser = Browser;
|
||
|
Browser.Events = {
|
||
|
Disconnected: 'disconnected'
|
||
|
};
|