#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>

#include "rep.h"

void
handle_client(int client_socket)
{
    char buffer[BUFFER_SIZE] = {0};
    read(client_socket, buffer, BUFFER_SIZE);

    printf("Domain query: %s\n", buffer);

    for (int i = 0; records[i].domain[0] != '\0'; i++) {
        if (strcmp(records[i].domain, buffer) == 0) {
            write(client_socket, records[i].ip, strlen(records[i].ip));
            write(client_socket, "\n", 1);
            close(client_socket);
            return;
        }
    }

    // If domain not found
    char *not_found = "Domain not found";
    write(client_socket, not_found, strlen(not_found));
    write(client_socket, "\n", 1);
    close(client_socket);
}

int
main()
{
    int server_fd, client_socket;
    struct sockaddr_in address;
    int opt = 1;
    int addrlen = sizeof(address);

    server_fd = socket(AF_INET, SOCK_STREAM, 0);
    setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt));

    address.sin_family = AF_INET;
    address.sin_addr.s_addr = INADDR_ANY;
    address.sin_port = htons(RNS_PORT);

    bind(server_fd, (struct sockaddr *)&address, sizeof(address));
    listen(server_fd, 3);

    while (1) {
        client_socket = accept(server_fd, (struct sockaddr *)&address, (socklen_t *)&addrlen);
        if (client_socket >= 0) {
            handle_client(client_socket);
        }
    }

    return 0;
}