created HasOne, HasMany relationship extension.

NzbDrone.Core/Datastore/MappingExtensions.cs

@@ -0,0 +1,49 @@
+using System;
+using System.Linq;
+using System.Reflection;
+using Marr.Data.Mapping;
+
+namespace NzbDrone.Core.Datastore
+{
+    public static class MappingExtensions
+    {
+        public static ColumnMapBuilder<T> RegisterModel<T>(this FluentMappings.MappingsFluentEntity<T> mapBuilder, string tableName) where T : ModelBase
+        {
+            return mapBuilder.Table.MapTable(tableName)
+                             .Columns
+                             .AutoMapPropertiesWhere(IsMappableProperty)
+                             .For(c => c.Id)
+                             .SetPrimaryKey()
+                             .SetReturnValue()
+                             .SetAutoIncrement();
+        }
+
+        public static bool IsMappableProperty(MemberInfo memberInfo)
+        {
+            var propertyInfo = memberInfo as PropertyInfo;
+
+            if (propertyInfo == null) return false;
+
+            if (propertyInfo.PropertyType.GetInterfaces().Any(i => i == typeof(IEmbeddedDocument)))
+            {
+                return true;
+            }
+
+            return propertyInfo.CanRead && propertyInfo.CanWrite && IsSimpleType(propertyInfo.PropertyType);
+        }
+
+        public static bool IsSimpleType(Type type)
+        {
+            if (type.IsGenericType && type.GetGenericTypeDefinition() == typeof(Nullable<>))
+            {
+                type = type.GetGenericArguments()[0];
+            }
+
+            return type.IsPrimitive
+                   || type.IsEnum
+                   || type == typeof(string)
+                   || type == typeof(DateTime)
+                   || type == typeof(Decimal);
+        }
+    }
+}