/*
 * dodfun 
 * Copyright (c) 2004 Lukasz Wlasinski
 *
 *
 *    This program is free software; you can redistribute it and/or modify it
 *    under the terms of the GNU General Public License as published by the
 *    Free Software Foundation; either version 2 of the License, or (at
 *    your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful, but
 *    WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *    General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with this program; if not, write to the Free Software Foundation,
 *    Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *    In addition, as a special exception, the author gives permission to
 *    link the code of this program with the Half-Life Game Engine ("HL
 *    Engine") and Modified Game Libraries ("MODs") developed by Valve,
 *    L.L.C ("Valve").  You must obey the GNU General Public License in all
 *    respects for all of the code used other than the HL Engine and MODs
 *    from Valve.  If you modify this file, you may extend this exception
 *    to your version of the file, but you are not obligated to do so.  If
 *    you do not wish to do so, delete this exception statement from your
 *    version.
 *
 */

#include <string.h>
#include "amxxmodule.h"
#include "dodfun.h"

funEventCall modMsgsEnd[MAX_REG_MSGS];
funEventCall modMsgs[MAX_REG_MSGS];
void (*function)(void*);
void (*endfunction)(void*);
CPlayer* mPlayer;
CPlayer* gPlayerRocket;
CPlayer players[33];

CObjective mObjects;

int mState;
int mDest;
int mPlayerIndex;

int iFGrenade;
int iFRocket;
int iFInitCP;

int gmsgCurWeapon;
int gmsgScoreShort;
int gmsgPTeam;
int gmsgInitObj;
int gmsgSetObj;
int gmsgFrags;
int gmsgObjScore;

struct sUserMsg {
	const char* name;
	int* id;
	funEventCall func;
	bool endmsg;
} g_user_msg[] = {
	{ "InitObj",&gmsgInitObj,Client_InitObj,false},
	{ "CurWeapon",&gmsgCurWeapon,Client_CurWeapon,false },
	{ "ScoreShort",&gmsgScoreShort,NULL,false },
	{ "PTeam",&gmsgPTeam,NULL,false },
	{ "SetObj",&gmsgSetObj,Client_SetObj,false },
	{ "Frags",&gmsgFrags,NULL,false },
	{ "ObjScore", &gmsgObjScore, NULL, false },

	{ 0,0,0,false }
};

int RegUserMsg_Post(const char *pszName, int iSize){
	for (int i = 0; g_user_msg[ i ].name; ++i ){
		if ( !*g_user_msg[i].id && strcmp( g_user_msg[ i ].name , pszName  ) == 0 ){
			int id = META_RESULT_ORIG_RET( int );

			*g_user_msg[ i ].id = id;
		
			if ( g_user_msg[ i ].endmsg )
				modMsgsEnd[ id  ] = g_user_msg[ i ].func;
			else
				modMsgs[ id  ] = g_user_msg[ i ].func;

			break;
		}
	}

	RETURN_META_VALUE(MRES_IGNORED, 0);
}

void ServerActivate_Post( edict_t *pEdictList, int edictCount, int clientMax ){

	for( int i = 1; i <= gpGlobals->maxClients; ++i )
		GET_PLAYER_POINTER_I(i)->Init( i , pEdictList + i );

	RETURN_META(MRES_IGNORED);
}

void ServerDeactivate() {
	for(int i = 1;i<=gpGlobals->maxClients; ++i){
		CPlayer *pPlayer = GET_PLAYER_POINTER_I(i);
		if (pPlayer->ingame)
			pPlayer->Disconnect();
	}
	RETURN_META(MRES_IGNORED);
}

BOOL ClientConnect_Post( edict_t *pEntity, const char *pszName, const char *pszAddress, char szRejectReason[ 128 ]  ){
	GET_PLAYER_POINTER(pEntity)->Connect();
	RETURN_META_VALUE(MRES_IGNORED, TRUE);
}

void ClientDisconnect( edict_t *pEntity ) {
	CPlayer *pPlayer = GET_PLAYER_POINTER(pEntity);
	if (pPlayer->ingame) 
		pPlayer->Disconnect();
	RETURN_META(MRES_IGNORED);
}

void ClientPutInServer_Post( edict_t *pEntity ) {
	GET_PLAYER_POINTER(pEntity)->PutInServer();
	RETURN_META(MRES_IGNORED);
}

void ClientUserInfoChanged_Post( edict_t *pEntity, char *infobuffer ) {
	CPlayer *pPlayer = GET_PLAYER_POINTER(pEntity);

	if ( !pPlayer->ingame && pPlayer->IsBot() ) {
		pPlayer->PutInServer();
	}

	RETURN_META(MRES_IGNORED);
}

void MessageBegin_Post(int msg_dest, int msg_type, const float *pOrigin, edict_t *ed) {
	if (ed){
		mPlayerIndex = ENTINDEX(ed);
		mPlayer = GET_PLAYER_POINTER_I(mPlayerIndex);
	} else {
		mPlayerIndex = 0;
		mPlayer = NULL;
	}
	mState = 0;
	mDest = msg_dest;
	if ( msg_type < 0 || msg_type >= MAX_REG_MSGS )
		msg_type = 0;
	function=modMsgs[msg_type];
	endfunction=modMsgsEnd[msg_type];
	RETURN_META(MRES_IGNORED);
}

void MessageEnd_Post(void) {
	if (endfunction) (*endfunction)(NULL);
	RETURN_META(MRES_IGNORED);
}

void WriteByte_Post(int iValue) {
	if (function) (*function)((void *)&iValue);
	RETURN_META(MRES_IGNORED);
}

void WriteChar_Post(int iValue) {
	if (function) (*function)((void *)&iValue);
	RETURN_META(MRES_IGNORED);
}

void WriteShort_Post(int iValue) {
	if (function) (*function)((void *)&iValue);
	RETURN_META(MRES_IGNORED);
}

void WriteLong_Post(int iValue) {
	if (function) (*function)((void *)&iValue);
	RETURN_META(MRES_IGNORED);
}

void WriteAngle_Post(float flValue) {
	if (function) (*function)((void *)&flValue);
	RETURN_META(MRES_IGNORED);
}

void WriteCoord_Post(float flValue) {
	if (function) (*function)((void *)&flValue);
	RETURN_META(MRES_IGNORED);
}

void WriteString_Post(const char *sz) {
	if (function) (*function)((void *)sz);
	RETURN_META(MRES_IGNORED);
}

void WriteEntity_Post(int iValue) {
	if (function) (*function)((void *)&iValue);
	RETURN_META(MRES_IGNORED);
}

void PlayerPreThink_Post(edict_t *pEntity) 
{
	CPlayer *pPlayer = GET_PLAYER_POINTER(pEntity);

	// Stamina
	if(pPlayer->staminaSet) 
	{
		if ( (int)pEntity->v.fuser4 > pPlayer->staminaMax)
			pEntity->v.fuser4 = (float)pPlayer->staminaMax;

		else if ( (int)pEntity->v.fuser4 < pPlayer->staminaMin)
			pEntity->v.fuser4 = (float)pPlayer->staminaMin;
	}

	if(pPlayer->current == 29 || pPlayer->current == 30 || pPlayer->current == 31)
	{
		if(!(pPlayer->pEdict->v.oldbuttons&IN_ATTACK) && (pPlayer->pEdict->v.button&IN_ATTACK))
			gPlayerRocket = GET_PLAYER_POINTER(pEntity);
	}

	RETURN_META(MRES_IGNORED);
}

void SetModel_Post(edict_t *e, const char *m)
{
	int w_id = 0;

	if(!e->v.owner || !e->v.dmgtime)
	{
		int owner = ENTINDEX(e->v.owner);

		if(owner && owner < 33 && m[7]=='w' && m[8]=='_')
		{
			CPlayer* pPlayer = GET_PLAYER_POINTER_I(owner);
			bool newNade = (pPlayer->current == 13 || pPlayer->current == 14) ? true : false;
			
			if(m[9]=='g' && m[10]=='r' && m[11]=='e' && m[12]=='n')
				w_id = newNade ? 13 : 16; // grenade

			else if(m[9]=='m' && m[10]=='i')
				w_id = newNade ? 36 : 16 ; // mills ; should I add mills_grenade_ex weapon ?

			else if(m[9]=='s' && m[10]=='t' && m[11]=='i')
				w_id = newNade ? 14 : 15; // stick
			
			if(!w_id)
				RETURN_META(MRES_IGNORED);

			if(w_id == 13 || w_id == 14 || w_id == 15 || w_id == 16 || w_id == 36)
			{
				MF_ExecuteForward(iFGrenade, pPlayer->index, ENTINDEX(e), w_id);

				/* fuse start */
				if(pPlayer->fuseSet)
				{
					if(newNade)
					{
						if(pPlayer->fuseType & 1<<0)
						{
							e->v.dmgtime += pPlayer->nadeFuse - 5.0;
						}
					}

					else
					{ 
						float fExp = e->v.dmgtime - gpGlobals->time;
						e->v.dmgtime += pPlayer->nadeFuse - fExp;
					}
				}
				/* fuse end */
			}
		}

		else if(strstr(m, "rocket") && gPlayerRocket)
		{
			if(strstr(m, "bazooka"))
				w_id = 29;

			else if(strstr(m, "piat"))
				w_id = 30;

			else if(strstr(m, "pschreck"))
				w_id = 31;
			
			MF_ExecuteForward(iFRocket, gPlayerRocket->index, ENTINDEX(e), w_id);

			gPlayerRocket = NULL;
		}
	}

	RETURN_META(MRES_IGNORED);
}

int AmxxCheckGame(const char *game)
{
	if (strcasecmp(game, "dod") == 0)
		return AMXX_GAME_OK;

	return AMXX_GAME_BAD;
}
void OnAmxxAttach() 
{
	MF_AddNatives( base_Natives );
	MF_AddNatives( pd_Natives );
}

void OnPluginsLoaded()
{
	iFGrenade = MF_RegisterForward("grenade_throw",ET_IGNORE,FP_CELL/*id*/,FP_CELL/*Grenade Ent*/,FP_CELL/*Weapon ID*/,FP_DONE);
	iFRocket = MF_RegisterForward("rocket_shoot",ET_IGNORE,FP_CELL/*id*/,FP_CELL/*Rocket Ent*/,FP_CELL/*Weapon ID*/,FP_DONE);
	iFInitCP = MF_RegisterForward("controlpoints_init",ET_IGNORE,FP_DONE);
}