#include <cstdlib>
#include <libpq-fe.h>
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/un.h>
#include <unistd.h>

// SERVER STRUCTS
#include "server_structs.h"
#include "stack.h"

#define SERVER_SOCKET_PATH "/tmp/ACplayer_socket"

const int STACK_SIZE = 512;

Stack api_queue(STACK_SIZE);

void checkConn(PGconn *conn) {
	if (PQstatus(conn) != CONNECTION_OK) {
		fprintf(stderr, "[!] Connection to database failed: %s", PQerrorMessage(conn));
		PQfinish(conn);
		exit(1);
	}
}

void *db_write_thread(void *arg) {

	(void)arg; // Unused parameter

	const char *conninfo = "host=localhost port=5432 dbname=acservers_players user=admin password=123wer789";
	PGconn *conn = PQconnectdb(conninfo);

	checkConn(conn);
	if (PQstatus(conn) == CONNECTION_OK) {
		printf("[+] Database connection established successfully.\n");
	}

	while (1) {
		api_packet packet;

		if (api_queue.pop(packet)) {

			printf("[W] Writing packet for Server with tracker ID: %d to database.\tQueue: %u/%d\n", packet.tracker_id, (unsigned int)api_queue.size(), STACK_SIZE);

			const char *query = "INSERT INTO servers ("
			                    "server_id, server_name, session_type, session_count, server_track, "
			                    "server_config, server_weather_graphics, typ, session_time, session_laps, "
			                    "session_wait_time, session_ambient_temp, session_road_temp, session_elapsed_time, connected_players"
			                    ") "
			                    "VALUES ($1, $2, $4, $3, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15) "
			                    "ON CONFLICT (server_id) DO UPDATE SET "
			                    "session_type = EXCLUDED.session_type, "
			                    "session_count = EXCLUDED.session_count, "
			                    "server_track = EXCLUDED.server_track, "
			                    "server_config = EXCLUDED.server_config, "
			                    "server_weather_graphics = EXCLUDED.server_weather_graphics, "
			                    "typ = EXCLUDED.typ, "
			                    "session_time = EXCLUDED.session_time, "
			                    "session_laps = EXCLUDED.session_laps, "
			                    "session_wait_time = EXCLUDED.session_wait_time, "
			                    "session_ambient_temp = EXCLUDED.session_ambient_temp, "
			                    "session_road_temp = EXCLUDED.session_road_temp, "
			                    "session_elapsed_time = EXCLUDED.session_elapsed_time, "
			                    "connected_players = EXCLUDED.connected_players "
			                    "RETURNING server_id;";

			const char *server_name = packet.track_info.server_name;
			const char *server_track = packet.track_info.track;
			const char *server_config = packet.track_info.track_config;
			const char *server_weather_graphics = packet.track_info.weather_graphics;

			std::string server_id_str = std::to_string(packet.tracker_id);
			std::string session_count_s = std::to_string(packet.track_info.session_count);
			std::string session_type_s = std::to_string(packet.track_info.session_type);
			std::string typ_s = std::to_string(packet.track_info.typ);
			std::string session_time_s = std::to_string(packet.track_info.time);
			std::string session_laps_s = std::to_string(packet.track_info.laps);
			std::string session_wait_s = std::to_string(packet.track_info.wait_time);
			std::string ambient_temp_s = std::to_string(packet.track_info.ambient_temp);
			std::string road_temp_s = std::to_string(packet.track_info.road_temp);
			std::string elapsed_time_s = std::to_string((u_int16_t)packet.track_info.elapsed_ms);

			std::string connected_players_s = std::to_string(packet.connected_cars);

			const char *paramValues[15] = {server_id_str.c_str(),
			                               server_name,
			                               session_count_s.c_str(),
			                               session_type_s.c_str(),
			                               server_track,
			                               server_config,
			                               server_weather_graphics,
			                               typ_s.c_str(),
			                               session_time_s.c_str(),
			                               session_laps_s.c_str(),
			                               session_wait_s.c_str(),
			                               ambient_temp_s.c_str(),
			                               road_temp_s.c_str(),
			                               elapsed_time_s.c_str(),
			                               connected_players_s.c_str()};

			PGresult *res = PQexecParams(conn, query, 15, nullptr, paramValues, nullptr, nullptr, 0);

			if (PQresultStatus(res) != PGRES_TUPLES_OK) {
				printf("[!] Insert failed: %s", PQerrorMessage(conn));
				PQclear(res);
				PQfinish(conn);
				break;
			}

			// Time for users table insert/UPDATE

            if (packet.connected_cars == 0) {
                PQclear(res);
                continue; // No connected cars to process
            }

			for (int i = 0; i < packet.connected_cars; i++) {

				if (packet.cars[i].isConnected == 0) {
					continue; // Skip disconnected cars
				}

				const char *user_query = "INSERT INTO users ("
				                         "driver_guid, driver_name, driver_team, car_model, car_skin, "
				                         "cuts_alltime, contacts_alltime, laps_completed, is_connect, is_loading, current_server"
				                         ") "
				                         "VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11) "
				                         "ON CONFLICT (driver_guid) DO UPDATE SET "
				                         "driver_name = EXCLUDED.driver_name, "
				                         "driver_team = EXCLUDED.driver_team, "
				                         "car_model = EXCLUDED.car_model, "
				                         "car_skin = EXCLUDED.car_skin, "
				                         "cuts_alltime = EXCLUDED.cuts_alltime, "
				                         "contacts_alltime = EXCLUDED.contacts_alltime, "
				                         "laps_completed = EXCLUDED.laps_completed, "
				                         "is_connect = EXCLUDED.is_connect, "
				                         "is_loading = EXCLUDED.is_loading, "
				                         "current_server = EXCLUDED.current_server;";

				const char *driver_guid = packet.cars[i].driver_GUID;
				const char *driver_name = packet.cars[i].driver_name;
				const char *driver_team = packet.cars[i].driver_team;
				const char *car_model = packet.cars[i].car_model;
				const char *car_skin = packet.cars[i].car_skin;

				std::string cuts_alltime_s = std::to_string(packet.cars[i].total_cuts_alltime);
				std::string contacts_alltime_s = std::to_string(packet.cars[i].total_contacts);
				std::string laps_completed_s = std::to_string(packet.cars[i].total_laps_completed);
				std::string is_connect_s = std::to_string(packet.cars[i].isConnected);
				std::string is_loading_s = std::to_string(packet.cars[i].isLoading);
				std::string current_server_s = server_id_str;

				const char *user_paramValues[11] = {driver_guid,
				                                    driver_name,
				                                    driver_team,
				                                    car_model,
				                                    car_skin,
				                                    cuts_alltime_s.c_str(),
				                                    contacts_alltime_s.c_str(),
				                                    laps_completed_s.c_str(),
				                                    is_connect_s.c_str(),
				                                    is_loading_s.c_str(),
				                                    current_server_s.c_str()};
				PGresult *user_res = PQexecParams(conn, user_query, 11, nullptr, user_paramValues, nullptr, nullptr, 0);

				PQclear(user_res);
			}

			PQclear(res);
		} else {
			usleep(1000); // Sleep for 1ms if no packets are available
		}
	}

	PQfinish(conn);
	return NULL;
}

int main(void) {

	int server_fd, client_fd;
	struct sockaddr_un addr;
	socklen_t addr_len;
	ssize_t n;

	api_packet api_packet_storage;

	// Create socket
	server_fd = socket(AF_UNIX, SOCK_STREAM, 0);
	if (server_fd < 0) {
		perror("socket");
		exit(1);
	}
	unlink(SERVER_SOCKET_PATH);

	addr.sun_family = AF_UNIX;
	strcpy(addr.sun_path, SERVER_SOCKET_PATH);
	if (bind(server_fd, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
		perror("bind");
		exit(1);
	}
	if (listen(server_fd, 5) < 0) {
		perror("listen");
		exit(1);
	}

	// Start DB write db_write_thread
	pthread_t db_thread;
	if (pthread_create(&db_thread, NULL, db_write_thread, NULL) != 0) {
		perror("pthread create");
		exit(1);
	}
	// Detach the thread so it cleans up after itself
	pthread_detach(db_thread);

	printf("[+] Server listening on %s\n", SERVER_SOCKET_PATH);

	while (1) {
		client_fd = accept(server_fd, NULL, NULL);
		if (client_fd < 0) {
			perror("[!] accept");
			continue;
		}

		printf("[+] Client connected\n");

		while ((n = read(client_fd, &api_packet_storage, sizeof(api_packet))) > 0) {
			printf("[+] Received %zd bytes\n", n);

			if (api_packet_storage.tracker_id == 65) {
				printf("[+] Handshake Received from server \"%s\" (%d).\n", api_packet_storage.track_info.server_name, api_packet_storage.tracker_id);
				continue;
			}

			// Print info
			printf("[*] Info: Tracker ID: %d (\"%s\")\tQueue: %u/%d\n", api_packet_storage.tracker_id, api_packet_storage.track_info.server_name, (unsigned int)api_queue.size(),
			       STACK_SIZE);
			if (!api_queue.push(api_packet_storage)) {
				printf("[!] API Queue full, dropping packet from Tracker ID: %d\n", api_packet_storage.tracker_id);
			}
			usleep(1500);
		}

		if (n < 0)
			perror("[!] Error on read");
		else
			printf("[+] Client disconnected\n");

		close(client_fd);
	}

	return 0;
}