You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
1.4 KiB
40 lines
1.4 KiB
4 years ago
|
using System;
|
||
|
using System.Threading.Tasks;
|
||
|
using Microsoft.AspNetCore.Authorization;
|
||
|
using Microsoft.Extensions.Options;
|
||
|
using NzbDrone.Core.Configuration;
|
||
|
|
||
|
namespace NzbDrone.Http.Authentication
|
||
|
{
|
||
|
public class UiAuthorizationPolicyProvider : IAuthorizationPolicyProvider
|
||
|
{
|
||
|
private const string POLICY_NAME = "UI";
|
||
|
private readonly IConfigFileProvider _config;
|
||
|
|
||
|
public DefaultAuthorizationPolicyProvider FallbackPolicyProvider { get; }
|
||
|
|
||
|
public UiAuthorizationPolicyProvider(IOptions<AuthorizationOptions> options,
|
||
|
IConfigFileProvider config)
|
||
|
{
|
||
|
FallbackPolicyProvider = new DefaultAuthorizationPolicyProvider(options);
|
||
|
_config = config;
|
||
|
}
|
||
|
|
||
|
public Task<AuthorizationPolicy> GetDefaultPolicyAsync() => FallbackPolicyProvider.GetDefaultPolicyAsync();
|
||
|
|
||
|
public Task<AuthorizationPolicy> GetFallbackPolicyAsync() => FallbackPolicyProvider.GetFallbackPolicyAsync();
|
||
|
|
||
|
public Task<AuthorizationPolicy> GetPolicyAsync(string policyName)
|
||
|
{
|
||
|
if (policyName.Equals(POLICY_NAME, StringComparison.OrdinalIgnoreCase))
|
||
|
{
|
||
|
var policy = new AuthorizationPolicyBuilder(_config.AuthenticationMethod.ToString())
|
||
|
.RequireAuthenticatedUser();
|
||
|
return Task.FromResult(policy.Build());
|
||
|
}
|
||
|
|
||
|
return FallbackPolicyProvider.GetPolicyAsync(policyName);
|
||
|
}
|
||
|
}
|
||
|
}
|