class TeamChangeChatCommand : public ChatCommandClass 
{
	void Triggered(int ID,const TokenClass &Text,int ChatType) 
	{
		GameObject *obj = Get_GameObj(ID);
		int gdiPlayers = (Tally_Team_Size(1));
		int nodPlayers = (Tally_Team_Size(0));
		int team = Get_Object_Type(obj);
		if (team == 1) 
		{
			printf("GDI\n");
			nodPlayers++;
			if (nodPlayers < gdiPlayers) 
			{
				printf("Change Team GDI to Nod\n");
				Change_Team_By_ID(ID,0);
				char message[256];
				sprintf(message,"msg Player %s has changed to Team Nod to make the teams even.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			else if (nodPlayers == 0)
			{
				char message[256];
				sprintf(message,"msg You are the only player on GDI", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			else if (nodPlayers == gdiPlayers)
			{
				char message[256];
				sprintf(message,"msg Teams are Even", Get_Player_Name_By_ID(ID));
				Console_Input(message);
				//Console_Input(StrFormat("msg Teams are Even.",Get_Player_ID(obj)).c_str();
			}
			else 
			{
				//Console_Input(StrFormat("msg Sorry GDI doesn't have enough Players!.",Get_Player_ID(obj)).c_str();
				char message[256];
				sprintf(message,"msg Sorry GDI doesn't have enough Players!", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
	
		}
		if (team == 0) 
		{
			printf("NOD\n");
			gdiPlayers++;
			if (gdiPlayers < nodPlayers) 
			{
				printf("Change Team Nod to GDI\n");
				Change_Team_By_ID (ID,1);
				char message[256];
				sprintf(message,"msg Player %s has changed to Team GDI to make the teams even.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			else if (gdiPlayers == 0)
			{
				char message[256];
				sprintf(message,"msg You are the only player on Nod.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
				//Console_Input(StrFormat("msg You are the only player on Nod.",Get_Player_ID(obj)).c_str();
			}
			else if (gdiPlayers == nodPlayers)
			{
				char message[256];
				sprintf(message,"msg Teams are Even", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			else 
			{
				//Console_Input(StrFormat("msg Sorry NOD doesn't have enough Players!.",Get_Player_ID(obj)).c_str();
				char message[256];
				sprintf(message,"msg Sorry NOD doesn't have enough Players!", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
		}
	}
};
ChatCommandRegistrant<TeamChangeChatCommand> TeamChangeChatCommandReg("!TeamChange;!tc;!switch;",CHATTYPE_TEAM,0,GAMEMODE_ALL);

		if (team == 1) 
		{
			printf("GDI\n");
			nodPlayers++;
		....

		if (team == 0) 
		{
			printf("NOD\n");
			gdiPlayers++;
		....

else if (gdiPlayers == 0) //or nodPlayers  in GDI ;)
			{
				char message[256];
				sprintf(message,"msg You are the only player on Nod.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
				//Console_Input(StrFormat("msg You are the only player on Nod.",Get_Player_ID(obj)).c_str();
			}

nodPlayers++;
gdiPlayers++;

else if (gdiPlayers == 0) //or nodPlayers  in GDI ;)
			{
				char message[256];
				sprintf(message,"msg You are the only player on Nod.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
				//Console_Input(StrFormat("msg You are the only player on Nod.",Get_Player_ID(obj)).c_str();
			}

else if (gdiPlayers == 0 && nodPlayers == 1) //or nodPlayers && gdiPlayers in GDI ;)
			{
				char message[256];
				sprintf(message,"msg You are the only player on Nod.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
				//Console_Input(StrFormat("msg You are the only player on Nod.",Get_Player_ID(obj)).c_str();
			}

nodPlayers++;
gdiPlayers++;

class TeamChangeChatCommand : public ChatCommandClass 
{
	void Triggered(int ID,const TokenClass &Text,int ChatType) 
	{
		if(!PluginSettings->TeamChange){return;}
		GameObject *obj = Get_GameObj(ID);
		int difference = 0;
		int gdiPlayers = (Tally_Team_Size(1));
		int nodPlayers = (Tally_Team_Size(0));
		int team = Get_Object_Type(obj);


		if (team == 1) 
		{
			if (gdiPlayers - nodPlayers )
			{
				difference = gdiPlayers - nodPlayers;
			}
			else if (gdiPlayers == 1)
			{
				Console_Input(StrFormat("ppage %d You are the only player on GDI.",Get_Player_ID(obj)).c_str()); 
			}
			else if (difference == 1)
			{
				char message[256];
				sprintf(message,"msg GDI has only one more player then NOD No need to Switch.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			else if (nodPlayers == gdiPlayers)
			{
				char message[256];
				sprintf(message,"msg Teams are Even", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			
			else if (nodPlayers < gdiPlayers + 1) 
			{
				printf("Change Team GDI to Nod\n");
				Change_Team_By_ID(ID,0);
				char message[256];
				sprintf(message,"msg Player %s has changed to Team Nod to make the teams even.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			else 
			{
				char message[256];
				sprintf(message,"msg Sorry GDI doesn't have enough Players!", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
	
		}
		if (team == 0) 
		{
			if (nodPlayers == 1)
			{
				Console_Input(StrFormat("ppage %d You are the only player on NOD.",Get_Player_ID(obj)).c_str()); 
			}
			else if (nodPlayers - gdiPlayers  )
			{
				difference = nodPlayers - gdiPlayers;
			}
			if (difference == 1)
			{
				char message[256];
				sprintf(message,"msg Nod has only one more player then GDI No need to Switch.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			else if (gdiPlayers == nodPlayers)
			{
				char message[256];
				sprintf(message,"msg Teams are Even", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			else if (gdiPlayers < nodPlayers + 1)
			{
				printf("Change Team Nod to GDI\n");
				Change_Team_By_ID (ID,1);
				char message[256];
				sprintf(message,"msg Player %s has changed to Team GDI to make the teams even.", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
			else 
			{
				char message[256];
				sprintf(message,"msg Sorry NOD doesn't have enough Players!", Get_Player_Name_By_ID(ID));
				Console_Input(message);
			}
		}
	}
};
ChatCommandRegistrant<TeamChangeChatCommand> TeamChangeChatCommandReg("!TeamChange;!tc;!switch;",CHATTYPE_TEAM,0,GAMEMODE_ALL);

void ObjectHookCall(void *data,GameObject *obj) {
	if(!PluginSettings->CreditSystem){return;}
		if(Commands->Is_A_Star(obj)){
			//Attach_Script_Once(obj,"Money","");
			int PlayerID;
			PlayerID = Get_Player_ID(obj);
			printf("Player Money Added\n");
			Set_Money(PlayerID,0);
			Commands->Give_Money(obj,1000,false);
		}
	if(!PluginSettings->NoFallingDamage){return;}
		if(Commands->Is_A_Star(obj)){
			Attach_Script_Once(obj,"M00_No_Falling_Damage_DME","0");
			printf("No Damge Script Attached\n");
		}
	if(!PluginSettings->DropItem){return;}
		if(Commands->Is_A_Star(obj)){
				Attach_Script_Once(obj,"Player","");
				printf("Player Script Aattached\n");
		}
	if(!PluginSettings->NoReload){return;}
		Attach_Script_Once(obj,"NoReload","");
}

if(PluginSettings->CreditSystem && Commands->Is_A_Star(obj)){
        //Attach_Script_Once(obj,"Money","");
	int PlayerID;
	PlayerID = Get_Player_ID(obj);
	printf("Player Money Added\n");
	Set_Money(PlayerID,0);
	Commands->Give_Money(obj,1000,false);
}

if(PluginSettings->NoFallingDamage && Commands->Is_A_Star(obj)){
			Attach_Script_Once(obj,"M00_No_Falling_Damage_DME","0");
	printf("No Damge Script Attached\n");
}

if(PluginSettings->DropItem && Commands->Is_A_Star(obj)){
				Attach_Script_Once(obj,"Player","");
		printf("Player Script Aattached\n");
}
if(PluginSettings->NoReload)
{
	Attach_Script_Once(obj,"NoReload","");
}

void tctimer::Created(GameObject *obj){
	Commands->Start_Timer(obj,this,30.0f,1);
	Commands->Start_Timer(obj,this,200.0f,2);
	Commands->Start_Timer(obj,this,8.0f,3);
}


void tctimer::Timer_Expired(GameObject *obj,int number){
	if(PluginSettings->TeamChange)
	{
		//printf("TC Timer Expired\n");
		int difference = 0;
		int gdiPlayers = (Tally_Team_Size(1));
		int nodPlayers = (Tally_Team_Size(0));
		int teamnod = 0;
		int teamgdi = 1;
		GameObject *ato = Find_First_Player(teamnod);
		GameObject *ato2 = Find_First_Player(teamgdi);
		int ID = Get_Player_ID(ato);
		int ID2 = Get_Player_ID(ato2);
		if (gdiPlayers - nodPlayers )
		{
			difference = gdiPlayers - nodPlayers;
			//printf("Player difference %d\n",difference);
		}
		if(number == 1)
		{
			if (difference >= 2)
			{
				Console_Input(StrFormat("msg Use the Team Change Chat Comamnd !tc to switch Teams").c_str());
			}
			if (difference <= -2)
			{
				Console_Input(StrFormat("msg Use the Team Change Chat Comamnd !tc to switch Teams").c_str());
			}
			Commands->Start_Timer(obj,this,30.0f,1);
		}
		else if(number == 2)
		{
			if (difference >= 2)
			{
				Console_Input(StrFormat("msg What are you doing idiots Even The Teams.").c_str());
			}
			if (difference <= -2)
			{
				Console_Input(StrFormat("msg What are you doing idiots Even The Teams.").c_str());
			}
			Commands->Start_Timer(obj,this,200.0f,2);
		}
		//auto switch
		else if(number == 3)
		{
			if (difference <= -3)
			{
				Change_Team_By_ID(ID,1);
				Console_Input(StrFormat("msg Force Team Change Engaged %s Switched to team GDI ",Get_Player_Name_By_ID(ID)).c_str());
			}
			if (difference >= 3)
			{
				Change_Team_By_ID(ID2,0);
				Console_Input(StrFormat("msg Force Team Change Engaged %s Switched to team Nod ",Get_Player_Name_By_ID(ID2)).c_str());
			}
			Commands->Start_Timer(obj,this,15.0f,3);
		}
	}
}

extern "C" {
	DLLEXPORT void SSGM_Level_Loaded_Hook() {
		PluginSettings->Load();
		if(PluginSettings->TeamChange)
		{
			GameObject *temp = Commands->Create_Object("Invisible_Object",Vector3(0.0f,0.0f,0.0f));
			Commands->Attach_Script(temp,"tctimer","");
		}
	}
}

#include "scripts.h"
#include <stdarg.h>
#ifdef WIN32
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#endif
#include "engine.h"
#include "gmmain.h"
#include "TeamReBalancer.h"

trbSettingsStruct *trbSettings = 0;
bool maploading, cannodchange, cangdichange = false;
int gdiflag, nodflag = 0;

void trbSettingsStruct::Load() {
	SettingsLoader::Load();
LoadInt(min,"MinimumPlayerDifferential");
LoadInt(time,"TimeAllowedForVolunteer");
}




void Plugin_Load() {
	trbSettings = new trbSettingsStruct("teamrebalancer.ini");
}
// Our own get team player count function, as the existing one is actually broken...
int reb_Get_Team_Player_Count(int Team)
{
	int Total = 0;
	GenericSLNode *x = BaseGameObjList->HeadNode;
	while (x)
	{
		GameObject *o = (GameObject *)x->NodeData;
		if (o && Commands->Is_A_Star(o))
		{
			if (Get_Team(Get_Player_ID(o)) == Team)
			{
			Total++;
		}
		}
		x = x->NodeNext;
	}
	return Total;
}




void Plugin_Unload() {
	delete trbSettings;
}

void reteam(){
	if(maploading == false){ // Make sure the map isn't loading
		// Get the team sizes
		int gdisize = reb_Get_Team_Player_Count(1);
		int nodsize = reb_Get_Team_Player_Count(0);
			if(nodsize - gdisize >= trbSettings->min || gdisize - nodsize >= trbSettings->min){ 
				// One team has more than the allowed differential
				if(nodsize > gdisize){ // Need to move a Nod player to GDI
					if(cannodchange == true){ 
						// means there is already a need rebalance, so sets a flag to make sure the old timer doesn't cut off the new one
						nodflag++;
					}
					cannodchange = true;
					GameObject *thingy = Commands->Create_Object("Invisible_Object",Vector3(0.0f,0.0f,0.0f));
					Commands->Attach_Script(thingy,"g_volunteer_timer","");
					Console_Input(StrFormat("msg Nod have more players on their team than the server owner deems to be fair, if you're on team Nod and want to change teams, please type \"!balance\". Nod players have %i seconds to volunteer before the server chooses a player for them.", trbSettings->time).c_str());
				}
				else{ // Need to move a GDI player to Nod
					if(cangdichange == true){ 
						// means there is already a need rebalance, so sets a flag to make sure the old timer doesn't cut off the new one
						gdiflag++;
					}
					cangdichange = true;
					GameObject *thingy = Commands->Create_Object("Invisible_Object",Vector3(0.0f,0.0f,0.0f));
					Commands->Attach_Script(thingy,"n_volunteer_timer","");
					Console_Input(StrFormat("msg GDI have more players on their team than the server owner deems to be fair, if you're on team GDI and want to change teams, please type \"!balance\". GDI players have %i seconds to volunteer before the server chooses a player for them.", trbSettings->time).c_str());
				}
			}
	}
	else{
		// Nothing, the map is loading so don't fuck up the player stats board at the end of the map.
		// I only included this else to show you why I made the conditional in the first place really.
	}
}

void g_volunteer_timer::Created(GameObject *obj){
	Commands->Start_Timer(obj,this,(float)trbSettings->time,1);
}

void g_volunteer_timer::Timer_Expired(GameObject *obj,int number){
	if(number == 1){
		if(gdiflag == 0){ //make sure this is the only active timer, so as not to stop a later one prematurely.
			if(cangdichange == true){ // make sure no one has volunteered already using the chat command
				int iterations = 0;
				int score = 0;
				GameObject *volunteer;
			GenericSLNode *x = BaseGameObjList->HeadNode;
			while (x){
				GameObject *o = (GameObject *)x->NodeData;
				if (o && Commands->Is_A_Star(o) && Get_Team(Get_Player_ID(o)) == 1){
					if(iterations == 0){
						score = (int)Get_Score(Get_Player_ID(o));
						volunteer = o;
					}
					else{
						int newscore = (int)Get_Score(Get_Player_ID(o));
							if(newscore < score){
								score = newscore;
								volunteer = o;
							}
						}
					}
					iterations++;
					x = x->NodeNext;
				}
			Change_Player_Team(volunteer,false,false,true);
			}
			cangdichange = false;
			reteam();
		}
		gdiflag--;
	}
}

ScriptRegistrant<g_volunteer_timer> g_volunteer_timer_Registrant("g_volunteer_timer","");

void n_volunteer_timer::Created(GameObject *obj){
	Commands->Start_Timer(obj,this,(float)trbSettings->time,1);
}

void n_volunteer_timer::Timer_Expired(GameObject *obj,int number){
	if(number == 1){
		if(nodflag == 0){ //make sure this is the only active timer, so as not to stop a later one prematurely.
			if(cannodchange == true){ // make sure no one has volunteered already using the chat command
				int iterations = 0;
				int score = 0;
				GameObject *volunteer;
			GenericSLNode *x = BaseGameObjList->HeadNode;
			while (x){
				GameObject *o = (GameObject *)x->NodeData;
				if (o && Commands->Is_A_Star(o) && Get_Team(Get_Player_ID(o)) == 0){
					if(iterations == 0){
						score = (int)Get_Score(Get_Player_ID(o));
						volunteer = o;
					}
					else{
						int newscore = (int)Get_Score(Get_Player_ID(o));
							if(newscore < score){
								score = newscore;
								volunteer = o;
							}
						}
					}
					iterations++;
					x = x->NodeNext;
			}
			Change_Player_Team(volunteer,false,false,true);
			}
			cannodchange = false;
			reteam();
		}
		nodflag--;
	}
}

ScriptRegistrant<n_volunteer_timer> n_volunteer_timer_Registrant("n_volunteer_timer","");


class balanceChatCommand : public ChatCommandClass {
	void Triggered(int ID,const TokenClass &Text,int ChatType){
		int team = Get_Team(ID);
		if(team == 1 && cangdichange == true){
			Change_Player_Team(Get_GameObj(ID),false,false,true);
			Console_Input(StrFormat("msg Player %s has changed teams to rebalance the game.",Get_Player_Name(Get_GameObj(ID))).c_str());
			Console_Input(StrFormat("ppage %d You have volunteered to balance the game and changed teams.",ID).c_str());
			cangdichange = false;
			reteam();
		}
		else if(team == 0 && cannodchange == true){
			Change_Player_Team(Get_GameObj(ID),false,false,true);
			Console_Input(StrFormat("msg Player %s has changed teams to rebalance the game.",Get_Player_Name(Get_GameObj(ID))).c_str());
			Console_Input(StrFormat("ppage %d You have volunteered to balance the game and changed teams.",ID).c_str());
			cannodchange = false;
			reteam();
		}
		else{
			Console_Input(StrFormat("ppage %d You do not need to balance the teams right now.",ID).c_str());
		}
	}
};
ChatCommandRegistrant<balanceChatCommand> balanceChatCommandReg("!Balance;!balance;!BALANCE",CHATTYPE_ALL,0,GAMEMODE_ALL);

extern "C" {



DLLEXPORT void SSGM_Player_Leave_Hook(int ID) {
reteam();
}


DLLEXPORT void SSGM_Player_Join_Hook(int ID, const char *Nick) {
	int gdisize = reb_Get_Team_Player_Count(1);
	int nodsize = reb_Get_Team_Player_Count(0);
	if(nodsize - gdisize < trbSettings->min && cannodchange == true){ 
		cannodchange = false;
		Console_Input(StrFormat("msg Nod's team no longer needs balancing now that %s has joined.", Nick).c_str());
	}
	else if(gdisize - nodsize < trbSettings->min && cangdichange == true){
		cangdichange = false;
		Console_Input(StrFormat("msg GDI's team no longer needs balancing now that %s has joined.", Nick).c_str());
	}
}

DLLEXPORT void SSGM_Level_Loaded_Hook() {
	trbSettings->Load();
maploading, cangdichange, cannodchange = false;
gdiflag, nodflag = 0;
}

DLLEXPORT void SSGM_GameOver_Hook() {
maploading = true;
}


}

#include "gmsettingsclass.h"

#define PluginName "reborn's team rebalancer plugin"
#define PluginVersion "1.0"

struct trbSettingsStruct : public virtual SettingsLoader {
	trbSettingsStruct(const char *ININame) : SettingsLoader(ININame) {
		min = 0;
		time = 30;
	}

	void Load();
	int min, time;
};

void Plugin_Load();

int reb_Get_Team_Player_Count(int Team);

void reteam();

class g_volunteer_timer : public ScriptImpClass {
	void Created(GameObject *obj);
	void Timer_Expired(GameObject *obj,int number);
};

class n_volunteer_timer : public ScriptImpClass {
	void Created(GameObject *obj);
	void Timer_Expired(GameObject *obj,int number);
};

void Plugin_Unload();