electron/lib/renderer/remote/callbacks-registry.ts

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);
    }
  }
}