using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Threading; using System.Threading.Tasks; using MediaBrowser.Common.Extensions; using MediaBrowser.Common.Net; using MediaBrowser.Controller; using MediaBrowser.Controller.Configuration; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Net; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Entities; using MediaBrowser.Model.IO; using MediaBrowser.Model.Providers; using MediaBrowser.Model.Services; using Microsoft.Extensions.Logging; namespace MediaBrowser.Api.Images { public class BaseRemoteImageRequest : IReturn { [ApiMember(Name = "Type", Description = "The image type", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")] public ImageType? Type { get; set; } /// /// Skips over a given number of items within the results. Use for paging. /// /// The start index. [ApiMember(Name = "StartIndex", Description = "Optional. The record index to start at. All items with a lower index will be dropped from the results.", IsRequired = false, DataType = "int", ParameterType = "query", Verb = "GET")] public int? StartIndex { get; set; } /// /// The maximum number of items to return /// /// The limit. [ApiMember(Name = "Limit", Description = "Optional. The maximum number of records to return", IsRequired = false, DataType = "int", ParameterType = "query", Verb = "GET")] public int? Limit { get; set; } [ApiMember(Name = "ProviderName", Description = "Optional. The image provider to use", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")] public string ProviderName { get; set; } [ApiMember(Name = "IncludeAllLanguages", Description = "Optional.", IsRequired = false, DataType = "bool", ParameterType = "query", Verb = "GET")] public bool IncludeAllLanguages { get; set; } } [Route("/Items/{Id}/RemoteImages", "GET", Summary = "Gets available remote images for an item")] [Authenticated] public class GetRemoteImages : BaseRemoteImageRequest { /// /// Gets or sets the id. /// /// The id. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")] public string Id { get; set; } } [Route("/Items/{Id}/RemoteImages/Providers", "GET", Summary = "Gets available remote image providers for an item")] [Authenticated] public class GetRemoteImageProviders : IReturn> { /// /// Gets or sets the id. /// /// The id. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")] public string Id { get; set; } } public class BaseDownloadRemoteImage : IReturnVoid { [ApiMember(Name = "Type", Description = "The image type", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "GET,POST")] public ImageType Type { get; set; } [ApiMember(Name = "ProviderName", Description = "The image provider", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET,POST")] public string ProviderName { get; set; } [ApiMember(Name = "ImageUrl", Description = "The image url", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET,POST")] public string ImageUrl { get; set; } } [Route("/Items/{Id}/RemoteImages/Download", "POST", Summary = "Downloads a remote image for an item")] [Authenticated(Roles = "Admin")] public class DownloadRemoteImage : BaseDownloadRemoteImage { /// /// Gets or sets the id. /// /// The id. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "POST")] public string Id { get; set; } } [Route("/Images/Remote", "GET", Summary = "Gets a remote image")] public class GetRemoteImage { [ApiMember(Name = "ImageUrl", Description = "The image url", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "GET")] public string ImageUrl { get; set; } } public class RemoteImageService : BaseApiService { private readonly IProviderManager _providerManager; private readonly IServerApplicationPaths _appPaths; private readonly IHttpClient _httpClient; private readonly IFileSystem _fileSystem; private readonly ILibraryManager _libraryManager; public RemoteImageService( ILogger logger, IServerConfigurationManager serverConfigurationManager, IHttpResultFactory httpResultFactory, IProviderManager providerManager, IServerApplicationPaths appPaths, IHttpClient httpClient, IFileSystem fileSystem, ILibraryManager libraryManager) : base(logger, serverConfigurationManager, httpResultFactory) { _providerManager = providerManager; _appPaths = appPaths; _httpClient = httpClient; _fileSystem = fileSystem; _libraryManager = libraryManager; } public object Get(GetRemoteImageProviders request) { var item = _libraryManager.GetItemById(request.Id); var result = GetImageProviders(item); return ToOptimizedResult(result); } private List GetImageProviders(BaseItem item) { return _providerManager.GetRemoteImageProviderInfo(item).ToList(); } public async Task Get(GetRemoteImages request) { var item = _libraryManager.GetItemById(request.Id); var images = await _providerManager.GetAvailableRemoteImages(item, new RemoteImageQuery(request.ProviderName) { IncludeAllLanguages = request.IncludeAllLanguages, IncludeDisabledProviders = true, ImageType = request.Type }, CancellationToken.None).ConfigureAwait(false); var imagesList = images.ToArray(); var allProviders = _providerManager.GetRemoteImageProviderInfo(item); if (request.Type.HasValue) { allProviders = allProviders.Where(i => i.SupportedImages.Contains(request.Type.Value)); } var result = new RemoteImageResult { TotalRecordCount = imagesList.Length, Providers = allProviders.Select(i => i.Name) .Distinct(StringComparer.OrdinalIgnoreCase) .ToArray() }; if (request.StartIndex.HasValue) { imagesList = imagesList.Skip(request.StartIndex.Value) .ToArray(); } if (request.Limit.HasValue) { imagesList = imagesList.Take(request.Limit.Value) .ToArray(); } result.Images = imagesList; return result; } /// /// Posts the specified request. /// /// The request. public Task Post(DownloadRemoteImage request) { var item = _libraryManager.GetItemById(request.Id); return DownloadRemoteImage(item, request); } /// /// Downloads the remote image. /// /// The item. /// The request. /// Task. private async Task DownloadRemoteImage(BaseItem item, BaseDownloadRemoteImage request) { await _providerManager.SaveImage(item, request.ImageUrl, request.Type, null, CancellationToken.None).ConfigureAwait(false); item.UpdateToRepository(ItemUpdateType.ImageUpdate, CancellationToken.None); } /// /// Gets the specified request. /// /// The request. /// System.Object. public async Task Get(GetRemoteImage request) { var urlHash = request.ImageUrl.GetMD5(); var pointerCachePath = GetFullCachePath(urlHash.ToString()); string contentPath; try { contentPath = File.ReadAllText(pointerCachePath); if (File.Exists(contentPath)) { return await ResultFactory.GetStaticFileResult(Request, contentPath).ConfigureAwait(false); } } catch (FileNotFoundException) { // Means the file isn't cached yet } catch (IOException) { // Means the file isn't cached yet } await DownloadImage(request.ImageUrl, urlHash, pointerCachePath).ConfigureAwait(false); // Read the pointer file again contentPath = File.ReadAllText(pointerCachePath); return await ResultFactory.GetStaticFileResult(Request, contentPath).ConfigureAwait(false); } /// /// Downloads the image. /// /// The URL. /// The URL hash. /// The pointer cache path. /// Task. private async Task DownloadImage(string url, Guid urlHash, string pointerCachePath) { using (var result = await _httpClient.GetResponse(new HttpRequestOptions { Url = url, BufferContent = false }).ConfigureAwait(false)) { var ext = result.ContentType.Split('/').Last(); var fullCachePath = GetFullCachePath(urlHash + "." + ext); Directory.CreateDirectory(Path.GetDirectoryName(fullCachePath)); using (var stream = result.Content) using (var filestream = new FileStream(fullCachePath, FileMode.Create, FileAccess.Write, FileShare.Read, IODefaults.FileStreamBufferSize, true)) { await stream.CopyToAsync(filestream).ConfigureAwait(false); } Directory.CreateDirectory(Path.GetDirectoryName(pointerCachePath)); File.WriteAllText(pointerCachePath, fullCachePath); } } /// /// Gets the full cache path. /// /// The filename. /// System.String. private string GetFullCachePath(string filename) { return Path.Combine(_appPaths.CachePath, "remote-images", filename.Substring(0, 1), filename); } } }