jellyfin/MediaBrowser.Providers/Omdb/OmdbItemProvider.cs

289 lines
11 KiB
C#
Raw Normal View History

2014-02-03 20:51:28 +00:00
using MediaBrowser.Common.Net;
2014-09-22 21:56:54 +00:00
using MediaBrowser.Controller.Channels;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Movies;
2014-02-03 20:51:28 +00:00
using MediaBrowser.Controller.Entities.TV;
2014-11-18 02:48:22 +00:00
using MediaBrowser.Controller.Library;
2014-02-03 20:51:28 +00:00
using MediaBrowser.Controller.Providers;
2014-09-22 21:56:54 +00:00
using MediaBrowser.Model.Channels;
2014-02-15 16:36:09 +00:00
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Logging;
using MediaBrowser.Model.Providers;
2014-02-03 20:51:28 +00:00
using MediaBrowser.Model.Serialization;
2014-02-15 16:36:09 +00:00
using MediaBrowser.Providers.Movies;
using MediaBrowser.Providers.TV;
using System;
using System.Collections.Generic;
2015-01-27 06:50:40 +00:00
using System.Globalization;
using System.Net;
2014-02-03 20:51:28 +00:00
using System.Threading;
using System.Threading.Tasks;
namespace MediaBrowser.Providers.Omdb
{
2014-02-15 16:36:09 +00:00
public class OmdbItemProvider : IRemoteMetadataProvider<Series, SeriesInfo>,
2014-12-03 03:13:03 +00:00
IRemoteMetadataProvider<Movie, MovieInfo>, IRemoteMetadataProvider<ChannelVideoItem, ChannelItemLookupInfo>
2014-02-03 20:51:28 +00:00
{
private readonly IJsonSerializer _jsonSerializer;
private readonly IHttpClient _httpClient;
2014-02-15 16:36:09 +00:00
private readonly ILogger _logger;
2014-11-18 02:48:22 +00:00
private readonly ILibraryManager _libraryManager;
2014-02-03 20:51:28 +00:00
2014-11-18 02:48:22 +00:00
public OmdbItemProvider(IJsonSerializer jsonSerializer, IHttpClient httpClient, ILogger logger, ILibraryManager libraryManager)
2014-02-03 20:51:28 +00:00
{
_jsonSerializer = jsonSerializer;
_httpClient = httpClient;
2014-02-15 16:36:09 +00:00
_logger = logger;
2014-11-18 02:48:22 +00:00
_libraryManager = libraryManager;
2014-02-03 20:51:28 +00:00
}
2015-01-27 06:50:40 +00:00
public Task<IEnumerable<RemoteSearchResult>> GetSearchResults(SeriesInfo searchInfo, CancellationToken cancellationToken)
{
2015-01-27 06:50:40 +00:00
return GetSearchResults(searchInfo, "series", cancellationToken);
}
2015-01-27 06:50:40 +00:00
public Task<IEnumerable<RemoteSearchResult>> GetSearchResults(MovieInfo searchInfo, CancellationToken cancellationToken)
{
2015-01-27 06:50:40 +00:00
return GetSearchResults(searchInfo, "movie", cancellationToken);
}
2015-01-27 06:50:40 +00:00
public async Task<IEnumerable<RemoteSearchResult>> GetSearchResults(ItemLookupInfo searchInfo, string type, CancellationToken cancellationToken)
{
2015-01-27 06:50:40 +00:00
var list = new List<RemoteSearchResult>();
var imdbId = searchInfo.GetProviderId(MetadataProviders.Imdb);
if (!string.IsNullOrWhiteSpace(imdbId))
{
return list;
}
var url = "http://www.omdbapi.com/?plot=short&r=json";
var name = searchInfo.Name;
var year = searchInfo.Year;
2015-01-28 18:10:26 +00:00
var parsedName = _libraryManager.ParseName(name);
var yearInName = parsedName.Year;
name = parsedName.Name;
year = year ?? yearInName;
2015-01-27 06:50:40 +00:00
if (year.HasValue)
{
url += "&y=" + year.Value.ToString(CultureInfo.InvariantCulture);
}
url += "&t=" + WebUtility.UrlEncode(name);
url += "&type=" + type;
using (var stream = await _httpClient.Get(new HttpRequestOptions
{
Url = url,
ResourcePool = OmdbProvider.ResourcePool,
CancellationToken = cancellationToken,
CacheMode = CacheMode.Unconditional,
CacheLength = TimeSpan.FromDays(7)
}).ConfigureAwait(false))
{
var result = _jsonSerializer.DeserializeFromStream<SearchResult>(stream);
if (string.Equals(result.Response, "true", StringComparison.OrdinalIgnoreCase))
{
var item = new RemoteSearchResult();
item.SearchProviderName = Name;
item.Name = result.Title;
item.SetProviderId(MetadataProviders.Imdb, result.imdbID);
int parsedYear;
if (int.TryParse(result.Year, NumberStyles.Any, CultureInfo.InvariantCulture, out parsedYear))
{
item.ProductionYear = parsedYear;
}
if (!string.IsNullOrWhiteSpace(result.Poster) && !string.Equals(result.Poster, "N/A", StringComparison.OrdinalIgnoreCase))
{
item.ImageUrl = result.Poster;
}
list.Add(item);
}
}
return list;
}
2014-09-22 21:56:54 +00:00
public Task<MetadataResult<ChannelVideoItem>> GetMetadata(ChannelItemLookupInfo info, CancellationToken cancellationToken)
{
2014-09-28 15:27:26 +00:00
if (info.ContentType != ChannelMediaContentType.MovieExtra || info.ExtraType != ExtraType.Trailer)
2014-09-22 21:56:54 +00:00
{
return Task.FromResult(new MetadataResult<ChannelVideoItem>());
}
return GetMovieResult<ChannelVideoItem>(info, cancellationToken);
}
2015-01-27 06:50:40 +00:00
public Task<IEnumerable<RemoteSearchResult>> GetSearchResults(ChannelItemLookupInfo searchInfo, CancellationToken cancellationToken)
2014-09-22 21:56:54 +00:00
{
2015-01-27 06:50:40 +00:00
if (searchInfo.ContentType != ChannelMediaContentType.MovieExtra || searchInfo.ExtraType != ExtraType.Trailer)
{
return Task.FromResult<IEnumerable<RemoteSearchResult>>(new List<RemoteSearchResult>());
}
return GetSearchResults(searchInfo, "movie", cancellationToken);
2014-09-22 21:56:54 +00:00
}
public string Name
{
2014-02-20 21:10:26 +00:00
get { return "The Open Movie Database"; }
}
2014-02-15 16:36:09 +00:00
public async Task<MetadataResult<Series>> GetMetadata(SeriesInfo info, CancellationToken cancellationToken)
2014-02-03 20:51:28 +00:00
{
2014-02-15 16:36:09 +00:00
var result = new MetadataResult<Series>
{
Item = new Series()
};
2014-02-24 03:27:13 +00:00
var searchResult = await GetSeriesImdbId(info, cancellationToken).ConfigureAwait(false);
2014-07-12 14:05:51 +00:00
result.Item.Name = searchResult.Item4;
2014-02-24 03:27:13 +00:00
2014-07-12 14:05:51 +00:00
if (!string.IsNullOrEmpty(searchResult.Item1))
2014-02-15 16:36:09 +00:00
{
2014-07-12 14:05:51 +00:00
result.Item.SetProviderId(MetadataProviders.Imdb, searchResult.Item1);
}
2014-02-15 16:36:09 +00:00
2014-07-12 14:05:51 +00:00
if (!string.IsNullOrEmpty(searchResult.Item2))
{
result.Item.SetProviderId(MetadataProviders.Tmdb, searchResult.Item2);
2014-02-15 16:36:09 +00:00
}
2014-07-12 14:05:51 +00:00
if (!string.IsNullOrEmpty(searchResult.Item3))
{
result.Item.SetProviderId(MetadataProviders.Tvdb, searchResult.Item3);
}
var imdbId = result.Item.GetProviderId(MetadataProviders.Imdb);
if (!string.IsNullOrEmpty(info.GetProviderId(MetadataProviders.Imdb)))
2014-02-15 16:36:09 +00:00
{
result.Item.SetProviderId(MetadataProviders.Imdb, imdbId);
result.HasMetadata = true;
await new OmdbProvider(_jsonSerializer, _httpClient).Fetch(result.Item, imdbId, cancellationToken)
.ConfigureAwait(false);
}
return result;
2014-02-03 20:51:28 +00:00
}
2014-02-15 16:36:09 +00:00
public Task<MetadataResult<Movie>> GetMetadata(MovieInfo info, CancellationToken cancellationToken)
2014-02-03 20:51:28 +00:00
{
2014-02-15 16:36:09 +00:00
return GetMovieResult<Movie>(info, cancellationToken);
}
2014-02-15 16:36:09 +00:00
private async Task<MetadataResult<T>> GetMovieResult<T>(ItemLookupInfo info, CancellationToken cancellationToken)
where T : Video, new()
{
var result = new MetadataResult<T>
{
Item = new T()
};
var imdbId = info.GetProviderId(MetadataProviders.Imdb);
2014-02-24 03:27:13 +00:00
var searchResult = await GetMovieImdbId(info, cancellationToken).ConfigureAwait(false);
result.Item.Name = searchResult.Item3;
2014-02-15 16:36:09 +00:00
if (string.IsNullOrEmpty(imdbId))
{
imdbId = searchResult.Item1;
if (!string.IsNullOrEmpty(searchResult.Item2))
{
result.Item.SetProviderId(MetadataProviders.Tmdb, searchResult.Item2);
}
}
if (!string.IsNullOrEmpty(imdbId))
{
result.Item.SetProviderId(MetadataProviders.Imdb, imdbId);
result.HasMetadata = true;
await new OmdbProvider(_jsonSerializer, _httpClient).Fetch(result.Item, imdbId, cancellationToken)
.ConfigureAwait(false);
}
return result;
}
2014-02-22 21:05:00 +00:00
private async Task<Tuple<string, string, string>> GetMovieImdbId(ItemLookupInfo info, CancellationToken cancellationToken)
2014-02-15 16:36:09 +00:00
{
2014-11-18 02:48:22 +00:00
var result = await new GenericMovieDbInfo<Movie>(_logger, _jsonSerializer, _libraryManager).GetMetadata(info, cancellationToken)
2014-02-15 16:36:09 +00:00
.ConfigureAwait(false);
var imdb = result.HasMetadata ? result.Item.GetProviderId(MetadataProviders.Imdb) : null;
var tmdb = result.HasMetadata ? result.Item.GetProviderId(MetadataProviders.Tmdb) : null;
2014-02-22 21:05:00 +00:00
var name = result.HasMetadata ? result.Item.Name : null;
2014-02-15 16:36:09 +00:00
2014-02-22 21:05:00 +00:00
return new Tuple<string, string, string>(imdb, tmdb, name);
2014-02-15 16:36:09 +00:00
}
2014-07-12 14:05:51 +00:00
private async Task<Tuple<string, string, string, string>> GetSeriesImdbId(SeriesInfo info, CancellationToken cancellationToken)
2014-02-15 16:36:09 +00:00
{
2014-07-12 14:05:51 +00:00
//var result = await TvdbSeriesProvider.Current.GetMetadata(info, cancellationToken)
// .ConfigureAwait(false);
//var imdb = result.HasMetadata ? result.Item.GetProviderId(MetadataProviders.Imdb) : null;
//var tvdb = result.HasMetadata ? result.Item.GetProviderId(MetadataProviders.Tvdb) : null;
//var name = result.HasMetadata ? result.Item.Name : null;
//return new Tuple<string, string, string>(imdb, tvdb, name);
var result = await MovieDbSeriesProvider.Current.GetMetadata(info, cancellationToken)
.ConfigureAwait(false);
2014-02-15 16:36:09 +00:00
var imdb = result.HasMetadata ? result.Item.GetProviderId(MetadataProviders.Imdb) : null;
2014-07-12 14:05:51 +00:00
var tmdb = result.HasMetadata ? result.Item.GetProviderId(MetadataProviders.Tmdb) : null;
2014-02-15 16:36:09 +00:00
var tvdb = result.HasMetadata ? result.Item.GetProviderId(MetadataProviders.Tvdb) : null;
2014-02-22 21:05:00 +00:00
var name = result.HasMetadata ? result.Item.Name : null;
2014-02-15 16:36:09 +00:00
2014-07-12 14:05:51 +00:00
return new Tuple<string, string, string, string>(imdb, tmdb, tvdb, name);
2014-02-03 20:51:28 +00:00
}
public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
{
2015-01-27 06:50:40 +00:00
return _httpClient.GetResponse(new HttpRequestOptions
{
CancellationToken = cancellationToken,
Url = url,
ResourcePool = OmdbProvider.ResourcePool
});
}
class SearchResult
{
public string Title { get; set; }
public string Year { get; set; }
public string Rated { get; set; }
public string Released { get; set; }
public string Runtime { get; set; }
public string Genre { get; set; }
public string Director { get; set; }
public string Writer { get; set; }
public string Actors { get; set; }
public string Plot { get; set; }
public string Language { get; set; }
public string Country { get; set; }
public string Awards { get; set; }
public string Poster { get; set; }
public string Metascore { get; set; }
public string imdbRating { get; set; }
public string imdbVotes { get; set; }
public string imdbID { get; set; }
public string Type { get; set; }
public string Response { get; set; }
}
2014-02-03 20:51:28 +00:00
}
}