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

    File: IOPipeProcessor.hpp
    Date: 2022-6-6
    Author: Reece
***/
#pragma once

#include <Aurora/Utility/ThroughputCalculator.hpp>

namespace Aurora::IO
{
    struct IOProcessor;
    struct IOPipeWork;
    struct IOPipeProcessor;

    struct IOWorkStart : IIOProcessorWorkUnit
    {
        IOWorkStart(IOPipeWork *parent);

        void OnRun() override;
        void OnCanceled()  override;

        IOPipeWork *parent {};
    };

    struct IOWorkEnd : IIOProcessorWorkUnit
    {
        IOWorkEnd(IOPipeWork *parent);

        void OnRun() override;
        void OnCanceled()  override;

        IOPipeWork *parent {};
    };
    
    struct IOPipeWork : IIOPipeWork, IIOEventListenerFunctional, AuEnableSharedFromThis<IIOPipeWork>
    {
        IOPipeWork(const AuSPtr<IOPipeProcessor> &pParent, const IOPipeRequestAIO &request);
        IOPipeWork(const AuSPtr<IOPipeProcessor> &pParent, const IOPipeRequestBasic &request);

        AuSPtr<IIOProcessorItem> pWatch;

        void Tick_FrameEpilogue() override;
        void Tick_Any()           override;

        void OnFailureCompletion() override;
        void OnNominalCompletion() override;

        virtual bool Start() override;

        virtual bool End() override;

        virtual AuInt64 GetLastTickMS() override;
        virtual AuInt64 GetStartTickMS() override;

        virtual AuUInt64 GetBytesProcessed() override;

        virtual double GetPredictedThroughput() override;

        void RunOnThread();
        void TerminateOnThread(bool bError = false);

        // INIT
        void PrepareStream();
        void PrepareAsync();

        // PUMP
        void AsyncPump();
        void StreamPump();

        // END/INIT
        void ReadNext();
        void ReadNextAsync();

        AuUInt32 TryPump();

        AuMemoryViewWrite nextWriteAsync_;
        IOPipeRequest request {};
        bool bShouldReadNext {false};

        bool IsAtRequestedEnd();
        AuByteBuffer *GetBuffer();

        AuSPtr<Protocol::IProtocolStack> pProtocolStack;

    private:
        AuSPtr<IOPipeProcessor> parent_;

        struct /*not a union. the following members are mutex*/
        {
            IOPipeInputData input_;
            AuSPtr<IAsyncTransaction> pAsyncTransaction_;
            AuSPtr<IAsyncStreamAdapter> pAsyncAdapter_;
            AuSPtr<IAsyncStreamReader> pAsyncStreamReader_;
        };

        IOPipeCallback output;
        IOWorkStart startCallback;
        IOWorkEnd endCallback;
        AuUInt64 iStartTickMS_ {};
        bool bActive {true};
        AuUInt32 uBufferSize_ {};
        AuUInt32 uFrameCap_ {};
        AuUInt uBytesWritten_ {};
        AuUInt uBytesWrittenLimit_ {};
        AuUInt uBytesWrittenTarget_ {};
        AuByteBuffer buffer_;
        Utility::ThroughputCalculator throughput_;
    };


    struct IOPipeProcessor : IIOPipeProcessor
    {
        IOPipeProcessor(IOProcessor *parent);

        virtual AuSPtr<IIOPipeWork> NewBasicPipe(const IOPipeRequestBasic &request) override;

        virtual AuSPtr<IIOPipeWork> NewAIOPipe(const IOPipeRequestAIO &request) override;
      
        IOProcessor *parent_;
    };
}