feat(main): expose on & once events

massCodeIO · Apr 3, 2022 · 008e11a · 008e11a
1 parent 3fce6b5
commit 008e11a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/main/preload.ts b/src/main/preload.ts
@@ -5,7 +5,9 @@ import type { ElectronBridge } from './types'
 
 contextBridge.exposeInMainWorld('electron', {
   ipc: {
-    invoke: (channel, payload) => ipcRenderer.invoke(channel, payload)
+    invoke: (channel, payload) => ipcRenderer.invoke(channel, payload),
+    on: (channel, cb) => ipcRenderer.on(channel, cb),
+    once: (channel, cb) => ipcRenderer.once(channel, cb)
   },
   store: {
     app: {

diff --git a/src/main/types/index.d.ts b/src/main/types/index.d.ts
@@ -1,7 +1,8 @@
+import type { IpcRendererEvent } from 'electron'
 import type { AppStore, PreferencesStore } from '../store/module/types'
 import type { DB, Folder, Tag, Snippet } from './db'
 
-type ChannelSubject = 'snippet' | 'snippet-fragment' | 'folder'
+type ChannelSubject = 'snippet' | 'snippet-fragment' | 'folder' | 'close'
 
 type ContextMenuAction =
   | 'rename'
@@ -34,6 +35,10 @@ export interface ContextMenuResponse {
   data: any
 }
 
+interface EventCallback {
+  (event?: IpcRendererEvent, ...args: any[]): void
+}
+
 interface StoreGet<T> {
   (name: keyof T): any
 }
@@ -51,6 +56,8 @@ interface StoreProperties<T> {
 export interface ElectronBridge {
   ipc: {
     invoke<T, U>(channel: Channel, payload: U): Promise<T>
+    on(channel: Channel, cb: EventCallback): void
+    once(channel: Channel, cb: EventCallback): void
   }
   store: {
     app: StoreProperties<AppStore>