#include "dht.h"
#include <stdio.h>
#include <string.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
#include <errno.h>

// #define cpy(x, y, z) char *x_c = x; char *y_c = y; while(*y_c && z--) *x_c++ = *y_c++

int server_socket_udp;

void build_lookup_request(lookup_request* lr){
    lr->hash_id = ntohs(lr->hash_id);
    lr->node_id = ntohs(lr->node_id);
    lr->node_ip.s_addr = ntohl(lr->node_ip.s_addr);
    lr->node_port = ntohs(lr->node_port);
    // snprintf(lr->node_request, HTTP_MAX_SIZE, "%d|%hd|%d|%u|%d\r\n\r\n", lr->message_type, ntohs(lr->hash_id), ntohs(lr->node_id), ntohl(ip.s_addr), ntohs(lr->node_port));
}



void lookup_request_init(lookup_request* lr, uint8_t message_type, uint16_t node_id, in_addr_t node_ip, uint16_t node_port, uint16_t hash_id)
{
    lr->message_type = message_type;
    lr->node_id = node_id;
    lr->node_ip.s_addr = node_ip;
    lr->node_port = node_port;
    lr->hash_id = hash_id;
}

void extract_lookup_request(lookup_request* lr){
    build_lookup_request(lr);
}

int call_network(lookup_request* lr, in_addr_t ip, uint16_t port){
    // int succ_sock;
    struct addrinfo hints, *servinfo; // *p;
    int rv;

    struct sockaddr_in addr;

    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = htonl(ip);
    addr.sin_port = htons(port);

    memset(&hints, 0, sizeof hints);
    hints.ai_family = AF_INET;
    hints.ai_socktype = SOCK_DGRAM;

    char port_str[6];
    snprintf(port_str, sizeof port_str, "%d", port);

    char ip_str[INET_ADDRSTRLEN];

    inet_ntop(AF_INET, &addr.sin_addr, ip_str, INET_ADDRSTRLEN);

    if ((rv = getaddrinfo(ip_str, port_str, &hints, &servinfo)) != 0) {
        fprintf(stderr, "Func: call_network - getaddrinfo: %s\n", gai_strerror(rv));
        return 1;
    }

    addr = *((struct sockaddr_in*) servinfo->ai_addr);
    freeaddrinfo(servinfo);

    perror("\n\n---CALL_NETWORK---\n");
    perror("[Deserialized]\n");
    print_lookup_request(lr);
    build_lookup_request(lr);
    // perror("[Serialized]\n");
    // print_lookup_request(lr);
    fprintf(stderr, "Sending to Port ---> %u\n", port);
    perror("------------------\n");

    if (sendto(server_socket_udp, lr, sizeof(lookup_request), 0, (struct sockaddr*) &addr, sizeof(struct sockaddr_in)) == -1) {
        fprintf(stderr, "Func: call_network - sendto failed with errno %d: %s\n", errno, strerror(errno));
        fprintf(stderr, "Request:\n\tIP:\t%s\n\tPORT:\t%d\n", ip_str, ntohs(addr.sin_port));
        fprintf(stderr, "\tUDP_SOCKET: %d\n", server_socket_udp);
        exit(1);
    }

    return 0;
}

void clone_lookup_request(lookup_request* dest, lookup_request* source){
    memcpy(dest, source, sizeof(lookup_request));
}

void print_lookup_request(lookup_request *lr){
    fprintf(stderr, "LOOKUP_REQUEST:\n\tMessage-type: %d\n\tNode-ID: %u\n\tNode-IP: %s\t(reversed)\n\tNode-PORT: %u\n\tHash-ID: %u\n", lr->message_type, lr->node_id, inet_ntoa(lr->node_ip), lr->node_port, lr->hash_id);
}