#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 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; using MediaBrowser.Providers.Plugins.StudioImages; namespace MediaBrowser.Providers.Studios { public class StudiosImageProvider : IRemoteImageProvider { private readonly IServerConfigurationManager _config; private readonly IHttpClientFactory _httpClientFactory; private readonly IFileSystem _fileSystem; private readonly String repositoryUrl; public StudiosImageProvider(IServerConfigurationManager config, IHttpClientFactory httpClientFactory, IFileSystem fileSystem) { _config = config; _httpClientFactory = httpClientFactory; _fileSystem = fileSystem; repositoryUrl = Plugin.Instance.Configuration.RepositoryUrl; } public string Name => "Artwork Repository"; public int Order => 0; public bool Supports(BaseItem item) { return item is Studio; } public IEnumerable GetSupportedImages(BaseItem item) { return new List { ImageType.Primary, ImageType.Thumb }; } public Task> GetImages(BaseItem item, CancellationToken cancellationToken) { return GetImages(item, true, true, cancellationToken); } private async Task> GetImages(BaseItem item, bool posters, bool thumbs, CancellationToken cancellationToken) { var list = new List(); if (posters) { var posterPath = Path.Combine(_config.ApplicationPaths.CachePath, "imagesbyname", "remotestudioposters.txt"); posterPath = await EnsurePosterList(posterPath, cancellationToken).ConfigureAwait(false); list.Add(GetImage(item, posterPath, ImageType.Primary, "folder")); } cancellationToken.ThrowIfCancellationRequested(); if (thumbs) { var thumbsPath = Path.Combine(_config.ApplicationPaths.CachePath, "imagesbyname", "remotestudiothumbs.txt"); thumbsPath = await EnsureThumbsList(thumbsPath, cancellationToken).ConfigureAwait(false); list.Add(GetImage(item, thumbsPath, ImageType.Thumb, "thumb")); } return list.Where(i => i != null); } 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}/{1}/{2}.jpg", repositoryUrl, image, filename); } private Task EnsureThumbsList(string file, CancellationToken cancellationToken) { string url = string.Format(CultureInfo.InvariantCulture, "{0}/studiothumbs.txt", repositoryUrl); return EnsureList(url, file, _fileSystem, cancellationToken); } private Task EnsurePosterList(string file, CancellationToken cancellationToken) { string url = string.Format(CultureInfo.InvariantCulture, "{0}/studioposters.txt", repositoryUrl); 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).ConfigureAwait(false); await using var fileStream = new FileStream(file, FileMode.Create); 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 = new FileStream(file, FileMode.Open, FileAccess.Read, FileShare.Read)) { using (var reader = new StreamReader(fileStream)) { var lines = new List(); while (!reader.EndOfStream) { var text = reader.ReadLine(); if (!string.IsNullOrWhiteSpace(text)) { lines.Add(text); } } return lines; } } } } }