Lidarr/src/NzbDrone.Core/Download/Aggregation/RemoteAlbumAggregationService.cs

using System;
using System.Collections.Generic;
using NLog;
using NzbDrone.Core.Download.Aggregation.Aggregators;
using NzbDrone.Core.Parser.Model;

namespace NzbDrone.Core.Download.Aggregation
{
    public interface IRemoteAlbumAggregationService
    {
        RemoteAlbum Augment(RemoteAlbum remoteAlbum);
    }

    public class RemoteAlbumAggregationService : IRemoteAlbumAggregationService
    {
        private readonly IEnumerable<IAggregateRemoteAlbum> _augmenters;
        private readonly Logger _logger;

        public RemoteAlbumAggregationService(IEnumerable<IAggregateRemoteAlbum> augmenters,
                                  Logger logger)
        {
            _augmenters = augmenters;
            _logger = logger;
        }

        public RemoteAlbum Augment(RemoteAlbum remoteAlbum)
        {
            foreach (var augmenter in _augmenters)
            {
                try
                {
                    augmenter.Aggregate(remoteAlbum);
                }
                catch (Exception ex)
                {
                    _logger.Warn(ex, ex.Message);
                }
            }

            return remoteAlbum;
        }
    }
}