TgsRestAuthorizeAttribute{TAuthority}.cs

Go to the documentation of this file.

using System;
using System.Reflection;
 
using Microsoft.AspNetCore.Authorization;
 
using Tgstation.Server.Host.Authority.Core;
 
namespace Tgstation.Server.Host.Security
{
    [AttributeUsage(AttributeTargets.Method, AllowMultiple = true, Inherited = true)]
    public sealed class TgsRestAuthorizeAttribute<TAuthority> : AuthorizeAttribute
        where TAuthority : IAuthority
    {
        public string MethodName { get; }
 
        public TgsRestAuthorizeAttribute(string methodName)
        {
            ArgumentNullException.ThrowIfNull(methodName);
 
            var authorityType = typeof(TAuthority);
            var authorityMethod = authorityType.GetMethod(methodName)
                ?? throw new InvalidOperationException($"Could not find method {methodName} on {authorityType}!");
 
            var authorizeAttribute = authorityMethod.GetCustomAttribute<TgsAuthorizeAttribute>()
                ?? throw new InvalidOperationException($"Could not find method {authorityType}.{methodName}() has no {nameof(TgsAuthorizeAttribute)}!");
 
            MethodName = methodName;
            Roles = authorizeAttribute.Roles;
        }
    }
}