131 lines
6.3 KiB
JavaScript
131 lines
6.3 KiB
JavaScript
/**
|
|
* Copyright 2021 Google LLC.
|
|
* 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.
|
|
*/
|
|
import { EventEmitter } from '../utils/EventEmitter.js';
|
|
import { LogType } from '../utils/log.js';
|
|
import { ProcessingQueue } from '../utils/ProcessingQueue.js';
|
|
import { CommandProcessor } from './CommandProcessor.js';
|
|
import { BluetoothProcessor } from './modules/bluetooth/BluetoothProcessor.js';
|
|
import { CdpTargetManager } from './modules/cdp/CdpTargetManager.js';
|
|
import { BrowsingContextStorage } from './modules/context/BrowsingContextStorage.js';
|
|
import { NetworkStorage } from './modules/network/NetworkStorage.js';
|
|
import { PreloadScriptStorage } from './modules/script/PreloadScriptStorage.js';
|
|
import { RealmStorage } from './modules/script/RealmStorage.js';
|
|
import { EventManager, } from './modules/session/EventManager.js';
|
|
export class BidiServer extends EventEmitter {
|
|
#messageQueue;
|
|
#transport;
|
|
#commandProcessor;
|
|
#eventManager;
|
|
#browsingContextStorage = new BrowsingContextStorage();
|
|
#realmStorage = new RealmStorage();
|
|
#preloadScriptStorage = new PreloadScriptStorage();
|
|
#bluetoothProcessor;
|
|
#logger;
|
|
#handleIncomingMessage = (message) => {
|
|
void this.#commandProcessor.processCommand(message).catch((error) => {
|
|
this.#logger?.(LogType.debugError, error);
|
|
});
|
|
};
|
|
#processOutgoingMessage = async (messageEntry) => {
|
|
const message = messageEntry.message;
|
|
if (messageEntry.channel !== null) {
|
|
message['channel'] = messageEntry.channel;
|
|
}
|
|
await this.#transport.sendMessage(message);
|
|
};
|
|
constructor(bidiTransport, cdpConnection, browserCdpClient, selfTargetId, defaultUserContextId, parser, logger) {
|
|
super();
|
|
this.#logger = logger;
|
|
this.#messageQueue = new ProcessingQueue(this.#processOutgoingMessage, this.#logger);
|
|
this.#transport = bidiTransport;
|
|
this.#transport.setOnMessage(this.#handleIncomingMessage);
|
|
this.#eventManager = new EventManager(this.#browsingContextStorage);
|
|
const networkStorage = new NetworkStorage(this.#eventManager, this.#browsingContextStorage, browserCdpClient, logger);
|
|
this.#bluetoothProcessor = new BluetoothProcessor(this.#eventManager, this.#browsingContextStorage);
|
|
this.#commandProcessor = new CommandProcessor(cdpConnection, browserCdpClient, this.#eventManager, this.#browsingContextStorage, this.#realmStorage, this.#preloadScriptStorage, networkStorage, this.#bluetoothProcessor, parser, async (options) => {
|
|
// This is required to ignore certificate errors when service worker is fetched.
|
|
await browserCdpClient.sendCommand('Security.setIgnoreCertificateErrors', {
|
|
ignore: options.acceptInsecureCerts ?? false,
|
|
});
|
|
new CdpTargetManager(cdpConnection, browserCdpClient, selfTargetId, this.#eventManager, this.#browsingContextStorage, this.#realmStorage, networkStorage, this.#bluetoothProcessor, this.#preloadScriptStorage, defaultUserContextId, options?.['goog:prerenderingDisabled'] ?? false, options?.unhandledPromptBehavior, logger);
|
|
// Needed to get events about new targets.
|
|
await browserCdpClient.sendCommand('Target.setDiscoverTargets', {
|
|
discover: true,
|
|
});
|
|
// Needed to automatically attach to new targets.
|
|
await browserCdpClient.sendCommand('Target.setAutoAttach', {
|
|
autoAttach: true,
|
|
waitForDebuggerOnStart: true,
|
|
flatten: true,
|
|
// Browser session should attach to tab instead of the page, so that
|
|
// prerendering is not blocked.
|
|
filter: [
|
|
{
|
|
type: 'page',
|
|
exclude: true,
|
|
},
|
|
{},
|
|
],
|
|
});
|
|
await this.#topLevelContextsLoaded();
|
|
}, this.#logger);
|
|
this.#eventManager.on("event" /* EventManagerEvents.Event */, ({ message, event }) => {
|
|
this.emitOutgoingMessage(message, event);
|
|
});
|
|
this.#commandProcessor.on("response" /* CommandProcessorEvents.Response */, ({ message, event }) => {
|
|
this.emitOutgoingMessage(message, event);
|
|
});
|
|
}
|
|
/**
|
|
* Creates and starts BiDi Mapper instance.
|
|
*/
|
|
static async createAndStart(bidiTransport, cdpConnection, browserCdpClient, selfTargetId, parser, logger) {
|
|
// The default context is not exposed in Target.getBrowserContexts but can
|
|
// be observed via Target.getTargets. To determine the default browser
|
|
// context, we check which one is mentioned in Target.getTargets and not in
|
|
// Target.getBrowserContexts.
|
|
const [{ browserContextIds }, { targetInfos }] = await Promise.all([
|
|
browserCdpClient.sendCommand('Target.getBrowserContexts'),
|
|
browserCdpClient.sendCommand('Target.getTargets'),
|
|
]);
|
|
let defaultUserContextId = 'default';
|
|
for (const info of targetInfos) {
|
|
if (info.browserContextId &&
|
|
!browserContextIds.includes(info.browserContextId)) {
|
|
defaultUserContextId = info.browserContextId;
|
|
break;
|
|
}
|
|
}
|
|
const server = new BidiServer(bidiTransport, cdpConnection, browserCdpClient, selfTargetId, defaultUserContextId, parser, logger);
|
|
return server;
|
|
}
|
|
/**
|
|
* Sends BiDi message.
|
|
*/
|
|
emitOutgoingMessage(messageEntry, event) {
|
|
this.#messageQueue.add(messageEntry, event);
|
|
}
|
|
close() {
|
|
this.#transport.close();
|
|
}
|
|
async #topLevelContextsLoaded() {
|
|
await Promise.all(this.#browsingContextStorage
|
|
.getTopLevelContexts()
|
|
.map((c) => c.lifecycleLoaded()));
|
|
}
|
|
}
|
|
//# sourceMappingURL=BidiServer.js.map
|