#include<arpa/inet.h>
#include<sys/socket.h>
#include<stdint.h>
#include<pthread.h>
#include<stdio.h>
#include<string.h>

#include "main.h"
#include "vec.h"
#include "net.h"

struct Connection {
    int socket;
    uint32_t id;
};

void * server_handle_conn(void * args_) {
    struct Connection * args = args_;

    printf("THREAD(%u): start\n", args->id);
    while(1) {
        uint8_t buf[1024];
        while(1) {
            int len = recv(args->socket, buf, 1024, MSG_WAITALL);
            if(len <= 0) break;
            Message msg;
            if(msg_deserialize(buf, len, &msg) == 0) {
                
            } else {
                printf("Couldn't parse message.\n");
            }
        }
        printf("THREAD(%u): connection closed\n", args->id);
    }

    free(args);
    return NULL;
}

void server_run(uint16_t port) {
    printf("SERVER: start\n");
    int sock = socket(AF_INET, SOCK_STREAM, 0);
    if(sock < 0) panicf("Couldn't open socket\n");
    struct sockaddr_in addr = {};
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = htonl(INADDR_ANY);
    addr.sin_port = htons(port);
    if(bind(sock, (struct sockaddr*)&addr, sizeof(addr)) != 0)
        panicf("Couldn't bind to the socket\n");
    if(listen(sock, 16) != 0)
        panicf("Couldn't listen on socket\n");

    uint32_t ids = 0;
    while(1) {
        struct sockaddr con_addr;
        socklen_t con_len = sizeof(con_addr);
        struct Connection conn;

        conn.socket = accept(sock, &con_addr, &con_len);

        if(conn.socket >= 0) {
            printf("SERVER: got connection\n");

            conn.id = ids++;

            struct Connection * conn_ptr = malloc(sizeof(struct Connection));
            memcpy(conn_ptr, &conn, sizeof(struct Connection));

            pthread_t thread;
            pthread_create(&thread, NULL, server_handle_conn, conn_ptr);
        } else {
            printf("Couldn't accept connection (%d)\n", conn.socket);
        }
    }
}