@page "/media/libraries"
@using ErsatzTV.Application.Emby
@using ErsatzTV.Application.Jellyfin
@using ErsatzTV.Application.Libraries
@using ErsatzTV.Application.MediaSources
@using ErsatzTV.Application.Plex
@using ErsatzTV.Core.Interfaces.Metadata
@using ErsatzTV.Core.Metadata
@using MediatR.Courier
@using PlexLibraryViewModel = ErsatzTV.Application.Libraries.PlexLibraryViewModel
@implements IDisposable
@inject IMediator Mediator
@inject IEntityLocker Locker
@inject ChannelWriter<IScannerBackgroundServiceRequest> ScannerWorkerChannel
@inject ICourier Courier
@inject IScannerProxyService ScannerProxyService

<MudForm Style="max-height: 100%">
    <div class="d-flex flex-column" style="height: 100vh; overflow-x: auto">
        <MudContainer MaxWidth="MaxWidth.ExtraLarge" Class="pt-8">
            <MudText Typo="Typo.h5" Class="mb-2">Libraries</MudText>
            <MudDivider Class="mb-6"/>
            <MudTable Hover="true" Items="_libraries" Dense="true" Class="mb-5">
                <ColGroup>
                    <MudHidden Breakpoint="Breakpoint.Xs">
                        <col/>
                        @if (_showServerNames)
                        {
                            <col/>
                        }
                        <col/>
                        <col/>
                        <col style="width: 180px;"/>
                    </MudHidden>
                </ColGroup>
                <HeaderContent>
                    <MudTh>Library Kind</MudTh>
                    @if (_showServerNames)
                    {
                        <MudTh>Server Name</MudTh>
                    }
                    <MudTh>Library Name</MudTh>
                    <MudTh>Media Kind</MudTh>
                    <MudTh/>
                </HeaderContent>
                <RowTemplate>
                    <MudTd DataLabel="Library Kind">@context.LibraryKind</MudTd>
                    @if (_showServerNames)
                    {
                        <MudTd DataLabel="Server Name">@context.MediaSourceName</MudTd>
                    }
                    <MudTd DataLabel="Library Name">@context.Name</MudTd>
                    <MudTd DataLabel="Media Kind">
                        @switch (context.MediaKind)
                        {
                            case LibraryMediaKind.MusicVideos:
                                @:Music Videos
                                break;
                            case LibraryMediaKind.OtherVideos:
                                @:Other Videos
                                break;
                            case LibraryMediaKind.RemoteStreams:
                                @:Remote Streams
                                break;
                            default:
                                @(context.MediaKind)
                                break;
                        }
                    </MudTd>
                    <MudTd>
                        <div style="align-items: center; display: flex;">
                            @if (IsLibraryLocked(context.Id))
                            {
                                <div style="width: 48px">
                                    @if (_progressByLibrary[context.Id] > 0)
                                    {
                                        <MudText Color="Color.Primary">
                                            @($"{_progressByLibrary[context.Id]} %")
                                        </MudText>
                                    }
                                </div>
                                <div style="align-items: center; display: flex; height: 48px; justify-content: center; width: 48px;">
                                    <MudProgressCircular Color="Color.Primary" Size="Size.Small" Indeterminate="true"/>
                                </div>
                            }
                            else
                            {
                                if (context is PlexLibraryViewModel or EmbyLibraryViewModel or JellyfinLibraryViewModel)
                                {
                                    <MudTooltip Text="Deep Scan Library">
                                        <MudIconButton Icon="@Icons.Material.Filled.FindReplace"
                                                       Disabled="@Locker.IsLibraryLocked(context.Id)"
                                                       OnClick="@(_ => ScanLibrary(context, true))">
                                        </MudIconButton>
                                    </MudTooltip>
                                }
                                else
                                {
                                    <div style="width: 48px"></div>
                                }

                                <MudTooltip Text="Scan Library">
                                    <MudIconButton Icon="@Icons.Material.Filled.Refresh"
                                                   Disabled="@Locker.IsLibraryLocked(context.Id)"
                                                   OnClick="@(_ => ScanLibrary(context))">
                                    </MudIconButton>
                                </MudTooltip>
                            }
                            <MudTooltip Text="Search Library">
                                <MudIconButton Icon="@Icons.Material.Filled.Search"
                                               Href="@($"search?query=library_id%3a{context.Id}")">
                                </MudIconButton>
                            </MudTooltip>
                        </div>
                    </MudTd>
                </RowTemplate>
            </MudTable>
            <MudText Typo="Typo.h5" Class="mt-10 mb-2">External Collections</MudText>
            <MudDivider Class="mb-6"/>
            @if (_externalCollections.Any())
            {
                <MudTable Hover="true" Items="_externalCollections" Dense="true">
                    <ColGroup>
                        <MudHidden Breakpoint="Breakpoint.Xs">
                            <col/>
                            <col style="width: 180px;"/>
                        </MudHidden>
                    </ColGroup>
                    <HeaderContent>
                        <MudTh>Library Kind</MudTh>
                        <MudTh/>
                    </HeaderContent>
                    <RowTemplate>
                        <MudTd>@context.LibraryKind</MudTd>
                        <MudTd>
                            <div style="align-items: center; display: flex;">
                                <div style="width: 48px"></div>
                                @if (AreCollectionsLocked(context.LibraryKind))
                                {
                                    <div style="align-items: center; display: flex; height: 48px; justify-content: center; width: 48px;">
                                        <MudProgressCircular Color="Color.Primary" Size="Size.Small" Indeterminate="true"/>
                                    </div>
                                }
                                else
                                {
                                    if (context.LibraryKind is "Plex" or "Emby" or "Jellyfin")
                                    {
                                        <MudTooltip Text="Deep Scan Collections">
                                            <MudIconButton Icon="@Icons.Material.Filled.FindReplace"
                                                           Disabled="@AreCollectionsLocked(context.LibraryKind)"
                                                           OnClick="@(_ => ScanExternalCollections(context, true))">
                                            </MudIconButton>
                                        </MudTooltip>
                                    }
                                    else
                                    {
                                        <div style="width: 48px"></div>
                                    }
                                    <MudTooltip Text="Scan Collections">
                                        <MudIconButton Icon="@Icons.Material.Filled.Refresh"
                                                       Disabled="@AreCollectionsLocked(context.LibraryKind)"
                                                       OnClick="@(_ => ScanExternalCollections(context))">
                                        </MudIconButton>
                                    </MudTooltip>
                                }
                                <div style="width: 48px"></div>
                            </div>
                        </MudTd>
                    </RowTemplate>
                </MudTable>
            }
        </MudContainer>
    </div>
</MudForm>

@code {
    private CancellationTokenSource _cts;

    private IList<LibraryViewModel> _libraries = new List<LibraryViewModel>();
    private IList<LibraryViewModel> _externalCollections = new List<LibraryViewModel>();
    private Dictionary<int, int> _progressByLibrary = new();
    private bool _showServerNames;

    protected override void OnInitialized()
    {
        Locker.OnLibraryChanged += LockChanged;
        Locker.OnEmbyCollectionsChanged += LockChanged;
        Locker.OnJellyfinCollectionsChanged += LockChanged;
        Locker.OnPlexCollectionsChanged += LockChanged;
        Courier.Subscribe<LibraryScanProgress>(HandleScanProgress);
    }

    protected override async Task OnParametersSetAsync()
    {
        _cts?.Cancel();
        _cts?.Dispose();
        _cts = new CancellationTokenSource();
        var token = _cts.Token;

        try
        {
            _libraries = await Mediator.Send(new GetConfiguredLibraries(), token);
            _showServerNames = _libraries.Any(l => l is PlexLibraryViewModel);
            token.ThrowIfCancellationRequested();

            _externalCollections = await Mediator.Send(new GetExternalCollections(), token);

            foreach (var library in _libraries)
            {
                decimal progress = ScannerProxyService.GetProgress(library.Id).IfNone(0);
                _progressByLibrary[library.Id] = (int)(progress * 100);
            }
        }
        catch (OperationCanceledException)
        {
            // do nothing
        }
    }

    private async Task ScanLibrary(LibraryViewModel library, bool deepScan = false)
    {
        if (Locker.LockLibrary(library.Id))
        {
            var token = _cts?.Token ?? CancellationToken.None;

            switch (library)
            {
                case LocalLibraryViewModel:
                    await ScannerWorkerChannel.WriteAsync(new ForceScanLocalLibrary(library.Id), token);
                    break;
                case PlexLibraryViewModel:
                    await ScannerWorkerChannel.WriteAsync(new SynchronizePlexLibraries(library.MediaSourceId), token);
                    await ScannerWorkerChannel.WriteAsync(new ForceSynchronizePlexLibraryById(library.Id, deepScan), token);
                    await ScannerWorkerChannel.WriteAsync(new SynchronizePlexNetworks(library.Id, true), token);
                    break;
                case JellyfinLibraryViewModel:
                    await ScannerWorkerChannel.WriteAsync(new SynchronizeJellyfinLibraries(library.MediaSourceId), token);
                    await ScannerWorkerChannel.WriteAsync(new ForceSynchronizeJellyfinLibraryById(library.Id, deepScan), token);
                    break;
                case EmbyLibraryViewModel:
                    await ScannerWorkerChannel.WriteAsync(new SynchronizeEmbyLibraries(library.MediaSourceId), token);
                    await ScannerWorkerChannel.WriteAsync(new ForceSynchronizeEmbyLibraryById(library.Id, deepScan), token);
                    break;
            }

            StateHasChanged();
        }
    }

    private async Task ScanExternalCollections(LibraryViewModel library, bool deepScan = false)
    {
        var token = _cts?.Token ?? CancellationToken.None;
        switch (library.LibraryKind.ToLowerInvariant())
        {
            case "emby":
                if (Locker.LockEmbyCollections())
                {
                    await ScannerWorkerChannel.WriteAsync(new SynchronizeEmbyCollections(library.MediaSourceId, true, deepScan), token);
                }

                break;
            case "jellyfin":
                if (Locker.LockJellyfinCollections())
                {
                    await ScannerWorkerChannel.WriteAsync(new SynchronizeJellyfinCollections(library.MediaSourceId, true, deepScan), token);
                }

                break;
            case "plex":
                if (Locker.LockPlexCollections())
                {
                    await ScannerWorkerChannel.WriteAsync(new SynchronizePlexCollections(library.MediaSourceId, true, deepScan), token);
                }

                break;
        }
    }

    private void LockChanged(object sender, EventArgs e) =>
        InvokeAsync(StateHasChanged);

    private bool AreCollectionsLocked(string libraryKind)
    {
        switch (libraryKind.ToLowerInvariant())
        {
            case "emby":
                return Locker.AreEmbyCollectionsLocked();
            case "jellyfin":
                return Locker.AreJellyfinCollectionsLocked();
            case "plex":
                return Locker.ArePlexCollectionsLocked();
        }

        return false;
    }

    private async Task HandleScanProgress(LibraryScanProgress libraryScanProgress, CancellationToken cancellationToken)
    {
        try
        {
            if (_progressByLibrary != null && _progressByLibrary.ContainsKey(libraryScanProgress.LibraryId))
            {
                _progressByLibrary[libraryScanProgress.LibraryId] = (int)(libraryScanProgress.Progress * 100);
                await InvokeAsync(StateHasChanged);
            }
        }
        catch (Exception)
        {
            // ignore
        }
    }

    void IDisposable.Dispose()
    {
        Locker.OnLibraryChanged -= LockChanged;
        Locker.OnEmbyCollectionsChanged -= LockChanged;
        Locker.OnJellyfinCollectionsChanged -= LockChanged;
        Locker.OnPlexCollectionsChanged -= LockChanged;
        Courier.UnSubscribe<LibraryScanProgress>(HandleScanProgress);

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

    private bool IsLibraryLocked(int libraryId)
    {
        if (Locker.IsLibraryLocked(libraryId))
        {
            return true;
        }

        _progressByLibrary[libraryId] = 0;
        return false;
    }

}