Lidarr/src/NzbDrone.Core/Extras/Files/ExtraFileRepository.cs

using System.Collections.Generic;
using System.Linq;
using NzbDrone.Core.Datastore;
using NzbDrone.Core.Messaging.Events;

namespace NzbDrone.Core.Extras.Files
{
    public interface IExtraFileRepository<TExtraFile> : IBasicRepository<TExtraFile>
        where TExtraFile : ExtraFile, new()
    {
        void DeleteForArtists(List<int> artistIds);
        void DeleteForAlbum(int artistId, int albumId);
        void DeleteForTrackFile(int trackFileId);
        List<TExtraFile> GetFilesByArtist(int artistId);
        List<TExtraFile> GetFilesByAlbum(int artistId, int albumId);
        List<TExtraFile> GetFilesByTrackFile(int trackFileId);
        TExtraFile FindByPath(int artistId, string path);
    }

    public class ExtraFileRepository<TExtraFile> : BasicRepository<TExtraFile>, IExtraFileRepository<TExtraFile>
        where TExtraFile : ExtraFile, new()
    {
        public ExtraFileRepository(IMainDatabase database, IEventAggregator eventAggregator)
            : base(database, eventAggregator)
        {
        }

        public void DeleteForArtists(List<int> artistIds)
        {
            Delete(c => artistIds.Contains(c.ArtistId));
        }

        public void DeleteForAlbum(int artistId, int albumId)
        {
            Delete(c => c.ArtistId == artistId && c.AlbumId == albumId);
        }

        public void DeleteForTrackFile(int trackFileId)
        {
            Delete(c => c.TrackFileId == trackFileId);
        }

        public List<TExtraFile> GetFilesByArtist(int artistId)
        {
            return Query(c => c.ArtistId == artistId);
        }

        public List<TExtraFile> GetFilesByAlbum(int artistId, int albumId)
        {
            return Query(c => c.ArtistId == artistId && c.AlbumId == albumId);
        }

        public List<TExtraFile> GetFilesByTrackFile(int trackFileId)
        {
            return Query(c => c.TrackFileId == trackFileId);
        }

        public TExtraFile FindByPath(int artistId, string path)
        {
            return Query(c => c.ArtistId == artistId && c.RelativePath == path).SingleOrDefault();
        }
    }
}