using System; using System.Collections.Generic; using System.Linq; using NLog; using NzbDrone.Core.Repository.Quality; using SubSonic.Repository; namespace NzbDrone.Core.Providers { public class QualityProvider { private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); private readonly IRepository _repository; public QualityProvider() { } public QualityProvider(IRepository repository) { _repository = repository; } public virtual int Add(QualityProfile profile) { return Convert.ToInt32(_repository.Add(profile)); } public virtual void Update(QualityProfile profile) { if (!_repository.Exists(q => q.QualityProfileId == profile.QualityProfileId)) { Logger.Error("Unable to update non-existing profile"); throw new InvalidOperationException("Unable to update non-existing profile"); } _repository.Update(profile); } public virtual void Delete(int profileId) { _repository.Delete(profileId); } public virtual List GetAllProfiles() { var profiles = _repository.All().ToList(); return profiles; } public virtual QualityProfile Find(int profileId) { return _repository.Single(q => q.QualityProfileId == profileId); } } }