feat: enhance JSON-RPC initialization

purocean · Dec 9, 2024 · ef0b9ca · ef0b9ca
1 parent 4f8f307
commit ef0b9ca
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/renderer/startup.ts b/src/renderer/startup.ts
@@ -260,7 +260,7 @@ whenEditorReady().then(() => {
 
 // json-rpc
 
-jsonrpc.init({ ctx })
+jsonrpc.init({ ctx }, whenEditorReady())
 
 setTimeout(() => {
   removeOldDatabases()

diff --git a/src/renderer/support/jsonrpc.ts b/src/renderer/support/jsonrpc.ts
@@ -2,6 +2,8 @@ import { JSONRPCError, JSONRPCRequest, JSONRPCResult, JSONRPCServer, JSONRPCServ
 import { FLAG_DEBUG } from '@fe/support/args'
 import { isElectron, nodeRequire } from './env'
 
+let rpcReady: Promise<void> | null = null
+
 class ElectronRendererServerChannel implements JSONRPCServerChannel {
   ipcRenderer: Electron.IpcRenderer
 
@@ -15,7 +17,15 @@ class ElectronRendererServerChannel implements JSONRPCServerChannel {
 
   setMessageHandler (callback: (message: JSONRPCRequest<any[]>) => void): void {
     this.ipcRenderer.on('jsonrpc', (_event, message) => {
-      callback(message)
+      // delay the message until the app is ready
+      if (rpcReady) {
+        rpcReady.then(() => {
+          rpcReady = null
+          callback(message)
+        })
+      } else {
+        callback(message)
+      }
     })
   }
 }
@@ -27,8 +37,9 @@ function initElectronRPCServer (modules: Record<string, any>) {
   }
 }
 
-export function init (modules: Record<string, any>) {
+export function init (modules: Record<string, any>, ready: Promise<any>) {
   if (isElectron) {
+    rpcReady = ready
     initElectronRPCServer(modules)
   }
 }