UNPKG

wxt

Version:

⚡ Next-gen Web Extension Framework

135 lines (134 loc) 5.51 kB
/** @module wxt/utils/content-script-context */ import { ContentScriptDefinition } from '../types'; import { WxtLocationChangeEvent } from './internal/custom-events'; /** * Implements [`AbortController`](https://developer.mozilla.org/en-US/docs/Web/API/AbortController). * Used to detect and stop content script code when the script is invalidated. * * It also provides several utilities like `ctx.setTimeout` and `ctx.setInterval` that should be used in * content scripts instead of `window.setTimeout` or `window.setInterval`. * * To create context for testing, you can use the class's constructor: * * ```ts * import { ContentScriptContext } from 'wxt/utils/content-scripts-context'; * * test("storage listener should be removed when context is invalidated", () => { * const ctx = new ContentScriptContext('test'); * const item = storage.defineItem("local:count", { defaultValue: 0 }); * const watcher = vi.fn(); * * const unwatch = item.watch(watcher); * ctx.onInvalidated(unwatch); // Listen for invalidate here * * await item.setValue(1); * expect(watcher).toBeCalledTimes(1); * expect(watcher).toBeCalledWith(1, 0); * * ctx.notifyInvalidated(); // Use this function to invalidate the context * await item.setValue(2); * expect(watcher).toBeCalledTimes(1); * }); * ``` */ export declare class ContentScriptContext implements AbortController { private readonly contentScriptName; readonly options?: Omit<ContentScriptDefinition, "main"> | undefined; private static SCRIPT_STARTED_MESSAGE_TYPE; private isTopFrame; private abortController; private locationWatcher; private receivedMessageIds; constructor(contentScriptName: string, options?: Omit<ContentScriptDefinition, "main"> | undefined); get signal(): AbortSignal; abort(reason?: any): void; get isInvalid(): boolean; get isValid(): boolean; /** * Add a listener that is called when the content script's context is invalidated. * * @returns A function to remove the listener. * * @example * browser.runtime.onMessage.addListener(cb); * const removeInvalidatedListener = ctx.onInvalidated(() => { * browser.runtime.onMessage.removeListener(cb); * }) * // ... * removeInvalidatedListener(); */ onInvalidated(cb: () => void): () => void; /** * Return a promise that never resolves. Useful if you have an async function that shouldn't run * after the context is expired. * * @example * const getValueFromStorage = async () => { * if (ctx.isInvalid) return ctx.block(); * * // ... * } */ block<T>(): Promise<T>; /** * Wrapper around `window.setInterval` that automatically clears the interval when invalidated. * * Intervals can be cleared by calling the normal `clearInterval` function. */ setInterval(handler: () => void, timeout?: number): number; /** * Wrapper around `window.setTimeout` that automatically clears the interval when invalidated. * * Timeouts can be cleared by calling the normal `setTimeout` function. */ setTimeout(handler: () => void, timeout?: number): number; /** * Wrapper around `window.requestAnimationFrame` that automatically cancels the request when * invalidated. * * Callbacks can be canceled by calling the normal `cancelAnimationFrame` function. */ requestAnimationFrame(callback: FrameRequestCallback): number; /** * Wrapper around `window.requestIdleCallback` that automatically cancels the request when * invalidated. * * Callbacks can be canceled by calling the normal `cancelIdleCallback` function. */ requestIdleCallback(callback: IdleRequestCallback, options?: IdleRequestOptions): number; /** * Call `target.addEventListener` and remove the event listener when the context is invalidated. * * Listeners can be canceled by calling the normal `removeEventListener` function. * * Includes additional events useful for content scripts: * * - `"wxt:locationchange"` - Triggered when HTML5 history mode is used to change URL. Content * scripts are not reloaded when navigating this way, so this can be used to reset the content * script state on URL change, or run custom code. * * @example * ctx.addEventListener(document, "visibilitychange", () => { * // ... * }); * ctx.addEventListener(window, "wxt:locationchange", () => { * // ... * }); */ addEventListener<TType extends keyof WxtWindowEventMap>(target: Window, type: TType, handler: (event: WxtWindowEventMap[TType]) => void, options?: AddEventListenerOptions): void; addEventListener<TType extends keyof DocumentEventMap>(target: Document, type: TType, handler: (event: DocumentEventMap[TType]) => void, options?: AddEventListenerOptions): void; addEventListener<TTarget extends EventTarget>(target: TTarget, ...params: Parameters<TTarget['addEventListener']>): void; /** * @internal * Abort the abort controller and execute all `onInvalidated` listeners. */ notifyInvalidated(): void; stopOldScripts(): void; verifyScriptStartedEvent(event: MessageEvent): boolean; listenForNewerScripts(options?: { ignoreFirstEvent?: boolean; }): void; } export interface WxtWindowEventMap extends WindowEventMap { 'wxt:locationchange': WxtLocationChangeEvent; }