using System; using System.Collections.Generic; using System.Collections.Immutable; using System.Linq; using System.Threading.Tasks; using API.Constants; using API.Data; using API.Data.Repositories; using API.DTOs; using API.Entities; using API.Entities.Enums; using API.Helpers; using API.Helpers.Builders; using EasyCaching.Core; using Flurl.Http; using Kavita.Common; using Kavita.Common.EnvironmentInfo; using Kavita.Common.Helpers; using Microsoft.Extensions.Logging; namespace API.Services.Plus; #nullable enable public interface IRatingService { Task> GetRatings(int seriesId); } public class RatingService : IRatingService { private readonly IUnitOfWork _unitOfWork; private readonly ILogger _logger; private readonly IEasyCachingProvider _cacheProvider; public const string CacheKey = "rating_"; public RatingService(IUnitOfWork unitOfWork, ILogger logger, IEasyCachingProviderFactory cachingProviderFactory) { _unitOfWork = unitOfWork; _logger = logger; FlurlHttp.ConfigureClient(Configuration.KavitaPlusApiUrl, cli => cli.Settings.HttpClientFactory = new UntrustedCertClientFactory()); _cacheProvider = cachingProviderFactory.GetCachingProvider(EasyCacheProfiles.KavitaPlusRatings); } /// /// Fetches Ratings for a given Series. Will check cache first /// /// /// public async Task> GetRatings(int seriesId) { var cacheKey = CacheKey + seriesId; var results = await _cacheProvider.GetAsync>(cacheKey); if (results.HasValue) { return results.Value; } var license = await _unitOfWork.SettingsRepository.GetSettingAsync(ServerSettingKey.LicenseKey); var series = await _unitOfWork.SeriesRepository.GetSeriesByIdAsync(seriesId, SeriesIncludes.Metadata | SeriesIncludes.Library | SeriesIncludes.Chapters | SeriesIncludes.Volumes); // Don't send any ratings back for Comic libraries as Kavita+ doesn't have any providers for that if (series == null || series.Library.Type == LibraryType.Comic) { await _cacheProvider.SetAsync(cacheKey, ImmutableList.Empty, TimeSpan.FromHours(24)); return ImmutableList.Empty; } var ratings = (await GetRatings(license.Value, series)).ToList(); await _cacheProvider.SetAsync(cacheKey, ratings, TimeSpan.FromHours(24)); _logger.LogDebug("Caching external rating for {Key}", cacheKey); return ratings; } private async Task> GetRatings(string license, Series series) { try { return await (Configuration.KavitaPlusApiUrl + "/api/rating") .WithHeader("Accept", "application/json") .WithHeader("User-Agent", "Kavita") .WithHeader("x-license-key", license) .WithHeader("x-installId", HashUtil.ServerToken()) .WithHeader("x-kavita-version", BuildInfo.Version) .WithHeader("Content-Type", "application/json") .WithTimeout(TimeSpan.FromSeconds(Configuration.DefaultTimeOutSecs)) .PostJsonAsync(new PlusSeriesDtoBuilder(series).Build()) .ReceiveJson>(); } catch (Exception e) { _logger.LogError(e, "An error happened during the request to Kavita+ API"); } return new List(); } }