#include "client.h"
#include "chatroom.h"
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#include <arpa/inet.h>

extern client_t *clients[];
extern pthread_mutex_t clients_mutex;

void send_to_client(int sock, const char *msg) {
    send(sock, msg, strlen(msg), 0);
}

void broadcast_message(const char *message, int sender_sock) {
    pthread_mutex_lock(&clients_mutex);

    client_t *sender = NULL;
    for (int i = 0; i < 10; i++) {
        if (clients[i] && clients[i]->socket == sender_sock) {
            sender = clients[i];
            break;
        }
    }

    if (sender == NULL || strlen(sender->room) == 0) {
        pthread_mutex_unlock(&clients_mutex);
        return;
    }

    for (int i = 0; i < 10; i++) {
        if (clients[i] && clients[i]->socket != sender_sock &&
            strcmp(clients[i]->room, sender->room) == 0) {
            send_to_client(clients[i]->socket, message);
        }
    }

    pthread_mutex_unlock(&clients_mutex);
}

void *handle_client(void *arg) {
    client_t *cli = (client_t *)arg;
    char buffer[BUFFER_SIZE];
    int bytes;

    while ((bytes = recv(cli->socket, buffer, BUFFER_SIZE - 1, 0)) > 0) {
        buffer[bytes] = '\0';

        if (strncmp(buffer, "/join ", 6) == 0) {
            char room[50];
            sscanf(buffer + 6, "%s", room);
            if (join_chatroom(room, cli)) {
                char msg[128];
                snprintf(msg, sizeof(msg), "You joined '%s'\n", room);
                send_to_client(cli->socket, msg);
            } else {
                send_to_client(cli->socket, "Room does not exist.\n");
            }
        }
        else if (strncmp(buffer, "/list", 5) == 0) {
            print_chatroom_list(cli->socket);
        }
        else if (strncmp(buffer, "/users", 6) == 0) {
            pthread_mutex_lock(&clients_mutex);
            int count = 0;
            for (int i = 0; i < 10; i++)
                if (clients[i] && strcmp(clients[i]->room, cli->room) == 0)
                    count++;

            char header[BUFFER_SIZE];
            snprintf(header, sizeof(header), "[you are in '%s' among %d]\n\n", cli->room, count);
            send_to_client(cli->socket, header);

            for (int i = 0; i < 10; i++) {
                if (clients[i] && strcmp(clients[i]->room, cli->room) == 0) {
                    char line[BUFFER_SIZE];
                    snprintf(line, sizeof(line), "%s@%s\n", clients[i]->username,
                             inet_ntoa(clients[i]->address.sin_addr));
                    send_to_client(cli->socket, line);
                }
            }
            pthread_mutex_unlock(&clients_mutex);
        }
        else {
            broadcast_message(buffer, cli->socket);
        }
    }

    close(cli->socket);
    pthread_mutex_lock(&clients_mutex);
    for (int i = 0; i < 10; i++) {
        if (clients[i] == cli) {
            leave_chatroom(cli);
            clients[i] = NULL;
            break;
        }
    }
    pthread_mutex_unlock(&clients_mutex);
    free(cli);
    pthread_exit(NULL);
}