#include <stdlib.h>
#include <curses.h>
#include "ui.h"
#include "game.h"
#include "net.h"
#include "player.h"

int main(int argc, char **argv)
{
	char* port = 0;
	char *addr;
	bool servermode = false;

	if (argc < 2)
	{
		printf("usage: '%s address port' for client mode, or '%s port' for server mode\n", argv[0], argv[0]);
		return EXIT_SUCCESS;
	}
	else if (argc == 2) // Server mode
	{
		int ssock;
		int* csocks;
		struct player_list players;
		int opponents = 3;
		const char* nickname = "deki";
		servermode = true;
		port = argv[1];

		// start listening
		ssock = server_start(port);

		// accept client connections
		csocks = server_get_players(ssock, opponents);

		players.count = opponents + 1;
		players.names[0] = malloc(strlen(nickname));
		strcpy(players.names[0], nickname);

		for(int i=0; i<opponents; i++)
		{
			// wait for greeting from client i and read nick
			char* name = server_recv(csocks[i], msg_type_hello);
			players.names[i+1] = name;
			printf("player connected: len:%d, name:%s\n", strlen(name), name);
		}

		// start game and send player list to clients
		server_start_game(csocks, opponents, &players);

		// cleanup
		for(int i=0; i<players.count; i++)
			free(players.names[i]);
		close(ssock);
		free(csocks);
	}
	else if (argc == 3) // Client mode
	{
		int sock;
		struct player_list* players;
		const char* nickname = "schnippi";
		addr = argv[1];
		port = argv[2];

		// connect to server
		sock = client_connect_server(addr, port);

		// greet server and tell nickname
		client_hello(sock, nickname);

		// retrieve list of all players from server
		players = client_recv(sock, msg_type_start_game);
		for(int i=0; i<players->count; i++)
			printf("Player %d: %s\n", i, players->names[i]);

		// cleanup
		for(int i=0; i<players->count; i++)
			free(players->names[i]);
		free(players);
		close(sock);
	}

	ui_init();
	start_game(servermode, addr, port);
	ui_fini();
	return EXIT_SUCCESS;
}