ipc-main-internal-utils.ts 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. import { ipcMainInternal } from '@electron/internal/browser/ipc-main-internal';
  2. type IPCHandler = (event: Electron.IpcMainInvokeEvent, ...args: any[]) => any
  3. export const handleSync = function <T extends IPCHandler> (channel: string, handler: T) {
  4. ipcMainInternal.on(channel, async (event, ...args) => {
  5. try {
  6. event.returnValue = [null, await handler(event, ...args)];
  7. } catch (error) {
  8. event.returnValue = [error];
  9. }
  10. });
  11. };
  12. let nextId = 0;
  13. export function invokeInWebContents<T> (sender: Electron.WebContentsInternal, sendToAll: boolean, command: string, ...args: any[]) {
  14. return new Promise<T>((resolve, reject) => {
  15. const requestId = ++nextId;
  16. const channel = `${command}_RESPONSE_${requestId}`;
  17. ipcMainInternal.on(channel, function handler (
  18. event, error: Electron.SerializedError, result: any
  19. ) {
  20. if (event.sender !== sender) {
  21. console.error(`Reply to ${command} sent by unexpected WebContents (${event.sender.id})`);
  22. return;
  23. }
  24. ipcMainInternal.removeListener(channel, handler);
  25. if (error) {
  26. reject(error);
  27. } else {
  28. resolve(result);
  29. }
  30. });
  31. if (sendToAll) {
  32. sender._sendInternalToAll(command, requestId, ...args);
  33. } else {
  34. sender._sendInternal(command, requestId, ...args);
  35. }
  36. });
  37. }