Aaru/Aaru.Gui/ViewModels/Tabs/AtaInfoViewModel.cs

// /***************************************************************************
// Aaru Data Preservation Suite
// ----------------------------------------------------------------------------
//
// Filename       : AtaInfoViewModel.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : GUI view models.
//
// --[ Description ] ----------------------------------------------------------
//
//     View model and code for the ATA information tab.
//
// --[ 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.Collections.Generic;
using System.IO;
using System.Reactive;
using Aaru.Decoders.ATA;
using Avalonia.Controls;
using JetBrains.Annotations;
using ReactiveUI;

namespace Aaru.Gui.ViewModels.Tabs;

public sealed class AtaInfoViewModel : ViewModelBase
{
    readonly byte[] _ata;
    readonly byte[] _atapi;
    readonly Window _view;

    public AtaInfoViewModel([CanBeNull] byte[] ataIdentify, byte[] atapiIdentify,
                            AtaErrorRegistersChs? ataMcptError, Window view)
    {
        SaveAtaBinaryCommand = ReactiveCommand.Create(ExecuteSaveAtaBinaryCommand);
        SaveAtaTextCommand   = ReactiveCommand.Create(ExecuteSaveAtaTextCommand);

        _ata   = ataIdentify;
        _atapi = atapiIdentify;
        _view  = view;

        if(ataIdentify   == null &&
           atapiIdentify == null)
            return;

        if(ataIdentify != null)
        {
            AtaMcptVisible = true;
            AtaMcptChecked = ataMcptError.HasValue;
            AtaOrAtapiText = "ATA IDENTIFY DEVICE";

            if(ataMcptError.HasValue)
            {
                switch(ataMcptError.Value.DeviceHead & 0x7)
                {
                    case 0:
                        AtaMcptText = "Device reports incorrect media card type";

                        break;
                    case 1:
                        AtaMcptText = "Device contains a Secure Digital card";

                        break;
                    case 2:
                        AtaMcptText = "Device contains a MultiMediaCard ";

                        break;
                    case 3:
                        AtaMcptText = "Device contains a Secure Digital I/O card";

                        break;
                    case 4:
                        AtaMcptText = "Device contains a Smart Media card";

                        break;
                    default:
                        AtaMcptText =
                            $"Device contains unknown media card type {ataMcptError.Value.DeviceHead & 0x07}";

                        break;
                }

                AtaMcptWriteProtectionChecked = (ataMcptError.Value.DeviceHead & 0x08) == 0x08;

                ushort specificData = (ushort)((ataMcptError.Value.CylinderHigh * 0x100) +
                                               ataMcptError.Value.CylinderLow);

                AtaMcptSpecificDataText = $"Card specific data: 0x{specificData:X4}";
            }

            AtaIdentifyText = Identify.Prettify(_ata);
        }
        else
        {
            AtaOrAtapiText  = "ATA PACKET IDENTIFY DEVICE";
            AtaIdentifyText = Identify.Prettify(_atapi);
        }
    }

    public string                      AtaIdentifyText               { get; }
    public string                      AtaMcptText                   { get; }
    public string                      AtaMcptSpecificDataText       { get; }
    public bool                        AtaMcptChecked                { get; }
    public bool                        AtaMcptWriteProtectionChecked { get; }
    public bool                        AtaMcptVisible                { get; }
    public ReactiveCommand<Unit, Unit> SaveAtaBinaryCommand          { get; }
    public ReactiveCommand<Unit, Unit> SaveAtaTextCommand            { get; }

    public string AtaOrAtapiText { get; }

    async void ExecuteSaveAtaBinaryCommand()
    {
        var dlgSaveBinary = new SaveFileDialog();

        dlgSaveBinary.Filters.Add(new FileDialogFilter
        {
            Extensions = new List<string>(new[]
            {
                "*.bin"
            }),
            Name = "Binary"
        });

        string result = await dlgSaveBinary.ShowAsync(_view);

        if(result is null)
            return;

        var saveFs = new FileStream(result, FileMode.Create);

        if(_ata != null)
            saveFs.Write(_ata, 0, _ata.Length);
        else if(_atapi != null)
            saveFs.Write(_atapi, 0, _atapi.Length);

        saveFs.Close();
    }

    async void ExecuteSaveAtaTextCommand()
    {
        var dlgSaveText = new SaveFileDialog();

        dlgSaveText.Filters.Add(new FileDialogFilter
        {
            Extensions = new List<string>(new[]
            {
                "*.txt"
            }),
            Name = "Text"
        });

        string result = await dlgSaveText.ShowAsync(_view);

        if(result is null)
            return;

        var saveFs = new FileStream(result, FileMode.Create);
        var saveSw = new StreamWriter(saveFs);
        saveSw.Write(AtaIdentifyText);
        saveFs.Close();
    }
}