/*--------------------------------------------------------------------------------------------- * Copyright (c) Microsoft Corporation. All rights reserved. * Licensed under the Source EULA. See License.txt in the project root for license information. *--------------------------------------------------------------------------------------------*/ import { CancellationToken } from 'vs/base/common/cancellation'; import { onUnexpectedError } from 'vs/base/common/errors'; import { once as onceFn } from 'vs/base/common/functional'; import { combinedDisposable, Disposable, DisposableStore, IDisposable, SafeDisposable, toDisposable } from 'vs/base/common/lifecycle'; import { LinkedList } from 'vs/base/common/linkedList'; import { IObservable, IObserver } from 'vs/base/common/observable'; import { StopWatch } from 'vs/base/common/stopwatch'; // ----------------------------------------------------------------------------------------------------------------------- // Uncomment the next line to print warnings whenever an emitter with listeners is disposed. That is a sign of code smell. // ----------------------------------------------------------------------------------------------------------------------- const _enableDisposeWithListenerWarning = false; // _enableDisposeWithListenerWarning = Boolean("TRUE"); // causes a linter warning so that it cannot be pushed // ----------------------------------------------------------------------------------------------------------------------- // Uncomment the next line to print warnings whenever a snapshotted event is used repeatedly without cleanup. // See https://github.com/microsoft/vscode/issues/142851 // ----------------------------------------------------------------------------------------------------------------------- const _enableSnapshotPotentialLeakWarning = false; // _enableSnapshotPotentialLeakWarning = Boolean("TRUE"); // causes a linter warning so that it cannot be pushed /** * To an event a function with one or zero parameters * can be subscribed. The event is the subscriber function itself. */ export interface Event { (listener: (e: T) => any, thisArgs?: any, disposables?: IDisposable[] | DisposableStore): IDisposable; } export namespace Event { export const None: Event = () => Disposable.None; function _addLeakageTraceLogic(options: EmitterOptions) { if (_enableSnapshotPotentialLeakWarning) { const { onListenerDidAdd: origListenerDidAdd } = options; const stack = Stacktrace.create(); let count = 0; options.onListenerDidAdd = () => { if (++count === 2) { console.warn('snapshotted emitter LIKELY used public and SHOULD HAVE BEEN created with DisposableStore. snapshotted here'); stack.print(); } origListenerDidAdd?.(); }; } } /** * Given an event, returns another event which only fires once. */ export function once(event: Event): Event { return (listener, thisArgs = null, disposables?) => { // we need this, in case the event fires during the listener call let didFire = false; let result: IDisposable | undefined = undefined; result = event(e => { if (didFire) { return; } else if (result) { result.dispose(); } else { didFire = true; } return listener.call(thisArgs, e); }, null, disposables); if (didFire) { result.dispose(); } return result; }; } /** * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the * returned event causes this utility to leak a listener on the original event. */ export function map(event: Event, map: (i: I) => O, disposable?: DisposableStore): Event { return snapshot((listener, thisArgs = null, disposables?) => event(i => listener.call(thisArgs, map(i)), null, disposables), disposable); } /** * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the * returned event causes this utility to leak a listener on the original event. */ export function forEach(event: Event, each: (i: I) => void, disposable?: DisposableStore): Event { return snapshot((listener, thisArgs = null, disposables?) => event(i => { each(i); listener.call(thisArgs, i); }, null, disposables), disposable); } /** * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the * returned event causes this utility to leak a listener on the original event. */ export function filter(event: Event, filter: (e: T | U) => e is T, disposable?: DisposableStore): Event; export function filter(event: Event, filter: (e: T) => boolean, disposable?: DisposableStore): Event; export function filter(event: Event, filter: (e: T | R) => e is R, disposable?: DisposableStore): Event; export function filter(event: Event, filter: (e: T) => boolean, disposable?: DisposableStore): Event { return snapshot((listener, thisArgs = null, disposables?) => event(e => filter(e) && listener.call(thisArgs, e), null, disposables), disposable); } /** * Given an event, returns the same event but typed as `Event`. */ export function signal(event: Event): Event { return event as Event as Event; } /** * Given a collection of events, returns a single event which emits * whenever any of the provided events emit. */ export function any(...events: Event[]): Event; export function any(...events: Event[]): Event; export function any(...events: Event[]): Event { return (listener, thisArgs = null, disposables?) => combinedDisposable(...events.map(event => event(e => listener.call(thisArgs, e), null, disposables))); } /** * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the * returned event causes this utility to leak a listener on the original event. */ export function reduce(event: Event, merge: (last: O | undefined, event: I) => O, initial?: O, disposable?: DisposableStore): Event { let output: O | undefined = initial; return map(event, e => { output = merge(output, e); return output; }, disposable); } function snapshot(event: Event, disposable: DisposableStore | undefined): Event { let listener: IDisposable | undefined; const options: EmitterOptions | undefined = { onFirstListenerAdd() { listener = event(emitter.fire, emitter); }, onLastListenerRemove() { listener?.dispose(); } }; if (!disposable) { _addLeakageTraceLogic(options); } const emitter = new Emitter(options); disposable?.add(emitter); return emitter.event; } /** * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the * returned event causes this utility to leak a listener on the original event. */ export function debounce(event: Event, merge: (last: T | undefined, event: T) => T, delay?: number, leading?: boolean, leakWarningThreshold?: number, disposable?: DisposableStore): Event; /** * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the * returned event causes this utility to leak a listener on the original event. */ export function debounce(event: Event, merge: (last: O | undefined, event: I) => O, delay?: number, leading?: boolean, leakWarningThreshold?: number, disposable?: DisposableStore): Event; export function debounce(event: Event, merge: (last: O | undefined, event: I) => O, delay: number = 100, leading = false, leakWarningThreshold?: number, disposable?: DisposableStore): Event { let subscription: IDisposable; let output: O | undefined = undefined; let handle: any = undefined; let numDebouncedCalls = 0; const options: EmitterOptions | undefined = { leakWarningThreshold, onFirstListenerAdd() { subscription = event(cur => { numDebouncedCalls++; output = merge(output, cur); if (leading && !handle) { emitter.fire(output); output = undefined; } clearTimeout(handle); handle = setTimeout(() => { const _output = output; output = undefined; handle = undefined; if (!leading || numDebouncedCalls > 1) { emitter.fire(_output!); } numDebouncedCalls = 0; }, delay); }); }, onLastListenerRemove() { subscription.dispose(); } }; if (!disposable) { _addLeakageTraceLogic(options); } const emitter = new Emitter(options); disposable?.add(emitter); return emitter.event; } /** * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the * returned event causes this utility to leak a listener on the original event. */ export function latch(event: Event, equals: (a: T, b: T) => boolean = (a, b) => a === b, disposable?: DisposableStore): Event { let firstCall = true; let cache: T; return filter(event, value => { const shouldEmit = firstCall || !equals(value, cache); firstCall = false; cache = value; return shouldEmit; }, disposable); } /** * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the * returned event causes this utility to leak a listener on the original event. */ export function split(event: Event, isT: (e: T | U) => e is T, disposable?: DisposableStore): [Event, Event] { return [ Event.filter(event, isT, disposable), Event.filter(event, e => !isT(e), disposable) as Event, ]; } /** * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the * returned event causes this utility to leak a listener on the original event. */ export function buffer(event: Event, flushAfterTimeout = false, _buffer: T[] = []): Event { let buffer: T[] | null = _buffer.slice(); let listener: IDisposable | null = event(e => { if (buffer) { buffer.push(e); } else { emitter.fire(e); } }); const flush = () => { buffer?.forEach(e => emitter.fire(e)); buffer = null; }; const emitter = new Emitter({ onFirstListenerAdd() { if (!listener) { listener = event(e => emitter.fire(e)); } }, onFirstListenerDidAdd() { if (buffer) { if (flushAfterTimeout) { setTimeout(flush); } else { flush(); } } }, onLastListenerRemove() { if (listener) { listener.dispose(); } listener = null; } }); return emitter.event; } export interface IChainableEvent extends IDisposable { event: Event; map(fn: (i: T) => O): IChainableEvent; forEach(fn: (i: T) => void): IChainableEvent; filter(fn: (e: T) => boolean): IChainableEvent; filter(fn: (e: T | R) => e is R): IChainableEvent; reduce(merge: (last: R | undefined, event: T) => R, initial?: R): IChainableEvent; latch(): IChainableEvent; debounce(merge: (last: T | undefined, event: T) => T, delay?: number, leading?: boolean, leakWarningThreshold?: number): IChainableEvent; debounce(merge: (last: R | undefined, event: T) => R, delay?: number, leading?: boolean, leakWarningThreshold?: number): IChainableEvent; on(listener: (e: T) => any, thisArgs?: any, disposables?: IDisposable[] | DisposableStore): IDisposable; once(listener: (e: T) => any, thisArgs?: any, disposables?: IDisposable[]): IDisposable; } class ChainableEvent implements IChainableEvent { private readonly disposables = new DisposableStore(); constructor(readonly event: Event) { } map(fn: (i: T) => O): IChainableEvent { return new ChainableEvent(map(this.event, fn, this.disposables)); } forEach(fn: (i: T) => void): IChainableEvent { return new ChainableEvent(forEach(this.event, fn, this.disposables)); } filter(fn: (e: T) => boolean): IChainableEvent; filter(fn: (e: T | R) => e is R): IChainableEvent; filter(fn: (e: T) => boolean): IChainableEvent { return new ChainableEvent(filter(this.event, fn, this.disposables)); } reduce(merge: (last: R | undefined, event: T) => R, initial?: R): IChainableEvent { return new ChainableEvent(reduce(this.event, merge, initial, this.disposables)); } latch(): IChainableEvent { return new ChainableEvent(latch(this.event, undefined, this.disposables)); } debounce(merge: (last: T | undefined, event: T) => T, delay?: number, leading?: boolean, leakWarningThreshold?: number): IChainableEvent; debounce(merge: (last: R | undefined, event: T) => R, delay?: number, leading?: boolean, leakWarningThreshold?: number): IChainableEvent; debounce(merge: (last: R | undefined, event: T) => R, delay: number = 100, leading = false, leakWarningThreshold?: number): IChainableEvent { return new ChainableEvent(debounce(this.event, merge, delay, leading, leakWarningThreshold, this.disposables)); } on(listener: (e: T) => any, thisArgs: any, disposables: IDisposable[] | DisposableStore) { return this.event(listener, thisArgs, disposables); } once(listener: (e: T) => any, thisArgs: any, disposables: IDisposable[]) { return once(this.event)(listener, thisArgs, disposables); } dispose() { this.disposables.dispose(); } } export function chain(event: Event): IChainableEvent { return new ChainableEvent(event); } export interface NodeEventEmitter { on(event: string | symbol, listener: Function): unknown; removeListener(event: string | symbol, listener: Function): unknown; } export function fromNodeEventEmitter(emitter: NodeEventEmitter, eventName: string, map: (...args: any[]) => T = id => id): Event { const fn = (...args: any[]) => result.fire(map(...args)); const onFirstListenerAdd = () => emitter.on(eventName, fn); const onLastListenerRemove = () => emitter.removeListener(eventName, fn); const result = new Emitter({ onFirstListenerAdd, onLastListenerRemove }); return result.event; } export interface DOMEventEmitter { addEventListener(event: string | symbol, listener: Function): void; removeEventListener(event: string | symbol, listener: Function): void; } export function fromDOMEventEmitter(emitter: DOMEventEmitter, eventName: string, map: (...args: any[]) => T = id => id): Event { const fn = (...args: any[]) => result.fire(map(...args)); const onFirstListenerAdd = () => emitter.addEventListener(eventName, fn); const onLastListenerRemove = () => emitter.removeEventListener(eventName, fn); const result = new Emitter({ onFirstListenerAdd, onLastListenerRemove }); return result.event; } export function toPromise(event: Event): Promise { return new Promise(resolve => once(event)(resolve)); } export function runAndSubscribe(event: Event, handler: (e: T | undefined) => any): IDisposable { handler(undefined); return event(e => handler(e)); } export function runAndSubscribeWithStore(event: Event, handler: (e: T | undefined, disposableStore: DisposableStore) => any): IDisposable { let store: DisposableStore | null = null; function run(e: T | undefined) { store?.dispose(); store = new DisposableStore(); handler(e, store); } run(undefined); const disposable = event(e => run(e)); return toDisposable(() => { disposable.dispose(); store?.dispose(); }); } class EmitterObserver implements IObserver { readonly emitter: Emitter; private _counter = 0; private _hasChanged = false; constructor(readonly obs: IObservable, store: DisposableStore | undefined) { const options = { onFirstListenerAdd: () => { obs.addObserver(this); }, onLastListenerRemove: () => { obs.removeObserver(this); } }; if (!store) { _addLeakageTraceLogic(options); } this.emitter = new Emitter(options); if (store) { store.add(this.emitter); } } beginUpdate(_observable: IObservable): void { // console.assert(_observable === this.obs); this._counter++; } handleChange(_observable: IObservable, _change: TChange): void { this._hasChanged = true; } endUpdate(_observable: IObservable): void { if (--this._counter === 0) { if (this._hasChanged) { this._hasChanged = false; this.emitter.fire(this.obs.get()); } } } } export function fromObservable(obs: IObservable, store?: DisposableStore): Event { const observer = new EmitterObserver(obs, store); return observer.emitter.event; } } export interface EmitterOptions { onFirstListenerAdd?: Function; onFirstListenerDidAdd?: Function; onListenerDidAdd?: Function; onLastListenerRemove?: Function; leakWarningThreshold?: number; /** * Pass in a delivery queue, which is useful for ensuring * in order event delivery across multiple emitters. */ deliveryQueue?: EventDeliveryQueue; /** ONLY enable this during development */ _profName?: string; } class EventProfiling { private static _idPool = 0; private _name: string; private _stopWatch?: StopWatch; private _listenerCount: number = 0; private _invocationCount = 0; private _elapsedOverall = 0; constructor(name: string) { this._name = `${name}_${EventProfiling._idPool++}`; } start(listenerCount: number): void { this._stopWatch = new StopWatch(true); this._listenerCount = listenerCount; } stop(): void { if (this._stopWatch) { const elapsed = this._stopWatch.elapsed(); this._elapsedOverall += elapsed; this._invocationCount += 1; console.info(`did FIRE ${this._name}: elapsed_ms: ${elapsed.toFixed(5)}, listener: ${this._listenerCount} (elapsed_overall: ${this._elapsedOverall.toFixed(2)}, invocations: ${this._invocationCount})`); this._stopWatch = undefined; } } } let _globalLeakWarningThreshold = -1; export function setGlobalLeakWarningThreshold(n: number): IDisposable { const oldValue = _globalLeakWarningThreshold; _globalLeakWarningThreshold = n; return { dispose() { _globalLeakWarningThreshold = oldValue; } }; } class LeakageMonitor { private _stacks: Map | undefined; private _warnCountdown: number = 0; constructor( readonly customThreshold?: number, readonly name: string = Math.random().toString(18).slice(2, 5), ) { } dispose(): void { if (this._stacks) { this._stacks.clear(); } } check(stack: Stacktrace, listenerCount: number): undefined | (() => void) { let threshold = _globalLeakWarningThreshold; if (typeof this.customThreshold === 'number') { threshold = this.customThreshold; } if (threshold <= 0 || listenerCount < threshold) { return undefined; } if (!this._stacks) { this._stacks = new Map(); } const count = (this._stacks.get(stack.value) || 0); this._stacks.set(stack.value, count + 1); this._warnCountdown -= 1; if (this._warnCountdown <= 0) { // only warn on first exceed and then every time the limit // is exceeded by 50% again this._warnCountdown = threshold * 0.5; // find most frequent listener and print warning let topStack: string | undefined; let topCount: number = 0; for (const [stack, count] of this._stacks) { if (!topStack || topCount < count) { topStack = stack; topCount = count; } } console.warn(`[${this.name}] potential listener LEAK detected, having ${listenerCount} listeners already. MOST frequent listener (${topCount}):`); console.warn(topStack!); } return () => { const count = (this._stacks!.get(stack.value) || 0); this._stacks!.set(stack.value, count - 1); }; } } class Stacktrace { static create() { return new Stacktrace(new Error().stack ?? ''); } private constructor(readonly value: string) { } print() { console.warn(this.value.split('\n').slice(2).join('\n')); } } class Listener { readonly subscription = new SafeDisposable(); constructor( readonly callback: (e: T) => void, readonly callbackThis: any | undefined, readonly stack: Stacktrace | undefined ) { } invoke(e: T) { this.callback.call(this.callbackThis, e); } } /** * The Emitter can be used to expose an Event to the public * to fire it from the insides. * Sample: class Document { private readonly _onDidChange = new Emitter<(value:string)=>any>(); public onDidChange = this._onDidChange.event; // getter-style // get onDidChange(): Event<(value:string)=>any> { // return this._onDidChange.event; // } private _doIt() { //... this._onDidChange.fire(value); } } */ export class Emitter { private readonly _options?: EmitterOptions; private readonly _leakageMon?: LeakageMonitor; private readonly _perfMon?: EventProfiling; private _disposed: boolean = false; private _event?: Event; private _deliveryQueue?: EventDeliveryQueue; protected _listeners?: LinkedList>; constructor(options?: EmitterOptions) { this._options = options; this._leakageMon = _globalLeakWarningThreshold > 0 ? new LeakageMonitor(this._options && this._options.leakWarningThreshold) : undefined; this._perfMon = this._options?._profName ? new EventProfiling(this._options._profName) : undefined; this._deliveryQueue = this._options?.deliveryQueue; } dispose() { if (!this._disposed) { this._disposed = true; // It is bad to have listeners at the time of disposing an emitter, it is worst to have listeners keep the emitter // alive via the reference that's embedded in their disposables. Therefore we loop over all remaining listeners and // unset their subscriptions/disposables. Looping and blaming remaining listeners is done on next tick because the // the following programming pattern is very popular: // // const someModel = this._disposables.add(new ModelObject()); // (1) create and register model // this._disposables.add(someModel.onDidChange(() => { ... }); // (2) subscribe and register model-event listener // ...later... // this._disposables.dispose(); disposes (1) then (2): don't warn after (1) but after the "overall dispose" is done if (this._listeners) { if (_enableDisposeWithListenerWarning) { const listeners = Array.from(this._listeners); queueMicrotask(() => { for (const listener of listeners) { if (listener.subscription.isset()) { listener.subscription.unset(); listener.stack?.print(); } } }); } this._listeners.clear(); } this._deliveryQueue?.clear(this); this._options?.onLastListenerRemove?.(); this._leakageMon?.dispose(); } } /** * For the public to allow to subscribe * to events from this Emitter */ get event(): Event { if (!this._event) { this._event = (callback: (e: T) => any, thisArgs?: any, disposables?: IDisposable[] | DisposableStore) => { if (!this._listeners) { this._listeners = new LinkedList(); } const firstListener = this._listeners.isEmpty(); if (firstListener && this._options?.onFirstListenerAdd) { this._options.onFirstListenerAdd(this); } let removeMonitor: Function | undefined; let stack: Stacktrace | undefined; if (this._leakageMon && this._listeners.size >= 30) { // check and record this emitter for potential leakage stack = Stacktrace.create(); removeMonitor = this._leakageMon.check(stack, this._listeners.size + 1); } if (_enableDisposeWithListenerWarning) { stack = stack ?? Stacktrace.create(); } const listener = new Listener(callback, thisArgs, stack); const removeListener = this._listeners.push(listener); if (firstListener && this._options?.onFirstListenerDidAdd) { this._options.onFirstListenerDidAdd(this); } if (this._options?.onListenerDidAdd) { this._options.onListenerDidAdd(this, callback, thisArgs); } const result = listener.subscription.set(() => { removeMonitor?.(); if (!this._disposed) { removeListener(); if (this._options && this._options.onLastListenerRemove) { const hasListeners = (this._listeners && !this._listeners.isEmpty()); if (!hasListeners) { this._options.onLastListenerRemove(this); } } } }); if (disposables instanceof DisposableStore) { disposables.add(result); } else if (Array.isArray(disposables)) { disposables.push(result); } return result; }; } return this._event; } /** * To be kept private to fire an event to * subscribers */ fire(event: T): void { if (this._listeners) { // put all [listener,event]-pairs into delivery queue // then emit all event. an inner/nested event might be // the driver of this if (!this._deliveryQueue) { this._deliveryQueue = new PrivateEventDeliveryQueue(); } for (const listener of this._listeners) { this._deliveryQueue.push(this, listener, event); } // start/stop performance insight collection this._perfMon?.start(this._deliveryQueue.size); this._deliveryQueue.deliver(); this._perfMon?.stop(); } } hasListeners(): boolean { if (!this._listeners) { return false; } return (!this._listeners.isEmpty()); } } export class EventDeliveryQueue { protected _queue = new LinkedList(); get size(): number { return this._queue.size; } push(emitter: Emitter, listener: Listener, event: T): void { this._queue.push(new EventDeliveryQueueElement(emitter, listener, event)); } clear(emitter: Emitter): void { const newQueue = new LinkedList(); for (const element of this._queue) { if (element.emitter !== emitter) { newQueue.push(element); } } this._queue = newQueue; } deliver(): void { while (this._queue.size > 0) { const element = this._queue.shift()!; try { element.listener.invoke(element.event); } catch (e) { onUnexpectedError(e); } } } } /** * An `EventDeliveryQueue` that is guaranteed to be used by a single `Emitter`. */ class PrivateEventDeliveryQueue extends EventDeliveryQueue { override clear(emitter: Emitter): void { // Here we can just clear the entire linked list because // all elements are guaranteed to belong to this emitter this._queue.clear(); } } class EventDeliveryQueueElement { constructor( readonly emitter: Emitter, readonly listener: Listener, readonly event: T ) { } } export interface IWaitUntil { token: CancellationToken; waitUntil(thenable: Promise): void; } export type IWaitUntilData = Omit, 'token'>; export class AsyncEmitter extends Emitter { private _asyncDeliveryQueue?: LinkedList<[Listener, IWaitUntilData]>; async fireAsync(data: IWaitUntilData, token: CancellationToken, promiseJoin?: (p: Promise, listener: Function) => Promise): Promise { if (!this._listeners) { return; } if (!this._asyncDeliveryQueue) { this._asyncDeliveryQueue = new LinkedList(); } for (const listener of this._listeners) { this._asyncDeliveryQueue.push([listener, data]); } while (this._asyncDeliveryQueue.size > 0 && !token.isCancellationRequested) { const [listener, data] = this._asyncDeliveryQueue.shift()!; const thenables: Promise[] = []; const event = { ...data, token, waitUntil: (p: Promise): void => { if (Object.isFrozen(thenables)) { throw new Error('waitUntil can NOT be called asynchronous'); } if (promiseJoin) { p = promiseJoin(p, listener.callback); } thenables.push(p); } }; try { listener.invoke(event); } catch (e) { onUnexpectedError(e); continue; } // freeze thenables-collection to enforce sync-calls to // wait until and then wait for all thenables to resolve Object.freeze(thenables); await Promise.allSettled(thenables).then(values => { for (const value of values) { if (value.status === 'rejected') { onUnexpectedError(value.reason); } } }); } } } export class PauseableEmitter extends Emitter { private _isPaused = 0; protected _eventQueue = new LinkedList(); private _mergeFn?: (input: T[]) => T; constructor(options?: EmitterOptions & { merge?: (input: T[]) => T }) { super(options); this._mergeFn = options?.merge; } pause(): void { this._isPaused++; } resume(): void { if (this._isPaused !== 0 && --this._isPaused === 0) { if (this._mergeFn) { // use the merge function to create a single composite // event. make a copy in case firing pauses this emitter const events = Array.from(this._eventQueue); this._eventQueue.clear(); super.fire(this._mergeFn(events)); } else { // no merging, fire each event individually and test // that this emitter isn't paused halfway through while (!this._isPaused && this._eventQueue.size !== 0) { super.fire(this._eventQueue.shift()!); } } } } override fire(event: T): void { if (this._listeners) { if (this._isPaused !== 0) { this._eventQueue.push(event); } else { super.fire(event); } } } } export class DebounceEmitter extends PauseableEmitter { private readonly _delay: number; private _handle: any | undefined; constructor(options: EmitterOptions & { merge: (input: T[]) => T; delay?: number }) { super(options); this._delay = options.delay ?? 100; } override fire(event: T): void { if (!this._handle) { this.pause(); this._handle = setTimeout(() => { this._handle = undefined; this.resume(); }, this._delay); } super.fire(event); } } /** * An emitter which queue all events and then process them at the * end of the event loop. */ export class MicrotaskEmitter extends Emitter { private _queuedEvents: T[] = []; private _mergeFn?: (input: T[]) => T; constructor(options?: EmitterOptions & { merge?: (input: T[]) => T }) { super(options); this._mergeFn = options?.merge; } override fire(event: T): void { this._queuedEvents.push(event); if (this._queuedEvents.length === 1) { queueMicrotask(() => { if (this._mergeFn) { super.fire(this._mergeFn(this._queuedEvents)); } else { this._queuedEvents.forEach(e => super.fire(e)); } this._queuedEvents = []; }); } } } export class EventMultiplexer implements IDisposable { private readonly emitter: Emitter; private hasListeners = false; private events: { event: Event; listener: IDisposable | null }[] = []; constructor() { this.emitter = new Emitter({ onFirstListenerAdd: () => this.onFirstListenerAdd(), onLastListenerRemove: () => this.onLastListenerRemove() }); } get event(): Event { return this.emitter.event; } add(event: Event): IDisposable { const e = { event: event, listener: null }; this.events.push(e); if (this.hasListeners) { this.hook(e); } const dispose = () => { if (this.hasListeners) { this.unhook(e); } const idx = this.events.indexOf(e); this.events.splice(idx, 1); }; return toDisposable(onceFn(dispose)); } private onFirstListenerAdd(): void { this.hasListeners = true; this.events.forEach(e => this.hook(e)); } private onLastListenerRemove(): void { this.hasListeners = false; this.events.forEach(e => this.unhook(e)); } private hook(e: { event: Event; listener: IDisposable | null }): void { e.listener = e.event(r => this.emitter.fire(r)); } private unhook(e: { event: Event; listener: IDisposable | null }): void { if (e.listener) { e.listener.dispose(); } e.listener = null; } dispose(): void { this.emitter.dispose(); } } /** * The EventBufferer is useful in situations in which you want * to delay firing your events during some code. * You can wrap that code and be sure that the event will not * be fired during that wrap. * * ``` * const emitter: Emitter; * const delayer = new EventDelayer(); * const delayedEvent = delayer.wrapEvent(emitter.event); * * delayedEvent(console.log); * * delayer.bufferEvents(() => { * emitter.fire(); // event will not be fired yet * }); * * // event will only be fired at this point * ``` */ export class EventBufferer { private buffers: Function[][] = []; wrapEvent(event: Event): Event { return (listener, thisArgs?, disposables?) => { return event(i => { const buffer = this.buffers[this.buffers.length - 1]; if (buffer) { buffer.push(() => listener.call(thisArgs, i)); } else { listener.call(thisArgs, i); } }, undefined, disposables); }; } bufferEvents(fn: () => R): R { const buffer: Array<() => R> = []; this.buffers.push(buffer); const r = fn(); this.buffers.pop(); buffer.forEach(flush => flush()); return r; } } /** * A Relay is an event forwarder which functions as a replugabble event pipe. * Once created, you can connect an input event to it and it will simply forward * events from that input event through its own `event` property. The `input` * can be changed at any point in time. */ export class Relay implements IDisposable { private listening = false; private inputEvent: Event = Event.None; private inputEventListener: IDisposable = Disposable.None; private readonly emitter = new Emitter({ onFirstListenerDidAdd: () => { this.listening = true; this.inputEventListener = this.inputEvent(this.emitter.fire, this.emitter); }, onLastListenerRemove: () => { this.listening = false; this.inputEventListener.dispose(); } }); readonly event: Event = this.emitter.event; set input(event: Event) { this.inputEvent = event; if (this.listening) { this.inputEventListener.dispose(); this.inputEventListener = event(this.emitter.fire, this.emitter); } } dispose() { this.inputEventListener.dispose(); this.emitter.dispose(); } }