TvdbEpisodeImageProvider.cs 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  1. using MediaBrowser.Common.IO;
  2. using MediaBrowser.Common.Net;
  3. using MediaBrowser.Controller.Configuration;
  4. using MediaBrowser.Controller.Entities;
  5. using MediaBrowser.Controller.Entities.TV;
  6. using MediaBrowser.Controller.Library;
  7. using MediaBrowser.Controller.Providers;
  8. using MediaBrowser.Model.Entities;
  9. using MediaBrowser.Model.Providers;
  10. using System;
  11. using System.Collections.Generic;
  12. using System.Globalization;
  13. using System.IO;
  14. using System.Linq;
  15. using System.Text;
  16. using System.Threading;
  17. using System.Threading.Tasks;
  18. using System.Xml;
  19. using CommonIO;
  20. namespace MediaBrowser.Providers.TV
  21. {
  22. public class TvdbEpisodeImageProvider : IRemoteImageProvider, IHasChangeMonitor
  23. {
  24. private readonly IServerConfigurationManager _config;
  25. private readonly CultureInfo _usCulture = new CultureInfo("en-US");
  26. private readonly IHttpClient _httpClient;
  27. private readonly IFileSystem _fileSystem;
  28. public TvdbEpisodeImageProvider(IServerConfigurationManager config, IHttpClient httpClient, IFileSystem fileSystem)
  29. {
  30. _config = config;
  31. _httpClient = httpClient;
  32. _fileSystem = fileSystem;
  33. }
  34. public string Name
  35. {
  36. get { return "TheTVDB"; }
  37. }
  38. public bool Supports(IHasImages item)
  39. {
  40. return item is Episode;
  41. }
  42. public IEnumerable<ImageType> GetSupportedImages(IHasImages item)
  43. {
  44. return new List<ImageType>
  45. {
  46. ImageType.Primary
  47. };
  48. }
  49. public Task<IEnumerable<RemoteImageInfo>> GetImages(IHasImages item, CancellationToken cancellationToken)
  50. {
  51. var episode = (Episode)item;
  52. var series = episode.Series;
  53. if (series != null && TvdbSeriesProvider.IsValidSeries(series.ProviderIds))
  54. {
  55. // Process images
  56. var seriesDataPath = TvdbSeriesProvider.GetSeriesDataPath(_config.ApplicationPaths, series.ProviderIds);
  57. var indexOffset = TvdbSeriesProvider.GetSeriesOffset(series.ProviderIds) ?? 0;
  58. var nodes = TvdbEpisodeProvider.Current.GetEpisodeXmlNodes(seriesDataPath, episode.GetLookupInfo());
  59. var result = nodes.Select(i => GetImageInfo(i, cancellationToken))
  60. .Where(i => i != null)
  61. .ToList();
  62. return Task.FromResult<IEnumerable<RemoteImageInfo>>(result);
  63. }
  64. return Task.FromResult<IEnumerable<RemoteImageInfo>>(new RemoteImageInfo[] { });
  65. }
  66. private RemoteImageInfo GetImageInfo(XmlReader reader, CancellationToken cancellationToken)
  67. {
  68. var height = 225;
  69. var width = 400;
  70. var url = string.Empty;
  71. // Use XmlReader for best performance
  72. using (reader)
  73. {
  74. reader.MoveToContent();
  75. // Loop through each element
  76. while (reader.Read())
  77. {
  78. cancellationToken.ThrowIfCancellationRequested();
  79. if (reader.NodeType == XmlNodeType.Element)
  80. {
  81. switch (reader.Name)
  82. {
  83. case "thumb_width":
  84. {
  85. var val = reader.ReadElementContentAsString();
  86. if (!string.IsNullOrWhiteSpace(val))
  87. {
  88. int rval;
  89. // int.TryParse is local aware, so it can be probamatic, force us culture
  90. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  91. {
  92. width = rval;
  93. }
  94. }
  95. break;
  96. }
  97. case "thumb_height":
  98. {
  99. var val = reader.ReadElementContentAsString();
  100. if (!string.IsNullOrWhiteSpace(val))
  101. {
  102. int rval;
  103. // int.TryParse is local aware, so it can be probamatic, force us culture
  104. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  105. {
  106. height = rval;
  107. }
  108. }
  109. break;
  110. }
  111. case "filename":
  112. {
  113. var val = reader.ReadElementContentAsString();
  114. if (!string.IsNullOrWhiteSpace(val))
  115. {
  116. url = TVUtils.BannerUrl + val;
  117. }
  118. break;
  119. }
  120. default:
  121. reader.Skip();
  122. break;
  123. }
  124. }
  125. }
  126. }
  127. if (string.IsNullOrEmpty(url))
  128. {
  129. return null;
  130. }
  131. return new RemoteImageInfo
  132. {
  133. Width = width,
  134. Height = height,
  135. ProviderName = Name,
  136. Url = url,
  137. Type = ImageType.Primary
  138. };
  139. }
  140. public int Order
  141. {
  142. get { return 0; }
  143. }
  144. public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
  145. {
  146. return _httpClient.GetResponse(new HttpRequestOptions
  147. {
  148. CancellationToken = cancellationToken,
  149. Url = url,
  150. ResourcePool = TvdbSeriesProvider.Current.TvDbResourcePool
  151. });
  152. }
  153. public bool HasChanged(IHasMetadata item, IDirectoryService directoryService, DateTime date)
  154. {
  155. var episode = (Episode)item;
  156. if (!episode.IsVirtualUnaired)
  157. {
  158. // For non-unaired items, only enable if configured
  159. if (!TvdbSeriesProvider.Current.GetTvDbOptions().EnableAutomaticUpdates)
  160. {
  161. return false;
  162. }
  163. }
  164. if (!item.HasImage(ImageType.Primary))
  165. {
  166. var series = episode.Series;
  167. if (series != null && TvdbSeriesProvider.IsValidSeries(series.ProviderIds))
  168. {
  169. // Process images
  170. var seriesXmlPath = TvdbSeriesProvider.Current.GetSeriesXmlPath(series.ProviderIds, series.GetPreferredMetadataLanguage());
  171. return _fileSystem.GetLastWriteTimeUtc(seriesXmlPath) > date;
  172. }
  173. }
  174. return false;
  175. }
  176. }
  177. }