From 6be577400143f4557d23daaeac0d214ad412ae79 Mon Sep 17 00:00:00 2001
From: 3top1a <3top1a.official@gmail.com>
Date: Tue, 22 Oct 2024 22:29:56 +0200
Subject: [PATCH] Added RPC connection retrying

---
 src/scripts/discord.ts | 38 +++++++++++++++++++++++---------------
 1 file changed, 23 insertions(+), 15 deletions(-)

diff --git a/src/scripts/discord.ts b/src/scripts/discord.ts
index 10c25d3..3f8dff5 100644
--- a/src/scripts/discord.ts
+++ b/src/scripts/discord.ts
@@ -13,6 +13,8 @@ const clientId = "833617820704440341";
 export let rpc: Client;
 
 const ACTIVITY_LISTENING = 2;
+const MAX_RETRIES = 5;
+const RETRY_DELAY = 10000;
 
 const observer = () => {
   if (rpc) {
@@ -77,26 +79,32 @@ const getActivity = (): SetActivity => {
   }
 };
 
+/**
+ * Try to login to RPC and retry if it errors
+ * @param retryCount Max retry count
+ */
+const connectWithRetry = async (retryCount = 0) => {
+  try {
+    await rpc.login();
+    Logger.log('Connected to Discord');
+    rpc.on("ready", updateActivity);
+    Object.values(globalEvents).forEach(event => ipcMain.on(event, observer));
+  } catch (error) {
+    if (retryCount < MAX_RETRIES) {
+      Logger.log(`Failed to connect to Discord, retrying in ${RETRY_DELAY/1000} seconds... (Attempt ${retryCount + 1}/${MAX_RETRIES})`);
+      setTimeout(() => connectWithRetry(retryCount + 1), RETRY_DELAY);
+    } else {
+      Logger.log('Failed to connect to Discord after maximum retry attempts');
+    }
+  }
+};
+
 /**
  * Set up the discord rpc and listen on globalEvents.updateInfo
  */
 export const initRPC = () => {
   rpc = new Client({ transport: {type: "ipc"}, clientId });
-  rpc.login().then(
-    () => {
-      rpc.on("ready", () => {
-        updateActivity();
-      });
-
-      const { updateInfo, play, pause, playPause } = globalEvents;
-      [updateInfo, play, pause, playPause].forEach((status) => {
-        ipcMain.on(status, observer);
-      });
-    },
-    () => {
-      Logger.log("Can't connect to Discord, is it running?");
-    }
-  );
+  connectWithRetry();
 };
 
 /**