#include "Packet.h"
#include <cstdlib>
#include <vector>
#include <cstring>

uint32_t to_network_uint32(uint32_t host) {
    return htonl(host);
}

uint32_t from_network_uint32(uint32_t net) {
    return ntohl(net);
}

bool send_all(SOCKET sock, const char* data, size_t len) {
    size_t totalSent = 0;
    while(totalSent < len) {
        int sent = send(sock, data + totalSent, static_cast<int>(len - totalSent), 0);
        if(sent <= 0)
            return false;
        totalSent += sent;
    }
    return true;
}

bool recv_all(SOCKET sock, char* buffer, size_t len) {
    size_t totalRecv = 0;
    while(totalRecv < len) {
        int recvd = recv(sock, buffer + totalRecv, static_cast<int>(len - totalRecv), 0);
        if(recvd <= 0)
            return false;
        totalRecv += recvd;
    }
    return true;
}

bool sendPacket(SOCKET sock, const Packet& packet) {
    // Payload length = 1 byte for type + data length.
    uint32_t payloadLen = 1 + static_cast<uint32_t>(packet.data.size());
    uint32_t netLen = to_network_uint32(payloadLen);
    if(!send_all(sock, reinterpret_cast<const char*>(&netLen), sizeof(netLen)))
        return false;
    if(!send_all(sock, reinterpret_cast<const char*>(&packet.type), sizeof(packet.type)))
        return false;
    if(!packet.data.empty()) {
        if(!send_all(sock, packet.data.data(), packet.data.size()))
            return false;
    }
    return true;
}

bool receivePacket(SOCKET sock, Packet &packet) {
    uint32_t netLen = 0;
    if(!recv_all(sock, reinterpret_cast<char*>(&netLen), sizeof(netLen)))
        return false;
    uint32_t len = from_network_uint32(netLen);
    if(len < 1)
        return false;
    char typeChar;
    if(!recv_all(sock, &typeChar, 1))
        return false;
    packet.type = static_cast<PacketType>(typeChar);
    size_t dataLen = len - 1;
    if(dataLen > 0) {
        std::vector<char> buf(dataLen);
        if(!recv_all(sock, buf.data(), dataLen))
            return false;
        packet.data = std::string(buf.begin(), buf.end());
    } else {
        packet.data.clear();
    }
    return true;
}