ChatManager.cs

Go to the documentation of this file.

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.Extensions.Logging;
 
using Newtonsoft.Json;
 
using Serilog.Context;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Api.Models.Internal;
using Tgstation.Server.Common.Extensions;
using Tgstation.Server.Host.Components.Chat.Commands;
using Tgstation.Server.Host.Components.Chat.Providers;
using Tgstation.Server.Host.Components.Interop;
using Tgstation.Server.Host.Core;
using Tgstation.Server.Host.Utils;
 
namespace Tgstation.Server.Host.Components.Chat
{
#pragma warning disable CA1506 // TODO: Decomplexify
    sealed class ChatManager : IChatManager, IRestartHandler
    {
        public const string CommonMention = "!tgs";
 
        readonly IProviderFactory providerFactory;
 
        readonly ICommandFactory commandFactory;
 
        readonly IRestartRegistration restartRegistration;
 
        readonly ILoggerFactory loggerFactory;
 
        readonly ILogger<ChatManager> logger;
 
        readonly Dictionary<string, ICommand> builtinCommands;
 
        readonly Dictionary<long, IProvider> providers;
 
        readonly ConcurrentDictionary<long, SemaphoreSlim> changeChannelSemaphores;
 
        readonly Dictionary<ulong, ChannelMapping> mappedChannels;
 
        readonly List<IChatTrackingContext> trackingContexts;
 
        readonly List<Models.ChatBot> activeChatBots;
 
        readonly CancellationTokenSource handlerCts;
 
        readonly object synchronizationLock;
 
        ICustomCommandHandler? customCommandHandler;
 
        Task? chatHandler;
 
        Task? initialProviderConnectionsTask;
 
        Task messageSendTask;
 
        TaskCompletionSource connectionsUpdated;
 
        ulong channelIdCounter;
 
        long messagesProcessed;
 
        public ChatManager(
            IProviderFactory providerFactory,
            ICommandFactory commandFactory,
            IServerControl serverControl,
            ILoggerFactory loggerFactory,
            ILogger<ChatManager> logger,
            IEnumerable<Models.ChatBot> initialChatBots)
        {
            this.providerFactory = providerFactory ?? throw new ArgumentNullException(nameof(providerFactory));
            this.commandFactory = commandFactory ?? throw new ArgumentNullException(nameof(commandFactory));
            ArgumentNullException.ThrowIfNull(serverControl);
            this.loggerFactory = loggerFactory ?? throw new ArgumentNullException(nameof(loggerFactory));
            this.logger = logger ?? throw new ArgumentNullException(nameof(logger));
            activeChatBots = initialChatBots?.ToList() ?? throw new ArgumentNullException(nameof(initialChatBots));
 
            restartRegistration = serverControl.RegisterForRestart(this);
 
            synchronizationLock = new object();
 
            builtinCommands = new Dictionary<string, ICommand>(StringComparer.OrdinalIgnoreCase);
            providers = new Dictionary<long, IProvider>();
            changeChannelSemaphores = new ConcurrentDictionary<long, SemaphoreSlim>();
            mappedChannels = new Dictionary<ulong, ChannelMapping>();
            trackingContexts = new List<IChatTrackingContext>();
            handlerCts = new CancellationTokenSource();
            connectionsUpdated = new TaskCompletionSource();
 
            messageSendTask = Task.CompletedTask;
            channelIdCounter = 1;
        }
 
        public async ValueTask DisposeAsync()
        {
            logger.LogTrace("Disposing...");
            restartRegistration.Dispose();
            handlerCts.Dispose();
            foreach (var providerKvp in providers)
                await providerKvp.Value.DisposeAsync();
 
            foreach (var providerKvp in changeChannelSemaphores)
                providerKvp.Value.Dispose();
 
            await messageSendTask;
        }
 
        public async ValueTask ChangeChannels(long connectionId, IEnumerable<Models.ChatChannel> newChannels, CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(newChannels);
 
            logger.LogTrace("ChangeChannels {connectionId}...", connectionId);
            var semaphore = changeChannelSemaphores.GetOrAdd(connectionId, _ =>
            {
                logger.LogTrace("Creating ChangeChannels semaphore for connection ID {connectionId}...", connectionId);
                return new SemaphoreSlim(1);
            });
            using (await SemaphoreSlimContext.Lock(semaphore, cancellationToken))
            {
                var provider = await RemoveProviderChannels(connectionId, false, cancellationToken);
                if (provider == null)
                    return;
 
                if (!provider.Connected)
                {
                    logger.LogDebug("Cannot map channels, provider {providerId} disconnected!", connectionId);
                    return;
                }
 
                var results = await provider.MapChannels(newChannels, cancellationToken);
                try
                {
                    lock (activeChatBots)
                    {
                        var botToUpdate = activeChatBots.FirstOrDefault(bot => bot.Id == connectionId);
                        if (botToUpdate != null)
                            botToUpdate.Channels = newChannels
                                .Select(apiModel => new Models.ChatChannel
                                {
                                    DiscordChannelId = apiModel.DiscordChannelId,
                                    IrcChannel = apiModel.IrcChannel,
                                    IsAdminChannel = apiModel.IsAdminChannel,
                                    IsUpdatesChannel = apiModel.IsUpdatesChannel,
                                    IsSystemChannel = apiModel.IsSystemChannel,
                                    IsWatchdogChannel = apiModel.IsWatchdogChannel,
                                    Tag = apiModel.Tag,
                                })
                                .ToList();
                    }
 
                    var newMappings = results.SelectMany(
                        kvp => kvp.Value.Select(
                            channelRepresentation => new ChannelMapping(channelRepresentation)
                            {
                                IsWatchdogChannel = kvp.Key.IsWatchdogChannel == true,
                                IsUpdatesChannel = kvp.Key.IsUpdatesChannel == true,
                                IsAdminChannel = kvp.Key.IsAdminChannel == true,
                                IsSystemChannel = kvp.Key.IsSystemChannel == true,
                                ProviderChannelId = channelRepresentation.RealId,
                                ProviderId = connectionId,
                            }));
 
                    ulong baseId;
                    lock (synchronizationLock)
                    {
                        baseId = channelIdCounter;
                        channelIdCounter += (ulong)results.Count;
                    }
 
                    lock (mappedChannels)
                    {
                        lock (providers)
                            if (!providers.TryGetValue(connectionId, out var verify) || verify != provider) // aborted again
                                return;
                        foreach (var newMapping in newMappings)
                        {
                            var newId = baseId++;
                            logger.LogTrace("Mapping channel {connectionName}:{channelFriendlyName} as {newId}", newMapping.Channel.ConnectionName, newMapping.Channel.FriendlyName, newId);
                            mappedChannels.Add(newId, newMapping);
                            newMapping.Channel.RealId = newId;
                        }
                    }
 
                    // we only want to update contexts if everything at startup has connected once already
                    // otherwise we could send an incomplete channel set to the DMAPI, which will then spout all its queued messages into it instead of all relevant chatbots
                    // The watchdog can call this if it needs to after starting up
                    if (initialProviderConnectionsTask!.IsCompleted)
                        await UpdateTrackingContexts(cancellationToken);
                }
                finally
                {
                    provider.InitialMappingComplete();
                }
            }
        }
 
        public async ValueTask ChangeSettings(Models.ChatBot newSettings, CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(newSettings);
 
            logger.LogTrace("ChangeSettings...");
 
            Task disconnectTask;
            IProvider? provider = null;
            var newSettingsId = Models.ModelExtensions.Require(newSettings, x => x.Id);
            var newSettingsEnabled = Models.ModelExtensions.Require(newSettings, x => x.Enabled);
            lock (providers)
            {
                // raw settings changes forces a rebuild of the provider
                if (providers.ContainsKey(newSettingsId))
                    disconnectTask = DeleteConnection(newSettingsId, cancellationToken);
                else
                    disconnectTask = Task.CompletedTask;
                if (newSettingsEnabled)
                {
                    provider = providerFactory.CreateProvider(newSettings);
                    providers.Add(newSettingsId, provider);
                }
            }
 
            lock (mappedChannels)
                foreach (var oldMappedChannelId in mappedChannels.Where(x => x.Value.ProviderId == newSettingsId).Select(x => x.Key).ToList())
                    mappedChannels.Remove(oldMappedChannelId);
 
            await disconnectTask;
 
            lock (synchronizationLock)
            {
                // same thread shennanigans
                var oldOne = connectionsUpdated;
                connectionsUpdated = new TaskCompletionSource();
                oldOne.SetResult();
            }
 
            var reconnectionUpdateTask = provider?.SetReconnectInterval(
                Models.ModelExtensions.Require(newSettings, x => x.ReconnectionInterval),
                newSettingsEnabled)
                ?? Task.CompletedTask;
            lock (activeChatBots)
            {
                var originalChatBot = activeChatBots.FirstOrDefault(bot => bot.Id == newSettings.Id);
                if (originalChatBot != null)
                    activeChatBots.Remove(originalChatBot);
 
                activeChatBots.Add(new Models.ChatBot(newSettings.Channels)
                {
                    Id = newSettings.Id,
                    ConnectionString = newSettings.ConnectionString,
                    Enabled = newSettings.Enabled,
                    Name = newSettings.Name,
                    ReconnectionInterval = newSettings.ReconnectionInterval,
                    Provider = newSettings.Provider,
                });
            }
 
            await reconnectionUpdateTask;
        }
 
        public void QueueMessage(MessageContent message, IEnumerable<ulong> channelIds)
        {
            ArgumentNullException.ThrowIfNull(message);
            ArgumentNullException.ThrowIfNull(channelIds);
 
            QueueMessageInternal(message, () => channelIds, false);
        }
 
        public void QueueWatchdogMessage(string message)
            => QueueMessageGeneric(mapping => mapping.IsWatchdogChannel, message, "WD");
 
        public void QueueRawDeploymentMessage(string message)
            => QueueMessageGeneric(mapping => mapping.IsUpdatesChannel, message, null);
 
        public Func<string?, string, Action<bool>> QueueDeploymentMessage(
            Models.RevisionInformation revisionInformation,
            Models.RevisionInformation? previousRevisionInformation,
            EngineVersion engineVersion,
            DateTimeOffset? estimatedCompletionTime,
            string? gitHubOwner,
            string? gitHubRepo,
            bool localCommitPushed)
        {
            List<ulong> wdChannels;
            lock (mappedChannels) // so it doesn't change while we're using it
                wdChannels = mappedChannels.Where(x => x.Value.IsUpdatesChannel).Select(x => x.Key).ToList();
 
            logger.LogTrace("Sending deployment message for RevisionInformation: {revisionInfoId}", revisionInformation.Id);
 
            var callbacks = new List<Func<string?, string, ValueTask<Func<bool, ValueTask>>>>();
 
            var task = Task.WhenAll(
                wdChannels.Select(
                    async x =>
                    {
                        ChannelMapping? channelMapping;
                        lock (mappedChannels)
                            if (!mappedChannels.TryGetValue(x, out channelMapping))
                                return;
                        IProvider? provider;
                        lock (providers)
                            if (!providers.TryGetValue(channelMapping.ProviderId, out provider))
                                return;
                        try
                        {
                            var callback = await provider.SendUpdateMessage(
                                revisionInformation,
                                previousRevisionInformation,
                                engineVersion,
                                estimatedCompletionTime,
                                gitHubOwner,
                                gitHubRepo,
                                channelMapping.ProviderChannelId,
                                localCommitPushed,
                                handlerCts.Token);
 
                            lock (callbacks)
                                callbacks.Add(callback);
                        }
                        catch (Exception ex)
                        {
                            logger.LogWarning(
                                ex,
                                "Error sending deploy message to provider {providerId}!",
                                channelMapping.ProviderId);
                        }
                    }));
 
            AddMessageTask(task);
 
            Task callbackTask;
            Func<bool, Task>? finalUpdateAction = null;
            async Task CallbackTask(string? errorMessage, string dreamMakerOutput)
            {
                await task;
                var callbackResults = await ValueTaskExtensions.WhenAll(
                    callbacks.Select(
                        x => x(
                            errorMessage,
                            dreamMakerOutput)),
                    callbacks.Count);
 
                finalUpdateAction = active => ValueTaskExtensions.WhenAll(callbackResults.Select(finalizerCallback => finalizerCallback(active))).AsTask();
            }
 
            async Task CompletionTask(bool active)
            {
                try
                {
                    await callbackTask;
                }
                catch
                {
                    // Handled in AddMessageTask
                    return;
                }
 
                AddMessageTask(finalUpdateAction!(active));
            }
 
            return (errorMessage, dreamMakerOutput) =>
            {
                callbackTask = CallbackTask(errorMessage, dreamMakerOutput);
                AddMessageTask(callbackTask);
                return active => AddMessageTask(CompletionTask(active));
            };
        }
 
        public async Task StartAsync(CancellationToken cancellationToken)
        {
            foreach (var tgsCommand in commandFactory.GenerateCommands())
                builtinCommands.Add(tgsCommand.Name.ToUpperInvariant(), tgsCommand);
            var initialChatBots = activeChatBots.ToList();
            await ValueTaskExtensions.WhenAll(initialChatBots.Select(x => ChangeSettings(x, cancellationToken)));
            initialProviderConnectionsTask = InitialConnection();
            chatHandler = MonitorMessages(handlerCts.Token);
        }
 
        public async Task StopAsync(CancellationToken cancellationToken)
        {
            handlerCts.Cancel();
            if (chatHandler != null)
                await chatHandler;
            await Task.WhenAll(providers.Select(x => x.Key).Select(x => DeleteConnection(x, cancellationToken)));
            await messageSendTask;
        }
 
        public IChatTrackingContext CreateTrackingContext()
        {
            if (customCommandHandler == null)
                throw new InvalidOperationException("RegisterCommandHandler() hasn't been called!");
 
            IChatTrackingContext context = null!;
            lock (mappedChannels)
                context = new ChatTrackingContext(
                    customCommandHandler,
                    mappedChannels.Select(y => y.Value.Channel),
                    loggerFactory.CreateLogger<ChatTrackingContext>(),
                    () =>
                    {
                        lock (trackingContexts)
                            trackingContexts.Remove(context);
                    });
 
            lock (trackingContexts)
                trackingContexts.Add(context);
 
            return context;
        }
 
        public async ValueTask UpdateTrackingContexts(CancellationToken cancellationToken)
        {
            var logMessageSent = 0;
            async Task UpdateTrackingContext(IChatTrackingContext channelSink, IEnumerable<ChannelRepresentation> channels)
            {
                if (Interlocked.Exchange(ref logMessageSent, 1) == 0)
 
                await channelSink.UpdateChannels(channels, cancellationToken);
            }
 
            var waitingForInitialConnection = !initialProviderConnectionsTask!.IsCompleted;
            if (waitingForInitialConnection)
            {
                logger.LogTrace("Waiting for initial chat bot connections before updating tracking contexts...");
                await initialProviderConnectionsTask.WaitAsync(cancellationToken);
            }
 
            List<Task> tasks;
            lock (mappedChannels)
                lock (trackingContexts)
                    tasks = trackingContexts.Select(x => UpdateTrackingContext(x, mappedChannels.Select(y => y.Value.Channel))).ToList();
 
            if (waitingForInitialConnection)
                if (tasks.Count > 0)
                    logger.LogTrace("Updating chat tracking contexts...");
                else
                    logger.LogTrace("No chat tracking contexts to update");
 
            await Task.WhenAll(tasks);
        }
 
        public void RegisterCommandHandler(ICustomCommandHandler customCommandHandler)
        {
            if (this.customCommandHandler != null)
                throw new InvalidOperationException("RegisterCommandHandler() already called!");
            this.customCommandHandler = customCommandHandler ?? throw new ArgumentNullException(nameof(customCommandHandler));
        }
 
        public async Task DeleteConnection(long connectionId, CancellationToken cancellationToken)
        {
            logger.LogTrace("DeleteConnection {connectionId}", connectionId);
            var hasSemaphore = changeChannelSemaphores.TryRemove(connectionId, out var semaphore);
            using (hasSemaphore
                ? semaphore
                : null)
            using (hasSemaphore
                ? await SemaphoreSlimContext.Lock(semaphore!, cancellationToken)
                : null)
            {
                var provider = await RemoveProviderChannels(connectionId, true, cancellationToken);
                if (provider != null)
                {
                    var startTime = DateTimeOffset.UtcNow;
                    try
                    {
                        await provider.Disconnect(cancellationToken);
                    }
                    catch (Exception ex)
                    {
                        logger.LogError(ex, "Error disconnecting connection {connectionId}!", connectionId);
                    }
 
                    await provider.DisposeAsync();
                    var duration = DateTimeOffset.UtcNow - startTime;
                    if (duration.TotalSeconds > 3)
                        logger.LogWarning("Disconnecting a {providerType} took {totalSeconds}s!", provider.GetType().Name, duration.TotalSeconds);
                }
                else
                    logger.LogTrace("DeleteConnection: ID {connectionId} doesn't exist!", connectionId);
            }
        }
 
        public ValueTask HandleRestart(Version? updateVersion, bool handlerMayDelayShutdownWithExtremelyLongRunningTasks, CancellationToken cancellationToken)
        {
            var message = updateVersion == null
                ? $"TGS: {(handlerMayDelayShutdownWithExtremelyLongRunningTasks ? "Graceful shutdown" : "Going down")}..."
                : $"TGS: Updating to version {updateVersion}...";
            List<ulong> systemChannels;
            lock (mappedChannels) // so it doesn't change while we're using it
                systemChannels = mappedChannels
                    .Where(x => x.Value.IsSystemChannel)
                    .Select(x => x.Key)
                    .ToList();
 
            return SendMessage(
                systemChannels,
                null,
                new MessageContent
                {
                    Text = message,
                },
                cancellationToken);
        }
 
        async ValueTask<IProvider?> RemoveProviderChannels(long connectionId, bool removeProvider, CancellationToken cancellationToken)
        {
            logger.LogTrace("RemoveProviderChannels {connectionId}...", connectionId);
            IProvider? provider;
            lock (providers)
            {
                if (!providers.TryGetValue(connectionId, out provider))
                {
                    logger.LogTrace("Aborted, no such provider!");
                    return null;
                }
 
                if (removeProvider)
                    providers.Remove(connectionId);
            }
 
            ValueTask trackingContextsUpdateTask;
            lock (mappedChannels)
            {
                foreach (var mappedConnectionChannel in mappedChannels.Where(x => x.Value.ProviderId == connectionId).Select(x => x.Key).ToList())
                    mappedChannels.Remove(mappedConnectionChannel);
 
                var newMappedChannels = mappedChannels.Select(y => y.Value.Channel).ToList();
 
                if (removeProvider)
                    lock (trackingContexts)
                        trackingContextsUpdateTask = ValueTaskExtensions.WhenAll(trackingContexts.Select(x => x.UpdateChannels(newMappedChannels, cancellationToken)));
                else
                    trackingContextsUpdateTask = ValueTask.CompletedTask;
            }
 
            await trackingContextsUpdateTask;
 
            return provider;
        }
 
        async ValueTask RemapProvider(IProvider provider, CancellationToken cancellationToken)
        {
            logger.LogTrace("Remapping channels for provider reconnection...");
            IEnumerable<Models.ChatChannel>? channelsToMap;
            long providerId;
            lock (providers)
                providerId = providers.Where(x => x.Value == provider).Select(x => x.Key).First();
 
            lock (activeChatBots)
                channelsToMap = activeChatBots.FirstOrDefault(x => x.Id == providerId)?.Channels;
 
            if (channelsToMap?.Any() ?? false)
                await ChangeChannels(providerId, channelsToMap, cancellationToken);
        }
 
#pragma warning disable CA1502
        async ValueTask ProcessMessage(IProvider provider, Message? message, bool recursed, CancellationToken cancellationToken)
#pragma warning restore CA1502
        {
            if (!provider.Connected)
            {
                logger.LogTrace("Abort message processing because provider is disconnected!");
                return;
            }
 
            // provider reconnected, remap channels.
            if (message == null)
            {
                await RemapProvider(provider, cancellationToken);
                return;
            }
 
            // map the channel if it's private and we haven't seen it
            var providerChannelId = message.User.Channel.RealId;
            KeyValuePair<ulong, ChannelMapping>? mappedChannel;
            long providerId;
            bool hasChannelZero;
            lock (providers)
            {
                // important, otherwise we could end up processing during shutdown
                cancellationToken.ThrowIfCancellationRequested();
 
                var providerIdNullable = providers
                    .Where(x => x.Value == provider)
                    .Select(x => (long?)x.Key)
                    .FirstOrDefault();
 
                if (!providerIdNullable.HasValue)
                {
                    // possible to have a message queued and then the provider immediately disconnects
                    logger.LogDebug("Unable to process command \"{command}\" due to provider disconnecting", message.Content);
                    return;
                }
 
                providerId = providerIdNullable.Value;
                mappedChannel = mappedChannels
                    .Where(x => x.Value.ProviderId == providerId && x.Value.ProviderChannelId == providerChannelId)
                    .Select(x => (KeyValuePair<ulong, ChannelMapping>?)x)
                    .FirstOrDefault();
                hasChannelZero = mappedChannels
                    .Where(x => x.Value.ProviderId == providerId && x.Value.ProviderChannelId == 0)
                    .Any();
            }
 
            if (!recursed && !mappedChannel.HasValue && !message.User.Channel.IsPrivateChannel && hasChannelZero)
            {
                logger.LogInformation("Receieved message from unmapped channel whose provider contains ID 0. Remapping...");
                await RemapProvider(provider, cancellationToken);
                logger.LogTrace("Resume processing original message...");
                await ProcessMessage(provider, message, true, cancellationToken);
                return;
            }
 
            ValueTask TextReply(string reply) => SendMessage(
                new List<ulong>
                {
                    message.User.Channel.RealId,
                },
                message,
                new MessageContent
                {
                    Text = reply,
                },
                cancellationToken);
 
            if (message.User.Channel.IsPrivateChannel)
                lock (mappedChannels)
                    if (!mappedChannel.HasValue)
                    {
                        ulong newId;
                        lock (synchronizationLock)
                            newId = channelIdCounter++;
                        logger.LogTrace(
                            "Mapping private channel {connectionName}:{channelFriendlyName} as {newId}",
                            message.User.Channel.ConnectionName,
                            message.User.FriendlyName,
                            newId);
                        mappedChannels.Add(newId, new ChannelMapping(message.User.Channel)
                        {
                            ProviderChannelId = message.User.Channel.RealId,
                            ProviderId = providerId,
                        });
 
                        logger.LogTrace(
                            "Mapping DM {connectionName}:{userId} ({userFriendlyName}) as {newId}",
                            message.User.Channel.ConnectionName,
                            message.User.RealId,
                            message.User.FriendlyName,
                            newId);
                        message.User.Channel.RealId = newId;
                    }
                    else
                        message.User.Channel.RealId = mappedChannel.Value.Key;
            else
            {
                if (!mappedChannel.HasValue)
                {
                    logger.LogError(
                        "Error mapping message: Provider ID: {providerId}, Channel Real ID: {realId}",
                        providerId,
                        message.User.Channel.RealId);
                    logger.LogTrace("message: {messageJson}", JsonConvert.SerializeObject(message));
                    lock (mappedChannels)
                        logger.LogTrace("mappedChannels: {mappedChannelsJson}", JsonConvert.SerializeObject(mappedChannels));
                    await TextReply("TGS: Processing error, check logs!");
                    return;
                }
 
                var mappingChannelRepresentation = mappedChannel.Value.Value.Channel;
 
                message.User.Channel.RealId = mappingChannelRepresentation.RealId;
                message.User.Channel.Tag = mappingChannelRepresentation.Tag;
                message.User.Channel.IsAdminChannel = mappingChannelRepresentation.IsAdminChannel;
            }
 
            var trimmedMessage = message.Content.Trim();
            if (trimmedMessage.Length == 0)
                return;
 
            var splits = new List<string>(trimmedMessage.Split(' ', StringSplitOptions.RemoveEmptyEntries));
            var address = splits[0];
            if (address.Length > 1 && (address.Last() == ':' || address.Last() == ','))
                address = address[0..^1];
 
            var addressed =
                address.Equals(CommonMention, StringComparison.OrdinalIgnoreCase)
                || address.Equals(provider.BotMention, StringComparison.OrdinalIgnoreCase);
 
            // no mention
            if (!addressed && !message.User.Channel.IsPrivateChannel)
                return;
 
            logger.LogTrace(
                "Start processing command: {message}. User (True provider Id): {profiderId}",
                message.Content,
                JsonConvert.SerializeObject(message.User));
            try
            {
                if (addressed)
                    splits.RemoveAt(0);
 
                if (splits.Count == 0)
                {
                    // just a mention
                    await TextReply("Hi!");
                    return;
                }
 
                var command = splits[0];
                splits.RemoveAt(0);
                var arguments = String.Join(" ", splits);
 
                Tuple<ICommand, IChatTrackingContext?>? GetCommand(string command)
                {
                    if (!builtinCommands.TryGetValue(command, out var handler))
                        return trackingContexts
                            .Where(trackingContext => trackingContext.Active)
                            .SelectMany(trackingContext => trackingContext.CustomCommands.Select(customCommand => Tuple.Create<ICommand, IChatTrackingContext?>(customCommand, trackingContext)))
                            .Where(tuple => tuple.Item1.Name.Equals(command, StringComparison.OrdinalIgnoreCase))
                            .FirstOrDefault();
 
                    return Tuple.Create<ICommand, IChatTrackingContext?>(handler, null);
                }
 
                const string UnknownCommandMessage = "TGS: Unknown command! Type '?' or 'help' for available commands.";
 
                if (command.Equals("help", StringComparison.OrdinalIgnoreCase) || command == "?")
                {
                    string helpText;
                    if (splits.Count == 0)
                    {
                        var allCommands = builtinCommands.Select(x => x.Value).ToList();
                        allCommands.AddRange(
                            trackingContexts
                                .SelectMany(
                                    x => x.CustomCommands));
                        helpText = String.Format(CultureInfo.InvariantCulture, "Available commands (Type '?' or 'help' and then a command name for more details): {0}", String.Join(", ", allCommands.Select(x => x.Name)));
                    }
                    else
                    {
                        var helpTuple = GetCommand(splits[0]);
                        if (helpTuple != default)
                        {
                            var (helpHandler, _) = helpTuple;
                            helpText = String.Format(CultureInfo.InvariantCulture, "{0}: {1}{2}", helpHandler.Name, helpHandler.HelpText, helpHandler.AdminOnly ? " - May only be used in admin channels" : String.Empty);
                        }
                        else
                            helpText = UnknownCommandMessage;
                    }
 
                    await TextReply(helpText);
                    return;
                }
 
                var tuple = GetCommand(command);
 
                if (tuple == default)
                {
                    await TextReply(UnknownCommandMessage);
                    return;
                }
 
                var (commandHandler, trackingContext) = tuple;
 
                if (trackingContext?.Active == false)
                {
                    await TextReply("TGS: The server is rebooting, please try again later");
                    return;
                }
 
                if (commandHandler.AdminOnly && !message.User.Channel.IsAdminChannel)
                {
                    await TextReply("TGS: Use this command in an admin channel!");
                    return;
                }
 
                var result = await commandHandler.Invoke(arguments, message.User, cancellationToken);
                if (result != null)
                    await SendMessage(new List<ulong> { message.User.Channel.RealId }, message, result, cancellationToken);
            }
            catch (OperationCanceledException ex)
            {
                logger.LogTrace(ex, "Command processing canceled!");
            }
            catch (Exception e)
            {
                // error bc custom commands should reply about why it failed
                logger.LogError(e, "Error processing chat command");
                await TextReply("TGS: Internal error processing command! Check server logs!");
            }
            finally
            {
                logger.LogTrace("Done processing command.");
            }
        }
 
        async Task MonitorMessages(CancellationToken cancellationToken)
        {
            logger.LogTrace("Starting processing loop...");
            var messageTasks = new Dictionary<IProvider, Task<Message?>>();
            ValueTask activeProcessingTask = ValueTask.CompletedTask;
            try
            {
                Task? updatedTask = null;
                while (!cancellationToken.IsCancellationRequested)
                {
                    if (updatedTask?.IsCompleted != false)
                        lock (synchronizationLock)
                            updatedTask = connectionsUpdated.Task;
 
                    // prune disconnected providers
                    foreach (var disposedProviderMessageTaskKvp in messageTasks.Where(x => x.Key.Disposed).ToList())
                        messageTasks.Remove(disposedProviderMessageTaskKvp.Key);
 
                    // add new ones
                    lock (providers)
                        foreach (var providerKvp in providers)
                            if (!messageTasks.ContainsKey(providerKvp.Value))
                                messageTasks.Add(
                                    providerKvp.Value,
                                    providerKvp.Value.NextMessage(cancellationToken));
 
                    if (messageTasks.Count == 0)
                    {
                        logger.LogTrace("No providers active, pausing messsage monitoring...");
                        await updatedTask.WaitAsync(cancellationToken);
                        logger.LogTrace("Resuming message monitoring...");
                        continue;
                    }
 
                    // wait for a message
                    await Task.WhenAny(updatedTask, Task.WhenAny(messageTasks.Select(x => x.Value)));
 
                    // process completed ones
                    foreach (var completedMessageTaskKvp in messageTasks.Where(x => x.Value.IsCompleted).ToList())
                    {
                        var provider = completedMessageTaskKvp.Key;
                        messageTasks.Remove(provider);
 
                        if (provider.Disposed) // valid to receive one, but don't process it
                            continue;
 
                        var message = await completedMessageTaskKvp.Value;
                        var messageNumber = Interlocked.Increment(ref messagesProcessed);
 
                        async ValueTask WrapProcessMessage()
                        {
                            var localActiveProcessingTask = activeProcessingTask;
                            using (LogContext.PushProperty(SerilogContextHelper.ChatMessageIterationContextProperty, messageNumber))
                                try
                                {
                                    await ProcessMessage(provider, message, false, cancellationToken);
                                }
                                catch (Exception ex)
                                {
                                    logger.LogError(ex, "Error processing message {messageNumber}!", messageNumber);
                                }
 
                            await localActiveProcessingTask;
                        }
 
                        activeProcessingTask = WrapProcessMessage();
                    }
                }
            }
            catch (OperationCanceledException ex)
            {
                logger.LogTrace(ex, "Message processing loop cancelled!");
            }
            catch (Exception e)
            {
                logger.LogError(e, "Message loop crashed!");
            }
            finally
            {
                await activeProcessingTask;
            }
 
            logger.LogTrace("Leaving message processing loop");
        }
 
        ValueTask SendMessage(IEnumerable<ulong> channelIds, Message? replyTo, MessageContent message, CancellationToken cancellationToken)
        {
            var channelIdsList = channelIds.ToList();
 
            logger.LogTrace(
                "Chat send \"{message}\"{embed} to channels: [{channelIdsCommaSeperated}]",
                message.Text,
                message.Embed != null ? " (with embed)" : String.Empty,
                String.Join(", ", channelIdsList));
 
            if (channelIdsList.Count == 0)
                return ValueTask.CompletedTask;
 
            return ValueTaskExtensions.WhenAll(
                channelIdsList.Select(x =>
                {
                    ChannelMapping? channelMapping;
                    lock (mappedChannels)
                        if (!mappedChannels.TryGetValue(x, out channelMapping))
                            return ValueTask.CompletedTask;
                    IProvider? provider;
                    lock (providers)
                        if (!providers.TryGetValue(channelMapping.ProviderId, out provider))
                            return ValueTask.CompletedTask;
                    return provider.SendMessage(replyTo, message, channelMapping.ProviderChannelId, cancellationToken);
                }));
        }
 
        async Task InitialConnection()
        {
            await Task.WhenAll(providers.Select(x => x.Value.InitialConnectionJob));
            logger.LogTrace("Initial provider connection task completed");
        }
 
        void AddMessageTask(Task task)
        {
            async Task Wrap(Task originalTask)
            {
                await originalTask;
                try
                {
                    await task;
                }
                catch (OperationCanceledException ex)
                {
                    logger.LogDebug(ex, "Async chat message cancelled!");
                }
                catch (Exception ex)
                {
                    logger.LogError(ex, "Error in asynchronous chat message!");
                }
            }
 
            lock (handlerCts)
                messageSendTask = Wrap(messageSendTask);
        }
 
        void QueueMessageInternal(MessageContent message, Func<IEnumerable<ulong>> channelIdsFactory, bool waitForConnections)
        {
            async Task SendMessageTask()
            {
                var cancellationToken = handlerCts.Token;
                if (waitForConnections)
                    await initialProviderConnectionsTask!.WaitAsync(cancellationToken);
 
                await SendMessage(
                    channelIdsFactory(),
                    null,
                    message,
                    cancellationToken);
            }
 
            AddMessageTask(SendMessageTask());
        }
 
        void QueueMessageGeneric(Predicate<ChannelMapping> channelSelector, string message, string? prefix)
        {
            ArgumentNullException.ThrowIfNull(message);
 
            if (prefix != null)
            {
                message = $"{prefix}: {message}";
            }
 
            if (!initialProviderConnectionsTask!.IsCompleted)
                logger.LogTrace("Waiting for initial provider connections before sending chat message...");
 
            // Reimplementing QueueMessage
            QueueMessageInternal(
                new MessageContent
                {
                    Text = message,
                },
                () =>
                {
                    // so it doesn't change while we're using it
                    lock (mappedChannels)
                        return mappedChannels.Where(x => channelSelector(x.Value)).Select(x => x.Key).ToList();
                },
                true);
        }
    }
}