refactor: version 2.0 (React)

packages/preload/index.ts

@@ -0,0 +1,36 @@
+import { contextBridge, ipcRenderer } from 'electron'
+import fs from 'fs'
+import { useLoading } from './loading'
+import { domReady } from './utils'
+
+const { appendLoading, removeLoading } = useLoading()
+
+;(async () => {
+  await domReady()
+
+  appendLoading()
+})()
+
+// --------- Expose some API to the Renderer process. ---------
+contextBridge.exposeInMainWorld('fs', fs)
+contextBridge.exposeInMainWorld('removeLoading', removeLoading)
+contextBridge.exposeInMainWorld('ipcRenderer', withPrototype(ipcRenderer))
+
+// `exposeInMainWorld` can't detect attributes and methods of `prototype`, manually patching it.
+function withPrototype(obj: Record<string, any>) {
+  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 APIs, like `NodeJS.EventEmitter['on']`, don't work in the Renderer process. Wrapping them into a function.
+      obj[key] = function (...args: any) {
+        return value.call(obj, ...args)
+      }
+    } else {
+      obj[key] = value
+    }
+  }
+  return obj
+}