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.
62 lines
1.9 KiB
62 lines
1.9 KiB
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NzbDrone.Api.NetImport;
|
|
using NzbDrone.Core.MediaCover;
|
|
using NzbDrone.Core.MetadataSource;
|
|
using NzbDrone.Core.NetImport;
|
|
using Radarr.Http;
|
|
|
|
namespace NzbDrone.Api.Movies
|
|
{
|
|
public class MovieDiscoverModule : RadarrRestModule<MovieResource>
|
|
{
|
|
private readonly IDiscoverNewMovies _searchProxy;
|
|
private readonly INetImportFactory _netImportFactory;
|
|
|
|
public MovieDiscoverModule(IDiscoverNewMovies searchProxy, INetImportFactory netImportFactory)
|
|
: base("/movies/discover")
|
|
{
|
|
_searchProxy = searchProxy;
|
|
_netImportFactory = netImportFactory;
|
|
Get("/lists", x => GetLists());
|
|
Get("/{action?recommendations}", x => Search(x.action));
|
|
}
|
|
|
|
private object Search(string action)
|
|
{
|
|
var imdbResults = _searchProxy.DiscoverNewMovies(action);
|
|
return MapToResource(imdbResults);
|
|
}
|
|
|
|
private object GetLists()
|
|
{
|
|
var lists = _netImportFactory.Discoverable();
|
|
|
|
return lists.Select(definition =>
|
|
{
|
|
var resource = new NetImportResource();
|
|
resource.Id = definition.Definition.Id;
|
|
|
|
resource.Name = definition.Definition.Name;
|
|
|
|
return resource;
|
|
});
|
|
}
|
|
|
|
private static IEnumerable<MovieResource> MapToResource(IEnumerable<Core.Movies.Movie> movies)
|
|
{
|
|
foreach (var currentSeries in movies)
|
|
{
|
|
var resource = currentSeries.ToResource();
|
|
var poster = currentSeries.Images.FirstOrDefault(c => c.CoverType == MediaCoverTypes.Poster);
|
|
if (poster != null)
|
|
{
|
|
resource.RemotePoster = poster.Url;
|
|
}
|
|
|
|
yield return resource;
|
|
}
|
|
}
|
|
}
|
|
}
|