#pragma once

#include "remote_server_session_semaphore.h"
#include "session_impl.h"

#ifdef _MSC_VER
#pragma warning(push)
#pragma warning(disable : 4250) //  'NBus::NPrivate::TRemoteClientSession' : inherits 'NBus::NPrivate::TBusSessionImpl::NBus::NPrivate::TBusSessionImpl::GetConfig' via dominance
#endif

namespace NBus {
    namespace NPrivate {
        class TRemoteServerSession: public TBusServerSession, public TBusSessionImpl {
            friend class TRemoteServerConnection;

        private:
            TObjectCounter<TRemoteServerSession> ObjectCounter;

            TRemoteServerSessionSemaphore ServerOwnedMessages;
            IBusServerHandler* const ServerHandler;

        public:
            TRemoteServerSession(TBusMessageQueue* queue, TBusProtocol* proto,
                                 IBusServerHandler* handler,
                                 const TBusSessionConfig& config, const TString& name);

            void OnMessageReceived(TRemoteConnection* c, TVectorSwaps<TBusMessagePtrAndHeader>& newMsg) override;
            void InvokeOnMessage(TBusMessagePtrAndHeader& request, TIntrusivePtr<TRemoteServerConnection>& conn);

            EMessageStatus SendReply(const TBusIdentity& ident, TBusMessage* pRep) override;

            EMessageStatus ForgetRequest(const TBusIdentity& ident) override;

            int GetInFlight() const noexcept override;
            void FillStatus() override;

            void Shutdown() override;

            void PauseInput(bool pause) override;
            unsigned GetActualListenPort() override;

            void AcquireInWorkRequests(TArrayRef<const TBusMessagePtrAndHeader> requests);
            void ReleaseInWorkResponses(TArrayRef<const TBusMessagePtrAndHeader> responses);
            void ReleaseInWorkRequests(TRemoteConnection&, TBusMessage*);
            void ReleaseInWork(TBusIdentity&);
            void ConvertInWork(TBusIdentity& req, TBusMessage* reply);
        };

#ifdef _MSC_VER
#pragma warning(pop)
#endif

    }
}