2013-03-25 06:13:53 +00:00
|
|
|
|
using System;
|
2013-03-31 23:40:19 +00:00
|
|
|
|
using System.Collections.Generic;
|
2013-04-01 00:20:01 +00:00
|
|
|
|
using System.Linq;
|
2013-03-25 06:13:53 +00:00
|
|
|
|
using Marr.Data;
|
|
|
|
|
using Marr.Data.Mapping;
|
2013-04-17 23:32:06 +00:00
|
|
|
|
using NzbDrone.Common;
|
2013-03-25 06:13:53 +00:00
|
|
|
|
using NzbDrone.Core.Configuration;
|
2013-04-01 02:43:58 +00:00
|
|
|
|
using NzbDrone.Core.DataAugmentation.Scene;
|
2013-03-25 06:13:53 +00:00
|
|
|
|
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;
|
2013-04-25 04:27:49 +00:00
|
|
|
|
using NzbDrone.Core.Organizer;
|
2013-03-25 06:13:53 +00:00
|
|
|
|
using NzbDrone.Core.Qualities;
|
|
|
|
|
using NzbDrone.Core.RootFolders;
|
2013-04-20 23:36:23 +00:00
|
|
|
|
using NzbDrone.Core.SeriesStats;
|
2013-03-25 06:13:53 +00:00
|
|
|
|
using NzbDrone.Core.Tv;
|
|
|
|
|
|
|
|
|
|
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");
|
2013-03-27 00:48:07 +00:00
|
|
|
|
Mapper.Entity<RootFolder>().RegisterModel("RootFolders").Ignore(r => r.FreeSpace);
|
2013-03-25 06:13:53 +00:00
|
|
|
|
|
2013-04-07 07:30:37 +00:00
|
|
|
|
Mapper.Entity<IndexerDefinition>().RegisterModel("IndexerDefinitions");
|
2013-05-09 06:38:20 +00:00
|
|
|
|
Mapper.Entity<ScheduledTask>().RegisterModel("ScheduledTasks");
|
2013-03-25 06:13:53 +00:00
|
|
|
|
Mapper.Entity<ExternalNotificationDefinition>().RegisterModel("ExternalNotificationDefinitions");
|
|
|
|
|
|
|
|
|
|
Mapper.Entity<SceneMapping>().RegisterModel("SceneMappings");
|
|
|
|
|
|
2013-05-03 07:28:02 +00:00
|
|
|
|
Mapper.Entity<History.History>().RegisterModel("History")
|
|
|
|
|
.Relationship()
|
|
|
|
|
.HasOne(h => h.Episode, h => h.EpisodeId)
|
|
|
|
|
.HasOne(h => h.Series, h => h.SeriesId);
|
2013-03-26 05:51:56 +00:00
|
|
|
|
|
2013-04-11 04:30:51 +00:00
|
|
|
|
Mapper.Entity<Series>().RegisterModel("Series")
|
2013-04-11 07:52:38 +00:00
|
|
|
|
.Ignore(s => s.Path)
|
2013-04-29 01:47:06 +00:00
|
|
|
|
.Relationship()
|
|
|
|
|
.HasOne(s => s.RootFolder, s => s.RootFolderId)
|
|
|
|
|
.HasOne(s => s.QualityProfile, s => s.QualityProfileId);
|
2013-03-26 05:51:56 +00:00
|
|
|
|
|
2013-03-25 06:13:53 +00:00
|
|
|
|
Mapper.Entity<Season>().RegisterModel("Seasons");
|
2013-05-03 00:25:51 +00:00
|
|
|
|
|
|
|
|
|
Mapper.Entity<Episode>().RegisterModel("Episodes")
|
2013-05-10 15:19:29 +00:00
|
|
|
|
.Ignore(e => e.SeriesTitle)
|
|
|
|
|
.Relationships.AutoMapICollectionOrComplexProperties();
|
2013-05-03 00:25:51 +00:00
|
|
|
|
|
2013-03-25 06:13:53 +00:00
|
|
|
|
Mapper.Entity<EpisodeFile>().RegisterModel("EpisodeFiles");
|
2013-04-06 01:03:14 +00:00
|
|
|
|
|
2013-04-29 01:47:06 +00:00
|
|
|
|
Mapper.Entity<QualityProfile>().RegisterModel("QualityProfiles");
|
2013-04-06 01:03:14 +00:00
|
|
|
|
|
2013-03-25 06:13:53 +00:00
|
|
|
|
Mapper.Entity<QualitySize>().RegisterModel("QualitySizes");
|
|
|
|
|
|
|
|
|
|
Mapper.Entity<Log>().RegisterModel("Logs");
|
2013-04-20 20:09:12 +00:00
|
|
|
|
|
2013-04-25 04:27:49 +00:00
|
|
|
|
Mapper.Entity<NamingConfig>().RegisterModel("NamingConfig");
|
|
|
|
|
|
2013-04-20 21:23:17 +00:00
|
|
|
|
Mapper.Entity<SeriesStatistics>().MapResultSet();
|
2013-03-25 06:13:53 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static void RegisterMappers()
|
|
|
|
|
{
|
2013-04-01 00:20:01 +00:00
|
|
|
|
RegisterEmbeddedConverter();
|
|
|
|
|
|
2013-03-25 06:13:53 +00:00
|
|
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Int32), new Int32Converter());
|
|
|
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Boolean), new BooleanIntConverter());
|
2013-03-26 05:51:56 +00:00
|
|
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Enum), new EnumIntConverter());
|
2013-03-26 08:02:31 +00:00
|
|
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Quality), new QualityIntConverter());
|
2013-03-25 06:13:53 +00:00
|
|
|
|
}
|
2013-04-01 00:20:01 +00:00
|
|
|
|
|
|
|
|
|
private static void RegisterEmbeddedConverter()
|
|
|
|
|
{
|
|
|
|
|
var embeddedTypes = typeof(IEmbeddedDocument).Assembly.GetTypes()
|
|
|
|
|
.Where(c => c.GetInterfaces().Any(i => i == typeof(IEmbeddedDocument)));
|
|
|
|
|
|
|
|
|
|
|
2013-04-17 23:32:06 +00:00
|
|
|
|
var embeddedConvertor = new EmbeddedDocumentConverter(new JsonSerializer());
|
2013-04-01 00:20:01 +00:00
|
|
|
|
var genericListDefinition = typeof(List<>).GetGenericTypeDefinition();
|
|
|
|
|
foreach (var embeddedType in embeddedTypes)
|
|
|
|
|
{
|
|
|
|
|
var embeddedListType = genericListDefinition.MakeGenericType(embeddedType);
|
|
|
|
|
MapRepository.Instance.RegisterTypeConverter(embeddedType, embeddedConvertor);
|
|
|
|
|
MapRepository.Instance.RegisterTypeConverter(embeddedListType, embeddedConvertor);
|
|
|
|
|
}
|
|
|
|
|
}
|
2013-03-25 06:13:53 +00:00
|
|
|
|
}
|
|
|
|
|
}
|