/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
#pragma once
#include <GridMate/Drillers/ReplicaDriller.h>
#include "ByteReporter.h"
#include <GridMate/Drillers/CarrierDriller.h>

namespace MultiplayerDiagnostics
{
    /**
     * \brief GridMate network live analysis tool via ImGui.
     */
    class ImGuiServerManager
        : public GridMate::Debug::ReplicaDrillerBus::Handler
        , public GridMate::Debug::CarrierDrillerBus::Handler
    {
    public:
        ImGuiServerManager();
        virtual ~ImGuiServerManager();

        // main update loop
        void OnImGuiUpdate();

        // ReplicaDrillerBus - receive

        void OnReceiveReplicaBegin(GridMate::Replica* replica, const void* data, size_t len) override;
        void OnReceiveReplicaEnd(GridMate::Replica* replica) override;
        void OnReceiveReplicaChunkEnd(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex) override;
        void OnReceiveDataSet(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, GridMate::DataSetBase* dataSet, GridMate::PeerId from, GridMate::PeerId to, const void* data, size_t len) override;
        void OnReceiveRpc(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, GridMate::Internal::RpcRequest* rpc, GridMate::PeerId from, GridMate::PeerId to, const void* data, size_t len) override;

        // ReplicaDrillerBus - sending

        void OnSendReplicaBegin(GridMate::Replica* replica) override;
        void OnSendReplicaEnd(GridMate::Replica* replica, const void* data, size_t len) override;
        void OnSendReplicaChunkEnd(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, const void* data, size_t len) override;
        void OnSendDataSet(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, GridMate::DataSetBase* dataSet, GridMate::PeerId from, GridMate::PeerId to, const void* data, size_t len) override;
        void OnSendRpc(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, GridMate::Internal::RpcRequest* rpc, GridMate::PeerId from, GridMate::PeerId to, const void* data, size_t len) override;

        // CarrierDrillerBus
        void OnIncomingConnection (GridMate::Carrier* carrier, GridMate::ConnectionID id) override;
        void OnFailedToConnect (GridMate::Carrier* carrier,
            GridMate::ConnectionID id,
            GridMate::CarrierDisconnectReason reason) override;
        void OnConnectionEstablished (GridMate::Carrier* carrier, GridMate::ConnectionID id) override;
        void OnDisconnect (GridMate::Carrier* carrier,
            GridMate::ConnectionID id,
            GridMate::CarrierDisconnectReason reason) override;
        void OnDriverError (GridMate::Carrier* carrier,
            GridMate::ConnectionID id,
            const GridMate::DriverError& error) override;
        void OnSecurityError (GridMate::Carrier* carrier,
            GridMate::ConnectionID id,
            const GridMate::SecurityError& error) override;
        void OnUpdateStatistics (const GridMate::string& address,
            const GridMate::TrafficControl::Statistics& lastSecond,
            const GridMate::TrafficControl::Statistics& lifeTime,
            const GridMate::TrafficControl::Statistics& effectiveLastSecond,
            const GridMate::TrafficControl::Statistics& effectiveLifeTime) override;
        void OnConnectionStateChanged (GridMate::Carrier* carrier,
            GridMate::ConnectionID id,
            GridMate::Carrier::ConnectionStates newState) override;

    private:

        bool m_showServerReportWindow = false;
        bool m_isTrackingMessages = false;

        AZStd::map<AZStd::string, EntityReporter> m_sendingEntityReports{};
        EntityReporter m_currentSendingEntityReport;

        AZStd::map<AZStd::string, EntityReporter> m_receivingEntityReports{};
        EntityReporter m_currentReceivingEntityReport;

        float m_replicatedStateKbpsWarn = 10.f;
        float m_replicatedStateMaxSizeWarn = 30.f;

        void UpdateTrafficStatistics();
        AZStd::map<GridMate::string, GridMate::TrafficControl::Statistics> m_lastSecondStats;
    };
}