/*
 * Copyright (c) Contributors to the Open 3D Engine Project.
 * For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#pragma once

#include <Source/AutoGen/MultiplayerEditor.AutoPacketDispatcher.h>
#include <AzCore/IO/ByteContainerStream.h>
#include <AzCore/Settings/SettingsRegistry.h>
#include <AzFramework/Spawnable/InMemorySpawnableAssetContainer.h>
#include <AzNetworking/ConnectionLayer/IConnectionListener.h>

namespace AzNetworking
{
    class INetworkInterface;
}

namespace Multiplayer
{
    //! MultiplayerEditorConnection is a connection listener to synchronize the Editor and a local server it launches
    class MultiplayerEditorConnection final
        : public AzNetworking::IConnectionListener
    {
    public:
        MultiplayerEditorConnection();
        ~MultiplayerEditorConnection();

        bool HandleRequest(AzNetworking::IConnection* connection, const AzNetworking::IPacketHeader& packetHeader, MultiplayerEditorPackets::EditorServerReadyForLevelData& packet);
        bool HandleRequest(AzNetworking::IConnection* connection, const AzNetworking::IPacketHeader& packetHeader, MultiplayerEditorPackets::EditorServerLevelData& packet);
        bool HandleRequest(AzNetworking::IConnection* connection, const AzNetworking::IPacketHeader& packetHeader, MultiplayerEditorPackets::EditorServerReady& packet);
        
        //! IConnectionListener interface
        //! @{
        AzNetworking::ConnectResult ValidateConnect(const AzNetworking::IpAddress& remoteAddress, const AzNetworking::IPacketHeader& packetHeader, AzNetworking::ISerializer& serializer) override;
        void OnConnect(AzNetworking::IConnection* connection) override;
        AzNetworking::PacketDispatchResult OnPacketReceived(AzNetworking::IConnection* connection, const AzNetworking::IPacketHeader& packetHeader, AzNetworking::ISerializer& serializer) override;
        void OnPacketLost([[maybe_unused]]AzNetworking::IConnection* connection, [[maybe_unused]]AzNetworking::PacketId packetId) override {}
        void OnDisconnect([[maybe_unused]]AzNetworking::IConnection* connection, [[maybe_unused]]AzNetworking::DisconnectReason reason, [[maybe_unused]]AzNetworking::TerminationEndpoint endpoint) override {}
        //! @}

    private:
        void ActivateDedicatedEditorServer() const;

        AzNetworking::INetworkInterface* m_networkEditorInterface = nullptr;
        AZStd::vector<uint8_t> m_buffer;
        AZ::IO::ByteContainerStream<AZStd::vector<uint8_t>> m_byteStream;
        mutable bool m_isActivated = false;
        AZ::SettingsRegistryInterface::NotifyEventHandler m_componentApplicationLifecycleHandler;
        AZStd::unique_ptr<AzFramework::InMemorySpawnableAssetContainer> m_inMemorySpawnableAssetContainer = nullptr;
    };
}