1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/core/Database.cpp	Thu Apr 05 07:08:39 2007 +0000
     1.3 @@ -0,0 +1,153 @@
     1.4 +#include "Database.h"
     1.5 +#include "HandleSys.h"
     1.6 +#include "sourcemod.h"
     1.7 +
     1.8 +DBManager g_DBMan;
     1.9 +
    1.10 +void DBManager::OnSourceModAllInitialized()
    1.11 +{
    1.12 +	HandleAccess sec;
    1.13 +
    1.14 +	g_HandleSys.InitAccessDefaults(NULL, &sec);
    1.15 +	sec.access[HandleAccess_Delete] |= HANDLE_RESTRICT_IDENTITY;
    1.16 +	sec.access[HandleAccess_Clone] |= HANDLE_RESTRICT_IDENTITY;
    1.17 +	
    1.18 +	m_DriverType = g_HandleSys.CreateType("IDriver", this, 0, NULL, &sec, g_pCoreIdent, NULL);
    1.19 +	m_DatabaseType = g_HandleSys.CreateType("IDatabase", this, 0, NULL, NULL, g_pCoreIdent, NULL);
    1.20 +	m_QueryType = g_HandleSys.CreateType("IQuery", this, 0, NULL, NULL, g_pCoreIdent, NULL);
    1.21 +}
    1.22 +
    1.23 +void DBManager::OnHandleDestroy(HandleType_t type, void *object)
    1.24 +{
    1.25 +	if (type == m_DriverType)
    1.26 +	{
    1.27 +		/* Ignore */
    1.28 +		return;
    1.29 +	}
    1.30 +
    1.31 +	if (g_HandleSys.TypeCheck(type, m_QueryType))
    1.32 +	{
    1.33 +		IQuery *pQuery = (IQuery *)object;
    1.34 +		pQuery->Release();
    1.35 +	} else if (g_HandleSys.TypeCheck(type, m_DatabaseType)) {
    1.36 +		IDatabase *pdb = (IDatabase *)object;
    1.37 +		pdb->Close();
    1.38 +	}
    1.39 +}
    1.40 +
    1.41 +void DBManager::AddDriver(IDBDriver *pDriver, IdentityToken_t *ident)
    1.42 +{
    1.43 +	if (!pDriver)
    1.44 +	{
    1.45 +		return;
    1.46 +	}
    1.47 +
    1.48 +	if (m_drivers.size() >= HANDLESYS_MAX_SUBTYPES)
    1.49 +	{
    1.50 +		return;
    1.51 +	}
    1.52 +
    1.53 +	db_driver *driver = new db_driver;
    1.54 +
    1.55 +	driver->driver = pDriver;
    1.56 +	driver->ident = ident;
    1.57 +
    1.58 +	HandleSecurity sec;
    1.59 +
    1.60 +	sec.pIdentity = ident;
    1.61 +	sec.pOwner = g_pCoreIdent;
    1.62 +	driver->hDriver = g_HandleSys.CreateHandleEx(m_DriverType, driver, &sec, NULL, NULL);
    1.63 +
    1.64 +	driver->htDatabase = g_HandleSys.CreateType(NULL, this, m_DatabaseType, NULL, NULL, g_pCoreIdent, NULL);
    1.65 +	driver->htQuery = g_HandleSys.CreateType(NULL, this, m_QueryType, NULL, NULL, g_pCoreIdent, NULL);
    1.66 +	
    1.67 +	m_drivers.push_back(driver);
    1.68 +}
    1.69 +
    1.70 +void DBManager::RemoveDriver(IDBDriver *pDriver)
    1.71 +{
    1.72 +	CVector<db_driver *>::iterator iter;
    1.73 +	for (iter = m_drivers.begin();
    1.74 +		 iter != m_drivers.end();
    1.75 +		 iter++)
    1.76 +	{
    1.77 +		if ((*iter)->driver == pDriver)
    1.78 +		{
    1.79 +			db_driver *driver = (*iter);
    1.80 +			HandleSecurity sec;
    1.81 +
    1.82 +			sec.pIdentity = driver->ident;
    1.83 +			sec.pOwner = g_pCoreIdent;
    1.84 +			g_HandleSys.RemoveType(driver->htQuery, driver->ident);
    1.85 +			g_HandleSys.RemoveType(driver->htDatabase, driver->ident);
    1.86 +			g_HandleSys.FreeHandle(driver->hDriver, &sec);
    1.87 +
    1.88 +			delete driver;
    1.89 +			m_drivers.erase(iter);
    1.90 +
    1.91 +			return;
    1.92 +		}
    1.93 +	}
    1.94 +}
    1.95 +
    1.96 +const char *DBManager::GetInterfaceName()
    1.97 +{
    1.98 +	return SMINTERFACE_DBI_NAME;
    1.99 +}
   1.100 +
   1.101 +unsigned int DBManager::GetInterfaceVersion()
   1.102 +{
   1.103 +	return SMINTERFACE_DBI_VERSION;
   1.104 +}
   1.105 +
   1.106 +unsigned int DBManager::GetDriverCount()
   1.107 +{
   1.108 +	return (unsigned int)m_drivers.size();
   1.109 +}
   1.110 +
   1.111 +IDBDriver *DBManager::GetDriver(unsigned int index, IdentityToken_t **pToken)
   1.112 +{
   1.113 +	if (index >= GetDriverCount())
   1.114 +	{
   1.115 +		return NULL;
   1.116 +	}
   1.117 +
   1.118 +	if (pToken)
   1.119 +	{
   1.120 +		*pToken = m_drivers[index]->ident;
   1.121 +	}
   1.122 +
   1.123 +	return m_drivers[index]->driver;
   1.124 +}
   1.125 +
   1.126 +const DatabaseInfo *DBManager::FindDatabaseConf(const char *name)
   1.127 +{
   1.128 +	/* :TODO: implement */
   1.129 +	return NULL;
   1.130 +}
   1.131 +
   1.132 +bool DBManager::Connect(const char *name, IDBDriver **pdr, IDatabase **pdb, bool persistent)
   1.133 +{
   1.134 +	const DatabaseInfo *pInfo = FindDatabaseConf(name);
   1.135 +
   1.136 +	unsigned int count = GetDriverCount();
   1.137 +	for (unsigned int i=0; i<count; i++)
   1.138 +	{
   1.139 +		IDBDriver *driver = GetDriver(i);
   1.140 +		if (strcmp(driver->GetIdentifier(), pInfo->driver) == 0)
   1.141 +		{
   1.142 +			*pdr = driver;
   1.143 +			if (persistent)
   1.144 +			{
   1.145 +				*pdb = driver->PConnect(pInfo);
   1.146 +			} else {
   1.147 +				*pdb = driver->Connect(pInfo);
   1.148 +			}
   1.149 +		}
   1.150 +	}
   1.151 +
   1.152 +	*pdr = NULL;
   1.153 +	*pdb = NULL;
   1.154 +
   1.155 +	return false;
   1.156 +}