mirror of
https://github.com/lidarr/Lidarr
synced 2024-12-25 01:02:05 +00:00
74 lines
2.1 KiB
C#
74 lines
2.1 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common.Messaging;
|
|
using NzbDrone.Core.Lifecycle;
|
|
|
|
namespace NzbDrone.Core.Qualities
|
|
{
|
|
public interface IQualitySizeService
|
|
{
|
|
void Update(QualitySize qualitySize);
|
|
void UpdateAll(List<QualitySize> qualitySizes);
|
|
List<QualitySize> All();
|
|
QualitySize Get(int qualityId);
|
|
}
|
|
|
|
public class QualitySizeService : IQualitySizeService, IHandle<ApplicationStartedEvent>
|
|
{
|
|
private readonly IQualitySizeRepository _qualitySizeRepository;
|
|
private readonly Logger _logger;
|
|
|
|
public QualitySizeService(IQualitySizeRepository qualitySizeRepository, Logger logger)
|
|
{
|
|
_qualitySizeRepository = qualitySizeRepository;
|
|
_logger = logger;
|
|
}
|
|
|
|
public virtual void Update(QualitySize qualitySize)
|
|
{
|
|
_qualitySizeRepository.Update(qualitySize);
|
|
}
|
|
|
|
public virtual void UpdateAll(List<QualitySize> qualitySizes)
|
|
{
|
|
_qualitySizeRepository.UpdateMany(qualitySizes);
|
|
}
|
|
|
|
public virtual List<QualitySize> All()
|
|
{
|
|
return _qualitySizeRepository.All().ToList();
|
|
}
|
|
|
|
public virtual QualitySize Get(int qualityId)
|
|
{
|
|
return _qualitySizeRepository.GetByQualityId(qualityId);
|
|
}
|
|
|
|
public void Init()
|
|
{
|
|
|
|
}
|
|
|
|
public void Handle(ApplicationStartedEvent message)
|
|
{
|
|
var existing = All();
|
|
|
|
_logger.Debug("Setting up default quality sizes");
|
|
|
|
foreach (var quality in Quality.All().Where(q => q.Id > 0))
|
|
{
|
|
if (!existing.Any(s => s.QualityId == quality.Id))
|
|
{
|
|
_qualitySizeRepository.Insert(new QualitySize
|
|
{
|
|
QualityId = quality.Id,
|
|
Name = quality.Name,
|
|
MinSize = 0,
|
|
MaxSize = 100
|
|
});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|