2
0

StudiosImageProvider.cs 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173
  1. #nullable disable
  2. #pragma warning disable CS1591
  3. using System;
  4. using System.Collections.Generic;
  5. using System.Globalization;
  6. using System.IO;
  7. using System.Linq;
  8. using System.Net.Http;
  9. using System.Threading;
  10. using System.Threading.Tasks;
  11. using Jellyfin.Extensions;
  12. using MediaBrowser.Common.Net;
  13. using MediaBrowser.Controller.Configuration;
  14. using MediaBrowser.Controller.Entities;
  15. using MediaBrowser.Controller.Providers;
  16. using MediaBrowser.Model.Entities;
  17. using MediaBrowser.Model.IO;
  18. using MediaBrowser.Model.Providers;
  19. using MediaBrowser.Providers.Plugins.StudioImages;
  20. namespace MediaBrowser.Providers.Studios
  21. {
  22. public class StudiosImageProvider : IRemoteImageProvider
  23. {
  24. private readonly IServerConfigurationManager _config;
  25. private readonly IHttpClientFactory _httpClientFactory;
  26. private readonly IFileSystem _fileSystem;
  27. private readonly string repositoryUrl;
  28. public StudiosImageProvider(IServerConfigurationManager config, IHttpClientFactory httpClientFactory, IFileSystem fileSystem)
  29. {
  30. _config = config;
  31. _httpClientFactory = httpClientFactory;
  32. _fileSystem = fileSystem;
  33. repositoryUrl = Plugin.Instance.Configuration.RepositoryUrl;
  34. }
  35. public string Name => "Artwork Repository";
  36. public int Order => 0;
  37. public bool Supports(BaseItem item)
  38. {
  39. return item is Studio;
  40. }
  41. public IEnumerable<ImageType> GetSupportedImages(BaseItem item)
  42. {
  43. return new List<ImageType>
  44. {
  45. ImageType.Thumb
  46. };
  47. }
  48. public async Task<IEnumerable<RemoteImageInfo>> GetImages(BaseItem item, CancellationToken cancellationToken)
  49. {
  50. var thumbsPath = Path.Combine(_config.ApplicationPaths.CachePath, "imagesbyname", "remotestudiothumbs.txt");
  51. thumbsPath = await EnsureThumbsList(thumbsPath, cancellationToken).ConfigureAwait(false);
  52. cancellationToken.ThrowIfCancellationRequested();
  53. var imageInfo = GetImage(item, thumbsPath, ImageType.Thumb, "thumb");
  54. if (imageInfo == null)
  55. {
  56. return Enumerable.Empty<RemoteImageInfo>();
  57. }
  58. return new RemoteImageInfo[]
  59. {
  60. imageInfo
  61. };
  62. }
  63. private RemoteImageInfo GetImage(BaseItem item, string filename, ImageType type, string remoteFilename)
  64. {
  65. var list = GetAvailableImages(filename);
  66. var match = FindMatch(item, list);
  67. if (!string.IsNullOrEmpty(match))
  68. {
  69. var url = GetUrl(match, remoteFilename);
  70. return new RemoteImageInfo
  71. {
  72. ProviderName = Name,
  73. Type = type,
  74. Url = url
  75. };
  76. }
  77. return null;
  78. }
  79. private string GetUrl(string image, string filename)
  80. {
  81. return string.Format(CultureInfo.InvariantCulture, "{0}/images/{1}/{2}.jpg", repositoryUrl, image, filename);
  82. }
  83. private Task<string> EnsureThumbsList(string file, CancellationToken cancellationToken)
  84. {
  85. string url = string.Format(CultureInfo.InvariantCulture, "{0}/thumbs.txt", repositoryUrl);
  86. return EnsureList(url, file, _fileSystem, cancellationToken);
  87. }
  88. public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
  89. {
  90. var httpClient = _httpClientFactory.CreateClient(NamedClient.Default);
  91. return httpClient.GetAsync(url, cancellationToken);
  92. }
  93. /// <summary>
  94. /// Ensures the list.
  95. /// </summary>
  96. /// <param name="url">The URL.</param>
  97. /// <param name="file">The file.</param>
  98. /// <param name="fileSystem">The file system.</param>
  99. /// <param name="cancellationToken">The cancellation token.</param>
  100. /// <returns>Task.</returns>
  101. public async Task<string> EnsureList(string url, string file, IFileSystem fileSystem, CancellationToken cancellationToken)
  102. {
  103. var fileInfo = fileSystem.GetFileInfo(file);
  104. if (!fileInfo.Exists || (DateTime.UtcNow - fileSystem.GetLastWriteTimeUtc(fileInfo)).TotalDays > 1)
  105. {
  106. var httpClient = _httpClientFactory.CreateClient(NamedClient.Default);
  107. Directory.CreateDirectory(Path.GetDirectoryName(file));
  108. await using var response = await httpClient.GetStreamAsync(url, cancellationToken).ConfigureAwait(false);
  109. await using var fileStream = new FileStream(file, FileMode.Create, FileAccess.Write, FileShare.None, IODefaults.FileStreamBufferSize, FileOptions.Asynchronous);
  110. await response.CopyToAsync(fileStream, cancellationToken).ConfigureAwait(false);
  111. }
  112. return file;
  113. }
  114. public string FindMatch(BaseItem item, IEnumerable<string> images)
  115. {
  116. var name = GetComparableName(item.Name);
  117. return images.FirstOrDefault(i => string.Equals(name, GetComparableName(i), StringComparison.OrdinalIgnoreCase));
  118. }
  119. private string GetComparableName(string name)
  120. {
  121. return name.Replace(" ", string.Empty, StringComparison.Ordinal)
  122. .Replace(".", string.Empty, StringComparison.Ordinal)
  123. .Replace("&", string.Empty, StringComparison.Ordinal)
  124. .Replace("!", string.Empty, StringComparison.Ordinal)
  125. .Replace(",", string.Empty, StringComparison.Ordinal)
  126. .Replace("/", string.Empty, StringComparison.Ordinal);
  127. }
  128. public IEnumerable<string> GetAvailableImages(string file)
  129. {
  130. using var fileStream = File.OpenRead(file);
  131. using var reader = new StreamReader(fileStream);
  132. foreach (var line in reader.ReadAllLines())
  133. {
  134. if (!string.IsNullOrWhiteSpace(line))
  135. {
  136. yield return line;
  137. }
  138. }
  139. }
  140. }
  141. }