2018-02-02 22:13:26 +00:00
|
|
|
using System;
|
2013-03-31 23:40:19 +00:00
|
|
|
using System.Collections.Generic;
|
2019-12-16 21:35:13 +00:00
|
|
|
using System.Linq;
|
2019-12-18 21:56:41 +00:00
|
|
|
using Dapper;
|
2013-09-21 06:39:26 +00:00
|
|
|
using NzbDrone.Common.Reflection;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.Authentication;
|
2021-08-19 19:13:14 +00:00
|
|
|
using NzbDrone.Core.Blocklisting;
|
2013-03-25 06:13:53 +00:00
|
|
|
using NzbDrone.Core.Configuration;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.CustomFilters;
|
|
|
|
using NzbDrone.Core.CustomFormats;
|
2013-03-25 06:13:53 +00:00
|
|
|
using NzbDrone.Core.Datastore.Converters;
|
2014-02-14 05:31:49 +00:00
|
|
|
using NzbDrone.Core.Download;
|
2020-03-29 20:05:48 +00:00
|
|
|
using NzbDrone.Core.Download.History;
|
2014-06-08 08:22:55 +00:00
|
|
|
using NzbDrone.Core.Download.Pending;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.Extras.Metadata;
|
|
|
|
using NzbDrone.Core.Extras.Metadata.Files;
|
|
|
|
using NzbDrone.Core.Extras.Others;
|
|
|
|
using NzbDrone.Core.Extras.Subtitles;
|
2020-03-29 20:05:48 +00:00
|
|
|
using NzbDrone.Core.History;
|
2020-09-04 02:50:56 +00:00
|
|
|
using NzbDrone.Core.ImportLists;
|
|
|
|
using NzbDrone.Core.ImportLists.ImportExclusions;
|
|
|
|
using NzbDrone.Core.ImportLists.ImportListMovies;
|
2013-03-25 06:13:53 +00:00
|
|
|
using NzbDrone.Core.Indexers;
|
|
|
|
using NzbDrone.Core.Instrumentation;
|
|
|
|
using NzbDrone.Core.Jobs;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.Languages;
|
2013-03-25 06:13:53 +00:00
|
|
|
using NzbDrone.Core.MediaFiles;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.Messaging.Commands;
|
|
|
|
using NzbDrone.Core.Movies;
|
|
|
|
using NzbDrone.Core.Movies.AlternativeTitles;
|
2019-11-02 19:32:05 +00:00
|
|
|
using NzbDrone.Core.Movies.Credits;
|
2020-05-26 01:55:10 +00:00
|
|
|
using NzbDrone.Core.Movies.Translations;
|
2013-05-19 23:17:32 +00:00
|
|
|
using NzbDrone.Core.Notifications;
|
2013-04-25 04:27:49 +00:00
|
|
|
using NzbDrone.Core.Organizer;
|
2014-06-08 08:22:55 +00:00
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
using NzbDrone.Core.Profiles;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.Profiles.Delay;
|
2013-03-25 06:13:53 +00:00
|
|
|
using NzbDrone.Core.Qualities;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.RemotePathMappings;
|
2014-10-28 04:37:35 +00:00
|
|
|
using NzbDrone.Core.Restrictions;
|
2013-03-25 06:13:53 +00:00
|
|
|
using NzbDrone.Core.RootFolders;
|
2014-09-10 00:02:55 +00:00
|
|
|
using NzbDrone.Core.Tags;
|
2013-09-21 06:39:26 +00:00
|
|
|
using NzbDrone.Core.ThingiProvider;
|
2021-01-25 04:33:54 +00:00
|
|
|
using NzbDrone.Core.Update.History;
|
2019-12-18 21:56:41 +00:00
|
|
|
using static Dapper.SqlMapper;
|
2013-03-25 06:13:53 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Datastore
|
|
|
|
{
|
|
|
|
public static class TableMapping
|
|
|
|
{
|
2019-12-18 21:56:41 +00:00
|
|
|
static TableMapping()
|
|
|
|
{
|
|
|
|
Mapper = new TableMapper();
|
|
|
|
}
|
|
|
|
|
|
|
|
public static TableMapper Mapper { get; private set; }
|
2013-03-25 06:13:53 +00:00
|
|
|
|
|
|
|
public static void Map()
|
|
|
|
{
|
|
|
|
RegisterMappers();
|
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<Config>("Config").RegisterModel();
|
2018-02-02 22:13:26 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<RootFolder>("RootFolders").RegisterModel()
|
2019-08-27 21:48:34 +00:00
|
|
|
.Ignore(r => r.Accessible)
|
2018-02-02 22:13:26 +00:00
|
|
|
.Ignore(r => r.FreeSpace)
|
|
|
|
.Ignore(r => r.TotalSpace);
|
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<ScheduledTask>("ScheduledTasks").RegisterModel();
|
2013-03-25 06:13:53 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<IndexerDefinition>("Indexers").RegisterModel()
|
|
|
|
.Ignore(x => x.ImplementationName)
|
2014-08-18 02:25:00 +00:00
|
|
|
.Ignore(i => i.Enable)
|
|
|
|
.Ignore(i => i.Protocol)
|
|
|
|
.Ignore(i => i.SupportsRss)
|
2018-11-23 07:03:32 +00:00
|
|
|
.Ignore(i => i.SupportsSearch)
|
|
|
|
.Ignore(d => d.Tags);
|
2014-08-18 02:25:00 +00:00
|
|
|
|
2020-09-04 02:50:56 +00:00
|
|
|
Mapper.Entity<ImportListDefinition>("ImportLists").RegisterModel()
|
2019-12-18 21:56:41 +00:00
|
|
|
.Ignore(x => x.ImplementationName)
|
2019-11-02 19:43:45 +00:00
|
|
|
.Ignore(i => i.ListType)
|
2019-12-18 21:56:41 +00:00
|
|
|
.Ignore(i => i.Enable);
|
2017-01-22 13:30:33 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<NotificationDefinition>("Notifications").RegisterModel()
|
|
|
|
.Ignore(x => x.ImplementationName)
|
2015-05-20 23:22:10 +00:00
|
|
|
.Ignore(i => i.SupportsOnGrab)
|
|
|
|
.Ignore(i => i.SupportsOnDownload)
|
|
|
|
.Ignore(i => i.SupportsOnUpgrade)
|
2019-11-04 20:44:33 +00:00
|
|
|
.Ignore(i => i.SupportsOnRename)
|
2021-03-05 23:30:08 +00:00
|
|
|
.Ignore(i => i.SupportsOnMovieDelete)
|
|
|
|
.Ignore(i => i.SupportsOnMovieFileDelete)
|
|
|
|
.Ignore(i => i.SupportsOnMovieFileDeleteForUpgrade)
|
2019-11-04 20:44:33 +00:00
|
|
|
.Ignore(i => i.SupportsOnHealthIssue);
|
2018-08-05 14:28:05 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<MetadataDefinition>("Metadata").RegisterModel()
|
|
|
|
.Ignore(x => x.ImplementationName)
|
2018-11-23 07:03:32 +00:00
|
|
|
.Ignore(d => d.Tags);
|
2014-08-18 02:25:00 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<DownloadClientDefinition>("DownloadClients").RegisterModel()
|
|
|
|
.Ignore(x => x.ImplementationName)
|
2018-11-23 07:03:32 +00:00
|
|
|
.Ignore(d => d.Protocol)
|
|
|
|
.Ignore(d => d.Tags);
|
2013-03-25 06:13:53 +00:00
|
|
|
|
2020-03-29 20:05:48 +00:00
|
|
|
Mapper.Entity<MovieHistory>("History").RegisterModel();
|
2019-12-18 21:56:41 +00:00
|
|
|
|
|
|
|
Mapper.Entity<MovieFile>("MovieFiles").RegisterModel()
|
|
|
|
.Ignore(f => f.Path);
|
|
|
|
|
|
|
|
Mapper.Entity<Movie>("Movies").RegisterModel()
|
2020-09-04 02:50:56 +00:00
|
|
|
.Ignore(s => s.RootFolderPath)
|
|
|
|
.Ignore(s => s.Translations);
|
|
|
|
|
|
|
|
Mapper.Entity<ImportListMovie>("ImportListMovies").RegisterModel();
|
2019-12-18 21:56:41 +00:00
|
|
|
|
|
|
|
Mapper.Entity<AlternativeTitle>("AlternativeTitles").RegisterModel();
|
|
|
|
|
2020-05-26 01:55:10 +00:00
|
|
|
Mapper.Entity<MovieTranslation>("MovieTranslations").RegisterModel();
|
|
|
|
|
2019-11-02 19:32:05 +00:00
|
|
|
Mapper.Entity<Credit>("Credits").RegisterModel();
|
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<ImportExclusion>("ImportExclusions").RegisterModel();
|
|
|
|
|
|
|
|
Mapper.Entity<QualityDefinition>("QualityDefinitions").RegisterModel()
|
|
|
|
.Ignore(d => d.GroupName)
|
|
|
|
.Ignore(d => d.Weight);
|
|
|
|
|
2020-01-22 21:47:33 +00:00
|
|
|
Mapper.Entity<CustomFormat>("CustomFormats").RegisterModel();
|
2019-12-18 21:56:41 +00:00
|
|
|
|
|
|
|
Mapper.Entity<Profile>("Profiles").RegisterModel();
|
|
|
|
Mapper.Entity<Log>("Logs").RegisterModel();
|
|
|
|
Mapper.Entity<NamingConfig>("NamingConfig").RegisterModel();
|
2021-08-19 19:13:14 +00:00
|
|
|
Mapper.Entity<Blocklist>("Blocklist").RegisterModel();
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<MetadataFile>("MetadataFiles").RegisterModel();
|
|
|
|
Mapper.Entity<SubtitleFile>("SubtitleFiles").RegisterModel();
|
|
|
|
Mapper.Entity<OtherExtraFile>("ExtraFiles").RegisterModel();
|
|
|
|
|
|
|
|
Mapper.Entity<PendingRelease>("PendingReleases").RegisterModel()
|
2018-03-14 20:41:36 +00:00
|
|
|
.Ignore(e => e.RemoteMovie);
|
2014-09-11 20:24:00 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<RemotePathMapping>("RemotePathMappings").RegisterModel();
|
|
|
|
Mapper.Entity<Tag>("Tags").RegisterModel();
|
|
|
|
Mapper.Entity<Restriction>("Restrictions").RegisterModel();
|
2014-11-24 00:07:46 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<DelayProfile>("DelayProfiles").RegisterModel();
|
|
|
|
Mapper.Entity<User>("Users").RegisterModel();
|
|
|
|
Mapper.Entity<CommandModel>("Commands").RegisterModel()
|
|
|
|
.Ignore(c => c.Message);
|
2015-06-27 09:43:17 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<IndexerStatus>("IndexerStatus").RegisterModel();
|
|
|
|
Mapper.Entity<DownloadClientStatus>("DownloadClientStatus").RegisterModel();
|
2020-09-04 02:50:56 +00:00
|
|
|
Mapper.Entity<ImportListStatus>("ImportListStatus").RegisterModel();
|
2018-11-23 07:03:32 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
Mapper.Entity<CustomFilter>("CustomFilters").RegisterModel();
|
2020-03-29 20:05:48 +00:00
|
|
|
|
|
|
|
Mapper.Entity<DownloadHistory>("DownloadHistory").RegisterModel();
|
2021-01-25 04:33:54 +00:00
|
|
|
|
|
|
|
Mapper.Entity<UpdateHistory>("UpdateHistory").RegisterModel();
|
2013-03-25 06:13:53 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private static void RegisterMappers()
|
|
|
|
{
|
2013-04-01 00:20:01 +00:00
|
|
|
RegisterEmbeddedConverter();
|
2013-09-21 06:39:26 +00:00
|
|
|
RegisterProviderSettingConverter();
|
2013-04-01 00:20:01 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
SqlMapper.RemoveTypeMap(typeof(DateTime));
|
|
|
|
SqlMapper.AddTypeHandler(new DapperUtcConverter());
|
|
|
|
SqlMapper.AddTypeHandler(new DapperQualityIntConverter());
|
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<ProfileQualityItem>>(new QualityIntConverter()));
|
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<ProfileFormatItem>>(new CustomFormatIntConverter()));
|
2020-02-18 21:03:05 +00:00
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<ICustomFormatSpecification>>(new CustomFormatSpecificationListConverter()));
|
2020-01-22 21:47:33 +00:00
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<QualityModel>(new QualityIntConverter()));
|
2019-12-18 21:56:41 +00:00
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<Dictionary<string, string>>());
|
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<IDictionary<string, string>>());
|
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<int>>());
|
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<KeyValuePair<string, int>>>());
|
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<KeyValuePair<string, int>>());
|
|
|
|
SqlMapper.AddTypeHandler(new DapperLanguageIntConverter());
|
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<Language>>(new LanguageIntConverter()));
|
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<string>>());
|
2020-01-22 21:47:33 +00:00
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<ParsedMovieInfo>(new QualityIntConverter(), new LanguageIntConverter()));
|
2019-12-18 21:56:41 +00:00
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<ReleaseInfo>());
|
2020-09-04 02:50:56 +00:00
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<MovieTranslation>>());
|
2019-12-18 21:56:41 +00:00
|
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<HashSet<int>>());
|
|
|
|
SqlMapper.AddTypeHandler(new OsPathConverter());
|
|
|
|
SqlMapper.RemoveTypeMap(typeof(Guid));
|
|
|
|
SqlMapper.RemoveTypeMap(typeof(Guid?));
|
|
|
|
SqlMapper.AddTypeHandler(new GuidConverter());
|
|
|
|
SqlMapper.AddTypeHandler(new CommandConverter());
|
2021-01-25 04:33:54 +00:00
|
|
|
SqlMapper.AddTypeHandler(new SystemVersionConverter());
|
2013-03-25 06:13:53 +00:00
|
|
|
}
|
2013-04-01 00:20:01 +00:00
|
|
|
|
2013-09-21 06:39:26 +00:00
|
|
|
private static void RegisterProviderSettingConverter()
|
|
|
|
{
|
2019-12-16 21:35:13 +00:00
|
|
|
var settingTypes = typeof(IProviderConfig).Assembly.ImplementationsOf<IProviderConfig>()
|
|
|
|
.Where(x => !x.ContainsGenericParameters);
|
2013-09-21 06:39:26 +00:00
|
|
|
|
|
|
|
var providerSettingConverter = new ProviderSettingConverter();
|
|
|
|
foreach (var embeddedType in settingTypes)
|
|
|
|
{
|
2019-12-18 21:56:41 +00:00
|
|
|
SqlMapper.AddTypeHandler(embeddedType, providerSettingConverter);
|
2013-09-21 06:39:26 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-04-01 00:20:01 +00:00
|
|
|
private static void RegisterEmbeddedConverter()
|
|
|
|
{
|
2013-09-21 06:39:26 +00:00
|
|
|
var embeddedTypes = typeof(IEmbeddedDocument).Assembly.ImplementationsOf<IEmbeddedDocument>();
|
2013-04-01 00:20:01 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
var embeddedConverterDefinition = typeof(EmbeddedDocumentConverter<>).GetGenericTypeDefinition();
|
2013-04-01 00:20:01 +00:00
|
|
|
var genericListDefinition = typeof(List<>).GetGenericTypeDefinition();
|
2014-09-10 00:02:55 +00:00
|
|
|
|
2013-04-01 00:20:01 +00:00
|
|
|
foreach (var embeddedType in embeddedTypes)
|
|
|
|
{
|
|
|
|
var embeddedListType = genericListDefinition.MakeGenericType(embeddedType);
|
2014-09-10 00:02:55 +00:00
|
|
|
|
2019-12-18 21:56:41 +00:00
|
|
|
RegisterEmbeddedConverter(embeddedType, embeddedConverterDefinition);
|
|
|
|
RegisterEmbeddedConverter(embeddedListType, embeddedConverterDefinition);
|
2013-04-01 00:20:01 +00:00
|
|
|
}
|
|
|
|
}
|
2019-12-18 21:56:41 +00:00
|
|
|
|
|
|
|
private static void RegisterEmbeddedConverter(Type embeddedType, Type embeddedConverterDefinition)
|
|
|
|
{
|
|
|
|
var embeddedConverterType = embeddedConverterDefinition.MakeGenericType(embeddedType);
|
|
|
|
var converter = (ITypeHandler)Activator.CreateInstance(embeddedConverterType);
|
|
|
|
|
|
|
|
SqlMapper.AddTypeHandler(embeddedType, converter);
|
|
|
|
}
|
2013-03-25 06:13:53 +00:00
|
|
|
}
|
2018-02-15 12:39:01 +00:00
|
|
|
}
|