You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
163 lines
6.6 KiB
163 lines
6.6 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Dapper;
|
|
using NzbDrone.Common.Reflection;
|
|
using NzbDrone.Core.Applications;
|
|
using NzbDrone.Core.Authentication;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.CustomFilters;
|
|
using NzbDrone.Core.Datastore.Converters;
|
|
using NzbDrone.Core.Download;
|
|
using NzbDrone.Core.IndexerProxies;
|
|
using NzbDrone.Core.Indexers;
|
|
using NzbDrone.Core.IndexerVersions;
|
|
using NzbDrone.Core.Instrumentation;
|
|
using NzbDrone.Core.Jobs;
|
|
using NzbDrone.Core.Messaging.Commands;
|
|
using NzbDrone.Core.Notifications;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Profiles;
|
|
using NzbDrone.Core.Tags;
|
|
using NzbDrone.Core.ThingiProvider;
|
|
using NzbDrone.Core.Update.History;
|
|
using static Dapper.SqlMapper;
|
|
|
|
namespace NzbDrone.Core.Datastore
|
|
{
|
|
public static class TableMapping
|
|
{
|
|
static TableMapping()
|
|
{
|
|
Mapper = new TableMapper();
|
|
}
|
|
|
|
public static TableMapper Mapper { get; private set; }
|
|
|
|
public static void Map()
|
|
{
|
|
RegisterMappers();
|
|
|
|
Mapper.Entity<Config>("Config").RegisterModel();
|
|
|
|
Mapper.Entity<ScheduledTask>("ScheduledTasks").RegisterModel()
|
|
.Ignore(i => i.Priority);
|
|
|
|
Mapper.Entity<IndexerDefinition>("Indexers").RegisterModel()
|
|
.Ignore(x => x.ImplementationName)
|
|
.Ignore(i => i.Description)
|
|
.Ignore(i => i.Language)
|
|
.Ignore(i => i.Encoding)
|
|
.Ignore(i => i.IndexerUrls)
|
|
.Ignore(i => i.LegacyUrls)
|
|
.Ignore(i => i.Protocol)
|
|
.Ignore(i => i.Privacy)
|
|
.Ignore(i => i.SupportsRss)
|
|
.Ignore(i => i.SupportsSearch)
|
|
.Ignore(i => i.SupportsRedirect)
|
|
.Ignore(i => i.Capabilities)
|
|
.HasOne(a => a.AppProfile, a => a.AppProfileId);
|
|
|
|
Mapper.Entity<DownloadClientDefinition>("DownloadClients").RegisterModel()
|
|
.Ignore(x => x.ImplementationName)
|
|
.Ignore(i => i.Protocol)
|
|
.Ignore(d => d.Tags);
|
|
|
|
Mapper.Entity<NotificationDefinition>("Notifications").RegisterModel()
|
|
.Ignore(x => x.ImplementationName)
|
|
.Ignore(i => i.SupportsOnHealthIssue)
|
|
.Ignore(i => i.SupportsOnApplicationUpdate);
|
|
|
|
Mapper.Entity<IndexerProxyDefinition>("IndexerProxies").RegisterModel()
|
|
.Ignore(x => x.ImplementationName);
|
|
|
|
Mapper.Entity<ApplicationDefinition>("Applications").RegisterModel()
|
|
.Ignore(x => x.ImplementationName);
|
|
|
|
Mapper.Entity<History.History>("History").RegisterModel();
|
|
|
|
Mapper.Entity<Log>("Logs").RegisterModel();
|
|
|
|
Mapper.Entity<Tag>("Tags").RegisterModel();
|
|
|
|
Mapper.Entity<AppIndexerMap>("ApplicationIndexerMapping").RegisterModel();
|
|
|
|
Mapper.Entity<User>("Users").RegisterModel();
|
|
Mapper.Entity<CommandModel>("Commands").RegisterModel()
|
|
.Ignore(c => c.Message);
|
|
|
|
Mapper.Entity<IndexerStatus>("IndexerStatus").RegisterModel();
|
|
|
|
Mapper.Entity<DownloadClientStatus>("DownloadClientStatus").RegisterModel();
|
|
|
|
Mapper.Entity<ApplicationStatus>("ApplicationStatus").RegisterModel();
|
|
|
|
Mapper.Entity<CustomFilter>("CustomFilters").RegisterModel();
|
|
Mapper.Entity<UpdateHistory>("UpdateHistory").RegisterModel();
|
|
|
|
Mapper.Entity<AppSyncProfile>("AppSyncProfiles").RegisterModel();
|
|
Mapper.Entity<IndexerDefinitionVersion>("IndexerDefinitionVersions").RegisterModel();
|
|
}
|
|
|
|
private static void RegisterMappers()
|
|
{
|
|
RegisterEmbeddedConverter();
|
|
RegisterProviderSettingConverter();
|
|
|
|
SqlMapper.RemoveTypeMap(typeof(DateTime));
|
|
SqlMapper.AddTypeHandler(new DapperUtcConverter());
|
|
SqlMapper.AddTypeHandler(new DapperTimeSpanConverter());
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<Dictionary<string, string>>());
|
|
SqlMapper.AddTypeHandler(new CookieConverter());
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<int>>());
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<KeyValuePair<string, int>>>());
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<KeyValuePair<string, int>>());
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<List<string>>());
|
|
SqlMapper.AddTypeHandler(new EmbeddedDocumentConverter<ReleaseInfo>());
|
|
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());
|
|
SqlMapper.AddTypeHandler(new SystemVersionConverter());
|
|
}
|
|
|
|
private static void RegisterProviderSettingConverter()
|
|
{
|
|
var settingTypes = typeof(IProviderConfig).Assembly.ImplementationsOf<IProviderConfig>()
|
|
.Where(x => !x.ContainsGenericParameters);
|
|
|
|
var providerSettingConverter = new ProviderSettingConverter();
|
|
foreach (var embeddedType in settingTypes)
|
|
{
|
|
SqlMapper.AddTypeHandler(embeddedType, providerSettingConverter);
|
|
}
|
|
}
|
|
|
|
private static void RegisterEmbeddedConverter()
|
|
{
|
|
var embeddedTypes = typeof(IEmbeddedDocument).Assembly.ImplementationsOf<IEmbeddedDocument>();
|
|
|
|
var embeddedConverterDefinition = typeof(EmbeddedDocumentConverter<>).GetGenericTypeDefinition();
|
|
var genericListDefinition = typeof(List<>).GetGenericTypeDefinition();
|
|
|
|
foreach (var embeddedType in embeddedTypes)
|
|
{
|
|
var embeddedListType = genericListDefinition.MakeGenericType(embeddedType);
|
|
|
|
RegisterEmbeddedConverter(embeddedType, embeddedConverterDefinition);
|
|
RegisterEmbeddedConverter(embeddedListType, embeddedConverterDefinition);
|
|
}
|
|
}
|
|
|
|
private static void RegisterEmbeddedConverter(Type embeddedType, Type embeddedConverterDefinition)
|
|
{
|
|
var embeddedConverterType = embeddedConverterDefinition.MakeGenericType(embeddedType);
|
|
var converter = (ITypeHandler)Activator.CreateInstance(embeddedConverterType);
|
|
|
|
SqlMapper.AddTypeHandler(embeddedType, converter);
|
|
}
|
|
}
|
|
}
|