electron/lib/browser/ipc-main-impl.ts

35 lines
1,007 B
TypeScript
Raw Normal View History

2020-03-20 20:28:31 +00:00
import { EventEmitter } from 'events';
import { IpcMainInvokeEvent } from 'electron/main';
export class IpcMainImpl extends EventEmitter {
private _invokeHandlers: Map<string, (e: IpcMainInvokeEvent, ...args: any[]) => void> = new Map();
2022-08-03 23:55:12 +00:00
constructor () {
super();
// Do not throw exception when channel name is "error".
this.on('error', () => {});
}
handle: Electron.IpcMain['handle'] = (method, fn) => {
if (this._invokeHandlers.has(method)) {
2020-03-20 20:28:31 +00:00
throw new Error(`Attempted to register a second handler for '${method}'`);
}
if (typeof fn !== 'function') {
2020-03-20 20:28:31 +00:00
throw new Error(`Expected handler to be a function, but found type '${typeof fn}'`);
}
2023-02-13 21:39:18 +00:00
this._invokeHandlers.set(method, fn);
};
handleOnce: Electron.IpcMain['handleOnce'] = (method, fn) => {
this.handle(method, (e, ...args) => {
2020-03-20 20:28:31 +00:00
this.removeHandler(method);
return fn(e, ...args);
});
};
removeHandler (method: string) {
2020-03-20 20:28:31 +00:00
this._invokeHandlers.delete(method);
}
}