/***
    Copyright (C) 2022 J Reece Wilson (a/k/a "Reece"). All rights reserved.

    File: ProtocolPiece.hpp
    Date: 2022-8-24
    Author: Reece
***/
#pragma once

namespace Aurora::IO::Protocol
{
    struct IProtocolNext;
    struct ProtocolStack;

    struct ProtocolPiece :
        IProtocolPiece,
        AuEnableSharedFromThis<ProtocolPiece>
    {
        ProtocolStack *pParent;
        AuSPtr<IProtocolInterceptor> pInterceptor;
        AuSPtr<IProtocolInterceptorEx> pInterceptorEx;
        AuSPtr<IProtocolNext> pWriteInteface;
        AuSPtr<ProtocolPiece> pNext;
        AuSPtr<ProtocolPiece> pBefore;
        AuByteBuffer outputBuffer;
        AuSPtr<IStreamWriter> pOuputWriter;
        bool bMultipleTick {};
        AuOptional<AuUInt> uMax;
        AuUInt64 uStartingSize {};
        friend struct ProtocolStack; // IProtocolPiece::PrivateUserDataClear()

        AuSPtr<IProtocolStack> GetParent() override;
        AuSPtr<IProtocolPiece> GetNextPiece() override;
        AuSPtr<IProtocolPiece> GetPreviousPiece() override;
        bool ReallocateDrainBuffer(AuUInt32 uOutputLength) override;
        bool ReallocateSourceBuffer(AuUInt32 uInputLength) override;
        void Remove() override;
        AuSPtr<IStreamWriter> GetNextWriter() override;
        AuSPtr<IStreamWriter> GetInputWriter() override;
        AuSPtr<IStreamReader> GetOutputReader() override;
        AuSPtr<Memory::ByteBuffer> GetNextPieceBuffer() override;
        AuSPtr<IProtocolInterceptorEx> GetExtendedInterceptor() override;
        AuSPtr<IProtocolInterceptor> GetShortPipeInterceptor() override;
        AuOptional<AuUInt> ExchangeMaximumBufferLength(AuOptional<AuUInt> optMax) override;
    };
}