Timeline/Controllers/ControllerAuthExtensions.cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

using Microsoft.AspNetCore.Mvc;
using System;
using System.Security.Claims;
using Timeline.Auth;
using static Timeline.Resources.Controllers.ControllerAuthExtensions;

namespace Timeline.Controllers
{
    public static class ControllerAuthExtensions
    {
        public static bool IsAdministrator(this ControllerBase controller)
        {
            return controller.User != null && controller.User.IsAdministrator();
        }

        public static long GetUserId(this ControllerBase controller)
        {
            var claim = controller.User.FindFirst(ClaimTypes.NameIdentifier);
            if (claim == null)
                throw new InvalidOperationException(ExceptionNoUserIdentifierClaim);

            if (long.TryParse(claim.Value, out var value))
                return value;

            throw new InvalidOperationException(ExceptionUserIdentifierClaimBadFormat);
        }

        public static long? GetOptionalUserId(this ControllerBase controller)
        {
            var claim = controller.User.FindFirst(ClaimTypes.NameIdentifier);
            if (claim == null)
                return null;

            if (long.TryParse(claim.Value, out var value))
                return value;

            throw new InvalidOperationException(ExceptionUserIdentifierClaimBadFormat);
        }
    }
}