mirror of
https://github.com/microsoft/playwright.git
synced 2025-06-26 21:40:17 +00:00
65 lines
2.8 KiB
TypeScript
65 lines
2.8 KiB
TypeScript
/**
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
*
|
|
* 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 type { ParsedStackTrace } from '../utils/stackTrace';
|
|
import type { BrowserContext } from './browserContext';
|
|
import type { APIRequestContext } from './fetch';
|
|
|
|
export interface ClientInstrumentation {
|
|
addListener(listener: ClientInstrumentationListener): void;
|
|
removeListener(listener: ClientInstrumentationListener): void;
|
|
removeAllListeners(): void;
|
|
onApiCallBegin(apiCall: string, params: Record<string, any>, stackTrace: ParsedStackTrace | null, wallTime: number, userData: any): void;
|
|
onApiCallEnd(userData: any, error?: Error): void;
|
|
onDidCreateBrowserContext(context: BrowserContext): Promise<void>;
|
|
onDidCreateRequestContext(context: APIRequestContext): Promise<void>;
|
|
onWillPause(): void;
|
|
onWillCloseBrowserContext(context: BrowserContext): Promise<void>;
|
|
onWillCloseRequestContext(context: APIRequestContext): Promise<void>;
|
|
}
|
|
|
|
export interface ClientInstrumentationListener {
|
|
onApiCallBegin?(apiCall: string, params: Record<string, any>, stackTrace: ParsedStackTrace | null, wallTime: number, userData: any): void;
|
|
onApiCallEnd?(userData: any, error?: Error): void;
|
|
onDidCreateBrowserContext?(context: BrowserContext): Promise<void>;
|
|
onDidCreateRequestContext?(context: APIRequestContext): Promise<void>;
|
|
onWillPause?(): void;
|
|
onWillCloseBrowserContext?(context: BrowserContext): Promise<void>;
|
|
onWillCloseRequestContext?(context: APIRequestContext): Promise<void>;
|
|
}
|
|
|
|
export function createInstrumentation(): ClientInstrumentation {
|
|
const listeners: ClientInstrumentationListener[] = [];
|
|
return new Proxy({}, {
|
|
get: (obj: any, prop: string | symbol) => {
|
|
if (typeof prop !== 'string')
|
|
return obj[prop];
|
|
if (prop === 'addListener')
|
|
return (listener: ClientInstrumentationListener) => listeners.push(listener);
|
|
if (prop === 'removeListener')
|
|
return (listener: ClientInstrumentationListener) => listeners.splice(listeners.indexOf(listener), 1);
|
|
if (prop === 'removeAllListeners')
|
|
return () => listeners.splice(0, listeners.length);
|
|
if (!prop.startsWith('on'))
|
|
return obj[prop];
|
|
return async (...params: any[]) => {
|
|
for (const listener of listeners)
|
|
await (listener as any)[prop]?.(...params);
|
|
};
|
|
},
|
|
});
|
|
}
|