hermes-client/Seven/Socket/Handlers/DispatchHandler.cs

using System.Text.Json;
using CommonSocketLibrary.Abstract;
using CommonSocketLibrary.Common;
using Serilog;
using TwitchChatTTS.Chat.Emotes;
using TwitchChatTTS.Seven.Socket.Data;

namespace TwitchChatTTS.Seven.Socket.Handlers
{
    public class DispatchHandler : IWebSocketHandler
    {
        public int OperationCode { get; } = 0;

        private readonly ILogger _logger;
        private readonly IEmoteDatabase _emotes;
        private readonly Mutex _lock;

        public DispatchHandler(IEmoteDatabase emotes, ILogger logger)
        {
            _emotes = emotes;
            _logger = logger;
            _lock = new Mutex();
        }

        public Task Execute<Data>(SocketClient<WebSocketMessage> sender, Data data)
        {
            if (data is not DispatchMessage message || message == null || message.Body == null)
                return Task.CompletedTask;

            ApplyChanges(message.Body.Pulled, cf => cf.OldValue, true);
            ApplyChanges(message.Body.Pushed, cf => cf.Value, false);
            ApplyChanges(message.Body.Removed, cf => cf.OldValue, true);
            ApplyChanges(message.Body.Updated, cf => cf.OldValue, false, cf => cf.Value);
            return Task.CompletedTask;
        }

        private void ApplyChanges(IEnumerable<ChangeField>? fields, Func<ChangeField, object> getter, bool removing, Func<ChangeField, object>? updater = null)
        {
            if (fields == null || !fields.Any() || removing && updater != null)
                return;

            foreach (var val in fields)
            {
                var value = getter(val);
                if (value == null)
                    continue;

                var o = JsonSerializer.Deserialize<EmoteField>(value.ToString()!, new JsonSerializerOptions()
                {
                    PropertyNameCaseInsensitive = false,
                    PropertyNamingPolicy = JsonNamingPolicy.SnakeCaseLower
                });
                if (o == null)
                    continue;

                try
                {
                    _lock.WaitOne();
                    if (removing)
                    {
                        if (_emotes.Get(o.Name) != o.Id)
                        {
                            _logger.Warning("Mismatched emote found while removing a 7tv emote.");
                            continue;
                        }
                        _emotes.Remove(o.Name);
                        _logger.Information($"Removed 7tv emote [name: {o.Name}][id: {o.Id}]");
                    }
                    else if (updater != null)
                    {
                        if (_emotes.Get(o.Name) != o.Id)
                        {
                            _logger.Warning("Mismatched emote found while updating a 7tv emote.");
                            continue;
                        }
                        _emotes.Remove(o.Name);
                        var update = updater(val);
                        if (update == null)
                            continue;

                        var u = JsonSerializer.Deserialize<EmoteField>(update.ToString()!, new JsonSerializerOptions()
                        {
                            PropertyNameCaseInsensitive = false,
                            PropertyNamingPolicy = JsonNamingPolicy.SnakeCaseLower
                        });

                        if (u != null)
                        {
                            _emotes.Add(u.Name, u.Id);
                            _logger.Information($"Updated 7tv emote [old name: {o.Name}][new name: {u.Name}][id: {u.Id}]");
                        }
                        else
                        {
                            _logger.Warning("Failed to update 7tv emote.");
                        }
                    }
                    else
                    {
                        _emotes.Add(o.Name, o.Id);
                        _logger.Information($"Added 7tv emote [name: {o.Name}][id: {o.Id}]");
                    }
                }
                finally
                {
                    _lock.ReleaseMutex();
                }
            }
        }
    }
}