dvander@370: /**
ds@653:  * vim: set ts=4 :
ds@1330:  * =============================================================================
dvander@1231:  * SourceMod
dvander@1231:  * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.
ds@1330:  * =============================================================================
dvander@370:  *
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>.
dvander@370:  *
dvander@370:  * Version: $Id$
dvander@370:  */
dvander@370: 
dvander@340: #include "sm_globals.h"
dvander@1406: #include <ITextParsers.h>
dvander@340: #include "HandleSys.h"
dvander@340: 
dvander@340: HandleType_t g_TypeSMC = 0;
dvander@340: 
dvander@340: class ParseInfo : public ITextListener_SMC
dvander@340: {
dvander@340: public:
dvander@340: 	ParseInfo()
dvander@340: 	{
dvander@340: 		parse_start = NULL;
dvander@340: 		parse_end = NULL;
dvander@340: 		new_section = NULL;
dvander@340: 		key_value = NULL;
dvander@340: 		end_section = NULL;
dvander@340: 		raw_line = NULL;
dvander@340: 		handle = 0;
dvander@340: 	}
dvander@340: public:
dvander@340: 	void ReadSMC_ParseStart()
dvander@340: 	{
dvander@340: 		if (parse_start)
dvander@340: 		{
dvander@340: 			cell_t result;
dvander@340: 			parse_start->PushCell(handle);
dvander@340: 			parse_start->Execute(&result);
dvander@340: 		}
dvander@340: 	}
dvander@340: 
dvander@340: 	void ReadSMC_ParseEnd(bool halted, bool failed)
dvander@340: 	{
dvander@340: 		if (parse_end)
dvander@340: 		{
dvander@340: 			cell_t result;
dvander@747: 			parse_end->PushCell(handle);
dvander@340: 			parse_end->PushCell(halted ? 1 : 0);
dvander@340: 			parse_end->PushCell(failed ? 1 : 0);
dvander@340: 			parse_end->Execute(&result);
dvander@340: 		}
dvander@340: 	}
dvander@340: 
dvander@340: 	SMCParseResult ReadSMC_NewSection(const char *name, bool opt_quotes)
dvander@340: 	{
dvander@340: 		cell_t result = SMCParse_Continue;
dvander@340: 
dvander@340: 		if (new_section)
dvander@340: 		{
dvander@340: 			new_section->PushCell(handle);
dvander@340: 			new_section->PushString(name);
dvander@340: 			new_section->PushCell(opt_quotes ? 1 : 0);
dvander@340: 			new_section->Execute(&result);
dvander@340: 		}
dvander@340: 
dvander@340: 		return (SMCParseResult)result;
dvander@340: 	}
dvander@340: 
dvander@340: 	SMCParseResult ReadSMC_KeyValue(const char *key, const char *value, bool key_quotes, bool value_quotes)
dvander@340: 	{
dvander@340: 		cell_t result = SMCParse_Continue;
dvander@340: 
dvander@340: 		if (key_value)
dvander@340: 		{
dvander@340: 			key_value->PushCell(handle);
dvander@340: 			key_value->PushString(key);
dvander@340: 			key_value->PushString(value);
dvander@340: 			key_value->PushCell(key_quotes ? 1 : 0);
dvander@340: 			key_value->PushCell(value_quotes ? 1 : 0);
dvander@340: 			key_value->Execute(&result);
dvander@340: 		}
dvander@340: 
dvander@340: 		return (SMCParseResult)result;
dvander@340: 	}
dvander@340: 
dvander@340: 	SMCParseResult ReadSMC_LeavingSection()
dvander@340: 	{
dvander@340: 		cell_t result = SMCParse_Continue;
dvander@340: 
dvander@340: 		if (end_section)
dvander@340: 		{
dvander@340: 			end_section->PushCell(handle);
dvander@340: 			end_section->Execute(&result);
dvander@340: 		}
dvander@340: 
dvander@340: 		return (SMCParseResult)result;
dvander@340: 	}
dvander@340: 
dvander@340: 	SMCParseResult ReadSMC_RawLine(const char *line, unsigned int curline)
dvander@340: 	{
dvander@340: 		cell_t result = SMCParse_Continue;
dvander@340: 
dvander@340: 		if (raw_line)
dvander@340: 		{
dvander@340: 			raw_line->PushCell(handle);
dvander@340: 			raw_line->PushString(line);
dvander@340: 			raw_line->PushCell(curline);
dvander@340: 			raw_line->Execute(&result);
dvander@340: 		}
dvander@340: 
dvander@340: 		return (SMCParseResult)result;
dvander@340: 	}
dvander@340: public:
dvander@340: 	IPluginFunction *parse_start;
dvander@340: 	IPluginFunction *parse_end;
dvander@340: 	IPluginFunction *new_section;
dvander@340: 	IPluginFunction *key_value;
dvander@340: 	IPluginFunction *end_section;
dvander@340: 	IPluginFunction *raw_line;
dvander@340: 	Handle_t handle;
dvander@340: };
dvander@340: 
dvander@340: class TextParseGlobals : 
dvander@340: 	public SMGlobalClass,
dvander@340: 	public IHandleTypeDispatch
dvander@340: {
dvander@340: public:
dvander@340: 	void OnSourceModAllInitialized()
dvander@340: 	{
dvander@340: 		HandleAccess sec;
dvander@340: 
dvander@340: 		/* These cannot be cloned, because they are locked to a specific plugin.
dvander@340: 		 * However, we let anyone read them because we don't care.
dvander@340: 		 */
dvander@340: 		g_HandleSys.InitAccessDefaults(NULL, &sec);
dvander@340: 		sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY;
dvander@340: 		sec.access[HandleAccess_Read] = 0;
dvander@340: 
dvander@340: 		g_TypeSMC = g_HandleSys.CreateType("SMCParser", this, 0, NULL, &sec, g_pCoreIdent, NULL);
dvander@340: 	}
dvander@340: 
dvander@340: 	void OnSourceModShutdown()
dvander@340: 	{
dvander@340: 		g_HandleSys.RemoveType(g_TypeSMC, g_pCoreIdent);
dvander@340: 	}
dvander@340: 
dvander@340: 	void OnHandleDestroy(HandleType_t type, void *object)
dvander@340: 	{
dvander@340: 		ParseInfo *parse = (ParseInfo *)object;
dvander@340: 		delete parse;
dvander@340: 	}
dvander@340: };
dvander@340: 
dvander@340: TextParseGlobals g_TextParseGlobals;
dvander@340: 
dvander@364: static cell_t SMC_CreateParser(IPluginContext *pContext, const cell_t *params)
dvander@340: {
dvander@340: 	ParseInfo *pInfo = new ParseInfo();
dvander@340: 
dvander@340: 	Handle_t hndl = g_HandleSys.CreateHandle(g_TypeSMC, pInfo, pContext->GetIdentity(), g_pCoreIdent, NULL);
dvander@340: 
dvander@340: 	/* Should never happen */
dvander@340: 	if (!hndl)
dvander@340: 	{
dvander@340: 		delete pInfo;
dvander@340: 		return 0;
dvander@340: 	}
dvander@340: 
dvander@340: 	pInfo->handle = hndl;
dvander@340: 
dvander@340: 	return hndl;
dvander@340: }
dvander@340: 
dvander@340: static cell_t SMC_SetParseStart(IPluginContext *pContext, const cell_t *params)
dvander@340: {
dvander@340: 	Handle_t hndl = (Handle_t)params[1];
dvander@340: 	HandleError err;
dvander@340: 	ParseInfo *parse;
dvander@340: 
dvander@340: 	if ((err=g_HandleSys.ReadHandle(hndl, g_TypeSMC, NULL, (void **)&parse))
dvander@340: 		!= HandleError_None)
dvander@340: 	{
dvander@340: 		return pContext->ThrowNativeError("Invalid SMC Parse Handle %x (error %d)", hndl, err);
dvander@340: 	}
dvander@340: 
dvander@340: 	parse->parse_start = pContext->GetFunctionById((funcid_t)params[2]);
dvander@340: 
dvander@340: 	return 1;
dvander@340: }
dvander@340: 
dvander@340: static cell_t SMC_SetParseEnd(IPluginContext *pContext, const cell_t *params)
dvander@340: {
dvander@340: 	Handle_t hndl = (Handle_t)params[1];
dvander@340: 	HandleError err;
dvander@340: 	ParseInfo *parse;
dvander@340: 
dvander@340: 	if ((err=g_HandleSys.ReadHandle(hndl, g_TypeSMC, NULL, (void **)&parse))
dvander@340: 		!= HandleError_None)
dvander@340: 	{
dvander@340: 		return pContext->ThrowNativeError("Invalid SMC Parse Handle %x (error %d)", hndl, err);
dvander@340: 	}
dvander@340: 
dvander@340: 	parse->parse_end = pContext->GetFunctionById((funcid_t)params[2]);
dvander@340: 
dvander@340: 	return 1;
dvander@340: }
dvander@340: 
dvander@340: static cell_t SMC_SetReaders(IPluginContext *pContext, const cell_t *params)
dvander@340: {
dvander@340: 	Handle_t hndl = (Handle_t)params[1];
dvander@340: 	HandleError err;
dvander@340: 	ParseInfo *parse;
dvander@340: 
dvander@340: 	if ((err=g_HandleSys.ReadHandle(hndl, g_TypeSMC, NULL, (void **)&parse))
dvander@340: 		!= HandleError_None)
dvander@340: 	{
dvander@340: 		return pContext->ThrowNativeError("Invalid SMC Parse Handle %x (error %d)", hndl, err);
dvander@340: 	}
dvander@340: 
dvander@340: 	parse->new_section = pContext->GetFunctionById((funcid_t)params[2]);
dvander@340: 	parse->key_value = pContext->GetFunctionById((funcid_t)params[3]);
dvander@340: 	parse->end_section = pContext->GetFunctionById((funcid_t)params[4]);
dvander@340: 
dvander@340: 	return 1;
dvander@340: }
dvander@340: 
dvander@340: static cell_t SMC_SetRawLine(IPluginContext *pContext, const cell_t *params)
dvander@340: {
dvander@340: 	Handle_t hndl = (Handle_t)params[1];
dvander@340: 	HandleError err;
dvander@340: 	ParseInfo *parse;
dvander@340: 
dvander@340: 	if ((err=g_HandleSys.ReadHandle(hndl, g_TypeSMC, NULL, (void **)&parse))
dvander@340: 		!= HandleError_None)
dvander@340: 	{
dvander@340: 		return pContext->ThrowNativeError("Invalid SMC Parse Handle %x (error %d)", hndl, err);
dvander@340: 	}
dvander@340: 
dvander@340: 	parse->raw_line = pContext->GetFunctionById((funcid_t)params[2]);
dvander@340: 
dvander@340: 	return 1;
dvander@340: }
dvander@340: 
dvander@340: static cell_t SMC_ParseFile(IPluginContext *pContext, const cell_t *params)
dvander@340: {
dvander@340: 	Handle_t hndl = (Handle_t)params[1];
dvander@340: 	HandleError err;
dvander@340: 	ParseInfo *parse;
dvander@340: 
dvander@340: 	if ((err=g_HandleSys.ReadHandle(hndl, g_TypeSMC, NULL, (void **)&parse))
dvander@340: 		!= HandleError_None)
dvander@340: 	{
dvander@340: 		return pContext->ThrowNativeError("Invalid SMC Parse Handle %x (error %d)", hndl, err);
dvander@340: 	}
dvander@340: 
dvander@340: 	char *file;
dvander@340: 	pContext->LocalToString(params[2], &file);
dvander@340: 
dvander@340: 	char path[PLATFORM_MAX_PATH];
dvander@340: 	g_SourceMod.BuildPath(Path_Game, path, sizeof(path), "%s", file);
dvander@340: 
dvander@340: 	unsigned int line = 0, col = 0;
dvander@1406: 	SMCParseError p_err = textparsers->ParseFile_SMC(path, parse, &line, &col);
dvander@340: 
dvander@340: 	cell_t *c_line, *c_col;
dvander@340: 	pContext->LocalToPhysAddr(params[3], &c_line);
dvander@340: 	pContext->LocalToPhysAddr(params[4], &c_col);
dvander@340: 
faluco@389: 	*c_line = line;
faluco@389: 	*c_col = col;
faluco@389: 
dvander@340: 	return (cell_t)p_err;
dvander@340: }
dvander@340: 
dvander@340: static cell_t SMC_GetErrorString(IPluginContext *pContext, const cell_t *params)
dvander@340: {
dvander@1406: 	const char *str = textparsers->GetSMCErrorString((SMCParseError)params[1]);
dvander@340: 
dvander@340: 	if (!str)
dvander@340: 	{
dvander@340: 		return 0;
dvander@340: 	}
dvander@340: 
dvander@340: 	pContext->StringToLocal(params[2], params[3], str);
dvander@340: 
dvander@340: 	return 1;
dvander@340: }
dvander@340: 
dvander@340: REGISTER_NATIVES(textNatives)
dvander@340: {
dvander@364: 	{"SMC_CreateParser",			SMC_CreateParser},
dvander@340: 	{"SMC_ParseFile",				SMC_ParseFile},
dvander@340: 	{"SMC_GetErrorString",			SMC_GetErrorString},
dvander@340: 	{"SMC_SetParseStart",			SMC_SetParseStart},
dvander@340: 	{"SMC_SetParseEnd",				SMC_SetParseEnd},
dvander@340: 	{"SMC_SetReaders",				SMC_SetReaders},
dvander@340: 	{"SMC_SetRawLine",				SMC_SetRawLine},
dvander@340: 	{NULL,							NULL},
dvander@340: };