/**

  * 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;

 }