hermes-server/Server.cs

using System.Net.WebSockets;
using System.Text;
using System.Text.Json;
using HermesSocketLibrary.Socket.Data;
using HermesSocketServer.Socket;
using ILogger = Serilog.ILogger;

namespace HermesSocketLibrary
{
    public class Server
    {
        private readonly HermesSocketManager _sockets;
        private readonly SocketHandlerManager _handlers;
        private readonly JsonSerializerOptions _options;
        private readonly ILogger _logger;


        public Server(
            HermesSocketManager sockets,
            SocketHandlerManager handlers,
            JsonSerializerOptions options,
            ILogger logger
        )
        {
            _sockets = sockets;
            _handlers = handlers;
            _options = options;
            _logger = logger;
        }


        public async Task Handle(WebSocketUser socket, HttpContext context)
        {
            _logger.Information($"Socket connected [ip: {socket.IPAddress}]");
            _sockets.Add(socket);
            var buffer = new byte[1024 * 8];

            while (socket.State == WebSocketState.Open)
            {
                try
                {
                    var result = await socket.Receive(new ArraySegment<byte>(buffer));
                    if (result == null || result.MessageType == WebSocketMessageType.Close || !socket.Connected)
                        break;

                    string message = Encoding.UTF8.GetString(buffer, 0, result.Count).TrimEnd('\0');
                    var obj = JsonSerializer.Deserialize<SocketMessage>(message, _options);
                    if (obj == null || !obj.OpCode.HasValue)
                        continue;

                    if (obj.OpCode != 0)
                        _logger.Information("Message: " + message);

                    /**
                    * 0: Heartbeat
                    * 1: Login RX
                    * 2: Login Ack TX
                    * 3: Request RX
                    * 4: Request Ack TX
                    * 5: Error RX/TX
                    */
                    if (obj.Data == null)
                    {
                        await socket.Send(5, new ErrorMessage("Received no data in the message."));
                        continue;
                    }
                    else if (obj.OpCode == 0)
                        obj.Data = JsonSerializer.Deserialize<HeartbeatMessage>(obj.Data.ToString(), _options);
                    else if (obj.OpCode == 1)
                        obj.Data = JsonSerializer.Deserialize<HermesLoginMessage>(obj.Data.ToString(), _options);
                    else if (obj.OpCode == 3)
                        obj.Data = JsonSerializer.Deserialize<RequestMessage>(obj.Data.ToString(), _options);
                    else if (obj.OpCode == 5)
                        obj.Data = JsonSerializer.Deserialize<ErrorMessage>(obj.Data.ToString(), _options);
                    else if (obj.OpCode == 6)
                        obj.Data = JsonSerializer.Deserialize<ChatterMessage>(obj.Data.ToString(), _options);
                    else if (obj.OpCode == 7)
                        obj.Data = JsonSerializer.Deserialize<EmoteDetailsMessage>(obj.Data.ToString(), _options);
                    else if (obj.OpCode == 8)
                        obj.Data = JsonSerializer.Deserialize<EmoteUsageMessage>(obj.Data.ToString(), _options);
                    else
                    {
                        await socket.Send(5, new ErrorMessage("Received an invalid message: " + message));
                        continue;
                    }
                    await _handlers.Execute(socket, obj.OpCode.Value, obj.Data);
                }
                catch (Exception e)
                {
                    _logger.Error(e, "Error trying to process a socket message");
                }
            }

            try
            {
                if (socket.Connected)
                    await socket.Close(socket.CloseStatus ?? WebSocketCloseStatus.NormalClosure, socket.CloseStatusDescription, CancellationToken.None);
            }
            catch (Exception)
            {
                _sockets.Remove(socket);
            }
            _logger.Information($"Client disconnected [username: {socket.Name}][id: {socket.Id}][ip: {socket.IPAddress}]");
        }
    }
}