// /***************************************************************************
// Aaru Data Preservation Suite
// ----------------------------------------------------------------------------
//
// Filename       : ErrorLog.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : Core algorithms.
//
// --[ 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-2023 Natalia Portillo
// ****************************************************************************/

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using Aaru.Decoders.ATA;
using Aaru.Decoders.SCSI;

namespace Aaru.Core.Logging;

/// <summary>Logs errors</summary>
public sealed class ErrorLog
{
    readonly StreamWriter _logSw;

    /// <summary>Initializes the error log</summary>
    /// <param name="outputFile">Output log file</param>
    public ErrorLog(string outputFile)
    {
        if(string.IsNullOrEmpty(outputFile)) return;

        _logSw = new StreamWriter(outputFile, true);

        _logSw.WriteLine(Localization.Core.Start_error_logging_on_0, DateTime.Now);
        _logSw.WriteLine(Localization.Core.Log_section_separator);
        _logSw.Flush();
    }

    /// <summary>Finishes and closes the error log</summary>
    public void Close()
    {
        _logSw.WriteLine(Localization.Core.Log_section_separator);
        _logSw.WriteLine(Localization.Core.End_logging_on_0, DateTime.Now);
        _logSw.Close();
    }

    /// <summary>Register an ATA error after sending a CHS command</summary>
    /// <param name="command">Command</param>
    /// <param name="osError"><c>true</c> if operating system returned an error status instead of the device</param>
    /// <param name="errno">Operating system error number</param>
    /// <param name="registers">Error registers</param>
    public void WriteLine(string command, bool osError, int errno, AtaErrorRegistersChs registers)
    {
        if(osError)
        {
            _logSw.WriteLine(Localization.Core.ATA_command_0_operating_system_error_1, command, errno);
            _logSw.Flush();
        }
        else
        {
            List<string> error  = new();
            List<string> status = new();

            if((registers.Status & 0x01) == 0x01) status.Add("ERR");

            if((registers.Status & 0x02) == 0x02) status.Add("IDX");

            if((registers.Status & 0x04) == 0x04) status.Add("CORR");

            if((registers.Status & 0x08) == 0x08) status.Add("DRQ");

            if((registers.Status & 0x10) == 0x10) status.Add("SRV");

            if((registers.Status & 0x20) == 0x20) status.Add("DF");

            if((registers.Status & 0x40) == 0x40) status.Add("RDY");

            if((registers.Status & 0x80) == 0x80) status.Add("BSY");

            if((registers.Error & 0x01) == 0x01) error.Add("AMNF");

            if((registers.Error & 0x02) == 0x02) error.Add("T0NF");

            if((registers.Error & 0x04) == 0x04) error.Add("ABRT");

            if((registers.Error & 0x08) == 0x08) error.Add("MCR");

            if((registers.Error & 0x10) == 0x10) error.Add("IDNF");

            if((registers.Error & 0x20) == 0x20) error.Add("MC");

            if((registers.Error & 0x40) == 0x40) error.Add("UNC");

            if((registers.Error & 0x80) == 0x80) error.Add("BBK");

            _logSw.WriteLine(Localization.Core.ATA_command_0_error_status_1_error_2,
                             command,
                             string.Join(' ', status),
                             string.Join(' ', error));

            _logSw.Flush();
        }
    }

    /// <summary>Register an ATA error after trying to read using CHS commands</summary>
    /// <param name="cylinder">Cylinder</param>
    /// <param name="head">Head</param>
    /// <param name="sector">Sector</param>
    /// <param name="osError"><c>true</c> if operating system returned an error status instead of the device</param>
    /// <param name="errno">Operating system error number</param>
    /// <param name="registers">Error registers</param>
    public void WriteLine(ushort               cylinder, byte head, byte sector, bool osError, int errno,
                          AtaErrorRegistersChs registers)
    {
        if(osError)
        {
            _logSw.WriteLine(Localization.Core.ATA_reading_CHS_0_1_2_operating_system_error_3,
                             cylinder,
                             head,
                             sector,
                             errno);

            _logSw.Flush();
        }
        else
        {
            List<string> error  = new();
            List<string> status = new();

            if((registers.Status & 0x01) == 0x01) status.Add("ERR");

            if((registers.Status & 0x02) == 0x02) status.Add("IDX");

            if((registers.Status & 0x04) == 0x04) status.Add("CORR");

            if((registers.Status & 0x08) == 0x08) status.Add("DRQ");

            if((registers.Status & 0x10) == 0x10) status.Add("SRV");

            if((registers.Status & 0x20) == 0x20) status.Add("DF");

            if((registers.Status & 0x40) == 0x40) status.Add("RDY");

            if((registers.Status & 0x80) == 0x80) status.Add("BSY");

            if((registers.Error & 0x01) == 0x01) error.Add("AMNF");

            if((registers.Error & 0x02) == 0x02) error.Add("T0NF");

            if((registers.Error & 0x04) == 0x04) error.Add("ABRT");

            if((registers.Error & 0x08) == 0x08) error.Add("MCR");

            if((registers.Error & 0x10) == 0x10) error.Add("IDNF");

            if((registers.Error & 0x20) == 0x20) error.Add("MC");

            if((registers.Error & 0x40) == 0x40) error.Add("UNC");

            if((registers.Error & 0x80) == 0x80) error.Add("BBK");

            _logSw.WriteLine(Localization.Core.ATA_reading_CHS_0_1_2_error_status_3_error_4,
                             cylinder,
                             head,
                             sector,
                             string.Join(' ', status),
                             string.Join(' ', error));

            _logSw.Flush();
        }
    }

    /// <summary>Register an ATA error after trying to read using 28-bit LBA commands</summary>
    /// <param name="block">Starting block</param>
    /// <param name="osError"><c>true</c> if operating system returned an error status instead of the device</param>
    /// <param name="errno">Operating system error number</param>
    /// <param name="registers">Error registers</param>
    public void WriteLine(ulong block, bool osError, int errno, AtaErrorRegistersLba28 registers)
    {
        if(osError)
        {
            _logSw.WriteLine(Localization.Core.ATA_reading_LBA_0_operating_system_error_1, block, errno);
            _logSw.Flush();
        }
        else
        {
            List<string> error  = new();
            List<string> status = new();

            if((registers.Status & 0x01) == 0x01) status.Add("ERR");

            if((registers.Status & 0x02) == 0x02) status.Add("IDX");

            if((registers.Status & 0x04) == 0x04) status.Add("CORR");

            if((registers.Status & 0x08) == 0x08) status.Add("DRQ");

            if((registers.Status & 0x10) == 0x10) status.Add("SRV");

            if((registers.Status & 0x20) == 0x20) status.Add("DF");

            if((registers.Status & 0x40) == 0x40) status.Add("RDY");

            if((registers.Status & 0x80) == 0x80) status.Add("BSY");

            if((registers.Error & 0x01) == 0x01) error.Add("AMNF");

            if((registers.Error & 0x02) == 0x02) error.Add("T0NF");

            if((registers.Error & 0x04) == 0x04) error.Add("ABRT");

            if((registers.Error & 0x08) == 0x08) error.Add("MCR");

            if((registers.Error & 0x10) == 0x10) error.Add("IDNF");

            if((registers.Error & 0x20) == 0x20) error.Add("MC");

            if((registers.Error & 0x40) == 0x40) error.Add("UNC");

            if((registers.Error & 0x80) == 0x80) error.Add("BBK");

            _logSw.WriteLine(Localization.Core.ATA_reading_LBA_0_error_status_1_error_2,
                             block,
                             string.Join(' ', status),
                             string.Join(' ', error));

            _logSw.Flush();
        }
    }

    /// <summary>Register an ATA error after trying to read using 48-bit LBA commands</summary>
    /// <param name="block">Starting block</param>
    /// <param name="osError"><c>true</c> if operating system returned an error status instead of the device</param>
    /// <param name="errno">Operating system error number</param>
    /// <param name="registers">Error registers</param>
    public void WriteLine(ulong block, bool osError, int errno, AtaErrorRegistersLba48 registers)
    {
        if(osError)
        {
            _logSw.WriteLine(Localization.Core.ATA_reading_LBA_0_operating_system_error_1, block, errno);
            _logSw.Flush();
        }
        else
        {
            List<string> error  = new();
            List<string> status = new();

            if((registers.Status & 0x01) == 0x01) status.Add("ERR");

            if((registers.Status & 0x02) == 0x02) status.Add("IDX");

            if((registers.Status & 0x04) == 0x04) status.Add("CORR");

            if((registers.Status & 0x08) == 0x08) status.Add("DRQ");

            if((registers.Status & 0x10) == 0x10) status.Add("SRV");

            if((registers.Status & 0x20) == 0x20) status.Add("DF");

            if((registers.Status & 0x40) == 0x40) status.Add("RDY");

            if((registers.Status & 0x80) == 0x80) status.Add("BSY");

            if((registers.Error & 0x01) == 0x01) error.Add("AMNF");

            if((registers.Error & 0x02) == 0x02) error.Add("T0NF");

            if((registers.Error & 0x04) == 0x04) error.Add("ABRT");

            if((registers.Error & 0x08) == 0x08) error.Add("MCR");

            if((registers.Error & 0x10) == 0x10) error.Add("IDNF");

            if((registers.Error & 0x20) == 0x20) error.Add("MC");

            if((registers.Error & 0x40) == 0x40) error.Add("UNC");

            if((registers.Error & 0x80) == 0x80) error.Add("BBK");

            _logSw.WriteLine(Localization.Core.ATA_reading_LBA_0_error_status_1_error_2,
                             block,
                             string.Join(' ', status),
                             string.Join(' ', error));

            _logSw.Flush();
        }
    }

    /// <summary>Register a SCSI error after sending a command</summary>
    /// <param name="command">Command</param>
    /// <param name="osError"><c>true</c> if operating system returned an error status instead of the device</param>
    /// <param name="errno">Operating system error number</param>
    /// <param name="senseBuffer">REQUEST SENSE response buffer</param>
    public void WriteLine(string command, bool osError, int errno, byte[] senseBuffer)
    {
        if(osError)
        {
            _logSw.WriteLine(Localization.Core.SCSI_command_0_operating_system_error_1, command, errno);
            _logSw.Flush();

            return;
        }

        DecodedSense? decodedSense = Sense.Decode(senseBuffer);
        string        prettySense  = Sense.PrettifySense(senseBuffer);
        var           hexSense     = string.Join(' ', senseBuffer.Select(b => $"{b:X2}"));

        if(decodedSense.HasValue)
        {
            if(prettySense != null)
            {
                if(prettySense.StartsWith("SCSI SENSE: ", StringComparison.Ordinal)) prettySense = prettySense[12..];

                if(prettySense.EndsWith('\n')) prettySense = prettySense[..^1];

                prettySense = prettySense.Replace("\n", " - ");

                _logSw.WriteLine(Localization.Core.SCSI_command_0_error_SENSE_1_ASC_2_ASCQ_3_4_5,
                                 command,
                                 decodedSense.Value.SenseKey,
                                 decodedSense.Value.ASC,
                                 decodedSense.Value.ASCQ,
                                 hexSense,
                                 prettySense);
            }
            else
            {
                _logSw.WriteLine(Localization.Core.SCSI_command_0_error_SENSE_1_ASC_2_ASCQ_3_4,
                                 command,
                                 decodedSense.Value.SenseKey,
                                 decodedSense.Value.ASC,
                                 decodedSense.Value.ASCQ,
                                 hexSense);
            }
        }
        else
        {
            if(prettySense != null)
            {
                if(prettySense.StartsWith("SCSI SENSE: ", StringComparison.Ordinal)) prettySense = prettySense[12..];

                if(prettySense.EndsWith('\n')) prettySense = prettySense[..^1];

                prettySense = prettySense.Replace("\n", " - ");

                _logSw.WriteLine(Localization.Core.SCSI_command_0_error_1_2, command, hexSense, prettySense);
            }
            else
                _logSw.WriteLine(Localization.Core.SCSI_command_0_error_1, command, hexSense);
        }

        _logSw.Flush();
    }

    /// <summary>Register an SCSI error after trying to read</summary>
    /// <param name="block">Starting block</param>
    /// <param name="osError"><c>true</c> if operating system returned an error status instead of the device</param>
    /// <param name="errno">Operating system error number</param>
    /// <param name="senseBuffer">REQUEST SENSE response buffer</param>
    public void WriteLine(ulong block, bool osError, int errno, byte[] senseBuffer)
    {
        if(osError)
        {
            _logSw.WriteLine(Localization.Core.SCSI_reading_LBA_0_operating_system_error_1, block, errno);
            _logSw.Flush();

            if(senseBuffer is null || senseBuffer.Length == 0 || senseBuffer.All(s => s == 0)) return;
        }

        DecodedSense? decodedSense = Sense.Decode(senseBuffer);
        string        prettySense  = Sense.PrettifySense(senseBuffer);
        var           hexSense     = string.Join(' ', senseBuffer.Select(b => $"{b:X2}"));

        if(decodedSense.HasValue)
        {
            if(prettySense != null)
            {
                if(prettySense.StartsWith("SCSI SENSE: ", StringComparison.Ordinal)) prettySense = prettySense[12..];

                if(prettySense.EndsWith('\n')) prettySense = prettySense[..^1];

                prettySense = prettySense.Replace("\n", " - ");

                _logSw.WriteLine(Localization.Core.SCSI_reading_LBA_0_error_SENSE_1_ASC_2_ASCQ_3_4_5,
                                 block,
                                 decodedSense.Value.SenseKey,
                                 decodedSense.Value.ASC,
                                 decodedSense.Value.ASCQ,
                                 hexSense,
                                 prettySense);
            }
            else
            {
                _logSw.WriteLine(Localization.Core.SCSI_reading_LBA_0_error_SENSE_1_ASC_2_ASCQ_3_4,
                                 block,
                                 decodedSense.Value.SenseKey,
                                 decodedSense.Value.ASC,
                                 decodedSense.Value.ASCQ,
                                 hexSense);
            }
        }
        else
        {
            if(prettySense != null)
            {
                if(prettySense.StartsWith("SCSI SENSE: ", StringComparison.Ordinal)) prettySense = prettySense[12..];

                if(prettySense.EndsWith('\n')) prettySense = prettySense[..^1];

                prettySense = prettySense.Replace("\n", " - ");

                _logSw.WriteLine(Localization.Core.SCSI_reading_LBA_0_error_1_2, block, hexSense, prettySense);
            }
            else
                _logSw.WriteLine(Localization.Core.SCSI_reading_LBA_0_error_1, block, hexSense);
        }

        _logSw.Flush();
    }

    /// <summary>Register a SecureDigital / MultiMediaCard error after sending a command</summary>
    /// <param name="command">Command</param>
    /// <param name="osError"><c>true</c> if operating system returned an error status instead of the device</param>
    /// <param name="errno">Operating system error number</param>
    /// <param name="response">Response</param>
    public void WriteLine(string command, bool osError, int errno, uint[] response)
    {
        if(osError)
        {
            _logSw.WriteLine(Localization.Core.SD_MMC_command_0_operating_system_error_1, command, errno);
            _logSw.Flush();

            return;
        }

        // TODO: Decode response
        _logSw.WriteLine(Localization.Core.SD_MMC_command_0_error_1,
                         command,
                         string.Join(" - ", response.Select(r => $"0x{r:X8}")));

        _logSw.Flush();
    }

    /// <summary>Register a SecureDigital / MultiMediaCard error after trying to read</summary>
    /// <param name="block">Starting block</param>
    /// <param name="osError"><c>true</c> if operating system returned an error status instead of the device</param>
    /// <param name="errno">Operating system error number</param>
    /// <param name="byteAddressed">Byte addressed</param>
    /// <param name="response">Response</param>
    public void WriteLine(ulong block, bool osError, int errno, bool byteAddressed, uint[] response)

    {
        if(osError)
        {
            _logSw.WriteLine(byteAddressed
                                 ? Localization.Core.SD_MMC_reading_LBA_0_byte_addressed_operating_system_error_1
                                 : Localization.Core.SD_MMC_reading_LBA_0_block_addressed_operating_system_error_1,
                             block,
                             errno);

            _logSw.Flush();

            return;
        }

        _logSw.WriteLine(byteAddressed
                             ? Localization.Core.SD_MMC_reading_LBA_0_byte_addressed_error_1
                             : Localization.Core.SD_MMC_reading_LBA_0_block_addressed_error_1,
                         block,
                         string.Join(" - ", response.Select(r => $"0x{r:X8}")));

        throw new NotImplementedException();
    }
}