#include <stdio.h>
#include <string.h>
#include <unistd.h>

#include "core.h"

ECPContext ctx_s;
ECPSocket sock_s;
ECPDHKey key_perma_s;
ECPConnHandler handler_s;

ECPContext ctx_c;
ECPSocket sock_c;
ECPDHKey key_perma_c;
ECPConnHandler handler_c;

ECPNode node;
ECPConnection conn;

#define CTYPE_TEST  0
#define MTYPE_MSG   8

ssize_t handle_open_c(ECPConnection *conn, ecp_seq_t sq, unsigned char t, unsigned char *p, ssize_t s) {
    uint32_t seq = 0;
    
    ecp_conn_handle_open(conn, sq, t, p, s);
    if (s < 0) {
        printf("OPEN ERR:%ld\n", s);
        return s;
    }
    
    unsigned char payload[ECP_SIZE_PLD(1000, 0)];
    unsigned char *buf = ecp_pld_get_buf(payload, 0);
    char *msg = "PERA JE CAR!";

    ecp_pld_set_type(payload, MTYPE_MSG);
    strcpy((char *)buf, msg);
    ssize_t _rv = ecp_pld_send(conn, payload, sizeof(payload));
    return s;
}

ssize_t handle_msg_c(ECPConnection *conn, ecp_seq_t sq, unsigned char t, unsigned char *p, ssize_t s) {
    printf("MSG C:%s size:%ld\n", p, s);
    return s;
}

ssize_t handle_msg_s(ECPConnection *conn, ecp_seq_t sq, unsigned char t, unsigned char *p, ssize_t s) {
    printf("MSG S:%s size:%ld\n", p, s);

    unsigned char payload[ECP_SIZE_PLD(1000, 0)];
    unsigned char *buf = ecp_pld_get_buf(payload, 0);
    char *msg = "VAISTINU JE CAR!";

    ecp_pld_set_type(payload, MTYPE_MSG);
    strcpy((char *)buf, msg);
    ssize_t _rv = ecp_pld_send(conn, payload, sizeof(payload));

    return s;
}

int main(int argc, char *argv[]) {
    int rv;
    
    rv = ecp_init(&ctx_s);
    printf("ecp_init RV:%d\n", rv);
    
    rv = ecp_conn_handler_init(&handler_s);
    handler_s.msg[MTYPE_MSG] = handle_msg_s;
    ctx_s.handler[CTYPE_TEST] = &handler_s;
    
    rv = ecp_dhkey_generate(&ctx_s, &key_perma_s);
    printf("ecp_dhkey_generate RV:%d\n", rv);
    
    rv = ecp_sock_create(&sock_s, &ctx_s, &key_perma_s);
    printf("ecp_sock_create RV:%d\n", rv);

    rv = ecp_sock_open(&sock_s, "0.0.0.0:3000");
    printf("ecp_sock_open RV:%d\n", rv);
    
    rv = ecp_start_receiver(&sock_s);
    printf("ecp_start_receiver RV:%d\n", rv);

    rv = ecp_init(&ctx_c);
    printf("ecp_init RV:%d\n", rv);

    rv = ecp_conn_handler_init(&handler_c);
    handler_c.msg[ECP_MTYPE_OPEN] = handle_open_c;
    handler_c.msg[MTYPE_MSG] = handle_msg_c;
    ctx_c.handler[CTYPE_TEST] = &handler_c;
    
    rv = ecp_dhkey_generate(&ctx_c, &key_perma_c);
    printf("ecp_dhkey_generate RV:%d\n", rv);
    
    rv = ecp_sock_create(&sock_c, &ctx_c, &key_perma_c);
    printf("ecp_sock_create RV:%d\n", rv);

    rv = ecp_sock_open(&sock_c, NULL);
    printf("ecp_sock_open RV:%d\n", rv);
    
    rv = ecp_start_receiver(&sock_c);
    printf("ecp_start_receiver RV:%d\n", rv);

    rv = ecp_node_init(&ctx_c, &node, &key_perma_s.public, "127.0.0.1:3000");
    printf("ecp_node_init RV:%d\n", rv);

    rv = ecp_conn_create(&conn, &sock_c, CTYPE_TEST);
    printf("ecp_conn_create RV:%d\n", rv);

    rv = ecp_conn_open(&conn, &node);
    printf("ecp_conn_open RV:%d\n", rv);
    
    while (1) sleep(1);
}