ChatTrackingContext.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.Extensions.Logging;
 
using Tgstation.Server.Host.Components.Chat.Commands;
using Tgstation.Server.Host.Utils;
 
namespace Tgstation.Server.Host.Components.Chat
{
    sealed class ChatTrackingContext : DisposeInvoker, IChatTrackingContext
    {
        public bool Active
        {
            get => active && !IsDisposed;
            set
            {
                if (active == value)
                    return;
                if (value)
                    logger.LogTrace("Activated");
                else
                    logger.LogTrace("Deactivated");
 
                active = value;
            }
        }
 
        public IReadOnlyCollection<ChannelRepresentation> Channels { get; private set; }
 
        public IEnumerable<CustomCommand> CustomCommands
        {
            get => customCommands;
            set
            {
                customCommands = (value ?? throw new InvalidOperationException("value cannot be null!"))
                .Select(customCommand =>
                {
                    customCommand.SetHandler(customCommandHandler);
                    return customCommand;
                })
                .ToList();
                logger.LogTrace("Custom commands set.");
            }
        }
 
        readonly ICustomCommandHandler customCommandHandler;
 
        readonly ILogger<ChatTrackingContext> logger;
 
        readonly object synchronizationLock;
 
        volatile IChannelSink? channelSink;
 
        IReadOnlyCollection<CustomCommand> customCommands;
 
        bool active;
 
        public ChatTrackingContext(
            ICustomCommandHandler customCommandHandler,
            IEnumerable<ChannelRepresentation> initialChannels,
            ILogger<ChatTrackingContext> logger,
            Action disposeAction)
            : base(disposeAction)
        {
            this.customCommandHandler = customCommandHandler ?? throw new ArgumentNullException(nameof(customCommandHandler));
            Channels = initialChannels?.ToList() ?? throw new ArgumentNullException(nameof(initialChannels));
            this.logger = logger ?? throw new ArgumentNullException(nameof(logger));
 
            synchronizationLock = new object();
            Active = true;
            customCommands = Array.Empty<CustomCommand>();
        }
 
        public void SetChannelSink(IChannelSink channelSink)
        {
            ArgumentNullException.ThrowIfNull(channelSink);
 
            var originalValue = Interlocked.CompareExchange(ref this.channelSink, channelSink, null);
            if (originalValue != null)
                throw new InvalidOperationException("channelSink already set!");
        }
 
        public ValueTask UpdateChannels(IEnumerable<ChannelRepresentation> newChannels, CancellationToken cancellationToken)
        {
            logger.LogTrace("UpdateChannels...");
            var completed = newChannels.ToList();
            ValueTask updateTask;
            lock (synchronizationLock)
            {
                Channels = completed;
                updateTask = channelSink?.UpdateChannels(newChannels, cancellationToken) ?? ValueTask.CompletedTask;
            }
 
            return updateTask;
        }
    }
}