// /***************************************************************************
// Aaru Data Preservation Suite
// ----------------------------------------------------------------------------
//
// Filename       : Write.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : Disk image plugins.
//
// --[ Description ] ----------------------------------------------------------
//
//     Writes CloneCD disc images.
//
// --[ License ] --------------------------------------------------------------
//
//     This library is free software; you can redistribute it and/or modify
//     it under the terms of the GNU Lesser General Public License as
//     published by the Free Software Foundation; either version 2.1 of the
//     License, or (at your option) any later version.
//
//     This library 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
//     Lesser General Public License for more details.
//
//     You should have received a copy of the GNU Lesser General Public
//     License along with this library; if not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2011-2025 Natalia Portillo
// ****************************************************************************/

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using Aaru.CommonTypes;
using Aaru.CommonTypes.AaruMetadata;
using Aaru.CommonTypes.Enums;
using Aaru.CommonTypes.Structs;
using Aaru.Decoders.CD;
using Aaru.Helpers;
using Aaru.Logging;
using Track = Aaru.CommonTypes.Structs.Track;
using TrackType = Aaru.CommonTypes.Enums.TrackType;

namespace Aaru.Images;

public sealed partial class CloneCd
{
#region IWritableOpticalImage Members

    /// <inheritdoc />
    public bool Create(string path, MediaType mediaType, Dictionary<string, string> options, ulong sectors,
                       uint   sectorSize)
    {
        if(!SupportedMediaTypes.Contains(mediaType))
        {
            ErrorMessage = string.Format(Localization.Unsupported_media_format_0, mediaType);

            return false;
        }

        _imageInfo = new ImageInfo
        {
            MediaType  = mediaType,
            SectorSize = sectorSize,
            Sectors    = sectors
        };

        try
        {
            _writingBaseName = Path.Combine(Path.GetDirectoryName(path) ?? "", Path.GetFileNameWithoutExtension(path));

            _descriptorStream = new StreamWriter(path, false, Encoding.ASCII);

            _dataStream = new FileStream(_writingBaseName + ".img",
                                         FileMode.OpenOrCreate,
                                         FileAccess.ReadWrite,
                                         FileShare.None);
        }
        catch(IOException ex)
        {
            ErrorMessage = string.Format(Localization.Could_not_create_new_image_file_exception_0, ex.Message);
            AaruLogging.Exception(ex, Localization.Could_not_create_new_image_file_exception_0, ex.Message);

            return false;
        }

        _imageInfo.MediaType = mediaType;

        _trackFlags = new Dictionary<byte, byte>();

        IsWriting    = true;
        ErrorMessage = null;

        return true;
    }

    /// <inheritdoc />
    public bool WriteMediaTag(byte[] data, MediaTagType tag)
    {
        if(!IsWriting)
        {
            ErrorMessage = Localization.Tried_to_write_on_a_non_writable_image;

            return false;
        }

        switch(tag)
        {
            case MediaTagType.CD_MCN:
                _catalog = Encoding.ASCII.GetString(data);

                return true;
            case MediaTagType.CD_FullTOC:
                _fullToc = data;

                return true;
            default:
                ErrorMessage = string.Format(Localization.Unsupported_media_tag_0, tag);

                return false;
        }
    }

    /// <inheritdoc />
    public bool WriteSector(byte[] data, ulong sectorAddress)
    {
        if(!IsWriting)
        {
            ErrorMessage = Localization.Tried_to_write_on_a_non_writable_image;

            return false;
        }

        // TODO: Implement ECC generation
        ErrorMessage = Localization.This_format_requires_sectors_to_be_raw_Generating_ECC_is_not_yet_implemented;

        return false;
    }

    /// <inheritdoc />
    public bool WriteSectors(byte[] data, ulong sectorAddress, uint length)
    {
        if(!IsWriting)
        {
            ErrorMessage = Localization.Tried_to_write_on_a_non_writable_image;

            return false;
        }

        // TODO: Implement ECC generation
        ErrorMessage = Localization.This_format_requires_sectors_to_be_raw_Generating_ECC_is_not_yet_implemented;

        return false;
    }

    /// <inheritdoc />
    public bool WriteSectorLong(byte[] data, ulong sectorAddress)
    {
        if(!IsWriting)
        {
            ErrorMessage = Localization.Tried_to_write_on_a_non_writable_image;

            return false;
        }

        Track track = Tracks.FirstOrDefault(trk => sectorAddress >= trk.StartSector && sectorAddress <= trk.EndSector);

        if(track is null)
        {
            ErrorMessage = string.Format(Localization.Cant_find_track_containing_0, sectorAddress);

            return false;
        }

        if(data.Length != track.RawBytesPerSector)
        {
            ErrorMessage = Localization.Incorrect_data_size;

            return false;
        }

        _dataStream.Seek((long)(track.FileOffset +
                                (sectorAddress - track.StartSector) * (ulong)track.RawBytesPerSector),
                         SeekOrigin.Begin);

        _dataStream.Write(data, 0, data.Length);

        return true;
    }

    /// <inheritdoc />
    public bool WriteSectorsLong(byte[] data, ulong sectorAddress, uint length)
    {
        if(!IsWriting)
        {
            ErrorMessage = Localization.Tried_to_write_on_a_non_writable_image;

            return false;
        }

        Track track = Tracks.FirstOrDefault(trk => sectorAddress >= trk.StartSector && sectorAddress <= trk.EndSector);

        if(track is null)
        {
            ErrorMessage = string.Format(Localization.Cant_find_track_containing_0, sectorAddress);

            return false;
        }

        if(sectorAddress + length > track.EndSector + 1)
        {
            ErrorMessage = Localization.Cant_cross_tracks;

            return false;
        }

        if(data.Length % track.RawBytesPerSector != 0)
        {
            ErrorMessage = Localization.Incorrect_data_size;

            return false;
        }

        _dataStream.Seek((long)(track.FileOffset +
                                (sectorAddress - track.StartSector) * (ulong)track.RawBytesPerSector),
                         SeekOrigin.Begin);

        _dataStream.Write(data, 0, data.Length);

        return true;
    }

    /// <inheritdoc />
    public bool SetTracks(List<Track> tracks)
    {
        ulong currentDataOffset       = 0;
        ulong currentSubchannelOffset = 0;

        Tracks = [];

        foreach(Track track in tracks.OrderBy(t => t.Sequence))
        {
            Track newTrack = track;

            if(newTrack.Session > 1)
            {
                Track firstSessionTrack = tracks.FirstOrDefault(t => t.Session == newTrack.Session);

                if(firstSessionTrack?.Sequence == newTrack.Sequence && newTrack.Pregap >= 150)
                {
                    newTrack.Pregap      -= 150;
                    newTrack.StartSector += 150;
                }
            }

            uint subchannelSize;

            switch(newTrack.SubchannelType)
            {
                case TrackSubchannelType.None:
                    subchannelSize = 0;

                    break;
                case TrackSubchannelType.Raw:
                case TrackSubchannelType.RawInterleaved:
                    subchannelSize = 96;

                    break;
                default:
                    ErrorMessage = string.Format(Localization.Unsupported_subchannel_type_0, newTrack.SubchannelType);

                    return false;
            }

            newTrack.FileOffset       = currentDataOffset;
            newTrack.SubchannelOffset = currentSubchannelOffset;

            currentDataOffset += (ulong)newTrack.RawBytesPerSector * (newTrack.EndSector - newTrack.StartSector + 1);

            currentSubchannelOffset += subchannelSize * (newTrack.EndSector - newTrack.StartSector + 1);

            Tracks.Add(newTrack);
        }

        return true;
    }

    /// <inheritdoc />
    public bool Close()
    {
        if(!IsWriting)
        {
            ErrorMessage = Localization.Image_is_not_opened_for_writing;

            return false;
        }

        _dataStream.Flush();
        _dataStream.Close();

        _subStream?.Flush();
        _subStream?.Close();

        FullTOC.CDFullTOC? nullableToc = null;

        // Easy, just decode the real toc
        if(_fullToc != null)
        {
            byte[] tmp = new byte[_fullToc.Length + 2];
            Array.Copy(BigEndianBitConverter.GetBytes((ushort)_fullToc.Length), 0, tmp, 0, 2);
            Array.Copy(_fullToc,                                                0, tmp, 2, _fullToc.Length);
            nullableToc = FullTOC.Decode(tmp);
        }

        // Not easy, create a toc from scratch
        FullTOC.CDFullTOC toc = nullableToc ?? FullTOC.Create(Tracks, _trackFlags, true);

        _descriptorStream.WriteLine("[CloneCD]");
        _descriptorStream.WriteLine("Version=2");
        _descriptorStream.WriteLine("[Disc]");
        _descriptorStream.WriteLine("TocEntries={0}", toc.TrackDescriptors.Length);
        _descriptorStream.WriteLine("Sessions={0}",   toc.LastCompleteSession);
        _descriptorStream.WriteLine("DataTracksScrambled=0");
        _descriptorStream.WriteLine("CDTextLength=0");

        if(!string.IsNullOrEmpty(_catalog)) _descriptorStream.WriteLine("CATALOG={0}", _catalog);

        for(int i = 1; i <= toc.LastCompleteSession; i++)
        {
            _descriptorStream.WriteLine("[Session {0}]", i);

            Track firstSessionTrack = Tracks.FirstOrDefault(t => t.Session == i);

            switch(firstSessionTrack?.Type)
            {
                case TrackType.Audio:
                    // CloneCD always writes this value for first track in disc, however the Rainbow Books
                    // say the first track pregap is no different from other session pregaps, same mode as
                    // the track they belong to.
                    _descriptorStream.WriteLine("PreGapMode=0");

                    break;
                case TrackType.Data:
                case TrackType.CdMode1:
                    _descriptorStream.WriteLine("PreGapMode=1");

                    break;
                case TrackType.CdMode2Formless:
                case TrackType.CdMode2Form1:
                case TrackType.CdMode2Form2:
                    _descriptorStream.WriteLine("PreGapMode=2");

                    break;
                default:
                    ErrorMessage = string.Format(Localization.Unexpected_first_session_track_type_0,
                                                 firstSessionTrack?.Type.ToString() ?? "null");

                    return false;
            }

            _descriptorStream.WriteLine("PreGapSubC=0");
        }

        for(int i = 0; i < toc.TrackDescriptors.Length; i++)
        {
            long alba = MsfToLba((toc.TrackDescriptors[i].Min, toc.TrackDescriptors[i].Sec,
                                  toc.TrackDescriptors[i].Frame));

            long plba = MsfToLba((toc.TrackDescriptors[i].PMIN, toc.TrackDescriptors[i].PSEC,
                                  toc.TrackDescriptors[i].PFRAME));

            if(alba > 405000) alba = (alba - 405000 + 300) * -1;

            if(plba > 405000) plba = (plba - 405000 + 300) * -1;

            _descriptorStream.WriteLine("[Entry {0}]",      i);
            _descriptorStream.WriteLine("Session={0}",      toc.TrackDescriptors[i].SessionNumber);
            _descriptorStream.WriteLine("Point=0x{0:x2}",   toc.TrackDescriptors[i].POINT);
            _descriptorStream.WriteLine("ADR=0x{0:x2}",     toc.TrackDescriptors[i].ADR);
            _descriptorStream.WriteLine("Control=0x{0:x2}", toc.TrackDescriptors[i].CONTROL);
            _descriptorStream.WriteLine("TrackNo={0}",      toc.TrackDescriptors[i].TNO);
            _descriptorStream.WriteLine("AMin={0}",         toc.TrackDescriptors[i].Min);
            _descriptorStream.WriteLine("ASec={0}",         toc.TrackDescriptors[i].Sec);
            _descriptorStream.WriteLine("AFrame={0}",       toc.TrackDescriptors[i].Frame);
            _descriptorStream.WriteLine("ALBA={0}",         alba);

            _descriptorStream.WriteLine("Zero={0}",
                                        ((toc.TrackDescriptors[i].HOUR & 0x0F) << 4) +
                                        (toc.TrackDescriptors[i].PHOUR & 0x0F));

            _descriptorStream.WriteLine("PMin={0}",   toc.TrackDescriptors[i].PMIN);
            _descriptorStream.WriteLine("PSec={0}",   toc.TrackDescriptors[i].PSEC);
            _descriptorStream.WriteLine("PFrame={0}", toc.TrackDescriptors[i].PFRAME);
            _descriptorStream.WriteLine("PLBA={0}",   plba);
        }

        _descriptorStream.Flush();
        _descriptorStream.Close();

        IsWriting    = false;
        ErrorMessage = "";

        return true;
    }

    /// <inheritdoc />
    public bool SetImageInfo(ImageInfo imageInfo) => true;

    /// <inheritdoc />
    public bool SetGeometry(uint cylinders, uint heads, uint sectorsPerTrack)
    {
        ErrorMessage = Localization.Unsupported_feature;

        return false;
    }

    /// <inheritdoc />
    public bool WriteSectorTag(byte[] data, ulong sectorAddress, SectorTagType tag)
    {
        if(!IsWriting)
        {
            ErrorMessage = Localization.Tried_to_write_on_a_non_writable_image;

            return false;
        }

        Track track = Tracks.FirstOrDefault(trk => sectorAddress >= trk.StartSector && sectorAddress <= trk.EndSector);

        if(track is null)
        {
            ErrorMessage = string.Format(Localization.Cant_find_track_containing_0, sectorAddress);

            return false;
        }

        switch(tag)
        {
            case SectorTagType.CdTrackFlags:
            {
                if(data.Length != 1)
                {
                    ErrorMessage = Localization.Incorrect_data_size_for_track_flags;

                    return false;
                }

                _trackFlags[(byte)sectorAddress] = data[0];

                return true;
            }
            case SectorTagType.CdSectorSubchannel:
            {
                if(track.SubchannelType == 0)
                {
                    ErrorMessage =
                        string.Format(Localization.Trying_to_write_subchannel_to_track_0_that_does_not_have_subchannel,
                                      track.Sequence);

                    return false;
                }

                if(data.Length != 96)
                {
                    ErrorMessage = Localization.Incorrect_data_size_for_subchannel;

                    return false;
                }

                if(_subStream == null)
                {
                    try
                    {
                        _subStream = new FileStream(_writingBaseName + ".sub",
                                                    FileMode.OpenOrCreate,
                                                    FileAccess.ReadWrite,
                                                    FileShare.None);
                    }
                    catch(IOException ex)
                    {
                        ErrorMessage = string.Format(Localization.Could_not_create_subchannel_file_exception_0,
                                                     ex.Message);

                        AaruLogging.Exception(ex,Localization.Could_not_create_subchannel_file_exception_0,
                                              ex.Message);

                        return false;
                    }
                }

                _subStream.Seek((long)(track.SubchannelOffset + (sectorAddress - track.StartSector) * 96),
                                SeekOrigin.Begin);

                _subStream.Write(Subchannel.Deinterleave(data), 0, data.Length);

                return true;
            }
            default:
                ErrorMessage = string.Format(Localization.Unsupported_tag_type_0, tag);

                return false;
        }
    }

    /// <inheritdoc />
    public bool WriteSectorsTag(byte[] data, ulong sectorAddress, uint length, SectorTagType tag)
    {
        if(!IsWriting)
        {
            ErrorMessage = Localization.Tried_to_write_on_a_non_writable_image;

            return false;
        }

        Track track = Tracks.FirstOrDefault(trk => sectorAddress >= trk.StartSector && sectorAddress <= trk.EndSector);

        if(track is null)
        {
            ErrorMessage = string.Format(Localization.Cant_find_track_containing_0, sectorAddress);

            return false;
        }

        switch(tag)
        {
            case SectorTagType.CdTrackFlags:
            case SectorTagType.CdTrackIsrc:
                return WriteSectorTag(data, sectorAddress, tag);
            case SectorTagType.CdSectorSubchannel:
            {
                if(track.SubchannelType == 0)
                {
                    ErrorMessage =
                        string.Format(Localization.Trying_to_write_subchannel_to_track_0_that_does_not_have_subchannel,
                                      track.Sequence);

                    return false;
                }

                if(data.Length % 96 != 0)
                {
                    ErrorMessage = Localization.Incorrect_data_size_for_subchannel;

                    return false;
                }

                if(_subStream == null)
                {
                    try
                    {
                        _subStream = new FileStream(_writingBaseName + ".sub",
                                                    FileMode.OpenOrCreate,
                                                    FileAccess.ReadWrite,
                                                    FileShare.None);
                    }
                    catch(IOException ex)
                    {
                        ErrorMessage = string.Format(Localization.Could_not_create_subchannel_file_exception_0,
                                                     ex.Message);

                        AaruLogging.Exception(ex,Localization.Could_not_create_subchannel_file_exception_0,
                                              ex.Message);

                        return false;
                    }
                }

                _subStream.Seek((long)(track.SubchannelOffset + (sectorAddress - track.StartSector) * 96),
                                SeekOrigin.Begin);

                _subStream.Write(Subchannel.Deinterleave(data), 0, data.Length);

                return true;
            }
            default:
                ErrorMessage = string.Format(Localization.Unsupported_tag_type_0, tag);

                return false;
        }
    }

    /// <inheritdoc />
    public bool SetDumpHardware(List<DumpHardware> dumpHardware) => false;

    /// <inheritdoc />
    public bool SetMetadata(Metadata metadata) => false;

#endregion
}