59 lines
1.6 KiB
TypeScript
59 lines
1.6 KiB
TypeScript
export class CallbacksRegistry {
|
|
private nextId: number = 0
|
|
private callbacks = new Map<number, Function>()
|
|
private callbackIds = new WeakMap<Function, number>();
|
|
private locationInfo = new WeakMap<Function, string>();
|
|
|
|
add (callback: Function) {
|
|
// The callback is already added.
|
|
let id = this.callbackIds.get(callback);
|
|
if (id != null) return id;
|
|
|
|
id = this.nextId += 1;
|
|
|
|
// Capture the location of the function and put it in the ID string,
|
|
// so that release errors can be tracked down easily.
|
|
const regexp = /at (.*)/gi;
|
|
const stackString = (new Error()).stack;
|
|
if (!stackString) return;
|
|
|
|
let filenameAndLine: string;
|
|
let match;
|
|
|
|
while ((match = regexp.exec(stackString)) !== null) {
|
|
const location = match[1];
|
|
if (location.includes('(native)')) continue;
|
|
if (location.includes('(<anonymous>)')) continue;
|
|
if (location.includes('electron/js2c')) continue;
|
|
|
|
const ref = /([^/^)]*)\)?$/gi.exec(location);
|
|
if (ref) filenameAndLine = ref![1];
|
|
break;
|
|
}
|
|
|
|
this.callbacks.set(id, callback);
|
|
this.callbackIds.set(callback, id);
|
|
this.locationInfo.set(callback, filenameAndLine!);
|
|
return id;
|
|
}
|
|
|
|
get (id: number) {
|
|
return this.callbacks.get(id) || function () {};
|
|
}
|
|
|
|
getLocation (callback: Function) {
|
|
return this.locationInfo.get(callback);
|
|
}
|
|
|
|
apply (id: number, ...args: any[]) {
|
|
return this.get(id).apply(global, ...args);
|
|
}
|
|
|
|
remove (id: number) {
|
|
const callback = this.callbacks.get(id);
|
|
if (callback) {
|
|
this.callbackIds.delete(callback);
|
|
this.callbacks.delete(id);
|
|
}
|
|
}
|
|
}
|