ConnectionMappingHub.cs

Go to the documentation of this file.

using System;
using System.Threading.Tasks;
 
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.SignalR;
 
using Tgstation.Server.Host.Security;
 
namespace Tgstation.Server.Host.Utils.SignalR
{
    [TgsAuthorize]
    abstract class ConnectionMappingHub<TChildHub, THubMethods> : Hub<THubMethods>
        where TChildHub : ConnectionMappingHub<TChildHub, THubMethods>
        where THubMethods : class
    {
        readonly IHubConnectionMapper<TChildHub, THubMethods> connectionMapper;
 
        readonly IAuthenticationContext authenticationContext;
 
        protected ConnectionMappingHub(
            IHubConnectionMapper<TChildHub, THubMethods> connectionMapper,
            IAuthenticationContext authenticationContext)
        {
            this.connectionMapper = connectionMapper ?? throw new ArgumentNullException(nameof(connectionMapper));
            this.authenticationContext = authenticationContext ?? throw new ArgumentNullException(nameof(authenticationContext));
        }
 
        public override async Task OnConnectedAsync()
        {
            await connectionMapper.UserConnected(authenticationContext, (TChildHub)this, Context.ConnectionAborted);
            await base.OnConnectedAsync();
        }
 
        [AllowAnonymous]
        public override Task OnDisconnectedAsync(Exception? exception)
        {
            connectionMapper.UserDisconnected(Context.ConnectionId);
            return base.OnDisconnectedAsync(exception);
        }
    }
}