VencordDesktop: Fix Updater

1 files changed, 6 insertions, 3 deletions

diff --git a/src/utils/updater.ts b/src/utils/updater.ts
index d0b1fdc..3606e5f 100644
--- a/src/utils/updater.ts
+++ b/src/utils/updater.ts
@@ -62,7 +62,7 @@ export function getRepo() {
     return Unwrap(VencordNative.ipc.invoke<IpcRes<string>>(IpcEvents.GET_REPO));
 }
 
-type Hashes = Record<"patcher.js" | "preload.js" | "renderer.js" | "renderer.css", string>;
+type Hashes = Record<"patcher.js" | "main.js" | "preload.js" | "renderer.js" | "renderer.css", string>;
 
 /**
  * @returns true if hard restart is required
@@ -75,8 +75,11 @@ export async function rebuild() {
 
     const newHashes = await Unwrap(VencordNative.ipc.invoke<IpcRes<Hashes>>(IpcEvents.GET_HASHES));
 
-    return oldHashes["patcher.js"] !== newHashes["patcher.js"] ||
-        oldHashes["preload.js"] !== newHashes["preload.js"];
+    if (oldHashes["preload.js"] !== newHashes["preload.js"]) return true;
+    if (IS_DISCORD_DESKTOP && oldHashes["patcher.js"] !== newHashes["patcher.js"]) return true;
+    if (IS_VENCORD_DESKTOP && oldHashes["main.js"] !== newHashes["main.js"]) return true;
+
+    return false;
 }
 
 export async function maybePromptToUpdate(confirmMessage: string, checkForDev = false) {