using System.Globalization; using ErsatzTV.Application.Libraries; using ErsatzTV.Core; using ErsatzTV.Core.Domain; using ErsatzTV.Core.Errors; using ErsatzTV.Core.Interfaces.Metadata; using ErsatzTV.Core.Interfaces.Repositories; using ErsatzTV.FFmpeg.Runtime; using ErsatzTV.Infrastructure.Data; using ErsatzTV.Infrastructure.Extensions; using Microsoft.EntityFrameworkCore; using Microsoft.Extensions.Logging; namespace ErsatzTV.Application.MediaSources; public class CallLocalLibraryScannerHandler : CallLibraryScannerHandler, IRequestHandler>, IRequestHandler> { private readonly IScannerProxyService _scannerProxyService; public CallLocalLibraryScannerHandler( IDbContextFactory dbContextFactory, IConfigElementRepository configElementRepository, IScannerProxyService scannerProxyService, IRuntimeInfo runtimeInfo, ILogger logger) : base(dbContextFactory, configElementRepository, runtimeInfo, logger) { _scannerProxyService = scannerProxyService; } Task> IRequestHandler>.Handle( ForceScanLocalLibrary request, CancellationToken cancellationToken) => Handle(request, cancellationToken); Task> IRequestHandler>.Handle( ScanLocalLibraryIfNeeded request, CancellationToken cancellationToken) => Handle(request, cancellationToken); private async Task> Handle(IScanLocalLibrary request, CancellationToken cancellationToken) { Validation validation = await Validate(request, cancellationToken); return await validation.Match( parameters => PerformScan(parameters, request, cancellationToken), error => { foreach (ScanIsNotRequired scanIsNotRequired in error.OfType()) { return Task.FromResult>(scanIsNotRequired); } return Task.FromResult>(error.Join()); }); } private async Task> PerformScan( ScanParameters parameters, IScanLocalLibrary request, CancellationToken cancellationToken) { Option maybeScanId = _scannerProxyService.StartScan(request.LibraryId); foreach (var scanId in maybeScanId) { try { var arguments = new List { "scan-local", request.LibraryId.ToString(CultureInfo.InvariantCulture), GetBaseUrl(scanId) }; if (request.ForceScan) { arguments.Add("--force"); } return await base.PerformScan(parameters, arguments, cancellationToken); } finally { _scannerProxyService.EndScan(scanId); } } return BaseError.New($"Library {request.LibraryId} is already scanning"); } protected override async Task> GetLastScan( TvContext dbContext, IScanLocalLibrary request, CancellationToken cancellationToken) { List libraryPaths = await dbContext.LibraryPaths .Filter(lp => lp.LibraryId == request.LibraryId) .ToListAsync(cancellationToken); DateTime minDateTime = libraryPaths.Count != 0 ? libraryPaths.Min(lp => lp.LastScan ?? SystemTime.MinValueUtc) : SystemTime.MaxValueUtc; string libraryName = await dbContext.Libraries .SelectOneAsync(l => l.Id, l => l.Id == request.LibraryId, cancellationToken) .Match(l => l.Name, () => string.Empty); return new Tuple(libraryName, new DateTimeOffset(minDateTime, TimeSpan.Zero)); } protected override bool ScanIsRequired( DateTimeOffset lastScan, int libraryRefreshInterval, IScanLocalLibrary request) { if (lastScan == SystemTime.MaxValueUtc) { return false; } DateTimeOffset nextScan = lastScan + TimeSpan.FromHours(libraryRefreshInterval); return request.ForceScan || libraryRefreshInterval > 0 && nextScan < DateTimeOffset.Now; } }