using System.Collections.Concurrent;

namespace SameBeat.Api.Monitoring;

/// <summary>
/// Prosty licznik otwartych polaczen SignalR, trzymany w pamieci procesu (ETAP 8). Wystarczajacy dla
/// MVP jednej instancji backendu -- NIE jest to licznik rozproszony: przy kilku instancjach API za
/// load balancerem kazda miałaby wlasny, czesciowy stan. Rozproszony licznik (np. przez Redis) jest
/// poza zakresem MVP.
///
/// Kazde polaczenie SignalR trafia od razu do grupy "widzowie" w StanImprezyHub.OnConnectedAsync i --
/// na dzien dzisiejszy -- nigdy z niej nie jest usuwane, wiec licznik ogolny pokrywa sie z licznikiem
/// widzow. Grupa "host" jest podzbiorem: polaczenie dolacza do niej wylacznie przez wywolanie
/// DolaczDoGrupyHosta (po pozytywnej autoryzacji [Authorize] na tej metodzie).
///
/// Watkowo bezpieczne dzieki ConcurrentDictionary -- SignalR moze wywolywac OnConnectedAsync/
/// OnDisconnectedAsync dla wielu polaczen rownolegle.
/// </summary>
public sealed class LicznikPolaczenSignalR
{
    private readonly ConcurrentDictionary<string, byte> _polaczenia = new();
    private readonly ConcurrentDictionary<string, byte> _polaczeniaHost = new();

    /// <summary>Rejestruje nowe polaczenie (kazde polaczenie automatycznie trafia do grupy "widzowie").</summary>
    public void Polaczono(string idPolaczenia) => _polaczenia[idPolaczenia] = 0;

    /// <summary>Rejestruje dolaczenie polaczenia do grupy "host" (po autoryzacji hosta).</summary>
    public void DolaczylDoGrupyHosta(string idPolaczenia) => _polaczeniaHost[idPolaczenia] = 0;

    /// <summary>Usuwa polaczenie ze wszystkich sledzonych grup -- wywolywane przy rozlaczeniu.</summary>
    public void Rozlaczono(string idPolaczenia)
    {
        _polaczenia.TryRemove(idPolaczenia, out _);
        _polaczeniaHost.TryRemove(idPolaczenia, out _);
    }

    /// <summary>Liczba wszystkich aktualnie otwartych polaczen SignalR (uzywane w /api/health).</summary>
    public int LiczbaPolaczenOgolem => _polaczenia.Count;

    /// <summary>Liczba polaczen w grupie "widzowie" -- dzis rowna liczbie ogolem, patrz komentarz klasy.</summary>
    public int LiczbaPolaczenWidzowie => _polaczenia.Count;

    /// <summary>Liczba polaczen w grupie "host".</summary>
    public int LiczbaPolaczenHost => _polaczeniaHost.Count;
}