From 79ab2b304d93b1029515bd3f954db4e5a73f4168 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 30 Jan 2020 20:26:52 +0800
Subject: ...

---
 Timeline/Controllers/ControllerAuthExtensions.cs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'Timeline/Controllers/ControllerAuthExtensions.cs')

diff --git a/Timeline/Controllers/ControllerAuthExtensions.cs b/Timeline/Controllers/ControllerAuthExtensions.cs
index 81fd2428..90da8a93 100644
--- a/Timeline/Controllers/ControllerAuthExtensions.cs
+++ b/Timeline/Controllers/ControllerAuthExtensions.cs
@@ -26,5 +26,20 @@ namespace Timeline.Controllers
 
             throw new InvalidOperationException("Failed to get user id because NameIdentifier claim is not a number.");
         }
+
+        public static long? GetOptionalUserId(this ControllerBase controller)
+        {
+            if (controller.User == null)
+                return 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.");
+        }
     }
 }
-- 
cgit v1.2.3