/**

  * vim: set ts=4 sw=4 tw=99 noet :

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

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

 #define _INCLUDE_DATABASE_MANAGER_H_

 #include <IDBDriver.h>

 #include "sm_globals.h"

 #include <sh_vector.h>

 #include <sh_string.h>

 #include <sh_list.h>

 #include <ITextParsers.h>

 #include "sm_memtable.h"

 #include <IThreader.h>

 #include <IPluginSys.h>

 #include <am-thread-utils.h>

 #include "sm_simple_prioqueue.h"

 using namespace SourceHook;

 struct ConfDbInfo

 {

 	ConfDbInfo() : realDriver(NULL)

 	{

 	}

 	String name;

 	String driver;

 	String host;

 	String user;

 	String pass;

 	String database;

 	IDBDriver *realDriver;

 	DatabaseInfo info;

 };

 class DBManager : 

 	public IDBManager,

 	public SMGlobalClass,

 	public IHandleTypeDispatch,

 	public ITextListener_SMC,

 	public IPluginsListener,

 	public ke::IRunnable

 {

 public:

 	DBManager();

 public:

 	const char *GetInterfaceName();

 	unsigned int GetInterfaceVersion();

 public: //SMGlobalClass

 	void OnSourceModAllInitialized();

 	void OnSourceModLevelChange(const char *mapName);

 	void OnSourceModIdentityDropped(IdentityToken_t *pToken);

 	void OnSourceModShutdown();

 public: //IHandleTypeDispatch

 	void OnHandleDestroy(HandleType_t type, void *object);

 public: //IDBManager

 	void AddDriver(IDBDriver *pDrivera);

 	void RemoveDriver(IDBDriver *pDriver);

 	const DatabaseInfo *FindDatabaseConf(const char *name);

 	bool Connect(const char *name, IDBDriver **pdr, IDatabase **pdb, bool persistent, char *error, size_t maxlength);

 	unsigned int GetDriverCount();

 	IDBDriver *GetDriver(unsigned int index);

 	Handle_t CreateHandle(DBHandleType type, void *ptr, IdentityToken_t *pToken);

 	HandleError ReadHandle(Handle_t hndl, DBHandleType type, void **ptr);

 	HandleError ReleaseHandle(Handle_t hndl, DBHandleType type, IdentityToken_t *token);

 	void AddDependency(IExtension *myself, IDBDriver *driver);

 public: //ITextListener_SMC

 	void ReadSMC_ParseStart();

 	SMCResult ReadSMC_NewSection(const SMCStates *states, const char *name);

 	SMCResult ReadSMC_KeyValue(const SMCStates *states, const char *key, const char *value);

 	SMCResult ReadSMC_LeavingSection(const SMCStates *states);

 	void ReadSMC_ParseEnd(bool halted, bool failed);

 public: //ke::IRunnable

 	void Run();

 	void ThreadMain();

 public: //IPluginsListener

 	void OnPluginUnloaded(IPlugin *plugin);

 public:

 	ConfDbInfo *GetDatabaseConf(const char *name);

 	IDBDriver *FindOrLoadDriver(const char *name);

 	IDBDriver *GetDefaultDriver();

 	const char *GetDefaultDriverName();

 	bool AddToThreadQueue(IDBThreadOperation *op, PrioQueueLevel prio);

 	void LockConfig();

 	void UnlockConfig();

 	void RunFrame();

 	inline HandleType_t GetDatabaseType()

 	{

 		return m_DatabaseType;

 	}

 private:

 	void ClearConfigs();

 	void KillWorkerThread();

 private:

 	CVector<IDBDriver *> m_drivers;

 	/* Threading stuff */

 	PrioQueue<IDBThreadOperation *> m_OpQueue;

 	Queue<IDBThreadOperation *> m_ThinkQueue;

 	CVector<bool> m_drSafety;			/* which drivers are safe? */

 	ke::AutoPtr<ke::Thread> m_Worker;

 	ke::ConditionVariable m_QueueEvent;

 	ke::Mutex m_ConfigLock;

 	ke::Mutex m_ThinkLock;

 	bool m_Terminate;

 	List<ConfDbInfo *> m_confs;

 	HandleType_t m_DriverType;

 	HandleType_t m_DatabaseType;

 	String m_DefDriver;

 	char m_Filename[PLATFORM_MAX_PATH];

 	unsigned int m_ParseLevel;

 	unsigned int m_ParseState;

 	IDBDriver *m_pDefault;

 };

 extern DBManager g_DBMan;

 #endif //_INCLUDE_DATABASE_MANAGER_H_