#include "uox3.h"
#include "CJSMapping.h"
#include "cScript.h"
#include "scriptc.h"
#include "ssection.h"
#include "StringUtility.hpp"

CJSMapping *JSMapping = nullptr;
//o-----------------------------------------------------------------------------------------------o
//|	File		-	CJSMapping.cpp
//|	Date		-	Feb 7, 2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	JavaScript File Mapping
//o-----------------------------------------------------------------------------------------------o
//| Changes		-	Version History
//|					1.0			 		Feb 7, 2005
//|					Initial Implementation
//|
//|					1.1			 		Feb 8, 2005
//|					Added the ability to reload JS files on a per-script basis
//|
//|					1.2			 		Feb 28, 2005
//|					Added the ability to reload JS files on a per-section basis
//o-----------------------------------------------------------------------------------------------o

//o-----------------------------------------------------------------------------------------------o
//|	Class		-	CJSMapping
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Base global class that holds within it an array of JS Mapping Sections
//o-----------------------------------------------------------------------------------------------o
CJSMapping::CJSMapping()
{
	Console.print( "Loading JS Scripts\n" );
}
CJSMapping::~CJSMapping()
{
	Cleanup();
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	void ResetDefaults( void )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Resets all parameters of the CJSMapping class to default
//o-----------------------------------------------------------------------------------------------o
void CJSMapping::ResetDefaults( void )
{
	for( size_t i = SCPT_NORMAL; i < SCPT_COUNT; ++i )
	{
		mapSection[i] = new CJSMappingSection( static_cast<SCRIPTTYPE>(i) );
	}

	envokeByID		= new CEnvoke( "object" );
	envokeByType	= new CEnvoke( "type" );

	Parse();
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	void Cleanup( void )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Frees all memory used by CJSMapping
//o-----------------------------------------------------------------------------------------------o
void CJSMapping::Cleanup( void )
{
	for( size_t i = SCPT_NORMAL; i < SCPT_COUNT; ++i )
	{
		if( mapSection[1] != nullptr )
		{
			delete mapSection[i];
			mapSection[i] = nullptr;
		}
	}

	if( envokeByID != nullptr )
	{
		delete envokeByID;
		envokeByID = nullptr;
	}
	if( envokeByType != nullptr )
	{
		delete envokeByType;
		envokeByType = nullptr;
	}
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	void Reload( UI16 scriptID )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Reloads the JS Scripts
//o-----------------------------------------------------------------------------------------------o
void CJSMapping::Reload( UI16 scriptID )
{
	if( scriptID != 0xFFFF )
	{
		Console.print( oldstrutil::format("CMD: Attempting Reload of JavaScript (ScriptID %u)\n", scriptID) );
		for( size_t i = SCPT_NORMAL; i < SCPT_COUNT; ++i )
		{
			if( mapSection[i]->IsInMap( scriptID ) )
			{
				mapSection[i]->Reload( scriptID );
				return;
			}
		}
		Console.warning( oldstrutil::format("Unable to locate specified JavaScript in the map (ScriptID %u)", scriptID) );
	}
	else
	{
		Console.print( oldstrutil::format("CMD: Loading JSE Scripts... \n") );
		Cleanup();
		ResetDefaults();
		envokeByID->Parse();
		envokeByType->Parse();
	}
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	void Reload( SCRIPTTYPE sectionID )
//|	Date		-	2/28/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Reloads a specific section of the JS Scripts
//o-----------------------------------------------------------------------------------------------o
void CJSMapping::Reload( SCRIPTTYPE sectionID )
{
	Console.print( oldstrutil::format("CMD: Attempting Reload of JavaScript (SectionID %u)\n", static_cast<SI32>(sectionID) ));
	if( mapSection[sectionID] != nullptr )
	{
		delete mapSection[sectionID];
		mapSection[sectionID] =  new CJSMappingSection( sectionID );

		Parse( sectionID );
	}
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	void Parse( SCRIPTTYPE toParse )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Parses through jse_fileassociations.scp doling out the work
//|						to each CJSMappingSection Parse() routine.
//o-----------------------------------------------------------------------------------------------o
void CJSMapping::Parse( SCRIPTTYPE toParse )
{
	std::string scpFileName = cwmWorldState->ServerData()->Directory( CSDDP_SCRIPTS ) + "jse_fileassociations.scp";
	if( !FileExists( scpFileName ) )
	{
		Console.error( oldstrutil::format("Failed to open %s", scpFileName.c_str() ));
		return;
	}

	Script *fileAssocData	= new Script( scpFileName, NUM_DEFS, false );
	if( fileAssocData != nullptr )
	{
		if( toParse != SCPT_COUNT )
		{
			if( mapSection[toParse] != nullptr )
				mapSection[toParse]->Parse( fileAssocData );
		}
		else
		{
			for( size_t i = SCPT_NORMAL; i < SCPT_COUNT; ++i )
			{
				if( mapSection[i] != nullptr )
					mapSection[i]->Parse( fileAssocData );
			}
		}

		delete fileAssocData;
	}
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	CJSMappingSection * GetSection( SCRIPTTYPE toGet )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Returns a pointer to the specified JSMappingSection
//o-----------------------------------------------------------------------------------------------o
CJSMappingSection * CJSMapping::GetSection( SCRIPTTYPE toGet )
{
	if( mapSection[toGet] != nullptr )
		return mapSection[toGet];

	return nullptr;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	UI16 GetScriptID( JSObject *toFind )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Returns the scriptID relating to the specified JSObject
//o-----------------------------------------------------------------------------------------------o
UI16 CJSMapping::GetScriptID( JSObject *toFind )
{
	UI16 retVal		= 0xFFFF;

	for( size_t i = SCPT_NORMAL; i < SCPT_COUNT; ++i )
	{
		retVal = mapSection[i]->GetScriptID( toFind );
		if( retVal != 0xFFFF )
			break;
	}
	return retVal;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	cScript * GetScript( JSObject *toFind )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Returns the cScript relating to the specified JSObject
//o-----------------------------------------------------------------------------------------------o
cScript * CJSMapping::GetScript( JSObject *toFind )
{
	cScript *retVal		= nullptr;
	cScript *toCheck	= nullptr;

	for( size_t i = SCPT_NORMAL; i < SCPT_COUNT; ++i )
	{
		toCheck = mapSection[i]->GetScript( toFind );
		if( toCheck != nullptr )
		{
			retVal = toCheck;
			break;
		}
	}
	return retVal;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	cScript * GetScript( UI16 toFind )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Returns the cScript relating to the specified scriptID
//o-----------------------------------------------------------------------------------------------o
cScript * CJSMapping::GetScript( UI16 toFind )
{
	cScript *retVal		= nullptr;
	cScript *toCheck	= nullptr;

	for( size_t i = SCPT_NORMAL; i < SCPT_COUNT; ++i )
	{
		toCheck = mapSection[i]->GetScript( toFind );
		if( toCheck != nullptr )
		{
			retVal = toCheck;
			break;
		}
	}
	return retVal;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	CEnvoke * GetEnvokeByID( void )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Returns a pointer to the CEnvoke class handling EnvokeByID
//o-----------------------------------------------------------------------------------------------o
CEnvoke * CJSMapping::GetEnvokeByID( void )
{
	return envokeByID;
}
//o-----------------------------------------------------------------------------------------------o
//|	Function	-	CEnvoke * GetEnvokeByType( void )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Returns a pointer to the CEnvoke class handling EnvokeByType
//o-----------------------------------------------------------------------------------------------o
CEnvoke * CJSMapping::GetEnvokeByType( void )
{
	return envokeByType;
}

//o-----------------------------------------------------------------------------------------------o
//|	Class		-	CJSMappingSection( SCRIPTTYPE sT )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Class containing the specified section of jse_fileassociations.scp
//o-----------------------------------------------------------------------------------------------o
CJSMappingSection::CJSMappingSection( SCRIPTTYPE sT )
{
	scriptType = sT;

	scriptIDMap.clear();
	scriptJSMap.clear();

	scriptIDIter = scriptIDMap.end();
}
CJSMappingSection::~CJSMappingSection()
{

	for( std::map< UI16, cScript * >::const_iterator sIter = scriptIDMap.begin(); sIter != scriptIDMap.end(); ++sIter )
	{
		cScript *toDelete = sIter->second;
		if( toDelete != nullptr )
			delete toDelete;
	}

	scriptIDMap.clear();
	scriptJSMap.clear();
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	void Parse( Script *fileAssocData )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Parses through specified section of jse_fileassociations.scp
//|						populating the map with each entry.
//o-----------------------------------------------------------------------------------------------o
void CJSMappingSection::Parse( Script *fileAssocData )
{
	std::string basePath	= cwmWorldState->ServerData()->Directory( CSDDP_SCRIPTS );
	ScriptSection *mSection = fileAssocData->FindEntry( ScriptNames[scriptType] );
	UI08 runTime			= 0;

	if( scriptType == SCPT_CONSOLE )
		runTime = 1;

	if( mSection != nullptr )
	{
		UI16 scriptID = 0xFFFF;
		size_t i = 0;
		std::string data, fullPath;
		for( std::string tag = mSection->First(); !mSection->AtEnd(); tag = mSection->Next() )
		{
			scriptID	= static_cast<UI16>(std::stoul(tag, nullptr, 0));
			data		= mSection->GrabData();
			fullPath	= basePath + data;

			if( !FileExists( fullPath ) )
				Console.error( oldstrutil::format("SE mapping of %i to %s failed, file does not exist!", scriptID, data.c_str() ));
			else
			{
				try
				{
					cScript *toAdd = new cScript( fullPath, runTime );
					if( toAdd != nullptr )
					{
						scriptIDMap[scriptID]			= toAdd;
						scriptJSMap[toAdd->Object()]	= scriptID;
						++i;
					}
				}
				catch( std::runtime_error &e )
				{
					Console.error( oldstrutil::format("Compiling %s caused a construction failure (Details: %s)", fullPath.c_str(), e.what()) );
				}
			}
		}
		Console.print( "  o Loaded " );
		Console.TurnYellow();
		Console.print( oldstrutil::format("%4u ", i) );
		Console.TurnNormal();
		Console.print( "scripts from section " );
		Console.TurnYellow();
		Console.print( oldstrutil::format("%s\n", ScriptNames[scriptType].c_str()) );
		Console.TurnNormal();
	}
	else
		Console.warning( oldstrutil::format("No JS file mappings found in section %s", ScriptNames[scriptType].c_str()) );
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	void Reload( UI16 toLoad )
//|	Date		-	2/8/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Reloads the specified JS file (by its scriptID)
//o-----------------------------------------------------------------------------------------------o
void CJSMappingSection::Reload( UI16 toLoad )
{
	std::string scpFileName = cwmWorldState->ServerData()->Directory( CSDDP_SCRIPTS ) + "jse_fileassociations.scp";
	if( !FileExists( scpFileName ) )
	{
		Console.error( oldstrutil::format("Failed to open %s", scpFileName.c_str()) );
		return;
	}

	Script *fileAssocData	= new Script( scpFileName, NUM_DEFS, false );
	if( fileAssocData != nullptr )
	{
		ScriptSection *mSection = fileAssocData->FindEntry( ScriptNames[scriptType] );
		if( mSection != nullptr )
		{
			UI16 scriptID = 0xFFFF;
			std::string basePath = cwmWorldState->ServerData()->Directory( CSDDP_SCRIPTS );
			std::string data, fullPath;
			UI08 runTime = 0;
			if( scriptType == SCPT_CONSOLE )
				runTime = 1;
			for( std::string tag = mSection->First(); !mSection->AtEnd(); tag = mSection->Next() )
			{
				scriptID		= static_cast<UI16>(std::stoul(tag, nullptr, 0));
				if( scriptID == toLoad )
				{
					data		= mSection->GrabData();
					fullPath	= basePath + data;

					if( !FileExists( fullPath ) )
						Console.error( oldstrutil::format("SE mapping of %i to %s failed, file does not exist!", scriptID, data.c_str() ));
					else
					{
						try
						{
							std::map< UI16, cScript * >::const_iterator iFind = scriptIDMap.find( toLoad );
							if( iFind != scriptIDMap.end() )
							{
								if( scriptIDMap[toLoad] != nullptr )
								{
									JSObject *jsObj = scriptIDMap[toLoad]->Object();
									std::map< JSObject *, UI16 >::iterator jFind = scriptJSMap.find( jsObj );
									if( jFind != scriptJSMap.end() )
										scriptJSMap.erase( jFind );

									delete scriptIDMap[toLoad];
									scriptIDMap[toLoad] = nullptr;
								}
							}
							cScript *toAdd = new cScript( fullPath, runTime );
							if( toAdd != nullptr )
							{
								scriptIDMap[scriptID]			= toAdd;
								scriptJSMap[toAdd->Object()]	= scriptID;
								Console.print( oldstrutil::format("Reload of JavaScript (ScriptID %u) Successful\n", toLoad) );
							}
						}
						catch( std::runtime_error &e )
						{
							Console.error( oldstrutil::format("Compiling %s caused a construction failure (Details: %s)", fullPath.c_str(), e.what()) );
						}
					}
					delete fileAssocData;
					return;
				}
			}
			Console.warning(oldstrutil::format( "Unable to locate the specified JavaScript in the file (ScriptID %u)", toLoad) );
		}
		else
			Console.warning( oldstrutil::format("No JS file mappings found in section %s", ScriptNames[scriptType].c_str()) );
		delete fileAssocData;
	}
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	bool IsInMap( UI16 scriptID )
//|	Date		-	2/8/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Checks if the specified scriptID exists in the map
//o-----------------------------------------------------------------------------------------------o
bool CJSMappingSection::IsInMap( UI16 scriptID )
{
	bool retVal = false;

	std::map< UI16, cScript * >::const_iterator sIter = scriptIDMap.find( scriptID );
	if( sIter != scriptIDMap.end() )
		retVal = true;

	return retVal;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	UI16 GetScriptID( JSObject *toFind )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Finds the scriptID associated with the specified JSObject
//o-----------------------------------------------------------------------------------------------o
UI16 CJSMappingSection::GetScriptID( JSObject *toFind )
{
	UI16 retVal = 0xFFFF;

	std::map< JSObject *, UI16 >::const_iterator sIter = scriptJSMap.find( toFind );
	if( sIter != scriptJSMap.end() )
		retVal = sIter->second;

	return retVal;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	cScript * GetScript( UI16 toFind )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Finds the cScript associated with the specified scriptID
//o-----------------------------------------------------------------------------------------------o
cScript * CJSMappingSection::GetScript( UI16 toFind )
{
	cScript *retVal = nullptr;

	std::map< UI16, cScript * >::const_iterator idIter = scriptIDMap.find( toFind );
	if( idIter != scriptIDMap.end() )
		retVal = idIter->second;

	return retVal;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	cScript * GetScript( JSObject *toFind )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Finds the cScript associated with the specified JSObject
//o-----------------------------------------------------------------------------------------------o
cScript * CJSMappingSection::GetScript( JSObject *toFind )
{
	UI16 scriptID = GetScriptID( toFind );
	return GetScript( scriptID );
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	cScript * First( void )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Returns the first cScript in the scriptIDMap
//o-----------------------------------------------------------------------------------------------o
cScript * CJSMappingSection::First( void )
{
	scriptIDIter = scriptIDMap.begin();
	if( !Finished() )
		return scriptIDIter->second;

	return nullptr;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	cScript * Next( void )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Returns the next cScript in the scriptIDMap
//o-----------------------------------------------------------------------------------------------o
cScript * CJSMappingSection::Next( void )
{
	if( !Finished() )
	{
		++scriptIDIter;
		if( !Finished() )
			return scriptIDIter->second;
	}
	return nullptr;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	bool Finished( void )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Tells us when we have reached the end of the scriptIDMap
//o-----------------------------------------------------------------------------------------------o
bool CJSMappingSection::Finished( void )
{
	return (scriptIDIter == scriptIDMap.end());
}

//o-----------------------------------------------------------------------------------------------o
//|	Class		-	CEnvoke( const std::string &eT )
//|	Date		-	2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Class containing the specified envoke file
//o-----------------------------------------------------------------------------------------------o
CEnvoke::CEnvoke( const std::string &eT )
{
	envokeType = eT;
	envokeList.clear();
}
CEnvoke::~CEnvoke()
{
	envokeList.clear();
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	bool Check( UI16 envokeID ) const
//|	Date		-	2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Checks if specified envokeID is in the map
//o-----------------------------------------------------------------------------------------------o
bool CEnvoke::Check( UI16 envokeID ) const
{
	std::map< UI16, UI16 >::const_iterator p = envokeList.find( envokeID );
	return ( p != envokeList.end() );
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	UI16 GetScript( UI16 envokeID ) const
//|	Date		-	2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Returns specified envokeID
//o-----------------------------------------------------------------------------------------------o
UI16 CEnvoke::GetScript( UI16 envokeID ) const
{
	std::map< UI16, UI16 >::const_iterator p = envokeList.find( envokeID );
	if( p != envokeList.end() )
		return p->second;
	return 0xFFFF;
}

//o-----------------------------------------------------------------------------------------------o
//|	Function	-	void Parse( void )
//|	Date		-	2/7/2005
//o-----------------------------------------------------------------------------------------------o
//|	Purpose		-	Parses through specific envoke file mapping the scriptID to the envoke type
//o-----------------------------------------------------------------------------------------------o
void CEnvoke::Parse( void )
{
	envokeList.clear();

	std::string filename = cwmWorldState->ServerData()->Directory( CSDDP_SCRIPTS ) + "jse_" + envokeType + "associations.scp";
	if( !FileExists( filename ) )
	{
		Console << "Unable to open " << filename << " for parsing" << myendl;
		return;
	}

	Script *fileAssocData	= new Script( filename, NUM_DEFS, false );
	if( fileAssocData != nullptr )
	{
		ScriptSection *mSection = fileAssocData->FindEntry( "ENVOKE" );
		if( mSection != nullptr )
		{
			std::string tag, data;
			for( tag = mSection->First(); !mSection->AtEnd(); tag = mSection->Next() )
			{
				if( !tag.empty() && tag != "\n" && tag != "\r" )
				{
					data			= mSection->GrabData();
					UI16 envokeID	= static_cast<UI16>(std::stoul(tag, nullptr, 0));
					UI16 scriptID	= static_cast<UI16>(std::stoul(data, nullptr, 0));
					cScript *verify	= JSMapping->GetScript( scriptID );
					if( verify != nullptr )
						envokeList[envokeID] = scriptID;
					else
						Console.error(oldstrutil::format( "(ENVOKE) Item %s refers to scriptID %u which does not exist.", tag.c_str(), scriptID) );
				}
			}
		}
		else
			Console.warning( "Envoke section not found, no hard id->script matching being done" );
		delete fileAssocData;
	}
}