Allow user to pass customized handlers (fixes #8)

stormpath · Sep 14, 2016 · d58da4d · d58da4d
1 parent 5a12652
commit d58da4d
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 2 deletions.
diff --git a/src/Stormpath.AspNet/StormpathAppBuilderExtensions.cs b/src/Stormpath.AspNet/StormpathAppBuilderExtensions.cs
@@ -42,7 +42,17 @@ public static IAppBuilder UseStormpath(this IAppBuilder app, StormpathMiddleware
                 Configuration = options?.Configuration,
                 ConfigurationFileRoot = AppDomain.CurrentDomain.BaseDirectory,
                 Logger = options?.Logger,
-                CacheProvider = options?.CacheProvider
+                CacheProvider = options?.CacheProvider,
+                PostChangePasswordHandler = options?.PostChangePasswordHandler,
+                PostLoginHandler = options?.PostLoginHandler,
+                PostLogoutHandler = options?.PostLogoutHandler,
+                PostRegistrationHandler = options?.PostRegistrationHandler,
+                PostVerifyEmailHandler = options?.PostVerifyEmailHandler,
+                PreChangePasswordHandler = options?.PreChangePasswordHandler,
+                PreLoginHandler = options?.PreLoginHandler,
+                PreLogoutHandler = options?.PreLogoutHandler,
+                PreRegistrationHandler = options?.PreRegistrationHandler,
+                PreVerifyEmailHandler = options?.PreVerifyEmailHandler,
             });
 
             app.Use(stormpathMiddleware);

diff --git a/src/Stormpath.AspNet/StormpathMiddlewareOptions.cs b/src/Stormpath.AspNet/StormpathMiddlewareOptions.cs
@@ -1,4 +1,8 @@
-using Stormpath.SDK.Cache;
+using System;
+using System.Threading;
+using System.Threading.Tasks;
+using Stormpath.Owin.Middleware;
+using Stormpath.SDK.Cache;
 using Stormpath.SDK.Logging;
 
 namespace Stormpath.AspNet
@@ -10,5 +14,25 @@ public sealed class StormpathMiddlewareOptions
         public ILogger Logger { get; set; }
 
         public ICacheProviderBuilder CacheProvider { get; set; }
+
+        public Func<PreChangePasswordContext, CancellationToken, Task> PreChangePasswordHandler { get; set; }
+
+        public Func<PostChangePasswordContext, CancellationToken, Task> PostChangePasswordHandler { get; set; }
+
+        public Func<PreLoginContext, CancellationToken, Task> PreLoginHandler { get; set; }
+
+        public Func<PostLoginContext, CancellationToken, Task> PostLoginHandler { get; set; }
+
+        public Func<PreLogoutContext, CancellationToken, Task> PreLogoutHandler { get; set; }
+
+        public Func<PostLogoutContext, CancellationToken, Task> PostLogoutHandler { get; set; }
+
+        public Func<PreRegistrationContext, CancellationToken, Task> PreRegistrationHandler { get; set; }
+
+        public Func<PostRegistrationContext, CancellationToken, Task> PostRegistrationHandler { get; set; }
+
+        public Func<PreVerifyEmailContext, CancellationToken, Task> PreVerifyEmailHandler { get; set; }
+
+        public Func<PostVerifyEmailContext, CancellationToken, Task> PostVerifyEmailHandler { get; set; }
     }
 }