/**

 * vim: set ts=4 :

 * ===============================================================

 * SourceMod BAT Support Extension

 * Copyright (C) 2004-2007 AlliedModders LLC. All rights reserved.

 * ===============================================================

 *

 * 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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

 *

 * Version: $Id$

 */

#include <IAdminSystem.h>

#include <IPlayerHelpers.h>

#include "extension.h"

/**

 * @file extension.cpp

 * @brief Implements BAT Support extension code.

 */

BatSupport g_BatSupport;		/**< Global singleton for your extension's main interface */

IAdminSystem *admins = NULL;

IPlayerManager *players = NULL;

SMEXT_LINK(&g_BatSupport);

bool BatSupport::SDK_OnLoad(char *error, size_t maxlength, bool late)

{

	SM_GET_IFACE(ADMINSYS, admins);

	SM_GET_IFACE(PLAYERMANAGER, players);

	players->AddClientListener(this);

	return true;

}

void BatSupport::SDK_OnUnload()

{

	players->RemoveClientListener(this);

	List<AdminInterfaceListner *>::iterator iter;

	AdminInterfaceListner *hook;

	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

	{

		hook = (*iter);

		hook->OnAdminInterfaceUnload();

	}

	/* In case plugins don't do this */

	m_hooks.clear();

}

bool BatSupport::SDK_OnMetamodLoad(char *error, size_t maxlength, bool late)

{

	g_SMAPI->AddListener(this, this);

	return true;

}

void BatSupport::OnClientAuthorized(int client, const char *authstring)

{

	List<AdminInterfaceListner *>::iterator iter;

	AdminInterfaceListner *hook;

	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

	{

		hook = (*iter);

		hook->Client_Authorized(client);

	}

}

const char *BatSupport::GetModName()

{

	return "SourceMod";

}

int BatSupport::GetInterfaceVersion()

{

	return ADMININTERFACE_VERSION;

}

void *BatSupport::OnMetamodQuery(const char *iface, int *ret)

{

	if (strcmp(iface, "AdminInterface") == 0)

	{

		AdminInterface *pThis = this;

		if (ret)

		{

			*ret = IFACE_OK;

		}

		return pThis;

	}

	if (ret)

	{

		*ret = IFACE_FAILED;

	}

	return NULL;

}

bool BatSupport::RegisterFlag(const char *Class,const char *Flag,const char *Description)

{

	/* No empty flags */

	if (Flag[0] == '\0')

	{

		g_pSM->LogError(myself, "BAT AdminInterface support tried to register a blank flag");

		return false;

	}

	/* We only support up to 6 custom flags for SourceMod */

	if (m_flags.size() >= 6)

	{

		g_pSM->LogError(myself, "BAT AdminInterface support reached maximum number of custom flags");

		return false;

	}

	List<CustomFlag>::iterator iter;

	for (iter=m_flags.begin(); iter!=m_flags.end(); iter++)

	{

		CustomFlag &cf = (*iter);

		/* Ignore already registered, in case plugin is reloading */

		if (cf.name.compare(Flag) == 0)

		{

			return true;

		}

	}

	g_pSM->LogMessage(myself, 

		"BAT AdminInterface support registered Admin_Custom%d (class \"%s\") (flag \"%s\") (descr \"%s\")",

		m_flags.size() + 1,

		Class,

		Flag,

		Description);

	unsigned int f = (unsigned int)Admin_Custom1;

	f += m_flags.size();

	CustomFlag cf;

	cf.bit = (1<<f);

	cf.flag = (AdminFlag)f;

	cf.name.assign(Flag);

	m_flags.push_back(cf);

	return true;

}

bool BatSupport::IsClient(int id)

{

	IGamePlayer *pPlayer = players->GetGamePlayer(id);

	if (!pPlayer)

	{

		return false;

	}

	if (!pPlayer->IsConnected())

	{

		return false;

	}

	if (pPlayer->IsFakeClient())

	{

		return false;

	}

	return true;

}

void BatSupport::AddEventListner(AdminInterfaceListner *ptr)

{

	m_hooks.push_back(ptr);

}

void BatSupport::RemoveListner(AdminInterfaceListner *ptr)

{

	m_hooks.remove(ptr);

}

bool BatSupport::HasFlag(int id,const char *Flag)

{

	IGamePlayer *pPlayer = players->GetGamePlayer(id);

	if (!pPlayer || !pPlayer->IsConnected())

	{

		return false;

	}

	AdminId admin = pPlayer->GetAdminId();

	if (admin == INVALID_ADMIN_ID)

	{

		return false;

	}

	FlagBits bits = admins->GetAdminFlags(admin, Access_Effective);

	/* Root has it all... except for immunity */

	if ((strcmp(Flag, "immunity") != 0)

	    && ((bits & ADMFLAG_ROOT) == ADMFLAG_ROOT))

	{

		return true;

	}

	if (!strcmp(Flag, "any"))

	{

		return ((bits & ~ADMFLAG_RESERVATION) != 0);

	} else if (!strcmp(Flag, "kick")) {

		return ((bits & ADMFLAG_KICK) == ADMFLAG_KICK);

	} else if (!strcmp(Flag, "slap")) {

		return ((bits & ADMFLAG_SLAY) == ADMFLAG_SLAY);

	} else if (!strcmp(Flag, "slay")) {

		return ((bits & ADMFLAG_SLAY) == ADMFLAG_SLAY);

	} else if (!strcmp(Flag, "ban")) {

		return ((bits & ADMFLAG_BAN) == ADMFLAG_BAN);

	} else if (!strcmp(Flag, "chat")) {

		return ((bits & ADMFLAG_CHAT) == ADMFLAG_CHAT);

	} else if (!strcmp(Flag, "rcon")) {

		return ((bits & ADMFLAG_RCON) == ADMFLAG_RCON);

	} else if (!strcmp(Flag, "map")) {

		return ((bits & ADMFLAG_CHANGEMAP) == ADMFLAG_CHANGEMAP);

	} else if (!strcmp(Flag, "reservedslots")) {

		return ((bits & ADMFLAG_RESERVATION) == ADMFLAG_RESERVATION);

	} else if (!strcmp(Flag, "immunuty")) {

		/* This is a bit different... */

		unsigned int count = admins->GetAdminGroupCount(admin);

		for (unsigned int i=0; i<count; i++)

		{

			GroupId gid = admins->GetAdminGroup(admin, i, NULL);

			if (admins->GetGroupGenericImmunity(gid, Immunity_Default)

				|| admins->GetGroupGenericImmunity(gid, Immunity_Global))

			{

				return true;

			}

		}

		return false;

	}

	List<CustomFlag>::iterator iter;

	for (iter=m_flags.begin(); iter!=m_flags.end(); iter++)

	{

		CustomFlag &cf = (*iter);

		if (cf.name.compare(Flag) == 0)

		{

			return ((bits & cf.bit) == cf.bit);

		}

	}

	return false;

}