1 files changed, 18 insertions, 1 deletions
diff --git a/src/preload.ts b/src/preload.ts
index 73d2eb5..c716d2c 100644
--- a/src/preload.ts
+++ b/src/preload.ts
@@ -1,7 +1,24 @@
-import { contextBridge, webFrame } from "electron";
+import electron, { contextBridge, webFrame } from "electron";
 import { readFileSync } from "fs";
 import { join } from "path";
 import VencordNative from "./VencordNative";
+import { ipcRenderer } from 'electron';
+import IpcEvents from "./utils/IpcEvents";
+
+if (electron.desktopCapturer === void 0) {
+    // Fix for desktopCapturer being main only in Electron 17+
+    // Discord accesses this in discord_desktop_core (DiscordNative.desktopCapture.getDesktopCaptureSources)
+    // and errors with cannot "read property getSources() of undefined"
+    // see discord_desktop_core/app/discord_native/renderer/desktopCapture.js
+    const electronPath = require.resolve("electron");
+    delete require.cache[electronPath]!.exports;
+    require.cache[electronPath]!.exports = {
+        ...electron,
+        desktopCapturer: {
+            getSources: (opts) => ipcRenderer.invoke(IpcEvents.GET_DESKTOP_CAPTURE_SOURCES, opts)
+        }
+    };
+}
 
 contextBridge.exposeInMainWorld("VencordNative", VencordNative);