@ai-coding-labs/playwright-mcp-plus
Version:
Enhanced Playwright Tools for MCP with Project Session Isolation
112 lines (111 loc) • 3.36 kB
JavaScript
/**
* 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.
*/
export class ManualPromise extends Promise {
_resolve;
_reject;
_isDone;
constructor() {
let resolve;
let reject;
super((f, r) => {
resolve = f;
reject = r;
});
this._isDone = false;
this._resolve = resolve;
this._reject = reject;
}
isDone() {
return this._isDone;
}
resolve(t) {
this._isDone = true;
this._resolve(t);
}
reject(e) {
this._isDone = true;
this._reject(e);
}
static get [Symbol.species]() {
return Promise;
}
get [Symbol.toStringTag]() {
return 'ManualPromise';
}
}
export class LongStandingScope {
_terminateError;
_closeError;
_terminatePromises = new Map();
_isClosed = false;
reject(error) {
this._isClosed = true;
this._terminateError = error;
for (const p of this._terminatePromises.keys())
p.resolve(error);
}
close(error) {
this._isClosed = true;
this._closeError = error;
for (const [p, frames] of this._terminatePromises)
p.resolve(cloneError(error, frames));
}
isClosed() {
return this._isClosed;
}
static async raceMultiple(scopes, promise) {
return Promise.race(scopes.map(s => s.race(promise)));
}
async race(promise) {
return this._race(Array.isArray(promise) ? promise : [promise], false);
}
async safeRace(promise, defaultValue) {
return this._race([promise], true, defaultValue);
}
async _race(promises, safe, defaultValue) {
const terminatePromise = new ManualPromise();
const frames = captureRawStack();
if (this._terminateError)
terminatePromise.resolve(this._terminateError);
if (this._closeError)
terminatePromise.resolve(cloneError(this._closeError, frames));
this._terminatePromises.set(terminatePromise, frames);
try {
return await Promise.race([
terminatePromise.then(e => safe ? defaultValue : Promise.reject(e)),
...promises
]);
}
finally {
this._terminatePromises.delete(terminatePromise);
}
}
}
function cloneError(error, frames) {
const clone = new Error();
clone.name = error.name;
clone.message = error.message;
clone.stack = [error.name + ':' + error.message, ...frames].join('\n');
return clone;
}
function captureRawStack() {
const stackTraceLimit = Error.stackTraceLimit;
Error.stackTraceLimit = 50;
const error = new Error();
const stack = error.stack || '';
Error.stackTraceLimit = stackTraceLimit;
return stack.split('\n');
}