Ragon/Ragon.Core/Application.cs

using System.Diagnostics;
using NLog;
using Ragon.Common;
using Ragon.Core.Lobby;
using Ragon.Core.Server;
using Ragon.Core.Time;
using Ragon.Server;
using Ragon.Server.ENet;

namespace Ragon.Core;

public class Application : INetworkListener
{
  private readonly Logger _logger = LogManager.GetCurrentClassLogger();
  private readonly INetworkServer _server;
  private readonly Thread _dedicatedThread;
  private readonly Executor _executor;
  private readonly Configuration _configuration;
  private readonly HandlerRegistry _handlerRegistry;
  private readonly ILobby _lobby;
  private readonly Loop _loop;
  private readonly Dictionary<ushort, PlayerContext> _contexts;

  public Application(Configuration configuration)
  {
    _configuration = configuration;
    _executor = new Executor();
    _dedicatedThread = new Thread(Execute);
    _dedicatedThread.IsBackground = true;
    _contexts = new Dictionary<ushort, PlayerContext>();
    _handlerRegistry = new HandlerRegistry();
    _lobby = new LobbyInMemory();
    _loop = new Loop();

    if (configuration.ServerType == "enet")
      _server = new ENetServer();
    
    if (configuration.ServerType == "websocket")
      _server = new NativeWebSocketServer(_executor);

    Debug.Assert(_server != null, $"Socket type not supported: {configuration.ServerType}. Supported: [enet, websocket]");
  }

  public void Execute()
  {
    while (true)
    {
      _executor.Execute();
      _loop.Tick();
      _server.Poll();

      Thread.Sleep((int)1000.0f / _configuration.ServerTickRate);
    }
  }

  public void Start()
  {
    var networkConfiguration = new NetworkConfiguration()
    {
      LimitConnections = _configuration.LimitConnections,
      Protocol = RagonVersion.Parse(_configuration.GameProtocol),
      Address = "0.0.0.0",
      Port = _configuration.Port,
    };

    _server.Start(this, networkConfiguration);
    _dedicatedThread.Start();
  }

  public void Stop()
  {
    _server.Stop();
    _dedicatedThread.Interrupt();
  }

  public void OnConnected(INetworkConnection connection)
  {
    var context = new PlayerContext(connection, new LobbyPlayer(connection));
    context.Lobby = _lobby;
    context.Loop = _loop;

    _logger.Trace($"Connected {connection.Id}");
    _contexts.Add(connection.Id, context);
  }

  public void OnDisconnected(INetworkConnection connection)
  {
    _logger.Trace($"Disconnected {connection.Id}");

    if (_contexts.Remove(connection.Id, out var context))
    {
      var room = context.Room;
      if (room != null)
      {
        room.RemovePlayer(context.RoomPlayer);

        _lobby.RemoveIfEmpty(room);
      }

      context.Dispose();
    }
  }

  public void OnTimeout(INetworkConnection connection)
  {
    if (_contexts.Remove(connection.Id, out var context))
      context.Dispose();
  }

  public void OnData(INetworkConnection connection, byte[] data)
  {
    if (_contexts.TryGetValue(connection.Id, out var context))
      _handlerRegistry.Handle(context, data);
  }
}