#ifndef _LIBRARY_UTILS_BUFFERWRAPPER_HPP_
#define _LIBRARY_UTILS_BUFFERWRAPPER_HPP_

// ------------------------------------------------------------------------------------------------
#include "Base/Shared.hpp"
#include "Base/Buffer.hpp"

// ------------------------------------------------------------------------------------------------
namespace SqMod {

// ------------------------------------------------------------------------------------------------
template < typename T > class BufferInterpreter;

/* ------------------------------------------------------------------------------------------------
 * Squirrel wrapper for the shared buffer class.
*/
class BufferWrapper
{
private:

    // --------------------------------------------------------------------------------------------
    typedef SharedPtr< Buffer > SRef; // Strong reference type to the managed memory buffer.
    typedef WeakPtr< Buffer >   WRef; // Weak reference type to the managed memory buffer.

    // --------------------------------------------------------------------------------------------
    SRef m_Buffer; // The managed memory buffer.

public:

    // --------------------------------------------------------------------------------------------
    typedef Buffer::Value   Value; // The type of value used to represent a byte.

    // --------------------------------------------------------------------------------------------
    typedef Value &         Reference; // A reference to the stored value type.
    typedef const Value &   ConstRef; // A const reference to the stored value type.

    // --------------------------------------------------------------------------------------------
    typedef Value *         Pointer; // A pointer to the stored value type.
    typedef const Value *   ConstPtr; // A const pointer to the stored value type.

    // --------------------------------------------------------------------------------------------
    typedef Buffer::SzType  SzType; // The type used to represent size in general.

public:

    /* --------------------------------------------------------------------------------------------
     * Create a memory buffer with the requested size.
    */
    static Object Create(SzType n);

    /* --------------------------------------------------------------------------------------------
     * Base constructor.
    */
    BufferWrapper(const SRef & ref)
        : m_Buffer(ref)
    {
        /* ... */
    }

    /* --------------------------------------------------------------------------------------------
     * Buffer constructor.
    */
    BufferWrapper(Buffer && b)
        : m_Buffer(new Buffer(std::move(b)))
    {
        /* ... */
    }

    /* --------------------------------------------------------------------------------------------
     * Copy constructor.
    */
    BufferWrapper(const BufferWrapper & o) = default;

    /* --------------------------------------------------------------------------------------------
     * Move constructor.
    */
    BufferWrapper(BufferWrapper && o) = default;

    /* --------------------------------------------------------------------------------------------
     * Destructor.
    */
    ~BufferWrapper() = default;

    /* --------------------------------------------------------------------------------------------
     * Copy assignment operator.
    */
    BufferWrapper & operator = (const BufferWrapper & o) = default;

    /* --------------------------------------------------------------------------------------------
     * Move assignment operator.
    */
    BufferWrapper & operator = (BufferWrapper && o) = default;

    /* --------------------------------------------------------------------------------------------
     * Retrieve a reference to the managed memory buffer.
    */
    const SRef & GetRef() const
    {
        return m_Buffer;
    }

    /* --------------------------------------------------------------------------------------------
     * Validate the managed memory buffer reference.
    */
    void Validate() const
    {
        // Do we even point to a valid buffer?
        if (!m_Buffer)
        {
            STHROWF("Invalid memory buffer reference");
        }
    }

    /* --------------------------------------------------------------------------------------------
     * Validate the managed memory buffer reference and the buffer itself.
    */
    void ValidateDeeper() const
    {
        // Do we even point to a valid buffer?
        if (!m_Buffer)
        {
            STHROWF("Invalid memory buffer reference");
        }
        // Validate the buffer itself
        else if (!(*m_Buffer))
        {
            STHROWF("Invalid memory buffer");
        }
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve a certain element type at the specified position.
    */
    Value Get(SzType n) const
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (n >= m_Buffer->Size())
        {
            STHROWF("Index (%u) is out of bounds (%u)", n, m_Buffer->Size());
        }
        // Return the requested element
        return m_Buffer->At(n);
    }

    /* --------------------------------------------------------------------------------------------
     * Modify a certain element type at the specified position.
    */
    void Set(SzType n, Value v)
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (n >= m_Buffer->Size())
        {
            STHROWF("Index (%u) is out of bounds (%u)", n, m_Buffer->Size());
        }
        // Return the requested element
        m_Buffer->At(n) = v;
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the element at the front of the buffer.
    */
    Value GetFront() const
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < sizeof(Value))
        {
            STHROWF("Value size (%u starting at 0) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        return m_Buffer->Front();
    }

    /* --------------------------------------------------------------------------------------------
     * Modify the element at the front of the buffer.
    */
    void SetFront(Value v)
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < sizeof(Value))
        {
            STHROWF("Value size (%u starting at 0) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        m_Buffer->Front() = v;
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the element after the first element in the buffer.
    */
    Value GetNext() const
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < (sizeof(Value) * 2))
        {
            STHROWF("Value size (%u starting at %u) is out of bounds (%u)",
                        sizeof(Value), sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        return m_Buffer->Next();
    }

    /* --------------------------------------------------------------------------------------------
     * Modify the element after the first element in the buffer.
    */
    void SetNext(Value v)
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < (sizeof(Value) * 2))
        {
            STHROWF("Value size (%u starting at %u) is out of bounds (%u)",
                        sizeof(Value), sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        m_Buffer->Next() = v;
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the element at the back of the buffer.
    */
    Value GetBack() const
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < sizeof(Value))
        {
            STHROWF("Value size (%u starting at 0) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        return m_Buffer->Back();
    }

    /* --------------------------------------------------------------------------------------------
     * Modify the element at the back of the buffer.
    */
    void SetBack(Value v)
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < sizeof(Value))
        {
            STHROWF("Value size (%u starting at 0) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        m_Buffer->Back() = v;
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the element before the last element in the buffer.
    */
    Value GetPrev() const
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < (sizeof(Value) * 2))
        {
            STHROWF("Value size (%u starting at %u) is out of bounds (%u)",
                        sizeof(Value), sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        return m_Buffer->Prev();
    }

    /* --------------------------------------------------------------------------------------------
     * Modify the element before the last element in the buffer.
    */
    void SetPrev(Value v)
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < (sizeof(Value) * 2))
        {
            STHROWF("Value size (%u starting at %u) is out of bounds (%u)",
                        sizeof(Value), sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        m_Buffer->Prev() = v;
    }

    /* --------------------------------------------------------------------------------------------
     * Reposition the edit cursor to the specified number of elements ahead.
    */
    void Advance(SzType n)
    {
        // Validate the managed buffer reference
        Validate();
        // Perform the requested operation
        m_Buffer->Advance(n);
    }

    /* --------------------------------------------------------------------------------------------
     * Reposition the edit cursor to the specified number of elements behind.
    */
    void Retreat(SzType n)
    {
        // Validate the managed buffer reference
        Validate();
        // Perform the requested operation
        m_Buffer->Retreat(n);
    }

    /* --------------------------------------------------------------------------------------------
     * Reposition the edit cursor to a fixed position within the buffer.
    */
    void Move(SzType n)
    {
        // Validate the managed buffer reference
        Validate();
        // Perform the requested operation
        m_Buffer->Move(n);
    }

    /* --------------------------------------------------------------------------------------------
     * Append a value to the current cursor location and advance the cursor.
    */
    void Push(Value v)
    {
        // Validate the managed buffer reference
        Validate();
        // Perform the requested operation
        m_Buffer->Push(v);
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the element at the cursor position.
    */
    Value GetCursor() const
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Position() >= m_Buffer->Size())
        {
            STHROWF("Value size (%u starting at %u) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Position(), m_Buffer->Capacity());
        }
        // Return the requested element
        return m_Buffer->Cursor();
    }

    /* --------------------------------------------------------------------------------------------
     * Modify the element at the cursor position.
    */
    void SetCursor(Value v)
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Position() >= m_Buffer->Size())
        {
            STHROWF("Value size (%u starting at %u) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Position(), m_Buffer->Capacity());
        }
        // Return the requested element
        m_Buffer->Cursor() = v;
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the element before the cursor position.
    */
    Value GetBefore() const
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Position() < sizeof(Value))
        {
            STHROWF("Value size (%u starting at 0) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        return m_Buffer->Before();
    }

    /* --------------------------------------------------------------------------------------------
     * Modify the element before the cursor position.
    */
    void SetBefore(Value v)
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Position() < sizeof(Value))
        {
            STHROWF("Value size (%u starting at 0) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Capacity());
        }
        // Return the requested element
        m_Buffer->Before() = v;
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the element after the cursor position.
    */
    Value GetAfter() const
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < sizeof(Value) ||
            (m_Buffer->Position() + sizeof(Value)) > (m_Buffer->Capacity() - sizeof(Value)))
        {
            STHROWF("Value size (%u starting at %u) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Position(), m_Buffer->Capacity());
        }
        // Return the requested element
        return m_Buffer->After();
    }

    /* --------------------------------------------------------------------------------------------
     * Modify the element after the cursor position.
    */
    void SetAfter(Value v)
    {
        // Validate the managed buffer reference
        Validate();
        // Are we out of the memory buffer range?
        if (m_Buffer->Capacity() < sizeof(Value) ||
            (m_Buffer->Position() + sizeof(Value)) > (m_Buffer->Capacity() - sizeof(Value)))
        {
            STHROWF("Value size (%u starting at %u) is out of bounds (%u)",
                        sizeof(Value), m_Buffer->Position(), m_Buffer->Capacity());
        }
        // Return the requested element
        m_Buffer->After() = v;
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve maximum elements it can hold for a certain type.
    */
    SzType GetMax() const
    {
        return Buffer::Max();
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the current buffer capacity in element count.
    */
    SzType GetSize() const
    {
        // Validate the managed buffer reference
        Validate();
        // Return the requested information
        return m_Buffer->Size();
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the current buffer capacity in byte count.
    */
    SzType GetCapacity() const
    {
        // Validate the managed buffer reference
        Validate();
        // Return the requested information
        return m_Buffer->Capacity();
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the current position of the cursor in the buffer.
    */
    SzType GetPosition() const
    {
        // Validate the managed buffer reference
        Validate();
        // Return the requested information
        return m_Buffer->Position();
    }

    /* --------------------------------------------------------------------------------------------
     * Retrieve the amount of unused buffer after the edit cursor.
    */
    SzType GetRemaining() const
    {
        // Validate the managed buffer reference
        Validate();
        // Return the requested information
        return m_Buffer->Remaining();
    }

    /* --------------------------------------------------------------------------------------------
     * Grow the size of the internal buffer by the specified amount of bytes.
    */
    void Grow(SzType n)
    {
        // Validate the managed buffer reference
        Validate();
        // Perform the requested operation
        return m_Buffer->Grow(n * sizeof(Value));
    }

    /* --------------------------------------------------------------------------------------------
     * Makes sure there is enough capacity to hold the specified element count.
    */
    void Adjust(SzType n)
    {
        // Validate the managed buffer reference
        Validate();
        // Attempt to perform the requested operation
        try
        {
            Buffer bkp(m_Buffer->Adjust(n * sizeof(Value)));
            // Copy the data into the new buffer
            m_Buffer->Write(0, bkp.Data(), bkp.Capacity());
            m_Buffer->Move(bkp.Position());
        }
        catch (const std::exception & e)
        {
            STHROWF("%s", e.what()); // Re-package
        }
    }

    /* --------------------------------------------------------------------------------------------
     * Write a 8 bit byte to the stream buffer.
    */
    void WriteByte(SQInteger val);

    /* --------------------------------------------------------------------------------------------
     * Write a 16 bit short to the stream buffer.
    */
    void WriteShort(SQInteger val);

    /* --------------------------------------------------------------------------------------------
     * Write a 32 bit integer to the stream buffer.
    */
    void WriteInt(SQInteger val);

    /* --------------------------------------------------------------------------------------------
     * Write a 32 bit float to the stream buffer.
    */
    void WriteFloat(SQFloat val);

    /* --------------------------------------------------------------------------------------------
     * Write a string to the stream buffer.
    */
    void WriteString(CSStr val);

    /* --------------------------------------------------------------------------------------------
     * Write a raw string to the stream buffer.
    */
    void WriteRawString(CSStr val);

    /* --------------------------------------------------------------------------------------------
     * Read a 8 bit byte to the stream buffer.
    */
    SQInteger ReadByte();

    /* --------------------------------------------------------------------------------------------
     * Read a 16 bit short to the stream buffer.
    */
    SQInteger ReadShort();

    /* --------------------------------------------------------------------------------------------
     * Read a 32 bit integer to the stream buffer.
    */
    SQInteger ReadInt();

    /* --------------------------------------------------------------------------------------------
     * Read a 32 bit float to the stream buffer.
    */
    SQFloat ReadFloat();

    /* --------------------------------------------------------------------------------------------
     * Read a string to the stream buffer.
    */
    Object ReadString();

    /* --------------------------------------------------------------------------------------------
     * Read a raw string to the stream buffer.
    */
    Object ReadRawString(Uint32 len);

    /* --------------------------------------------------------------------------------------------
     * Retrieve a signed 8 bit interpreter to this buffer.
    */
    BufferInterpreter< Int8 > GetInt8Interpreter() const;

    /* --------------------------------------------------------------------------------------------
     * Retrieve an unsigned 8 bit interpreter to this buffer.
    */
    BufferInterpreter< Uint8 > GetUint8Interpreter() const;

    /* --------------------------------------------------------------------------------------------
     * Retrieve a signed 16 bit interpreter to this buffer.
    */
    BufferInterpreter< Int16 > GetInt16Interpreter() const;

    /* --------------------------------------------------------------------------------------------
     * Retrieve an unsigned 16 bit interpreter to this buffer.
    */
    BufferInterpreter< Uint16 > GetUint16Interpreter() const;

    /* --------------------------------------------------------------------------------------------
     * Retrieve a signed 32 bit interpreter to this buffer.
    */
    BufferInterpreter< Int32 > GetInt32Interpreter() const;

    /* --------------------------------------------------------------------------------------------
     * Retrieve an unsigned 32 bit interpreter to this buffer.
    */
    BufferInterpreter< Uint32 > GetUint32Interpreter() const;

    /* --------------------------------------------------------------------------------------------
     * Retrieve a signed 64 bit interpreter to this buffer.
    */
    BufferInterpreter< Int64 > GetInt64Interpreter() const;

    /* --------------------------------------------------------------------------------------------
     * Retrieve an unsigned 64 bit interpreter to this buffer.
    */
    BufferInterpreter< Uint64 > GetUint64Interpreter() const;

    /* --------------------------------------------------------------------------------------------
     * Retrieve a 32 bit floating point interpreter to this buffer.
    */
    BufferInterpreter< Float32 > GetFloat32Interpreter() const;

    /* --------------------------------------------------------------------------------------------
     * Retrieve a 64 bit floating point interpreter to this buffer.
    */
    BufferInterpreter< Float64 > GetFloat64Interpreter() const;
};

} // Namespace:: SqMod

#endif // _LIBRARY_UTILS_BUFFERWRAPPER_HPP_
