You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
502 lines
19 KiB
502 lines
19 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using NLog;
|
|
using NzbDrone.Common.Cloud;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Common.Http;
|
|
using NzbDrone.Core.Exceptions;
|
|
using NzbDrone.Core.MediaCover;
|
|
using NzbDrone.Core.MetadataSource.SkyHook.Resource;
|
|
using NzbDrone.Core.Music;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Profiles.Metadata;
|
|
|
|
namespace NzbDrone.Core.MetadataSource.SkyHook
|
|
{
|
|
public class SkyHookProxy : IProvideArtistInfo, ISearchForNewArtist, IProvideAlbumInfo, ISearchForNewAlbum
|
|
{
|
|
private readonly IHttpClient _httpClient;
|
|
private readonly Logger _logger;
|
|
|
|
private readonly IArtistService _artistService;
|
|
private readonly IAlbumService _albumService;
|
|
private readonly IHttpRequestBuilderFactory _requestBuilder;
|
|
private readonly IConfigService _configService;
|
|
private readonly IMetadataProfileService _metadataProfileService;
|
|
|
|
private IHttpRequestBuilderFactory _customerRequestBuilder;
|
|
|
|
public SkyHookProxy(IHttpClient httpClient,
|
|
ILidarrCloudRequestBuilder requestBuilder,
|
|
IArtistService artistService,
|
|
IAlbumService albumService,
|
|
Logger logger,
|
|
IConfigService configService,
|
|
IMetadataProfileService metadataProfileService)
|
|
{
|
|
_httpClient = httpClient;
|
|
_configService = configService;
|
|
_metadataProfileService = metadataProfileService;
|
|
_requestBuilder = requestBuilder.Search;
|
|
_artistService = artistService;
|
|
_albumService = albumService;
|
|
_logger = logger;
|
|
}
|
|
|
|
public Artist GetArtistInfo(string foreignArtistId, int metadataProfileId)
|
|
{
|
|
|
|
_logger.Debug("Getting Artist with LidarrAPI.MetadataID of {0}", foreignArtistId);
|
|
|
|
SetCustomProvider();
|
|
|
|
var metadataProfile = _metadataProfileService.Exists(metadataProfileId) ? _metadataProfileService.Get(metadataProfileId) : _metadataProfileService.All().First();
|
|
|
|
var primaryTypes = metadataProfile.PrimaryAlbumTypes.Where(s => s.Allowed).Select(s => s.PrimaryAlbumType.Name);
|
|
var secondaryTypes = metadataProfile.SecondaryAlbumTypes.Where(s => s.Allowed).Select(s => s.SecondaryAlbumType.Name);
|
|
var releaseStatuses = metadataProfile.ReleaseStatuses.Where(s => s.Allowed).Select(s => s.ReleaseStatus.Name);
|
|
|
|
var httpRequest = _customerRequestBuilder.Create()
|
|
.SetSegment("route", "artist/" + foreignArtistId)
|
|
.AddQueryParam("primTypes", string.Join("|", primaryTypes))
|
|
.AddQueryParam("secTypes", string.Join("|", secondaryTypes))
|
|
.AddQueryParam("releaseStatuses", string.Join("|", releaseStatuses))
|
|
.Build();
|
|
|
|
httpRequest.AllowAutoRedirect = true;
|
|
httpRequest.SuppressHttpError = true;
|
|
|
|
var httpResponse = _httpClient.Get<ArtistResource>(httpRequest);
|
|
|
|
|
|
if (httpResponse.HasHttpError)
|
|
{
|
|
if (httpResponse.StatusCode == HttpStatusCode.NotFound)
|
|
{
|
|
throw new ArtistNotFoundException(foreignArtistId);
|
|
}
|
|
else if (httpResponse.StatusCode == HttpStatusCode.BadRequest)
|
|
{
|
|
throw new BadRequestException(foreignArtistId);
|
|
}
|
|
else
|
|
{
|
|
throw new HttpException(httpRequest, httpResponse);
|
|
}
|
|
}
|
|
|
|
var artist = new Artist();
|
|
artist.Metadata = MapArtistMetadata(httpResponse.Resource);
|
|
artist.CleanName = Parser.Parser.CleanArtistName(artist.Metadata.Value.Name);
|
|
artist.SortName = Parser.Parser.NormalizeTitle(artist.Metadata.Value.Name);
|
|
artist.Albums = httpResponse.Resource.Albums.Select(x => MapAlbum(x, null)).ToList();
|
|
|
|
return artist;
|
|
}
|
|
|
|
public Tuple<string, Album, List<ArtistMetadata>> GetAlbumInfo(string foreignAlbumId)
|
|
{
|
|
_logger.Debug("Getting Album with LidarrAPI.MetadataID of {0}", foreignAlbumId);
|
|
|
|
SetCustomProvider();
|
|
|
|
var httpRequest = _customerRequestBuilder.Create()
|
|
.SetSegment("route", "album/" + foreignAlbumId)
|
|
.Build();
|
|
|
|
httpRequest.AllowAutoRedirect = true;
|
|
httpRequest.SuppressHttpError = true;
|
|
|
|
var httpResponse = _httpClient.Get<AlbumResource>(httpRequest);
|
|
|
|
if (httpResponse.HasHttpError)
|
|
{
|
|
if (httpResponse.StatusCode == HttpStatusCode.NotFound)
|
|
{
|
|
throw new AlbumNotFoundException(foreignAlbumId);
|
|
}
|
|
else if (httpResponse.StatusCode == HttpStatusCode.BadRequest)
|
|
{
|
|
throw new BadRequestException(foreignAlbumId);
|
|
}
|
|
else
|
|
{
|
|
throw new HttpException(httpRequest, httpResponse);
|
|
}
|
|
}
|
|
|
|
var artists = httpResponse.Resource.Artists.Select(MapArtistMetadata).ToList();
|
|
var artistDict = artists.ToDictionary(x => x.ForeignArtistId, x => x);
|
|
var album = MapAlbum(httpResponse.Resource, artistDict);
|
|
|
|
return new Tuple<string, Album, List<ArtistMetadata>>(httpResponse.Resource.ArtistId, album, artists);
|
|
}
|
|
|
|
public List<Artist> SearchForNewArtist(string title)
|
|
{
|
|
try
|
|
{
|
|
var lowerTitle = title.ToLowerInvariant();
|
|
|
|
if (lowerTitle.StartsWith("lidarr:") || lowerTitle.StartsWith("lidarrid:") || lowerTitle.StartsWith("mbid:"))
|
|
{
|
|
var slug = lowerTitle.Split(':')[1].Trim();
|
|
|
|
Guid searchGuid;
|
|
|
|
bool isValid = Guid.TryParse(slug, out searchGuid);
|
|
|
|
if (slug.IsNullOrWhiteSpace() || slug.Any(char.IsWhiteSpace) || isValid == false)
|
|
{
|
|
return new List<Artist>();
|
|
}
|
|
|
|
try
|
|
{
|
|
var existingArtist = _artistService.FindById(searchGuid.ToString());
|
|
if (existingArtist != null)
|
|
{
|
|
return new List<Artist> { existingArtist };
|
|
}
|
|
|
|
var metadataProfile = _metadataProfileService.All().First().Id; //Change this to Use last Used profile?
|
|
|
|
return new List<Artist> { GetArtistInfo(searchGuid.ToString(), metadataProfile) };
|
|
}
|
|
catch (ArtistNotFoundException)
|
|
{
|
|
return new List<Artist>();
|
|
}
|
|
}
|
|
|
|
SetCustomProvider();
|
|
|
|
var httpRequest = _customerRequestBuilder.Create()
|
|
.SetSegment("route", "search")
|
|
.AddQueryParam("type", "artist")
|
|
.AddQueryParam("query", title.ToLower().Trim())
|
|
//.AddQueryParam("images","false") // Should pass these on import search to avoid looking to fanart and wiki
|
|
//.AddQueryParam("overview","false")
|
|
.Build();
|
|
|
|
|
|
|
|
var httpResponse = _httpClient.Get<List<ArtistResource>>(httpRequest);
|
|
|
|
return httpResponse.Resource.SelectList(MapSearchResult);
|
|
}
|
|
catch (HttpException)
|
|
{
|
|
throw new SkyHookException("Search for '{0}' failed. Unable to communicate with LidarrAPI.", title);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Warn(ex, ex.Message);
|
|
throw new SkyHookException("Search for '{0}' failed. Invalid response received from LidarrAPI.", title);
|
|
}
|
|
}
|
|
|
|
public List<Album> SearchForNewAlbum(string title, string artist)
|
|
{
|
|
try
|
|
{
|
|
var lowerTitle = title.ToLowerInvariant();
|
|
|
|
if (lowerTitle.StartsWith("lidarr:") || lowerTitle.StartsWith("lidarrid:") || lowerTitle.StartsWith("mbid:"))
|
|
{
|
|
var slug = lowerTitle.Split(':')[1].Trim();
|
|
|
|
Guid searchGuid;
|
|
|
|
bool isValid = Guid.TryParse(slug, out searchGuid);
|
|
|
|
if (slug.IsNullOrWhiteSpace() || slug.Any(char.IsWhiteSpace) || isValid == false)
|
|
{
|
|
return new List<Album>();
|
|
}
|
|
|
|
try
|
|
{
|
|
var existingAlbum = _albumService.FindById(searchGuid.ToString());
|
|
|
|
if (existingAlbum == null)
|
|
{
|
|
return new List<Album> { GetAlbumInfo(searchGuid.ToString()).Item2 };
|
|
}
|
|
|
|
existingAlbum.Artist = _artistService.GetArtist(existingAlbum.ArtistId);
|
|
return new List<Album>{existingAlbum};
|
|
|
|
}
|
|
catch (ArtistNotFoundException)
|
|
{
|
|
return new List<Album>();
|
|
}
|
|
}
|
|
|
|
SetCustomProvider();
|
|
|
|
var httpRequest = _customerRequestBuilder.Create()
|
|
.SetSegment("route", "search")
|
|
.AddQueryParam("type", "album")
|
|
.AddQueryParam("query", title.ToLower().Trim())
|
|
.AddQueryParam("artist", artist.IsNotNullOrWhiteSpace() ? artist.ToLower().Trim() : string.Empty)
|
|
.Build();
|
|
|
|
|
|
|
|
var httpResponse = _httpClient.Get<List<AlbumResource>>(httpRequest);
|
|
|
|
return httpResponse.Resource.SelectList(MapSearchResult);
|
|
}
|
|
catch (HttpException)
|
|
{
|
|
throw new SkyHookException("Search for '{0}' failed. Unable to communicate with LidarrAPI.", title);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Warn(ex, ex.Message);
|
|
throw new SkyHookException("Search for '{0}' failed. Invalid response received from LidarrAPI.", title);
|
|
}
|
|
}
|
|
|
|
private Artist MapSearchResult(ArtistResource resource)
|
|
{
|
|
var artist = _artistService.FindById(resource.Id);
|
|
if (artist == null)
|
|
{
|
|
artist = new Artist();
|
|
artist.Metadata = MapArtistMetadata(resource);
|
|
}
|
|
|
|
return artist;
|
|
}
|
|
|
|
private Album MapSearchResult(AlbumResource resource)
|
|
{
|
|
var album = _albumService.FindById(resource.Id) ?? MapAlbum(resource, null);
|
|
|
|
if (album.Artist == null)
|
|
{
|
|
album.Artist = _artistService.GetArtist(album.ArtistId);
|
|
}
|
|
|
|
return album;
|
|
}
|
|
|
|
private static Album MapAlbum(AlbumResource resource, Dictionary<string, ArtistMetadata> artistDict)
|
|
{
|
|
Album album = new Album();
|
|
album.ForeignAlbumId = resource.Id;
|
|
album.Title = resource.Title;
|
|
album.Overview = resource.Overview;
|
|
album.Disambiguation = resource.Disambiguation;
|
|
album.ReleaseDate = resource.ReleaseDate;
|
|
|
|
if (resource.Images != null)
|
|
{
|
|
album.Images = resource.Images.Select(MapImage).ToList();
|
|
}
|
|
|
|
album.AlbumType = resource.Type;
|
|
album.SecondaryTypes = resource.SecondaryTypes.Select(MapSecondaryTypes).ToList();
|
|
album.Ratings = MapRatings(resource.Rating);
|
|
album.Links = resource.Links?.Select(MapLink).ToList();
|
|
album.CleanTitle = Parser.Parser.CleanArtistName(album.Title);
|
|
|
|
if (resource.Releases != null)
|
|
{
|
|
album.AlbumReleases = resource.Releases.Select(x => MapRelease(x, artistDict)).ToList();
|
|
album.AlbumReleases.Value.OrderByDescending(x => x.TrackCount).First().Monitored = true;
|
|
}
|
|
|
|
album.AnyReleaseOk = true;
|
|
|
|
return album;
|
|
}
|
|
|
|
private static AlbumRelease MapRelease(ReleaseResource resource, Dictionary<string, ArtistMetadata> artistDict)
|
|
{
|
|
AlbumRelease release = new AlbumRelease();
|
|
release.ForeignReleaseId = resource.Id;
|
|
release.Title = resource.Title;
|
|
release.Status = resource.Status;
|
|
release.Label = resource.Label;
|
|
release.Disambiguation = resource.Disambiguation;
|
|
release.Country = resource.Country;
|
|
release.ReleaseDate = resource.ReleaseDate;
|
|
release.TrackCount = resource.TrackCount;
|
|
release.Tracks = resource.Tracks.Select(x => MapTrack(x, artistDict)).ToList();
|
|
release.Media = resource.Media.Select(MapMedium).ToList();
|
|
if (!release.Media.Any())
|
|
{
|
|
foreach(int n in release.Tracks.Value.Select(x => x.MediumNumber).Distinct())
|
|
{
|
|
release.Media.Add(new Medium { Name = "Unknown", Number = n, Format = "Unknown" });
|
|
}
|
|
}
|
|
release.Duration = release.Tracks.Value.Sum(x => x.Duration);
|
|
|
|
return release;
|
|
}
|
|
|
|
private static Medium MapMedium(MediumResource resource)
|
|
{
|
|
Medium medium = new Medium
|
|
{
|
|
Name = resource.Name,
|
|
Number = resource.Position,
|
|
Format = resource.Format
|
|
};
|
|
|
|
return medium;
|
|
}
|
|
|
|
private static Track MapTrack(TrackResource resource, Dictionary<string, ArtistMetadata> artistDict)
|
|
{
|
|
Track track = new Track
|
|
{
|
|
ArtistMetadata = artistDict[resource.ArtistId],
|
|
Title = resource.TrackName,
|
|
ForeignTrackId = resource.Id,
|
|
ForeignRecordingId = resource.RecordingId,
|
|
TrackNumber = resource.TrackNumber,
|
|
AbsoluteTrackNumber = resource.TrackPosition,
|
|
Duration = resource.DurationMs,
|
|
MediumNumber = resource.MediumNumber
|
|
};
|
|
|
|
return track;
|
|
}
|
|
|
|
private static ArtistMetadata MapArtistMetadata(ArtistResource resource)
|
|
{
|
|
|
|
ArtistMetadata artist = new ArtistMetadata();
|
|
|
|
artist.Name = resource.ArtistName;
|
|
artist.ForeignArtistId = resource.Id;
|
|
artist.Genres = resource.Genres;
|
|
artist.Overview = resource.Overview;
|
|
artist.Disambiguation = resource.Disambiguation;
|
|
artist.Type = resource.Type;
|
|
artist.Status = MapArtistStatus(resource.Status);
|
|
artist.Ratings = MapRatings(resource.Rating);
|
|
artist.Images = resource.Images?.Select(MapImage).ToList();
|
|
artist.Links = resource.Links?.Select(MapLink).ToList();
|
|
return artist;
|
|
}
|
|
|
|
private static ArtistStatusType MapArtistStatus(string status)
|
|
{
|
|
if (status == null)
|
|
{
|
|
return ArtistStatusType.Continuing;
|
|
}
|
|
|
|
if (status.Equals("ended", StringComparison.InvariantCultureIgnoreCase))
|
|
{
|
|
return ArtistStatusType.Ended;
|
|
}
|
|
|
|
return ArtistStatusType.Continuing;
|
|
}
|
|
|
|
private static Ratings MapRatings(RatingResource rating)
|
|
{
|
|
if (rating == null)
|
|
{
|
|
return new Ratings();
|
|
}
|
|
|
|
return new Ratings
|
|
{
|
|
Votes = rating.Count,
|
|
Value = rating.Value
|
|
};
|
|
}
|
|
|
|
private static MediaCover.MediaCover MapImage(ImageResource arg)
|
|
{
|
|
return new MediaCover.MediaCover
|
|
{
|
|
Url = arg.Url,
|
|
CoverType = MapCoverType(arg.CoverType)
|
|
};
|
|
}
|
|
|
|
private static Links MapLink(LinkResource arg)
|
|
{
|
|
return new Links
|
|
{
|
|
Url = arg.Target,
|
|
Name = arg.Type
|
|
};
|
|
}
|
|
|
|
private static MediaCoverTypes MapCoverType(string coverType)
|
|
{
|
|
switch (coverType.ToLower())
|
|
{
|
|
case "poster":
|
|
return MediaCoverTypes.Poster;
|
|
case "banner":
|
|
return MediaCoverTypes.Banner;
|
|
case "fanart":
|
|
return MediaCoverTypes.Fanart;
|
|
case "cover":
|
|
return MediaCoverTypes.Cover;
|
|
case "disc":
|
|
return MediaCoverTypes.Disc;
|
|
case "logo":
|
|
return MediaCoverTypes.Logo;
|
|
default:
|
|
return MediaCoverTypes.Unknown;
|
|
}
|
|
}
|
|
|
|
private static SecondaryAlbumType MapSecondaryTypes(string albumType)
|
|
{
|
|
switch (albumType.ToLowerInvariant())
|
|
{
|
|
case "compilation":
|
|
return SecondaryAlbumType.Compilation;
|
|
case "soundtrack":
|
|
return SecondaryAlbumType.Soundtrack;
|
|
case "spokenword":
|
|
return SecondaryAlbumType.Spokenword;
|
|
case "interview":
|
|
return SecondaryAlbumType.Interview;
|
|
case "audiobook":
|
|
return SecondaryAlbumType.Audiobook;
|
|
case "live":
|
|
return SecondaryAlbumType.Live;
|
|
case "remix":
|
|
return SecondaryAlbumType.Remix;
|
|
case "dj-mix":
|
|
return SecondaryAlbumType.DJMix;
|
|
case "mixtape/street":
|
|
return SecondaryAlbumType.Mixtape;
|
|
case "demo":
|
|
return SecondaryAlbumType.Demo;
|
|
default:
|
|
return SecondaryAlbumType.Studio;
|
|
}
|
|
}
|
|
|
|
private void SetCustomProvider()
|
|
{
|
|
if (_configService.MetadataSource.IsNotNullOrWhiteSpace())
|
|
{
|
|
_customerRequestBuilder = new HttpRequestBuilder(_configService.MetadataSource.TrimEnd("/") + "/{route}").CreateFactory();
|
|
}
|
|
else
|
|
{
|
|
_customerRequestBuilder = _requestBuilder;
|
|
}
|
|
}
|
|
}
|
|
}
|