AuthenticationContextFactory.cs

Go to the documentation of this file.

using System;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
 
using Tgstation.Server.Host.Configuration;
using Tgstation.Server.Host.Database;
using Tgstation.Server.Host.Models;
 
namespace Tgstation.Server.Host.Security
{
    sealed class AuthenticationContextFactory : IAuthenticationContextFactory, IDisposable
    {
        public IAuthenticationContext CurrentAuthenticationContext => currentAuthenticationContext;
 
        readonly IDatabaseContext databaseContext;
 
        readonly IIdentityCache identityCache;
 
        readonly ILogger<AuthenticationContextFactory> logger;
 
        readonly SwarmConfiguration swarmConfiguration;
 
        readonly AuthenticationContext currentAuthenticationContext;
 
        int initialized;
 
        public AuthenticationContextFactory(
            IDatabaseContext databaseContext,
            IIdentityCache identityCache,
            IOptions<SwarmConfiguration> swarmConfigurationOptions,
            ILogger<AuthenticationContextFactory> logger)
        {
            this.databaseContext = databaseContext ?? throw new ArgumentNullException(nameof(databaseContext));
            this.identityCache = identityCache ?? throw new ArgumentNullException(nameof(identityCache));
            swarmConfiguration = swarmConfigurationOptions?.Value ?? throw new ArgumentNullException(nameof(swarmConfigurationOptions));
            this.logger = logger ?? throw new ArgumentNullException(nameof(logger));
 
            currentAuthenticationContext = new AuthenticationContext();
        }
 
        public void Dispose() => currentAuthenticationContext.Dispose();
 
        public async ValueTask<IAuthenticationContext> CreateAuthenticationContext(long userId, long? instanceId, DateTimeOffset notBefore, CancellationToken cancellationToken)
        {
            if (Interlocked.Exchange(ref initialized, 1) != 0)
                throw new InvalidOperationException("Authentication context has already been loaded");
 
            var user = await databaseContext
                .Users
                .AsQueryable()
                .Where(x => x.Id == userId)
                .Include(x => x.CreatedBy)
                .Include(x => x.PermissionSet)
                .Include(x => x.Group)
                    .ThenInclude(x => x!.PermissionSet)
                .Include(x => x.OAuthConnections)
                .FirstOrDefaultAsync(cancellationToken);
            if (user == default)
            {
                logger.LogWarning("Unable to find user with ID {userId}!", userId);
                return currentAuthenticationContext;
            }
 
            ISystemIdentity? systemIdentity;
            if (user.SystemIdentifier != null)
                systemIdentity = identityCache.LoadCachedIdentity(user);
            else
            {
                if (user.LastPasswordUpdate.HasValue && user.LastPasswordUpdate >= notBefore)
                {
                    logger.LogDebug("Rejecting token for user {userId} created before last password update: {lastPasswordUpdate}", userId, user.LastPasswordUpdate.Value);
                    return currentAuthenticationContext;
                }
 
                systemIdentity = null;
            }
 
            var userPermissionSet = user.PermissionSet ?? user.Group!.PermissionSet;
            try
            {
                InstancePermissionSet? instancePermissionSet = null;
                if (instanceId.HasValue)
                {
                    instancePermissionSet = await databaseContext.InstancePermissionSets
                        .AsQueryable()
                        .Where(x => x.PermissionSetId == userPermissionSet!.Id && x.InstanceId == instanceId && x.Instance!.SwarmIdentifer == swarmConfiguration.Identifier)
                        .Include(x => x.Instance)
                        .FirstOrDefaultAsync(cancellationToken);
 
                    if (instancePermissionSet == null)
                        logger.LogDebug("User {userId} does not have permissions on instance {instanceId}!", userId, instanceId.Value);
                }
 
                currentAuthenticationContext.Initialize(
                    systemIdentity,
                    user,
                    instancePermissionSet);
                return currentAuthenticationContext;
            }
            catch
            {
                systemIdentity?.Dispose();
                throw;
            }
        }
    }
}