Ragon/Ragon.Server/Sources/RagonServer.cs

/*
 * Copyright 2023-2024 Eduard Kargin <kargin.eduard@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

using System.Diagnostics;
using Ragon.Protocol;
using Ragon.Server.Handler;
using Ragon.Server.IO;
using Ragon.Server.Lobby;
using Ragon.Server.Logging;
using Ragon.Server.Plugin;
using Ragon.Server.Project;
using Ragon.Server.Time;

namespace Ragon.Server;

public class RagonServer : IRagonServer, INetworkListener
{
  private const string ServerVersion = "1.4.3";

  private readonly IRagonLogger _logger = LoggerManager.GetLogger(nameof(RagonServer));
  private readonly INetworkServer _server;
  private readonly BaseOperation[] _handlers;
  private readonly IRagonLobby _lobby;
  private readonly IServerPlugin _serverPlugin;
  private readonly RagonServerConfiguration _configuration;
  private readonly RagonBuffer _reader;
  private readonly RagonBuffer _writer;
  private readonly RagonScheduler _scheduler;
  private readonly Dictionary<ushort, RagonContext> _contextsByConnection;
  private readonly Dictionary<string, RagonContext> _contextsByPlayerId;
  private readonly ProjectRegistry _projectRegistry;
  private readonly RagonLobbyDispatcher _lobbySerializer;
  private bool _isRunning = false;

  public bool IsRunning => _isRunning;
  public ProjectRegistry ProjectRegistry => _projectRegistry;

  public RagonServer(
    INetworkServer server,
    IServerPlugin plugin,
    RagonServerConfiguration configuration)
  {
    _server = server;
    _configuration = configuration;
    _serverPlugin = plugin;
    _contextsByConnection = new Dictionary<ushort, RagonContext>();
    _contextsByPlayerId = new Dictionary<string, RagonContext>();
    _projectRegistry = new ProjectRegistry(configuration.LimitConnectionsPerProject);
    _lobby = new LobbyInMemory();
    _lobbySerializer = new RagonLobbyDispatcher(_lobby);
    _scheduler = new RagonScheduler();
    _reader = new RagonBuffer();
    _writer = new RagonBuffer();

    var contextObserver = new RagonContextObserver(_contextsByPlayerId);
    _scheduler.Run(new RagonActionTimer(SendRoomList, 2.0f));
    _scheduler.Run(new RagonActionTimer(SendPlayerUserData, 0.1f));
    _scheduler.Run(new RagonActionTimer(SendRoomUserData, 0.1f));
    _scheduler.Run(new RagonActionTimer(SendTimestamp, 1.0f / _configuration.ServerTickRate));
    
    _serverPlugin.OnAttached(this);

    _handlers = new BaseOperation[byte.MaxValue];
    _handlers[(byte)RagonOperation.AUTHORIZE] = new AuthorizationOperation(_reader, _writer, this, _serverPlugin, contextObserver, configuration, _projectRegistry);
    _handlers[(byte)RagonOperation.JOIN_OR_CREATE_ROOM] = new RoomJoinOrCreateOperation(_reader, _writer, plugin, _configuration);
    _handlers[(byte)RagonOperation.CREATE_ROOM] = new RoomCreateOperation(_reader, _writer, plugin, _configuration);
    _handlers[(byte)RagonOperation.JOIN_ROOM] = new RoomJoinOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.LEAVE_ROOM] = new RoomLeaveOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.LOAD_SCENE] = new SceneLoadOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.SCENE_LOADED] = new SceneLoadedOperation(_reader, _writer, _configuration);
    _handlers[(byte)RagonOperation.CREATE_ENTITY] = new EntityCreateOperation(_reader, _writer, _configuration);
    _handlers[(byte)RagonOperation.REMOVE_ENTITY] = new EntityDestroyOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.REPLICATE_ENTITY_EVENT] = new EntityEventOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.REPLICATE_ENTITY_STATE] = new EntityStateOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.TRANSFER_ROOM_OWNERSHIP] = new EntityOwnershipOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.TRANSFER_ENTITY_OWNERSHIP] = new EntityOwnershipOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.TIMESTAMP_SYNCHRONIZATION] = new TimestampSyncOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.REPLICATE_ROOM_EVENT] = new RoomEventOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.REPLICATE_RAW_DATA] = new RoomDataOperation(_reader, _writer);
    _handlers[(byte)RagonOperation.ROOM_DATA_UPDATED] = new RoomUserDataOperation(_reader, _writer, _configuration.LimitUserDataSize);
    _handlers[(byte)RagonOperation.PLAYER_DATA_UPDATED] = new PlayerUserDataOperation(_reader, _writer, _configuration.LimitUserDataSize);
  }
  public void Tick()
  {
    var deltaTime = 1.0f / _configuration.ServerTickRate;
    _scheduler.Update(deltaTime);
    _server.Update();
  }

  public void Start(bool executeInDedicatedThread = false)
  {
    CopyrightInfo();

    var networkConfiguration = new NetworkConfiguration()
    {
      LimitConnections = _configuration.LimitConnections,
      Protocol = RagonVersion.Parse(_configuration.Protocol),
      Address = _configuration.ServerAddress,
      Port = _configuration.Port,
    };

    _server.Listen(this, networkConfiguration);
    _serverPlugin.OnAttached(this);

    _isRunning = true;
  }

  public void Dispose()
  {
    _serverPlugin.OnDetached();
    _server.Stop();
  }

  public void OnConnected(INetworkConnection connection)
  {
    var context = new RagonContext(connection, _lobby, _scheduler, _configuration.LimitBufferedEvents);

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

  public void OnDisconnected(INetworkConnection connection)
  {
    if (_contextsByConnection.Remove(connection.Id, out var context))
    {
      var room = context.Room;
      if (room != null)
      {
        room.DetachPlayer(context.RoomPlayer);

        _lobby.RemoveIfEmpty(room);
      }

      if (context.ConnectionStatus == ConnectionStatus.Authorized)
      {
        _contextsByPlayerId.Remove(context.LobbyPlayer.Id);
        _projectRegistry.UnregisterConnection(context.LobbyPlayer.ProjectId);
      }

      _logger.Trace($"Disconnected: {connection.Id}");
    }
    else
    {
      _logger.Trace($"Disconnected without context: {connection.Id}");
    }
  }

  public void OnTimeout(INetworkConnection connection)
  {
    if (_contextsByConnection.Remove(connection.Id, out var context) && context.ConnectionStatus == ConnectionStatus.Authorized)
    {
      var room = context.Room;
      if (room != null)
      {
        room.DetachPlayer(context.RoomPlayer);
        _lobby.RemoveIfEmpty(room);
      }

      if (context.ConnectionStatus == ConnectionStatus.Authorized)
      {
        _contextsByPlayerId.Remove(context.LobbyPlayer.Id);
        _projectRegistry.UnregisterConnection(context.LobbyPlayer.ProjectId);
      }

      _logger.Trace($"Timeout: {connection.Id}|{context.LobbyPlayer.Name}|{context.LobbyPlayer.Id}");
    }
    else
    {
      _logger.Trace($"Timeout: {connection.Id}");
    }
  }

  public void OnData(INetworkConnection connection, NetworkChannel channel, byte[] data)
  {
    try
    {
      if (_contextsByConnection.TryGetValue(connection.Id, out var context))
      {
        _writer.Clear();
        _reader.Clear();
        _reader.FromArray(data);

        var operation = _reader.ReadByte();

        if (operation >= _handlers.Length || _handlers[operation] == null)
        {
          _logger.Warning($"Invalid operation code: {operation} from connection {connection.Id}");
          return;
        }

        _handlers[operation].Handle(context, channel);
      }
    }
    catch (Exception ex)
    {
      _logger.Error(ex);
    }
  }

  public void SendTimestamp()
  {
    var timestamp = RagonTime.CurrentTimestamp();
    var value = new DoubleToUInt
    {
      Double = timestamp,
    };

    _writer.Clear();
    _writer.WriteOperation(RagonOperation.TIMESTAMP_SYNCHRONIZATION);
    _writer.Write(value.Int0, 32);
    _writer.Write(value.Int1, 32);

    var sendData = _writer.ToArray();
    _server.Broadcast(sendData, NetworkChannel.UNRELIABLE);
  }

  public void SendRoomList()
  {
    foreach (var (_, context) in _contextsByPlayerId)
    {
      if (context.Room == null) // If only in lobby, then send room list data
      {
        _lobbySerializer.Write(_writer, context.LobbyPlayer.ProjectId);
        var sendData = _writer.ToArray();
        context.Connection.Reliable.Send(sendData);
      }
    }
  }

  public void SendPlayerUserData()
  {
    foreach (var (_, value) in _contextsByPlayerId)
    {
      if (value.UserData.IsDirty)
      {
        _writer.Clear();
        _writer.WriteOperation(RagonOperation.PLAYER_DATA_UPDATED);
        _writer.WriteUShort(value.Connection.Id);
        
        value.UserData.Write(_writer);
        
        var sendData = _writer.ToArray();
        _server.Broadcast(sendData, NetworkChannel.RELIABLE);
      }
    }
  }
  
  public void SendRoomUserData()
  {
    foreach (var room in _lobby.Rooms)
    {
      if (room.UserData.IsDirty)
      {
        _writer.Clear();
        _writer.WriteOperation(RagonOperation.ROOM_DATA_UPDATED);
        
        room.UserData.Write(_writer);
        
        var sendData = _writer.ToArray();
        _server.Broadcast(sendData, NetworkChannel.RELIABLE);
      }
    }
  }

  public BaseOperation ResolveHandler(RagonOperation operation)
  {
    return _handlers[(byte)operation];
  }
  
  public RagonContext? GetContextByConnectionId(ushort peerId)
  {
    return _contextsByConnection.TryGetValue(peerId, out var context) ? context : null;
  }

  public RagonContext? GetContextById(string playerId)
  {
    return _contextsByPlayerId.TryGetValue(playerId, out var context) ? context : null;
  }

  private void CopyrightInfo()
  {
    _logger.Info($"Server Version: {ServerVersion}");
    _logger.Info($"Machine Name: {Environment.MachineName}");
    _logger.Info($"OS: {Environment.OSVersion}");
    _logger.Info($"Processors: {Environment.ProcessorCount}");
    _logger.Info($"Runtime Version: {Environment.Version}");
    _logger.Info($"Server Tick Rate: {_configuration.ServerTickRate}");
  }
}