using System; using System.IO; using System.Linq; using System.Threading; using System.Threading.Tasks; using ErsatzTV.Core; using ErsatzTV.Core.Domain; using ErsatzTV.Core.Interfaces.Locking; using ErsatzTV.Core.Interfaces.Plex; using ErsatzTV.Core.Interfaces.Repositories; using ErsatzTV.Core.Plex; using LanguageExt; using MediatR; using Microsoft.Extensions.Logging; using Unit = LanguageExt.Unit; namespace ErsatzTV.Application.Plex.Commands { public class SynchronizePlexLibraryByIdHandler : IRequestHandler>, IRequestHandler> { private readonly IConfigElementRepository _configElementRepository; private readonly IEntityLocker _entityLocker; private readonly ILibraryRepository _libraryRepository; private readonly ILogger _logger; private readonly IMediaSourceRepository _mediaSourceRepository; private readonly IPlexMovieLibraryScanner _plexMovieLibraryScanner; private readonly IPlexSecretStore _plexSecretStore; private readonly IPlexTelevisionLibraryScanner _plexTelevisionLibraryScanner; public SynchronizePlexLibraryByIdHandler( IMediaSourceRepository mediaSourceRepository, IConfigElementRepository configElementRepository, IPlexSecretStore plexSecretStore, IPlexMovieLibraryScanner plexMovieLibraryScanner, IPlexTelevisionLibraryScanner plexTelevisionLibraryScanner, ILibraryRepository libraryRepository, IEntityLocker entityLocker, ILogger logger) { _mediaSourceRepository = mediaSourceRepository; _configElementRepository = configElementRepository; _plexSecretStore = plexSecretStore; _plexMovieLibraryScanner = plexMovieLibraryScanner; _plexTelevisionLibraryScanner = plexTelevisionLibraryScanner; _libraryRepository = libraryRepository; _entityLocker = entityLocker; _logger = logger; } public Task> Handle( ForceSynchronizePlexLibraryById request, CancellationToken cancellationToken) => Handle(request); public Task> Handle( SynchronizePlexLibraryByIdIfNeeded request, CancellationToken cancellationToken) => Handle(request); private Task> Handle(ISynchronizePlexLibraryById request) => Validate(request) .MapT(parameters => Synchronize(parameters).Map(_ => parameters.Library.Name)) .Bind(v => v.ToEitherAsync()); private async Task Synchronize(RequestParameters parameters) { var lastScan = new DateTimeOffset(parameters.Library.LastScan ?? SystemTime.MinValueUtc, TimeSpan.Zero); DateTimeOffset nextScan = lastScan + TimeSpan.FromHours(parameters.LibraryRefreshInterval); if (parameters.ForceScan || nextScan < DateTimeOffset.Now) { switch (parameters.Library.MediaKind) { case LibraryMediaKind.Movies: await _plexMovieLibraryScanner.ScanLibrary( parameters.ConnectionParameters.ActiveConnection, parameters.ConnectionParameters.PlexServerAuthToken, parameters.Library, parameters.FFprobePath); break; case LibraryMediaKind.Shows: await _plexTelevisionLibraryScanner.ScanLibrary( parameters.ConnectionParameters.ActiveConnection, parameters.ConnectionParameters.PlexServerAuthToken, parameters.Library, parameters.FFprobePath); break; } parameters.Library.LastScan = DateTime.UtcNow; await _libraryRepository.UpdateLastScan(parameters.Library); } else { _logger.LogDebug( "Skipping unforced scan of plex media library {Name}", parameters.Library.Name); } _entityLocker.UnlockLibrary(parameters.Library.Id); return Unit.Default; } private async Task> Validate(ISynchronizePlexLibraryById request) => (await ValidateConnection(request), await PlexLibraryMustExist(request), await ValidateLibraryRefreshInterval(), await ValidateFFprobePath()) .Apply( (connectionParameters, plexLibrary, libraryRefreshInterval, ffprobePath) => new RequestParameters( connectionParameters, plexLibrary, request.ForceScan, libraryRefreshInterval, ffprobePath )); private Task> ValidateConnection( ISynchronizePlexLibraryById request) => PlexMediaSourceMustExist(request) .BindT(MediaSourceMustHaveActiveConnection) .BindT(MediaSourceMustHaveToken); private Task> PlexMediaSourceMustExist( ISynchronizePlexLibraryById request) => _mediaSourceRepository.GetPlexByLibraryId(request.PlexLibraryId) .Map( v => v.ToValidation( $"Plex media source for library {request.PlexLibraryId} does not exist.")); private Validation MediaSourceMustHaveActiveConnection( PlexMediaSource plexMediaSource) { Option maybeConnection = plexMediaSource.Connections.SingleOrDefault(c => c.IsActive); return maybeConnection.Map(connection => new ConnectionParameters(plexMediaSource, connection)) .ToValidation("Plex media source requires an active connection"); } private async Task> MediaSourceMustHaveToken( ConnectionParameters connectionParameters) { Option maybeToken = await _plexSecretStore.GetServerAuthToken(connectionParameters.PlexMediaSource.ClientIdentifier); return maybeToken.Map(token => connectionParameters with { PlexServerAuthToken = token }) .ToValidation("Plex media source requires a token"); } private Task> PlexLibraryMustExist( ISynchronizePlexLibraryById request) => _mediaSourceRepository.GetPlexLibrary(request.PlexLibraryId) .Map(v => v.ToValidation($"Plex library {request.PlexLibraryId} does not exist.")); private Task> ValidateLibraryRefreshInterval() => _configElementRepository.GetValue(ConfigElementKey.LibraryRefreshInterval) .FilterT(lri => lri > 0) .Map(lri => lri.ToValidation("Library refresh interval is invalid")); private Task> ValidateFFprobePath() => _configElementRepository.GetValue(ConfigElementKey.FFprobePath) .FilterT(File.Exists) .Map( ffprobePath => ffprobePath.ToValidation("FFprobe path does not exist on the file system")); private record RequestParameters( ConnectionParameters ConnectionParameters, PlexLibrary Library, bool ForceScan, int LibraryRefreshInterval, string FFprobePath); private record ConnectionParameters(PlexMediaSource PlexMediaSource, PlexConnection ActiveConnection) { public PlexServerAuthToken PlexServerAuthToken { get; set; } } } }