diff options
Diffstat (limited to 'source/cLuaCommandBinder.cpp')
| -rw-r--r-- | source/cLuaCommandBinder.cpp | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/source/cLuaCommandBinder.cpp b/source/cLuaCommandBinder.cpp new file mode 100644 index 000000000..a800513e1 --- /dev/null +++ b/source/cLuaCommandBinder.cpp @@ -0,0 +1,106 @@ +#include "cLuaCommandBinder.h"
+#include "cMCLogger.h"
+#include "cPlayer.h"
+#include "cPlugin_Lua.h"
+
+#include "tolua++.h"
+
+extern std::vector<std::string> StringSplit(std::string str, std::string delim);
+extern bool report_errors(lua_State* lua, int status);
+
+cLuaCommandBinder::cLuaCommandBinder()
+{
+}
+
+cLuaCommandBinder::~cLuaCommandBinder()
+{
+}
+
+void cLuaCommandBinder::ClearBindings()
+{
+ m_BoundCommands.clear();
+}
+
+void cLuaCommandBinder::RemoveBindingsForPlugin( cPlugin* a_Plugin )
+{
+ for( CommandMap::iterator itr = m_BoundCommands.begin(); itr != m_BoundCommands.end(); )
+ {
+ if( itr->second.Plugin == a_Plugin )
+ {
+ LOGINFO("Unbinding %s ", itr->first.c_str( ) );
+ luaL_unref( itr->second.LuaState, LUA_REGISTRYINDEX, itr->second.Reference ); // unreference
+ CommandMap::iterator eraseme = itr;
+ ++itr;
+ m_BoundCommands.erase( eraseme );
+ continue;
+ }
+ ++itr;
+ }
+}
+
+bool cLuaCommandBinder::BindCommand( const std::string & a_Command, const std::string & a_Permission, cPlugin* a_Plugin, lua_State * a_LuaState, int a_FunctionReference )
+{
+ if( m_BoundCommands.find( a_Command ) != m_BoundCommands.end() )
+ {
+ LOGERROR("ERROR: Trying to bind command \"%s\" that has already been bound.", a_Command.c_str() );
+ return false;
+ }
+ LOGINFO("Binding %s (%s)", a_Command.c_str(), a_Permission.c_str() );
+ m_BoundCommands[ a_Command ] = BoundFunction( a_Plugin, a_LuaState, a_FunctionReference, a_Permission );
+ return true;
+}
+
+bool cLuaCommandBinder::HandleCommand( const std::string & a_Command, cPlayer* a_Player )
+{
+ std::vector<std::string> Split = StringSplit( a_Command, " ");
+ CommandMap::iterator FoundCommand = m_BoundCommands.find( Split[0] );
+ if( FoundCommand != m_BoundCommands.end() )
+ {
+ const BoundFunction & func = FoundCommand->second;
+ if( func.Permission.size() > 0 )
+ {
+ if( !a_Player->HasPermission( func.Permission.c_str() ) )
+ {
+ return false;
+ }
+ }
+
+ // For enabling 'self' in the function, it's kind of a hack I'm not sure this is the way to go
+ lua_pushvalue(func.LuaState, LUA_GLOBALSINDEX);
+ lua_pushstring(func.LuaState, "self");
+ tolua_pushusertype( func.LuaState, func.Plugin, "cPlugin" );
+ lua_rawset(func.LuaState, -3);
+ lua_pop(func.LuaState, 1);
+
+ LOGINFO("1. Stack size: %i", lua_gettop(func.LuaState) );
+ lua_rawgeti( func.LuaState, LUA_REGISTRYINDEX, func.Reference); // same as lua_getref()
+
+ // Push the split
+ LOGINFO("2. Stack size: %i", lua_gettop(func.LuaState) );
+ lua_createtable(func.LuaState, Split.size(), 0);
+ int newTable = lua_gettop(func.LuaState);
+ int index = 1;
+ std::vector<std::string>::const_iterator iter = Split.begin();
+ while(iter != Split.end()) {
+ tolua_pushstring( func.LuaState, (*iter).c_str() );
+ lua_rawseti(func.LuaState, newTable, index);
+ ++iter;
+ ++index;
+ }
+ LOGINFO("3. Stack size: %i", lua_gettop(func.LuaState) );
+ // Push player
+ tolua_pushusertype( func.LuaState, a_Player, "cPlayer" );
+ LOGINFO("Calling bound function! :D");
+ int s = lua_pcall(func.LuaState, 2, 1, 0);
+ if( report_errors( func.LuaState, s ) )
+ {
+ LOGINFO("error. Stack size: %i", lua_gettop(func.LuaState) );
+ return false;
+ }
+ bool RetVal = (tolua_toboolean(func.LuaState, -1, 0) > 0);
+ lua_pop(func.LuaState, 1); // Pop return value
+ LOGINFO("ok. Stack size: %i", lua_gettop(func.LuaState) );
+ return RetVal;
+ }
+ return false;
+}
|