// /***************************************************************************
// Aaru Data Preservation Suite
// ----------------------------------------------------------------------------
//
// Filename       : Dir.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : Apple DOS filesystem plugin.
//
// --[ Description ] ----------------------------------------------------------
//
//     Methods to handle Apple DOS filesystem catalog (aka directory).
//
// --[ 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 Aaru.CommonTypes.Enums;
using Aaru.CommonTypes.Interfaces;
using Aaru.Helpers;

namespace Aaru.Filesystems;

public sealed partial class AppleDOS
{
    ErrorNumber ReadCatalog()
    {
        var catalogMs = new MemoryStream();
        var lba       = (ulong)(_vtoc.catalogTrack * _sectorsPerTrack + _vtoc.catalogSector);
        _totalFileEntries = 0;
        _catalogCache     = new Dictionary<string, ushort>();
        _fileTypeCache    = new Dictionary<string, byte>();
        _lockedFiles      = [];

        if(lba == 0 || lba > _device.Info.Sectors) return ErrorNumber.InvalidArgument;

        while(lba != 0)
        {
            _usedSectors++;
            ErrorNumber errno = _device.ReadSector(lba, false, out byte[] catSectorB, out _);

            if(errno != ErrorNumber.NoError) return errno;

            _totalFileEntries += 7;

            if(_debug) catalogMs.Write(catSectorB, 0, catSectorB.Length);

            // Read the catalog sector
            CatalogSector catSector = Marshal.ByteArrayToStructureLittleEndian<CatalogSector>(catSectorB);

            foreach(FileEntry entry in catSector.entries.Where(static entry => entry.extentTrack > 0))
            {
                _track1UsedByFiles |= entry.extentTrack == 1;
                _track2UsedByFiles |= entry.extentTrack == 2;

                var filenameB = new byte[30];
                var ts        = (ushort)(entry.extentTrack << 8 | entry.extentSector);

                // Apple DOS has high byte set over ASCII.
                for(var i = 0; i < 30; i++) filenameB[i] = (byte)(entry.filename[i] & 0x7F);

                string filename = StringHandlers.SpacePaddedToString(filenameB, _encoding);

                _catalogCache.TryAdd(filename, ts);
                _fileTypeCache.TryAdd(filename, (byte)(entry.typeAndFlags & 0x7F));

                if((entry.typeAndFlags & 0x80) == 0x80 && !_lockedFiles.Contains(filename)) _lockedFiles.Add(filename);
            }

            lba = (ulong)(catSector.trackOfNext * _sectorsPerTrack + catSector.sectorOfNext);

            if(lba > _device.Info.Sectors) break;
        }

        if(_debug) _catalogBlocks = catalogMs.ToArray();

        return ErrorNumber.NoError;
    }

#region IReadOnlyFilesystem Members

    /// <inheritdoc />
    public ErrorNumber ReadLink(string path, out string dest)
    {
        dest = null;

        return !_mounted ? ErrorNumber.AccessDenied : ErrorNumber.NotSupported;
    }

    /// <inheritdoc />
    public ErrorNumber OpenDir(string path, out IDirNode node)
    {
        node = null;

        if(!_mounted) return ErrorNumber.AccessDenied;

        if(!string.IsNullOrEmpty(path) && !string.Equals(path, "/", StringComparison.OrdinalIgnoreCase))
            return ErrorNumber.NotSupported;

        var contents = _catalogCache.Keys.ToList();

        if(_debug)
        {
            contents.Add("$");
            contents.Add("$Boot");
            contents.Add("$Vtoc");
        }

        contents.Sort();

        node = new AppleDosDirNode
        {
            Path     = path,
            Position = 0,
            Contents = contents.ToArray()
        };

        return ErrorNumber.NoError;
    }

    /// <inheritdoc />
    public ErrorNumber ReadDir(IDirNode node, out string filename)
    {
        filename = null;

        if(!_mounted) return ErrorNumber.AccessDenied;

        if(node is not AppleDosDirNode mynode) return ErrorNumber.InvalidArgument;

        if(mynode.Position < 0) return ErrorNumber.InvalidArgument;

        if(mynode.Position >= mynode.Contents.Length) return ErrorNumber.NoError;

        filename = mynode.Contents[mynode.Position++];

        return ErrorNumber.NoError;
    }

    /// <inheritdoc />
    public ErrorNumber CloseDir(IDirNode node)
    {
        if(node is not AppleDosDirNode mynode) return ErrorNumber.InvalidArgument;

        mynode.Position = -1;
        mynode.Contents = null;

        return ErrorNumber.NoError;
    }

#endregion
}