111 lines
4.2 KiB
C#
111 lines
4.2 KiB
C#
using System;
|
|
using System.Reflection;
|
|
using Marr.Data;
|
|
using Marr.Data.Mapping;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Datastore.Converters;
|
|
using NzbDrone.Core.ExternalNotification;
|
|
using NzbDrone.Core.Indexers;
|
|
using NzbDrone.Core.Instrumentation;
|
|
using NzbDrone.Core.Jobs;
|
|
using NzbDrone.Core.MediaFiles;
|
|
using NzbDrone.Core.Qualities;
|
|
using NzbDrone.Core.ReferenceData;
|
|
using NzbDrone.Core.RootFolders;
|
|
using NzbDrone.Core.Tv;
|
|
using BooleanIntConverter = NzbDrone.Core.Datastore.Converters.BooleanIntConverter;
|
|
using System.Linq;
|
|
|
|
namespace NzbDrone.Core.Datastore
|
|
{
|
|
public static class TableMapping
|
|
{
|
|
|
|
private static readonly FluentMappings Mapper = new FluentMappings(true);
|
|
|
|
public static void Map()
|
|
{
|
|
RegisterMappers();
|
|
|
|
Mapper.Entity<Config>().RegisterModel("Config");
|
|
Mapper.Entity<RootFolder>().RegisterModel("RootFolders");
|
|
|
|
Mapper.Entity<Indexer>().RegisterModel("IndexerDefinitions");
|
|
Mapper.Entity<NewznabDefinition>().RegisterModel("NewznabDefinitions");
|
|
Mapper.Entity<JobDefinition>().RegisterModel("JobDefinitions");
|
|
Mapper.Entity<ExternalNotificationDefinition>().RegisterModel("ExternalNotificationDefinitions");
|
|
|
|
Mapper.Entity<SceneMapping>().RegisterModel("SceneMappings");
|
|
|
|
Mapper.Entity<History.History>().RegisterModel("History")
|
|
.Relationships.AutoMapComplexTypeProperties<ILazyLoaded>()
|
|
.For(c => c.Episode)
|
|
.LazyLoad((db, history) => db.Query<Episode>().Single(ep => ep.Id == history.EpisodeId));
|
|
|
|
Mapper.Entity<Series>().RegisterModel("Series")
|
|
.Relationships.AutoMapComplexTypeProperties<ILazyLoaded>()
|
|
.For(c => c.Covers)
|
|
.LazyLoad((db, series) => db.Query<MediaCover.MediaCover>().Where(cover => cover.SeriesId == series.Id).ToList());
|
|
|
|
|
|
Mapper.Entity<Season>().RegisterModel("Seasons");
|
|
Mapper.Entity<Episode>().RegisterModel("Episodes");
|
|
Mapper.Entity<EpisodeFile>().RegisterModel("EpisodeFiles");
|
|
Mapper.Entity<MediaCover.MediaCover>().RegisterModel("MediaCovers");
|
|
|
|
Mapper.Entity<QualityProfile>().RegisterModel("QualityProfiles");
|
|
Mapper.Entity<QualitySize>().RegisterModel("QualitySizes");
|
|
|
|
Mapper.Entity<Log>().RegisterModel("Logs");
|
|
|
|
}
|
|
|
|
private static void RegisterMappers()
|
|
{
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Int32), new Int32Converter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Boolean), new BooleanIntConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Enum), new EnumIntConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(QualityModel), new EmbeddedDocumentConverter());
|
|
}
|
|
|
|
|
|
private 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();
|
|
}
|
|
|
|
private 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);
|
|
}
|
|
|
|
private 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);
|
|
}
|
|
}
|
|
} |