mirror of https://github.com/Radarr/Radarr
63 lines
1.8 KiB
C#
63 lines
1.8 KiB
C#
using System.Collections.Generic;
|
|
using Nancy;
|
|
using NzbDrone.Core.MediaCover;
|
|
using NzbDrone.Core.MetadataSource;
|
|
using System.Linq;
|
|
using NzbDrone.Core.NetImport;
|
|
using NzbDrone.Core.Movies;
|
|
using Radarr.Http;
|
|
using Radarr.Http.Extensions;
|
|
|
|
namespace Radarr.Api.V3.Movies
|
|
{
|
|
public class FetchMovieListModule : RadarrRestModule<MovieResource>
|
|
{
|
|
private readonly IFetchNetImport _fetchNetImport;
|
|
private readonly ISearchForNewMovie _movieSearch;
|
|
|
|
public FetchMovieListModule(IFetchNetImport netImport, ISearchForNewMovie movieSearch)
|
|
: base("/netimport/movies")
|
|
{
|
|
_fetchNetImport = netImport;
|
|
_movieSearch = movieSearch;
|
|
Get("/", x => Search());
|
|
}
|
|
|
|
|
|
private object Search()
|
|
{
|
|
var results = _fetchNetImport.FetchAndFilter((int) Request.Query.listId, false);
|
|
|
|
List<Movie> realResults = new List<Movie>();
|
|
|
|
foreach (var movie in results)
|
|
{
|
|
var mapped = _movieSearch.MapMovieToTmdbMovie(movie);
|
|
|
|
if (mapped != null)
|
|
{
|
|
realResults.Add(mapped);
|
|
}
|
|
}
|
|
|
|
return MapToResource(realResults);
|
|
}
|
|
|
|
|
|
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;
|
|
}
|
|
}
|
|
}
|
|
}
|