Lidarr/NzbDrone.Core/Providers/Core/ConfigProvider.cs

320 lines
8.8 KiB
C#
Raw Normal View History

2010-09-23 03:19:47 +00:00
using System;
using System.Collections.Generic;
using Ninject;
2010-10-05 06:21:18 +00:00
using NLog;
using NzbDrone.Core.Model;
using NzbDrone.Core.Repository;
2011-06-17 02:27:10 +00:00
using PetaPoco;
2010-09-23 03:19:47 +00:00
2011-04-04 03:50:12 +00:00
namespace NzbDrone.Core.Providers.Core
2010-09-23 03:19:47 +00:00
{
public class ConfigProvider
2010-09-23 03:19:47 +00:00
{
2010-10-05 06:21:18 +00:00
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
2011-06-17 02:27:10 +00:00
private readonly IDatabase _database;
2010-09-23 03:19:47 +00:00
[Inject]
2011-06-17 02:27:10 +00:00
public ConfigProvider(IDatabase database)
2010-09-23 03:19:47 +00:00
{
2011-06-17 02:27:10 +00:00
_database = database;
2010-09-23 03:19:47 +00:00
}
public ConfigProvider()
{
}
public IList<Config> All()
2010-09-23 03:19:47 +00:00
{
return _database.Fetch<Config>();
}
2010-09-24 05:21:45 +00:00
public virtual String NzbMatrixUsername
{
get { return GetValue("NzbMatrixUsername"); }
set { SetValue("NzbMatrixUsername", value); }
}
public virtual String NzbMatrixApiKey
{
get { return GetValue("NzbMatrixApiKey"); }
set { SetValue("NzbMatrixApiKey", value); }
}
public virtual String NzbsOrgUId
{
get { return GetValue("NzbsOrgUId"); }
set { SetValue("NzbsOrgUId", value); }
}
public virtual String NzbsOrgHash
{
get { return GetValue("NzbsOrgHash"); }
set { SetValue("NzbsOrgHash", value); }
}
public virtual String NzbsrusUId
{
get { return GetValue("NzbsrusUId"); }
set { SetValue("NzbsrusUId", value); }
}
public virtual String NzbsrusHash
{
get { return GetValue("NzbsrusHash"); }
set { SetValue("NzbsrusHash", value); }
}
public virtual String NewzbinUsername
{
get { return GetValue("NewzbinUsername"); }
set { SetValue("NewzbinUsername", value); }
}
public virtual String NewzbinPassword
{
get { return GetValue("NewzbinPassword"); }
set { SetValue("NewzbinPassword", value); }
}
public virtual String SabHost
{
2011-06-17 02:27:10 +00:00
get { return GetValue("SabHost", "localhost"); }
set { SetValue("SabHost", value); }
}
public virtual int SabPort
{
get { return GetValueInt("SabPort", 8080); }
set { SetValue("SabPort", value); }
}
public virtual String SabApiKey
{
get { return GetValue("SabApiKey"); }
set { SetValue("SabApiKey", value); }
}
public virtual String SabUsername
{
get { return GetValue("SabUsername"); }
set { SetValue("SabUsername", value); }
}
public virtual String SabPassword
{
get { return GetValue("SabPassword"); }
set { SetValue("SabPassword", value); }
}
public virtual String SabTvCategory
{
get { return GetValue("SabTvCategory", "tv"); }
set { SetValue("SabTvCategory", value); }
}
public virtual SabnzbdPriorityType SabTvPriority
{
get { return (SabnzbdPriorityType)GetValueInt("SabTvPriority"); }
set { SetValue("SabTvPriority", (int)value); }
}
public virtual String SabDropDirectory
{
2011-06-17 02:27:10 +00:00
get { return GetValue("SabTvDropDirectory"); }
set { SetValue("SabTvDropDirectory", value); }
}
2011-07-08 03:57:44 +00:00
public virtual bool SortingIncludeSeriesName
{
get { return GetValueBoolean("Sorting_SeriesName", true); }
set { SetValue("Sorting_SeriesName", value); }
}
2011-07-08 03:57:44 +00:00
public virtual bool SortingIncludeEpisodeTitle
{
get { return GetValueBoolean("Sorting_EpisodeName", true); }
set { SetValue("Sorting_EpisodeName", value); }
}
2011-07-08 03:36:02 +00:00
public virtual bool SortingReplaceSpaces
{
2011-07-08 03:57:44 +00:00
get { return GetValueBoolean("Sorting_ReplaceSpaces"); }
set { SetValue("Sorting_ReplaceSpaces", value); }
}
2011-07-08 03:36:02 +00:00
public virtual bool SortingAppendQuality
{
get { return GetValueBoolean("Sorting_AppendQaulity", true); }
set { SetValue("Sorting_AppendQaulity", value); }
}
public virtual bool UseSeasonFolder
{
2011-07-08 03:36:02 +00:00
get { return GetValueBoolean("UseSeasonFolder", true); }
2011-07-08 03:36:02 +00:00
set { SetValue("UseSeasonFolder", value); }
}
2011-07-08 03:36:02 +00:00
public virtual string SortingSeasonFolderFormat
{
2011-06-17 02:27:10 +00:00
get { return GetValue("Sorting_SeasonFolderFormat", "Season %s"); }
set { SetValue("Sorting_SeasonFolderFormat", value); }
}
2011-07-08 03:36:02 +00:00
public virtual int SortingSeparatorStyle
{
get { return GetValueInt("Sorting_SeparatorStyle"); }
set { SetValue("Sorting_SeparatorStyle", value); }
}
2011-07-08 03:36:02 +00:00
public virtual int SortingNumberStyle
{
get { return GetValueInt("Sorting_NumberStyle", 2); }
set { SetValue("Sorting_NumberStyle", value); }
}
2011-07-08 03:36:02 +00:00
public virtual int SortingMultiEpisodeStyle
{
get { return GetValueInt("Sorting_MultiEpisodeStyle"); }
set { SetValue("Sorting_MultiEpisodeStyle", value); }
}
public virtual int DefaultQualityProfile
{
get { return GetValueInt("DefaultQualityProfile", 1); }
set { SetValue("DefaultQualityProfile", value); }
}
public virtual Boolean XbmcNotifyOnGrab
{
get { return GetValueBoolean("XbmcNotifyOnGrab"); }
set { SetValue("XbmcNotifyOnGrab", value); }
}
public virtual Boolean XbmcNotifyOnDownload
{
get { return GetValueBoolean("XbmcNotifyOnDownload"); }
set { SetValue("XbmcNotifyOnDownload", value); }
}
public virtual Boolean XbmcUpdateLibrary
{
get { return GetValueBoolean("XbmcUpdateLibrary"); }
set { SetValue("XbmcUpdateLibrary", value); }
}
public virtual Boolean XbmcCleanLibrary
{
get { return GetValueBoolean("XbmcCleanLibrary"); }
set { SetValue("XbmcCleanLibrary", value); }
}
public virtual string XbmcHosts
{
get { return GetValue("XbmcHosts", "localhost:8080"); }
set { SetValue("XbmcHosts", value); }
}
public virtual string XbmcUsername
{
get { return GetValue("XbmcUsername", "xbmc"); }
set { SetValue("XbmcUsername", value); }
}
public virtual string XbmcPassword
{
get { return GetValue("XbmcPassword", String.Empty); }
set { SetValue("XbmcPassword", value); }
}
public virtual string UpdateUrl
{
get { return GetValue("UpdateUrl", @"http://update.nzbdrone.com/master/"); }
set { SetValue("UpdateUrl", value); }
}
private string GetValue(string key)
{
2011-06-17 02:27:10 +00:00
return GetValue(key, String.Empty);
2010-09-23 03:19:47 +00:00
}
private bool GetValueBoolean(string key, bool defaultValue = false)
{
2011-06-17 02:27:10 +00:00
return Convert.ToBoolean(GetValue(key, defaultValue));
}
private int GetValueInt(string key, int defaultValue = 0)
{
2011-06-17 02:27:10 +00:00
return Convert.ToInt16(GetValue(key, defaultValue));
}
2011-06-17 02:27:10 +00:00
public virtual string GetValue(string key, object defaultValue)
2010-09-23 03:19:47 +00:00
{
2011-06-23 06:56:17 +00:00
var dbValue = _database.SingleOrDefault<Config>("WHERE [Key] =@0", key);
2010-09-23 03:19:47 +00:00
2010-09-28 06:09:24 +00:00
if (dbValue != null && !String.IsNullOrEmpty(dbValue.Value))
return dbValue.Value;
2010-09-23 03:19:47 +00:00
Logger.Trace("Unable to find config key '{0}' defaultValue:'{1}'", key, defaultValue);
return defaultValue.ToString();
2010-09-23 03:19:47 +00:00
}
private void SetValue(string key, Boolean value)
{
SetValue(key, value.ToString());
}
private void SetValue(string key, int value)
{
SetValue(key, value.ToString());
}
public void SetValue(string key, string value)
2010-09-23 03:19:47 +00:00
{
2010-09-28 06:09:24 +00:00
if (String.IsNullOrEmpty(key))
throw new ArgumentOutOfRangeException("key");
if (value == null)
throw new ArgumentNullException("key");
2010-09-24 05:37:48 +00:00
Logger.Trace("Writing Setting to file. Key:'{0}' Value:'{1}'", key, value);
2010-09-23 03:19:47 +00:00
2011-06-23 06:56:17 +00:00
var dbValue = _database.SingleOrDefault<Config>("WHERE [KEY]=@0", key);
2010-09-28 06:09:24 +00:00
if (dbValue == null)
{
2011-06-17 02:27:10 +00:00
_database.Insert(new Config { Key = key, Value = value });
2010-09-28 06:09:24 +00:00
}
else
{
dbValue.Value = value;
2011-06-23 06:56:17 +00:00
using (var tran = _database.GetTransaction())
{
_database.Update(dbValue);
tran.Complete();
}
}
2010-09-23 03:19:47 +00:00
}
}
}