path: root/Software/Visual_Studio/FSE/Modules/Tango.FSE.PPCConsole/ViewModels/FileSystemViewVM.cs

using MaterialDesignThemes.Wpf;
using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Input;
using Tango.Core.Commands;
using Tango.FileSystem;
using Tango.FSE.Common;
using Tango.FSE.Common.Connection;
using Tango.FSE.Common.FileSystem;
using static Tango.SharedUI.Controls.NavigationControl;

namespace Tango.FSE.PPCConsole.ViewModels
{
    public class FileSystemViewVM : FSEViewModel, INavigationViewModel
    {
        private FileSystemItem _currentItem;
        public FileSystemItem CurrentItem
        {
            get { return _currentItem; }
            set { _currentItem = value; RaisePropertyChangedAuto(); OnCurrentItemChanged(); }
        }

        private String _currentPath;
        public String CurrentPath
        {
            get { return _currentPath; }
            set { _currentPath = value; RaisePropertyChangedAuto(); }
        }

        private List<DriveItem> _drives;
        public List<DriveItem> Drives
        {
            get { return _drives; }
            set { _drives = value; RaisePropertyChangedAuto(); }
        }

        public ObservableCollection<FileSystemItem> SelectedItems { get; set; }
        public ObservableCollection<FileSystemHandler> FileSystemHandlers { get; set; }

        public RelayCommand NavigateCommand { get; set; }
        public RelayCommand<FileSystemItem> OpenItemCommand { get; set; }
        public RelayCommand BackCommand { get; set; }
        public RelayCommand<String> NavigateSpecialFolderCommand { get; set; }
        public RelayCommand<String> NavigateToFolderCommand { get; set; }
        public RelayCommand<IList<FileSystemItem>> DeleteCommand { get; set; }
        public RelayCommand<List<FileSystemItem>> DropCommand { get; set; }
        public RelayCommand<List<DragItem>> DragCommand { get; set; }
        public RelayCommand<FileSystemHandler> DeleteFileSystemHandlerCommand { get; set; }
        public RelayCommand<FileSystemHandler> OpenFileSystemHandlerDestinationCommand { get; set; }
        public RelayCommand<FileSystemHandler> RetryFailedFileSystemHandlerCommand { get; set; }
        public RelayCommand<List<FileSystemItem>> CopyPasteCommand { get; set; }
        public RelayCommand<List<FileSystemItem>> CutPasteCommand { get; set; }
        public RelayCommand<List<FileSystemItem>> DownloadCommand { get; set; }
        public RelayCommand<FileSystemItem> RenameCommand { get; set; }


        public FileSystemViewVM()
        {
            SelectedItems = new ObservableCollection<FileSystemItem>();
            FileSystemHandlers = new ObservableCollection<FileSystemHandler>();
            NavigateCommand = new RelayCommand(NavigateToCurrentPath);
            OpenItemCommand = new RelayCommand<FileSystemItem>(OpenFileSystemItem);
            BackCommand = new RelayCommand(NavigateBack, () => !(CurrentItem is FolderItem) || !(CurrentItem as FolderItem).IsRoot);
            NavigateSpecialFolderCommand = new RelayCommand<string>(NavigateToSpecialFolder);
            NavigateToFolderCommand = new RelayCommand<string>(async (x) => await Navigate(x));
            DeleteCommand = new RelayCommand<IList<FileSystemItem>>(DeleteSelectedItems);
            DragCommand = new RelayCommand<List<DragItem>>(OnItemsDraggedOut);
            DeleteFileSystemHandlerCommand = new RelayCommand<FileSystemHandler>(DeleteFileSystemHandler);
            OpenFileSystemHandlerDestinationCommand = new RelayCommand<FileSystemHandler>(OpenFileSystemHandlerDestination);
            RetryFailedFileSystemHandlerCommand = new RelayCommand<FileSystemHandler>(RetryFailedFileSystemHandler);
            CopyPasteCommand = new RelayCommand<List<FileSystemItem>>((items) => PasteItems(items, false));
            CutPasteCommand = new RelayCommand<List<FileSystemItem>>((items) => PasteItems(items, true));
            DownloadCommand = new RelayCommand<List<FileSystemItem>>(DownloadSelectedItems);
            RenameCommand = new RelayCommand<FileSystemItem>(RenameFileSystemItem);
        }

        private async void NavigateBack()
        {
            if (CurrentItem.Path.Length == 3)
            {
                await Navigate(null);
            }
            else
            {
                String parent = Path.GetDirectoryName(CurrentItem.Path);
                await Navigate(parent);
            }
        }

        public override void OnApplicationStarted()
        {
            base.OnApplicationStarted();
            MachineProvider.MachineConnected += MachineProvider_MachineConnected;
        }

        public override void OnApplicationReady()
        {
            base.OnApplicationReady();
        }

        private async void MachineProvider_MachineConnected(object sender, MachineConnectedEventArgs e)
        {
            if (e.DifferentFromPrevious)
            {
                await Navigate(null);
            }
        }

        private async void NavigateToCurrentPath()
        {
            await Navigate(CurrentPath);
        }

        private async void OpenFileSystemItem(FileSystemItem item)
        {
            if (item == null) return;

            if (item.Type == FileSystemItemType.Folder || item.Type == FileSystemItemType.Drive)
            {
                await Navigate(item.Path);
            }
            else if (item.Type == FileSystemItemType.File)
            {
                //TODO: Download/Open file?...
            }
        }

        private async void DeleteSelectedItems(IList<FileSystemItem> items)
        {
            if (items != null && items.Count > 0)
            {
                if (await NotificationProvider.ShowWarningQuestion($"Are you sure you want to delete {(items.Count == 1 ? $"'{items.First().Name}'" : $"the {items.Count} selected files/folders")}?", "DELETE"))
                {
                    using (var task = NotificationProvider.PushTaskItem("Removing..."))
                    {
                        int remainingItems = items.Count;

                        foreach (var item in items)
                        {
                            task.UpdateProgress($"Removing '{item.Name}'...");

                            try
                            {
                                remainingItems--;
                                await FileSystemProvider.Delete(item);
                            }
                            catch (Exception ex)
                            {
                                LogManager.Log(ex, $"Could not remove '{item.Name}'.");

                                if (remainingItems > 0)
                                {
                                    if (!await NotificationProvider.ShowWarningQuestion($"Could not remove '{item.Name}'.\n{ex.FlattenMessage()}\nDo you wish to continue removing the remaining items?"))
                                    {
                                        break;
                                    }
                                }
                                else
                                {
                                    await NotificationProvider.ShowError($"Could not remove '{item.Name}'.\n{ex.FlattenMessage()}");
                                }
                            }
                        }
                    }

                    NavigateToCurrentPath();
                }
            }
        }

        private async void OnItemsDraggedOut(List<DragItem> items)
        {
            foreach (var item in items.Where(x => x.FileSystemItem.Type != FileSystemItemType.Drive))
            {
                Debug.WriteLine($"Dropped out: {item.FileSystemItem.Name} => {item.Destination}");

                if (File.Exists(Path.Combine(item.Destination, item.FileSystemItem.Name)) || Directory.Exists(Path.Combine(item.Destination, item.FileSystemItem.Name)))
                {
                    if (!await NotificationProvider.ShowWarningQuestion($"'{item.FileSystemItem.Name}' already exists on '{Path.GetDirectoryName(item.Destination)}'. Do you want to overwrite?"))
                    {
                        continue;
                    }
                }

                var handler = await FileSystemProvider.Download(item.FileSystemItem, item.Destination);
                FileSystemHandlers.Insert(0, handler);
            }
        }

        private async void DeleteFileSystemHandler(FileSystemHandler handler)
        {
            if (handler.Status != FileSystemHandlerStatus.Completed && handler.Status != FileSystemHandlerStatus.Failed)
            {
                if (await NotificationProvider.ShowWarningQuestion($"This item is currently {handler.Status}. Do you wish to abort and delete this item?"))
                {
                    try
                    {
                        handler.Abort();
                    }
                    catch { }

                    FileSystemHandlers.Remove(handler);
                }
            }
            else
            {
                FileSystemHandlers.Remove(handler);
            }
        }

        private void OpenFileSystemHandlerDestination(FileSystemHandler handler)
        {
            String destination = String.Empty;

            if (File.Exists(handler.Destination) || Directory.Exists(handler.Destination))
            {
                destination = handler.Destination;
                Process.Start("explorer.exe", string.Format("/select,\"{0}\"", destination));
            }
            else
            {
                destination = Path.GetDirectoryName(handler.Destination);
                Process.Start("explorer.exe", destination);
            }
        }

        private async void RetryFailedFileSystemHandler(FileSystemHandler handler)
        {
            if (handler.Status == FileSystemHandlerStatus.Failed)
            {
                var newHandler = await FileSystemProvider.Download(handler.FileSystemItem, Path.GetDirectoryName(handler.Destination));
                FileSystemHandlers.Replace(handler, newHandler);
            }
        }

        private async void NavigateToSpecialFolder(string folder)
        {
            Environment.SpecialFolder specialFolder = (Environment.SpecialFolder)Enum.Parse(typeof(Environment.SpecialFolder), folder);
            await Navigate(null, specialFolder);
        }

        private async Task Navigate(String path, Environment.SpecialFolder? specialFolder = null)
        {
            try
            {
                IsFree = false;

                Mouse.OverrideCursor = Cursors.AppStarting;

                if (path != null)
                {
                    CurrentItem = await FileSystemProvider.GetFolder(path) as FileSystemItem;
                }
                else if (specialFolder != null)
                {
                    CurrentItem = await FileSystemProvider.GetSpecialFolder(specialFolder.Value) as FileSystemItem;
                }
                else
                {
                    CurrentItem = await FileSystemProvider.GetThisPC() as FileSystemItem;
                    try
                    {
                        Drives = (CurrentItem as IFileSystemContainer).Items.Cast<DriveItem>().ToList();
                    }
                    catch (Exception ex)
                    {
                        LogManager.Log(ex, "Error setting file system drives menu.");
                    }
                }
            }
            catch (Exception ex)
            {
                IsFree = true;
                Mouse.OverrideCursor = null;
                await NotificationProvider.ShowError($"Error navigating to the specified path.\n{ex.FlattenMessage()}");
            }
            finally
            {
                IsFree = true;
                Mouse.OverrideCursor = null;
            }
        }

        private async void PasteItems(List<FileSystemItem> items, bool move = false)
        {
            using (var task = NotificationProvider.PushTaskItem("Please wait..."))
            {
                int remainingItems = items.Count;

                foreach (var item in items)
                {
                    Debug.WriteLine($"{(move ? "Cut" : "Copy")} Paste Item '{item.Name}' To '{CurrentItem.Name}'.");

                    try
                    {
                        remainingItems--;

                        if (move)
                        {
                            task.UpdateProgress($"Moving '{item.Name}'...");
                            await FileSystemProvider.Move(item, CurrentItem);
                        }
                        else
                        {
                            task.UpdateProgress($"Copying '{item.Name}'...");
                            await FileSystemProvider.Copy(item, CurrentItem);
                        }
                    }
                    catch (Exception ex)
                    {
                        string operation = move ? "move" : "copy";

                        LogManager.Log(ex, $"Could not {operation} '{item.Name}'.");

                        if (remainingItems > 0)
                        {
                            if (!await NotificationProvider.ShowWarningQuestion($"Could not {operation} '{item.Name}'.\n{ex.FlattenMessage()}\nDo you wish to continue with the remaining items?"))
                            {
                                break;
                            }
                        }
                        else
                        {
                            await NotificationProvider.ShowError($"Could not {operation} '{item.Name}'.\n{ex.FlattenMessage()}");
                        }
                    }
                }

                NavigateToCurrentPath();
            }
        }

        private async void DownloadSelectedItems(List<FileSystemItem> items)
        {
            var result = await StorageProvider.SelectFolder("Select download destination folder");
            if (result)
            {
                Debug.WriteLine($"Download to {result.SelectedItem}");
            }
        }

        private async void RenameFileSystemItem(FileSystemItem item)
        {
            if (item.Type != FileSystemItemType.Drive)
            {
                var result = await NotificationProvider.ShowInputBox(
                    "Rename",
                    $"Please enter a new {(item.Type == FileSystemItemType.File ? "file" : "folder")} name and press 'ENTER'.",
                    PackIconKind.Rename, item.Name,
                    $"{(item.Type == FileSystemItemType.File ? "file" : "folder")} name",
                    100,
                    "RENAME");

                if (result.Confirmed && result.Input != item.Name)
                {
                    try
                    {
                        using (NotificationProvider.PushTaskItem("Renaming..."))
                        {
                            await FileSystemProvider.Rename(item, result.Input);
                            item.Path = Path.Combine(Path.GetDirectoryName(item.Path), result.Input);
                        }
                    }
                    catch (Exception ex)
                    {
                        LogManager.Log(ex, $"Error renaming '{item.Path}' to {result.Input}.");
                        await NotificationProvider.ShowError($"Error renaming '{item.Name}'.\n{ex.FlattenMessage()}");
                    }
                }
            }
        }

        private void OnCurrentItemChanged()
        {
            CurrentPath = CurrentItem.Path;
            InvalidateRelayCommands();
        }
    }
}