Dispose of client if connection drops

insomniac-eeper · May 30, 2024 · b945748 · b945748
1 parent 92f6d7b
commit b945748
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/BTTDRichDiscordPresence/Constants.cs b/BTTDRichDiscordPresence/Constants.cs
@@ -21,9 +21,9 @@ internal static class Constants
     internal static int MaxRetries { get; } = 5;
 
     /// <summary>
-    /// Delay between retries in milliseconds.
+    /// Delay between retries in seconds.
     /// </summary>
-    internal static int RetryDelay { get; } = 5000;
+    internal static int RetryDelay { get; } = 5;
 
     /// <summary>
     /// Mapping of map ids to map names.

diff --git a/BTTDRichDiscordPresence/DiscordRichPresence.cs b/BTTDRichDiscordPresence/DiscordRichPresence.cs
@@ -61,7 +61,16 @@ public void SetPresence(
             },
         };
 
-        activityManager.UpdateActivity(presence, resultCallback);
+        activityManager.UpdateActivity(presence, result =>
+        {
+            if (result == Result.NotRunning)
+            {
+                this.client?.Dispose();
+                this.client = null;
+                this.retries = 0;
+            }
+            resultCallback?.Invoke(result);
+        });
     }
 
     /// <summary>