mirror of https://github.com/Radarr/Radarr
65 lines
1.9 KiB
C#
65 lines
1.9 KiB
C#
using System.Collections.Generic;
|
|
using Nancy;
|
|
using Radarr.Http.Extensions;
|
|
using NzbDrone.Core.Movies;
|
|
using NzbDrone.Core.MediaCover;
|
|
using NzbDrone.Core.MetadataSource;
|
|
using System.Linq;
|
|
using System;
|
|
using Radarr.Http;
|
|
using NzbDrone.Core.NetImport;
|
|
using Radarr.Api.V2.NetImport;
|
|
|
|
namespace Radarr.Api.V2.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<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;
|
|
}
|
|
}
|
|
}
|
|
}
|