From 3c739b6e4244c7dba7f111f8e1573ac31eaae0e9 Mon Sep 17 00:00:00 2001
From: Young <iamlovedit@163.com>
Date: Thu, 10 Oct 2024 11:06:00 +0800
Subject: [PATCH] renamed and optimised

---
 src/Infrastructure/Security/ITokenBuilder.cs     | 16 ++++++++--------
 .../Security/{TokenInfo.cs => JwtTokenInfo.cs}   |  2 +-
 2 files changed, 9 insertions(+), 9 deletions(-)
 rename src/Infrastructure/Security/{TokenInfo.cs => JwtTokenInfo.cs} (70%)

diff --git a/src/Infrastructure/Security/ITokenBuilder.cs b/src/Infrastructure/Security/ITokenBuilder.cs
index b809158..a8339bc 100644
--- a/src/Infrastructure/Security/ITokenBuilder.cs
+++ b/src/Infrastructure/Security/ITokenBuilder.cs
@@ -14,7 +14,7 @@ public interface ITokenBuilder
 
     void SetUserContext(TokenValidatedContext context);
 
-    TokenInfo GenerateJwtTokenInfo(IReadOnlyCollection<Claim> claims);
+    JwtTokenInfo GenerateJwtTokenInfo(IReadOnlyCollection<Claim> claims);
 }
 
 public class TokenBuilder(
@@ -46,16 +46,16 @@ public class TokenBuilder(
                           throw new NullReferenceException(nameof(IUserContext));
         var principal = context.Principal ?? throw new NullReferenceException(nameof(context.Principal));
         userContext.Id = long.Parse(
-            principal.Claims.FirstOrDefault(c => c.Type == JwtRegisteredClaimNames.NameId)?.Value);
+            principal.Claims.First(c => c.Type == JwtRegisteredClaimNames.NameId).Value);
         userContext.Username =
-            principal.Claims.FirstOrDefault(c => c.Type == JwtRegisteredClaimNames.UniqueName)?.Value!;
-        userContext.Name = principal.Claims.FirstOrDefault(c => c.Type == JwtRegisteredClaimNames.Name)?.Value!;
-        userContext.Email = principal.Claims.FirstOrDefault(c => c.Type == JwtRegisteredClaimNames.Email)?.Value!;
+            principal.Claims.First(c => c.Type == JwtRegisteredClaimNames.UniqueName).Value;
+        userContext.Name = principal.Claims.First(c => c.Type == JwtRegisteredClaimNames.Name).Value;
+        userContext.Email = principal.Claims.First(c => c.Type == JwtRegisteredClaimNames.Email).Value;
         userContext.RoleIds = principal.Claims.Where(c => c.Type == ClaimTypes.Role).Select(c => c.Value).ToArray();
-        userContext.RemoteIpAddress = context.HttpContext.GetRequestIp();
+        userContext.RemoteIpAddress = context.HttpContext.GetRequestIp()!;
     }
 
-    public TokenInfo GenerateJwtTokenInfo(IReadOnlyCollection<Claim> claims)
+    public JwtTokenInfo GenerateJwtTokenInfo(IReadOnlyCollection<Claim> claims)
     {
         var jwtToken = new JwtSecurityToken(
             issuer: jwtOptions.Issuer,
@@ -65,7 +65,7 @@ public class TokenBuilder(
             expires: DateTime.Now.AddSeconds(jwtOptions.Expiration),
             signingCredentials: jwtOptions.SigningCredentials);
         var token = jwtSecurityTokenHandler.WriteToken(jwtToken);
-        return new TokenInfo(encryptionService.Encrypt(token), jwtOptions.Expiration,
+        return new JwtTokenInfo(encryptionService.Encrypt(token), jwtOptions.Expiration,
             JwtBearerDefaults.AuthenticationScheme);
     }
 }
\ No newline at end of file
diff --git a/src/Infrastructure/Security/TokenInfo.cs b/src/Infrastructure/Security/JwtTokenInfo.cs
similarity index 70%
rename from src/Infrastructure/Security/TokenInfo.cs
rename to src/Infrastructure/Security/JwtTokenInfo.cs
index 9db0294..8404a98 100644
--- a/src/Infrastructure/Security/TokenInfo.cs
+++ b/src/Infrastructure/Security/JwtTokenInfo.cs
@@ -1,6 +1,6 @@
 namespace Infrastructure.Security;
 
-public class TokenInfo(string token, double expiredIn, string tokenType)
+public class JwtTokenInfo(string token, double expiredIn, string tokenType)
 {
     public string? Token { get; } = token;