Kyoo/back/src/Kyoo.Core/Controllers/ThumbnailsManager.cs

// Kyoo - A portable and vast media library solution.
// Copyright (c) Kyoo.
//
// See AUTHORS.md and LICENSE file in the project root for full license information.
//
// Kyoo is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// any later version.
//
// Kyoo is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Kyoo. If not, see <https://www.gnu.org/licenses/>.

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net.Http;
using System.Security.Cryptography;
using System.Text;
using System.Threading.Tasks;
using Blurhash.SkiaSharp;
using Kyoo.Abstractions.Controllers;
using Kyoo.Abstractions.Models;
using Kyoo.Abstractions.Models.Exceptions;
using Microsoft.Extensions.Logging;
using SkiaSharp;

namespace Kyoo.Core.Controllers
{
	/// <summary>
	/// Download images and retrieve the path of those images for a resource.
	/// </summary>
	public class ThumbnailsManager(
		IHttpClientFactory clientFactory,
		ILogger<ThumbnailsManager> logger,
		Lazy<IRepository<User>> users
	) : IThumbnailsManager
	{
		private static readonly Dictionary<string, TaskCompletionSource<object>> _downloading =
			new();

		private static async Task _WriteTo(SKBitmap bitmap, string path, int quality)
		{
			SKData data = bitmap.Encode(SKEncodedImageFormat.Webp, quality);
			await using Stream reader = data.AsStream();
			await using Stream file = File.Create(path);
			await reader.CopyToAsync(file);
		}

		private async Task _DownloadImage(Image? image, string localPath, string what)
		{
			if (image == null)
				return;
			try
			{
				logger.LogInformation("Downloading image {What}", what);

				HttpClient client = clientFactory.CreateClient();
				HttpResponseMessage response = await client.GetAsync(image.Source);
				response.EnsureSuccessStatusCode();
				await using Stream reader = await response.Content.ReadAsStreamAsync();
				using SKCodec codec = SKCodec.Create(reader);
				if (codec == null)
				{
					logger.LogError("Unsupported codec for {What}", what);
					return;
				}

				SKImageInfo info = codec.Info;
				info.ColorType = SKColorType.Rgba8888;
				using SKBitmap original = SKBitmap.Decode(codec, info);

				using SKBitmap high = original.Resize(
					new SKSizeI(original.Width, original.Height),
					SKFilterQuality.High
				);
				await _WriteTo(
					original,
					$"{localPath}.{ImageQuality.High.ToString().ToLowerInvariant()}.webp",
					90
				);

				using SKBitmap medium = high.Resize(
					new SKSizeI((int)(high.Width / 1.5), (int)(high.Height / 1.5)),
					SKFilterQuality.Medium
				);
				await _WriteTo(
					medium,
					$"{localPath}.{ImageQuality.Medium.ToString().ToLowerInvariant()}.webp",
					75
				);

				using SKBitmap low = medium.Resize(
					new SKSizeI(original.Width / 2, original.Height / 2),
					SKFilterQuality.Low
				);
				await _WriteTo(
					low,
					$"{localPath}.{ImageQuality.Low.ToString().ToLowerInvariant()}.webp",
					50
				);

				image.Blurhash = Blurhasher.Encode(low, 4, 3);
			}
			catch (Exception ex)
			{
				logger.LogError(ex, "{What} could not be downloaded", what);
			}
		}

		/// <inheritdoc />
		public async Task DownloadImages<T>(T item)
			where T : IThumbnails
		{
			string name = item is IResource res ? res.Slug : "???";

			string posterPath =
				$"{_GetBaseImagePath(item, "poster")}.{ImageQuality.High.ToString().ToLowerInvariant()}.webp";
			bool duplicated = false;
			TaskCompletionSource<object>? sync = null;
			try
			{
				lock (_downloading)
				{
					if (_downloading.ContainsKey(posterPath))
					{
						duplicated = true;
						sync = _downloading.GetValueOrDefault(posterPath);
					}
					else
					{
						sync = new();
						_downloading.Add(posterPath, sync);
					}
				}
				if (duplicated)
				{
					object? dup = sync != null ? await sync.Task : null;
					if (dup != null)
						throw new DuplicatedItemException(dup);
				}

				await _DownloadImage(
					item.Poster,
					_GetBaseImagePath(item, "poster"),
					$"The poster of {name}"
				);
				await _DownloadImage(
					item.Thumbnail,
					_GetBaseImagePath(item, "thumbnail"),
					$"The poster of {name}"
				);
				await _DownloadImage(
					item.Logo,
					_GetBaseImagePath(item, "logo"),
					$"The poster of {name}"
				);
			}
			finally
			{
				if (!duplicated)
				{
					lock (_downloading)
					{
						_downloading.Remove(posterPath);
						sync!.SetResult(item);
					}
				}
			}
		}

		private static string _GetBaseImagePath<T>(T item, string image)
		{
			string directory = item switch
			{
				IResource res
					=> Path.Combine("./metadata", item.GetType().Name.ToLowerInvariant(), res.Slug),
				_ => Path.Combine("./metadata", typeof(T).Name.ToLowerInvariant())
			};
			Directory.CreateDirectory(directory);
			return Path.Combine(directory, image);
		}

		/// <inheritdoc />
		public string GetImagePath<T>(T item, string image, ImageQuality quality)
			where T : IThumbnails
		{
			return $"{_GetBaseImagePath(item, image)}.{quality.ToString().ToLowerInvariant()}.webp";
		}

		/// <inheritdoc />
		public Task DeleteImages<T>(T item)
			where T : IThumbnails
		{
			IEnumerable<string> images = new[] { "poster", "thumbnail", "logo" }
				.SelectMany(x => _GetBaseImagePath(item, x))
				.SelectMany(x =>
					new[]
					{
						ImageQuality.High.ToString().ToLowerInvariant(),
						ImageQuality.Medium.ToString().ToLowerInvariant(),
						ImageQuality.Low.ToString().ToLowerInvariant(),
					}.Select(quality => $"{x}.{quality}.webp")
				);

			foreach (string image in images)
				File.Delete(image);
			return Task.CompletedTask;
		}

		public async Task<Stream> GetUserImage(Guid userId)
		{
			try
			{
				return File.Open($"/metadata/user/{userId}.webp", FileMode.Open);
			}
			catch (FileNotFoundException) { }
			catch (DirectoryNotFoundException) { }

			User user = await users.Value.Get(userId);
			if (user.Email == null)
				throw new ItemNotFoundException();
			using MD5 md5 = MD5.Create();
			string hash = Convert
				.ToHexString(md5.ComputeHash(Encoding.ASCII.GetBytes(user.Email)))
				.ToLower();
			try
			{
				HttpClient client = clientFactory.CreateClient();
				HttpResponseMessage response = await client.GetAsync(
					$"https://www.gravatar.com/avatar/{hash}.jpg?d=404&s=250"
				);
				response.EnsureSuccessStatusCode();
				return await response.Content.ReadAsStreamAsync();
			}
			catch
			{
				throw new ItemNotFoundException();
			}
		}

		public async Task SetUserImage(Guid userId, Stream? image)
		{
			if (image == null)
			{
				try
				{
					File.Delete($"/metadata/user/{userId}.webp");
				}
				catch { }
				return;
			}
			using SKCodec codec = SKCodec.Create(image);
			SKImageInfo info = codec.Info;
			info.ColorType = SKColorType.Rgba8888;
			using SKBitmap original = SKBitmap.Decode(codec, info);
			using SKBitmap ret = original.Resize(new SKSizeI(250, 250), SKFilterQuality.High);
			Directory.CreateDirectory("/metadata/user");
			await _WriteTo(ret, $"/metadata/user/{userId}.webp", 75);
		}
	}
}