hermes-server/Socket/Handlers/RequestHandler.cs

using HermesSocketLibrary.Socket.Data;
using HermesSocketServer.Requests;
using ILogger = Serilog.ILogger;

namespace HermesSocketServer.Socket.Handlers
{
    public class RequestHandler : ISocketHandler
    {
        public int OperationCode { get; } = 3;
        private readonly IRequestManager _requests;
        private readonly HermesSocketManager _sockets;
        private readonly ILogger _logger;

        public RequestHandler(IRequestManager requests, HermesSocketManager sockets, ILogger logger)
        {
            _requests = requests;
            _sockets = sockets;
            _logger = logger;
        }


        public async Task Execute<T>(WebSocketUser sender, T message, HermesSocketManager sockets)
        {
            if (message is not RequestMessage data || sender.Id == null)
                return;
            if (sender.Slave && !sender.WebLogin && !data.Type.StartsWith("get"))
            {
                _logger.Warning($"Received a non-get request from a slave client [request type: {data.Type}][sender id: {sender.Id}][sender session: {sender.SessionId}]");
                return;
            }

            RequestResult? result = null;
            _logger.Debug("Executing request handler: " + data.Type);
            try
            {
                result = await _requests.Grant(sender.Id, data);
            }
            catch (Exception e)
            {
                _logger.Error(e, $"Failed to grant a request of type '{data.Type}'.");
            }

            if (result == null)
                return;

            var ack = new RequestAckMessage()
            {
                Request = data,
                Data = result.Success ? result.Result : null,
                Nounce = data.Nounce,
                Error = result.Success ? null : result.Result?.ToString()
            };

            if (!result.NotifyClientsOnAccount || !result.Success)
            {
                await sender.Send(4, ack);
                return;
            }

            var recipients = _sockets.GetSockets(sender.Id);
            foreach (var socket in recipients)
            {
                try
                {
                    _logger.Verbose($"Sending {data.Type} to socket [ip: {socket.IPAddress}].");
                    await socket.Send(4, ack);
                }
                catch (Exception)
                {
                    _logger.Warning($"Failed to send {data.Type} to socket [ip: {socket.IPAddress}].");
                }
            }
        }
    }
}