#include "TracyProtocolServer.hpp"
#include <cassert>
#include <cstring>
#include <iostream>

namespace tracy
{
std::optional<tracy::BroadcastMessage> ParseBroadcastMessage( const char* msg, size_t msgLen )
{
    if( msgLen < sizeof( uint16_t ) )
    {
        std::cout << "Received too short broadcast message" << std::endl;
        return std::nullopt;
    }
    uint16_t broadcastVersion;
    memcpy( &broadcastVersion, msg, sizeof( uint16_t ) );
    if( broadcastVersion > tracy::BroadcastVersion )
    {
        std::cout << "Received broadcast message with unsupported version: " << broadcastVersion << std::endl;
        return std::nullopt;
    }
    switch( broadcastVersion )
    {
    case 3:
    {
        if( msgLen > sizeof( tracy::BroadcastMessage ) )
        {
            std::cout << "Received unexpected size broadcast v3 message" << std::endl;
            return std::nullopt;
        }
        tracy::BroadcastMessage bm;
        memcpy( &bm, msg, msgLen );
        return bm;
        break;
    }
    case 2:
    {
        if( msgLen > sizeof( tracy::BroadcastMessage_v2 ) )
        {
            std::cout << "Received unexpected size broadcast v2 message" << std::endl;
            return std::nullopt;
        }
        tracy::BroadcastMessage_v2 bm;
        memcpy( &bm, msg, msgLen );

        tracy::BroadcastMessage out;
        out.broadcastVersion = broadcastVersion;
        out.protocolVersion = bm.protocolVersion;
        out.activeTime = bm.activeTime;
        out.listenPort = bm.listenPort;
        strcpy( out.programName, bm.programName );
        out.pid = 0;
        return out;
        break;
    }
    case 1:
    {
        if( msgLen > sizeof( tracy::BroadcastMessage_v1 ) )
        {
            std::cout << "Received unexpected size broadcast v1 message" << std::endl;
            return std::nullopt;
        }
        tracy::BroadcastMessage_v1 bm;
        memcpy( &bm, msg, msgLen );

        tracy::BroadcastMessage out;
        out.broadcastVersion = broadcastVersion;
        out.protocolVersion = bm.protocolVersion;
        out.activeTime = bm.activeTime;
        out.listenPort = bm.listenPort;
        strcpy( out.programName, bm.programName );
        out.pid = 0;
        return out;
        break;
    }
    case 0:
    {
        if( msgLen > sizeof( tracy::BroadcastMessage_v0 ) )
        {
            std::cout << "Received unexpected size broadcast v0 message" << std::endl;
            return std::nullopt;
        }
        tracy::BroadcastMessage_v0 bm;
        memcpy( &bm, msg, msgLen );

        tracy::BroadcastMessage out;
        out.broadcastVersion = broadcastVersion;
        out.protocolVersion = bm.protocolVersion;
        out.activeTime = bm.activeTime;
        out.listenPort = tracy::DEFAULT_CLIENT_DATA_TCP_PORT;
        strcpy( out.programName, bm.programName );
        out.pid = 0;
        return out;
        break;
    }
    default:
        assert( false );
        break;
    }
    return std::nullopt;
}

uint64_t ClientUniqueID( tracy::IpAddress const& addr, uint16_t port )
{
    return uint64_t( addr.GetNumber() ) | ( uint64_t( port ) << 32 );
}
}