/**

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

  */

 #include <ITextParsers.h>

 #include "CoreConfig.h"

 #include "sourcemod.h"

 #include "sourcemm_api.h"

 #include "sm_srvcmds.h"

 #include <sourcemod_version.h>

 #include "sm_stringutil.h"

 #include "LibrarySys.h"

 #include "Logger.h"

 #include "ForwardSys.h"

 #include "frame_hooks.h"

 #include "logic_bridge.h"

 #ifdef PLATFORM_WINDOWS

 ConVar sm_corecfgfile("sm_corecfgfile", "addons\\sourcemod\\configs\\core.cfg", 0, "SourceMod core configuration file");

 #elif defined PLATFORM_LINUX || defined PLATFORM_APPLE

 ConVar sm_corecfgfile("sm_corecfgfile", "addons/sourcemod/configs/core.cfg", 0, "SourceMod core configuration file");

 #endif

 IForward *g_pOnServerCfg = NULL;

 IForward *g_pOnConfigsExecuted = NULL;

 IForward *g_pOnAutoConfigsBuffered = NULL;

 CoreConfig g_CoreConfig;

 bool g_bConfigsExecd = false;

 bool g_bServerExecd = false;

 bool g_bGotServerStart = false;

 bool g_bGotTrigger = false;

 ConCommand *g_pExecPtr = NULL;

 ConVar *g_ServerCfgFile = NULL;

 void CheckAndFinalizeConfigs();

 #if SOURCE_ENGINE == SE_DOTA

 SH_DECL_EXTERN2_void(ConCommand, Dispatch, SH_NOATTRIB, false, const CCommandContext &, const CCommand &);

 void Hook_ExecDispatchPre(const CCommandContext &context, const CCommand &cmd)

 #elif SOURCE_ENGINE >= SE_ORANGEBOX

 SH_DECL_EXTERN1_void(ConCommand, Dispatch, SH_NOATTRIB, false, const CCommand &);

 void Hook_ExecDispatchPre(const CCommand &cmd)

 #elif SOURCE_ENGINE == SE_DARKMESSIAH

 SH_DECL_EXTERN0_void(ConCommand, Dispatch, SH_NOATTRIB, false);

 void Hook_ExecDispatchPre()

 #else

 # if SH_IMPL_VERSION >= 4

  extern int __SourceHook_FHAddConCommandDispatch(void *,bool,class fastdelegate::FastDelegate0<void>);

 # else

  extern bool __SourceHook_FHAddConCommandDispatch(void *,bool,class fastdelegate::FastDelegate0<void>);

 # endif

 extern bool __SourceHook_FHRemoveConCommandDispatch(void *,bool,class fastdelegate::FastDelegate0<void>);

 void Hook_ExecDispatchPre()

 #endif

 {

 #if SOURCE_ENGINE <= SE_DARKMESSIAH

 	CCommand cmd;

 #endif

 	const char *arg = cmd.Arg(1);

 	if (!g_bServerExecd && arg != NULL && strcmp(arg, g_ServerCfgFile->GetString()) == 0)

 	{

 		g_bGotTrigger = true;

 	}

 }

 #if SOURCE_ENGINE == SE_DOTA

 void Hook_ExecDispatchPost(const CCommandContext &context, const CCommand &cmd)

 #elif SOURCE_ENGINE >= SE_ORANGEBOX

 void Hook_ExecDispatchPost(const CCommand &cmd)

 #else

 void Hook_ExecDispatchPost()

 #endif

 {

 	if (g_bGotTrigger)

 	{

 		g_bGotTrigger = false;

 		g_bServerExecd = true;

 		CheckAndFinalizeConfigs();

 	}

 }

 void CheckAndFinalizeConfigs()

 {

 	if ((g_bServerExecd || g_ServerCfgFile == NULL) && g_bGotServerStart)

 	{

         g_PendingInternalPush = true;

 	}

 }

 void CoreConfig::OnSourceModAllInitialized()

 {

 	g_RootMenu.AddRootConsoleCommand("config", "Set core configuration options", this);

 	g_pOnServerCfg = g_Forwards.CreateForward("OnServerCfg", ET_Ignore, 0, NULL);

 	g_pOnConfigsExecuted = g_Forwards.CreateForward("OnConfigsExecuted", ET_Ignore, 0, NULL);

 	g_pOnAutoConfigsBuffered = g_Forwards.CreateForward("OnAutoConfigsBuffered", ET_Ignore, 0, NULL);

 }

 CoreConfig::CoreConfig() : m_Strings(512)

 {

 }

 CoreConfig::~CoreConfig()

 {

 }

 void CoreConfig::OnSourceModShutdown()

 {

 	g_RootMenu.RemoveRootConsoleCommand("config", this);

 	g_Forwards.ReleaseForward(g_pOnServerCfg);

 	g_Forwards.ReleaseForward(g_pOnConfigsExecuted);

 	g_Forwards.ReleaseForward(g_pOnAutoConfigsBuffered);

 	if (g_pExecPtr != NULL)

 	{

 		SH_REMOVE_HOOK(ConCommand, Dispatch, g_pExecPtr, SH_STATIC(Hook_ExecDispatchPre), false);

 		SH_REMOVE_HOOK(ConCommand, Dispatch, g_pExecPtr, SH_STATIC(Hook_ExecDispatchPost), true);

 		g_pExecPtr = NULL;

 	}

 }

 void CoreConfig::OnSourceModLevelChange(const char *mapName)

 {

 	static bool already_checked = false;

 	if (!already_checked)

 	{

 		if (engine->IsDedicatedServer())

 		{

 			g_ServerCfgFile = icvar->FindVar("servercfgfile");

 		}

 		else

 		{

 			g_ServerCfgFile = icvar->FindVar("lservercfgfile");

 		}

 		if (g_ServerCfgFile != NULL)

 		{

 			g_pExecPtr = FindCommand("exec");

 			if (g_pExecPtr != NULL)

 			{

 				SH_ADD_HOOK(ConCommand, Dispatch, g_pExecPtr, SH_STATIC(Hook_ExecDispatchPre), false);

 				SH_ADD_HOOK(ConCommand, Dispatch, g_pExecPtr, SH_STATIC(Hook_ExecDispatchPost), true);

 			}

 			else

 			{

 				g_ServerCfgFile = NULL;

 			}

 		}

 		already_checked = true;

 	}

 	g_bConfigsExecd = false;

 	g_bServerExecd = false;

 	g_bGotServerStart = false;

 	g_bGotTrigger = false;

 }

 void CoreConfig::OnRootConsoleCommand(const char *cmdname, const CCommand &command)

 {

 	int argcount = command.ArgC();

 	if (argcount >= 4)

 	{

 		const char *option = command.Arg(2);

 		const char *value = command.Arg(3);

 		char error[255];

 		ConfigResult res = SetConfigOption(option, value, ConfigSource_Console, error, sizeof(error));

 		if (res == ConfigResult_Reject)

 		{

 			g_RootMenu.ConsolePrint("[SM] Could not set config option \"%s\" to \"%s\". (%s)", option, value, error);

 		} else if (res == ConfigResult_Ignore) {

 			g_RootMenu.ConsolePrint("[SM] No such config option \"%s\" exists.", option);

 		} else {

 			g_RootMenu.ConsolePrint("[SM] Config option \"%s\" successfully set to \"%s\".", option, value);

 		}

 		return;

 	} else if (argcount >= 3) {

 		const char *option = command.Arg(2);

 		const char *value = GetCoreConfigValue(option);

 		if (value == NULL)

 		{

 			g_RootMenu.ConsolePrint("[SM] No such config option \"%s\" exists.", option);

 		} else {

 			g_RootMenu.ConsolePrint("[SM] Config option \"%s\" is set to \"%s\".", option, value);

 		}

 		return;

 	}

 	g_RootMenu.ConsolePrint("[SM] Usage: sm config <option> [value]");

 }

 void CoreConfig::Initialize()

 {

 	SMCError err;

 	char filePath[PLATFORM_MAX_PATH];

 	/* Try to get command line value of core config convar */

 	const char *corecfg = icvar->GetCommandLineValue("sm_corecfgfile");

 	/* If sm_corecfgfile is on the command line, use that

 	 * If sm_corecfgfile isn't there, check sm_basepath on the command line and build the path off that

 	 * If sm_basepath isn't there, just use the default path for the cfg

 	 */

 	if (corecfg)

 	{

 		g_LibSys.PathFormat(filePath, sizeof(filePath), "%s/%s", g_SourceMod.GetGamePath(), corecfg);

 	}

 	else

 	{

 		const char *basepath = icvar->GetCommandLineValue("sm_basepath");

 		/* Format path to config file */

 		if (basepath)

 		{

 			g_LibSys.PathFormat(filePath, sizeof(filePath), "%s/%s/%s", g_SourceMod.GetGamePath(), basepath, "configs/core.cfg");

 		}

 		else

 		{

 			g_LibSys.PathFormat(filePath, sizeof(filePath), "%s/%s", g_SourceMod.GetGamePath(), sm_corecfgfile.GetDefault());

 		}

 	}

 	/* Reset cached key values */

 	m_KeyValues.clear();

 	m_Strings.Reset();

 	/* Parse config file */

 	if ((err=textparsers->ParseFile_SMC(filePath, this, NULL)) != SMCError_Okay)

 	{

  		/* :TODO: This won't actually log or print anything :( - So fix that somehow */

 		const char *error = textparsers->GetSMCErrorString(err);

 		g_Logger.LogFatal("[SM] Error encountered parsing core config file: %s", error ? error : "");

 	}

 }

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

 {

 	char error[255];

 	ConfigResult err = SetConfigOption(key, value, ConfigSource_File, error, sizeof(error));

 	if (err == ConfigResult_Reject)

 	{

 		/* This is a fatal error */

 		g_Logger.LogFatal("Config error (key: %s) (value: %s) %s", key, value, error);

 	}

 	return SMCResult_Continue;

 }

 ConfigResult CoreConfig::SetConfigOption(const char *option, const char *value, ConfigSource source, char *error, size_t maxlength)

 {

 	ConfigResult result;

 	/* Notify! */

 	SMGlobalClass *pBase = SMGlobalClass::head;

 	while (pBase)

 	{

 		if ((result = pBase->OnSourceModConfigChanged(option, value, source, error, maxlength)) != ConfigResult_Ignore)

 		{

 			return result;

 		}

 		pBase = pBase->m_pGlobalClassNext;

 	}

 	m_KeyValues.replace(option, m_Strings.AddString(value));

 	return ConfigResult_Ignore;

 }

 const char *CoreConfig::GetCoreConfigValue(const char *key)

 {

 	int address;

 	if (!m_KeyValues.retrieve(key, &address))

 		return NULL;

 	return m_Strings.GetString(address);

 }

 bool SM_AreConfigsExecuted()

 {

 	return g_bConfigsExecd;

 }

 inline bool IsPathSepChar(char c)

 {

 #if defined PLATFORM_WINDOWS

 	return (c == '\\' || c == '/');

 #elif defined PLATFORM_LINUX || defined PLATFORM_POSIX

 	return (c == '/');

 #endif

 }

 bool SM_ExecuteConfig(IPlugin *pl, AutoConfig *cfg, bool can_create)

 {

 	bool will_create = false;

 	/* See if we should be creating */

 	if (can_create && cfg->create)

 	{

 		will_create = true;

 		/* If the folder does not exist, attempt to create it.  

 		 * We're awfully nice.

 		 */

 		const char *folder = cfg->folder.c_str();

 		char path[PLATFORM_MAX_PATH];

 		char build[PLATFORM_MAX_PATH];

 		g_SourceMod.BuildPath(Path_Game, path, sizeof(path), "cfg/%s", folder);

 		if (!g_LibSys.IsPathDirectory(path))

 		{

 			char *cur_ptr = path;

 			size_t len;

 			g_LibSys.PathFormat(path, sizeof(path), "%s", folder);

 			len = g_SourceMod.BuildPath(Path_Game, build, sizeof(build), "cfg");

 			do

 			{

 				/* Find next suitable path */

 				char *next_ptr = cur_ptr;

 				while (*next_ptr != '\0')

 				{

 					if (IsPathSepChar(*next_ptr))

 					{

 						*next_ptr = '\0';

 						next_ptr++;

 						break;

 					}

 					next_ptr++;

 				}

 				if (*next_ptr == '\0')

 				{

 					next_ptr = NULL;

 				}

 				len += g_LibSys.PathFormat(&build[len], 

 					sizeof(build)-len,

 					"/%s",

 					cur_ptr);

 				if (!g_LibSys.CreateFolder(build))

 				{

 					break;

 				}

 				cur_ptr = next_ptr;

 			} while (cur_ptr);

 		}

 	}

 	/* Check if the file exists. */

 	char file[PLATFORM_MAX_PATH];

 	char local[PLATFORM_MAX_PATH];

 	if (cfg->folder.size())

 	{

 		g_LibSys.PathFormat(local, 

 			sizeof(local), 

 			"%s/%s.cfg",

 			cfg->folder.c_str(),

 			cfg->autocfg.c_str());

 	} else {

 		g_LibSys.PathFormat(local,

 			sizeof(local),

 			"%s.cfg",

 			cfg->autocfg.c_str());

 	}

 	g_SourceMod.BuildPath(Path_Game, file, sizeof(file), "cfg/%s", local);

 	bool file_exists = g_LibSys.IsPathFile(file);

 	if (!file_exists && will_create)

 	{

 		List<const ConVar *> *convars = NULL;

 		if (pl->GetProperty("ConVarList", (void **)&convars, false) && convars)

 		{

 			/* Attempt to create it */

 			FILE *fp = fopen(file, "wt");

 			if (fp)

 			{

 				fprintf(fp, "// This file was auto-generated by SourceMod (v%s)\n", SM_VERSION_STRING);

 				fprintf(fp, "// ConVars for plugin \"%s\"\n", pl->GetFilename());

 				fprintf(fp, "\n\n");

 				List<const ConVar *>::iterator iter;

 				float x;

 				for (iter = convars->begin(); iter != convars->end(); iter++)

 				{

 					const ConVar *cvar = (*iter);

 					if ((cvar->GetFlags() & FCVAR_DONTRECORD) == FCVAR_DONTRECORD)

 					{

 						continue;

 					}

 					char descr[255];

 					char *dptr = descr;

 					/* Print comments until there is no more */

 					strncopy(descr, cvar->GetHelpText(), sizeof(descr));

 					while (*dptr != '\0')

 					{

 						/* Find the next line */

 						char *next_ptr = dptr;

 						while (*next_ptr != '\0')

 						{

 							if (*next_ptr == '\n')

 							{

 								*next_ptr = '\0';

 								next_ptr++;

 								break;

 							}

 							next_ptr++;

 						}

 						fprintf(fp, "// %s\n", dptr);

 						dptr = next_ptr;

 					}

 					fprintf(fp, "// -\n");

 					fprintf(fp, "// Default: \"%s\"\n", cvar->GetDefault());

 					if (cvar->GetMin(x))

 					{

 						fprintf(fp, "// Minimum: \"%02f\"\n", x);

 					}

 					if (cvar->GetMax(x))

 					{

 						fprintf(fp, "// Maximum: \"%02f\"\n", x);

 					}

 					fprintf(fp, "%s \"%s\"\n", cvar->GetName(), cvar->GetDefault());

 					fprintf(fp, "\n");

 				}

 				fprintf(fp, "\n");

 				file_exists = true;

 				can_create = false;

 				fclose(fp);

 			}

 			else

 			{

 				g_Logger.LogError("Failed to auto generate config for %s, make sure the directory has write permission.", pl->GetFilename());

 				return can_create;

 			}

 		}

 	}

 	if (file_exists)

 	{

 		char cmd[255];

 		UTIL_Format(cmd, sizeof(cmd), "exec %s\n", local);

 		engine->ServerCommand(cmd);

 	}

 	return can_create;

 }

 void SM_DoSingleExecFwds(IPluginContext *ctx)

 {

 	IPluginFunction *pf;

 	if ((pf = ctx->GetFunctionByName("OnServerCfg")) != NULL)

 	{

 		pf->Execute(NULL);

 	}

 	if ((pf = ctx->GetFunctionByName("OnConfigsExecuted")) != NULL)

 	{

 		pf->Execute(NULL);

 	}

 }

 void SM_ConfigsExecuted_Plugin(unsigned int serial)

 {

 	IPluginIterator *iter = scripts->GetPluginIterator();

 	while (iter->MorePlugins())

 	{

 		IPlugin *plugin = iter->GetPlugin();

 		if (plugin->GetSerial() == serial)

 		{

 			SM_DoSingleExecFwds(plugin->GetBaseContext());

 			break;

 		}

 		iter->NextPlugin();

 	}

 	iter->Release();

 }

 void SM_ExecuteForPlugin(IPluginContext *ctx)

 {

 	SMPlugin *plugin = scripts->FindPluginByContext(ctx->GetContext());

 	unsigned int num = plugin->GetConfigCount();

 	if (!num)

 	{

 		SM_DoSingleExecFwds(ctx);

 	}

 	else

 	{

 		bool can_create = true;

 		for (unsigned int i=0; i<num; i++)

 		{

 			can_create = SM_ExecuteConfig(plugin, plugin->GetConfig(i), can_create);

 		}

 		char cmd[255];

 		UTIL_Format(cmd, sizeof(cmd), "sm internal 2 %d\n", plugin->GetSerial());

 		engine->ServerCommand(cmd);

 	}

 }

 void SM_ExecuteAllConfigs()

 {

 	if (g_bGotServerStart)

 	{

 		return;

 	}

 	engine->ServerCommand("exec sourcemod/sourcemod.cfg\n");

 	AutoPluginList plugins(scripts);

 	for (size_t i = 0; i < plugins->size(); i++)

 	{

 		SMPlugin *plugin = plugins->at(i);

 		unsigned int num = plugin->GetConfigCount();

 		bool can_create = true;

 		for (unsigned int i=0; i<num; i++)

 		{

 			can_create = SM_ExecuteConfig(plugin, plugin->GetConfig(i), can_create);

 		}

 	}

 	g_bGotServerStart = true;

 	CheckAndFinalizeConfigs();

 }

 void SM_ConfigsExecuted_Global()

 {

 	if (g_bConfigsExecd)

 	{

 		return;

 	}

 	g_bConfigsExecd = true;

 	g_pOnServerCfg->Execute(NULL);

 	g_pOnConfigsExecuted->Execute(NULL);

 }

 void SM_InternalCmdTrigger()

 {

 	/* Order is important here.  We need to buffer things before we send the command out. */

 	g_pOnAutoConfigsBuffered->Execute(NULL);

 	engine->ServerCommand("sm internal 1\n");

     g_PendingInternalPush = false;

 }