ersatztv/ErsatzTV/Controllers/Api/TroubleshootController.cs

using System.Threading.Channels;
using ErsatzTV.Application;
using ErsatzTV.Application.MediaItems;
using ErsatzTV.Application.Troubleshooting;
using ErsatzTV.Application.Troubleshooting.Queries;
using ErsatzTV.Core;
using ErsatzTV.Core.Interfaces.FFmpeg;
using ErsatzTV.Core.Interfaces.Metadata;
using ErsatzTV.Core.Interfaces.Troubleshooting;
using MediatR;
using Microsoft.AspNetCore.Mvc;

namespace ErsatzTV.Controllers.Api;

[ApiController]
public class TroubleshootController(
    ChannelWriter<IFFmpegWorkerRequest> channelWriter,
    ILocalFileSystem localFileSystem,
    ITroubleshootingNotifier notifier,
    IMediator mediator) : ControllerBase
{
    [HttpHead("api/troubleshoot/playback.m3u8")]
    [HttpGet("api/troubleshoot/playback.m3u8")]
    public async Task<IActionResult> TroubleshootPlayback(
        [FromQuery]
        int mediaItem,
        [FromQuery]
        int ffmpegProfile,
        [FromQuery]
        List<int> watermark,
        [FromQuery]
        int? subtitleId,
        [FromQuery]
        bool startFromBeginning,
        CancellationToken cancellationToken)
    {
        try
        {
            Either<BaseError, PlayoutItemResult> result = await mediator.Send(
                new PrepareTroubleshootingPlayback(mediaItem, ffmpegProfile, watermark, subtitleId, startFromBeginning),
                cancellationToken);

            if (result.IsLeft)
            {
                return NotFound();
            }

            foreach (var playoutItemResult in result.RightToSeq())
            {
                Either<BaseError, MediaItemInfo> maybeMediaInfo =
                    await mediator.Send(new GetMediaItemInfo(mediaItem), cancellationToken);
                foreach (MediaItemInfo mediaInfo in maybeMediaInfo.RightToSeq())
                {
                    var sessionId = Guid.NewGuid();

                    try
                    {
                        TroubleshootingInfo troubleshootingInfo = await mediator.Send(
                            new GetTroubleshootingInfo(),
                            cancellationToken);

                        // filter ffmpeg profiles
                        troubleshootingInfo.FFmpegProfiles.RemoveAll(p => p.Id != ffmpegProfile);

                        // filter watermarks
                        troubleshootingInfo.Watermarks.RemoveAll(p => !watermark.Contains(p.Id));

                        await channelWriter.WriteAsync(
                            new StartTroubleshootingPlayback(sessionId, playoutItemResult, mediaInfo,
                                troubleshootingInfo),
                            cancellationToken);

                        string playlistFile = Path.Combine(
                            FileSystemLayout.TranscodeFolder,
                            ".troubleshooting",
                            "live.m3u8");

                        while (!localFileSystem.FileExists(playlistFile))
                        {
                            await Task.Delay(TimeSpan.FromMilliseconds(250), cancellationToken);
                            if (cancellationToken.IsCancellationRequested || notifier.IsFailed(sessionId))
                            {
                                break;
                            }
                        }

                        if (!notifier.IsFailed(sessionId))
                        {
                            return Redirect("~/iptv/session/.troubleshooting/live.m3u8");
                        }

                    }
                    finally
                    {
                        notifier.RemoveSession(sessionId);
                    }
                }
            }
        }
        catch (Exception)
        {
            // do nothing
        }

        return NotFound();
    }

    [HttpHead("api/troubleshoot/playback/archive")]
    [HttpGet("api/troubleshoot/playback/archive")]
    public async Task<IActionResult> TroubleshootPlaybackArchive(
        [FromQuery]
        int mediaItem,
        [FromQuery]
        int ffmpegProfile,
        [FromQuery]
        List<int> watermark,
        [FromQuery]
        bool startFromBeginning,
        CancellationToken cancellationToken)
    {
        Option<string> maybeArchivePath = await mediator.Send(
            new ArchiveTroubleshootingResults(mediaItem, ffmpegProfile, watermark, startFromBeginning),
            cancellationToken);

        foreach (string archivePath in maybeArchivePath)
        {
            FileStream fs = System.IO.File.OpenRead(archivePath);
            return File(
                fs,
                "application/zip",
                $"ersatztv-troubleshooting-{DateTimeOffset.Now.ToUnixTimeSeconds()}.zip");
        }

        return NotFound();
    }

}