using MediaBrowser.Common.Configuration; using MediaBrowser.Common.IO; using MediaBrowser.Common.Net; using MediaBrowser.Controller.Configuration; using MediaBrowser.Controller.Entities.Movies; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using MediaBrowser.Model.Serialization; using MediaBrowser.Providers.Movies; using System; using System.Collections.Generic; using System.Globalization; using System.IO; using System.Threading; using System.Threading.Tasks; namespace MediaBrowser.Providers.BoxSets { public class MovieDbBoxSetProvider : IRemoteMetadataProvider { private readonly CultureInfo _enUs = new CultureInfo("en-US"); private const string GetCollectionInfo3 = @"http://api.themoviedb.org/3/collection/{0}?api_key={1}&append_to_response=images"; internal static MovieDbBoxSetProvider Current; private readonly ILogger _logger; private readonly IJsonSerializer _json; private readonly IServerConfigurationManager _config; private readonly IFileSystem _fileSystem; public MovieDbBoxSetProvider(ILogger logger, IJsonSerializer json, IServerConfigurationManager config, IFileSystem fileSystem) { _logger = logger; _json = json; _config = config; _fileSystem = fileSystem; Current = this; } public async Task> GetMetadata(ItemId id, CancellationToken cancellationToken) { var tmdbId = id.GetProviderId(MetadataProviders.Tmdb); // We don't already have an Id, need to fetch it if (string.IsNullOrEmpty(tmdbId)) { tmdbId = await GetTmdbId(id, cancellationToken).ConfigureAwait(false); } var result = new MetadataResult(); if (!string.IsNullOrEmpty(tmdbId)) { var mainResult = await GetMovieDbResult(tmdbId, id.MetadataLanguage, cancellationToken).ConfigureAwait(false); if (mainResult != null) { result.HasMetadata = true; result.Item = GetItem(mainResult); } } return result; } internal async Task GetMovieDbResult(string tmdbId, string language, CancellationToken cancellationToken) { if (string.IsNullOrEmpty(tmdbId)) { throw new ArgumentNullException("tmdbId"); } await EnsureInfo(tmdbId, language, cancellationToken).ConfigureAwait(false); var dataFilePath = GetDataFilePath(_config.ApplicationPaths, tmdbId, language); if (!string.IsNullOrEmpty(dataFilePath)) { return _json.DeserializeFromFile(dataFilePath); } return null; } private BoxSet GetItem(RootObject obj) { var item = new BoxSet(); item.Name = obj.name; item.Overview = obj.overview; item.SetProviderId(MetadataProviders.Tmdb, obj.id.ToString(_enUs)); return item; } private async Task DownloadInfo(string tmdbId, string preferredMetadataLanguage, CancellationToken cancellationToken) { var mainResult = await FetchMainResult(tmdbId, preferredMetadataLanguage, cancellationToken).ConfigureAwait(false); if (mainResult == null) return; var dataFilePath = GetDataFilePath(_config.ApplicationPaths, tmdbId, preferredMetadataLanguage); Directory.CreateDirectory(Path.GetDirectoryName(dataFilePath)); _json.SerializeToFile(mainResult, dataFilePath); } private async Task FetchMainResult(string id, string language, CancellationToken cancellationToken) { var url = string.Format(GetCollectionInfo3, id, MovieDbSearch.ApiKey); // Get images in english and with no language url += "&include_image_language=en,null"; if (!string.IsNullOrEmpty(language)) { // If preferred language isn't english, get those images too if (!string.Equals(language, "en", StringComparison.OrdinalIgnoreCase)) { url += string.Format(",{0}", language); } url += string.Format("&language={0}", language); } cancellationToken.ThrowIfCancellationRequested(); RootObject mainResult = null; using (var json = await MovieDbProvider.Current.GetMovieDbResponse(new HttpRequestOptions { Url = url, CancellationToken = cancellationToken, AcceptHeader = MovieDbSearch.AcceptHeader }).ConfigureAwait(false)) { mainResult = _json.DeserializeFromStream(json); } cancellationToken.ThrowIfCancellationRequested(); if (mainResult != null && string.IsNullOrEmpty(mainResult.overview)) { if (!string.IsNullOrEmpty(language) && !string.Equals(language, "en", StringComparison.OrdinalIgnoreCase)) { url = string.Format(GetCollectionInfo3, id, MovieDbSearch.ApiKey) + "&include_image_language=en,null&language=en"; using (var json = await MovieDbProvider.Current.GetMovieDbResponse(new HttpRequestOptions { Url = url, CancellationToken = cancellationToken, AcceptHeader = MovieDbSearch.AcceptHeader }).ConfigureAwait(false)) { mainResult = _json.DeserializeFromStream(json); } if (String.IsNullOrEmpty(mainResult.overview)) { _logger.Error("Unable to find information for (id:" + id + ")"); return null; } } } return mainResult; } internal Task EnsureInfo(string tmdbId, string preferredMetadataLanguage, CancellationToken cancellationToken) { var path = GetDataFilePath(_config.ApplicationPaths, tmdbId, preferredMetadataLanguage); var fileInfo = _fileSystem.GetFileSystemInfo(path); if (fileInfo.Exists) { // If it's recent or automatic updates are enabled, don't re-download if ((DateTime.UtcNow - _fileSystem.GetLastWriteTimeUtc(fileInfo)).TotalDays <= 7) { return Task.FromResult(true); } } return DownloadInfo(tmdbId, preferredMetadataLanguage, cancellationToken); } private Task GetTmdbId(ItemId id, CancellationToken cancellationToken) { return new MovieDbSearch(_logger, _json).FindCollectionId(id, cancellationToken); } public string Name { get { return "TheMovieDb"; } } private static string GetDataFilePath(IApplicationPaths appPaths, string tmdbId, string preferredLanguage) { var path = GetDataPath(appPaths, tmdbId); var filename = string.Format("all-{0}.json", preferredLanguage ?? string.Empty); return Path.Combine(path, filename); } private static string GetDataPath(IApplicationPaths appPaths, string tmdbId) { var dataPath = GetCollectionsDataPath(appPaths); return Path.Combine(dataPath, tmdbId); } private static string GetCollectionsDataPath(IApplicationPaths appPaths) { var dataPath = Path.Combine(appPaths.DataPath, "tmdb-collections"); return dataPath; } internal class Part { public string title { get; set; } public int id { get; set; } public string release_date { get; set; } public string poster_path { get; set; } public string backdrop_path { get; set; } } internal class Backdrop { public double aspect_ratio { get; set; } public string file_path { get; set; } public int height { get; set; } public string iso_639_1 { get; set; } public double vote_average { get; set; } public int vote_count { get; set; } public int width { get; set; } } internal class Poster { public double aspect_ratio { get; set; } public string file_path { get; set; } public int height { get; set; } public string iso_639_1 { get; set; } public double vote_average { get; set; } public int vote_count { get; set; } public int width { get; set; } } internal class Images { public List backdrops { get; set; } public List posters { get; set; } } internal class RootObject { public int id { get; set; } public string name { get; set; } public string overview { get; set; } public string poster_path { get; set; } public string backdrop_path { get; set; } public List parts { get; set; } public Images images { get; set; } } } }