ipc-renderer-internal.ts 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. const { ipc } = process.electronBinding('ipc');
  2. const v8Util = process.electronBinding('v8_util');
  3. // Created by init.js.
  4. export const ipcRendererInternal = v8Util.getHiddenValue<Electron.IpcRendererInternal>(global, 'ipc-internal');
  5. const internal = true;
  6. if (!ipcRendererInternal.send) {
  7. ipcRendererInternal.send = function (channel, ...args) {
  8. return ipc.send(internal, channel, args);
  9. };
  10. ipcRendererInternal.sendSync = function (channel, ...args) {
  11. return ipc.sendSync(internal, channel, args)[0];
  12. };
  13. ipcRendererInternal.sendTo = function (webContentsId, channel, ...args) {
  14. return ipc.sendTo(internal, false, webContentsId, channel, args);
  15. };
  16. ipcRendererInternal.sendToAll = function (webContentsId, channel, ...args) {
  17. return ipc.sendTo(internal, true, webContentsId, channel, args);
  18. };
  19. ipcRendererInternal.invoke = async function<T> (channel: string, ...args: any[]) {
  20. const { error, result } = await ipc.invoke<T>(internal, channel, args);
  21. if (error) {
  22. throw new Error(`Error invoking remote method '${channel}': ${error}`);
  23. }
  24. return result;
  25. };
  26. ipcRendererInternal.onMessageFromMain = function (channel: string, listener: (event: Electron.IpcRendererEvent, ...args: any[]) => void) {
  27. return ipcRendererInternal.on(channel, (event, ...args) => {
  28. if (event.senderId !== 0) {
  29. console.error(`Message ${channel} sent by unexpected WebContents (${event.senderId})`);
  30. return;
  31. }
  32. listener(event, ...args);
  33. });
  34. };
  35. ipcRendererInternal.onceMessageFromMain = function (channel: string, listener: (event: Electron.IpcRendererEvent, ...args: any[]) => void) {
  36. return ipcRendererInternal.on(channel, function wrapper (event, ...args) {
  37. if (event.senderId !== 0) {
  38. console.error(`Message ${channel} sent by unexpected WebContents (${event.senderId})`);
  39. return;
  40. }
  41. ipcRendererInternal.removeListener(channel, wrapper);
  42. listener(event, ...args);
  43. });
  44. };
  45. }