Added call to PrincipalCreated after authenticating with a refresh token

azzlack · May 21, 2015 · f36037d · f36037d
1 parent 874d0f9
commit f36037d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/src/Sentinel.OAuth.Core/Models/SentinelAuthorizationServerEvents.cs b/src/Sentinel.OAuth.Core/Models/SentinelAuthorizationServerEvents.cs
@@ -14,8 +14,9 @@ public class SentinelAuthorizationServerEvents
         public Func<TokenIssuedEventArgs, Task> TokenIssued;
 
         /// <summary>
-        /// Activated when the user is logged in and the principal is created. Use this event to add any
-        /// custom claims to the user before the token is created.
+        /// Activated when the user is logged in (either via a username and password, or a refresh token)
+        /// and the principal is created. Use this event to add any custom claims to the user before the
+        /// token is created.
         /// </summary>
         public Func<PrincipalCreatedEventArgs, Task> PrincipalCreated;
 

diff --git a/src/Sentinel.OAuth/Providers/OAuth/SentinelAuthorizationServerProvider.cs b/src/Sentinel.OAuth/Providers/OAuth/SentinelAuthorizationServerProvider.cs
@@ -345,7 +345,19 @@ public override async Task GrantRefreshToken(OAuthGrantRefreshTokenContext conte
         {
             this.options.Logger.Debug("Authenticating refresh token flow");
 
-            context.Validated();
+            var user = new SentinelPrincipal(context.Ticket.Identity);
+
+            // Activate event if subscribed to
+            if (this.options.Events.PrincipalCreated != null)
+            {
+                var args = new PrincipalCreatedEventArgs(user, context);
+
+                await this.options.Events.PrincipalCreated(args);
+
+                user = new SentinelPrincipal(args.Principal);
+            }
+
+            context.Validated(user.Identity.AsClaimsIdentity());
         }
 
         /// <summary>Called before the TokenEndpoint redirects its response to the caller.</summary>