#include <errno.h>
#include "smud.h"
#include "smud_threads.h"

#define LISTEN_PORT 6500
#define LISTEN_QUEUE_LENGTH 6

char fileNames[NUM_FILES][30] = {
	"core.games.txt",
	"sdktools.games.txt",
	"sdktools.games.ep2.txt",
	"sdktools.games.l4d.txt",
	"sm-cstrike.games.txt",
	"sm-tf2.games.txt",
};

void *fileLocations[NUM_FILES];
int fileLength[NUM_FILES];

int main(int argc, char **argv)
{
	ThreadPool *pool;
	struct protoent *pProtocol;
	struct sockaddr_in serverAddress;
	struct sockaddr_in clientAddress;
	int serverSocket;
	int clientSocket;
	int addressLen;
	int opts;
	int file;
	char filename[100];
	struct stat sbuf;

#if defined DEBUG
	fprintf(stdout, "Loading Gamedata files into memory\n");
#endif

	for (int i=0; i<NUM_FILES; i++)
	{
		snprintf(filename, sizeof(filename), "./md5/%s", fileNames[i]);
		file = open(filename, O_RDWR);

		if (!file)
		{
			fprintf(stderr, "Could not find file: %s", filename);
			return 1;
		}

		if (stat(filename, &sbuf) == -1)
		{
			fprintf(stderr, "Could not stat file: %s (error: %s)", filename, strerror(errno));
			return 1;
		}

		if ((fileLocations[i] = mmap(NULL, sbuf.st_size, PROT_READ, MAP_PRIVATE, file, 0)) == MAP_FAILED)
		{
			fprintf(stderr, "Could not mmap file: %s (error: %s)", filename, strerror(errno));
			return 1;
		}

		fileLength[i] = sbuf.st_size;

#if defined DEBUG
		fprintf(stdout, "Initialised file of %s of length %i\n", fileNames[i], fileLength[i]);
#endif
	}

#if defined DEBUG
	fprintf(stdout, "Initializing Thread Pool\n");
#endif

	pool = new ThreadPool();
	
	if (!pool->Start())
	{
		fprintf(stderr, "Could not initialize thread pool!\n");
		return 1;
	}

#if defined DEBUG
	printf("Create Server Socket\n");
#endif

	memset(&serverAddress, 0, sizeof(serverAddress));
	serverAddress.sin_family = AF_INET;
	serverAddress.sin_addr.s_addr = INADDR_ANY;
	serverAddress.sin_port = htons(LISTEN_PORT);

	pProtocol = getprotobyname("tcp");

	if (pProtocol == NULL)
	{
		fprintf(stderr, "Could not get tcp proto: %s", strerror(errno));
		return 1;
	}

	serverSocket = socket(AF_INET, SOCK_STREAM, pProtocol->p_proto);

	if (serverSocket < 0) 
	{
		fprintf(stderr, "Could not open socket: %s", strerror(errno));
		return 1;
	}

	opts = 1;
	setsockopt(serverSocket, SOL_SOCKET, SO_REUSEADDR, &opts, sizeof(opts));

	if (bind(serverSocket, (struct sockaddr *)&serverAddress, sizeof(serverAddress)) < 0) 
	{
		fprintf(stderr, "Could not bind socket: %s", strerror(errno));
		return 1;
	}

	if (listen(serverSocket, LISTEN_QUEUE_LENGTH) < 0) 
	{
		fprintf(stderr, "Could not listen on socket: %s", strerror(errno));
		return 1;
	}

	fprintf(stdout, "Server has started.\n");

	while (1)
	{
		addressLen = sizeof(clientAddress);

		clientSocket = accept(serverSocket,
							  (struct sockaddr *)&clientAddress,
							  (socklen_t *)&addressLen);
		if (clientSocket < 0)
		{
			fprintf(stderr, "Could not accept client: %s", strerror(errno));
			continue;
		}

		opts = fcntl(clientSocket, F_GETFL, 0);
		if (fcntl(clientSocket, F_SETFL, opts|O_NONBLOCK) < 0)
		{
			fprintf(stderr, "Could not non-block client: %s", strerror(errno));
			closesocket(clientSocket);
			continue;
		}

#if defined DEBUG
		fprintf(stdout,
				"Accepting connection from client (sock %d, ip %s)",
				clientSocket,
				inet_ntoa(clientAddress.sin_addr));
#endif

		pool->AddConnection(clientSocket);
	}

	delete pool;
}