1 files changed, 423 insertions, 0 deletions

diff --git a/source/cPluginManager.cpp b/source/cPluginManager.cpp
new file mode 100644
index 000000000..db776f0a7
--- /dev/null
+++ b/source/cPluginManager.cpp
@@ -0,0 +1,423 @@
+#include "cPluginManager.h"

+#include "cPlugin.h"

+#include "cPlugin_Lua.h"

+#include "cMCLogger.h"

+#include "cWebAdmin.h"

+#include "cItem.h"

+#include "cRoot.h"

+#include "cLuaCommandBinder.h"

+

+#include <stdarg.h>

+

+#include "../iniFile/iniFile.h"

+

+extern std::vector<std::string> StringSplit(std::string str, std::string delim);

+

+typedef std::list< cPlugin_Lua* > LuaPluginList;

+typedef std::map< cPluginManager::PluginHook, cPluginManager::PluginList > HookMap;

+

+struct cPluginManager::sPluginManagerState

+{

+	LuaPluginList LuaPlugins;

+	cPluginManager::PluginList Plugins;

+	HookMap Hooks;

+};

+

+cPluginManager* cPluginManager::GetPluginManager()

+{

+	LOGWARN("WARNING: Using deprecated function cPluginManager::GetPluginManager() use cRoot::Get()->GetPluginManager() instead!");

+	return cRoot::Get()->GetPluginManager();

+}

+

+cPluginManager::cPluginManager()

+	: m_pState( new sPluginManagerState )

+	, m_LuaCommandBinder( new cLuaCommandBinder() )

+	, m_bReloadPlugins(false)

+{

+}

+

+cPluginManager::~cPluginManager()

+{

+	UnloadPluginsNow();

+	delete m_LuaCommandBinder;

+	delete m_pState;

+}

+

+void cPluginManager::ReloadPlugins()

+{

+	m_bReloadPlugins = true;

+}

+

+void cPluginManager::ReloadPluginsNow()

+{

+	LOG("--Loading plugins--");

+	m_bReloadPlugins = false;

+	UnloadPluginsNow();

+

+	cIniFile IniFile("settings.ini");

+	if( IniFile.ReadFile() )

+	{

+		unsigned int KeyNum = IniFile.FindKey("Plugins");

+		unsigned int NumPlugins = IniFile.GetNumValues( KeyNum );

+		if( NumPlugins > 0 )

+		{

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

+			{

+				std::string ValueName = IniFile.GetValueName(KeyNum, i );

+				if( ValueName.compare("Plugin") == 0 )

+				{	// It's a plugin

+					std::string PluginFile = IniFile.GetValue(KeyNum, i );

+					if( PluginFile.compare("") != 0 )

+					{

+						// allow for comma separated plugin list

+						// degrades and works fine for the plugin

+						// per line

+						std::vector< std::string > split = StringSplit( PluginFile, "," );

+						for (unsigned int j = 0; j < split.size(); j++) {

+							cPlugin_Lua* Plugin = new cPlugin_Lua( (split[j] + std::string(".lua") ).c_str() );

+							if( !AddLuaPlugin( Plugin ) )

+							{

+								delete Plugin;

+							}

+						}

+					}

+				}

+			}

+		}

+

+		if( GetNumPlugins() == 0 )

+		{

+			LOG("No plugins loaded");

+		}

+		else

+		{

+			LOG("Loaded %i plugin(s)", GetNumPlugins() );

+		}

+	}

+	else

+	{

+		LOG("WARNING: Can't find settings.ini, so can't load any plugins.");

+	}

+	LOG("--Done loading plugins--");

+}

+

+void cPluginManager::Tick(float a_Dt)

+{

+	if( m_bReloadPlugins )

+	{

+		ReloadPluginsNow();

+	}

+

+

+	HookMap::iterator Plugins = m_pState->Hooks.find( E_PLUGIN_TICK );

+	if( Plugins != m_pState->Hooks.end() )

+	{

+		for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+		{

+			(*itr)->Tick( a_Dt );

+		}

+	}

+}

+

+bool cPluginManager::CallHook( PluginHook a_Hook, unsigned int a_NumArgs, ... )

+{

+	HookMap::iterator Plugins = m_pState->Hooks.find( a_Hook );

+

+	// Special case for chat hook, since you can also bind commands (bound commands don't use chat hook)

+	if( a_Hook == E_PLUGIN_CHAT )

+	{

+		if( a_NumArgs != 2 ) return false;

+		va_list argptr;

+		va_start( argptr, a_NumArgs);

+		const char* Message = va_arg(argptr, const char* );

+		cPlayer* Player = va_arg(argptr, cPlayer* );

+		va_end (argptr);

+

+		if( m_LuaCommandBinder->HandleCommand( std::string( Message ), Player ) )

+			return true;

+

+		if( Plugins != m_pState->Hooks.end() )

+		{

+			for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+			{

+				if( (*itr)->OnChat( Message, Player ) )

+					return true;

+			}

+		}

+

+		return false;

+	}

+

+	if( Plugins != m_pState->Hooks.end() )

+	{

+		switch( a_Hook )

+		{

+		case E_PLUGIN_COLLECT_ITEM:

+			{

+				if( a_NumArgs != 2 ) break;

+				va_list argptr;

+				va_start( argptr, a_NumArgs);

+				cPickup* Pickup = va_arg(argptr, cPickup* );

+				cPlayer* Player = va_arg(argptr, cPlayer* );

+				va_end (argptr);

+				for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+				{

+					if( (*itr)->OnCollectItem( Pickup, Player ) )

+						return true;

+				}

+			}

+			break;

+		case E_PLUGIN_BLOCK_DIG:

+			{

+				if( a_NumArgs != 2 ) break;

+				va_list argptr;

+				va_start( argptr, a_NumArgs);

+				cPacket_BlockDig* Packet = va_arg(argptr, cPacket_BlockDig* );

+				cPlayer* Player = va_arg(argptr, cPlayer* );

+				cItem* Item = va_arg( argptr, cItem* );

+				va_end (argptr);

+				for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+				{

+					if( (*itr)->OnBlockDig( Packet, Player, Item ) )

+						return true;

+				}

+			}

+			break;

+		case E_PLUGIN_BLOCK_PLACE:

+			{

+				if( a_NumArgs != 2 ) break;

+				va_list argptr;

+				va_start( argptr, a_NumArgs);

+				cPacket_BlockPlace* Packet = va_arg(argptr, cPacket_BlockPlace* );

+				cPlayer* Player = va_arg(argptr, cPlayer* );

+				va_end (argptr);

+				for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+				{

+					if( (*itr)->OnBlockPlace( Packet, Player ) )

+						return true;

+				}

+			}

+			break;

+		case E_PLUGIN_DISCONNECT:

+			{

+				if( a_NumArgs != 2 ) break;

+				va_list argptr;

+				va_start( argptr, a_NumArgs);

+				const char* Reason = va_arg(argptr, const char* );

+				cPlayer* Player = va_arg(argptr, cPlayer* );

+				va_end (argptr);

+				for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+				{

+					if( (*itr)->OnDisconnect( Reason, Player ) )

+						return true;

+				}

+			}

+			break;

+		case E_PLUGIN_LOGIN:

+			{

+				if( a_NumArgs != 1 ) break;

+				va_list argptr;

+				va_start( argptr, a_NumArgs);

+				cPacket_Login* Packet = va_arg(argptr, cPacket_Login* );

+				va_end (argptr);

+				for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+				{

+					if( (*itr)->OnLogin( Packet ) )

+						return true;

+				}

+			}

+			break;

+		case E_PLUGIN_PLAYER_JOIN:

+			{

+				if( a_NumArgs != 1 ) break;

+				va_list argptr;

+				va_start( argptr, a_NumArgs);

+				cPlayer* Player = va_arg(argptr, cPlayer* );

+				va_end (argptr);

+				for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+				{

+					if( (*itr)->OnPlayerJoin( Player ) )

+						return true;

+				}

+			}

+			break;

+		case E_PLUGIN_PLAYER_MOVE:

+			{

+				if( a_NumArgs != 1 ) break;

+				va_list argptr;

+				va_start( argptr, a_NumArgs);

+				cPlayer* Player = va_arg(argptr, cPlayer* );

+				va_end (argptr);

+				for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+				{

+					(*itr)->OnPlayerMove( Player );

+				}

+			}

+			break;

+		case E_PLUGIN_TAKE_DAMAGE:

+			{

+				if( a_NumArgs != 2 ) break;

+				va_list argptr;

+				va_start( argptr, a_NumArgs);

+				cPawn* Pawn = va_arg(argptr, cPawn* );

+				TakeDamageInfo* TDI = va_arg(argptr, TakeDamageInfo* );

+				va_end (argptr);

+				for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+				{

+					(*itr)->OnTakeDamage( Pawn, TDI );

+				}

+			}

+			break;

+		case E_PLUGIN_KILLED:

+			{

+				if( a_NumArgs != 2 ) break;

+				va_list argptr;

+				va_start( argptr, a_NumArgs);

+				cPawn* Killed = va_arg(argptr, cPawn* );

+				cEntity* Killer = va_arg(argptr, cEntity* );

+				va_end (argptr);

+				for( PluginList::iterator itr = Plugins->second.begin(); itr != Plugins->second.end(); ++itr )

+				{

+					if( (*itr)->OnKilled( Killed, Killer ) )

+						return true;

+				}

+			}

+			break;

+		default:

+			LOG("WARNING: Calling Unknown hook: %i", a_Hook );

+			break;

+		}

+	}

+	return false;

+}

+

+cPlugin* cPluginManager::GetPlugin( std::string a_Plugin )

+{

+	for( PluginList::iterator itr = m_pState->Plugins.begin(); itr != m_pState->Plugins.end(); ++itr )

+	{

+		if( (*itr)->GetName().compare( a_Plugin ) == 0 )

+		{

+			return *itr;

+		}

+	}

+	return 0;

+}

+

+const cPluginManager::PluginList & cPluginManager::GetAllPlugins()

+{

+	return m_pState->Plugins;

+}

+

+void cPluginManager::UnloadPluginsNow()

+{

+	m_pState->Hooks.clear();

+

+	while( m_pState->LuaPlugins.size() > 0 )

+	{

+		cPlugin_Lua* LuaPlugin = *m_pState->LuaPlugins.begin();

+		if( LuaPlugin )

+		{

+			cWebAdmin* WebAdmin = cRoot::Get()->GetWebAdmin();

+			if( WebAdmin ) WebAdmin->RemovePlugin( LuaPlugin->GetLuaState() );

+			delete LuaPlugin;

+		}

+		m_pState->LuaPlugins.remove( LuaPlugin );

+	}

+

+	while( m_pState->Plugins.size() > 0 )

+	{

+		RemovePlugin( *m_pState->Plugins.begin(), true );

+	}

+}

+

+void cPluginManager::RemovePlugin( cPlugin* a_Plugin, bool a_bDelete /* = false */ )

+{

+	if( a_bDelete )

+	{

+		m_LuaCommandBinder->RemoveBindingsForPlugin( a_Plugin );

+		m_pState->Plugins.remove( a_Plugin );

+		a_Plugin->OnDisable();

+		delete a_Plugin;

+	}

+	else

+	{

+		for( LuaPluginList::iterator itr = m_pState->LuaPlugins.begin(); itr != m_pState->LuaPlugins.end(); ++itr )

+		{

+			(*itr)->RemovePlugin( a_Plugin );

+		}

+	}

+}

+

+bool cPluginManager::AddPlugin( cPlugin* a_Plugin )

+{

+	if( a_Plugin->Initialize() )

+	{

+		m_pState->Plugins.remove( a_Plugin );

+		m_pState->Plugins.push_back( a_Plugin );

+		return true;

+	}

+	return false;

+}

+

+bool cPluginManager::AddPlugin( lua_State* a_LuaState, cPlugin* a_Plugin )

+{

+	cPlugin_Lua* LuaPlugin = GetLuaPlugin( a_LuaState );

+	if( LuaPlugin && a_Plugin->Initialize() )

+	{

+		m_pState->Plugins.remove( a_Plugin );

+		m_pState->Plugins.push_back( a_Plugin );

+		LuaPlugin->AddPlugin( a_Plugin );

+		return true;

+	}

+	return false;

+}

+

+bool cPluginManager::AddLuaPlugin( cPlugin_Lua* a_Plugin )

+{

+	m_pState->LuaPlugins.push_back( a_Plugin ); // It HAS to be in here before calling Initialize, so it can be found by AddPlugin()

+	if(a_Plugin->Initialize() )

+	{

+		return true;

+	}

+	LOG(">>>>>>> Could not initialize a plugin! ");

+	m_pState->LuaPlugins.remove( a_Plugin );

+	return false;

+}

+

+void cPluginManager::RemoveLuaPlugin( std::string a_FileName )

+{

+	for( LuaPluginList::iterator itr = m_pState->LuaPlugins.begin(); itr != m_pState->LuaPlugins.end(); ++itr )

+	{

+		if( (*itr)->GetFileName() == a_FileName )

+		{

+			cPlugin_Lua* Plugin = *itr;

+			delete Plugin;

+			m_pState->LuaPlugins.remove( Plugin );

+			return;

+		}

+	}

+}

+

+cPlugin_Lua* cPluginManager::GetLuaPlugin( lua_State* a_State )

+{

+	for( LuaPluginList::iterator itr = m_pState->LuaPlugins.begin(); itr != m_pState->LuaPlugins.end(); ++itr )

+	{

+		if( (*itr)->GetLuaState() == a_State )

+		{

+			return *itr;

+		}

+	}

+	return 0;

+}

+

+void cPluginManager::AddHook( cPlugin* a_Plugin, PluginHook a_Hook )

+{

+	PluginList & Plugins = m_pState->Hooks[ a_Hook ];

+	Plugins.remove( a_Plugin );

+	Plugins.push_back( a_Plugin );

+}

+

+unsigned int cPluginManager::GetNumPlugins()

+{

+	return m_pState->Plugins.size(); 

+}
\ No newline at end of file