#include <cstddef>
#include <iterator>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>

#include "parcer.h"
#include "server_structs.h"
#include "socket.h"

const int SERVER_OUT_PORT = 12000;
const int SERVER_IN_PORT = 11000;
const char *SERVER_OUT_IP = "127.0.0.1";

int main(void) {

	printf("[+] Starting server...\n");
	printf("[+] Server listening on port %d\n", SERVER_IN_PORT);
	printf("[+] Server sending to %s:%d\n\n", SERVER_OUT_IP, SERVER_OUT_PORT);

	int send_sock_fd = connect_udp_socket("127.0.0.1", 11000);    // SEND requests to Server
	int recv_sock_fd = create_bound_udp_socket("0.0.0.0", 12000); // LISTEN to server updates

	if (send_sock_fd < 0 || recv_sock_fd < 0) {
		fprintf(stderr, "[!] Failed to create sockets (%d)(%d)\n", send_sock_fd, recv_sock_fd);
		return EXIT_FAILURE;
	}

	unsigned long offset = 0;
	int ok = 1;
	char buffer[1024];

	trackAtributes trackInfo;
	carAtributes update;

	trackInfo.server_name = (char *)malloc(128);
	trackInfo.track = (char *)malloc(128);
	trackInfo.track_config = (char *)malloc(64);
	trackInfo.weather_graphics = (char *)malloc(64);
	trackInfo.session_name = (char *)malloc(64);

    char message[124] = {0}; // just a place to put info messages from the server itself

	while (1) {
		offset = 0;
		u_int8_t str_len_8 = 0;
		memset(buffer, 0, sizeof(buffer));

		size_t recv_bytes = (size_t)recvfrom(recv_sock_fd, buffer, sizeof(buffer), 0, NULL, NULL);
		if (recv_bytes < 0) {
			perror("[!] recvfrom() failed");
			close(send_sock_fd);
			close(recv_sock_fd);
			return EXIT_FAILURE;
		}

		// Convert buffer to utf-8 string and print it
		printf("[+] Received %zd bytes from client\n", recv_bytes);

		// The first byte of the messages is always an ACSP_MessageType
		switch ((int)buffer[0]) {
		// ============================
		// SERVER → CLIENT MESSAGES
		// ============================

		case ACSP_NEW_SESSION:
			printf("[+] Message Type: ACSP_NEW_SESSION\n");
			offset = 1; // Reset offset to 1 to read after message types

			trackInfo.protocol_version = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
			trackInfo.session_index = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
			trackInfo.current_session_index = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
			trackInfo.session_count = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);

			str_len_8 = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
			read_utf32le_string((const u_int8_t *)buffer, recv_bytes, &offset, trackInfo.server_name, str_len_8, &ok);

			str_len_8 = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
            read_string((const u_int8_t *)buffer, recv_bytes, &offset, trackInfo.track, str_len_8, &ok);       

            str_len_8 = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
			read_string((const u_int8_t *)buffer, recv_bytes, &offset, trackInfo.track_config, str_len_8, &ok);

            str_len_8 = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
            read_string((const u_int8_t *)buffer, recv_bytes, &offset, trackInfo.session_name, str_len_8, &ok);

            trackInfo.typ = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
            trackInfo.time = read_uint16((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
            trackInfo.laps = read_uint16((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
            trackInfo.wait_time = read_uint16((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
            trackInfo.ambient_temp = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
            trackInfo.road_temp = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);

            str_len_8 = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
            read_string((const u_int8_t *)buffer, recv_bytes, &offset, trackInfo.weather_graphics, str_len_8, &ok);

            trackInfo.elapsed_ms = read_int32((const u_int8_t *)buffer, recv_bytes, &offset, &ok);


			if (!ok) {
				fprintf(stderr, "[-] Error parsing NEW_SESSION packet (offset=%zu)\n", offset);
				break;
			}

			printf("\tServer Name: \"%s\"\n", trackInfo.server_name);
			printf("\tTrack: \"%s\"\n", trackInfo.track);
			printf("\tTrack Config: \"%s\"\n", trackInfo.track_config);
            printf("\tSession Name: \"%s\"\n", trackInfo.session_name);

			printf("\tProtocol Version: %d\tSession Index: %d/%d\tCurrent Session Index: %d\n", trackInfo.protocol_version, trackInfo.session_index, trackInfo.session_count,
			       trackInfo.current_session_index);

            printf("\tType: %d\tTime: %d mins\tLaps: %d\tWait Time: %d secs\n", trackInfo.typ, trackInfo.time, trackInfo.laps, trackInfo.wait_time);
            printf("\tAmbient Temp: %d C\tRoad Temp: %d C\n", trackInfo.ambient_temp, trackInfo.road_temp);
            printf("\tWeather Graphics: %s\n", trackInfo.weather_graphics);
            printf("\tElapsed Time: %d ms\n", trackInfo.elapsed_ms);

			break;

		case ACSP_NEW_CONNECTION:
			printf("[+] Message Type: ACSP_NEW_CONNECTION\n");
			break;

		case ACSP_CONNECTION_CLOSED:
			printf("[+] Message Type: ACSP_CONNECTION_CLOSED\n");
			break;

		case ACSP_CAR_UPDATE:
			printf("[+] Message Type: ACSP_CAR_UPDATE\n");

			memcpy(&update.carID, buffer + 1, sizeof(uint8_t));
			offset = 1 + sizeof(uint8_t);
			memcpy(&update.position, buffer + offset, sizeof(postion));
			offset += sizeof(postion);
			memcpy(&update.velocity, buffer + offset, sizeof(postion));
			offset += sizeof(postion);
			memcpy(&update.carGear, buffer + offset, sizeof(u_int8_t));
			offset += sizeof(u_int8_t);
			memcpy(&update.carRPM, buffer + offset, sizeof(uint16_t));
			offset += sizeof(uint16_t);
			memcpy(&update.normalizedSplinePos, buffer + offset, sizeof(float_t));
			offset += sizeof(float_t);

			printf("\tCar %d Position: X: %.2f Y: %.2f Z: %.2f ", update.carID, update.position.x, update.position.y, update.position.z);
			printf("Gear: %d RPM: %d\n", update.carGear, update.carRPM);
			break;

		case ACSP_CAR_INFO:
			printf("[+] Message Type: ACSP_CAR_INFO\n");
			break;

		case ACSP_END_SESSION:
			printf("[+] Message Type: ACSP_END_SESSION\n");
			break;

		case ACSP_VERSION:
			printf("[+] Message Type: ACSP_VERSION\n");
			break;

		case ACSP_CHAT:
			printf("[+] Message Type: ACSP_CHAT\n");
			break;

		case ACSP_CLIENT_LOADED:
			printf("[+] Message Type: ACSP_CLIENT_LOADED\n");
			break;

		case ACSP_SESSION_INFO:
			printf("[+] Message Type: ACSP_SESSION_INFO\n");
			break;

		case ACSP_ERROR:
			printf("[+] Message Type: ACSP_ERROR\n");
            offset = 1; // Reset offset to 1 to read after message types
            
			str_len_8 = read_uint8((const u_int8_t *)buffer, recv_bytes, &offset, &ok);
			read_utf32le_string((const u_int8_t *)buffer, recv_bytes, &offset, message, str_len_8, &ok);

        	printf("\tServer Message: \"%s\"\n", message);
			break;

		case ACSP_LAP_COMPLETED:
			printf("[+] Message Type: ACSP_LAP_COMPLETED\n");
			break;

		// ============================
		// EVENTS
		// ============================
		case ACSP_CLIENT_EVENT:
			printf("[+] Message Type: ACSP_CLIENT_EVENT\n");
			break;

		// ============================
		// EVENT TYPES
		// ============================
		case ACSP_CE_COLLISION_WITH_CAR:
			printf("[+] Event Type: ACSP_CE_COLLISION_WITH_CAR\n");
			break;

		case ACSP_CE_COLLISION_WITH_ENV:
			printf("[+] Event Type: ACSP_CE_COLLISION_WITH_ENV\n");
			break;

		// ============================
		// CLIENT → SERVER COMMANDS
		// ============================
		case ACSP_REALTIMEPOS_INTERVAL:
			printf("[+] Command: ACSP_REALTIMEPOS_INTERVAL\n");
			break;

		case ACSP_GET_CAR_INFO:
			printf("[+] Command: ACSP_GET_CAR_INFO\n");
			break;

		case ACSP_SEND_CHAT:
			printf("[+] Command: ACSP_SEND_CHAT\n");
			break;

		case ACSP_BROADCAST_CHAT:
			printf("[+] Command: ACSP_BROADCAST_CHAT\n");
			break;

		case ACSP_GET_SESSION_INFO:
			printf("[+] Command: ACSP_GET_SESSION_INFO\n");
			break;

		case ACSP_SET_SESSION_INFO:
			printf("[+] Command: ACSP_SET_SESSION_INFO\n");
			break;

		case ACSP_KICK_USER:
			printf("[+] Command: ACSP_KICK_USER\n");
			break;

		case ACSP_NEXT_SESSION:
			printf("[+] Command: ACSP_NEXT_SESSION\n");
			break;

		case ACSP_RESTART_SESSION:
			printf("[+] Command: ACSP_RESTART_SESSION\n");
			break;

		case ACSP_ADMIN_COMMAND:
			printf("[+] Command: ACSP_ADMIN_COMMAND\n");
			break;

		// ============================
		// DEFAULT HANDLER
		// ============================
		default:
			printf("[!] Unknown Message Type: %d\n", buffer[0]);
			break;
		}

		usleep(10000); // Sleep for 10ms
	}
	close(send_sock_fd);
	close(recv_sock_fd);

	free(trackInfo.server_name);
	free(trackInfo.track);
	free(trackInfo.track_config);
	free(trackInfo.weather_graphics);
    free(trackInfo.session_name);
	return 0;
}