From d458ca6486d4a4ed0e8493ec6f3d8c4afce6b6a7 Mon Sep 17 00:00:00 2001
From: JudahGabriel <judahgabriel@gmail.com>
Date: Tue, 5 Sep 2017 12:25:38 -0500
Subject: [PATCH] Added overload for AddRavenDbIdentity to allow identity
 configuration options.

---
 RavenDB.Identity/RavenDB.Identity.csproj      |  2 +-
 .../ServiceCollectionExtensions.cs            | 20 +++++++++++++++++++
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/RavenDB.Identity/RavenDB.Identity.csproj b/RavenDB.Identity/RavenDB.Identity.csproj
index f43b5bc..5556bcd 100644
--- a/RavenDB.Identity/RavenDB.Identity.csproj
+++ b/RavenDB.Identity/RavenDB.Identity.csproj
@@ -18,7 +18,7 @@
     <GenerateAssemblyConfigurationAttribute>false</GenerateAssemblyConfigurationAttribute>
     <GenerateAssemblyCompanyAttribute>false</GenerateAssemblyCompanyAttribute>
     <GenerateAssemblyProductAttribute>false</GenerateAssemblyProductAttribute>
-    <Version>2.0.2</Version>
+    <Version>2.0.3</Version>
     <GeneratePackageOnBuild>False</GeneratePackageOnBuild>
   </PropertyGroup>
 
diff --git a/RavenDB.Identity/ServiceCollectionExtensions.cs b/RavenDB.Identity/ServiceCollectionExtensions.cs
index 9c5b6c4..cfd01dd 100644
--- a/RavenDB.Identity/ServiceCollectionExtensions.cs
+++ b/RavenDB.Identity/ServiceCollectionExtensions.cs
@@ -94,5 +94,25 @@ namespace RavenDB.Identity
 
             return services;
         }
+
+        /// <summary>
+        /// Registers a RavenDB as the user store.
+        /// </summary>
+        /// <typeparam name="TUser">The type of user. This should be a class you created derived from <see cref="IdentityUser"/>.</typeparam>
+        /// <param name="services"></param>
+        /// <param name="setupAction">Identity options configuration.</param>
+        /// <returns>The same service collection so that multiple calls can be chained.</returns>
+        public static IServiceCollection AddRavenDbIdentity<TUser>(this IServiceCollection services, Action<IdentityOptions> setupAction)
+            where TUser : IdentityUser
+        {
+            // Add the AspNet identity system to work with our RavenDB identity objects.
+            services.AddIdentity<TUser, IdentityRole>(setupAction)
+                .AddDefaultTokenProviders();
+
+            services.AddScoped<Microsoft.AspNetCore.Identity.IUserStore<TUser>, UserStore<TUser>>();
+            services.AddScoped<Microsoft.AspNetCore.Identity.IRoleStore<IdentityRole>, RoleStore<IdentityRole>>();
+
+            return services;
+        }
     }
 }