using System;

namespace Ryujinx.HLE.HOS.Services.Am.AppletAE
{
    class IStorageAccessor : IpcService
    {
        private IStorage _storage;

        public IStorageAccessor(IStorage storage)
        {
            _storage = storage;
        }

        [CommandHipc(0)]
        // GetSize() -> u64
        public ResultCode GetSize(ServiceCtx context)
        {
            context.ResponseData.Write((long)_storage.Data.Length);

            return ResultCode.Success;
        }

        [CommandHipc(10)]
        // Write(u64, buffer<bytes, 0x21>)
        public ResultCode Write(ServiceCtx context)
        {
            if (_storage.IsReadOnly)
            {
                return ResultCode.ObjectInvalid;
            }

            ulong writePosition = context.RequestData.ReadUInt64();

            if (writePosition > (ulong)_storage.Data.Length)
            {
                return ResultCode.OutOfBounds;
            }

            (ulong position, ulong size) = context.Request.GetBufferType0x21();

            size = Math.Min(size, (ulong)_storage.Data.Length - writePosition);

            if (size > 0)
            {
                ulong maxSize = (ulong)_storage.Data.Length - writePosition;

                if (size > maxSize)
                {
                    size = maxSize;
                }

                byte[] data = new byte[size];

                context.Memory.Read(position, data);

                Buffer.BlockCopy(data, 0, _storage.Data, (int)writePosition, (int)size);
            }

            return ResultCode.Success;
        }

        [CommandHipc(11)]
        // Read(u64) -> buffer<bytes, 0x22>
        public ResultCode Read(ServiceCtx context)
        {
            ulong readPosition = context.RequestData.ReadUInt64();

            if (readPosition > (ulong)_storage.Data.Length)
            {
                return ResultCode.OutOfBounds;
            }

            (ulong position, ulong size) = context.Request.GetBufferType0x22();

            size = Math.Min(size, (ulong)_storage.Data.Length - readPosition);

            byte[] data = new byte[size];

            Buffer.BlockCopy(_storage.Data, (int)readPosition, data, 0, (int)size);

            context.Memory.Write(position, data);

            return ResultCode.Success;
        }
    }
}