1 files changed, 255 insertions, 0 deletions
diff --git a/MediaBrowser.Providers/BoxSets/MovieDbBoxSetProvider.cs b/MediaBrowser.Providers/BoxSets/MovieDbBoxSetProvider.cs
new file mode 100644
index 000000000..2fe18fafa
--- /dev/null
+++ b/MediaBrowser.Providers/BoxSets/MovieDbBoxSetProvider.cs
@@ -0,0 +1,255 @@
+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<BoxSet>
+    {
+        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"; 
+        
+        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;
+        }
+
+        public async Task<MetadataResult<BoxSet>> 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<BoxSet>();
+
+            if (!string.IsNullOrEmpty(tmdbId))
+            {
+                await EnsureInfo(tmdbId, id.MetadataLanguage, cancellationToken).ConfigureAwait(false);
+
+                var dataFilePath = GetDataFilePath(_config.ApplicationPaths, tmdbId, id.MetadataLanguage);
+
+                if (!string.IsNullOrEmpty(dataFilePath))
+                {
+                    var mainResult = _json.DeserializeFromFile<RootObject>(dataFilePath);
+
+                    result.HasMetadata = true;
+                    result.Item = GetItem(mainResult);
+                }
+                
+            }
+
+            return result;
+        }
+
+        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<RootObject> 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<RootObject>(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<RootObject>(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<string> 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 object 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 object vote_average { get; set; }
+            public int vote_count { get; set; }
+            public int width { get; set; }
+        }
+
+        internal class Images
+        {
+            public List<Backdrop> backdrops { get; set; }
+            public List<Poster> 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<Part> parts { get; set; }
+            public Images images { get; set; }
+        }
+    }
+}