mirror of https://github.com/Sonarr/Sonarr
215 lines
10 KiB
C#
215 lines
10 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using Marr.Data;
|
|
using Marr.Data.Mapping;
|
|
using NzbDrone.Common.Reflection;
|
|
using NzbDrone.Core.Blacklisting;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.DataAugmentation.Scene;
|
|
using NzbDrone.Core.Datastore.Converters;
|
|
using NzbDrone.Core.Datastore.Extensions;
|
|
using NzbDrone.Core.Download;
|
|
using NzbDrone.Core.Download.Pending;
|
|
using NzbDrone.Core.Indexers;
|
|
using NzbDrone.Core.Instrumentation;
|
|
using NzbDrone.Core.Jobs;
|
|
using NzbDrone.Core.MediaFiles;
|
|
using NzbDrone.Core.Profiles.Delay;
|
|
using NzbDrone.Core.RemotePathMappings;
|
|
using NzbDrone.Core.Notifications;
|
|
using NzbDrone.Core.Organizer;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Profiles.Qualities;
|
|
using NzbDrone.Core.Qualities;
|
|
using NzbDrone.Core.RootFolders;
|
|
using NzbDrone.Core.SeriesStats;
|
|
using NzbDrone.Core.Tags;
|
|
using NzbDrone.Core.ThingiProvider;
|
|
using NzbDrone.Core.Tv;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Core.Authentication;
|
|
using NzbDrone.Core.CustomFilters;
|
|
using NzbDrone.Core.Download.History;
|
|
using NzbDrone.Core.Extras.Metadata;
|
|
using NzbDrone.Core.Extras.Metadata.Files;
|
|
using NzbDrone.Core.Extras.Others;
|
|
using NzbDrone.Core.Extras.Subtitles;
|
|
using NzbDrone.Core.History;
|
|
using NzbDrone.Core.Messaging.Commands;
|
|
using NzbDrone.Core.Languages;
|
|
using NzbDrone.Core.Profiles.Languages;
|
|
using NzbDrone.Core.Profiles.Releases;
|
|
using NzbDrone.Core.Update.History;
|
|
using NzbDrone.Core.ImportLists.Exclusions;
|
|
using NzbDrone.Core.ImportLists;
|
|
|
|
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")
|
|
.Ignore(r => r.Accessible)
|
|
.Ignore(r => r.FreeSpace)
|
|
.Ignore(r => r.TotalSpace);
|
|
|
|
Mapper.Entity<ScheduledTask>().RegisterModel("ScheduledTasks");
|
|
|
|
Mapper.Entity<IndexerDefinition>().RegisterDefinition("Indexers")
|
|
.Ignore(i => i.Enable)
|
|
.Ignore(i => i.Protocol)
|
|
.Ignore(i => i.SupportsRss)
|
|
.Ignore(i => i.SupportsSearch)
|
|
.Ignore(d => d.Tags);
|
|
|
|
Mapper.Entity<NotificationDefinition>().RegisterDefinition("Notifications")
|
|
.Ignore(i => i.SupportsOnGrab)
|
|
.Ignore(i => i.SupportsOnDownload)
|
|
.Ignore(i => i.SupportsOnUpgrade)
|
|
.Ignore(i => i.SupportsOnRename)
|
|
.Ignore(i => i.SupportsOnSeriesDelete)
|
|
.Ignore(i => i.SupportsOnEpisodeFileDelete)
|
|
.Ignore(i => i.SupportsOnHealthIssue);
|
|
|
|
Mapper.Entity<MetadataDefinition>().RegisterDefinition("Metadata")
|
|
.Ignore(d => d.Tags);
|
|
|
|
Mapper.Entity<DownloadClientDefinition>().RegisterDefinition("DownloadClients")
|
|
.Ignore(d => d.Protocol)
|
|
.Ignore(d => d.Tags);
|
|
|
|
Mapper.Entity<ImportListDefinition>().RegisterDefinition("ImportLists")
|
|
.Ignore(i => i.ListType)
|
|
.Ignore(i => i.Enable);
|
|
|
|
Mapper.Entity<SceneMapping>().RegisterModel("SceneMappings");
|
|
|
|
Mapper.Entity<EpisodeHistory>().RegisterModel("History")
|
|
.AutoMapChildModels();
|
|
|
|
Mapper.Entity<Series>().RegisterModel("Series")
|
|
.Ignore(s => s.RootFolderPath)
|
|
.Relationship()
|
|
.HasOne(s => s.QualityProfile, s => s.QualityProfileId)
|
|
.HasOne(s => s.LanguageProfile, s => s.LanguageProfileId);
|
|
|
|
Mapper.Entity<EpisodeFile>().RegisterModel("EpisodeFiles")
|
|
.Ignore(f => f.Path)
|
|
.Relationships.AutoMapICollectionOrComplexProperties()
|
|
.For("Episodes")
|
|
.LazyLoad(condition: parent => parent.Id > 0,
|
|
query: (db, parent) => db.Query<Episode>().Where(c => c.EpisodeFileId == parent.Id).ToList())
|
|
.HasOne(file => file.Series, file => file.SeriesId);
|
|
|
|
Mapper.Entity<Episode>().RegisterModel("Episodes")
|
|
.Ignore(e => e.SeriesTitle)
|
|
.Ignore(e => e.Series)
|
|
.Ignore(e => e.HasFile)
|
|
.Relationship()
|
|
.HasOne(episode => episode.EpisodeFile, episode => episode.EpisodeFileId);
|
|
|
|
Mapper.Entity<QualityDefinition>().RegisterModel("QualityDefinitions")
|
|
.Ignore(d => d.GroupName)
|
|
.Ignore(d => d.Weight);
|
|
|
|
Mapper.Entity<QualityProfile>().RegisterModel("QualityProfiles");
|
|
Mapper.Entity<LanguageProfile>().RegisterModel("LanguageProfiles");
|
|
Mapper.Entity<Log>().RegisterModel("Logs");
|
|
Mapper.Entity<NamingConfig>().RegisterModel("NamingConfig");
|
|
Mapper.Entity<SeasonStatistics>().MapResultSet();
|
|
Mapper.Entity<Blacklist>().RegisterModel("Blacklist");
|
|
Mapper.Entity<MetadataFile>().RegisterModel("MetadataFiles");
|
|
Mapper.Entity<SubtitleFile>().RegisterModel("SubtitleFiles");
|
|
Mapper.Entity<OtherExtraFile>().RegisterModel("ExtraFiles");
|
|
|
|
Mapper.Entity<PendingRelease>().RegisterModel("PendingReleases")
|
|
.Ignore(e => e.RemoteEpisode);
|
|
|
|
Mapper.Entity<RemotePathMapping>().RegisterModel("RemotePathMappings");
|
|
Mapper.Entity<Tag>().RegisterModel("Tags");
|
|
Mapper.Entity<ReleaseProfile>().RegisterModel("ReleaseProfiles");
|
|
|
|
Mapper.Entity<DelayProfile>().RegisterModel("DelayProfiles");
|
|
Mapper.Entity<User>().RegisterModel("Users");
|
|
Mapper.Entity<CommandModel>().RegisterModel("Commands")
|
|
.Ignore(c => c.Message);
|
|
|
|
Mapper.Entity<IndexerStatus>().RegisterModel("IndexerStatus");
|
|
Mapper.Entity<DownloadClientStatus>().RegisterModel("DownloadClientStatus");
|
|
Mapper.Entity<ImportListStatus>().RegisterModel("ImportListStatus");
|
|
|
|
Mapper.Entity<CustomFilter>().RegisterModel("CustomFilters");
|
|
|
|
Mapper.Entity<DownloadHistory>().RegisterModel("DownloadHistory")
|
|
.AutoMapChildModels();
|
|
|
|
Mapper.Entity<UpdateHistory>().RegisterModel("UpdateHistory");
|
|
Mapper.Entity<ImportListExclusion>().RegisterModel("ImportListExclusions");
|
|
}
|
|
|
|
private static void RegisterMappers()
|
|
{
|
|
RegisterEmbeddedConverter();
|
|
RegisterProviderSettingConverter();
|
|
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(int), new Int32Converter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(double), new DoubleConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(DateTime), new UtcConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(bool), new BooleanIntConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Enum), new EnumIntConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Quality), new QualityIntConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(List<QualityProfileQualityItem>), new EmbeddedDocumentConverter(new QualityIntConverter()));
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(QualityModel), new EmbeddedDocumentConverter(new QualityIntConverter()));
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Dictionary<string, string>), new EmbeddedDocumentConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(List<int>), new EmbeddedDocumentConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(List<KeyValuePair<string, int>>), new EmbeddedDocumentConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Language), new LanguageIntConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(List<string>), new EmbeddedDocumentConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(List<LanguageProfileItem>), new EmbeddedDocumentConverter(new LanguageIntConverter()));
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(ParsedEpisodeInfo), new EmbeddedDocumentConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(ReleaseInfo), new EmbeddedDocumentConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(HashSet<int>), new EmbeddedDocumentConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(OsPath), new OsPathConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Guid), new GuidConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Command), new CommandConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(TimeSpan), new TimeSpanConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(TimeSpan?), new TimeSpanConverter());
|
|
MapRepository.Instance.RegisterTypeConverter(typeof(Version), new SystemVersionConverter());
|
|
}
|
|
|
|
private static void RegisterProviderSettingConverter()
|
|
{
|
|
var settingTypes = typeof(IProviderConfig).Assembly.ImplementationsOf<IProviderConfig>();
|
|
|
|
var providerSettingConverter = new ProviderSettingConverter();
|
|
foreach (var embeddedType in settingTypes)
|
|
{
|
|
MapRepository.Instance.RegisterTypeConverter(embeddedType, providerSettingConverter);
|
|
}
|
|
}
|
|
|
|
private static void RegisterEmbeddedConverter()
|
|
{
|
|
var embeddedTypes = typeof(IEmbeddedDocument).Assembly.ImplementationsOf<IEmbeddedDocument>();
|
|
|
|
var embeddedConvertor = new EmbeddedDocumentConverter();
|
|
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);
|
|
}
|
|
}
|
|
}
|
|
}
|