Added room functionality

This commit is contained in:
2023-03-15 11:52:58 +01:00
parent 390190cc35
commit 90e27f250c
6 changed files with 179 additions and 118 deletions

View File

@@ -0,0 +1,54 @@
using MauMau_Server.Websockets;
using Microsoft.AspNetCore.Mvc;
namespace MauMau_Server.Controllers;
[ApiController]
[Route("[controller]")]
public class RoomController : ControllerBase
{
private readonly ILogger<RoomController> _logger;
private readonly IRoomManager _roomManager;
public RoomController(ILogger<RoomController> logger, IRoomManager roomManager)
{
_logger = logger;
_roomManager = roomManager;
}
[HttpGet]
public IActionResult GetAll()
{
var rooms = _roomManager.GetAllRooms();
return Ok(rooms);
}
[HttpGet("{id}")]
public async Task ConnectToRoom(string id)
{
if (HttpContext.WebSockets.IsWebSocketRequest)
{
if (_roomManager.RoomExists(id))
{
using var webSocket = await HttpContext.WebSockets.AcceptWebSocketAsync();
var room = _roomManager.GetRoom(id);
await room.InstantiateConnection(webSocket);
}
else
{
HttpContext.Response.StatusCode = 404;
}
}
else
{
HttpContext.Response.StatusCode = 400;
}
}
[HttpPost]
public IActionResult Post()
{
var id = _roomManager.CreateRoom();
return Ok(id);
}
}

View File

@@ -1,79 +0,0 @@
using System.Net.WebSockets;
using System.Text;
using MauMau_Server.Websockets;
using Microsoft.AspNetCore.Mvc;
namespace MauMau_Server.Controllers;
[ApiController]
[Route("[controller]")]
public class WebSocketsController : ControllerBase
{
private readonly ILogger<WebSocketsController> _logger;
private readonly IWebsocketManager _websocketManager;
private string _id = string.Empty;
public WebSocketsController(ILogger<WebSocketsController> logger, IWebsocketManager websocketManager)
{
_logger = logger;
_websocketManager = websocketManager;
}
[HttpGet("/ws")]
public async Task Get()
{
if (HttpContext.WebSockets.IsWebSocketRequest)
{
using var webSocket = await HttpContext.WebSockets.AcceptWebSocketAsync();
_id = _websocketManager.AddConnection(webSocket);
_logger.Log(LogLevel.Information, $"WebSocket connection {_id} established");
await Echo(webSocket);
}
else
{
HttpContext.Response.StatusCode = 400;
}
}
private async Task Echo(WebSocket webSocket)
{
var buffer = EmptyBuffer();
var result = await ReceiveAsync(webSocket, buffer);
while (!result.CloseStatus.HasValue)
{
var message = Encoding.UTF8.GetString(buffer);
BroadcastAsync(message);
buffer = EmptyBuffer();
result = await ReceiveAsync(webSocket, buffer);
}
await webSocket.CloseAsync(result.CloseStatus.Value, result.CloseStatusDescription, CancellationToken.None);
_websocketManager.RemoveConnection(_id);
_logger.Log(LogLevel.Information, $"WebSocket connection {_id} closed");
}
private async Task<WebSocketReceiveResult?> ReceiveAsync(WebSocket webSocket, byte[] buffer)
{
var arraySegment = new ArraySegment<byte>(buffer);
var result = await webSocket.ReceiveAsync(arraySegment, CancellationToken.None);
_logger.Log(LogLevel.Information, $"Message received from Client {_id}");
return result;
}
private void BroadcastAsync(string message)
{
var bytes = Encoding.Default.GetBytes($"{_id}: {message}");
var arraySegment = new ArraySegment<byte>(bytes);
foreach (var (id, socket) in _websocketManager.GetAllConnections())
{
socket.SendAsync(arraySegment, WebSocketMessageType.Text, true, CancellationToken.None);
_logger.Log(LogLevel.Information, $"Message from Client {_id} sent to Client {id}");
}
}
private static byte[] EmptyBuffer()
{
return new byte[4096];
}
}