2021-10-26 15:49:01 +02:00
|
|
|
#nullable disable
|
|
|
|
|
2020-06-19 20:24:13 +02:00
|
|
|
#pragma warning disable CS1591
|
|
|
|
|
2019-01-13 21:03:10 +01:00
|
|
|
using System;
|
2013-12-30 17:32:01 +01:00
|
|
|
using System.Collections.Generic;
|
2020-08-07 19:26:28 +02:00
|
|
|
using System.Globalization;
|
2013-12-30 17:32:01 +01:00
|
|
|
using System.IO;
|
|
|
|
using System.Linq;
|
2019-07-29 23:47:25 +02:00
|
|
|
using System.Net.Http;
|
2013-12-30 17:32:01 +01:00
|
|
|
using System.Threading;
|
|
|
|
using System.Threading.Tasks;
|
2021-06-19 18:02:33 +02:00
|
|
|
using Jellyfin.Extensions;
|
2020-08-31 19:05:21 +02:00
|
|
|
using MediaBrowser.Common.Net;
|
2019-01-13 20:26:31 +01:00
|
|
|
using MediaBrowser.Controller.Configuration;
|
|
|
|
using MediaBrowser.Controller.Entities;
|
|
|
|
using MediaBrowser.Controller.Providers;
|
|
|
|
using MediaBrowser.Model.Entities;
|
|
|
|
using MediaBrowser.Model.IO;
|
|
|
|
using MediaBrowser.Model.Providers;
|
2021-01-12 21:35:06 +01:00
|
|
|
using MediaBrowser.Providers.Plugins.StudioImages;
|
2013-12-30 17:32:01 +01:00
|
|
|
|
2014-01-28 19:37:01 +01:00
|
|
|
namespace MediaBrowser.Providers.Studios
|
2013-12-30 17:32:01 +01:00
|
|
|
{
|
2014-01-28 19:37:01 +01:00
|
|
|
public class StudiosImageProvider : IRemoteImageProvider
|
2013-12-30 17:32:01 +01:00
|
|
|
{
|
|
|
|
private readonly IServerConfigurationManager _config;
|
2020-08-17 19:46:40 +02:00
|
|
|
private readonly IHttpClientFactory _httpClientFactory;
|
2013-12-30 17:32:01 +01:00
|
|
|
private readonly IFileSystem _fileSystem;
|
2021-12-24 18:28:27 +01:00
|
|
|
private readonly string repositoryUrl;
|
2013-12-30 17:32:01 +01:00
|
|
|
|
2020-08-17 19:46:40 +02:00
|
|
|
public StudiosImageProvider(IServerConfigurationManager config, IHttpClientFactory httpClientFactory, IFileSystem fileSystem)
|
2013-12-30 17:32:01 +01:00
|
|
|
{
|
|
|
|
_config = config;
|
2020-08-17 19:46:40 +02:00
|
|
|
_httpClientFactory = httpClientFactory;
|
2013-12-30 17:32:01 +01:00
|
|
|
_fileSystem = fileSystem;
|
2021-01-12 21:35:06 +01:00
|
|
|
repositoryUrl = Plugin.Instance.Configuration.RepositoryUrl;
|
2013-12-30 17:32:01 +01:00
|
|
|
}
|
|
|
|
|
2021-01-12 21:35:06 +01:00
|
|
|
public string Name => "Artwork Repository";
|
2013-12-30 17:32:01 +01:00
|
|
|
|
2020-09-08 16:12:47 +02:00
|
|
|
public int Order => 0;
|
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
public bool Supports(BaseItem item)
|
2013-12-30 17:32:01 +01:00
|
|
|
{
|
|
|
|
return item is Studio;
|
|
|
|
}
|
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
public IEnumerable<ImageType> GetSupportedImages(BaseItem item)
|
2014-01-28 19:37:01 +01:00
|
|
|
{
|
|
|
|
return new List<ImageType>
|
|
|
|
{
|
|
|
|
ImageType.Thumb
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2022-03-08 16:36:05 +01:00
|
|
|
public async Task<IEnumerable<RemoteImageInfo>> GetImages(BaseItem item, CancellationToken cancellationToken)
|
2013-12-30 17:32:01 +01:00
|
|
|
{
|
2022-03-08 16:36:05 +01:00
|
|
|
var thumbsPath = Path.Combine(_config.ApplicationPaths.CachePath, "imagesbyname", "remotestudiothumbs.txt");
|
2013-12-30 17:32:01 +01:00
|
|
|
|
2022-03-08 16:36:05 +01:00
|
|
|
thumbsPath = await EnsureThumbsList(thumbsPath, cancellationToken).ConfigureAwait(false);
|
2013-12-30 17:32:01 +01:00
|
|
|
|
|
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
|
2022-03-08 16:36:05 +01:00
|
|
|
var imageInfo = GetImage(item, thumbsPath, ImageType.Thumb, "thumb");
|
2013-12-30 17:32:01 +01:00
|
|
|
|
2022-03-08 16:36:05 +01:00
|
|
|
if (imageInfo == null)
|
|
|
|
{
|
|
|
|
return Enumerable.Empty<RemoteImageInfo>();
|
2013-12-30 17:32:01 +01:00
|
|
|
}
|
|
|
|
|
2022-03-08 16:36:05 +01:00
|
|
|
return new RemoteImageInfo[]
|
|
|
|
{
|
|
|
|
imageInfo
|
|
|
|
};
|
2013-12-30 17:32:01 +01:00
|
|
|
}
|
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
private RemoteImageInfo GetImage(BaseItem item, string filename, ImageType type, string remoteFilename)
|
2013-12-30 17:32:01 +01:00
|
|
|
{
|
2020-01-08 17:52:50 +01:00
|
|
|
var list = GetAvailableImages(filename);
|
2013-12-30 17:32:01 +01:00
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
var match = FindMatch(item, list);
|
2013-12-30 17:32:01 +01:00
|
|
|
|
|
|
|
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)
|
|
|
|
{
|
2022-03-04 07:39:21 +01:00
|
|
|
return string.Format(CultureInfo.InvariantCulture, "{0}/images/{1}/{2}.jpg", repositoryUrl, image, filename);
|
2013-12-30 17:32:01 +01:00
|
|
|
}
|
|
|
|
|
2017-04-29 08:22:33 +02:00
|
|
|
private Task<string> EnsureThumbsList(string file, CancellationToken cancellationToken)
|
2013-12-30 17:32:01 +01:00
|
|
|
{
|
2022-03-04 07:39:21 +01:00
|
|
|
string url = string.Format(CultureInfo.InvariantCulture, "{0}/thumbs.txt", repositoryUrl);
|
2013-12-30 17:32:01 +01:00
|
|
|
|
2020-08-17 19:46:40 +02:00
|
|
|
return EnsureList(url, file, _fileSystem, cancellationToken);
|
2013-12-30 17:32:01 +01:00
|
|
|
}
|
|
|
|
|
2020-08-17 19:52:09 +02:00
|
|
|
public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
|
2014-01-28 19:37:01 +01:00
|
|
|
{
|
2020-08-31 19:05:21 +02:00
|
|
|
var httpClient = _httpClientFactory.CreateClient(NamedClient.Default);
|
2020-08-17 19:52:09 +02:00
|
|
|
return httpClient.GetAsync(url, cancellationToken);
|
2014-01-28 19:37:01 +01:00
|
|
|
}
|
2018-09-12 19:26:21 +02:00
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
/// Ensures the list.
|
|
|
|
/// </summary>
|
|
|
|
/// <param name="url">The URL.</param>
|
|
|
|
/// <param name="file">The file.</param>
|
|
|
|
/// <param name="fileSystem">The file system.</param>
|
|
|
|
/// <param name="cancellationToken">The cancellation token.</param>
|
|
|
|
/// <returns>Task.</returns>
|
2020-08-17 19:46:40 +02:00
|
|
|
public async Task<string> EnsureList(string url, string file, IFileSystem fileSystem, CancellationToken cancellationToken)
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
|
|
|
var fileInfo = fileSystem.GetFileInfo(file);
|
|
|
|
|
|
|
|
if (!fileInfo.Exists || (DateTime.UtcNow - fileSystem.GetLastWriteTimeUtc(fileInfo)).TotalDays > 1)
|
|
|
|
{
|
2020-08-31 19:05:21 +02:00
|
|
|
var httpClient = _httpClientFactory.CreateClient(NamedClient.Default);
|
2018-09-12 19:26:21 +02:00
|
|
|
|
2020-08-17 19:46:40 +02:00
|
|
|
Directory.CreateDirectory(Path.GetDirectoryName(file));
|
2021-02-15 14:19:08 +01:00
|
|
|
await using var response = await httpClient.GetStreamAsync(url, cancellationToken).ConfigureAwait(false);
|
2021-09-25 19:44:40 +02:00
|
|
|
await using var fileStream = new FileStream(file, FileMode.Create, FileAccess.Write, FileShare.None, IODefaults.FileStreamBufferSize, FileOptions.Asynchronous);
|
2020-08-17 19:46:40 +02:00
|
|
|
await response.CopyToAsync(fileStream, cancellationToken).ConfigureAwait(false);
|
2018-09-12 19:26:21 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
return file;
|
|
|
|
}
|
|
|
|
|
|
|
|
public string FindMatch(BaseItem item, IEnumerable<string> images)
|
|
|
|
{
|
|
|
|
var name = GetComparableName(item.Name);
|
|
|
|
|
|
|
|
return images.FirstOrDefault(i => string.Equals(name, GetComparableName(i), StringComparison.OrdinalIgnoreCase));
|
|
|
|
}
|
|
|
|
|
|
|
|
private string GetComparableName(string name)
|
|
|
|
{
|
2020-09-08 16:12:47 +02:00
|
|
|
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);
|
2018-09-12 19:26:21 +02:00
|
|
|
}
|
|
|
|
|
2020-01-08 17:52:50 +01:00
|
|
|
public IEnumerable<string> GetAvailableImages(string file)
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
2021-10-03 19:52:38 +02:00
|
|
|
using var fileStream = File.OpenRead(file);
|
2021-05-13 16:47:00 +02:00
|
|
|
using var reader = new StreamReader(fileStream);
|
2018-09-12 19:26:21 +02:00
|
|
|
|
2021-05-10 15:05:12 +02:00
|
|
|
foreach (var line in reader.ReadAllLines())
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
2021-05-10 15:05:12 +02:00
|
|
|
if (!string.IsNullOrWhiteSpace(line))
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
2021-10-03 19:52:38 +02:00
|
|
|
yield return line;
|
2018-09-12 19:26:21 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2013-12-30 17:32:01 +01:00
|
|
|
}
|
|
|
|
}
|