TmdbSeasonProvider.cs 9.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241
  1. #pragma warning disable CS1591
  2. using System;
  3. using System.Collections.Generic;
  4. using System.Globalization;
  5. using System.IO;
  6. using System.Net;
  7. using System.Net.Http;
  8. using System.Net.Http.Headers;
  9. using System.Threading;
  10. using System.Threading.Tasks;
  11. using MediaBrowser.Common.Net;
  12. using MediaBrowser.Controller.Configuration;
  13. using MediaBrowser.Controller.Providers;
  14. using MediaBrowser.Model.Entities;
  15. using MediaBrowser.Model.Globalization;
  16. using MediaBrowser.Model.IO;
  17. using MediaBrowser.Model.Net;
  18. using MediaBrowser.Model.Providers;
  19. using MediaBrowser.Model.Serialization;
  20. using MediaBrowser.Providers.Plugins.Tmdb.Models.TV;
  21. using MediaBrowser.Providers.Plugins.Tmdb.Movies;
  22. using Microsoft.Extensions.Logging;
  23. using Season = MediaBrowser.Controller.Entities.TV.Season;
  24. namespace MediaBrowser.Providers.Plugins.Tmdb.TV
  25. {
  26. public class TmdbSeasonProvider : IRemoteMetadataProvider<Season, SeasonInfo>
  27. {
  28. private const string GetTvInfo3 = TmdbUtils.BaseTmdbApiUrl + @"3/tv/{0}/season/{1}?api_key={2}&append_to_response=images,keywords,external_ids,credits,videos";
  29. private readonly IHttpClientFactory _httpClientFactory;
  30. private readonly IServerConfigurationManager _configurationManager;
  31. private readonly IJsonSerializer _jsonSerializer;
  32. private readonly IFileSystem _fileSystem;
  33. private readonly ILogger<TmdbSeasonProvider> _logger;
  34. internal static TmdbSeasonProvider Current { get; private set; }
  35. public TmdbSeasonProvider(
  36. IHttpClientFactory httpClientFactory,
  37. IServerConfigurationManager configurationManager,
  38. IFileSystem fileSystem,
  39. IJsonSerializer jsonSerializer,
  40. ILogger<TmdbSeasonProvider> logger)
  41. {
  42. _httpClientFactory = httpClientFactory;
  43. _configurationManager = configurationManager;
  44. _fileSystem = fileSystem;
  45. _jsonSerializer = jsonSerializer;
  46. _logger = logger;
  47. Current = this;
  48. }
  49. public string Name => TmdbUtils.ProviderName;
  50. public async Task<MetadataResult<Season>> GetMetadata(SeasonInfo info, CancellationToken cancellationToken)
  51. {
  52. var result = new MetadataResult<Season>();
  53. info.SeriesProviderIds.TryGetValue(MetadataProvider.Tmdb.ToString(), out string seriesTmdbId);
  54. var seasonNumber = info.IndexNumber;
  55. if (!string.IsNullOrWhiteSpace(seriesTmdbId) && seasonNumber.HasValue)
  56. {
  57. try
  58. {
  59. var seasonInfo = await GetSeasonInfo(seriesTmdbId, seasonNumber.Value, info.MetadataLanguage, cancellationToken)
  60. .ConfigureAwait(false);
  61. result.HasMetadata = true;
  62. result.Item = new Season();
  63. // Don't use moviedb season names for now until if/when we have field-level configuration
  64. // result.Item.Name = seasonInfo.name;
  65. result.Item.Name = info.Name;
  66. result.Item.IndexNumber = seasonNumber;
  67. result.Item.Overview = seasonInfo.Overview;
  68. if (seasonInfo.External_Ids != null && seasonInfo.External_Ids.Tvdb_Id > 0)
  69. {
  70. result.Item.SetProviderId(MetadataProvider.Tvdb, seasonInfo.External_Ids.Tvdb_Id.Value.ToString(CultureInfo.InvariantCulture));
  71. }
  72. var credits = seasonInfo.Credits;
  73. if (credits != null)
  74. {
  75. // Actors, Directors, Writers - all in People
  76. // actors come from cast
  77. if (credits.Cast != null)
  78. {
  79. // foreach (var actor in credits.cast.OrderBy(a => a.order)) result.Item.AddPerson(new PersonInfo { Name = actor.name.Trim(), Role = actor.character, Type = PersonType.Actor, SortOrder = actor.order });
  80. }
  81. // and the rest from crew
  82. if (credits.Crew != null)
  83. {
  84. // foreach (var person in credits.crew) result.Item.AddPerson(new PersonInfo { Name = person.name.Trim(), Role = person.job, Type = person.department });
  85. }
  86. }
  87. result.Item.PremiereDate = seasonInfo.Air_Date;
  88. result.Item.ProductionYear = result.Item.PremiereDate.Value.Year;
  89. }
  90. catch (HttpException ex)
  91. {
  92. _logger.LogError(ex, "No metadata found for {0}", seasonNumber.Value);
  93. if (ex.StatusCode.HasValue && ex.StatusCode.Value == HttpStatusCode.NotFound)
  94. {
  95. return result;
  96. }
  97. throw;
  98. }
  99. }
  100. return result;
  101. }
  102. public Task<IEnumerable<RemoteSearchResult>> GetSearchResults(SeasonInfo searchInfo, CancellationToken cancellationToken)
  103. {
  104. return Task.FromResult<IEnumerable<RemoteSearchResult>>(new List<RemoteSearchResult>());
  105. }
  106. public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
  107. {
  108. return _httpClientFactory.CreateClient(NamedClient.Default).GetAsync(url, cancellationToken);
  109. }
  110. private async Task<SeasonResult> GetSeasonInfo(
  111. string seriesTmdbId,
  112. int season,
  113. string preferredMetadataLanguage,
  114. CancellationToken cancellationToken)
  115. {
  116. await EnsureSeasonInfo(seriesTmdbId, season, preferredMetadataLanguage, cancellationToken)
  117. .ConfigureAwait(false);
  118. var dataFilePath = GetDataFilePath(seriesTmdbId, season, preferredMetadataLanguage);
  119. return _jsonSerializer.DeserializeFromFile<SeasonResult>(dataFilePath);
  120. }
  121. internal Task EnsureSeasonInfo(string tmdbId, int seasonNumber, string language, CancellationToken cancellationToken)
  122. {
  123. if (string.IsNullOrEmpty(tmdbId))
  124. {
  125. throw new ArgumentNullException(nameof(tmdbId));
  126. }
  127. if (string.IsNullOrEmpty(language))
  128. {
  129. throw new ArgumentNullException(nameof(language));
  130. }
  131. var path = GetDataFilePath(tmdbId, seasonNumber, language);
  132. var fileInfo = _fileSystem.GetFileSystemInfo(path);
  133. if (fileInfo.Exists)
  134. {
  135. // If it's recent or automatic updates are enabled, don't re-download
  136. if ((DateTime.UtcNow - _fileSystem.GetLastWriteTimeUtc(fileInfo)).TotalDays <= 2)
  137. {
  138. return Task.CompletedTask;
  139. }
  140. }
  141. return DownloadSeasonInfo(tmdbId, seasonNumber, language, cancellationToken);
  142. }
  143. internal string GetDataFilePath(string tmdbId, int seasonNumber, string preferredLanguage)
  144. {
  145. if (string.IsNullOrEmpty(tmdbId))
  146. {
  147. throw new ArgumentNullException(nameof(tmdbId));
  148. }
  149. if (string.IsNullOrEmpty(preferredLanguage))
  150. {
  151. throw new ArgumentNullException(nameof(preferredLanguage));
  152. }
  153. var path = TmdbSeriesProvider.GetSeriesDataPath(_configurationManager.ApplicationPaths, tmdbId);
  154. var filename = string.Format(
  155. CultureInfo.InvariantCulture,
  156. "season-{0}-{1}.json",
  157. seasonNumber.ToString(CultureInfo.InvariantCulture),
  158. preferredLanguage);
  159. return Path.Combine(path, filename);
  160. }
  161. internal async Task DownloadSeasonInfo(string id, int seasonNumber, string preferredMetadataLanguage, CancellationToken cancellationToken)
  162. {
  163. var mainResult = await FetchMainResult(id, seasonNumber, preferredMetadataLanguage, cancellationToken).ConfigureAwait(false);
  164. var dataFilePath = GetDataFilePath(id, seasonNumber, preferredMetadataLanguage);
  165. Directory.CreateDirectory(Path.GetDirectoryName(dataFilePath));
  166. _jsonSerializer.SerializeToFile(mainResult, dataFilePath);
  167. }
  168. internal async Task<SeasonResult> FetchMainResult(string id, int seasonNumber, string language, CancellationToken cancellationToken)
  169. {
  170. var url = string.Format(
  171. CultureInfo.InvariantCulture,
  172. GetTvInfo3,
  173. id,
  174. seasonNumber.ToString(CultureInfo.InvariantCulture),
  175. TmdbUtils.ApiKey);
  176. if (!string.IsNullOrEmpty(language))
  177. {
  178. url += string.Format(CultureInfo.InvariantCulture, "&language={0}", TmdbMovieProvider.NormalizeLanguage(language));
  179. }
  180. var includeImageLanguageParam = TmdbMovieProvider.GetImageLanguagesParam(language);
  181. // Get images in english and with no language
  182. url += "&include_image_language=" + includeImageLanguageParam;
  183. cancellationToken.ThrowIfCancellationRequested();
  184. using var requestMessage = new HttpRequestMessage(HttpMethod.Get, url);
  185. foreach (var header in TmdbUtils.AcceptHeaders)
  186. {
  187. requestMessage.Headers.Accept.Add(new MediaTypeWithQualityHeaderValue(header));
  188. }
  189. using var response = await TmdbMovieProvider.Current.GetMovieDbResponse(requestMessage, cancellationToken).ConfigureAwait(false);
  190. await using var stream = await response.Content.ReadAsStreamAsync().ConfigureAwait(false);
  191. return await _jsonSerializer.DeserializeFromStreamAsync<SeasonResult>(stream).ConfigureAwait(false);
  192. }
  193. }
  194. }