using System.Threading.Channels; using ErsatzTV.Application.Playouts; using ErsatzTV.Application.Search; using ErsatzTV.Core; using ErsatzTV.Core.Domain; using ErsatzTV.Core.Interfaces.Repositories; using ErsatzTV.Core.Scheduling; using ErsatzTV.Infrastructure.Data; using ErsatzTV.Infrastructure.Extensions; using Microsoft.EntityFrameworkCore; namespace ErsatzTV.Application.MediaCollections; public class AddMediaItemToCollectionHandler : IRequestHandler> { private readonly ChannelWriter _channel; private readonly IDbContextFactory _dbContextFactory; private readonly IMediaCollectionRepository _mediaCollectionRepository; private readonly ChannelWriter _searchChannel; public AddMediaItemToCollectionHandler( IDbContextFactory dbContextFactory, IMediaCollectionRepository mediaCollectionRepository, ChannelWriter channel, ChannelWriter searchChannel) { _dbContextFactory = dbContextFactory; _mediaCollectionRepository = mediaCollectionRepository; _channel = channel; _searchChannel = searchChannel; } public async Task> Handle( AddMediaItemToCollection request, CancellationToken cancellationToken) { await using TvContext dbContext = await _dbContextFactory.CreateDbContextAsync(cancellationToken); Validation validation = await Validate(dbContext, request); return await validation.Apply(parameters => ApplyAddMediaItemRequest(dbContext, parameters)); } private async Task ApplyAddMediaItemRequest(TvContext dbContext, Parameters parameters) { parameters.Collection.MediaItems.Add(parameters.MediaItem); if (await dbContext.SaveChangesAsync() > 0) { await _searchChannel.WriteAsync(new ReindexMediaItems([parameters.MediaItem.Id])); // refresh all playouts that use this collection foreach (int playoutId in await _mediaCollectionRepository .PlayoutIdsUsingCollection(parameters.Collection.Id)) { await _channel.WriteAsync(new BuildPlayout(playoutId, PlayoutBuildMode.Refresh)); } } return Unit.Default; } private static async Task> Validate( TvContext dbContext, AddMediaItemToCollection request) => (await CollectionMustExist(dbContext, request), await ValidateMediaItem(dbContext, request)) .Apply((collection, episode) => new Parameters(collection, episode)); private static Task> CollectionMustExist( TvContext dbContext, AddMediaItemToCollection request) => dbContext.Collections .Include(c => c.MediaItems) .SelectOneAsync(c => c.Id, c => c.Id == request.CollectionId) .Map(o => o.ToValidation("Collection does not exist.")); private static Task> ValidateMediaItem( TvContext dbContext, AddMediaItemToCollection request) => dbContext.MediaItems .SelectOneAsync(m => m.Id, e => e.Id == request.MediaItemId) .Map(o => o.ToValidation("MediaItem does not exist")); private sealed record Parameters(Collection Collection, MediaItem MediaItem); }