/**

  * vim: set ts=4 :

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

  * SourceMod Client Preferences Extension

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

  */

 #include "extension.h"

 /**

  * @file extension.cpp

  * @brief Implement extension code here.

  */

 ClientPrefs g_ClientPrefs;		/**< Global singleton for extension's main interface */

 SMEXT_LINK(&g_ClientPrefs);

 HandleType_t g_CookieType = 0;

 CookieTypeHandler g_CookieTypeHandler;

 HandleType_t g_CookieIterator = 0;

 CookieIteratorHandler g_CookieIteratorHandler;

 int driver = 0;

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

 {	

 	const DatabaseInfo *DBInfo = dbi->FindDatabaseConf("clientprefs");

 	if (DBInfo == NULL)

 	{

 		DBInfo = dbi->FindDatabaseConf("default");

 		if (DBInfo == NULL)

 		{

 			snprintf(error, maxlength, "Could not find \"clientprefs\" or \"default\" database configs");

 			return false;

 		}

 	}

 	if (DBInfo->driver[0] != '\0')

 	{

 		Driver = dbi->FindOrLoadDriver(DBInfo->driver);

 	}

 	else

 	{

 		Driver = dbi->GetDefaultDriver();

 	}

 	if (Driver == NULL)

 	{

 		snprintf(error, maxlength, "Could not load DB Driver \"%s\"", DBInfo->driver);

 		return false;

 	}

 	Database = Driver->Connect(DBInfo, true, error, maxlength);

 	if (Database == NULL)

 	{

 		return false;

 	}

 	dbi->AddDependency(myself, Driver);

 	const char *identifier = Driver->GetIdentifier();

 	if (strcmp(identifier, "sqlite") == 0)

 	{

 		driver = DRIVER_SQLITE;

 		TQueryOp *op = new TQueryOp(

 						Database, 

 						"CREATE TABLE IF NOT EXISTS sm_cookies  \

 						( \

 							id INTEGER PRIMARY KEY AUTOINCREMENT, \

 							name varchar(30) NOT NULL UNIQUE, \

 							description varchar(255), \

 							access INTEGER \

 						)", 

 						Query_CreateTable, 

 						0);

 		dbi->AddToThreadQueue(op, PrioQueue_Normal);

 		op = new TQueryOp(

 						Database, 

 						"CREATE TABLE IF NOT EXISTS sm_cookie_cache \

 						( \

 							player varchar(65) NOT NULL, \

 							cookie_id int(10) NOT NULL, \

 							value varchar(100), \

 							timestamp int, \

 							PRIMARY KEY (player, cookie_id) \

 						)", 

 						Query_CreateTable, 

 						0);

 		dbi->AddToThreadQueue(op, PrioQueue_Normal);

 	}

 	else if (strcmp(identifier, "mysql") == 0)

 	{

 		driver = DRIVER_MYSQL;

 		TQueryOp *op = new TQueryOp(

 						Database, 

 						"CREATE TABLE IF NOT EXISTS sm_cookies \

 						( \

 							id INTEGER unsigned NOT NULL auto_increment, \

 							name varchar(30) NOT NULL UNIQUE, \

 							description varchar(255), \

 							access INTEGER, \

 							PRIMARY KEY (id) \

 						)", 

 						Query_CreateTable, 

 						0);

 		dbi->AddToThreadQueue(op, PrioQueue_Normal);

 		op = new TQueryOp(

 						Database, 

 						"CREATE TABLE IF NOT EXISTS sm_cookie_cache \

 						( \

 							player varchar(65) NOT NULL, \

 							cookie_id int(10) NOT NULL, \

 							value varchar(100), \

 							timestamp int NOT NULL, \

 							PRIMARY KEY (player, cookie_id) \

 						)", 

 						Query_CreateTable, 

 						0);

 		dbi->AddToThreadQueue(op, PrioQueue_Normal);

 	}

 	else

 	{

 		snprintf(error, maxlength, "Unsupported driver \"%s\"", identifier);

 		return false;

 	}

 	sharesys->AddNatives(myself, g_ClientPrefNatives);

 	sharesys->RegisterLibrary(myself, "clientprefs");

 	g_CookieManager.cookieDataLoadedForward = forwards->CreateForward("OnClientCookiesCached", ET_Ignore, 1, NULL, Param_Cell);

 	g_CookieType = handlesys->CreateType("Cookie", 

 		&g_CookieTypeHandler, 

 		0, 

 		NULL, 

 		NULL, 

 		myself->GetIdentity(), 

 		NULL);

 	g_CookieIterator = handlesys->CreateType("CookieIterator", 

 		&g_CookieIteratorHandler, 

 		0, 

 		NULL, 

 		NULL, 

 		myself->GetIdentity(), 

 		NULL);

 	IMenuStyle *style = menus->GetDefaultStyle();

 	g_CookieManager.clientMenu = style->CreateMenu(&g_Handler, NULL);

 	g_CookieManager.clientMenu->SetDefaultTitle("Client Settings:");

 	plsys->AddPluginsListener(&g_CookieManager);

 	phrases = translator->CreatePhraseCollection();

 	phrases->AddPhraseFile("clientprefs.phrases");

 	phrases->AddPhraseFile("common.phrases");

 	return true;

 }

 void ClientPrefs::SDK_OnAllLoaded()

 {

 	playerhelpers->AddClientListener(&g_CookieManager);

 }

 bool ClientPrefs::QueryInterfaceDrop(SMInterface *pInterface)

 {

 	if ((void *)pInterface == (void *)(Database->GetDriver()))

 	{

 		return false;

 	}

 	return true;

 }

 void ClientPrefs::NotifyInterfaceDrop(SMInterface *pInterface)

 {

 	if (Database != NULL && (void *)pInterface == (void *)(Database->GetDriver()))

 	{

 		Database->Close();

 		Database = NULL;

 	}

 }

 void ClientPrefs::SDK_OnUnload()

 {

 	handlesys->RemoveType(g_CookieType, myself->GetIdentity());

 	handlesys->RemoveType(g_CookieIterator, myself->GetIdentity());

 	g_CookieManager.Unload();

 	if (Database != NULL)

 	{

 		Database->Close();

 	}

 	forwards->ReleaseForward(g_CookieManager.cookieDataLoadedForward);

 	g_CookieManager.clientMenu->Destroy();

 	phrases->Destroy();

 	plsys->RemovePluginsListener(&g_CookieManager);

 	playerhelpers->RemoveClientListener(&g_CookieManager);

 }

 size_t UTIL_Format(char *buffer, size_t maxlength, const char *fmt, ...)

 {

 	va_list ap;

 	va_start(ap, fmt);

 	size_t len = vsnprintf(buffer, maxlength, fmt, ap);

 	va_end(ap);

 	if (len >= maxlength)

 	{

 		buffer[maxlength - 1] = '\0';

 		return (maxlength - 1);

 	}

 	else

 	{

 		return len;

 	}

 }

 bool Translate(char *buffer, 

 				   size_t maxlength,

 				   const char *format,

 				   unsigned int numparams,

 				   size_t *pOutLength,

 				   ...)

 {

 	va_list ap;

 	unsigned int i;

 	const char *fail_phrase;

 	void *params[MAX_TRANSLATE_PARAMS];

 	if (numparams > MAX_TRANSLATE_PARAMS)

 	{

 		assert(false);

 		return false;

 	}

 	va_start(ap, pOutLength);

 	for (i = 0; i < numparams; i++)

 	{

 		params[i] = va_arg(ap, void *);

 	}

 	va_end(ap);

 	if (!g_ClientPrefs.phrases->FormatString(buffer,

 		maxlength, 

 		format, 

 		params,

 		numparams,

 		pOutLength,

 		&fail_phrase))

 	{

 		if (fail_phrase != NULL)

 		{

 			g_pSM->LogError(myself, "[SM] Could not find core phrase: %s", fail_phrase);

 		}

 		else

 		{

 			g_pSM->LogError(myself, "[SM] Unknown fatal error while translating a core phrase.");

 		}

 		return false;

 	}

 	return true;

 }