mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-08-31 08:34:33 +02:00
416 lines
16 KiB
C#
416 lines
16 KiB
C#
using MediaBrowser.Common.Net;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.Providers;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Logging;
|
|
using MediaBrowser.Model.Providers;
|
|
using MediaBrowser.Model.Serialization;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Globalization;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MediaBrowser.Providers.Movies
|
|
{
|
|
public class MovieDbSearch
|
|
{
|
|
private static readonly CultureInfo EnUs = new CultureInfo("en-US");
|
|
private const string Search3 = @"https://api.themoviedb.org/3/search/{3}?api_key={1}&query={0}&language={2}";
|
|
|
|
internal static string ApiKey = "f6bd687ffa63cd282b6ff2c6877f2669";
|
|
internal static string AcceptHeader = "application/json,image/*";
|
|
|
|
private readonly ILogger _logger;
|
|
private readonly IJsonSerializer _json;
|
|
private readonly ILibraryManager _libraryManager;
|
|
|
|
public MovieDbSearch(ILogger logger, IJsonSerializer json, ILibraryManager libraryManager)
|
|
{
|
|
_logger = logger;
|
|
_json = json;
|
|
_libraryManager = libraryManager;
|
|
}
|
|
|
|
public Task<IEnumerable<RemoteSearchResult>> GetSearchResults(SeriesInfo idInfo, CancellationToken cancellationToken)
|
|
{
|
|
return GetSearchResults(idInfo, "tv", cancellationToken);
|
|
}
|
|
|
|
public Task<IEnumerable<RemoteSearchResult>> GetMovieSearchResults(ItemLookupInfo idInfo, CancellationToken cancellationToken)
|
|
{
|
|
return GetSearchResults(idInfo, "movie", cancellationToken);
|
|
}
|
|
|
|
public Task<IEnumerable<RemoteSearchResult>> GetSearchResults(BoxSetInfo idInfo, CancellationToken cancellationToken)
|
|
{
|
|
return GetSearchResults(idInfo, "collection", cancellationToken);
|
|
}
|
|
|
|
private async Task<IEnumerable<RemoteSearchResult>> GetSearchResults(ItemLookupInfo idInfo, string searchType, CancellationToken cancellationToken)
|
|
{
|
|
var name = idInfo.Name;
|
|
var year = idInfo.Year;
|
|
|
|
var tmdbSettings = await MovieDbProvider.Current.GetTmdbSettings(cancellationToken).ConfigureAwait(false);
|
|
|
|
var tmdbImageUrl = tmdbSettings.images.secure_base_url + "original";
|
|
|
|
if (!string.IsNullOrWhiteSpace(name))
|
|
{
|
|
var parsedName = _libraryManager.ParseName(name);
|
|
var yearInName = parsedName.Year;
|
|
name = parsedName.Name;
|
|
year = year ?? yearInName;
|
|
}
|
|
|
|
_logger.Info("MovieDbProvider: Finding id for item: " + name);
|
|
var language = idInfo.MetadataLanguage.ToLower();
|
|
|
|
//nope - search for it
|
|
//var searchType = item is BoxSet ? "collection" : "movie";
|
|
|
|
var results = await GetSearchResults(name, searchType, year, language, tmdbImageUrl, cancellationToken).ConfigureAwait(false);
|
|
|
|
if (results.Count == 0)
|
|
{
|
|
//try in english if wasn't before
|
|
if (!string.Equals(language, "en", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
results = await GetSearchResults(name, searchType, year, "en", tmdbImageUrl, cancellationToken).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
if (results.Count == 0)
|
|
{
|
|
// try with dot and _ turned to space
|
|
var originalName = name;
|
|
|
|
name = name.Replace(",", " ");
|
|
name = name.Replace(".", " ");
|
|
name = name.Replace("_", " ");
|
|
name = name.Replace("-", " ");
|
|
name = name.Replace("!", " ");
|
|
name = name.Replace("?", " ");
|
|
|
|
name = name.Trim();
|
|
|
|
// Search again if the new name is different
|
|
if (!string.Equals(name, originalName))
|
|
{
|
|
results = await GetSearchResults(name, searchType, year, language, tmdbImageUrl, cancellationToken).ConfigureAwait(false);
|
|
|
|
if (results.Count == 0 && !string.Equals(language, "en", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
//one more time, in english
|
|
results = await GetSearchResults(name, searchType, year, "en", tmdbImageUrl, cancellationToken).ConfigureAwait(false);
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
return results.Where(i =>
|
|
{
|
|
if (year.HasValue && i.ProductionYear.HasValue)
|
|
{
|
|
// Allow one year tolerance
|
|
return Math.Abs(year.Value - i.ProductionYear.Value) <= 1;
|
|
}
|
|
|
|
return true;
|
|
});
|
|
}
|
|
|
|
private async Task<List<RemoteSearchResult>> GetSearchResults(string name, string type, int? year, string language, string baseImageUrl, CancellationToken cancellationToken)
|
|
{
|
|
switch (type)
|
|
{
|
|
case "tv":
|
|
return await GetSearchResultsTv(name, year, language, baseImageUrl, cancellationToken);
|
|
default:
|
|
return await GetSearchResultsGeneric(name, type, year, language, baseImageUrl, cancellationToken);
|
|
}
|
|
}
|
|
|
|
private async Task<List<RemoteSearchResult>> GetSearchResultsGeneric(string name, string type, int? year, string language, string baseImageUrl, CancellationToken cancellationToken)
|
|
{
|
|
|
|
var url3 = string.Format(Search3, WebUtility.UrlEncode(name), ApiKey, language, type);
|
|
|
|
using (var json = await MovieDbProvider.Current.GetMovieDbResponse(new HttpRequestOptions
|
|
{
|
|
Url = url3,
|
|
CancellationToken = cancellationToken,
|
|
AcceptHeader = AcceptHeader
|
|
|
|
}).ConfigureAwait(false))
|
|
{
|
|
var searchResults = _json.DeserializeFromStream<TmdbMovieSearchResults>(json);
|
|
|
|
var results = searchResults.results ?? new List<TmdbMovieSearchResult>();
|
|
|
|
var index = 0;
|
|
var resultTuples = results.Select(result => new Tuple<TmdbMovieSearchResult, int>(result, index++)).ToList();
|
|
|
|
return resultTuples.OrderBy(i => GetSearchResultOrder(i.Item1, year))
|
|
.ThenBy(i => i.Item2)
|
|
.Select(i => i.Item1)
|
|
.Select(i =>
|
|
{
|
|
var remoteResult = new RemoteSearchResult
|
|
{
|
|
SearchProviderName = MovieDbProvider.Current.Name,
|
|
Name = i.title ?? i.name ?? i.original_title,
|
|
ImageUrl = string.IsNullOrWhiteSpace(i.poster_path) ? null : baseImageUrl + i.poster_path
|
|
};
|
|
|
|
if (!string.IsNullOrWhiteSpace(i.release_date))
|
|
{
|
|
DateTime r;
|
|
|
|
// These dates are always in this exact format
|
|
if (DateTime.TryParseExact(i.release_date, "yyyy-MM-dd", EnUs, DateTimeStyles.None, out r))
|
|
{
|
|
remoteResult.PremiereDate = r.ToUniversalTime();
|
|
remoteResult.ProductionYear = remoteResult.PremiereDate.Value.Year;
|
|
}
|
|
}
|
|
|
|
remoteResult.SetProviderId(MetadataProviders.Tmdb, i.id.ToString(EnUs));
|
|
|
|
return remoteResult;
|
|
|
|
})
|
|
.ToList();
|
|
}
|
|
}
|
|
|
|
private async Task<List<RemoteSearchResult>> GetSearchResultsTv(string name, int? year, string language, string baseImageUrl, CancellationToken cancellationToken)
|
|
{
|
|
var url3 = string.Format(Search3, WebUtility.UrlEncode(name), ApiKey, language, "tv");
|
|
|
|
using (var json = await MovieDbProvider.Current.GetMovieDbResponse(new HttpRequestOptions
|
|
{
|
|
Url = url3,
|
|
CancellationToken = cancellationToken,
|
|
AcceptHeader = AcceptHeader
|
|
|
|
}).ConfigureAwait(false))
|
|
{
|
|
var searchResults = _json.DeserializeFromStream<TmdbTvSearchResults>(json);
|
|
|
|
var results = searchResults.results ?? new List<TvResult>();
|
|
|
|
var index = 0;
|
|
var resultTuples = results.Select(result => new Tuple<TvResult, int>(result, index++)).ToList();
|
|
|
|
return resultTuples.OrderBy(i => GetSearchResultOrder(i.Item1, year))
|
|
.ThenBy(i => i.Item2)
|
|
.Select(i => i.Item1)
|
|
.Select(i =>
|
|
{
|
|
var remoteResult = new RemoteSearchResult
|
|
{
|
|
SearchProviderName = MovieDbProvider.Current.Name,
|
|
Name = i.name ?? i.original_name,
|
|
ImageUrl = string.IsNullOrWhiteSpace(i.poster_path) ? null : baseImageUrl + i.poster_path
|
|
};
|
|
|
|
if (!string.IsNullOrWhiteSpace(i.first_air_date))
|
|
{
|
|
DateTime r;
|
|
|
|
// These dates are always in this exact format
|
|
if (DateTime.TryParseExact(i.first_air_date, "yyyy-MM-dd", EnUs, DateTimeStyles.None, out r))
|
|
{
|
|
remoteResult.PremiereDate = r.ToUniversalTime();
|
|
remoteResult.ProductionYear = remoteResult.PremiereDate.Value.Year;
|
|
}
|
|
}
|
|
|
|
remoteResult.SetProviderId(MetadataProviders.Tmdb, i.id.ToString(EnUs));
|
|
|
|
return remoteResult;
|
|
|
|
})
|
|
.ToList();
|
|
}
|
|
}
|
|
|
|
private int GetSearchResultOrder(TmdbMovieSearchResult result, int? year)
|
|
{
|
|
if (year.HasValue)
|
|
{
|
|
DateTime r;
|
|
|
|
// These dates are always in this exact format
|
|
if (DateTime.TryParseExact(result.release_date, "yyyy-MM-dd", EnUs, DateTimeStyles.None, out r))
|
|
{
|
|
// Allow one year tolernace, preserve order from Tmdb
|
|
return Math.Abs(r.Year - year.Value);
|
|
}
|
|
}
|
|
|
|
return int.MaxValue;
|
|
}
|
|
|
|
private int GetSearchResultOrder(TvResult result, int? year)
|
|
{
|
|
if (year.HasValue)
|
|
{
|
|
DateTime r;
|
|
|
|
// These dates are always in this exact format
|
|
if (DateTime.TryParseExact(result.first_air_date, "yyyy-MM-dd", EnUs, DateTimeStyles.None, out r))
|
|
{
|
|
// Allow one year tolernace, preserve order from Tmdb
|
|
return Math.Abs(r.Year - year.Value);
|
|
}
|
|
}
|
|
|
|
return int.MaxValue;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Class TmdbMovieSearchResult
|
|
/// </summary>
|
|
public class TmdbMovieSearchResult
|
|
{
|
|
/// <summary>
|
|
/// Gets or sets a value indicating whether this <see cref="TmdbMovieSearchResult" /> is adult.
|
|
/// </summary>
|
|
/// <value><c>true</c> if adult; otherwise, <c>false</c>.</value>
|
|
public bool adult { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the backdrop_path.
|
|
/// </summary>
|
|
/// <value>The backdrop_path.</value>
|
|
public string backdrop_path { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the id.
|
|
/// </summary>
|
|
/// <value>The id.</value>
|
|
public int id { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the original_title.
|
|
/// </summary>
|
|
/// <value>The original_title.</value>
|
|
public string original_title { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the original_name.
|
|
/// </summary>
|
|
/// <value>The original_name.</value>
|
|
public string original_name { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the release_date.
|
|
/// </summary>
|
|
/// <value>The release_date.</value>
|
|
public string release_date { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the poster_path.
|
|
/// </summary>
|
|
/// <value>The poster_path.</value>
|
|
public string poster_path { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the popularity.
|
|
/// </summary>
|
|
/// <value>The popularity.</value>
|
|
public double popularity { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the title.
|
|
/// </summary>
|
|
/// <value>The title.</value>
|
|
public string title { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the vote_average.
|
|
/// </summary>
|
|
/// <value>The vote_average.</value>
|
|
public double vote_average { get; set; }
|
|
/// <summary>
|
|
/// For collection search results
|
|
/// </summary>
|
|
public string name { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the vote_count.
|
|
/// </summary>
|
|
/// <value>The vote_count.</value>
|
|
public int vote_count { get; set; }
|
|
}
|
|
|
|
/// <summary>
|
|
/// Class TmdbMovieSearchResults
|
|
/// </summary>
|
|
private class TmdbMovieSearchResults
|
|
{
|
|
/// <summary>
|
|
/// Gets or sets the page.
|
|
/// </summary>
|
|
/// <value>The page.</value>
|
|
public int page { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the results.
|
|
/// </summary>
|
|
/// <value>The results.</value>
|
|
public List<TmdbMovieSearchResult> results { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the total_pages.
|
|
/// </summary>
|
|
/// <value>The total_pages.</value>
|
|
public int total_pages { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the total_results.
|
|
/// </summary>
|
|
/// <value>The total_results.</value>
|
|
public int total_results { get; set; }
|
|
}
|
|
|
|
public class TvResult
|
|
{
|
|
public string backdrop_path { get; set; }
|
|
public string first_air_date { get; set; }
|
|
public int id { get; set; }
|
|
public string original_name { get; set; }
|
|
public string poster_path { get; set; }
|
|
public double popularity { get; set; }
|
|
public string name { get; set; }
|
|
public double vote_average { get; set; }
|
|
public int vote_count { get; set; }
|
|
}
|
|
|
|
/// <summary>
|
|
/// Class TmdbTvSearchResults
|
|
/// </summary>
|
|
private class TmdbTvSearchResults
|
|
{
|
|
/// <summary>
|
|
/// Gets or sets the page.
|
|
/// </summary>
|
|
/// <value>The page.</value>
|
|
public int page { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the results.
|
|
/// </summary>
|
|
/// <value>The results.</value>
|
|
public List<TvResult> results { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the total_pages.
|
|
/// </summary>
|
|
/// <value>The total_pages.</value>
|
|
public int total_pages { get; set; }
|
|
/// <summary>
|
|
/// Gets or sets the total_results.
|
|
/// </summary>
|
|
/// <value>The total_results.</value>
|
|
public int total_results { get; set; }
|
|
}
|
|
|
|
public class ExternalIdLookupResult
|
|
{
|
|
public List<object> movie_results { get; set; }
|
|
public List<object> person_results { get; set; }
|
|
public List<TvResult> tv_results { get; set; }
|
|
}
|
|
}
|
|
}
|