ds@661: /**
ds@661:  * vim: set ts=4 :
ds@1330:  * =============================================================================
dvander@1231:  * SourceMod
dvander@1231:  * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.
ds@1330:  * =============================================================================
ds@661:  *
ds@1330:  * This program is free software; you can redistribute it and/or modify it under
ds@1330:  * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330:  * Free Software Foundation.
dvander@1231:  * 
ds@1330:  * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
ds@1330:  * details.
dvander@1231:  *
ds@1330:  * You should have received a copy of the GNU General Public License along with
ds@1330:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1231:  *
ds@1330:  * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330:  * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330:  * by the Valve Corporation.  You must obey the GNU General Public License in
ds@1330:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
ds@1330:  * this exception to all derivative works.  AlliedModders LLC defines further
ds@1330:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330:  * or <http://www.sourcemod.net/license.php>.
ds@661:  *
ds@661:  * Version: $Id$
ds@661:  */
ds@661: 
dvander@1406: #include <ITextParsers.h>
ds@661: #include "CoreConfig.h"
ds@661: #include "sourcemod.h"
ds@661: #include "sourcemm_api.h"
ds@661: #include "sm_srvcmds.h"
dvander@978: #include "sm_version.h"
dvander@978: #include "sm_stringutil.h"
ds@661: #include "LibrarySys.h"
ds@661: #include "Logger.h"
dvander@978: #include "PluginSys.h"
dvander@978: #include "ForwardSys.h"
ds@661: 
ds@661: #ifdef PLATFORM_WINDOWS
ds@661: ConVar sm_corecfgfile("sm_corecfgfile", "addons\\sourcemod\\configs\\core.cfg", 0, "SourceMod core configuration file");
dvander@663: #elif defined PLATFORM_LINUX
ds@661: ConVar sm_corecfgfile("sm_corecfgfile", "addons/sourcemod/configs/core.cfg", 0, "SourceMod core configuration file");
ds@661: #endif
ds@661: 
dvander@978: IForward *g_pOnServerCfg = NULL;
dvander@978: IForward *g_pOnConfigsExecuted = NULL;
ds@661: CoreConfig g_CoreConfig;
dvander@978: bool g_bConfigsExecd = false;
ds@661: 
ds@661: void CoreConfig::OnSourceModAllInitialized()
ds@661: {
ds@661: 	g_RootMenu.AddRootConsoleCommand("config", "Set core configuration options", this);
dvander@978: 	g_pOnServerCfg = g_Forwards.CreateForward("OnServerCfg", ET_Ignore, 0, NULL);
dvander@978: 	g_pOnConfigsExecuted = g_Forwards.CreateForward("OnConfigsExecuted", ET_Ignore, 0, NULL);
ds@661: }
ds@661: 
ds@661: void CoreConfig::OnSourceModShutdown()
ds@661: {
ds@661: 	g_RootMenu.RemoveRootConsoleCommand("config", this);
dvander@978: 	g_Forwards.ReleaseForward(g_pOnServerCfg);
dvander@978: 	g_Forwards.ReleaseForward(g_pOnConfigsExecuted);
dvander@978: }
dvander@978: 
dvander@978: void CoreConfig::OnSourceModLevelChange(const char *mapName)
dvander@978: {
dvander@978: 	g_bConfigsExecd = false;
ds@661: }
ds@661: 
ds@661: void CoreConfig::OnRootConsoleCommand(const char *command, unsigned int argcount)
ds@661: {
ds@661: 	if (argcount >= 4)
ds@661: 	{
ds@661: 		const char *option = engine->Cmd_Argv(2);
ds@661: 		const char *value = engine->Cmd_Argv(3);
ds@661: 
dvander@663: 		char error[255];
ds@661: 
ds@669: 		ConfigResult res = SetConfigOption(option, value, ConfigSource_Console, error, sizeof(error));
dvander@663: 
ds@669: 		if (res == ConfigResult_Reject)
ds@661: 		{
ds@669: 			g_RootMenu.ConsolePrint("[SM] Could not set config option \"%s\" to \"%s\" (%s)", option, value, error);
ds@669: 		} else if (res == ConfigResult_Ignore) {
ds@669: 			g_RootMenu.ConsolePrint("[SM] No such config option \"%s\" exists.", option);
dvander@665: 		} else {
dvander@665: 			g_RootMenu.ConsolePrint("Config option \"%s\" successfully set to \"%s.\"", option, value);
ds@661: 		}
ds@661: 
ds@661: 		return;
ds@661: 	}
ds@661: 
ds@661: 	g_RootMenu.ConsolePrint("[SM] Usage: sm config <option> <value>");
ds@661: }
ds@661: 
ds@661: void CoreConfig::Initialize()
ds@661: {
ds@661: 	SMCParseError err;
ds@661: 	char filePath[PLATFORM_MAX_PATH];
ds@661: 
ds@661: 	/* Try to get command line value of core config convar */
ds@661: 	const char *corecfg = icvar->GetCommandLineValue("sm_corecfgfile");
ds@661: 
ds@661: 	/* If sm_corecfgfile not specified on command line, use default value */
ds@661: 	if (!corecfg)
ds@661: 	{
ds@661: 		corecfg = sm_corecfgfile.GetDefault();
ds@661: 	}
ds@661: 
ds@661: 	/* Format path to config file */
ds@729: 	g_LibSys.PathFormat(filePath, sizeof(filePath), "%s/%s", g_SourceMod.GetGamePath(), corecfg);
ds@661: 
ds@661: 	/* Parse config file */
dvander@1406: 	if ((err=textparsers->ParseFile_SMC(filePath, this, NULL, NULL)) != SMCParse_Okay)
ds@661: 	{
dvander@665:  		/* :TODO: This won't actually log or print anything :( - So fix that somehow */
dvander@1406: 		const char *error = textparsers->GetSMCErrorString(err);
dvander@663: 		g_Logger.LogFatal("[SM] Error encountered parsing core config file: %s", error ? error : "");
ds@661: 	}
ds@661: }
ds@661: 
ds@661: SMCParseResult CoreConfig::ReadSMC_KeyValue(const char *key, const char *value, bool key_quotes, bool value_quotes)
ds@661: {
dvander@663: 	char error[255];
dvander@663: 	ConfigResult err = SetConfigOption(key, value, ConfigSource_File, error, sizeof(error));
ds@661: 
dvander@663: 	if (err == ConfigResult_Reject)
ds@661: 	{
dvander@663: 		/* This is a fatal error */
dvander@665: 		g_Logger.LogFatal("Config error (key: %s) (value: %s) %s", key, value, error);
ds@661: 	}
ds@661: 
ds@661: 	return SMCParse_Continue;
ds@661: }
ds@661: 
dvander@663: ConfigResult CoreConfig::SetConfigOption(const char *option, const char *value, ConfigSource source, char *error, size_t maxlength)
ds@661: {
dvander@663: 	ConfigResult result;
ds@661: 
ds@661: 	/* Notify! */
ds@661: 	SMGlobalClass *pBase = SMGlobalClass::head;
ds@661: 	while (pBase)
ds@661: 	{
dvander@663: 		if ((result = pBase->OnSourceModConfigChanged(option, value, source, error, maxlength)) != ConfigResult_Ignore)
ds@661: 		{
dvander@663: 			return result;
ds@661: 		}
ds@661: 		pBase = pBase->m_pGlobalClassNext;
ds@661: 	}
ds@661: 
dvander@663: 	return ConfigResult_Ignore;
ds@661: }
dvander@978: 
dvander@978: bool SM_AreConfigsExecuted()
dvander@978: {
dvander@978: 	return g_bConfigsExecd;
dvander@978: }
dvander@978: 
dvander@978: inline bool IsPathSepChar(char c)
dvander@978: {
dvander@978: #if defined PLATFORM_WINDOWS
dvander@978: 	return (c == '\\' || c == '/');
dvander@978: #elif defined PLATFORM_LINUX
dvander@978: 	return (c == '/');
dvander@978: #endif
dvander@978: }
dvander@978: 
dvander@978: bool SM_ExecuteConfig(CPlugin *pl, AutoConfig *cfg, bool can_create)
dvander@978: {
dvander@978: 	bool will_create = false;
dvander@978: 	
dvander@978: 	/* See if we should be creating */
dvander@978: 	if (can_create && cfg->create)
dvander@978: 	{
dvander@978: 		will_create = true;
dvander@978: 		
dvander@978: 		/* If the folder does not exist, attempt to create it.  
dvander@978: 		 * We're awfully nice.
dvander@978: 		 */
dvander@978: 		const char *folder = cfg->folder.c_str();
dvander@978: 		char path[PLATFORM_MAX_PATH];
dvander@978: 		char build[PLATFORM_MAX_PATH];
dvander@978: 
dvander@978: 		g_SourceMod.BuildPath(Path_Game, path, sizeof(path), "cfg/%s", folder);
dvander@978: 
dvander@978: 		if (!g_LibSys.IsPathDirectory(path))
dvander@978: 		{
dvander@978: 			char *cur_ptr = path;
dvander@978: 			size_t len;
dvander@978: 			
dvander@978: 			g_LibSys.PathFormat(path, sizeof(path), "%s", folder);
dvander@978: 			len = g_SourceMod.BuildPath(Path_Game, build, sizeof(build), "cfg");
dvander@978: 
dvander@978: 			do
dvander@978: 			{
dvander@978: 				/* Find next suitable path */
dvander@978: 				char *next_ptr = cur_ptr;
dvander@978: 				while (*next_ptr != '\0')
dvander@978: 				{
dvander@978: 					if (IsPathSepChar(*next_ptr))
dvander@978: 					{
dvander@978: 						*next_ptr = '\0';
dvander@978: 						next_ptr++;
dvander@978: 						break;
dvander@978: 					}
dvander@978: 					next_ptr++;
dvander@978: 				}
dvander@978: 				if (*next_ptr == '\0')
dvander@978: 				{
dvander@978: 					next_ptr = NULL;
dvander@978: 				}
dvander@978: 				len += g_LibSys.PathFormat(&build[len], 
dvander@978: 					sizeof(build)-len,
dvander@978: 					"/%s",
dvander@978: 					cur_ptr);
dvander@1192: 				if (!g_LibSys.CreateFolder(build))
dvander@978: 				{
dvander@978: 					break;
dvander@978: 				}
dvander@978: 				cur_ptr = next_ptr;
dvander@978: 			} while (cur_ptr);
dvander@978: 		}
dvander@978: 	}
dvander@978: 
dvander@978: 	/* Check if the file exists. */
dvander@978: 	char file[PLATFORM_MAX_PATH];
dvander@978: 	char local[PLATFORM_MAX_PATH];
dvander@978: 
dvander@978: 	if (cfg->folder.size())
dvander@978: 	{
dvander@978: 		g_LibSys.PathFormat(local, 
dvander@978: 			sizeof(local), 
dvander@978: 			"%s/%s.cfg",
dvander@978: 			cfg->folder.c_str(),
dvander@978: 			cfg->autocfg.c_str());
dvander@978: 	} else {
dvander@978: 		g_LibSys.PathFormat(local,
dvander@978: 			sizeof(local),
dvander@978: 			"%s.cfg",
dvander@978: 			cfg->autocfg.c_str());
dvander@978: 	}
dvander@978: 	
dvander@978: 	g_SourceMod.BuildPath(Path_Game, file, sizeof(file), "cfg/%s", local);
dvander@978: 
dvander@978: 	bool file_exists = g_LibSys.IsPathFile(file);
dvander@978: 	if (!file_exists && will_create)
dvander@978: 	{
dvander@978: 		List<const ConVar *> *convars = NULL;
dvander@978: 		if (pl->GetProperty("ConVarList", (void **)&convars, false) && convars)
dvander@978: 		{
dvander@978: 			/* Attempt to create it */
dvander@978: 			FILE *fp = fopen(file, "wt");
dvander@978: 			if (fp)
dvander@978: 			{
dvander@978: 				fprintf(fp, "// This file was auto-generated by SourceMod (v%s)\n", SVN_FULL_VERSION);
dvander@978: 				fprintf(fp, "// ConVars for plugin \"%s\"\n", pl->GetFilename());
dvander@978: 				fprintf(fp, "\n\n");
dvander@978: 
dvander@978: 				List<const ConVar *>::iterator iter;
dvander@978: 				float x;
dvander@978: 				for (iter = convars->begin(); iter != convars->end(); iter++)
dvander@978: 				{
ds@1020: 					const ConVar *cvar = (*iter);
dvander@1019: 
dvander@1019: 					if ((cvar->GetFlags() & FCVAR_DONTRECORD) == FCVAR_DONTRECORD)
dvander@1019: 					{
dvander@1019: 						continue;
dvander@1019: 					}
dvander@1019: 
dvander@978: 					char descr[255];
dvander@978: 					char *dptr = descr;
dvander@978: 
dvander@978: 					/* Print comments until there is no more */
dvander@978: 					strncopy(descr, cvar->GetHelpText(), sizeof(descr));
dvander@978: 					while (*dptr != '\0')
dvander@978: 					{
dvander@978: 						/* Find the next line */
dvander@978: 						char *next_ptr = dptr;
dvander@978: 						while (*next_ptr != '\0')
dvander@978: 						{
dvander@978: 							if (*next_ptr == '\n')
dvander@978: 							{
dvander@978: 								*next_ptr = '\0';
dvander@978: 								next_ptr++;
dvander@978: 								break;
dvander@978: 							}
dvander@978: 							next_ptr++;
dvander@978: 						}
dvander@978: 						fprintf(fp, "// %s\n", dptr);
dvander@978: 						dptr = next_ptr;
dvander@978: 					}
dvander@978: 
dvander@978: 					fprintf(fp, "// -\n");
dvander@978: 					fprintf(fp, "// Default: \"%s\"\n", cvar->GetDefault());
dvander@978: 					if (cvar->GetMin(x))
dvander@978: 					{
dvander@978: 						fprintf(fp, "// Minimum: \"%02f\"\n", x);
dvander@978: 					}
dvander@978: 					if (cvar->GetMax(x))
dvander@978: 					{
dvander@978: 						fprintf(fp, "// Maximum: \"%02f\"\n", x);
dvander@978: 					}
dvander@978: 					fprintf(fp, "%s \"%s\"\n", cvar->GetName(), cvar->GetDefault());
dvander@978: 					fprintf(fp, "\n");
dvander@978: 				}
dvander@978: 				fprintf(fp, "\n");
dvander@978: 
dvander@978: 				file_exists = true;
dvander@978: 				can_create = false;
dvander@978: 				fclose(fp);
dvander@978: 			}
dvander@978: 		}
dvander@978: 	}
dvander@978: 
dvander@978: 	if (file_exists)
dvander@978: 	{
dvander@978: 		char cmd[255];
dvander@978: 		UTIL_Format(cmd, sizeof(cmd), "exec %s\n", local);
dvander@978: 		engine->ServerCommand(cmd);
dvander@978: 	}
dvander@978: 
dvander@978: 	return can_create;
dvander@978: }
dvander@978: 
dvander@978: void SM_DoSingleExecFwds(IPluginContext *ctx)
dvander@978: {
dvander@978: 	IPluginFunction *pf;
dvander@978: 
dvander@978: 	if ((pf = ctx->GetFunctionByName("OnServerCfg")) != NULL)
dvander@978: 	{
dvander@978: 		pf->Execute(NULL);
dvander@978: 	}
dvander@978: 
dvander@978: 	if ((pf = ctx->GetFunctionByName("OnConfigsExecuted")) != NULL)
dvander@978: 	{
dvander@978: 		pf->Execute(NULL);
dvander@978: 	}
dvander@978: }
dvander@978: 
dvander@978: void SM_ConfigsExecuted_Plugin(unsigned int serial)
dvander@978: {
dvander@978: 	IPluginIterator *iter = g_PluginSys.GetPluginIterator();
dvander@978: 	while (iter->MorePlugins())
dvander@978: 	{
dvander@978: 		CPlugin *plugin = (CPlugin *)(iter->GetPlugin());
dvander@978: 		if (plugin->GetSerial() == serial)
dvander@978: 		{
dvander@978: 			SM_DoSingleExecFwds(plugin->GetBaseContext());
dvander@978: 			break;
dvander@978: 		}
dvander@978: 		iter->NextPlugin();
dvander@978: 	}
dvander@978: 	iter->Release();
dvander@978: }
dvander@978: 
dvander@978: void SM_ExecuteForPlugin(IPluginContext *ctx)
dvander@978: {
dvander@978: 	CPlugin *plugin = (CPlugin *)g_PluginSys.GetPluginByCtx(ctx->GetContext());
dvander@978: 
dvander@978: 	unsigned int num = plugin->GetConfigCount();
dvander@978: 	if (!num)
dvander@978: 	{
dvander@978: 		SM_DoSingleExecFwds(ctx);
dvander@978: 	} else {
dvander@978: 		bool can_create = true;
dvander@978: 		for (unsigned int i=0; i<num; i++)
dvander@978: 		{
dvander@978: 			can_create = SM_ExecuteConfig(plugin, plugin->GetConfig(i), can_create);
dvander@978: 		}
dvander@978: 		char cmd[255];
dvander@978: 		UTIL_Format(cmd, sizeof(cmd), "sm internal 2 %d\n", plugin->GetSerial());
dvander@978: 		engine->ServerCommand(cmd);
dvander@978: 	}
dvander@978: }
dvander@978: 
dvander@978: void SM_ExecuteAllConfigs()
dvander@978: {
dvander@978: 	if (g_bConfigsExecd)
dvander@978: 	{
dvander@978: 		return;
dvander@978: 	}
dvander@978: 
dvander@1021: 	engine->ServerCommand("exec sourcemod/sourcemod.cfg\n");
dvander@978: 
dvander@978: 	IPluginIterator *iter = g_PluginSys.GetPluginIterator();
dvander@978: 	while (iter->MorePlugins())
dvander@978: 	{
dvander@978: 		CPlugin *plugin = (CPlugin *)(iter->GetPlugin());
dvander@978: 		unsigned int num = plugin->GetConfigCount();
dvander@978: 		bool can_create = true;
dvander@978: 		for (unsigned int i=0; i<num; i++)
dvander@978: 		{
dvander@978: 			can_create = SM_ExecuteConfig(plugin, plugin->GetConfig(i), can_create);
dvander@978: 		}
dvander@978: 		iter->NextPlugin();
dvander@978: 	}
dvander@978: 	iter->Release();
dvander@978: 
dvander@978: 	engine->ServerCommand("sm internal 1\n");
dvander@978: }
dvander@978: 
dvander@978: void SM_ConfigsExecuted_Global()
dvander@978: {
dvander@978: 	if (g_bConfigsExecd)
dvander@978: 	{
dvander@978: 		return;
dvander@978: 	}
dvander@978: 
dvander@978: 	g_bConfigsExecd = true;
dvander@978: 
dvander@978: 	g_pOnServerCfg->Execute(NULL);
dvander@978: 	g_pOnConfigsExecuted->Execute(NULL);
dvander@978: }