Merge pull request #101 from matteofabbri/MongoStoreExtension

AddMongoDbStores extension to register only stores.

src/AspNetCore.Identity.Mongo/Mongo/TypeConverterResolver.cs

@@ -0,0 +1,18 @@
+using System;
+using System.Collections.Generic;
+using System.ComponentModel;
+using System.Text;
+
+namespace AspNetCore.Identity.Mongo.Mongo
+{
+    internal static class TypeConverterResolver
+    {
+        internal static void RegisterTypeConverter<T, TC>() where TC : TypeConverter
+        {
+            Attribute[] attr = new Attribute[1];
+            TypeConverterAttribute vConv = new TypeConverterAttribute(typeof(TC));
+            attr[0] = vConv;
+            TypeDescriptor.AddAttributes(typeof(T), attr);
+        }
+    }
+}

src/AspNetCore.Identity.Mongo/MongoIdentityExtensions.cs

@@ -100,7 +100,7 @@ public static IdentityBuilder AddIdentityMongoDbProvider<TUser, TRole, TKey>(thi
             // register custom ObjectId TypeConverter
             if (typeof(TKey) == typeof(ObjectId))
             {
-                RegisterTypeConverter<ObjectId, ObjectIdConverter>();
+                TypeConverterResolver.RegisterTypeConverter<ObjectId, ObjectIdConverter>();
             }
 
             // Identity Services
@@ -109,13 +109,5 @@ public static IdentityBuilder AddIdentityMongoDbProvider<TUser, TRole, TKey>(thi
 
             return builder;
         }
-
-        private static void RegisterTypeConverter<T, TC>() where TC : TypeConverter
-        {
-            Attribute[] attr = new Attribute[1];
-            TypeConverterAttribute vConv = new TypeConverterAttribute(typeof(TC));
-            attr[0] = vConv;
-            TypeDescriptor.AddAttributes(typeof(T), attr);
-        }
     }
 }

src/AspNetCore.Identity.Mongo/MongoStoreExtensions.cs

@@ -0,0 +1,55 @@
+using AspNetCore.Identity.Mongo.Migrations;
+using AspNetCore.Identity.Mongo.Model;
+using AspNetCore.Identity.Mongo.Mongo;
+using AspNetCore.Identity.Mongo.Stores;
+using Microsoft.AspNetCore.Identity;
+using Microsoft.Extensions.DependencyInjection;
+using MongoDB.Bson;
+using System;
+using System.Collections.Generic;
+using System.ComponentModel;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace AspNetCore.Identity.Mongo
+{
+    public static class MongoStoreExtensions
+    {
+        public static IdentityBuilder AddMongoDbStores<TUser>(this IdentityBuilder builder, Action<MongoIdentityOptions> setupDatabaseAction)
+            where TUser : MongoUser
+        {
+            return AddMongoDbStores<TUser, MongoRole, ObjectId>(builder, setupDatabaseAction);
+        }
+
+        public static IdentityBuilder AddMongoDbStores<TUser, TRole, TKey>(this IdentityBuilder builder, Action<MongoIdentityOptions> setupDatabaseAction)
+            where TKey : IEquatable<TKey>
+            where TUser : MongoUser<TKey>
+            where TRole : MongoRole<TKey>
+        {
+            var dbOptions = new MongoIdentityOptions();
+            setupDatabaseAction(dbOptions);
+
+            var migrationCollection = MongoUtil.FromConnectionString<MigrationHistory>(dbOptions, dbOptions.MigrationCollection);
+
+            Task.WaitAny(Migrator.Apply(migrationCollection));
+
+            var userCollection = MongoUtil.FromConnectionString<TUser>(dbOptions, dbOptions.UsersCollection);
+            var roleCollection = MongoUtil.FromConnectionString<TRole>(dbOptions, dbOptions.RolesCollection);
+
+            builder.Services.AddSingleton(x => userCollection);
+            builder.Services.AddSingleton(x => roleCollection);
+
+            // register custom ObjectId TypeConverter
+            if (typeof(TKey) == typeof(ObjectId))
+            {
+                TypeConverterResolver.RegisterTypeConverter<ObjectId, ObjectIdConverter>();
+            }
+
+            // Identity Services
+            builder.Services.AddTransient<IRoleStore<TRole>>(x => new RoleStore<TRole, TKey>(roleCollection));
+            builder.Services.AddTransient<IUserStore<TUser>>(x => new UserStore<TUser, TRole, TKey>(userCollection, new RoleStore<TRole, TKey>(roleCollection), x.GetService<ILookupNormalizer>()));
+
+            return builder;
+        }
+    }
+}