38 lines
1.1 KiB
TypeScript
38 lines
1.1 KiB
TypeScript
import fs from 'fs'
|
|
import { contextBridge, ipcRenderer } from 'electron'
|
|
import { domReady } from './utils'
|
|
import { useLoading } from './loading'
|
|
|
|
const isDev = process.env.NODE_ENV === 'development'
|
|
const { appendLoading, removeLoading } = useLoading()
|
|
|
|
; (async () => {
|
|
await domReady()
|
|
|
|
appendLoading()
|
|
})();
|
|
|
|
// --------- Expose some API to Renderer process. ---------
|
|
contextBridge.exposeInMainWorld('fs', fs)
|
|
contextBridge.exposeInMainWorld('removeLoading', removeLoading)
|
|
contextBridge.exposeInMainWorld('ipcRenderer', withPrototype(ipcRenderer))
|
|
|
|
// `exposeInMainWorld` can not detect `prototype` attribute and methods, manually patch 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 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
|
|
}
|