2020-03-20 20:28:31 +00:00
|
|
|
import { EventEmitter } from 'events';
|
2020-07-13 16:58:49 +00:00
|
|
|
import { IpcMainInvokeEvent } from 'electron/main';
|
2019-08-23 22:45:50 +00:00
|
|
|
|
|
|
|
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', () => {});
|
|
|
|
}
|
|
|
|
|
2019-08-23 22:45:50 +00:00
|
|
|
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}'`);
|
2019-08-23 22:45:50 +00:00
|
|
|
}
|
|
|
|
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}'`);
|
2019-08-23 22:45:50 +00:00
|
|
|
}
|
2023-02-13 21:39:18 +00:00
|
|
|
this._invokeHandlers.set(method, fn);
|
2023-05-30 11:10:22 +00:00
|
|
|
};
|
2019-08-23 22:45:50 +00:00
|
|
|
|
|
|
|
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);
|
|
|
|
});
|
2023-05-30 11:10:22 +00:00
|
|
|
};
|
2019-08-23 22:45:50 +00:00
|
|
|
|
|
|
|
removeHandler (method: string) {
|
2020-03-20 20:28:31 +00:00
|
|
|
this._invokeHandlers.delete(method);
|
2019-08-23 22:45:50 +00:00
|
|
|
}
|
|
|
|
}
|