Radarr/NzbDrone.Core/Providers/ConfigProvider.cs

83 lines
2.3 KiB
C#

using System;
using NLog;
using NzbDrone.Core.Repository;
using SubSonic.Repository;
namespace NzbDrone.Core.Providers
{
public class ConfigProvider : IConfigProvider
{
private const string SERIES_ROOTS = "SeriesRoots";
private const string EPISODE_NAME_FORMAT = "EpisodeNameFormat";
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
private readonly IRepository _sonicRepo;
public ConfigProvider(IRepository dataRepository)
{
_sonicRepo = dataRepository;
}
private string GetValue(string key)
{
return GetValue(key, String.Empty, false);
}
public String EpisodeNameFormat
{
get { return GetValue(EPISODE_NAME_FORMAT); }
set { SetValue(EPISODE_NAME_FORMAT, value); }
}
public String SeriesRoot
{
get { return GetValue(SERIES_ROOTS); }
set { SetValue(SERIES_ROOTS, value); }
}
public string GetValue(string key, object defaultValue, bool makePermanent)
{
string value;
var dbValue = _sonicRepo.Single<Config>(key);
if (dbValue != null && !String.IsNullOrEmpty(dbValue.Value))
return dbValue.Value;
Logger.Debug("Unable to find config key '{0}' defaultValue:'{1}'", key, defaultValue);
if (makePermanent)
SetValue(key, defaultValue.ToString());
value = defaultValue.ToString();
return value;
}
public void SetValue(string key, string value)
{
if (String.IsNullOrEmpty(key))
throw new ArgumentOutOfRangeException("key");
if (value == null)
throw new ArgumentNullException("key");
Logger.Debug("Writing Setting to file. Key:'{0}' Value:'{1}'", key, value);
var dbValue = _sonicRepo.Single<Config>(key);
if (dbValue == null)
{
_sonicRepo.Add(new Config
{
Key = key,
Value = value
});
}
else
{
dbValue.Value = value;
_sonicRepo.Update(dbValue);
}
}
}
}