Aaru/Aaru.Gui/ViewModels/Windows/MainWindowViewModel.cs

// /***************************************************************************
// Aaru Data Preservation Suite
// ----------------------------------------------------------------------------
//
// Filename       : MainWindowViewModel.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : GUI view models.
//
// --[ Description ] ----------------------------------------------------------
//
//     View model and code for the main window.
//
// --[ License ] --------------------------------------------------------------
//
//     This program is free software: you can redistribute it and/or modify
//     it under the terms of the GNU General public License as
//     published by the Free Software Foundation, either version 3 of the
//     License, or (at your option) any later version.
//
//     This program is distributed in the hope that it will be useful,
//     but WITHOUT ANY WARRANTY; without even the implied warranty of
//     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//     GNU General public License for more details.
//
//     You should have received a copy of the GNU General public License
//     along with this program.  If not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2011-2022 Natalia Portillo
// ****************************************************************************/

using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.IO;
using System.Linq;
using System.Reactive;
using System.Threading.Tasks;
using Aaru.CommonTypes;
using Aaru.CommonTypes.Enums;
using Aaru.CommonTypes.Interfaces;
using Aaru.CommonTypes.Interop;
using Aaru.CommonTypes.Structs.Devices.SCSI;
using Aaru.Console;
using Aaru.Core;
using Aaru.Core.Media.Info;
using Aaru.Database;
using Aaru.Devices;
using Aaru.Gui.Models;
using Aaru.Gui.ViewModels.Dialogs;
using Aaru.Gui.ViewModels.Panels;
using Aaru.Gui.Views.Dialogs;
using Aaru.Gui.Views.Panels;
using Aaru.Gui.Views.Windows;
using Avalonia;
using Avalonia.Controls;
using Avalonia.Controls.ApplicationLifetimes;
using Avalonia.Media.Imaging;
using Avalonia.Platform;
using JetBrains.Annotations;
using MessageBox.Avalonia;
using MessageBox.Avalonia.Enums;
using ReactiveUI;
using DeviceInfo = Aaru.Core.Devices.Info.DeviceInfo;
using ImageInfo = Aaru.Gui.Views.Panels.ImageInfo;
using Partition = Aaru.Gui.Views.Panels.Partition;
using PlatformID = Aaru.CommonTypes.Interop.PlatformID;

namespace Aaru.Gui.ViewModels.Windows;

public sealed class MainWindowViewModel : ViewModelBase
{
    readonly IAssetLoader     _assets;
    readonly DevicesRootModel _devicesRoot;
    readonly Bitmap           _ejectIcon;
    readonly Bitmap           _genericFolderIcon;
    readonly Bitmap           _genericHddIcon;
    readonly Bitmap           _genericOpticalIcon;
    readonly Bitmap           _genericTapeIcon;
    readonly ImagesRootModel  _imagesRoot;
    readonly Bitmap           _removableIcon;
    readonly Bitmap           _sdIcon;
    readonly Bitmap           _usbIcon;
    readonly MainWindow       _view;
    Views.Dialogs.Console     _console;
    object                    _contentPanel;
    bool                      _devicesSupported;
    object                    _treeViewSelectedItem;

    public MainWindowViewModel(MainWindow view)
    {
        AboutCommand                = ReactiveCommand.Create(ExecuteAboutCommand);
        EncodingsCommand            = ReactiveCommand.Create(ExecuteEncodingsCommand);
        PluginsCommand              = ReactiveCommand.Create(ExecutePluginsCommand);
        StatisticsCommand           = ReactiveCommand.Create(ExecuteStatisticsCommand);
        ExitCommand                 = ReactiveCommand.Create(ExecuteExitCommand);
        SettingsCommand             = ReactiveCommand.Create(ExecuteSettingsCommand);
        ConsoleCommand              = ReactiveCommand.Create(ExecuteConsoleCommand);
        OpenCommand                 = ReactiveCommand.Create(ExecuteOpenCommand);
        CalculateEntropyCommand     = ReactiveCommand.Create(ExecuteCalculateEntropyCommand);
        VerifyImageCommand          = ReactiveCommand.Create(ExecuteVerifyImageCommand);
        ChecksumImageCommand        = ReactiveCommand.Create(ExecuteChecksumImageCommand);
        ConvertImageCommand         = ReactiveCommand.Create(ExecuteConvertImageCommand);
        CreateSidecarCommand        = ReactiveCommand.Create(ExecuteCreateSidecarCommand);
        ViewImageSectorsCommand     = ReactiveCommand.Create(ExecuteViewImageSectorsCommand);
        DecodeImageMediaTagsCommand = ReactiveCommand.Create(ExecuteDecodeImageMediaTagsCommand);
        RefreshDevicesCommand       = ReactiveCommand.Create(ExecuteRefreshDevicesCommand);
        _view                       = view;
        TreeRoot                    = new ObservableCollection<RootModel>();
        _assets                     = AvaloniaLocator.Current.GetService<IAssetLoader>();
        ContentPanel                = Greeting;

        _imagesRoot = new ImagesRootModel
        {
            Name = "Images"
        };

        TreeRoot.Add(_imagesRoot);

        switch(DetectOS.GetRealPlatformID())
        {
            case PlatformID.Win32NT:
            case PlatformID.Linux:
            case PlatformID.FreeBSD:
                _devicesRoot = new DevicesRootModel
                {
                    Name = "Devices"
                };

                TreeRoot.Add(_devicesRoot);
                DevicesSupported = true;

                break;
        }

        if(_assets == null)
            return;

        _genericHddIcon =
            new Bitmap(_assets.Open(new Uri("avares://Aaru.Gui/Assets/Icons/oxygen/32x32/drive-harddisk.png")));

        _genericOpticalIcon =
            new Bitmap(_assets.Open(new Uri("avares://Aaru.Gui/Assets/Icons/oxygen/32x32/drive-optical.png")));

        _genericTapeIcon =
            new Bitmap(_assets.Open(new Uri("avares://Aaru.Gui/Assets/Icons/oxygen/32x32/media-tape.png")));

        _genericFolderIcon =
            new Bitmap(_assets.Open(new Uri("avares://Aaru.Gui/Assets/Icons/oxygen/32x32/inode-directory.png")));

        _usbIcon =
            new
                Bitmap(_assets.Open(new Uri("avares://Aaru.Gui/Assets/Icons/oxygen/32x32/drive-removable-media-usb.png")));

        _removableIcon =
            new Bitmap(_assets.Open(new Uri("avares://Aaru.Gui/Assets/Icons/oxygen/32x32/drive-removable-media.png")));

        _sdIcon =
            new Bitmap(_assets.Open(new Uri("avares://Aaru.Gui/Assets/Icons/oxygen/32x32/media-flash-sd-mmc.png")));

        _ejectIcon = new Bitmap(_assets.Open(new Uri("avares://Aaru.Gui/Assets/Icons/oxygen/32x32/media-eject.png")));
    }

    public string FileLabel                 => "_File";
    public string OpenLabel                 => "_Open";
    public string SettingsLabel             => "_Settings";
    public string ExitLabel                 => "E_xit";
    public string DevicesLabel              => "_Devices";
    public string RefreshDevicesLabel       => "_Refresh";
    public string WindowLabel               => "_Window";
    public string ConsoleLabel              => "_Console";
    public string HelpLabel                 => "_Help";
    public string EncodingsLabel            => "_Encodings";
    public string PluginsLabel              => "_Plugins";
    public string StatisticsLabel           => "_Statistics";
    public string AboutLabel                => "_About";
    public string RefreshDevicesFullLabel   => "_Refresh devices";
    public string CloseAllImagesLabel       => "_Close all images";
    public string CalculateEntropyLabel     => "Calculate entropy";
    public string VerifyImageLabel          => "Verify";
    public string ChecksumImageLabel        => "Checksum";
    public string CreateSidecarLabel        => "Create CICM XML sidecar...";
    public string ViewImageSectorsLabel     => "View sectors";
    public string DecodeImageMediaTagsLabel => "Decode media tags";

    public bool DevicesSupported
    {
        get => _devicesSupported;
        set => this.RaiseAndSetIfChanged(ref _devicesSupported, value);
    }

    public bool NativeMenuSupported =>
        NativeMenu.GetIsNativeMenuExported((Application.Current?.ApplicationLifetime as
                                                IClassicDesktopStyleApplicationLifetime)?.MainWindow);

    [NotNull]
    public string Greeting => "Welcome to Aaru!";
    public ObservableCollection<RootModel> TreeRoot                    { get; }
    public ReactiveCommand<Unit, Unit>     AboutCommand                { get; }
    public ReactiveCommand<Unit, Unit>     ConsoleCommand              { get; }
    public ReactiveCommand<Unit, Unit>     EncodingsCommand            { get; }
    public ReactiveCommand<Unit, Unit>     PluginsCommand              { get; }
    public ReactiveCommand<Unit, Unit>     StatisticsCommand           { get; }
    public ReactiveCommand<Unit, Unit>     ExitCommand                 { get; }
    public ReactiveCommand<Unit, Task>     SettingsCommand             { get; }
    public ReactiveCommand<Unit, Task>     OpenCommand                 { get; }
    public ReactiveCommand<Unit, Unit>     CalculateEntropyCommand     { get; }
    public ReactiveCommand<Unit, Unit>     VerifyImageCommand          { get; }
    public ReactiveCommand<Unit, Unit>     ChecksumImageCommand        { get; }
    public ReactiveCommand<Unit, Unit>     ConvertImageCommand         { get; }
    public ReactiveCommand<Unit, Unit>     CreateSidecarCommand        { get; }
    public ReactiveCommand<Unit, Unit>     ViewImageSectorsCommand     { get; }
    public ReactiveCommand<Unit, Unit>     DecodeImageMediaTagsCommand { get; }
    public ReactiveCommand<Unit, Unit>     RefreshDevicesCommand       { get; }

    public object ContentPanel
    {
        get => _contentPanel;
        set => this.RaiseAndSetIfChanged(ref _contentPanel, value);
    }

    public object TreeViewSelectedItem
    {
        get => _treeViewSelectedItem;
        set
        {
            if(value == _treeViewSelectedItem)
                return;

            this.RaiseAndSetIfChanged(ref _treeViewSelectedItem, value);

            ContentPanel = null;

            switch(value)
            {
                case ImageModel imageModel:
                    ContentPanel = new ImageInfo
                    {
                        DataContext = imageModel.ViewModel
                    };

                    break;
                case PartitionModel partitionModel:
                    ContentPanel = new Partition
                    {
                        DataContext = partitionModel.ViewModel
                    };

                    break;
                case FileSystemModel fileSystemModel:
                    ContentPanel = new FileSystem
                    {
                        DataContext = fileSystemModel.ViewModel
                    };

                    break;
                case SubdirectoryModel subdirectoryModel:
                    ContentPanel = new Subdirectory
                    {
                        DataContext = new SubdirectoryViewModel(subdirectoryModel, _view)
                    };

                    break;
                case DeviceModel deviceModel:
                {
                    if(deviceModel.ViewModel is null)
                    {
                        var dev = Device.Create(deviceModel.Path, out ErrorNumber devErrno);

                        switch(dev)
                        {
                            case null:
                                ContentPanel = $"Error {devErrno} opening device";

                                return;
                            case Devices.Remote.Device remoteDev:
                                Statistics.AddRemote(remoteDev.RemoteApplication, remoteDev.RemoteVersion,
                                                     remoteDev.RemoteOperatingSystem,
                                                     remoteDev.RemoteOperatingSystemVersion,
                                                     remoteDev.RemoteArchitecture);

                                break;
                        }

                        if(dev.Error)
                        {
                            ContentPanel = $"Error {dev.LastError} opening device";

                            return;
                        }

                        var devInfo = new DeviceInfo(dev);

                        deviceModel.ViewModel = new DeviceInfoViewModel(devInfo, _view);

                        if(!dev.IsRemovable)
                            deviceModel.Media.Add(new MediaModel
                            {
                                NonRemovable = true,
                                Name         = "Non-removable device commands not yet implemented"
                            });
                        else
                        {
                            // TODO: Removable non-SCSI?
                            var scsiInfo = new ScsiInfo(dev);

                            if(!scsiInfo.MediaInserted)
                                deviceModel.Media.Add(new MediaModel
                                {
                                    NoMediaInserted = true,
                                    Icon            = _ejectIcon,
                                    Name            = "No media inserted"
                                });
                            else
                            {
                                var mediaResource =
                                    new Uri($"avares://Aaru.Gui/Assets/Logos/Media/{scsiInfo.MediaType}.png");

                                deviceModel.Media.Add(new MediaModel
                                {
                                    DevicePath = deviceModel.Path,
                                    Icon = _assets.Exists(mediaResource) ? new Bitmap(_assets.Open(mediaResource))
                                               : null,
                                    Name      = $"{scsiInfo.MediaType}",
                                    ViewModel = new MediaInfoViewModel(scsiInfo, deviceModel.Path, _view)
                                });
                            }
                        }

                        dev.Close();
                    }

                    ContentPanel = new Views.Panels.DeviceInfo
                    {
                        DataContext = deviceModel.ViewModel
                    };

                    break;
                }
                case MediaModel { NonRemovable: true }:
                    ContentPanel = "Non-removable device commands not yet implemented";

                    break;
                case MediaModel { NoMediaInserted: true }:
                    ContentPanel = "No media inserted";

                    break;
                case MediaModel mediaModel:
                {
                    if(mediaModel.ViewModel != null)
                        ContentPanel = new MediaInfo
                        {
                            DataContext = mediaModel.ViewModel
                        };

                    break;
                }
            }
        }
    }

    void ExecuteCalculateEntropyCommand()
    {
        if(TreeViewSelectedItem is not ImageModel imageModel)
            return;

        var imageEntropyWindow = new ImageEntropy();
        imageEntropyWindow.DataContext = new ImageEntropyViewModel(imageModel.Image, imageEntropyWindow);

        imageEntropyWindow.Closed += (_, _) => imageEntropyWindow = null;

        imageEntropyWindow.Show();
    }

    void ExecuteVerifyImageCommand()
    {
        if(TreeViewSelectedItem is not ImageModel imageModel)
            return;

        var imageVerifyWindow = new ImageVerify();
        imageVerifyWindow.DataContext = new ImageVerifyViewModel(imageModel.Image, imageVerifyWindow);

        imageVerifyWindow.Closed += (_, _) => imageVerifyWindow = null;

        imageVerifyWindow.Show();
    }

    void ExecuteChecksumImageCommand()
    {
        if(TreeViewSelectedItem is not ImageModel imageModel)
            return;

        var imageChecksumWindow = new ImageChecksum();
        imageChecksumWindow.DataContext = new ImageChecksumViewModel(imageModel.Image, imageChecksumWindow);

        imageChecksumWindow.Closed += (_, _) => imageChecksumWindow = null;

        imageChecksumWindow.Show();
    }

    void ExecuteConvertImageCommand()
    {
        if(TreeViewSelectedItem is not ImageModel imageModel)
            return;

        var imageConvertWindow = new ImageConvert();

        imageConvertWindow.DataContext =
            new ImageConvertViewModel(imageModel.Image, imageModel.Path, imageConvertWindow);

        imageConvertWindow.Closed += (_, _) => imageConvertWindow = null;

        imageConvertWindow.Show();
    }

    void ExecuteCreateSidecarCommand()
    {
        if(TreeViewSelectedItem is not ImageModel imageModel)
            return;

        var imageSidecarWindow = new ImageSidecar();

        // TODO: Pass thru chosen default encoding
        imageSidecarWindow.DataContext =
            new ImageSidecarViewModel(imageModel.Image, imageModel.Path, imageModel.Filter.Id, null,
                                      imageSidecarWindow);

        imageSidecarWindow.Show();
    }

    void ExecuteViewImageSectorsCommand()
    {
        if(TreeViewSelectedItem is not ImageModel imageModel)
            return;

        new ViewSector
        {
            DataContext = new ViewSectorViewModel(imageModel.Image)
        }.Show();
    }

    void ExecuteDecodeImageMediaTagsCommand()
    {
        if(TreeViewSelectedItem is not ImageModel imageModel)
            return;

        new DecodeMediaTags
        {
            DataContext = new DecodeMediaTagsViewModel(imageModel.Image)
        }.Show();
    }

    internal void ExecuteAboutCommand()
    {
        var dialog = new About();
        dialog.DataContext = new AboutViewModel(dialog);
        dialog.ShowDialog(_view);
    }

    void ExecuteEncodingsCommand()
    {
        var dialog = new Encodings();
        dialog.DataContext = new EncodingsViewModel(dialog);
        dialog.ShowDialog(_view);
    }

    void ExecutePluginsCommand()
    {
        var dialog = new PluginsDialog();
        dialog.DataContext = new PluginsViewModel(dialog);
        dialog.ShowDialog(_view);
    }

    void ExecuteStatisticsCommand()
    {
        using var ctx = AaruContext.Create(Settings.Settings.LocalDbPath);

        if(!ctx.Commands.Any()     &&
           !ctx.Filesystems.Any()  &&
           !ctx.Filters.Any()      &&
           !ctx.MediaFormats.Any() &&
           !ctx.Medias.Any()       &&
           !ctx.Partitions.Any()   &&
           !ctx.SeenDevices.Any())
        {
            MessageBoxManager.GetMessageBoxStandardWindow("Warning", "There are no statistics.").ShowDialog(_view);

            return;
        }

        var dialog = new StatisticsDialog();
        dialog.DataContext = new StatisticsViewModel(dialog);
        dialog.ShowDialog(_view);
    }

    internal async Task ExecuteSettingsCommand()
    {
        var dialog = new SettingsDialog();
        dialog.DataContext = new SettingsViewModel(dialog, false);
        await dialog.ShowDialog(_view);
    }

    internal void ExecuteExitCommand() =>
        (Application.Current?.ApplicationLifetime as ClassicDesktopStyleApplicationLifetime)?.Shutdown();

    void ExecuteConsoleCommand()
    {
        if(_console is null)
        {
            _console             = new Views.Dialogs.Console();
            _console.DataContext = new ConsoleViewModel(_console);
        }

        _console.Show();
    }

    async Task ExecuteOpenCommand()
    {
        // TODO: Extensions
        var dlgOpenImage = new OpenFileDialog
        {
            Title         = "Choose image to open",
            AllowMultiple = false
        };

        string[] result = await dlgOpenImage.ShowAsync(_view);

        if(result?.Length != 1)
            return;

        var     filtersList = new FiltersList();
        IFilter inputFilter = filtersList.GetFilter(result[0]);

        if(inputFilter == null)
        {
            MessageBoxManager.GetMessageBoxStandardWindow("Error", "Cannot open specified file.", ButtonEnum.Ok,
                                                          Icon.Error);

            return;
        }

        try
        {
            if(ImageFormat.Detect(inputFilter) is not IMediaImage imageFormat)
            {
                MessageBoxManager.GetMessageBoxStandardWindow("Error", "Image format not identified.", ButtonEnum.Ok,
                                                              Icon.Error);

                return;
            }

            AaruConsole.WriteLine("Image format identified by {0} ({1}).", imageFormat.Name, imageFormat.Id);

            try
            {
                ErrorNumber opened = imageFormat.Open(inputFilter);

                if(opened != ErrorNumber.NoError)
                {
                    MessageBoxManager.GetMessageBoxStandardWindow("Error", $"Error {opened} opening image format.",
                                                                  ButtonEnum.Ok, Icon.Error);

                    AaruConsole.ErrorWriteLine("Unable to open image format");
                    AaruConsole.ErrorWriteLine("No error given");

                    return;
                }

                var mediaResource = new Uri($"avares://Aaru.Gui/Assets/Logos/Media/{imageFormat.Info.MediaType}.png");

                var imageModel = new ImageModel
                {
                    Path = result[0],
                    Icon = _assets.Exists(mediaResource)
                               ? new Bitmap(_assets.Open(mediaResource))
                               : imageFormat.Info.XmlMediaType == XmlMediaType.BlockMedia
                                   ? _genericHddIcon
                                   : imageFormat.Info.XmlMediaType == XmlMediaType.OpticalDisc
                                       ? _genericOpticalIcon
                                       : _genericFolderIcon,
                    FileName  = Path.GetFileName(result[0]),
                    Image     = imageFormat,
                    ViewModel = new ImageInfoViewModel(result[0], inputFilter, imageFormat, _view),
                    Filter    = inputFilter
                };

                List<CommonTypes.Partition> partitions = Core.Partitions.GetAll(imageFormat);
                Core.Partitions.AddSchemesToStats(partitions);

                bool         checkRaw = false;
                List<string> idPlugins;
                IFilesystem  plugin;
                PluginBase   plugins = GetPluginBase.Instance;

                if(partitions.Count == 0)
                {
                    AaruConsole.DebugWriteLine("Fs-info command", "No partitions found");

                    checkRaw = true;
                }
                else
                {
                    AaruConsole.WriteLine("{0} partitions found.", partitions.Count);

                    foreach(string scheme in partitions.Select(p => p.Scheme).Distinct().OrderBy(s => s))
                    {
                        // TODO: Add icons to partition schemes
                        var schemeModel = new PartitionSchemeModel
                        {
                            Name = scheme
                        };

                        foreach(CommonTypes.Partition partition in partitions.Where(p => p.Scheme == scheme).
                                                                              OrderBy(p => p.Start))
                        {
                            var partitionModel = new PartitionModel
                            {
                                // TODO: Add icons to partition types
                                Name      = $"{partition.Name} ({partition.Type})",
                                Partition = partition,
                                ViewModel = new PartitionViewModel(partition)
                            };

                            AaruConsole.WriteLine("Identifying filesystem on partition");

                            Core.Filesystems.Identify(imageFormat, out idPlugins, partition);

                            if(idPlugins.Count == 0)
                                AaruConsole.WriteLine("Filesystem not identified");
                            else
                            {
                                AaruConsole.WriteLine($"Identified by {idPlugins.Count} plugins");

                                foreach(string pluginName in idPlugins)
                                    if(plugins.PluginsList.TryGetValue(pluginName, out plugin))
                                    {
                                        plugin.GetInformation(imageFormat, partition, out string information, null);

                                        var fsPlugin = plugin as IReadOnlyFilesystem;

                                        if(fsPlugin != null)
                                        {
                                            ErrorNumber error =
                                                fsPlugin.Mount(imageFormat, partition, null,
                                                               new Dictionary<string, string>(), null);

                                            if(error != ErrorNumber.NoError)
                                                fsPlugin = null;
                                        }

                                        var filesystemModel = new FileSystemModel
                                        {
                                            VolumeName =
                                                plugin.XmlFsType.VolumeName is null ? $"{plugin.XmlFsType.Type}"
                                                    : $"{plugin.XmlFsType.VolumeName} ({plugin.XmlFsType.Type})",
                                            Filesystem         = plugin,
                                            ReadOnlyFilesystem = fsPlugin,
                                            ViewModel          = new FileSystemViewModel(plugin.XmlFsType, information)
                                        };

                                        // TODO: Trap expanding item
                                        if(fsPlugin != null)
                                        {
                                            filesystemModel.Roots.Add(new SubdirectoryModel
                                            {
                                                Name   = "/",
                                                Path   = "",
                                                Plugin = fsPlugin
                                            });

                                            Statistics.AddCommand("ls");
                                        }

                                        Statistics.AddFilesystem(plugin.XmlFsType.Type);
                                        partitionModel.FileSystems.Add(filesystemModel);
                                    }
                            }

                            schemeModel.Partitions.Add(partitionModel);
                        }

                        imageModel.PartitionSchemesOrFileSystems.Add(schemeModel);
                    }
                }

                if(checkRaw)
                {
                    var wholePart = new CommonTypes.Partition
                    {
                        Name   = "Whole device",
                        Length = imageFormat.Info.Sectors,
                        Size   = imageFormat.Info.Sectors * imageFormat.Info.SectorSize
                    };

                    Core.Filesystems.Identify(imageFormat, out idPlugins, wholePart);

                    if(idPlugins.Count == 0)
                        AaruConsole.WriteLine("Filesystem not identified");
                    else
                    {
                        AaruConsole.WriteLine($"Identified by {idPlugins.Count} plugins");

                        foreach(string pluginName in idPlugins)
                            if(plugins.PluginsList.TryGetValue(pluginName, out plugin))
                            {
                                plugin.GetInformation(imageFormat, wholePart, out string information, null);

                                var fsPlugin = plugin as IReadOnlyFilesystem;

                                if(fsPlugin != null)
                                {
                                    ErrorNumber error = fsPlugin.Mount(imageFormat, wholePart, null,
                                                                       new Dictionary<string, string>(), null);

                                    if(error != ErrorNumber.NoError)
                                        fsPlugin = null;
                                }

                                var filesystemModel = new FileSystemModel
                                {
                                    VolumeName = plugin.XmlFsType.VolumeName is null ? $"{plugin.XmlFsType.Type}"
                                                     : $"{plugin.XmlFsType.VolumeName} ({plugin.XmlFsType.Type})",
                                    Filesystem         = plugin,
                                    ReadOnlyFilesystem = fsPlugin,
                                    ViewModel          = new FileSystemViewModel(plugin.XmlFsType, information)
                                };

                                // TODO: Trap expanding item
                                if(fsPlugin != null)
                                {
                                    filesystemModel.Roots.Add(new SubdirectoryModel
                                    {
                                        Name   = "/",
                                        Path   = "",
                                        Plugin = fsPlugin
                                    });

                                    Statistics.AddCommand("ls");
                                }

                                Statistics.AddFilesystem(plugin.XmlFsType.Type);
                                imageModel.PartitionSchemesOrFileSystems.Add(filesystemModel);
                            }
                    }
                }

                Statistics.AddMediaFormat(imageFormat.Format);
                Statistics.AddMedia(imageFormat.Info.MediaType, false);
                Statistics.AddFilter(inputFilter.Name);

                _imagesRoot.Images.Add(imageModel);
            }
            catch(Exception ex)
            {
                MessageBoxManager.GetMessageBoxStandardWindow("Error", "Unable to open image format.", ButtonEnum.Ok,
                                                              Icon.Error);

                AaruConsole.ErrorWriteLine("Unable to open image format");
                AaruConsole.ErrorWriteLine("Error: {0}", ex.Message);
                AaruConsole.DebugWriteLine("Image-info command", "Stack trace: {0}", ex.StackTrace);
            }
        }
        catch(Exception ex)
        {
            MessageBoxManager.GetMessageBoxStandardWindow("Error", "Exception reading file.", ButtonEnum.Ok,
                                                          Icon.Error);

            AaruConsole.ErrorWriteLine($"Error reading file: {ex.Message}");
            AaruConsole.DebugWriteLine("Image-info command", ex.StackTrace);
        }

        Statistics.AddCommand("image-info");
    }

    internal void LoadComplete() => RefreshDevices();

    void ExecuteRefreshDevicesCommand() => RefreshDevices();

    void RefreshDevices()
    {
        if(!DevicesSupported)
            return;

        try
        {
            AaruConsole.WriteLine("Refreshing devices");
            _devicesRoot.Devices.Clear();

            foreach(Devices.DeviceInfo device in Device.ListDevices().Where(d => d.Supported).OrderBy(d => d.Vendor).
                                                        ThenBy(d => d.Model))
            {
                AaruConsole.DebugWriteLine("Main window",
                                           "Found supported device model {0} by manufacturer {1} on bus {2} and path {3}",
                                           device.Model, device.Vendor, device.Bus, device.Path);

                var deviceModel = new DeviceModel
                {
                    Icon = _genericHddIcon,
                    Name = $"{device.Vendor} {device.Model} ({device.Bus})",
                    Path = device.Path
                };

                var dev = Device.Create(device.Path, out _);

                if(dev != null)
                {
                    if(dev is Devices.Remote.Device remoteDev)
                        Statistics.AddRemote(remoteDev.RemoteApplication, remoteDev.RemoteVersion,
                                             remoteDev.RemoteOperatingSystem, remoteDev.RemoteOperatingSystemVersion,
                                             remoteDev.RemoteArchitecture);

                    switch(dev.Type)
                    {
                        case DeviceType.ATAPI:
                        case DeviceType.SCSI:
                            switch(dev.ScsiType)
                            {
                                case PeripheralDeviceTypes.DirectAccess:
                                case PeripheralDeviceTypes.SCSIZonedBlockDevice:
                                case PeripheralDeviceTypes.SimplifiedDevice:
                                    deviceModel.Icon = dev.IsRemovable ? dev.IsUsb ? _usbIcon : _removableIcon
                                                           : _genericHddIcon;

                                    break;
                                case PeripheralDeviceTypes.SequentialAccess:
                                    deviceModel.Icon = _genericTapeIcon;

                                    break;
                                case PeripheralDeviceTypes.OpticalDevice:
                                case PeripheralDeviceTypes.WriteOnceDevice:
                                case PeripheralDeviceTypes.OCRWDevice:
                                    deviceModel.Icon = _removableIcon;

                                    break;
                                case PeripheralDeviceTypes.MultiMediaDevice:
                                    deviceModel.Icon = _genericOpticalIcon;

                                    break;
                            }

                            break;
                        case DeviceType.SecureDigital:
                        case DeviceType.MMC:
                            deviceModel.Icon = _sdIcon;

                            break;
                        case DeviceType.NVMe:
                            deviceModel.Icon = null;

                            break;
                    }

                    dev.Close();
                }

                _devicesRoot.Devices.Add(deviceModel);
            }
        }
        catch(InvalidOperationException ex)
        {
            AaruConsole.ErrorWriteLine(ex.Message);
        }
    }
}