using System.Linq;
using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using API.Data;
using API.SignalR.Presence;
using Microsoft.AspNetCore.SignalR;
namespace API.SignalR;
///
/// Responsible for ushering events to the UI and allowing simple DI hook to send data
///
public interface IEventHub
{
Task SendMessageAsync(string method, SignalRMessage message, bool onlyAdmins = true);
Task SendMessageToAsync(string method, SignalRMessage message, int userId);
}
public class EventHub : IEventHub
{
private readonly IHubContext _messageHub;
private readonly IPresenceTracker _presenceTracker;
public EventHub(IHubContext messageHub, IPresenceTracker presenceTracker)
{
_messageHub = messageHub;
_presenceTracker = presenceTracker;
// TODO: When sending a message, queue the message up and on re-connect, reply the queued messages. Queue messages expire on a rolling basis (rolling array)
}
public async Task SendMessageAsync(string method, SignalRMessage message, bool onlyAdmins = true)
{
// TODO: If libraryId and NOT onlyAdmins, then perform RBS check before sending the event
var users = _messageHub.Clients.All;
if (onlyAdmins)
{
var admins = await _presenceTracker.GetOnlineAdminIds();
users = _messageHub.Clients.Users(admins.Select(i => i.ToString()).ToArray());
}
await users.SendAsync(method, message);
}
///
/// Sends a message directly to a user if they are connected
///
///
///
///
///
public async Task SendMessageToAsync(string method, SignalRMessage message, int userId)
{
await _messageHub.Clients.Users(new List() {userId + string.Empty}).SendAsync(method, message);
}
}