using Microsoft.Extensions.DependencyInjection;

namespace Hyperbar.Widget.MediaController.Windows;

public class MediaControllerHandler(IPublisher publisher,
    IServiceScopeProvider<MediaController> scopeProvider,
    ICache<MediaController, MediaControllerViewModel> cache) :
    INotificationHandler<Create<MediaController>>,
    INotificationHandler<Remove<MediaController>>
{
    public async Task Handle(Create<MediaController> notification,
        CancellationToken cancellationToken)
    {
        if (notification.Value is MediaController mediaController &&
            scopeProvider.TryGet(mediaController, out IServiceScope? serviceScope) &&
            serviceScope?.ServiceProvider.GetService<IFactory<MediaController, MediaControllerViewModel?>>() 
                is IFactory<MediaController, MediaControllerViewModel?> factory &&
            factory.Create(mediaController) is MediaControllerViewModel viewModel)
        {
            cache.Add(mediaController, viewModel);
            await publisher.PublishAsync(new Create<MediaControllerViewModel>(viewModel), cancellationToken);
        }
    }

    public async Task Handle(Remove<MediaController> notification, CancellationToken cancellationToken)
    {
        if (notification.Value is MediaController mediaController &&
            cache.TryGetValue(mediaController, out MediaControllerViewModel? viewModel) &&
            viewModel is not null)
        {
            await publisher.PublishAsync(new Remove<MediaControllerViewModel>(viewModel), cancellationToken);
            cache.Remove(mediaController);
        }
    }
}