#include "rnet.h"

void on_client_connect(rnet_socket_t *sock) { printf("%s connected\n", sock->name); }
void on_client_read(rnet_socket_t *sock) {
    unsigned char *data = net_socket_read(sock, 4096);
    if (!data)
        return;
    char *http_headers = "HTTP/1.1 200 OK\r\nContent-Length: 10\r\nConnection: close\r\n\r\n";
    net_socket_write(sock, (unsigned char *)http_headers, strlen(http_headers));
    rnet_safe_str((char *)data, sock->bytes_received);
    // data[11] = 0;
    printf("%s: %.30s\n", sock->name, data);
    net_socket_write(sock, data, strlen((char *)data));
    if (!strncmp((char *)data, "GET ", 4))
        net_socket_close(sock);
}
void on_client_close(rnet_socket_t *sock) { printf("%s disconnected\n", sock->name); }

int main(int argc, char *argv[]) {
    if (argc < 2) {
        printf("usage: [port].\n");
        return 1;
    }
    for (int i = 0; i < argc; i++) {
        if (strcmp(argv[i], "test") == 0) {
            printf("Skipping rnet tests.\n");
            return 0;
        }
    }
    rnet_server_t *server = net_socket_serve((unsigned int)atoi(argv[1]), 10);
    server->on_connect = on_client_connect;
    server->on_read = on_client_read;
    server->on_close = on_client_close;
    while (true) {
        if (net_socket_select(server)) {
            printf("Handled all events.\n");
        } else {
            printf("No events to handle.\n");
        }
    }
    return 0;
}