ErsatzTV/ErsatzTV/Pages/FFmpegEditor.razor

@page "/ffmpeg/{Id:int}"
@page "/ffmpeg/add"
@using Microsoft.Extensions.Caching.Memory
@using ErsatzTV.Application.Resolutions
@using ErsatzTV.Application.FFmpegProfiles
@using System.Runtime.InteropServices
@using ErsatzTV.Core.FFmpeg
@using ErsatzTV.FFmpeg.Format
@implements IDisposable
@inject NavigationManager _navigationManager
@inject ILogger<FFmpegEditor> _logger
@inject ISnackbar _snackbar
@inject IMediator _mediator
@inject IMemoryCache _memoryCache
@inject PersistentComponentState ApplicationState

<MudContainer MaxWidth="MaxWidth.ExtraLarge" Class="pt-8">
    @if (_editContext is not null)
    {
        <EditForm EditContext="_editContext" OnSubmit="@HandleSubmitAsync">
            <FluentValidationValidator/>
            <MudCard>
                <MudCardHeader>
                    <CardHeaderContent>
                        <MudText Typo="Typo.h5">@(IsEdit ? "Edit FFmpeg Profile" : "Add FFmpeg Profile")</MudText>
                    </CardHeaderContent>
                </MudCardHeader>
                <MudCardContent>
                    <MudGrid>
                        <MudItem xs="12">
                            <MudGrid Spacing="4" Justify="Justify.Center">
                                <MudItem>
                                    <MudText Typo="Typo.h6">General</MudText>
                                    <MudTextField Label="Name" @bind-Value="_model.Name" For="@(() => _model.Name)"/>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudTextField Label="Thread Count" @bind-Value="@_model.ThreadCount" For="@(() => _model.ThreadCount)"/>
                                    </MudElement>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudSelect Label="Preferred Resolution" @bind-Value="_model.Resolution" For="@(() => _model.Resolution)">
                                            @foreach (ResolutionViewModel resolution in _resolutions)
                                            {
                                                <MudSelectItem Value="@resolution">@resolution.Name</MudSelectItem>
                                            }
                                        </MudSelect>
                                    </MudElement>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudSelect Label="Scaling Behavior" @bind-Value="_model.ScalingBehavior" For="@(() => _model.ScalingBehavior)">
                                            <MudSelectItem Value="@ScalingBehavior.ScaleAndPad">Scale and Pad</MudSelectItem>
                                            <MudSelectItem Value="@ScalingBehavior.Stretch">Stretch</MudSelectItem>
                                            <MudSelectItem Value="@ScalingBehavior.Crop">Crop</MudSelectItem>
                                        </MudSelect>
                                    </MudElement>
                                </MudItem>
                                <MudItem>
                                    <MudText Typo="Typo.h6">Video</MudText>
                                    <MudSelect Label="Format" @bind-Value="_model.VideoFormat" For="@(() => _model.VideoFormat)">
                                        <MudSelectItem Value="@FFmpegProfileVideoFormat.H264">h264</MudSelectItem>
                                        <MudSelectItem Value="@FFmpegProfileVideoFormat.Hevc">hevc</MudSelectItem>
                                        <MudSelectItem Value="@FFmpegProfileVideoFormat.Mpeg2Video">mpeg-2</MudSelectItem>
                                    </MudSelect>
                                    <MudSelect Label="Profile"
                                               @bind-Value="_model.VideoProfile"
                                               For="@(() => _model.VideoProfile)"
                                               Disabled="@(_model.VideoFormat != FFmpegProfileVideoFormat.H264 || (_model.HardwareAcceleration != HardwareAccelerationKind.Nvenc && _model.HardwareAcceleration != HardwareAccelerationKind.None))">
                                        <MudSelectItem Value="@VideoProfile.Main">main</MudSelectItem>
                                        <MudSelectItem Value="@VideoProfile.High">high</MudSelectItem>
                                    </MudSelect>
                                    <MudSelect Label="Bit Depth" @bind-Value="_model.BitDepth" For="@(() => _model.BitDepth)">
                                        <MudSelectItem Value="@FFmpegProfileBitDepth.EightBit">8-bit</MudSelectItem>
                                        <MudSelectItem Value="@FFmpegProfileBitDepth.TenBit">10-bit</MudSelectItem>
                                    </MudSelect>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudTextField Label="Bitrate" @bind-Value="_model.VideoBitrate" For="@(() => _model.VideoBitrate)" Adornment="Adornment.End" AdornmentText="kBit/s"/>
                                    </MudElement>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudTextField Label="Buffer Size" @bind-Value="_model.VideoBufferSize" For="@(() => _model.VideoBufferSize)" Adornment="Adornment.End" AdornmentText="kBit"/>
                                    </MudElement>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudSelect Label="Hardware Acceleration" @bind-Value="_model.HardwareAcceleration" For="@(() => _model.HardwareAcceleration)">
                                            @foreach (HardwareAccelerationKind hwAccel in _hardwareAccelerationKinds)
                                            {
                                                <MudSelectItem Value="@hwAccel">@hwAccel</MudSelectItem>
                                            }
                                        </MudSelect>
                                    </MudElement>
                                    @if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
                                    {
                                        @if (_model.HardwareAcceleration == HardwareAccelerationKind.Vaapi)
                                        {
                                            <MudElement HtmlTag="div" Class="mt-3">
                                                <MudSelect Disabled="@(_model.HardwareAcceleration != HardwareAccelerationKind.Vaapi)" Label="VAAPI Driver" @bind-Value="_model.VaapiDriver" For="@(() => _model.VaapiDriver)">
                                                    @foreach (VaapiDriver driver in Enum.GetValues<VaapiDriver>())
                                                    {
                                                        <MudSelectItem Value="@driver">@driver</MudSelectItem>
                                                    }
                                                </MudSelect>
                                            </MudElement>
                                        }

                                        @if (_model.HardwareAcceleration is HardwareAccelerationKind.Vaapi or HardwareAccelerationKind.Qsv)
                                        {
                                            <MudElement HtmlTag="div" Class="mt-3">
                                                <MudSelect Disabled="@(_model.HardwareAcceleration != HardwareAccelerationKind.Vaapi && _model.HardwareAcceleration != HardwareAccelerationKind.Qsv)"
                                                           Label="@(_model.HardwareAcceleration == HardwareAccelerationKind.Vaapi ? "VAAPI Device" : "QSV Device")"
                                                           @bind-Value="_model.VaapiDevice"
                                                           For="@(() => _model.VaapiDevice)">
                                                    @foreach (string device in _vaapiDevices)
                                                    {
                                                        <MudSelectItem Value="@device">@device</MudSelectItem>
                                                    }
                                                </MudSelect>
                                            </MudElement>
                                        }
                                    }
                                    @if (_model.HardwareAcceleration == HardwareAccelerationKind.Qsv)
                                    {
                                        <MudElement HtmlTag="div" Class="mt-3">
                                            <MudTextField Disabled="@(_model.HardwareAcceleration != HardwareAccelerationKind.Qsv)" Label="QSV Extra Hardware Frames" @bind-Value="_model.QsvExtraHardwareFrames" For="@(() => _model.QsvExtraHardwareFrames)"/>
                                        </MudElement>
                                    }
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudCheckBox Label="Normalize Frame Rate" @bind-Checked="@_model.NormalizeFramerate" For="@(() => _model.NormalizeFramerate)"/>
                                    </MudElement>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudCheckBox Label="Auto Deinterlace Video" @bind-Checked="@_model.DeinterlaceVideo" For="@(() => _model.DeinterlaceVideo)"/>
                                    </MudElement>
                                </MudItem>
                                <MudItem>
                                    <MudText Typo="Typo.h6">Audio</MudText>
                                    <MudSelect Label="Format" @bind-Value="_model.AudioFormat" For="@(() => _model.AudioFormat)">
                                        <MudSelectItem Value="@FFmpegProfileAudioFormat.Aac">aac</MudSelectItem>
                                        <MudSelectItem Value="@FFmpegProfileAudioFormat.Ac3">ac3</MudSelectItem>
                                    </MudSelect>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudTextField Label="Bitrate" @bind-Value="_model.AudioBitrate" For="@(() => _model.AudioBitrate)" Adornment="Adornment.End" AdornmentText="kBit/s"/>
                                    </MudElement>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudTextField Label="Buffer Size" @bind-Value="_model.AudioBufferSize" For="@(() => _model.AudioBufferSize)" Adornment="Adornment.End" AdornmentText="kBit"/>
                                    </MudElement>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudTextField Label="Channels" @bind-Value="_model.AudioChannels" For="@(() => _model.AudioChannels)"/>
                                    </MudElement>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudTextField Label="Sample Rate" @bind-Value="_model.AudioSampleRate" For="@(() => _model.AudioSampleRate)" Adornment="Adornment.End" AdornmentText="kHz"/>
                                    </MudElement>
                                    <MudElement HtmlTag="div" Class="mt-3">
                                        <MudSelect Label="Normalize Loudness" @bind-Value="_model.NormalizeLoudnessMode" For="@(() => _model.NormalizeLoudnessMode)">
                                            <MudSelectItem Value="@NormalizeLoudnessMode.Off">Off</MudSelectItem>
                                            <MudSelectItem Value="@NormalizeLoudnessMode.LoudNorm">loudnorm</MudSelectItem>
                                        </MudSelect>
                                    </MudElement>
                                </MudItem>
                            </MudGrid>
                        </MudItem>
                    </MudGrid>
                </MudCardContent>
                <MudCardActions>
                    <MudButton ButtonType="ButtonType.Submit" Variant="Variant.Filled" Color="Color.Primary">
                        @(IsEdit ? "Save Changes" : "Add Profile")
                    </MudButton>
                </MudCardActions>
            </MudCard>
        </EditForm>
    }
</MudContainer>

@code {
    private readonly CancellationTokenSource _cts = new();

    [Parameter]
    public int Id { get; set; }

    private FFmpegProfileEditViewModel _model = new();
    private EditContext _editContext;
    private ValidationMessageStore _messageStore;

    private List<ResolutionViewModel> _resolutions = new();
    private List<HardwareAccelerationKind> _hardwareAccelerationKinds = new();
    private List<string> _vaapiDevices = new();
    private PersistingComponentStateSubscription _persistingSubscription;

    public void Dispose()
    {
        _persistingSubscription.Dispose();

        _cts.Cancel();
        _cts.Dispose();
    }

    protected override Task OnInitializedAsync()
    {
        _persistingSubscription = ApplicationState.RegisterOnPersisting(PersistData);

        return base.OnInitializedAsync();
    }

    protected override async Task OnParametersSetAsync()
    {
        if (!ApplicationState.TryTakeFromJson("_resolutions", out List<ResolutionViewModel> restoredResolutions))
        {
            _resolutions = await _mediator.Send(new GetAllResolutions(), _cts.Token);
        }
        else
        {
            _resolutions = restoredResolutions;
        }

        if (!ApplicationState.TryTakeFromJson("_hardwareAccelerationKinds", out List<HardwareAccelerationKind> restoredHardwareAccelerationKinds))
        {
            _hardwareAccelerationKinds = await _mediator.Send(new GetSupportedHardwareAccelerationKinds(), _cts.Token);
        }
        else
        {
            _hardwareAccelerationKinds = restoredHardwareAccelerationKinds;
        }

        if (IsEdit)
        {
            if (!ApplicationState.TryTakeFromJson("_model", out FFmpegProfileEditViewModel restoredProfile))
            {
                Option<FFmpegProfileViewModel> maybeProfile = await _mediator.Send(new GetFFmpegProfileById(Id), _cts.Token);
                foreach (FFmpegProfileViewModel profile in maybeProfile)
                {
                    _model = new FFmpegProfileEditViewModel(profile);
                }

                if (maybeProfile.IsNone)
                {
                    _navigationManager.NavigateTo("404");
                }
            }
            else
            {
                _model = restoredProfile;
            }
        }
        else
        {
            _model = new FFmpegProfileEditViewModel(await _mediator.Send(new NewFFmpegProfile(), _cts.Token));
        }

        if (!_hardwareAccelerationKinds.Contains(_model.HardwareAcceleration))
        {
            _model.HardwareAcceleration = HardwareAccelerationKind.None;
        }

        _editContext = new EditContext(_model);
        _messageStore = new ValidationMessageStore(_editContext);

        if (!_memoryCache.TryGetValue("ffmpeg.render_devices", out List<string> vaapiDevices))
        {
            vaapiDevices = new List<string> { "/dev/dri/renderD128" };
        }

        _vaapiDevices = vaapiDevices.OrderBy(s => s).ToList();
    }

    private Task PersistData()
    {
        ApplicationState.PersistAsJson("_model", _model);
        ApplicationState.PersistAsJson("_resolutions", _resolutions);
        ApplicationState.PersistAsJson("_hardwareAccelerationKinds", _hardwareAccelerationKinds);

        return Task.CompletedTask;
    }

    private bool IsEdit => Id != 0;

    private async Task HandleSubmitAsync()
    {
        _messageStore.Clear();
        if (_editContext.Validate())
        {
            Seq<BaseError> errorMessage = IsEdit ? (await _mediator.Send(_model.ToUpdate(), _cts.Token)).LeftToSeq() : (await _mediator.Send(_model.ToCreate(), _cts.Token)).LeftToSeq();

            errorMessage.HeadOrNone().Match(
                error =>
                {
                    _snackbar.Add("Unexpected error saving ffmpeg profile");
                    _logger.LogError("Unexpected error saving ffmpeg profile: {Error}", error.Value);
                },
                () => _navigationManager.NavigateTo("/ffmpeg"));
        }
    }

}