using MediaBrowser.Common.Net; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Logging; 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 = @"http://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; public MovieDbSearch(ILogger logger, IJsonSerializer json) { _logger = logger; _json = json; } public Task FindSeriesId(ItemLookupInfo idInfo, CancellationToken cancellationToken) { return FindId(idInfo, "tv", cancellationToken); } public Task FindMovieId(ItemLookupInfo idInfo, CancellationToken cancellationToken) { return FindId(idInfo, "movie", cancellationToken); } public Task FindCollectionId(ItemLookupInfo idInfo, CancellationToken cancellationToken) { return FindId(idInfo, "collection", cancellationToken); } private async Task FindId(ItemLookupInfo idInfo, string searchType, CancellationToken cancellationToken) { var name = idInfo.Name; var year = idInfo.Year; int? yearInName = null; NameParser.ParseName(name, out name, out yearInName); 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 id = await AttemptFindId(name, searchType, year, language, cancellationToken).ConfigureAwait(false); if (id == null) { //try in english if wasn't before if (language != "en") { id = await AttemptFindId(name, searchType, year, "en", cancellationToken).ConfigureAwait(false); } else { // 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)) { id = await AttemptFindId(name, searchType, year, language, cancellationToken).ConfigureAwait(false); if (id == null && language != "en") { //one more time, in english id = await AttemptFindId(name, searchType, year, "en", cancellationToken).ConfigureAwait(false); } } } } return id; } private async Task AttemptFindId(string name, string type, int? year, string language, 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 searchResult = _json.DeserializeFromStream(json); return FindIdOfBestResult(searchResult.results, name, year); } } private string FindIdOfBestResult(List results, string name, int? year) { if (year.HasValue) { // Take the first result from the same year var id = results.Where(i => { // Make sure it has a name if (!string.IsNullOrEmpty(i.title ?? i.name)) { DateTime r; // These dates are always in this exact format if (DateTime.TryParseExact(i.release_date, "yyyy-MM-dd", EnUs, DateTimeStyles.None, out r)) { return r.Year == year.Value; } } return false; }) .Select(i => i.id.ToString(CultureInfo.InvariantCulture)) .FirstOrDefault(); if (!string.IsNullOrEmpty(id)) { return id; } // Take the first result within one year id = results.Where(i => { // Make sure it has a name if (!string.IsNullOrEmpty(i.title ?? i.name)) { DateTime r; // These dates are always in this exact format if (DateTime.TryParseExact(i.release_date, "yyyy-MM-dd", EnUs, DateTimeStyles.None, out r)) { return Math.Abs(r.Year - year.Value) <= 1; } } return false; }) .Select(i => i.id.ToString(CultureInfo.InvariantCulture)) .FirstOrDefault(); if (!string.IsNullOrEmpty(id)) { return id; } } // Just take the first one return results.Where(i => !string.IsNullOrEmpty(i.title ?? i.name)) .Select(i => i.id.ToString(CultureInfo.InvariantCulture)) .FirstOrDefault(); } /// /// Class TmdbMovieSearchResult /// private class TmdbMovieSearchResult { /// /// Gets or sets a value indicating whether this is adult. /// /// true if adult; otherwise, false. public bool adult { get; set; } /// /// Gets or sets the backdrop_path. /// /// The backdrop_path. public string backdrop_path { get; set; } /// /// Gets or sets the id. /// /// The id. public int id { get; set; } /// /// Gets or sets the original_title. /// /// The original_title. public string original_title { get; set; } /// /// Gets or sets the release_date. /// /// The release_date. public string release_date { get; set; } /// /// Gets or sets the poster_path. /// /// The poster_path. public string poster_path { get; set; } /// /// Gets or sets the popularity. /// /// The popularity. public double popularity { get; set; } /// /// Gets or sets the title. /// /// The title. public string title { get; set; } /// /// Gets or sets the vote_average. /// /// The vote_average. public double vote_average { get; set; } /// /// For collection search results /// public string name { get; set; } /// /// Gets or sets the vote_count. /// /// The vote_count. public int vote_count { get; set; } } /// /// Class TmdbMovieSearchResults /// private class TmdbMovieSearchResults { /// /// Gets or sets the page. /// /// The page. public int page { get; set; } /// /// Gets or sets the results. /// /// The results. public List results { get; set; } /// /// Gets or sets the total_pages. /// /// The total_pages. public int total_pages { get; set; } /// /// Gets or sets the total_results. /// /// The total_results. public int total_results { get; set; } } public class TvResult { public string backdrop_path { get; set; } public int id { get; set; } public string original_name { get; set; } public string first_air_date { 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; } } public class ExternalIdLookupResult { public List movie_results { get; set; } public List person_results { get; set; } public List tv_results { get; set; } } } }