using MediaBrowser.Common.Net; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.LiveTv; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using System; using System.Collections.Generic; using System.Linq; using System.Threading; using System.Threading.Tasks; namespace MediaBrowser.Server.Implementations.LiveTv { public class ProgramImageProvider : IDynamicImageProvider, IHasItemChangeMonitor, IHasOrder { private readonly ILiveTvManager _liveTvManager; private readonly IHttpClient _httpClient; private readonly ILogger _logger; public ProgramImageProvider(ILiveTvManager liveTvManager, IHttpClient httpClient, ILogger logger) { _liveTvManager = liveTvManager; _httpClient = httpClient; _logger = logger; } public IEnumerable GetSupportedImages(IHasImages item) { return new[] { ImageType.Primary }; } public async Task GetImage(IHasImages item, ImageType type, CancellationToken cancellationToken) { var liveTvItem = (LiveTvProgram)item; var imageResponse = new DynamicImageResponse(); if (!string.IsNullOrEmpty(liveTvItem.ExternalImagePath)) { if (liveTvItem.ExternalImagePath.StartsWith("http", StringComparison.OrdinalIgnoreCase)) { var options = new HttpRequestOptions { CancellationToken = cancellationToken, Url = liveTvItem.ExternalImagePath }; var response = await _httpClient.GetResponse(options).ConfigureAwait(false); if (response.ContentType.StartsWith("image/", StringComparison.OrdinalIgnoreCase)) { imageResponse.HasImage = true; imageResponse.Stream = response.Content; imageResponse.SetFormatFromMimeType(response.ContentType); } else { _logger.Error("Provider did not return an image content type."); } } else { imageResponse.Path = liveTvItem.ExternalImagePath; imageResponse.HasImage = true; } } else { var service = _liveTvManager.Services.FirstOrDefault(i => string.Equals(i.Name, liveTvItem.ServiceName, StringComparison.OrdinalIgnoreCase)); if (service != null) { try { var channel = _liveTvManager.GetInternalChannel(liveTvItem.ChannelId); var response = await service.GetProgramImageAsync(liveTvItem.ExternalId, channel.ExternalId, cancellationToken).ConfigureAwait(false); if (response != null) { imageResponse.HasImage = true; imageResponse.Stream = response.Stream; imageResponse.Format = response.Format; } } catch (NotImplementedException) { } } } return imageResponse; } public string Name { get { return "Live TV Service Provider"; } } public bool Supports(IHasImages item) { return item is LiveTvProgram; } public int Order { get { // Let the better providers run first return 100; } } public bool HasChanged(IHasMetadata item, MetadataStatus status, IDirectoryService directoryService) { var liveTvItem = item as LiveTvProgram; if (liveTvItem != null) { return !liveTvItem.HasImage(ImageType.Primary); } return false; } } }