feat: expose withPrototype

src/preload/index.ts

@@ -14,10 +14,23 @@ domReady().then(() => {
 // --------- Expose some API to Renderer process. ---------
 contextBridge.exposeInMainWorld('fs', fs)
 contextBridge.exposeInMainWorld('removeLoading', removeLoading)
-contextBridge.exposeInMainWorld('ipcRenderer', {
+contextBridge.exposeInMainWorld('ipcRenderer', withPrototype(ipcRenderer))
-  ...ipcRenderer,
+
-  // `exposeInMainWorld` will not expose attribute and mothods from the prototype
+// `exposeInMainWorld` can not detect `prototype` attribute and methods, manually patch it.
-  on(...args: Parameters<IpcRenderer['on']>) {
+function withPrototype(obj: Record<string, any>) {
-    return ipcRenderer.on(...args)
+  const protos = Object.getPrototypeOf(obj)
+
+  for (const [key, value] of Object.entries(protos)) {
+    if (Object.prototype.hasOwnProperty.call(obj, key)) continue
+
+    if (typeof value === 'function') {
+      // Some native API not work in Renderer-process, like `NodeJS.EventEmitter['on']`. Wrap a function patch it.
+      obj[key] = function (...args: any) {
+        return value.call(obj, ...args)
+      }
+    } else {
+      obj[key] = value
+    }
   }
-})
+  return obj
+}