Daybreak/Daybreak.API/Controllers/WebSocket/MainPlayer/MainPlayerStateRoute.cs

using System.Buffers;
using System.Core.Extensions;
using System.Extensions.Core;
using System.Net.WebSockets;
using Daybreak.API.Models;
using Daybreak.API.Services;
using Daybreak.API.WebSockets;
using Daybreak.Shared.Models.Api;

namespace Daybreak.API.Controllers.WebSocket.MainPlayer;

public sealed class MainPlayerStateRoute(
    MainPlayerService mainPlayerStateService,
    ChatService chatService,
    ILogger<MainPlayerStateRoute> logger)
        : UpdateWebSocketRoute
{
    private const int DefaultFrequency = 1000;

    private readonly MainPlayerService mainPlayerStateService = mainPlayerStateService.ThrowIfNull();
    private readonly ChatService chatService = chatService.ThrowIfNull();
    private readonly ILogger<MainPlayerStateRoute> logger = logger.ThrowIfNull();

    private CallbackRegistration? mainPlayerStateRegistration;

    protected override int BufferSize => 256;

    public override Task ExecuteAsync(WebSocketMessageType type, ReadOnlySequence<byte> data, CancellationToken cancellationToken)
    {
        return Task.CompletedTask;
    }

    public override async Task SocketAccepted(CancellationToken cancellationToken)
    {
        var scopedLogger = this.logger.CreateScopedLogger();
        if (!int.TryParse(this.Context?.Request.Query["f"].FirstOrDefault(), out var freq))
        {
            freq = DefaultFrequency;
        }

        var freqTimeSpan = TimeSpan.FromMilliseconds(freq);
        scopedLogger.LogInformation("WebSocket accepted {id} with frequency {frequency}", this.Context?.Connection.Id ?? string.Empty, freqTimeSpan);

        this.mainPlayerStateRegistration = this.mainPlayerStateService.RegisterMainStateConsumer(freqTimeSpan, this.SendUpdate);

        await this.chatService.AddMessageAsync(
            message: $"{nameof(MainPlayerState)} subscriber added: {this.Context?.Connection.RemoteIpAddress?.ToString()}:{this.Context?.Connection.RemotePort} with frequency {freq}ms",
            sender: "Daybreak.API",
            channel: Channel.Whisper,
            cancellationToken: cancellationToken);
    }

    public override async Task SocketClosed()
    {
        var scopedLogger = this.logger.CreateScopedLogger();

        this.mainPlayerStateRegistration?.Dispose();
        this.mainPlayerStateRegistration = default;

        scopedLogger.LogInformation("WebSocket closed {id}", this.Context?.Connection.Id ?? string.Empty);
        await this.chatService.AddMessageAsync(
            message: $"{nameof(MainPlayerState)} subscriber removed: {this.Context?.Connection.RemoteIpAddress?.ToString()}:{this.Context?.Connection.RemotePort}",
            sender: "Daybreak.API",
            channel: Channel.Whisper,
            cancellationToken: CancellationToken.None);
    }
}