AuthorityInvokerBase{TAuthority}.cs

Go to the documentation of this file.

using System;
using System.Linq;
using System.Threading.Tasks;
 
using Microsoft.AspNetCore.Authorization;
 
namespace Tgstation.Server.Host.Authority.Core
{
    abstract class AuthorityInvokerBase<TAuthority> : IAuthorityInvoker<TAuthority>
        where TAuthority : IAuthority
    {
        protected TAuthority Authority { get; }
 
        readonly Security.IAuthorizationService authorizationService;
 
        public AuthorityInvokerBase(
            TAuthority authority,
            Security.IAuthorizationService authorizationService)
        {
            Authority = authority ?? throw new ArgumentNullException(nameof(authority));
            this.authorizationService = authorizationService ?? throw new ArgumentNullException(nameof(authorizationService));
        }
 
        async ValueTask<IQueryable<TResult>?> IAuthorityInvoker<TAuthority>.InvokeQueryable<TResult>(Func<TAuthority, RequirementsGated<IQueryable<TResult>>> authorityInvoker)
        {
            ArgumentNullException.ThrowIfNull(authorityInvoker);
 
            var requirementsGate = authorityInvoker(Authority);
            return await ExecuteIfRequirementsSatisfied(requirementsGate);
        }
 
        protected async ValueTask<TResult?> ExecuteIfRequirementsSatisfied<TResult>(RequirementsGated<TResult> requirementsGate)
            where TResult : class
        {
            var requirements = await requirementsGate.GetRequirements();
            var authorizationResult = await authorizationService.AuthorizeAsync(requirements, requirementsGate.InstanceId);
 
            if (!authorizationResult.Succeeded)
            {
                OnRequirementsFailure(authorizationResult.Failure);
                return null;
            }
 
            return await requirementsGate.Execute(authorizationService);
        }
 
        protected virtual void OnRequirementsFailure(AuthorizationFailure authFailure)
        {
        }
    }
}