Merge pull request #107 from TurnerSoftware/drop-bsonclassmap-automap

Drop BsonClassMap AutoMap

Author: Turnerj

src/MongoFramework/Infrastructure/Mapping/DefaultMappingPack.cs

@@ -13,13 +13,12 @@ private DefaultMappingPack()
 			{
 				new CollectionNameProcessor(),
 				new HierarchyProcessor(),
+				new PropertyMappingProcessor(),
 				new EntityIdProcessor(),
-				new MappedPropertiesProcessor(),
 				new NestedTypeProcessor(),
 				new ExtraElementsProcessor(),
 				new TypeDiscoveryProcessor(),
 				new BsonKnownTypesProcessor(),
-				new ClassMapPropertiesProcessor(),
 				new IndexProcessor(),
 				new EntityRelationshipProcessor()
 			};

src/MongoFramework/Infrastructure/Mapping/EntityMapping.cs

@@ -67,9 +67,7 @@ public static IEntityDefinition RegisterType(Type entityType)
 					try
 					{
 						var classMap = new BsonClassMap(entityType);
-
 						BsonClassMap.RegisterClassMap(classMap);
-						classMap.AutoMap();
 
 						foreach (var processor in MappingProcessors)
 						{

src/MongoFramework/Infrastructure/Mapping/Processors/ClassMapPropertiesProcessor.cs

@@ -15,29 +15,37 @@ public void ApplyMapping(IEntityDefinition definition, BsonClassMap classMap)
 			var entityType = definition.EntityType;
 
 			//Find the first property with the "Key" attribute to use as the Id
-			var properties = entityType.GetProperties(BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly);
-			var idProperty = properties.Where(p => p.GetCustomAttribute<KeyAttribute>() != null).FirstOrDefault();
-			if (idProperty != null)
+			var properties = definition.Properties;
+			var idProperty = properties.Where(p => p.PropertyInfo.GetCustomAttribute<KeyAttribute>() != null).FirstOrDefault();
+			if (idProperty == null)
 			{
-				classMap.MapIdMember(idProperty);
+				idProperty = properties
+					.Where(p => p.ElementName.Equals("id", StringComparison.InvariantCultureIgnoreCase))
+					.FirstOrDefault();
 			}
 
-			//If there is no Id generator, set a default based on the member type
-			if (classMap.IdMemberMap != null && classMap.IdMemberMap.IdGenerator == null)
+			if (idProperty is EntityProperty entityProperty)
 			{
-				var idMemberMap = classMap.IdMemberMap;
-				var memberType = BsonClassMap.GetMemberInfoType(idMemberMap.MemberInfo);
-				if (memberType == typeof(string))
-				{
-					idMemberMap.SetIdGenerator(StringObjectIdGenerator.Instance);
-				}
-				else if (memberType == typeof(Guid))
-				{
-					idMemberMap.SetIdGenerator(CombGuidGenerator.Instance);
-				}
-				else if (memberType == typeof(ObjectId))
+				classMap.MapIdMember(idProperty.PropertyInfo);
+				entityProperty.IsKey = true;
+
+				//If there is no Id generator, set a default based on the member type
+				if (classMap.IdMemberMap.IdGenerator == null)
 				{
-					idMemberMap.SetIdGenerator(ObjectIdGenerator.Instance);
+					var idMemberMap = classMap.IdMemberMap;
+					var memberType = BsonClassMap.GetMemberInfoType(idMemberMap.MemberInfo);
+					if (memberType == typeof(string))
+					{
+						idMemberMap.SetIdGenerator(StringObjectIdGenerator.Instance);
+					}
+					else if (memberType == typeof(Guid))
+					{
+						idMemberMap.SetIdGenerator(CombGuidGenerator.Instance);
+					}
+					else if (memberType == typeof(ObjectId))
+					{
+						idMemberMap.SetIdGenerator(ObjectIdGenerator.Instance);
+					}
 				}
 			}
 		}

src/MongoFramework/Infrastructure/Mapping/Processors/EntityIdProcessor.cs

@@ -16,6 +16,8 @@ public void ApplyMapping(IEntityDefinition definition, BsonClassMap classMap)
 		{
 			definition.Relationships = GetEntityRelationships(definition).ToArray();
 
+			var removeProperties = new HashSet<string>();
+
 			foreach (var relationship in definition.Relationships)
 			{
 				if (relationship.IsCollection)
@@ -27,9 +29,13 @@ public void ApplyMapping(IEntityDefinition definition, BsonClassMap classMap)
 				}
 				else
 				{
+					removeProperties.Add(relationship.NavigationProperty.FullPath);
 					classMap.UnmapMember(relationship.NavigationProperty.PropertyInfo);
 				}
 			}
+
+			//Remove navigation properties
+			definition.Properties = definition.Properties.Where(p => !removeProperties.Contains(p.FullPath)).ToArray();
 		}
 		private IEnumerable<IEntityRelationship> GetEntityRelationships(IEntityDefinition definition)
 		{

src/MongoFramework/Infrastructure/Mapping/Processors/EntityRelationshipProcessor.cs

@@ -22,18 +22,20 @@ public void ApplyMapping(IEntityDefinition definition, BsonClassMap classMap)
 			}
 			else
 			{
+				classMap.SetIgnoreExtraElements(false);
+
 				//If any of the Entity's properties have the "ExtraElementsAttribute", assign that against the BsonClassMap
-				var extraElementsProperty = entityType.GetProperties(BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly)
+				var extraElementsProperty = definition.Properties
 					.Select(p => new
 					{
-						PropertyInfo = p,
-						ExtraElementsAttribute = p.GetCustomAttribute<ExtraElementsAttribute>()
+						Property = p,
+						ExtraElementsAttribute = p.PropertyInfo.GetCustomAttribute<ExtraElementsAttribute>()
 					}).Where(p => p.ExtraElementsAttribute != null).FirstOrDefault();
 
-				if (extraElementsProperty != null && typeof(IDictionary<string, object>).IsAssignableFrom(extraElementsProperty.PropertyInfo.PropertyType))
+				if (extraElementsProperty != null && typeof(IDictionary<string, object>).IsAssignableFrom(extraElementsProperty.Property.PropertyType))
 				{
 					var memberMap = classMap.DeclaredMemberMaps
-						.Where(m => m.MemberInfo == extraElementsProperty.PropertyInfo)
+						.Where(m => m.ElementName == extraElementsProperty.Property.ElementName)
 						.FirstOrDefault();
 					classMap.SetExtraElementsMember(memberMap);
 				}

src/MongoFramework/Infrastructure/Mapping/Processors/ExtraElementsProcessor.cs

@@ -12,7 +12,7 @@ public class NestedTypeProcessor : IMappingProcessor
 		public void ApplyMapping(IEntityDefinition definition, BsonClassMap classMap)
 		{
 			var entityType = definition.EntityType;
-			var properties = entityType.GetProperties(BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly);
+			var properties = definition.Properties;
 
 			foreach (var property in properties)
 			{

src/MongoFramework/Infrastructure/Mapping/Processors/MappedPropertiesProcessor.cs

@@ -0,0 +1,67 @@
+using MongoDB.Bson.Serialization;
+using System;
+using System.ComponentModel.DataAnnotations.Schema;
+using System.Linq;
+using System.Reflection;
+
+namespace MongoFramework.Infrastructure.Mapping.Processors
+{
+	public class PropertyMappingProcessor : IMappingProcessor
+	{
+		public void ApplyMapping(IEntityDefinition definition, BsonClassMap classMap)
+		{
+			var entityType = definition.EntityType;
+			var properties = entityType.GetProperties(BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly);
+
+			foreach (var property in properties)
+			{
+				if (!property.CanRead || !property.CanWrite)
+				{
+					continue;
+				}
+
+				//Skip overridden properties 
+				var getMethod = property.GetMethod;
+				if (property.GetMethod.IsVirtual && getMethod.GetBaseDefinition().DeclaringType != entityType)
+				{
+					continue;
+				}
+
+				//Skip indexer properties (eg. "this[int index]")
+				if (property.GetIndexParameters().Length > 0)
+				{
+					continue;
+				}
+
+				//Skip properties with the "NotMappedAttribute"
+				var notMappedAttribute = property.GetCustomAttribute<NotMappedAttribute>();
+				if (notMappedAttribute != null)
+				{
+					continue;
+				}
+
+				//Do the mapping
+				var memberMap = classMap.MapMember(property);
+				
+				//Set custom element name with the "ColumnAttribute"
+				var columnAttribute = property.GetCustomAttribute<ColumnAttribute>();
+				if (columnAttribute != null)
+				{
+					var mappedName = columnAttribute.Name;
+					memberMap.SetElementName(mappedName);
+				}
+			}
+
+			definition.Properties = classMap.DeclaredMemberMaps
+				.Select(m => new EntityProperty
+				{
+					EntityType = definition.EntityType,
+					IsKey = m == classMap.IdMemberMap,
+					ElementName = m.ElementName,
+					FullPath = m.ElementName,
+					PropertyType = (m.MemberInfo as PropertyInfo).PropertyType,
+					PropertyInfo = (m.MemberInfo as PropertyInfo)
+				}).ToArray();
+		}
+	}
+}

src/MongoFramework/Infrastructure/Mapping/Processors/NestedTypeProcessor.cs

@@ -14,8 +14,4 @@
     <ProjectReference Include="..\..\src\MongoFramework\MongoFramework.csproj" />
   </ItemGroup>
 
-  <ItemGroup>
-    <Folder Include="docs\" />
-  </ItemGroup>
-
 </Project>