ClaimsPrincipalExtensions.cs

Go to the documentation of this file.

using System;
using System.Globalization;
using System.Security.Claims;
 
using Microsoft.IdentityModel.JsonWebTokens;
using Microsoft.IdentityModel.Tokens;
 
namespace Tgstation.Server.Host.Extensions
{
    static class ClaimsPrincipalExtensions
    {
        public static long? GetTgsUserId(this ClaimsPrincipal principal)
        {
            ArgumentNullException.ThrowIfNull(principal);
 
            var userIdClaim = principal.FindFirst(JwtRegisteredClaimNames.Sub);
            if (userIdClaim == default)
                return null;
 
            long userId;
            try
            {
                userId = Int64.Parse(userIdClaim.Value, CultureInfo.InvariantCulture);
            }
            catch (Exception e)
            {
                throw new InvalidOperationException("Failed to parse user ID!", e);
            }
 
            return userId;
        }
 
        public static long RequireTgsUserId(this ClaimsPrincipal principal)
            => principal.GetTgsUserId() ?? throw new InvalidOperationException($"Missing '{JwtRegisteredClaimNames.Sub}' claim!");
 
        public static DateTimeOffset ParseTime(this ClaimsPrincipal principal, string claimName)
        {
            ArgumentNullException.ThrowIfNull(principal);
            ArgumentNullException.ThrowIfNull(claimName);
 
            var claim = principal.FindFirst(claimName);
            if (claim == null)
                throw new InvalidOperationException($"Missing '{claimName}' claim!");
 
            try
            {
                return new DateTimeOffset(
                    EpochTime.DateTime(
                        Int64.Parse(claim.Value, CultureInfo.InvariantCulture)));
            }
            catch (Exception ex)
            {
                throw new InvalidOperationException($"Failed to parse claim {claimName}: '{claim.Value}'!", ex);
            }
        }
    }
}