FileTime2/src/AppCommon/FileTime.App.Core/Services/UserCommandHandler/ToolUserCommandHandlerService.cs

using System.Diagnostics;
using DeclarativeProperty;
using FileTime.App.ContainerSizeScanner;
using FileTime.App.Core.UserCommand;
using FileTime.App.Core.ViewModels;
using FileTime.App.Search;
using FileTime.Core.ContentAccess;
using FileTime.Core.Enums;
using FileTime.Core.Interactions;
using FileTime.Core.Models;
using FileTime.Core.Timeline;
using FileTime.Providers.Remote;
using FileTime.Server.Common;
using FileTime.Server.Common.Connections.SignalR;
using Microsoft.Extensions.Logging;

namespace FileTime.App.Core.Services.UserCommandHandler;

public class ToolUserCommandHandlerService : UserCommandHandlerServiceBase
{
    private readonly ISystemClipboardService _systemClipboardService;
    private readonly IUserCommunicationService _userCommunicationService;
    private readonly ISearchContentProvider _searchContentProvider;
    private readonly IItemNameConverterService _itemNameConverterService;
    private readonly ITimelessContentProvider _timelessContentProvider;
    private readonly IUserCommandHandlerService _userCommandHandlerService;
    private readonly IContentAccessorFactory _contentAccessorFactory;
    private readonly IContainerSizeScanProvider _containerSizeScanProvider;
    private readonly IProgramsService _programsService;
    private readonly IContentProviderRegistry _contentProviderRegistry;
    private readonly IServiceProvider _serviceProvider;
    private readonly ILogger<ToolUserCommandHandlerService> _logger;
    private IDeclarativeProperty<IContainer?>? _currentLocation;
    private IDeclarativeProperty<IItemViewModel?>? _currentSelectedItem;
    private ITabViewModel? _currentSelectedTab;

    public ToolUserCommandHandlerService(
        IAppState appState,
        ISystemClipboardService systemClipboardService,
        IUserCommunicationService userCommunicationService,
        ISearchContentProvider searchContentProvider,
        IItemNameConverterService itemNameConverterService,
        ITimelessContentProvider timelessContentProvider,
        IUserCommandHandlerService userCommandHandlerService,
        IContentAccessorFactory contentAccessorFactory,
        IContainerSizeScanProvider containerSizeScanProvider,
        IProgramsService programsService,
        IContentProviderRegistry contentProviderRegistry,
        IServiceProvider serviceProvider,
        ILogger<ToolUserCommandHandlerService> logger) : base(appState)
    {
        _systemClipboardService = systemClipboardService;
        _userCommunicationService = userCommunicationService;
        _searchContentProvider = searchContentProvider;
        _itemNameConverterService = itemNameConverterService;
        _timelessContentProvider = timelessContentProvider;
        _userCommandHandlerService = userCommandHandlerService;
        _contentAccessorFactory = contentAccessorFactory;
        _containerSizeScanProvider = containerSizeScanProvider;
        _programsService = programsService;
        _contentProviderRegistry = contentProviderRegistry;
        _serviceProvider = serviceProvider;
        _logger = logger;
        SaveCurrentLocation(l => _currentLocation = l);
        SaveCurrentSelectedItem(i => _currentSelectedItem = i);
        SaveSelectedTab(t => _currentSelectedTab = t);

        AddCommandHandler(new IUserCommandHandler[]
        {
            new TypeUserCommandHandler<AddRemoteContentProviderCommand>(AddRemoteContentProvider),
            new TypeUserCommandHandler<OpenInDefaultFileExplorerCommand>(OpenInDefaultFileExplorer),
            new TypeUserCommandHandler<CopyNativePathCommand>(CopyNativePath),
            new TypeUserCommandHandler<CopyBase64Command>(CopyBase64),
            new TypeUserCommandHandler<EditCommand>(Edit),
            new TypeUserCommandHandler<SearchCommand>(Search),
            new TypeUserCommandHandler<ScanSizeCommand>(ScanSize),
            new TypeUserCommandHandler<SortItemsCommand>(SortItems),
        });
    }

    private async Task AddRemoteContentProvider()
    {
        var containerNameInput = new TextInputElement("Path");
        var providerName = new TextInputElement("Provider name")
        {
            Value = "remote_" + Guid.NewGuid().ToString("N")[..8]
        };
        var inputs = new IInputElement[] {containerNameInput, providerName};
        var result = await _userCommunicationService.ReadInputs(inputs);

        if (!result) return;

        var path = containerNameInput.Value!;

        Func<Task<IRemoteConnection>>? connection = null;
        if (path.StartsWith("http"))
        {
            connection = async () => await SignalRConnection.GetOrCreateForAsync(path, providerName.Value);
        }

        if (connection is null)
        {
            _userCommunicationService.ShowToastMessage("Unknown path type");
            return;
        }

        if (string.IsNullOrWhiteSpace(providerName.Value))
        {
            _userCommunicationService.ShowToastMessage("Provider name can not be empty");
            return;
        }

        var remoteContentProvider = new RemoteContentProvider(
            _timelessContentProvider,
            _serviceProvider,
            connection,
            "local",
            providerName.Value);

        _contentProviderRegistry.AddContentProvider(remoteContentProvider);

        await _userCommandHandlerService.HandleCommandAsync(
            new OpenContainerCommand(new AbsolutePath(_timelessContentProvider, remoteContentProvider)));
        
        await remoteContentProvider.InitializeChildren();
    }

    private Task Edit()
    {
        if (_currentSelectedTab?.CurrentSelectedItem.Value?.BaseItem is not IElement {NativePath: { } filePath})
            return Task.CompletedTask;

        var getNext = false;
        while (true)
        {
            string? execPath = null;
            try
            {
                var editorProgram = _programsService.GetEditorProgram(getNext);
                if (editorProgram is null)
                {
                    break;
                }

                if (editorProgram.Path is { } executablePath)
                {
                    execPath = executablePath;
                    if (string.IsNullOrWhiteSpace(editorProgram.Arguments))
                    {
                        Process.Start(executablePath, "\"" + filePath + "\"");
                    }
                    else
                    {
                        var parts = editorProgram.Arguments.Split("%%1");
                        var arguments = string.Join("%%1", parts.Select(p => p.Replace("%1", "\"" + filePath + "\""))).Replace("%%1", "%1");
                        Process.Start(executablePath, arguments);
                    }
                }

                //TODO: else
                break;
            }
            catch (System.ComponentModel.Win32Exception e)
            {
                _logger.LogError(e, "Error while running editor program, possible the executable path does not exists. {ExecutablePath}", execPath);
            }
            catch (Exception e)
            {
                _logger.LogError(e, "Unknown error while running editor program");
            }

            getNext = true;
        }

        //TODO: else
        return Task.CompletedTask;
    }

    private async Task ScanSize()
    {
        if (_currentLocation?.Value is null) return;

        var searchTask = _containerSizeScanProvider.StartSizeScan(_currentLocation.Value);
        var openContainerCommand = new OpenContainerCommand(new AbsolutePath(_timelessContentProvider, searchTask.SizeSizeScanContainer));
        await _userCommandHandlerService.HandleCommandAsync(openContainerCommand);
    }

    private async Task SortItems(SortItemsCommand sortItemsCommand)
    {
        if (_currentSelectedTab is null) return;

        await _currentSelectedTab.Tab.Ordering.SetValue(sortItemsCommand.Ordering);
    }

    private async Task CopyBase64()
    {
        var item = _currentSelectedItem?.Value?.BaseItem;
        if (item?.Type != AbsolutePathType.Element || item is not IElement element) return;

        var contentReader = await _contentAccessorFactory.GetContentReaderFactory(element.Provider).CreateContentReaderAsync(element);
        using var ms = new MemoryStream();
        while (true)
        {
            //TODO handle large files
            var data = await contentReader.ReadBytesAsync(1048576);
            if (data.Length == 0) break;
            await ms.WriteAsync(data);
        }

        var base64Hash = Convert.ToBase64String(ms.ToArray());
        await _systemClipboardService.CopyToClipboardAsync(base64Hash);
    }

    private async Task Search(SearchCommand searchCommand)
    {
        if (_currentLocation?.Value is null) return;

        var searchQuery = searchCommand.SearchText;
        if (string.IsNullOrEmpty(searchQuery))
        {
            var title = searchCommand.SearchType switch
            {
                SearchType.NameContains => "Search by Name",
                SearchType.NameRegex => "Search by Name (Regex)",
                _ => throw new ArgumentOutOfRangeException()
            };
            var containerNameInput = new TextInputElement(title);
            await _userCommunicationService.ReadInputs(containerNameInput);

            if (containerNameInput.Value is not null)
            {
                searchQuery = containerNameInput.Value;
            }
        }

        //TODO proper error message
        if (string.IsNullOrWhiteSpace(searchQuery)) return;

        ISearchMatcher searchMatcher = searchCommand.SearchType switch
        {
            SearchType.NameContains => new NameContainsMatcher(_itemNameConverterService, searchQuery),
            SearchType.NameRegex => new RegexMatcher(searchQuery),
            _ => throw new ArgumentOutOfRangeException()
        };

        var searchTask = await _searchContentProvider.StartSearchAsync(searchMatcher, _currentLocation.Value);
        var openContainerCommand = new OpenContainerCommand(new AbsolutePath(_timelessContentProvider, searchTask.SearchContainer));
        await _userCommandHandlerService.HandleCommandAsync(openContainerCommand);
    }

    private async Task CopyNativePath()
    {
        if (_currentSelectedItem?.Value?.BaseItem?.NativePath is null) return;
        await _systemClipboardService.CopyToClipboardAsync(_currentSelectedItem.Value.BaseItem.NativePath.Path);
    }

    private Task OpenInDefaultFileExplorer()
    {
        if (_currentLocation?.Value?.NativePath is null) return Task.CompletedTask;
        Process.Start("explorer.exe", "\"" + _currentLocation.Value.NativePath.Path + "\"");
        return Task.CompletedTask;
    }
}