using ErsatzTV.Core; using ErsatzTV.Core.Domain; using ErsatzTV.Core.Domain.MediaServer; using ErsatzTV.Core.Errors; using ErsatzTV.Core.Extensions; using ErsatzTV.Core.Interfaces.Metadata; using ErsatzTV.Core.Interfaces.Repositories; using ErsatzTV.Core.MediaSources; using ErsatzTV.Core.Metadata; using Microsoft.Extensions.Logging; namespace ErsatzTV.Scanner.Core.Metadata; public abstract class MediaServerTelevisionLibraryScanner where TConnectionParameters : MediaServerConnectionParameters where TLibrary : Library where TShow : Show where TSeason : Season where TEpisode : Episode where TEtag : MediaServerItemEtag { private readonly ILocalFileSystem _localFileSystem; private readonly ILogger _logger; private readonly IMediator _mediator; private readonly IMetadataRepository _metadataRepository; protected MediaServerTelevisionLibraryScanner( ILocalFileSystem localFileSystem, IMetadataRepository metadataRepository, IMediator mediator, ILogger logger) { _localFileSystem = localFileSystem; _metadataRepository = metadataRepository; _mediator = mediator; _logger = logger; } protected virtual bool ServerSupportsRemoteStreaming => false; protected virtual bool ServerReturnsStatisticsWithMetadata => false; protected async Task> ScanLibrary( IMediaServerTelevisionRepository televisionRepository, TConnectionParameters connectionParameters, TLibrary library, Func getLocalPath, bool deepScan, CancellationToken cancellationToken) { try { return await ScanLibrary( televisionRepository, connectionParameters, library, getLocalPath, GetShowLibraryItems(connectionParameters, library), deepScan, cancellationToken); } catch (Exception ex) when (ex is TaskCanceledException or OperationCanceledException) { return new ScanCanceled(); } } protected abstract IAsyncEnumerable> GetShowLibraryItems( TConnectionParameters connectionParameters, TLibrary library); protected abstract string MediaServerItemId(TShow show); protected abstract string MediaServerItemId(TSeason season); protected abstract string MediaServerItemId(TEpisode episode); protected abstract string MediaServerEtag(TShow show); protected abstract string MediaServerEtag(TSeason season); protected abstract string MediaServerEtag(TEpisode episode); private async Task> ScanLibrary( IMediaServerTelevisionRepository televisionRepository, TConnectionParameters connectionParameters, TLibrary library, Func getLocalPath, IAsyncEnumerable> showEntries, bool deepScan, CancellationToken cancellationToken) { var incomingItemIds = new List(); List existingShows = await televisionRepository.GetExistingShows(library); await foreach ((TShow incoming, int totalShowCount) in showEntries.WithCancellation(cancellationToken)) { if (cancellationToken.IsCancellationRequested) { return new ScanCanceled(); } incomingItemIds.Add(MediaServerItemId(incoming)); decimal percentCompletion = Math.Clamp((decimal)incomingItemIds.Count / totalShowCount, 0, 1); await _mediator.Publish( new ScannerProgressUpdate( library.Id, library.Name, percentCompletion, Array.Empty(), Array.Empty()), cancellationToken); Either> maybeShow = await televisionRepository .GetOrAdd(library, incoming) .BindT(existing => UpdateMetadata(connectionParameters, library, existing, incoming, deepScan)); if (maybeShow.IsLeft) { foreach (BaseError error in maybeShow.LeftToSeq()) { _logger.LogWarning( "Error processing show {Title}: {Error}", incoming.ShowMetadata.Head().Title, error.Value); } continue; } foreach (MediaItemScanResult result in maybeShow.RightToSeq()) { Either scanResult = await ScanSeasons( televisionRepository, library, getLocalPath, result.Item, result.IsUpdated, connectionParameters, GetSeasonLibraryItems(library, connectionParameters, result.Item), deepScan, cancellationToken); foreach (ScanCanceled error in scanResult.LeftToSeq().OfType()) { return error; } await televisionRepository.SetEtag(result.Item, MediaServerEtag(incoming)); Option flagResult = await televisionRepository.FlagNormal(library, result.Item); if (flagResult.IsSome) { result.IsUpdated = true; } if (result.IsAdded || result.IsUpdated) { await _mediator.Publish( new ScannerProgressUpdate( library.Id, null, null, new[] { result.Item.Id }, Array.Empty()), cancellationToken); } } } // trash shows that are no longer present on the media server var fileNotFoundItemIds = existingShows.Map(s => s.MediaServerItemId).Except(incomingItemIds).ToList(); List ids = await televisionRepository.FlagFileNotFoundShows(library, fileNotFoundItemIds); await _mediator.Publish( new ScannerProgressUpdate(library.Id, null, null, ids.ToArray(), Array.Empty()), cancellationToken); await _mediator.Publish( new ScannerProgressUpdate( library.Id, library.Name, 0, Array.Empty(), Array.Empty()), cancellationToken); return Unit.Default; } protected abstract IAsyncEnumerable> GetSeasonLibraryItems( TLibrary library, TConnectionParameters connectionParameters, TShow show); protected abstract IAsyncEnumerable> GetEpisodeLibraryItems( TLibrary library, TConnectionParameters connectionParameters, TShow show, TSeason season); protected abstract Task> GetFullMetadata( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TShow incoming, bool deepScan); protected abstract Task> GetFullMetadata( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TSeason incoming, bool deepScan); protected abstract Task> GetFullMetadata( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TEpisode incoming, bool deepScan); protected virtual Task> GetMediaServerStatistics( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TEpisode incoming) => Task.FromResult(Option.None); protected abstract Task>> GetFullMetadataAndStatistics( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TEpisode incoming); protected abstract Task>> UpdateMetadata( MediaItemScanResult result, ShowMetadata fullMetadata); protected abstract Task>> UpdateMetadata( MediaItemScanResult result, SeasonMetadata fullMetadata); protected abstract Task>> UpdateMetadata( MediaItemScanResult result, EpisodeMetadata fullMetadata); private async Task> ScanSeasons( IMediaServerTelevisionRepository televisionRepository, TLibrary library, Func getLocalPath, TShow show, bool showIsUpdated, TConnectionParameters connectionParameters, IAsyncEnumerable> seasonEntries, bool deepScan, CancellationToken cancellationToken) { var incomingItemIds = new List(); List existingSeasons = await televisionRepository.GetExistingSeasons(library, show); await foreach ((TSeason incoming, int _) in seasonEntries.WithCancellation(cancellationToken)) { incoming.ShowId = show.Id; if (cancellationToken.IsCancellationRequested) { return new ScanCanceled(); } incomingItemIds.Add(MediaServerItemId(incoming)); Either> maybeSeason = await televisionRepository .GetOrAdd(library, incoming) .BindT(existing => UpdateMetadata(connectionParameters, library, existing, incoming, deepScan)); if (maybeSeason.IsLeft) { foreach (BaseError error in maybeSeason.LeftToSeq()) { _logger.LogWarning( "Error processing show {Title} season {SeasonNumber}: {Error}", show.ShowMetadata.Head().Title, incoming.SeasonNumber, error.Value); } continue; } foreach (MediaItemScanResult result in maybeSeason.RightToSeq()) { Either scanResult = await ScanEpisodes( televisionRepository, library, getLocalPath, show, showIsUpdated, result.Item, connectionParameters, GetEpisodeLibraryItems(library, connectionParameters, show, result.Item), deepScan, cancellationToken); foreach (ScanCanceled error in scanResult.LeftToSeq().OfType()) { return error; } await televisionRepository.SetEtag(result.Item, MediaServerEtag(incoming)); Option flagResult = await televisionRepository.FlagNormal(library, result.Item); if (flagResult.IsSome) { result.IsUpdated = true; } result.Item.Show = show; if (result.IsAdded || result.IsUpdated || showIsUpdated) { await _mediator.Publish( new ScannerProgressUpdate( library.Id, null, null, new[] { result.Item.Id }, Array.Empty()), cancellationToken); } } } // trash seasons that are no longer present on the media server var fileNotFoundItemIds = existingSeasons.Map(s => s.MediaServerItemId).Except(incomingItemIds).ToList(); List ids = await televisionRepository.FlagFileNotFoundSeasons(library, fileNotFoundItemIds); await _mediator.Publish( new ScannerProgressUpdate(library.Id, null, null, ids.ToArray(), Array.Empty()), cancellationToken); return Unit.Default; } private async Task> ScanEpisodes( IMediaServerTelevisionRepository televisionRepository, TLibrary library, Func getLocalPath, TShow show, bool showIsUpdated, TSeason season, TConnectionParameters connectionParameters, IAsyncEnumerable> episodeEntries, bool deepScan, CancellationToken cancellationToken) { var incomingItemIds = new List(); List existingEpisodes = await televisionRepository.GetExistingEpisodes(library, season); await foreach ((TEpisode incoming, int _) in episodeEntries.WithCancellation(cancellationToken)) { if (cancellationToken.IsCancellationRequested) { return new ScanCanceled(); } incomingItemIds.Add(MediaServerItemId(incoming)); string localPath = getLocalPath(incoming); if (await ShouldScanItem( televisionRepository, library, show, season, existingEpisodes, incoming, localPath, deepScan) == false) { continue; } incoming.SeasonId = season.Id; Either> maybeEpisode; if (ServerReturnsStatisticsWithMetadata) { maybeEpisode = await televisionRepository .GetOrAdd(library, incoming, deepScan) .MapT( result => { result.LocalPath = localPath; return result; }) .BindT( existing => UpdateMetadataAndStatistics( connectionParameters, library, existing, incoming, deepScan)); } else { maybeEpisode = await televisionRepository .GetOrAdd(library, incoming, deepScan) .MapT( result => { result.LocalPath = localPath; return result; }) .BindT( existing => UpdateMetadata(connectionParameters, library, existing, incoming, deepScan, None)) .BindT( existing => UpdateStatistics( connectionParameters, library, existing, incoming, deepScan, None)) .BindT(UpdateSubtitles); } if (maybeEpisode.IsLeft) { foreach (BaseError error in maybeEpisode.LeftToSeq()) { _logger.LogWarning( "Error processing episode {Title} s{SeasonNumber:00}e{EpisodeNumber:00}: {Error}", show.ShowMetadata.Head().Title, season.SeasonNumber, incoming.EpisodeMetadata.Head().EpisodeNumber, error.Value); } continue; } foreach (MediaItemScanResult result in maybeEpisode.RightToSeq()) { await televisionRepository.SetEtag(result.Item, MediaServerEtag(incoming)); if (_localFileSystem.FileExists(result.LocalPath)) { Option flagResult = await televisionRepository.FlagNormal(library, result.Item); if (flagResult.IsSome) { result.IsUpdated = true; } } else if (ServerSupportsRemoteStreaming) { Option flagResult = await televisionRepository.FlagRemoteOnly(library, result.Item); if (flagResult.IsSome) { result.IsUpdated = true; } } else { Option flagResult = await televisionRepository.FlagUnavailable(library, result.Item); if (flagResult.IsSome) { result.IsUpdated = true; } } if (result.IsAdded || result.IsUpdated || showIsUpdated) { await _mediator.Publish( new ScannerProgressUpdate( library.Id, null, null, new[] { result.Item.Id }, Array.Empty()), cancellationToken); } } } // trash episodes that are no longer present on the media server var fileNotFoundItemIds = existingEpisodes.Map(m => m.MediaServerItemId).Except(incomingItemIds).ToList(); List ids = await televisionRepository.FlagFileNotFoundEpisodes(library, fileNotFoundItemIds); await _mediator.Publish( new ScannerProgressUpdate(library.Id, null, null, ids.ToArray(), Array.Empty()), cancellationToken); return Unit.Default; } private async Task ShouldScanItem( IMediaServerTelevisionRepository televisionRepository, TLibrary library, Show show, Season season, List existingEpisodes, TEpisode incoming, string localPath, bool deepScan) { // deep scan will always pull every episode if (deepScan) { return true; } Option maybeExisting = existingEpisodes.Find(m => m.MediaServerItemId == MediaServerItemId(incoming)); string existingEtag = await maybeExisting.Map(e => e.Etag ?? string.Empty).IfNoneAsync(string.Empty); MediaItemState existingState = await maybeExisting.Map(e => e.State).IfNoneAsync(MediaItemState.Normal); if (existingState is MediaItemState.Unavailable or MediaItemState.FileNotFound && existingEtag == MediaServerEtag(incoming)) { // skip scanning unavailable/file not found items that are unchanged and still don't exist locally if (!_localFileSystem.FileExists(localPath) && !ServerSupportsRemoteStreaming) { return false; } } else if (existingEtag == MediaServerEtag(incoming)) { // item is unchanged, but file does not exist // don't scan, but mark as unavailable if (!_localFileSystem.FileExists(localPath)) { if (ServerSupportsRemoteStreaming) { if (existingState is not MediaItemState.RemoteOnly) { foreach (int id in await televisionRepository.FlagRemoteOnly(library, incoming)) { await _mediator.Publish( new ScannerProgressUpdate(library.Id, null, null, new[] { id }, Array.Empty()), CancellationToken.None); } } } else { if (existingState is not MediaItemState.Unavailable) { foreach (int id in await televisionRepository.FlagUnavailable(library, incoming)) { await _mediator.Publish( new ScannerProgressUpdate(library.Id, null, null, new[] { id }, Array.Empty()), CancellationToken.None); } } } } return false; } if (maybeExisting.IsNone) { _logger.LogDebug( "INSERT: new episode {Show} s{SeasonNumber:00}e{EpisodeNumber:00}", show.ShowMetadata.Head().Title, season.SeasonNumber, incoming.EpisodeMetadata.Head().EpisodeNumber); } else { _logger.LogDebug( "UPDATE: Etag has changed for episode {Show} s{SeasonNumber:00}e{EpisodeNumber:00}", show.ShowMetadata.Head().Title, season.SeasonNumber, incoming.EpisodeMetadata.Head().EpisodeNumber); } return true; } private async Task>> UpdateMetadata( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TShow incoming, bool deepScan) { foreach (ShowMetadata fullMetadata in await GetFullMetadata( connectionParameters, library, result, incoming, deepScan)) { // TODO: move some of this code into this scanner // will have to merge JF, Emby, Plex logic return await UpdateMetadata(result, fullMetadata); } return result; } private async Task>> UpdateMetadata( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TSeason incoming, bool deepScan) { foreach (SeasonMetadata fullMetadata in await GetFullMetadata( connectionParameters, library, result, incoming, deepScan)) { // TODO: move some of this code into this scanner // will have to merge JF, Emby, Plex logic return await UpdateMetadata(result, fullMetadata); } return result; } private async Task>> UpdateMetadataAndStatistics( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TEpisode incoming, bool deepScan) { Option> maybeMetadataAndStatistics = await GetFullMetadataAndStatistics( connectionParameters, library, result, incoming); foreach ((EpisodeMetadata fullMetadata, MediaVersion mediaVersion) in maybeMetadataAndStatistics) { Either> metadataResult = await UpdateMetadata( connectionParameters, library, result, incoming, deepScan, fullMetadata); foreach (BaseError error in metadataResult.LeftToSeq()) { return error; } foreach (MediaItemScanResult r in metadataResult.RightToSeq()) { result = r; } Either> statisticsResult = await UpdateStatistics( connectionParameters, library, result, incoming, deepScan, mediaVersion); foreach (BaseError error in statisticsResult.LeftToSeq()) { return error; } foreach (MediaItemScanResult r in metadataResult.RightToSeq()) { result = r; } } return result; } private async Task>> UpdateMetadata( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TEpisode incoming, bool deepScan, Option maybeFullMetadata) { if (maybeFullMetadata.IsNone) { maybeFullMetadata = await GetFullMetadata(connectionParameters, library, result, incoming, deepScan); } foreach (EpisodeMetadata fullMetadata in maybeFullMetadata) { // TODO: move some of this code into this scanner // will have to merge JF, Emby, Plex logic return await UpdateMetadata(result, fullMetadata); } return result; } private async Task>> UpdateStatistics( TConnectionParameters connectionParameters, TLibrary library, MediaItemScanResult result, TEpisode incoming, bool deepScan, Option maybeMediaVersion) { TEpisode existing = result.Item; if (deepScan || result.IsAdded || MediaServerEtag(existing) != MediaServerEtag(incoming) || existing.MediaVersions.Head().Streams.Count == 0) { // if (maybeMediaVersion.IsNone && _localFileSystem.FileExists(result.LocalPath)) // { // _logger.LogDebug("Refreshing {Attribute} for {Path}", "Statistics", result.LocalPath); // Either refreshResult = // await _localStatisticsProvider.RefreshStatistics( // ffmpegPath, // ffprobePath, // existing, // result.LocalPath); // // foreach (BaseError error in refreshResult.LeftToSeq()) // { // _logger.LogWarning( // "Unable to refresh {Attribute} for media item {Path}. Error: {Error}", // "Statistics", // result.LocalPath, // error.Value); // } // // foreach (bool _ in refreshResult.RightToSeq()) // { // result.IsUpdated = true; // } // } // else // { if (maybeMediaVersion.IsNone) { maybeMediaVersion = await GetMediaServerStatistics( connectionParameters, library, result, incoming); } foreach (MediaVersion mediaVersion in maybeMediaVersion) { if (await _metadataRepository.UpdateStatistics(result.Item, mediaVersion)) { result.IsUpdated = true; } } // } } return result; } private async Task>> UpdateSubtitles( MediaItemScanResult existing) { try { MediaVersion version = existing.Item.GetHeadVersion(); Option maybeMetadata = existing.Item.EpisodeMetadata.HeadOrNone(); foreach (EpisodeMetadata metadata in maybeMetadata) { List subtitles = version.Streams .Filter(s => s.MediaStreamKind is MediaStreamKind.Subtitle or MediaStreamKind.ExternalSubtitle) .Map(Subtitle.FromMediaStream) .ToList(); if (await _metadataRepository.UpdateSubtitles(metadata, subtitles)) { return existing; } } return BaseError.New("Failed to update media server subtitles"); } catch (Exception ex) { return BaseError.New(ex.ToString()); } } }