mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 06:17:14 -04:00
54 lines
1.9 KiB
C#
54 lines
1.9 KiB
C#
|
using System.Security.Claims;
|
||
|
using System.Text.Encodings.Web;
|
||
|
using System.Threading.Tasks;
|
||
|
using MediaBrowser.Controller.Net;
|
||
|
using Microsoft.AspNetCore.Authentication;
|
||
|
using Microsoft.Extensions.Logging;
|
||
|
using Microsoft.Extensions.Options;
|
||
|
|
||
|
namespace Jellyfin.Api.Auth
|
||
|
{
|
||
|
public class CustomAuthenticationHandler : AuthenticationHandler<AuthenticationSchemeOptions>
|
||
|
{
|
||
|
private readonly IAuthService _authService;
|
||
|
|
||
|
public CustomAuthenticationHandler(
|
||
|
IAuthService authService,
|
||
|
IOptionsMonitor<AuthenticationSchemeOptions> options,
|
||
|
ILoggerFactory logger,
|
||
|
UrlEncoder encoder,
|
||
|
ISystemClock clock) : base(options, logger, encoder, clock)
|
||
|
{
|
||
|
_authService = authService;
|
||
|
}
|
||
|
|
||
|
protected override Task<AuthenticateResult> HandleAuthenticateAsync()
|
||
|
{
|
||
|
var authenticatedAttribute = new AuthenticatedAttribute();
|
||
|
try
|
||
|
{
|
||
|
var user = _authService.Authenticate(Request, authenticatedAttribute);
|
||
|
if (user == null)
|
||
|
{
|
||
|
return Task.FromResult(AuthenticateResult.Fail("Invalid user"));
|
||
|
}
|
||
|
|
||
|
var claims = new[]
|
||
|
{
|
||
|
new Claim(ClaimTypes.Name, user.Name),
|
||
|
new Claim(ClaimTypes.Role, user.Policy.IsAdministrator ? "Administrator" : "User"),
|
||
|
};
|
||
|
var identity = new ClaimsIdentity(claims, Scheme.Name);
|
||
|
var principal = new ClaimsPrincipal(identity);
|
||
|
var ticket = new AuthenticationTicket(principal, Scheme.Name);
|
||
|
|
||
|
return Task.FromResult(AuthenticateResult.Success(ticket));
|
||
|
}
|
||
|
catch (SecurityException ex)
|
||
|
{
|
||
|
return Task.FromResult(AuthenticateResult.Fail(ex));
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|