ersatztv/ErsatzTV.Application/ProgramSchedules/Commands/AddProgramScheduleItemHandler.cs

using System.Linq;
using System.Threading;
using System.Threading.Channels;
using System.Threading.Tasks;
using ErsatzTV.Application.Playouts.Commands;
using ErsatzTV.Core;
using ErsatzTV.Core.Domain;
using ErsatzTV.Infrastructure.Data;
using LanguageExt;
using MediatR;
using Microsoft.EntityFrameworkCore;
using static ErsatzTV.Application.ProgramSchedules.Mapper;

namespace ErsatzTV.Application.ProgramSchedules.Commands
{
    public class AddProgramScheduleItemHandler : ProgramScheduleItemCommandBase,
        IRequestHandler<AddProgramScheduleItem, Either<BaseError, ProgramScheduleItemViewModel>>
    {
        private readonly ChannelWriter<IBackgroundServiceRequest> _channel;
        private readonly IDbContextFactory<TvContext> _dbContextFactory;

        public AddProgramScheduleItemHandler(
            IDbContextFactory<TvContext> dbContextFactory,
            ChannelWriter<IBackgroundServiceRequest> channel)
        {
            _dbContextFactory = dbContextFactory;
            _channel = channel;
        }

        public async Task<Either<BaseError, ProgramScheduleItemViewModel>> Handle(
            AddProgramScheduleItem request,
            CancellationToken cancellationToken)
        {
            await using TvContext dbContext = _dbContextFactory.CreateDbContext();
            Validation<BaseError, ProgramSchedule> validation = await Validate(dbContext, request);
            return await validation.Apply(ps => PersistItem(dbContext, request, ps));
        }

        private async Task<ProgramScheduleItemViewModel> PersistItem(
            TvContext dbContext,
            AddProgramScheduleItem request,
            ProgramSchedule programSchedule)
        {
            int nextIndex = programSchedule.Items.Select(i => i.Index).DefaultIfEmpty(0).Max() + 1;

            ProgramScheduleItem item = BuildItem(programSchedule, nextIndex, request);
            programSchedule.Items.Add(item);

            await dbContext.SaveChangesAsync();

            // rebuild any playouts that use this schedule
            foreach (Playout playout in programSchedule.Playouts)
            {
                await _channel.WriteAsync(new BuildPlayout(playout.Id, true));
            }

            return ProjectToViewModel(item);
        }

        private static Task<Validation<BaseError, ProgramSchedule>> Validate(
            TvContext dbContext,
            AddProgramScheduleItem request) =>
            ProgramScheduleMustExist(dbContext, request.ProgramScheduleId)
                .BindT(programSchedule => PlayoutModeMustBeValid(request, programSchedule));
    }
}