Merge pull request #81 from nullinside-development-group/fix/disconnect-chats

fix: Disconnect twitch client proxy

Author: ProgrammingByPermutation

src/Nullinside.Api.Common/Twitch/ITwitchClientProxy.cs

@@ -39,9 +39,10 @@ public interface ITwitchClientProxy : IDisposable, IAsyncDisposable {
   /// <summary>
   ///   Removes a callback for when the channel receives a new chat message.
   /// </summary>
+  /// <param name="channel">The name of the channel to add the callback for.</param>
   /// <param name="callback">The callback to remove.</param>
   /// <returns>An asynchronous task.</returns>
-  void RemoveMessageCallback(Action<OnMessageReceivedArgs> callback);
+  void RemoveMessageCallback(string channel, Action<OnMessageReceivedArgs> callback);
 
   /// <summary>
   ///   Adds a callback for when users are banned from the chat.
@@ -53,8 +54,9 @@ public interface ITwitchClientProxy : IDisposable, IAsyncDisposable {
   /// <summary>
   ///   Removes a callback for when users are banned from the chat.
   /// </summary>
+  /// <param name="channel">The name of the channel to add the callback for.</param>
   /// <param name="callback">The callback to remove from when a user is banned.</param>
-  void RemoveBannedCallback(Action<OnUserBannedArgs> callback);
+  void RemoveBannedCallback(string channel, Action<OnUserBannedArgs> callback);
 
   /// <summary>
   ///   Adds a callback for when the channel receives a raid.
@@ -67,7 +69,8 @@ public interface ITwitchClientProxy : IDisposable, IAsyncDisposable {
   /// <summary>
   ///   Removes a callback for when the channel receives a raid.
   /// </summary>
+  /// <param name="channel">The name of the channel to add the callback for.</param>
   /// <param name="callback">The callback to remove.</param>
   /// <returns>An asynchronous task.</returns>
-  void RemoveRaidCallback(Action<OnRaidNotificationArgs> callback);
+  void RemoveRaidCallback(string channel, Action<OnRaidNotificationArgs> callback);
 }

src/Nullinside.Api.Common/Twitch/TwitchClientProxy.cs

@@ -29,11 +29,6 @@ public class TwitchClientProxy : ITwitchClientProxy {
   /// </summary>
   private static TwitchClientProxy? instance;
 
-  /// <summary>
-  ///   The callback(s) to invoke when a new instance is created.
-  /// </summary>
-  private static Action<TwitchClientProxy>? onInstanceCreated;
-
   /// <summary>
   ///   The list of chats we attempted to join with the bot.
   /// </summary>
@@ -61,17 +56,17 @@ public class TwitchClientProxy : ITwitchClientProxy {
   /// <summary>
   ///   The callback(s) to invoke when a channel receives a chat message.
   /// </summary>
-  private Action<OnMessageReceivedArgs>? onMessageReceived;
+  private Dictionary<string, Action<OnMessageReceivedArgs>?> _onMessageReceived = new();
 
   /// <summary>
   ///   The callback(s) to invoke when a channel is raided.
   /// </summary>
-  private Action<OnRaidNotificationArgs>? onRaid;
+  private Dictionary<string, Action<OnRaidNotificationArgs>?> onRaid = new();
 
   /// <summary>
   ///   The callback(s) to invoke when a channel receives a ban message.
   /// </summary>
-  private Action<OnUserBannedArgs>? onUserBanReceived;
+  private Dictionary<string, Action<OnUserBannedArgs>?> onUserBanReceived = new();
 
   /// <summary>
   ///   The web socket to connect to twitch chat with.
@@ -98,7 +93,6 @@ public static TwitchClientProxy Instance {
     get {
       if (null == instance) {
         instance = new TwitchClientProxy();
-        onInstanceCreated?.Invoke(instance);
       }
 
       return instance;
@@ -162,39 +156,60 @@ public async Task<bool> SendMessage(string channel, string message, uint retryCo
   /// <inheritdoc />
   public async Task AddMessageCallback(string channel, Action<OnMessageReceivedArgs> callback) {
     await JoinChannel(channel);
-    onMessageReceived -= callback;
-    onMessageReceived += callback;
+    var channelSan = channel.ToLowerInvariant();
+    lock (_onMessageReceived) {
+      _onMessageReceived[channelSan] = callback;
+    }
   }
 
   /// <inheritdoc />
-  public void RemoveMessageCallback(Action<OnMessageReceivedArgs> callback) {
-    onMessageReceived -= callback;
+  public void RemoveMessageCallback(string channel, Action<OnMessageReceivedArgs> callback) {
+    bool shouldRemove = false;
+    var channelSan = channel.ToLowerInvariant();
+    lock (_onMessageReceived) {
+      _onMessageReceived.Remove(channel);
+    }
+
+    if (shouldRemove) {
+      client?.LeaveChannel(channelSan);
+      
+      // First add the channel to the master list.
+      lock (joinedChannels) {
+        joinedChannels.Add(channelSan);
+      }
+    }
   }
 
   /// <inheritdoc />
   public async Task AddBannedCallback(string channel, Action<OnUserBannedArgs> callback) {
     await JoinChannel(channel);
 
-    onUserBanReceived -= callback;
-    onUserBanReceived += callback;
+    lock (onUserBanReceived) {
+      onUserBanReceived[channel] = callback;
+    }
   }
 
   /// <inheritdoc />
-  public void RemoveBannedCallback(Action<OnUserBannedArgs> callback) {
-    onUserBanReceived -= callback;
+  public void RemoveBannedCallback(string channel, Action<OnUserBannedArgs> callback) {
+    lock (onUserBanReceived) {
+      onUserBanReceived.Remove(channel);
+    }
   }
 
   /// <inheritdoc />
   public async Task AddRaidCallback(string channel, Action<OnRaidNotificationArgs> callback) {
     await JoinChannel(channel);
 
-    onRaid -= callback;
-    onRaid += callback;
+    lock (onRaid) {
+      onRaid[channel] = callback;
+    }
   }
 
   /// <inheritdoc />
-  public void RemoveRaidCallback(Action<OnRaidNotificationArgs> callback) {
-    onRaid -= callback;
+  public void RemoveRaidCallback(string channel, Action<OnRaidNotificationArgs> callback) {
+    lock (onRaid) {
+      onRaid.Remove(channel);
+    }
   }
 
   /// <inheritdoc />
@@ -203,17 +218,6 @@ public ValueTask DisposeAsync() {
     return ValueTask.CompletedTask;
   }
 
-  /// <inheritdoc />
-  public void AddInstanceCallback(Action<TwitchClientProxy> callback) {
-    onInstanceCreated -= callback;
-    onInstanceCreated += callback;
-  }
-
-  /// <inheritdoc />
-  public void RemoveInstanceCallback(Action<TwitchClientProxy> callback) {
-    onInstanceCreated -= callback;
-  }
-
   /// <summary>
   ///   Joins a twitch channel.
   /// </summary>
@@ -303,6 +307,10 @@ private async Task<bool> Connect() {
       try {
         bool isConnected = false;
         lock (twitchClientLock) {
+          if (client?.IsConnected ?? false) {
+            return true;
+          }
+          
           // If this is a first time initialization, create a brand-new client.
           bool haveNoClient = null == client;
           if (haveNoClient) {
@@ -377,7 +385,13 @@ private async Task<bool> Connect() {
   /// <param name="sender">The twitch client.</param>
   /// <param name="e">The event arguments.</param>
   private void TwitchChatClient_OnRaidNotification(object? sender, OnRaidNotificationArgs e) {
-    Delegate[]? invokeList = onRaid?.GetInvocationList();
+    Action<OnRaidNotificationArgs>? callback;
+    var channel = e.Channel.ToLowerInvariant();
+    lock (onRaid) {
+      onRaid.TryGetValue(channel, out callback);
+    }
+    
+    Delegate[]? invokeList = callback?.GetInvocationList();
     if (null == invokeList) {
       return;
     }
@@ -393,7 +407,19 @@ private void TwitchChatClient_OnRaidNotification(object? sender, OnRaidNotificat
   /// <param name="sender">The twitch client.</param>
   /// <param name="e">The event arguments.</param>
   private void TwitchChatClient_OnMessageReceived(object? sender, OnMessageReceivedArgs e) {
-    Delegate[]? invokeList = onMessageReceived?.GetInvocationList();
+    Action<OnMessageReceivedArgs>? callbacks = null;
+    var channelSan = e.ChatMessage.Channel.ToLowerInvariant();
+    lock (_onMessageReceived) {
+      if (_onMessageReceived.TryGetValue(channelSan, out Action<OnMessageReceivedArgs>? messageReceivedCallback)) {
+        callbacks = messageReceivedCallback;
+      }
+    }
+
+    if (null == callbacks) {
+      return;
+    }
+
+    Delegate[]? invokeList = callbacks?.GetInvocationList();
     if (null == invokeList) {
       return;
     }
@@ -409,7 +435,13 @@ private void TwitchChatClient_OnMessageReceived(object? sender, OnMessageReceive
   /// <param name="sender">The twitch client.</param>
   /// <param name="e">The event arguments.</param>
   private void TwitchChatClient_OnUserBanned(object? sender, OnUserBannedArgs e) {
-    Delegate[]? invokeList = onUserBanReceived?.GetInvocationList();
+    Action<OnUserBannedArgs>? callback;
+    var channel = e.UserBan.Channel.ToLowerInvariant();
+    lock (onUserBanReceived) {
+      onUserBanReceived.TryGetValue(channel, out callback);
+    }
+    
+    Delegate[]? invokeList = callback?.GetInvocationList();
     if (null == invokeList) {
       return;
     }