#nullable disable #pragma warning disable CS1591 using System; using System.Collections.Generic; using System.Globalization; using System.IO; using System.Linq; using System.Net.Http; using System.Threading; using System.Threading.Tasks; using Jellyfin.Extensions; using MediaBrowser.Common.Net; using MediaBrowser.Controller.Configuration; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Entities; using MediaBrowser.Model.IO; using MediaBrowser.Model.Providers; namespace MediaBrowser.Providers.Plugins.StudioImages { public class StudiosImageProvider : IRemoteImageProvider { private readonly IServerConfigurationManager _config; private readonly IHttpClientFactory _httpClientFactory; private readonly IFileSystem _fileSystem; public StudiosImageProvider(IServerConfigurationManager config, IHttpClientFactory httpClientFactory, IFileSystem fileSystem) { _config = config; _httpClientFactory = httpClientFactory; _fileSystem = fileSystem; } public string Name => "Artwork Repository"; public bool Supports(BaseItem item) { return item is Studio; } public IEnumerable GetSupportedImages(BaseItem item) { return new List { ImageType.Thumb }; } public async Task> GetImages(BaseItem item, CancellationToken cancellationToken) { var thumbsPath = Path.Combine(_config.ApplicationPaths.CachePath, "imagesbyname", "remotestudiothumbs.txt"); thumbsPath = await EnsureThumbsList(thumbsPath, cancellationToken).ConfigureAwait(false); cancellationToken.ThrowIfCancellationRequested(); var imageInfo = GetImage(item, thumbsPath, ImageType.Thumb, "thumb"); if (imageInfo == null) { return Enumerable.Empty(); } return new RemoteImageInfo[] { imageInfo }; } private RemoteImageInfo GetImage(BaseItem item, string filename, ImageType type, string remoteFilename) { var list = GetAvailableImages(filename); var match = FindMatch(item, list); if (!string.IsNullOrEmpty(match)) { var url = GetUrl(match, remoteFilename); return new RemoteImageInfo { ProviderName = Name, Type = type, Url = url }; } return null; } private string GetUrl(string image, string filename) { return string.Format(CultureInfo.InvariantCulture, "{0}/images/{1}/{2}.jpg", GetRepositoryUrl(), image, filename); } private Task EnsureThumbsList(string file, CancellationToken cancellationToken) { string url = string.Format(CultureInfo.InvariantCulture, "{0}/thumbs.txt", GetRepositoryUrl()); return EnsureList(url, file, _fileSystem, cancellationToken); } public Task GetImageResponse(string url, CancellationToken cancellationToken) { var httpClient = _httpClientFactory.CreateClient(NamedClient.Default); return httpClient.GetAsync(url, cancellationToken); } /// /// Ensures the list. /// /// The URL. /// The file. /// The file system. /// The cancellation token. /// Task. public async Task EnsureList(string url, string file, IFileSystem fileSystem, CancellationToken cancellationToken) { var fileInfo = fileSystem.GetFileInfo(file); if (!fileInfo.Exists || (DateTime.UtcNow - fileSystem.GetLastWriteTimeUtc(fileInfo)).TotalDays > 1) { var httpClient = _httpClientFactory.CreateClient(NamedClient.Default); Directory.CreateDirectory(Path.GetDirectoryName(file)); await using var response = await httpClient.GetStreamAsync(url, cancellationToken).ConfigureAwait(false); await using var fileStream = new FileStream(file, FileMode.Create, FileAccess.Write, FileShare.None, IODefaults.FileStreamBufferSize, FileOptions.Asynchronous); await response.CopyToAsync(fileStream, cancellationToken).ConfigureAwait(false); } return file; } public string FindMatch(BaseItem item, IEnumerable images) { var name = GetComparableName(item.Name); return images.FirstOrDefault(i => string.Equals(name, GetComparableName(i), StringComparison.OrdinalIgnoreCase)); } private string GetComparableName(string name) { return name.Replace(" ", string.Empty, StringComparison.Ordinal) .Replace(".", string.Empty, StringComparison.Ordinal) .Replace("&", string.Empty, StringComparison.Ordinal) .Replace("!", string.Empty, StringComparison.Ordinal) .Replace(",", string.Empty, StringComparison.Ordinal) .Replace("/", string.Empty, StringComparison.Ordinal); } public IEnumerable GetAvailableImages(string file) { using var fileStream = File.OpenRead(file); using var reader = new StreamReader(fileStream); foreach (var line in reader.ReadAllLines()) { if (!string.IsNullOrWhiteSpace(line)) { yield return line; } } } private string GetRepositoryUrl() => Plugin.Instance.Configuration.RepositoryUrl; } }