// Copyright (c) 2012-2021 Wojciech Figat. All rights reserved.

#pragma once

#include "Engine/Networking/Types.h"
#include "Engine/Networking/INetworkDriver.h"
#include "Engine/Networking/NetworkConnection.h"
#include "Engine/Networking/NetworkConfig.h"

#include "Engine/Core/Collections/Dictionary.h"
#include "Engine/Scripting/ScriptingType.h"

/// <summary>
/// Low-level network transport interface implementation based on ENet library.
/// </summary>
API_CLASS(Namespace="FlaxEngine.Networking", Sealed) class FLAXENGINE_API ENetDriver : public INetworkDriver
{
DECLARE_SCRIPTING_TYPE_MINIMAL(ENetDriver);
public:
    void Initialize(NetworkPeer* host, const NetworkConfig& config) override;
    void Dispose() override;
    
    bool Listen() override;
    bool Connect() override;
    void Disconnect() override;
    void Disconnect(const NetworkConnection& connection) override;

    bool PopEvent(NetworkEvent* eventPtr) override;
    
    void SendMessage(NetworkChannelType channelType, const NetworkMessage& message) override;
    void SendMessage(NetworkChannelType channelType, const NetworkMessage& message, NetworkConnection target) override;
    void SendMessage(NetworkChannelType channelType, const NetworkMessage& message, const Array<NetworkConnection, HeapAllocation>& targets) override;

private:
    bool IsServer() const
    {
        return _host != nullptr && _peer == nullptr;
    }
    
private:
    NetworkConfig _config;
    NetworkPeer* _networkHost;
    void* _host = nullptr;
    void* _peer = nullptr;
    
    Dictionary<uint32, void*> _peerMap;
};