2017-09-16 20:27:38 +00:00
|
|
|
using System;
|
2013-06-13 01:37:05 +00:00
|
|
|
using System.Collections.Generic;
|
2013-03-04 05:53:02 +00:00
|
|
|
using System.IO;
|
2019-05-07 23:10:09 +00:00
|
|
|
using System.Linq;
|
2013-07-05 05:16:49 +00:00
|
|
|
using System.Net;
|
2013-03-04 05:53:02 +00:00
|
|
|
using NLog;
|
2014-01-06 06:20:08 +00:00
|
|
|
using NzbDrone.Common.Disk;
|
2013-06-28 00:04:52 +00:00
|
|
|
using NzbDrone.Common.EnvironmentInfo;
|
2014-12-02 06:26:25 +00:00
|
|
|
using NzbDrone.Common.Extensions;
|
2014-05-11 04:37:07 +00:00
|
|
|
using NzbDrone.Common.Http;
|
2014-01-02 06:56:19 +00:00
|
|
|
using NzbDrone.Core.Configuration;
|
2013-09-14 06:36:07 +00:00
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2017-09-17 03:26:56 +00:00
|
|
|
using NzbDrone.Core.Music;
|
|
|
|
using NzbDrone.Core.Music.Events;
|
2013-03-04 05:53:02 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.MediaCover
|
|
|
|
{
|
2014-01-22 05:22:09 +00:00
|
|
|
public interface IMapCoversToLocal
|
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
void ConvertToLocalUrls(int entityId, MediaCoverEntity coverEntity, IEnumerable<MediaCover> covers);
|
|
|
|
string GetCoverPath(int entityId, MediaCoverEntity coverEntity, MediaCoverTypes mediaCoverTypes, int? height = null);
|
2014-01-22 05:22:09 +00:00
|
|
|
}
|
|
|
|
|
2013-03-05 19:49:34 +00:00
|
|
|
public class MediaCoverService :
|
2017-09-17 03:26:56 +00:00
|
|
|
IHandleAsync<ArtistUpdatedEvent>,
|
|
|
|
IHandleAsync<ArtistDeletedEvent>,
|
2013-06-13 01:37:05 +00:00
|
|
|
IMapCoversToLocal
|
2013-03-04 05:53:02 +00:00
|
|
|
{
|
2015-01-22 22:12:35 +00:00
|
|
|
private readonly IImageResizer _resizer;
|
2017-09-17 03:26:56 +00:00
|
|
|
private readonly IAlbumService _albumService;
|
2014-09-11 23:49:41 +00:00
|
|
|
private readonly IHttpClient _httpClient;
|
2013-05-10 23:53:50 +00:00
|
|
|
private readonly IDiskProvider _diskProvider;
|
2013-05-29 04:10:23 +00:00
|
|
|
private readonly ICoverExistsSpecification _coverExistsSpecification;
|
2014-01-02 06:56:19 +00:00
|
|
|
private readonly IConfigFileProvider _configFileProvider;
|
2014-01-26 07:14:55 +00:00
|
|
|
private readonly IEventAggregator _eventAggregator;
|
2013-03-04 05:53:02 +00:00
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
private readonly string _coverRootFolder;
|
|
|
|
|
2015-01-22 22:12:35 +00:00
|
|
|
public MediaCoverService(IImageResizer resizer,
|
2017-09-17 03:26:56 +00:00
|
|
|
IAlbumService albumService,
|
2015-01-22 22:12:35 +00:00
|
|
|
IHttpClient httpClient,
|
2014-01-26 07:14:55 +00:00
|
|
|
IDiskProvider diskProvider,
|
|
|
|
IAppFolderInfo appFolderInfo,
|
2014-09-11 23:49:41 +00:00
|
|
|
ICoverExistsSpecification coverExistsSpecification,
|
|
|
|
IConfigFileProvider configFileProvider,
|
2014-01-26 07:14:55 +00:00
|
|
|
IEventAggregator eventAggregator,
|
|
|
|
Logger logger)
|
2013-03-04 05:53:02 +00:00
|
|
|
{
|
2015-01-22 22:12:35 +00:00
|
|
|
_resizer = resizer;
|
2017-09-17 03:26:56 +00:00
|
|
|
_albumService = albumService;
|
2014-09-11 23:49:41 +00:00
|
|
|
_httpClient = httpClient;
|
2013-03-04 05:53:02 +00:00
|
|
|
_diskProvider = diskProvider;
|
2013-05-29 04:10:23 +00:00
|
|
|
_coverExistsSpecification = coverExistsSpecification;
|
2014-01-02 06:56:19 +00:00
|
|
|
_configFileProvider = configFileProvider;
|
2014-01-26 07:14:55 +00:00
|
|
|
_eventAggregator = eventAggregator;
|
2013-03-04 05:53:02 +00:00
|
|
|
_logger = logger;
|
|
|
|
|
2013-07-05 04:43:28 +00:00
|
|
|
_coverRootFolder = appFolderInfo.GetMediaCoverPath();
|
2013-03-04 05:53:02 +00:00
|
|
|
}
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
public string GetCoverPath(int entityId, MediaCoverEntity coverEntity, MediaCoverTypes coverTypes, int? height = null)
|
2013-03-04 05:53:02 +00:00
|
|
|
{
|
2015-01-22 22:12:35 +00:00
|
|
|
var heightSuffix = height.HasValue ? "-" + height.ToString() : "";
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
if (coverEntity == MediaCoverEntity.Album)
|
2017-09-17 03:26:56 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
return Path.Combine(GetAlbumCoverPath(entityId), coverTypes.ToString().ToLower() + heightSuffix + ".jpg");
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
return Path.Combine(GetArtistCoverPath(entityId), coverTypes.ToString().ToLower() + heightSuffix + ".jpg");
|
2017-09-17 03:26:56 +00:00
|
|
|
}
|
2014-01-22 05:22:09 +00:00
|
|
|
}
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
public void ConvertToLocalUrls(int entityId, MediaCoverEntity coverEntity, IEnumerable<MediaCover> covers)
|
2014-01-22 05:22:09 +00:00
|
|
|
{
|
|
|
|
foreach (var mediaCover in covers)
|
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
var filePath = GetCoverPath(entityId, coverEntity, mediaCover.CoverType, null);
|
2014-01-22 05:22:09 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
if (coverEntity == MediaCoverEntity.Album)
|
2017-09-17 03:26:56 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
mediaCover.Url = _configFileProvider.UrlBase + @"/MediaCover/Albums/" + entityId + "/" + mediaCover.CoverType.ToString().ToLower() + ".jpg";
|
2017-09-17 03:26:56 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
mediaCover.Url = _configFileProvider.UrlBase + @"/MediaCover/" + entityId + "/" + mediaCover.CoverType.ToString().ToLower() + ".jpg";
|
2017-09-17 03:26:56 +00:00
|
|
|
}
|
2014-01-22 05:22:09 +00:00
|
|
|
|
|
|
|
if (_diskProvider.FileExists(filePath))
|
|
|
|
{
|
2014-10-14 21:31:35 +00:00
|
|
|
var lastWrite = _diskProvider.FileGetLastWrite(filePath);
|
2014-01-22 05:22:09 +00:00
|
|
|
mediaCover.Url += "?lastWrite=" + lastWrite.Ticks;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-09-17 03:26:56 +00:00
|
|
|
private string GetArtistCoverPath(int artistId)
|
|
|
|
{
|
|
|
|
return Path.Combine(_coverRootFolder, artistId.ToString());
|
|
|
|
}
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
private string GetAlbumCoverPath(int albumId)
|
2014-01-22 05:22:09 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
return Path.Combine(_coverRootFolder, "Albums", albumId.ToString());
|
2017-09-17 03:26:56 +00:00
|
|
|
}
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
private void EnsureArtistCovers(Artist artist)
|
2017-09-17 03:26:56 +00:00
|
|
|
{
|
2018-12-15 00:02:43 +00:00
|
|
|
foreach (var cover in artist.Metadata.Value.Images)
|
2017-09-17 03:26:56 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
var fileName = GetCoverPath(artist.Id, MediaCoverEntity.Artist, cover.CoverType);
|
2017-09-17 03:26:56 +00:00
|
|
|
var alreadyExists = false;
|
2018-05-25 01:08:35 +00:00
|
|
|
|
2017-09-17 03:26:56 +00:00
|
|
|
try
|
|
|
|
{
|
2018-05-25 01:08:35 +00:00
|
|
|
var lastModifiedServer = GetCoverModifiedDate(cover.Url);
|
|
|
|
|
|
|
|
alreadyExists = _coverExistsSpecification.AlreadyExists(lastModifiedServer, fileName);
|
|
|
|
|
2017-09-17 03:26:56 +00:00
|
|
|
if (!alreadyExists)
|
|
|
|
{
|
2018-05-25 01:08:35 +00:00
|
|
|
DownloadCover(artist, cover, lastModifiedServer);
|
2017-09-17 03:26:56 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (WebException e)
|
|
|
|
{
|
|
|
|
_logger.Warn("Couldn't download media cover for {0}. {1}", artist, e.Message);
|
|
|
|
}
|
|
|
|
catch (Exception e)
|
|
|
|
{
|
|
|
|
_logger.Error(e, "Couldn't download media cover for {0}", artist);
|
|
|
|
}
|
|
|
|
|
|
|
|
EnsureResizedCovers(artist, cover, !alreadyExists);
|
|
|
|
}
|
2013-03-04 05:53:02 +00:00
|
|
|
}
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
private void EnsureAlbumCovers(Album album)
|
|
|
|
{
|
|
|
|
foreach (var cover in album.Images.Where(e => e.CoverType == MediaCoverTypes.Cover))
|
|
|
|
{
|
|
|
|
var fileName = GetCoverPath(album.Id, MediaCoverEntity.Album, cover.CoverType, null);
|
|
|
|
var alreadyExists = false;
|
|
|
|
try
|
|
|
|
{
|
|
|
|
var lastModifiedServer = GetCoverModifiedDate(cover.Url);
|
|
|
|
alreadyExists = _coverExistsSpecification.AlreadyExists(lastModifiedServer, fileName);
|
|
|
|
if (!alreadyExists)
|
|
|
|
{
|
|
|
|
DownloadAlbumCover(album, cover, lastModifiedServer);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (WebException e)
|
|
|
|
{
|
|
|
|
_logger.Warn("Couldn't download media cover for {0}. {1}", album, e.Message);
|
|
|
|
}
|
|
|
|
catch (Exception e)
|
|
|
|
{
|
|
|
|
_logger.Error(e, "Couldn't download media cover for {0}", album);
|
|
|
|
}
|
|
|
|
|
|
|
|
EnsureResizedAlbumCovers(album, cover, !alreadyExists);
|
|
|
|
}
|
|
|
|
}
|
2018-05-25 01:08:35 +00:00
|
|
|
|
|
|
|
private void DownloadCover(Artist artist, MediaCover cover, DateTime lastModified)
|
2017-09-17 03:26:56 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
var fileName = GetCoverPath(artist.Id, MediaCoverEntity.Artist, cover.CoverType);
|
2017-09-17 03:26:56 +00:00
|
|
|
|
|
|
|
_logger.Info("Downloading {0} for {1} {2}", cover.CoverType, artist, cover.Url);
|
|
|
|
_httpClient.DownloadFile(cover.Url, fileName);
|
2018-05-25 01:08:35 +00:00
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
_diskProvider.FileSetLastWriteTime(fileName, lastModified);
|
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
|
|
|
_logger.Debug(ex, "Unable to set modified date for {0} image for artist {1}", cover.CoverType, artist);
|
|
|
|
}
|
2017-09-17 03:26:56 +00:00
|
|
|
}
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
private void DownloadAlbumCover(Album album, MediaCover cover, DateTime lastModified)
|
|
|
|
{
|
|
|
|
var fileName = GetCoverPath(album.Id, MediaCoverEntity.Album, cover.CoverType, null);
|
2013-08-30 02:00:40 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
_logger.Info("Downloading {0} for {1} {2}", cover.CoverType, album, cover.Url);
|
|
|
|
_httpClient.DownloadFile(cover.Url, fileName);
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
_diskProvider.FileSetLastWriteTime(fileName, lastModified);
|
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
|
|
|
_logger.Debug(ex, "Unable to set modified date for {0} image for album {1}", cover.CoverType, album);
|
|
|
|
}
|
|
|
|
}
|
2013-03-04 05:53:02 +00:00
|
|
|
|
2017-09-17 03:26:56 +00:00
|
|
|
private void EnsureResizedCovers(Artist artist, MediaCover cover, bool forceResize, Album album = null)
|
2015-01-22 22:12:35 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
int[] heights = GetDefaultHeights(cover.CoverType);
|
2015-01-22 22:12:35 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
foreach (var height in heights)
|
2015-01-22 22:12:35 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
var mainFileName = GetCoverPath(artist.Id, MediaCoverEntity.Artist, cover.CoverType);
|
|
|
|
var resizeFileName = GetCoverPath(artist.Id, MediaCoverEntity.Artist, cover.CoverType, height);
|
2015-01-22 22:12:35 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
if (forceResize || !_diskProvider.FileExists(resizeFileName) || _diskProvider.GetFileSize(resizeFileName) == 0)
|
2015-01-22 22:12:35 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
_logger.Debug("Resizing {0}-{1} for {2}", cover.CoverType, height, artist);
|
2015-01-22 22:12:35 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
try
|
|
|
|
{
|
|
|
|
_resizer.Resize(mainFileName, resizeFileName, height);
|
|
|
|
}
|
|
|
|
catch
|
2015-02-05 21:15:30 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
_logger.Debug("Couldn't resize media cover {0}-{1} for artist {2}, using full size image instead.", cover.CoverType, height, artist);
|
2015-02-05 21:15:30 +00:00
|
|
|
}
|
2017-09-17 03:26:56 +00:00
|
|
|
}
|
|
|
|
}
|
2019-05-07 23:10:09 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private void EnsureResizedAlbumCovers(Album album, MediaCover cover, bool forceResize)
|
|
|
|
{
|
|
|
|
int[] heights = GetDefaultHeights(cover.CoverType);
|
|
|
|
|
|
|
|
foreach (var height in heights)
|
2017-09-17 03:26:56 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
var mainFileName = GetCoverPath(album.Id, MediaCoverEntity.Album, cover.CoverType, null);
|
|
|
|
var resizeFileName = GetCoverPath(album.Id, MediaCoverEntity.Album, cover.CoverType, height);
|
2017-09-17 03:26:56 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
if (forceResize || !_diskProvider.FileExists(resizeFileName) || _diskProvider.GetFileSize(resizeFileName) == 0)
|
|
|
|
{
|
|
|
|
_logger.Debug("Resizing {0}-{1} for {2}", cover.CoverType, height, album);
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
_resizer.Resize(mainFileName, resizeFileName, height);
|
|
|
|
}
|
|
|
|
catch
|
2015-02-05 21:15:30 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
_logger.Debug("Couldn't resize media cover {0}-{1} for album {2}, using full size image instead.", cover.CoverType, height, album);
|
2015-02-05 21:15:30 +00:00
|
|
|
}
|
2015-01-22 22:12:35 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
private DateTime GetCoverModifiedDate(string url)
|
2014-01-22 05:22:09 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
var lastModifiedServer = DateTime.Now;
|
2017-09-17 03:26:56 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
var headers = _httpClient.Head(new HttpRequest(url)).Headers;
|
2017-10-07 21:40:17 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
if (headers.LastModified.HasValue)
|
|
|
|
{
|
|
|
|
lastModifiedServer = headers.LastModified.Value;
|
|
|
|
}
|
|
|
|
|
|
|
|
return lastModifiedServer;
|
2014-01-22 05:22:09 +00:00
|
|
|
}
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
private int[] GetDefaultHeights(MediaCoverTypes coverType)
|
2013-03-05 19:49:34 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
switch (coverType)
|
2013-03-05 19:49:34 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
default:
|
|
|
|
return new int[] { };
|
|
|
|
|
|
|
|
case MediaCoverTypes.Poster:
|
|
|
|
case MediaCoverTypes.Disc:
|
|
|
|
case MediaCoverTypes.Logo:
|
|
|
|
case MediaCoverTypes.Headshot:
|
|
|
|
return new[] { 500, 250 };
|
|
|
|
|
|
|
|
case MediaCoverTypes.Banner:
|
|
|
|
return new[] { 70, 35 };
|
|
|
|
|
|
|
|
case MediaCoverTypes.Fanart:
|
|
|
|
case MediaCoverTypes.Screenshot:
|
|
|
|
return new[] { 360, 180 };
|
|
|
|
case MediaCoverTypes.Cover:
|
|
|
|
return new[] { 250 };
|
2013-03-05 19:49:34 +00:00
|
|
|
}
|
|
|
|
}
|
2018-05-25 01:08:35 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
public void HandleAsync(ArtistUpdatedEvent message)
|
2018-05-25 01:08:35 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
EnsureArtistCovers(message.Artist);
|
2018-05-25 01:08:35 +00:00
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
var albums = _albumService.GetAlbumsByArtist(message.Artist.Id);
|
|
|
|
foreach (Album album in albums)
|
2018-05-25 01:08:35 +00:00
|
|
|
{
|
2019-05-07 23:10:09 +00:00
|
|
|
EnsureAlbumCovers(album);
|
2018-05-25 01:08:35 +00:00
|
|
|
}
|
|
|
|
|
2019-05-07 23:10:09 +00:00
|
|
|
_eventAggregator.PublishEvent(new MediaCoversUpdatedEvent(message.Artist));
|
2018-05-25 01:08:35 +00:00
|
|
|
}
|
2019-05-07 23:10:09 +00:00
|
|
|
|
|
|
|
public void HandleAsync(ArtistDeletedEvent message)
|
|
|
|
{
|
|
|
|
var path = GetArtistCoverPath(message.Artist.Id);
|
|
|
|
if (_diskProvider.FolderExists(path))
|
|
|
|
{
|
|
|
|
_diskProvider.DeleteFolder(path, true);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-03-04 05:53:02 +00:00
|
|
|
}
|
|
|
|
}
|