#include <launch-socket.c>
#include <pthread.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

// dont use stdio in these
sig_atomic_t running = 1;
void sig_int_handler(int num)
{
    running = 0;
}

void on_message(struct Server* server, char message[], ssize_t amountRecv, int socketFD)
{
    if (amountRecv > 0) {
        printf("%s\n", message);

        struct AcceptedSocket* sel_socket = server->accepted_socket;
        while (sel_socket != NULL) {
            if (sel_socket->socketFD != socketFD) {
                send(sel_socket->socketFD, message, 30000, 0);
            }
            sel_socket = sel_socket->next;
        }
    }
}

int main(int argc, char* argv[])
{
    if (argc < 2) {
        fprintf(stderr, "arg1 must be set for port\n");
        exit(-1);
    }
    int port = atoi(argv[1]);
    if (port == 0) {
        fprintf(stderr, "port must not be 0\n");
        exit(-1);
    }

    struct Server server = server_constructor(AF_INET, SOCK_STREAM, 0, INADDR_ANY, port, 10, launch_socket);
    server.on_message = on_message;
    pthread_t server_thread = server.launch(&server);

    struct sigaction act = { 0 };
    act.sa_handler = &sig_int_handler;
    sigaction(SIGINT, &act, NULL);

    while (running) { }

    server_close_socket(&server);
    pthread_join(server_thread, NULL);

    return 0;
}