From f60ccb766f30e7518bc348013b72f071f86ecbc6 Mon Sep 17 00:00:00 2001
From: Vendicated <vendicated@riseup.net>
Date: Wed, 31 Aug 2022 22:08:05 +0200
Subject: Add Plugin.start, make Settings actually start/stop plugins

---
 src/VencordNative.ts | 23 +++++++++++++++--------
 1 file changed, 15 insertions(+), 8 deletions(-)

(limited to 'src/VencordNative.ts')

diff --git a/src/VencordNative.ts b/src/VencordNative.ts
index 755c471..e5ccaaa 100644
--- a/src/VencordNative.ts
+++ b/src/VencordNative.ts
@@ -1,24 +1,31 @@
 import IPC_EVENTS from './utils/IpcEvents';
 import { IpcRenderer, ipcRenderer } from 'electron';
 
+function assertEventAllowed(event: string) {
+    if (!(event in IPC_EVENTS)) throw new Error(`Event ${event} not allowed.`);
+}
 export default {
     getVersions: () => process.versions,
     ipc: {
         send(event: string, ...args: any[]) {
-            if (event in IPC_EVENTS) ipcRenderer.send(event, ...args);
-            else throw new Error(`Event ${event} not allowed.`);
+            assertEventAllowed(event);
+            ipcRenderer.send(event, ...args);
         },
         sendSync<T = any>(event: string, ...args: any[]): T {
-            if (event in IPC_EVENTS) return ipcRenderer.sendSync(event, ...args);
-            else throw new Error(`Event ${event} not allowed.`);
+            assertEventAllowed(event);
+            return ipcRenderer.sendSync(event, ...args);
         },
         on(event: string, listener: Parameters<IpcRenderer["on"]>[1]) {
-            if (event in IPC_EVENTS) ipcRenderer.on(event, listener);
-            else throw new Error(`Event ${event} not allowed.`);
+            assertEventAllowed(event);
+            ipcRenderer.on(event, listener);
+        },
+        off(event: string, listener: Parameters<IpcRenderer["off"]>[1]) {
+            assertEventAllowed(event);
+            ipcRenderer.off(event, listener);
         },
         invoke<T = any>(event: string, ...args: any[]): Promise<T> {
-            if (event in IPC_EVENTS) return ipcRenderer.invoke(event, ...args);
-            else throw new Error(`Event ${event} not allowed.`);
+            assertEventAllowed(event);
+            return ipcRenderer.invoke(event, ...args);
         }
     },
     require(mod: string) {
-- 
cgit