/**

  * vim: set ts=4 :

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

  * SourceMod

  * Copyright (C) 2004-2008 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, version 3.0, as published by the

  * Free Software Foundation.

  * 

  * 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, see <http://www.gnu.org/licenses/>.

  *

  * As a special exception, AlliedModders LLC gives you permission to link the

  * code of this program (as well as its derivative works) to "Half-Life 2," the

  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

  * by the Valve Corporation.  You must obey the GNU General Public License in

  * all respects for all other code used.  Additionally, AlliedModders LLC grants

  * this exception to all derivative works.  AlliedModders LLC defines further

  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

  * or <http://www.sourcemod.net/license.php>.

  *

  * Version: $Id$

  */

 #ifndef _INCLUDE_SOURCEMOD_ADMINCACHE_H_

 #define _INCLUDE_SOURCEMOD_ADMINCACHE_H_

 #include "sm_memtable.h"

 #include <sm_trie.h>

 #include <sh_list.h>

 #include <sh_string.h>

 #include <IAdminSystem.h>

 #include "sm_globals.h"

 #include <IForwardSys.h>

 using namespace SourceHook;

 #define GRP_MAGIC_SET		0xDEADFADE

 #define GRP_MAGIC_UNSET		0xFACEFACE

 #define USR_MAGIC_SET		0xDEADFACE

 #define USR_MAGIC_UNSET		0xFADEDEAD

 struct AdminGroup

 {

 	uint32_t magic;					/* Magic flag, for memory validation (ugh) */

 	unsigned int immunity_level;	/* Immunity level */

 	/* Immune from target table (-1 = nonexistent)

 	 * [0] = number of entries

 	 * [1...N] = immune targets

 	 */

 	int immune_table;			

 	Trie *pCmdTable;				/* Command override table (can be NULL) */

 	Trie *pCmdGrpTable;				/* Command group override table (can be NULL) */

 	int next_grp;					/* Next group in the chain */

 	int prev_grp;					/* Previous group in the chain */

 	int nameidx;					/* Name */

 	FlagBits addflags;				/* Additive flags */

 };

 struct AuthMethod

 {

 	String name;

 	Trie *table;

 };

 struct UserAuth

 {

 	unsigned int index;				/* Index into auth table */

 	int identidx;					/* Index into the string table */

 };

 struct AdminUser

 {

 	uint32_t magic;					/* Magic flag, for memory validation */

 	FlagBits flags;					/* Flags */

 	FlagBits eflags;				/* Effective flags */

 	int nameidx;					/* Name index */

 	int password;					/* Password index */

 	unsigned int grp_count;			/* Number of groups */

 	unsigned int grp_size;			/* Size of groups table */

 	int grp_table;					/* Group table itself */

 	int next_user;					/* Next user in the list */

 	int prev_user;					/* Previous user in the list */

 	UserAuth auth;					/* Auth method for this user */

 	unsigned int immunity_level;	/* Immunity level */

 	unsigned int serialchange;		/* Serial # for changes */

 };

 class AdminCache : 

 	public IAdminSystem,

 	public SMGlobalClass

 {

 public:

 	AdminCache();

 	~AdminCache();

 public: //SMGlobalClass

 	void OnSourceModStartup(bool late);

 	void OnSourceModAllInitialized();

 	void OnSourceModLevelChange(const char *mapName);

 	void OnSourceModShutdown();

 	void OnSourceModPluginsLoaded();

 public: //IAdminSystem

 	/** Command cache stuff */

 	void AddCommandOverride(const char *cmd, OverrideType type, FlagBits flags);

 	bool GetCommandOverride(const char *cmd, OverrideType type, FlagBits *flags);

 	void UnsetCommandOverride(const char *cmd, OverrideType type);

 	/** Group cache stuff */

 	GroupId AddGroup(const char *group_name);

 	GroupId FindGroupByName(const char *group_name);

 	void SetGroupAddFlag(GroupId id, AdminFlag flag, bool enabled);

 	bool GetGroupAddFlag(GroupId id, AdminFlag flag);

 	FlagBits GetGroupAddFlags(GroupId id);

 	void SetGroupGenericImmunity(GroupId id, ImmunityType type, bool enabled);

 	bool GetGroupGenericImmunity(GroupId id, ImmunityType type);

 	void InvalidateGroup(GroupId id);

 	void AddGroupImmunity(GroupId id, GroupId other_id);

 	unsigned int GetGroupImmunityCount(GroupId id);

 	GroupId GetGroupImmunity(GroupId id, unsigned int number);

 	void AddGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule rule);

 	bool GetGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule *pRule);

 	void DumpAdminCache(AdminCachePart part, bool rebuild);

 	void AddAdminListener(IAdminListener *pListener);

 	void RemoveAdminListener(IAdminListener *pListener);

 	/** User stuff */

 	void RegisterAuthIdentType(const char *name);

 	AdminId CreateAdmin(const char *name);

 	const char *GetAdminName(AdminId id);

 	bool BindAdminIdentity(AdminId id, const char *auth, const char *ident);

 	virtual void SetAdminFlag(AdminId id, AdminFlag flag, bool enabled);

 	bool GetAdminFlag(AdminId id, AdminFlag flag, AccessMode mode);

 	FlagBits GetAdminFlags(AdminId id, AccessMode mode);

 	bool AdminInheritGroup(AdminId id, GroupId gid);

 	unsigned int GetAdminGroupCount(AdminId id);

 	GroupId GetAdminGroup(AdminId id, unsigned int index, const char **name);

 	void SetAdminPassword(AdminId id, const char *password);

 	const char *GetAdminPassword(AdminId id);

 	AdminId FindAdminByIdentity(const char *auth, const char *identity);

 	bool InvalidateAdmin(AdminId id);

 	unsigned int FlagBitsToBitArray(FlagBits bits, bool array[], unsigned int maxSize);

 	FlagBits FlagBitArrayToBits(const bool array[], unsigned int maxSize);

 	FlagBits FlagArrayToBits(const AdminFlag array[], unsigned int numFlags);

 	unsigned int FlagBitsToArray(FlagBits bits, AdminFlag array[], unsigned int maxSize);

 	bool CheckAdminFlags(AdminId id, FlagBits bits);

 	bool CanAdminTarget(AdminId id, AdminId target);

 	void SetAdminFlags(AdminId id, AccessMode mode, FlagBits bits);

 	bool FindFlag(const char *str, AdminFlag *pFlag);

 	bool FindFlag(char c, AdminFlag *pAdmFlag);

 	FlagBits ReadFlagString(const char *flags, const char **end);

 	unsigned int GetAdminSerialChange(AdminId id);

 	bool CanAdminUseCommand(int client, const char *cmd);

 	const char *GetGroupName(GroupId gid);

 	unsigned int SetGroupImmunityLevel(GroupId gid, unsigned int level);

 	unsigned int GetGroupImmunityLevel(GroupId gid);

 	unsigned int SetAdminImmunityLevel(AdminId id, unsigned int level);

 	unsigned int GetAdminImmunityLevel(AdminId id);

 	bool CheckAccess(int client, 

 		const char *cmd, 

 		FlagBits flags, 

 		bool override_only);

 public:

 	bool IsValidAdmin(AdminId id);

 private:

 	void _UnsetCommandOverride(const char *cmd);

 	void _UnsetCommandGroupOverride(const char *group);

 	void InvalidateGroupCache();

 	void InvalidateAdminCache(bool unlink_admins);

 	void DumpCommandOverrideCache(OverrideType type);

 	Trie *GetMethodByIndex(unsigned int index);

 	bool GetMethodIndex(const char *name, unsigned int *_index);

 	void NameFlag(const char *str, AdminFlag flag);

 public:

 	BaseStringTable *m_pStrings;

 	BaseMemTable *m_pMemory;

 	Trie *m_pCmdOverrides;

 	Trie *m_pCmdGrpOverrides;

 	int m_FirstGroup;

 	int m_LastGroup;

 	int m_FreeGroupList;

 	Trie *m_pGroups;

 	List<IAdminListener *> m_hooks;

 	List<AuthMethod> m_AuthMethods;

 	Trie *m_pAuthTables;

 	IForward *m_pCacheFwd;

 	int m_FirstUser;

 	int m_LastUser;

 	int m_FreeUserList;

 	bool m_InvalidatingAdmins;

 	bool m_destroying;

 	Trie *m_pLevelNames;

 };

 extern AdminCache g_Admins;

 #endif //_INCLUDE_SOURCEMOD_ADMINCACHE_H_