// /***************************************************************************
// Aaru Data Preservation Suite
// ----------------------------------------------------------------------------
//
// Filename       : File.cs
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : Opera filesystem plugin.
//
// --[ Description ] ----------------------------------------------------------
//
//     Methods to handle files.
//
// --[ 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-2022 Natalia Portillo
// ****************************************************************************/

namespace Aaru.Filesystems;

using System;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using Aaru.CommonTypes.Enums;
using Aaru.CommonTypes.Structs;

public sealed partial class OperaFS
{
    /// <inheritdoc />
    public ErrorNumber MapBlock(string path, long fileBlock, out long deviceBlock)
    {
        deviceBlock = 0;

        if(!_mounted)
            return ErrorNumber.AccessDenied;

        ErrorNumber err = GetFileEntry(path, out DirectoryEntryWithPointers entry);

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

        if((entry.Entry.flags & FLAGS_MASK) == (uint)FileFlags.Directory &&
           !_debug)
            return ErrorNumber.IsDirectory;

        deviceBlock = entry.Pointers[0] + fileBlock;

        return ErrorNumber.NoError;
    }

    /// <inheritdoc />
    public ErrorNumber GetAttributes(string path, out FileAttributes attributes)
    {
        attributes = new FileAttributes();

        if(!_mounted)
            return ErrorNumber.AccessDenied;

        ErrorNumber err = Stat(path, out FileEntryInfo stat);

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

        attributes = stat.Attributes;

        return ErrorNumber.NoError;
    }

    /// <inheritdoc />
    public ErrorNumber Read(string path, long offset, long size, ref byte[] buf)
    {
        buf = null;

        if(!_mounted)
            return ErrorNumber.AccessDenied;

        ErrorNumber err = GetFileEntry(path, out DirectoryEntryWithPointers entry);

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

        if((entry.Entry.flags & FLAGS_MASK) == (uint)FileFlags.Directory &&
           !_debug)
            return ErrorNumber.IsDirectory;

        if(entry.Pointers.Length < 1)
            return ErrorNumber.InvalidArgument;

        if(entry.Entry.byte_count == 0)
        {
            buf = Array.Empty<byte>();

            return ErrorNumber.NoError;
        }

        if(offset >= entry.Entry.byte_count)
            return ErrorNumber.InvalidArgument;

        if(size + offset >= entry.Entry.byte_count)
            size = entry.Entry.byte_count - offset;

        long firstBlock    = offset                 / entry.Entry.block_size;
        long offsetInBlock = offset                 % entry.Entry.block_size;
        long sizeInBlocks  = (size + offsetInBlock) / entry.Entry.block_size;

        if((size + offsetInBlock) % entry.Entry.block_size > 0)
            sizeInBlocks++;

        uint fileBlockSizeRatio;

        if(_image.Info.SectorSize is 2336 or 2352 or 2448)
            fileBlockSizeRatio = entry.Entry.block_size / 2048;
        else
            fileBlockSizeRatio = entry.Entry.block_size / _image.Info.SectorSize;

        ErrorNumber errno = _image.ReadSectors((ulong)(entry.Pointers[0] + firstBlock * fileBlockSizeRatio),
                                               (uint)(sizeInBlocks * fileBlockSizeRatio), out byte[] buffer);

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

        buf = new byte[size];
        Array.Copy(buffer, offsetInBlock, buf, 0, size);

        return ErrorNumber.NoError;
    }

    /// <inheritdoc />
    public ErrorNumber Stat(string path, out FileEntryInfo stat)
    {
        stat = null;

        if(!_mounted)
            return ErrorNumber.AccessDenied;

        ErrorNumber err = GetFileEntry(path, out DirectoryEntryWithPointers entryWithPointers);

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

        DirectoryEntry entry = entryWithPointers.Entry;

        stat = new FileEntryInfo
        {
            Attributes = new FileAttributes(),
            Blocks     = entry.block_count,
            BlockSize  = entry.block_size,
            Length     = entry.byte_count,
            Inode      = entry.id,
            Links      = (ulong)entryWithPointers.Pointers.Length
        };

        var flags = (FileFlags)(entry.flags & FLAGS_MASK);

        switch(flags)
        {
            case FileFlags.Directory:
                stat.Attributes |= FileAttributes.Directory;

                break;
            case FileFlags.Special:
                stat.Attributes |= FileAttributes.Device;

                break;
        }

        return ErrorNumber.NoError;
    }

    ErrorNumber GetFileEntry(string path, out DirectoryEntryWithPointers entry)
    {
        entry = null;

        string cutPath = path.StartsWith("/", StringComparison.Ordinal)
                             ? path.Substring(1).ToLower(CultureInfo.CurrentUICulture)
                             : path.ToLower(CultureInfo.CurrentUICulture);

        string[] pieces = cutPath.Split(new[]
        {
            '/'
        }, StringSplitOptions.RemoveEmptyEntries);

        if(pieces.Length == 0)
            return ErrorNumber.InvalidArgument;

        var parentPath = string.Join("/", pieces, 0, pieces.Length - 1);

        if(!_directoryCache.TryGetValue(parentPath, out _))
        {
            ErrorNumber err = ReadDir(parentPath, out _);

            if(err != ErrorNumber.NoError)
                return err;
        }

        Dictionary<string, DirectoryEntryWithPointers> parent;

        if(pieces.Length == 1)
            parent = _rootDirectoryCache;
        else if(!_directoryCache.TryGetValue(parentPath, out parent))
            return ErrorNumber.InvalidArgument;

        KeyValuePair<string, DirectoryEntryWithPointers> dirent =
            parent.FirstOrDefault(t => t.Key.ToLower(CultureInfo.CurrentUICulture) == pieces[^1]);

        if(string.IsNullOrEmpty(dirent.Key))
            return ErrorNumber.NoSuchFile;

        entry = dirent.Value;

        return ErrorNumber.NoError;
    }
}