/*
   core/gameserver.h
   This file is part of the Osirion project and is distributed under
   the terms of the GNU General Public License version 2
*/

#include <iomanip>

#include "auxiliary/functions.h"
#include "sys/sys.h"
#include "core/application.h"
#include "core/cvar.h"
#include "core/func.h"
#include "core/gameserver.h"
#include "core/netserver.h"

namespace core
{

Player *console_find_player(std::string const & target)
{
	Player *targetplayer = server()->find_player(target);
	if (!targetplayer) {
		con_print << "^BPlayer " + target + "^B not found.";
	}
	return targetplayer;
}

void func_who(std::string const &args)
{
	server()->list_players();
}

void func_time(std::string const &args)
{
	server()->showtime();
}

void func_mute(std::string const &args)
{
	Player *targetplayer = 0;
	if (!(targetplayer = console_find_player(args)))
		return;

	if (targetplayer->mute()) {
		con_print << "^BPlayer " + targetplayer->name() + "^B has already been muted.";
		return;
	}

	targetplayer->player_mute = true;
	server()->broadcast("^B" + targetplayer->name() + "^B has been muted.", targetplayer);
	server()->send(targetplayer, "^BYou have been muted.");
}

void func_unmute(std::string const &args)
{
	Player *targetplayer = 0;
	if (!(targetplayer = console_find_player(args)))
		return;

	if (!targetplayer->mute()) {
		con_print << "^BPlayer " + targetplayer->name() + "^B has not been muted.";
		return;
	}

	targetplayer->player_mute = false;
	server()->broadcast("^B" +targetplayer->name() + "^B has been unmuted.", targetplayer);
	server()->send(targetplayer, "^BYou have been unmuted.");
}

void func_kick(std::string const &args)
{
	std::stringstream str(args);
	std::string name;

	if (str >> name) {
		Player *targetplayer = 0;
		if (!(targetplayer = console_find_player(name)))
			return;
		std::string reason;
		if (args.size() > name.size()+1)
			reason.assign(args.substr(name.size()+1));
		else
			reason.assign("forcefully removed.");

		server()->kick(targetplayer, reason);
	}
}

void func_grant_rcon(std::string const &args)
{
	Player *targetplayer = 0;
	if (!(targetplayer = console_find_player(args)))
		return;}

void func_revoke_rcon(std::string const &args)
{
	Player *targetplayer = 0;
	if (!(targetplayer = console_find_player(args)))
		return;
}

GameServer *GameServer::server_instance = 0;

GameServer::GameServer() : GameInterface()
{
	con_print << "^BInitializing game server...\n";
	server_instance = this;
	server_network = 0;
	server_time = 0;
	server_previoustime = 0;
	server_frametime = 0.0f;
	server_maxplayerid = 1;

	server_module = Module::preload();
	if (!server_module) {
		con_error << "No module loaded.\n";
		abort();
		return;
	}

	// set the name of the game
	core::Cvar::set("g_name", server_module->name().c_str(), core::Cvar::Game | core::Cvar::ReadOnly);

	server_module->init();
	if (!server_module->running()) {
		con_error << "Could not initialize module '" << server_module->name() << "'\n";
		abort();
		return;
	}

	con_print << "  module '^B" << server_module->name() << "^N'\n";

	if ((Cvar::sv_dedicated->value() || Cvar::sv_private->value())) {
		server_network = new NetServer(Cvar::net_host->str(), (unsigned int) Cvar::net_port->value());
		if (!server_network->valid()) {
			delete server_network;
			server_network = 0;
			con_error << "Could not initialize network server\n";
			abort();
			return;
		}
	} else {
		con_print << "  network server disabled.\n";
		server_network = 0;
	}

	Func *func = 0;

	/* -- admin functions -- */
	func = Func::add("mute", func_mute);
	func->set_info("[player] mute a player");

	func = Func::add("unmute", func_unmute);
	func->set_info("[player] unmute a player");

	func = Func::add("kick", func_kick);
	func->set_info("[player] [reason] kick a player from the server");
/*
	func = Func::add("grant_rcon", func_grant_rcon);
	func->set_info("[player] grant rcon rights");

	func = Func::add("revoke_rcon", func_grant_rcon);
	func->set_info("[player] revoke rcon rights");
*/
	/* -- player functions --*/
	func = Func::add("time", func_time, Func::Shared);
	func->set_info("get the server uptime and current server localtime");

	func = Func::add("who", func_who, Func::Shared);
	func->set_info("get a list of connected players");

	if (!Cvar::sv_dedicated->value()) {
		player_connect(localplayer());
	}

	server_running = true;
}

GameServer::~GameServer()
{
	server_running = false;

	con_print << "^BShutting down game server...\n";

	if (server_network) {
		delete server_network;
		server_network = 0;
	}

	if (server_module) {
		if (server_module->running() && !Cvar::sv_dedicated->value())
			player_disconnect(localplayer());

		server_module->shutdown();

		if (server_module != Module::preload())
			delete server_module;
	}

	Func::remove("kick");

	Func::remove("mute");
	Func::remove("unmute");

/*	Func::remove("grant_rcon");
	Func::remove("revoke_rcon");
*/
	Func::remove("time");
	Func::remove("who");

	server_instance = 0;

	players.clear();
}

void GameServer::abort()
{
	server_running = false;
}

void GameServer::list_players()
{
	using namespace std;
	stringstream msgstr;
	int count = 0;

	for (std::list<Player *>:: iterator it = players.begin(); it != players.end(); it++) {
		msgstr.str("");
		con_print << setw(3) << (*it)->id() << aux::pad_left((*it)->name(), 24) << std::endl;
		count++;
	}

	con_print << count << " connected " << aux::plural("player", count) << std::endl;
}

void GameServer::showtime()
{
	using namespace std;

	int minutes = (int) floorf(server_time / 60.0f);
	int seconds = (int) floorf(server_time - (float) minutes* 60.0f);

	int syshours = sys::time() / 3600;
	int sysminutes = (sys::time() - syshours * 3600) / 60;
	int sysseconds = sys::time() % 60;

	con_print << 
		"Uptime " << minutes << ":" << setfill('0') << setw(2) << seconds << 
		" Server localtime " << setfill(' ') << setw(2) << syshours << ":" << setfill('0') << setw(2) << sysminutes << ":" << setw(2) << sysseconds << setfill(' ') << std::endl;
}

Player *GameServer::find_player(std::string const search)
{
	using aux::lowercase;

	std::istringstream searchstr(search);
	int id = 0;
	if (searchstr >> id) {
		for (std::list<Player *>:: iterator it = players.begin(); it != players.end(); it++) {
			if ((*it)->id() == id) {
				return (*it);
			}
		}
	}

	if (search.size() <3)
		return 0;

	for (std::list<Player *>:: iterator it = players.begin(); it != players.end(); it++) {
		if (aux::text_strip_lowercase((*it)->name()).find(lowercase(search)) != std::string::npos)
			return (*it);
	}

	return 0;
}

void GameServer::say(Player *player, std::string const &message)
{
	if (!message.size())
		return;

	if (player->mute()) {
		send(player, "^BYou have been muted.");
		return;
	}

	std::string notification("^B");
	notification.append(player->name());
	notification.append("^F:^B ");
	notification.append(message);

	// send to application
	application()->notify_message(notification);
	application()->notify_sound("com/chat");
	
	// broadcast to remote clients
	if (server_network) {
		server_network->broadcast_message("public", notification);
	}
}

// FIXME kicked by
void GameServer::kick(Player *player, std::string const &reason)
{
	if (!server_network) {
		con_print << "Not running a networked server." << std::endl;
		return;
	}

	NetClient *client = server_network->find_client(player);
	if (client) {
		broadcast("^B" + player->name() + "^B has been kicked: " + reason, player);
		server_network->send_message(client, "info", "^WYou have been kicked: " + reason);
		server_network->send_disconnect(client);
	} else {
		con_print << "Network client not found." << std::endl;
	}
}

// broadcast an "info" message to all players
void GameServer::broadcast(std::string const & message, Player *ignore_player) 
{
	if (!message.size())
		return;

	// send to application
	if (ignore_player != game()->localplayer())
		application()->notify_message(message);
	
	// broadcast to remote clients
	if (server_network) {
		server_network->broadcast_message("info", message, ignore_player);
	}
}

// send and "info" message to a single player
void GameServer::send(Player *player, std::string message)
{
	if (!message.size())
		return;

	// send to application
	if (player == localplayer()) {
		application()->notify_message(message);
		return;
	}

	// send to remote clients
	if (server_network) {
		NetClient *client = server_network->find_client(player);
		if (client) {
			server_network->send_message(client, "info", message);
		}
	}
}

// broadcast a sound event to all players
void GameServer::broadcast_sound(std::string const & sound, Player *ignore_player)
{
	if (!sound.size())
		return;

	// send to application
	if (ignore_player != game()->localplayer()) {
		application()->notify_sound(sound.c_str());
	}

	// broadcast to remote clients
	if (server_network) {
		server_network->broadcast_message("snd", sound, ignore_player);
	}
}

// send a sound event to a single player
void GameServer::send_sound(Player *player, std::string sound)
{
	if (!sound.size())
		return;

	// send to application
	if (player == localplayer()) {
		application()->notify_sound(sound.c_str());
		return;
	}

	// send to remote client
	if (server_network) {
		NetClient *client = server_network->find_client(player);
		if (client) {
			server_network->send_message(client, "snd", sound);
		}
	}
}

// send an rcon message to a single player
void GameServer::send_rcon(Player *player, std::string message)
{
	// send to application
	if (player == localplayer()) {
		con_print << message << std::endl;
		return;
	}

	// send to remote clients
	if (server_network) {
		NetClient *client = server_network->find_client(player);
		if (client) {
			server_network->send_message(client, "rcon", message);
		}
	}
}

// execute a command for a remote player
void GameServer::exec(Player *player, std::string const & cmdline)
{
	std::string command;
	std::stringstream cmdstream;
	cmdstream.str(cmdline);
	cmdstream >> command;

	Func *function = Func::find(command);
	if (function ) {

		std::string args;
		if (cmdline.size() > command.size() +1 )
			args.assign(cmdline.substr(command.size()+1));

		if ((function ->flags() & Func::Game) == Func::Game) {
			function->exec(player, args);
			return;
		} else if ((function->flags() & Func::Shared) == Func::Shared) {
			
			// enable rcon buffering
			console()->buffer_rcon(true);
			function->exec(args);

			char line[MAXCMDSIZE];
		
			while(console()->buffer().getline(line, MAXCMDSIZE-1)) {
				send(player, std::string(line));
			}

			// disable rcon buffering
			console()->buffer_rcon(false);
			return;
		}
	}

	std::string message("Unknown command '");
	message.append(command);
	message.append("^N'");
	send(player, message);
}

void GameServer::player_connect(Player *player)
{
	if (Cvar::sv_dedicated->value() && (player == localplayer())) {
		return;
	}

	player->player_id = server_maxplayerid++;

	std::string message("^B");
	message.append(player->name());
	message.append("^B connects.");
	broadcast(message, player);

	// notify the game module
	server_module->player_connect(player);

	// manage player list
	players.push_back(player);
}

void GameServer::player_disconnect(Player *player)
{
	std::string message("^B");
	message.append(player->name());
	message.append("^B disconnects.");
	broadcast(message, player);

	// clear all player assets
	player->clear_assets();

	// notify the game module
	server_module->player_disconnect(player);

	// manage player list
	std::list<Player *>:: iterator it = players.begin(); 
	while (((*it)->id() != player->id()) && (it != players.end())) {
		it++;
	}
	if (it != players.end()) {
			players.erase(it);
	}
}

void GameServer::frame(float seconds)
{
	if (error())
		return;

	server_time += seconds;
	server_frametime += seconds;

	// process incoming network messages
	if (server_network) {
		server_network->receive();

		if (server_network->error()) {
			abort();
			return;
		}
	}

	if (localplayer()->dirty())
	 	localplayer()->update_info();

	if (!Cvar::sv_dedicated->value()) {
		update_clientstate(seconds);
	}

	if ((Cvar::sv_dedicated->value() || Cvar::sv_private->value())) {
		if (core::Cvar::sv_framerate->value()) {		
			float f =  1.0f / core::Cvar::sv_framerate->value();
			if (server_frametime < f) {
				return;
			}
		}
	} 
	
	// copy the previous entity state to the client state
	if (!Cvar::sv_dedicated->value()) {
		reset_clientstate(server_time, server_previoustime);
	}

	// run a time frame on each entity
	for (Entity::Registry::iterator it=Entity::registry().begin(); it != Entity::registry().end(); it++) {
		Entity *entity = (*it).second;

		if ((entity->type() == Entity::Controlable) || (entity->type() == Entity::Dynamic)) {
			entity->frame(server_frametime);
		}
	}

	// run a frame on the module
	if (server_module) {
		server_module->frame(server_frametime);
		if (server_module->error()) {
			abort();
			return;
		}
	}

	// send updates
	if (server_network) {
		// FIXME prevent connecting clients from receiving update frames

		// transmit buffered sends
		server_network->transmit();

		// start server frame
		server_network->broadcast_frame(server_time, server_previoustime);

		// send changes in the world
		for (Entity::Registry::iterator it=Entity::registry().begin(); it != Entity::registry().end(); ) {

			Entity *entity = (*it).second;

			if (entity->entity_destroyed) {

				if (!entity->entity_created) {
					server_network->broadcast_entity_delete(entity);
				}

				delete entity;
				(*it).second = entity = 0;
				Entity::registry().erase(it++);

			} else {
				if (entity->entity_created) {
	
					server_network->broadcast_entity_create(entity);
					entity->entity_created = false;
					entity->entity_dirty = false;
	
				} else if (entity->dirty()) {	
	
					server_network->broadcast_entity_update(entity);
					entity->entity_dirty = false;
				}
				++it;
			}
		}

		// update player info
		server_network->broadcast_player_update();

		// transmit buffered sends
		server_network->transmit();

	} else {

		// local update stub
		for (Entity::Registry::iterator it=Entity::registry().begin(); it != Entity::registry().end(); ) {
			Entity *entity = (*it).second;

			if (entity->entity_destroyed) {
				delete entity;
				(*it).second = entity = 0;
				Entity::registry().erase(it++);
			} else {
				entity->entity_created = false;
				entity->entity_dirty = false;
				++it;
			}
		}
	}

	if (!Cvar::sv_dedicated->value()) {
		update_clientstate(0);
	}

	server_frametime = 0;
	server_previoustime = server_time;
}


}