2011-02-03 02:07:36 +01:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
2011-06-14 03:23:04 +02:00
|
|
|
|
using Ninject;
|
2011-02-04 03:58:02 +01:00
|
|
|
|
using NLog;
|
2011-02-03 02:07:36 +01:00
|
|
|
|
using NzbDrone.Core.Repository.Quality;
|
2011-06-17 22:31:25 +02:00
|
|
|
|
using PetaPoco;
|
2011-02-03 02:07:36 +01:00
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Providers
|
|
|
|
|
{
|
2011-04-08 06:03:46 +02:00
|
|
|
|
public class QualityProvider
|
2011-02-03 02:07:36 +01:00
|
|
|
|
{
|
2011-02-04 03:58:02 +01:00
|
|
|
|
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
|
2011-06-17 22:31:25 +02:00
|
|
|
|
private readonly IDatabase _database;
|
2011-02-03 02:07:36 +01:00
|
|
|
|
|
2011-04-08 08:50:30 +02:00
|
|
|
|
public QualityProvider()
|
|
|
|
|
{
|
|
|
|
|
}
|
2011-04-08 06:03:46 +02:00
|
|
|
|
|
2011-06-14 03:23:04 +02:00
|
|
|
|
[Inject]
|
2011-06-17 22:31:25 +02:00
|
|
|
|
public QualityProvider(IDatabase database)
|
2011-02-03 02:07:36 +01:00
|
|
|
|
{
|
2011-06-17 22:31:25 +02:00
|
|
|
|
_database = database;
|
2011-02-03 02:07:36 +01:00
|
|
|
|
}
|
|
|
|
|
|
2011-04-22 02:30:19 +02:00
|
|
|
|
public virtual int Add(QualityProfile profile)
|
2011-02-03 02:07:36 +01:00
|
|
|
|
{
|
2011-06-17 22:31:25 +02:00
|
|
|
|
return Convert.ToInt32(_database.Insert(profile));
|
2011-02-03 02:07:36 +01:00
|
|
|
|
}
|
|
|
|
|
|
2011-04-08 06:03:46 +02:00
|
|
|
|
public virtual void Update(QualityProfile profile)
|
2011-02-03 02:07:36 +01:00
|
|
|
|
{
|
2011-06-17 22:31:25 +02:00
|
|
|
|
if (!_database.Exists<QualityProfile>("WHERE QualityProfileid = @0", profile.QualityProfileId))
|
2011-02-03 02:07:36 +01:00
|
|
|
|
{
|
2011-02-04 03:58:02 +01:00
|
|
|
|
Logger.Error("Unable to update non-existing profile");
|
|
|
|
|
throw new InvalidOperationException("Unable to update non-existing profile");
|
2011-02-03 02:07:36 +01:00
|
|
|
|
}
|
|
|
|
|
|
2011-06-17 22:31:25 +02:00
|
|
|
|
_database.Update(profile);
|
2011-02-03 02:07:36 +01:00
|
|
|
|
}
|
|
|
|
|
|
2011-04-08 06:03:46 +02:00
|
|
|
|
public virtual void Delete(int profileId)
|
2011-02-03 02:07:36 +01:00
|
|
|
|
{
|
2011-06-17 22:31:25 +02:00
|
|
|
|
_database.Delete<QualityProfile>(profileId);
|
2011-02-03 02:07:36 +01:00
|
|
|
|
}
|
|
|
|
|
|
2011-07-08 07:41:08 +02:00
|
|
|
|
public virtual List<QualityProfile> All()
|
2011-02-03 02:07:36 +01:00
|
|
|
|
{
|
2011-06-17 22:31:25 +02:00
|
|
|
|
var profiles = _database.Fetch<QualityProfile>().ToList();
|
2011-02-03 02:07:36 +01:00
|
|
|
|
|
|
|
|
|
return profiles;
|
|
|
|
|
}
|
|
|
|
|
|
2011-06-18 06:39:02 +02:00
|
|
|
|
public virtual QualityProfile Get(int profileId)
|
2011-02-05 07:07:25 +01:00
|
|
|
|
{
|
2011-06-19 01:03:58 +02:00
|
|
|
|
return _database.Single<QualityProfile>(profileId);
|
2011-02-05 07:07:25 +01:00
|
|
|
|
}
|
2011-06-13 05:45:22 +02:00
|
|
|
|
|
|
|
|
|
public virtual void SetupDefaultProfiles()
|
|
|
|
|
{
|
2011-07-08 07:41:08 +02:00
|
|
|
|
if (All().Count != 0)
|
2011-07-08 05:57:31 +02:00
|
|
|
|
return;
|
2011-06-13 05:45:22 +02:00
|
|
|
|
|
2011-07-08 05:57:31 +02:00
|
|
|
|
Logger.Info("Setting up default quality profiles");
|
2011-06-13 05:45:22 +02:00
|
|
|
|
|
|
|
|
|
var sd = new QualityProfile { Name = "SD", Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }, Cutoff = QualityTypes.SDTV };
|
|
|
|
|
|
|
|
|
|
var hd = new QualityProfile
|
|
|
|
|
{
|
|
|
|
|
Name = "HD",
|
2012-10-15 06:30:07 +02:00
|
|
|
|
Allowed = new List<QualityTypes> { QualityTypes.HDTV, QualityTypes.WEBDL720p, QualityTypes.Bluray720p },
|
2011-06-13 05:45:22 +02:00
|
|
|
|
Cutoff = QualityTypes.HDTV
|
|
|
|
|
};
|
|
|
|
|
|
2011-07-08 05:57:31 +02:00
|
|
|
|
Add(sd);
|
|
|
|
|
Add(hd);
|
2011-06-13 05:45:22 +02:00
|
|
|
|
|
|
|
|
|
}
|
2011-02-03 02:07:36 +01:00
|
|
|
|
}
|
2011-04-10 04:44:01 +02:00
|
|
|
|
}
|