...

1 files changed, 30 insertions, 0 deletions

diff --git a/Timeline/Controllers/ControllerAuthExtensions.cs b/Timeline/Controllers/ControllerAuthExtensions.cs
new file mode 100644
index 00000000..81fd2428
--- /dev/null
+++ b/Timeline/Controllers/ControllerAuthExtensions.cs
@@ -0,0 +1,30 @@
+using Microsoft.AspNetCore.Mvc;

+using System.Security.Claims;

+using Timeline.Auth;

+using System;

+

+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)

+        {

+            if (controller.User == null)

+                throw new InvalidOperationException("Failed to get user id because User is null.");

+

+            var claim = controller.User.FindFirst(ClaimTypes.NameIdentifier);

+            if (claim == null)

+                throw new InvalidOperationException("Failed to get user id because User has no NameIdentifier claim.");

+

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

+                return value;

+

+            throw new InvalidOperationException("Failed to get user id because NameIdentifier claim is not a number.");

+        }

+    }

+}