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.
79 lines
2.5 KiB
79 lines
2.5 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Lidarr.Api.V1.Albums;
|
|
using Lidarr.Api.V1.Artist;
|
|
using Lidarr.Http;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using NzbDrone.Core.MediaCover;
|
|
using NzbDrone.Core.MetadataSource;
|
|
using NzbDrone.Core.Organizer;
|
|
|
|
namespace Lidarr.Api.V1.Search
|
|
{
|
|
[V1ApiController]
|
|
public class SearchController : Controller
|
|
{
|
|
private readonly ISearchForNewEntity _searchProxy;
|
|
private readonly IBuildFileNames _fileNameBuilder;
|
|
|
|
public SearchController(ISearchForNewEntity searchProxy, IBuildFileNames fileNameBuilder)
|
|
{
|
|
_searchProxy = searchProxy;
|
|
_fileNameBuilder = fileNameBuilder;
|
|
}
|
|
|
|
[HttpGet]
|
|
public object Search([FromQuery] string term)
|
|
{
|
|
var searchResults = _searchProxy.SearchForNewEntity(term);
|
|
return MapToResource(searchResults).ToList();
|
|
}
|
|
|
|
private IEnumerable<SearchResource> MapToResource(IEnumerable<object> results)
|
|
{
|
|
var id = 1;
|
|
foreach (var result in results)
|
|
{
|
|
var resource = new SearchResource();
|
|
resource.Id = id++;
|
|
|
|
if (result is NzbDrone.Core.Music.Artist artist)
|
|
{
|
|
resource.Artist = artist.ToResource();
|
|
resource.ForeignId = artist.ForeignArtistId;
|
|
|
|
var poster = artist.Metadata.Value.Images.FirstOrDefault(c => c.CoverType == MediaCoverTypes.Poster);
|
|
|
|
if (poster != null)
|
|
{
|
|
resource.Artist.RemotePoster = poster.Url;
|
|
}
|
|
|
|
resource.Artist.Folder = _fileNameBuilder.GetArtistFolder(artist);
|
|
}
|
|
else if (result is NzbDrone.Core.Music.Album album)
|
|
{
|
|
resource.Album = album.ToResource();
|
|
resource.ForeignId = album.ForeignAlbumId;
|
|
|
|
var cover = album.Images.FirstOrDefault(c => c.CoverType == MediaCoverTypes.Cover);
|
|
|
|
if (cover != null)
|
|
{
|
|
resource.Album.RemoteCover = cover.Url;
|
|
}
|
|
|
|
resource.Album.Artist.Folder = _fileNameBuilder.GetArtistFolder(album.Artist);
|
|
}
|
|
else
|
|
{
|
|
throw new NotImplementedException("Bad response from search all proxy");
|
|
}
|
|
|
|
yield return resource;
|
|
}
|
|
}
|
|
}
|
|
}
|