using ErsatzTV.Core; using ErsatzTV.Core.Domain; using ErsatzTV.Core.Interfaces.Emby; using ErsatzTV.Core.Interfaces.Repositories; using ErsatzTV.Core.MediaSources; using Microsoft.Extensions.Logging; namespace ErsatzTV.Scanner.Core.Emby; public class EmbyCollectionScanner : IEmbyCollectionScanner { private readonly IEmbyApiClient _embyApiClient; private readonly IEmbyCollectionRepository _embyCollectionRepository; private readonly ILogger _logger; private readonly IMediator _mediator; public EmbyCollectionScanner( IMediator mediator, IEmbyCollectionRepository embyCollectionRepository, IEmbyApiClient embyApiClient, ILogger logger) { _mediator = mediator; _embyCollectionRepository = embyCollectionRepository; _embyApiClient = embyApiClient; _logger = logger; } public async Task> ScanCollections(string address, string apiKey) { try { // need to call get libraries to find library that contains collections (box sets) await _embyApiClient.GetLibraries(address, apiKey); var incomingItemIds = new List(); // get all collections from db (item id, etag) List existingCollections = await _embyCollectionRepository.GetCollections(); await foreach ((EmbyCollection collection, int _) in _embyApiClient.GetCollectionLibraryItems(address, apiKey)) { incomingItemIds.Add(collection.ItemId); Option maybeExisting = existingCollections.Find(c => c.ItemId == collection.ItemId); // // skip if unchanged (etag) // if (await maybeExisting.Map(e => e.Etag ?? string.Empty).IfNoneAsync(string.Empty) == // collection.Etag) // { // _logger.LogDebug("Emby collection {Name} is unchanged", collection.Name); // continue; // } // add if new if (maybeExisting.IsNone) { _logger.LogDebug("Emby collection {Name} is new", collection.Name); await _embyCollectionRepository.AddCollection(collection); } await SyncCollectionItems(address, apiKey, collection); // save collection etag await _embyCollectionRepository.SetEtag(collection); } // remove missing collections (and remove any lingering tags from those collections) foreach (EmbyCollection collection in existingCollections.Filter(e => !incomingItemIds.Contains(e.ItemId))) { await _embyCollectionRepository.RemoveCollection(collection); } } catch (Exception ex) { _logger.LogWarning(ex, "Failed to get collections from Emby"); return BaseError.New(ex.Message); } return Unit.Default; } private async Task SyncCollectionItems( string address, string apiKey, EmbyCollection collection) { try { // get collection items from Emby IAsyncEnumerable> items = _embyApiClient.GetCollectionItems(address, apiKey, collection.ItemId); List removedIds = await _embyCollectionRepository.RemoveAllTags(collection); // sync tags on items var addedIds = new List(); await foreach ((MediaItem item, int _) in items) { addedIds.Add(await _embyCollectionRepository.AddTag(item, collection)); } _logger.LogDebug("Emby collection {Name} contains {Count} items", collection.Name, addedIds.Count); int[] changedIds = removedIds.Concat(addedIds).Distinct().ToArray(); await _mediator.Publish( new ScannerProgressUpdate(0, null, null, changedIds.ToArray(), Array.Empty()), CancellationToken.None); } catch (Exception ex) { _logger.LogWarning(ex, "Failed to synchronize Emby collection {Name}", collection.Name); } } }