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

    File: CompressionInterceptor.hpp
    Date: 2022-9-14
    Author: Reece
***/
#pragma once

#include <IO/Protocol/Protocol.hpp>

namespace Aurora::Compression
{
    struct CompressionInterceptor : ICompressionInterceptor, IO::IStreamReader
    {
        CompressionInterceptor();

        void Init(const AuSPtr<ICompressionStream> &pStream,
                  const AuSPtr<BaseStream> &pBaseStream);

        bool OnDataAvailable(const AuSPtr<Memory::ByteBuffer> &pReadInByteBuffer,
                             const AuSPtr<Memory::ByteBuffer> &pWriteOutByteBuffer) override;
        
        inline virtual IO::EStreamError IsOpen() override;

        inline virtual IO::EStreamError Read(const Memory::MemoryViewStreamWrite &parameters) override;

        inline virtual void Close() override;

        bool HasFailed() override;

    private:
        bool bErrorFlag_ {};
        AuSPtr<ICompressionStream> pStream_;
        AuSPtr<BaseStream> pBaseStream_;
        AuSPtr<Memory::ByteBuffer> pLastBuffer_;
    };
}