518 lines
13 KiB
C++
518 lines
13 KiB
C++
|
|
#include "Globals.h" // NOTE: MSVC stupidness requires this to be the same across all modules
|
|
|
|
#include "ManualBindings.h"
|
|
#include "tolua++.h"
|
|
|
|
#include "cRoot.h"
|
|
#include "cWorld.h"
|
|
#include "cPlugin.h"
|
|
#include "cPluginManager.h"
|
|
#include "cWebPlugin_Lua.h"
|
|
#include "cLuaCommandBinder.h"
|
|
#include "cPlayer.h"
|
|
#include "cWebAdmin.h"
|
|
#include "cStringMap.h"
|
|
#include "md5/md5.h"
|
|
|
|
|
|
static bool report_errors(lua_State* lua, int status)
|
|
{
|
|
if ( status!=0 )
|
|
{
|
|
std::string s = lua_tostring(lua, -1);
|
|
LOGERROR("-- %s", s.c_str() );
|
|
lua_pop(lua, 1);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
|
|
/****************************
|
|
* Lua bound functions with special return types
|
|
**/
|
|
|
|
static int tolua_StringSplit(lua_State* tolua_S)
|
|
{
|
|
std::string str = ((std::string) tolua_tocppstring(tolua_S,1,0));
|
|
std::string delim = ((std::string) tolua_tocppstring(tolua_S,2,0));
|
|
|
|
AStringVector Split = StringSplit( str, delim );
|
|
|
|
lua_createtable(tolua_S, Split.size(), 0);
|
|
int newTable = lua_gettop(tolua_S);
|
|
int index = 1;
|
|
std::vector<std::string>::const_iterator iter = Split.begin();
|
|
while(iter != Split.end()) {
|
|
tolua_pushstring( tolua_S, (*iter).c_str() );
|
|
lua_rawseti(tolua_S, newTable, index);
|
|
++iter;
|
|
++index;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_LOG(lua_State* tolua_S)
|
|
{
|
|
const char* str = tolua_tocppstring(tolua_S,1,0);
|
|
cMCLogger::GetInstance()->LogSimple( str, 0 );
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_LOGINFO(lua_State* tolua_S)
|
|
{
|
|
const char* str = tolua_tocppstring(tolua_S,1,0);
|
|
cMCLogger::GetInstance()->LogSimple( str, 1 );
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_LOGWARN(lua_State* tolua_S)
|
|
{
|
|
const char* str = tolua_tocppstring(tolua_S,1,0);
|
|
cMCLogger::GetInstance()->LogSimple( str, 2 );
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_LOGERROR(lua_State* tolua_S)
|
|
{
|
|
const char* str = tolua_tocppstring(tolua_S,1,0);
|
|
cMCLogger::GetInstance()->LogSimple( str, 3 );
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_cWorld_ForEachPlayer(lua_State* tolua_S)
|
|
{
|
|
int NumArgs = lua_gettop( tolua_S )-1; // This includes 'self'
|
|
if( NumArgs != 1 && NumArgs != 2)
|
|
{
|
|
LOGWARN("Error in function call 'ForEachPlayer': Requires 1 or 2 arguments, got %i", NumArgs );
|
|
return 0;
|
|
}
|
|
|
|
cWorld* self = (cWorld*) tolua_tousertype(tolua_S,1,0);
|
|
|
|
if( !lua_isfunction( tolua_S, 2 ) )
|
|
{
|
|
LOGWARN("Error in function call 'ForEachPlayer': Expected a function for parameter #1");
|
|
return 0;
|
|
}
|
|
|
|
// luaL_ref gets reference to value on top of the stack, the table is the last argument and therefore on the top
|
|
int TableRef = LUA_REFNIL;
|
|
if( NumArgs == 2 )
|
|
{
|
|
TableRef = luaL_ref(tolua_S, LUA_REGISTRYINDEX);
|
|
if( TableRef == LUA_REFNIL )
|
|
{
|
|
LOGWARN("Error in function call 'ForEachPlayer': Could not get value reference of parameter #2");
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
// table value is popped, and now function is on top of the stack
|
|
int FuncRef = luaL_ref(tolua_S, LUA_REGISTRYINDEX);
|
|
if( FuncRef == LUA_REFNIL )
|
|
{
|
|
LOGWARN("Error in function call 'ForEachPlayer': Could not get function reference of parameter #1");
|
|
return 0;
|
|
}
|
|
|
|
class cLuaPlayerCallback : public cPlayerListCallback
|
|
{
|
|
public:
|
|
cLuaPlayerCallback( lua_State* a_LuaState, int a_FuncRef, int a_TableRef )
|
|
: LuaState( a_LuaState )
|
|
, FuncRef( a_FuncRef )
|
|
, TableRef( a_TableRef )
|
|
{}
|
|
|
|
private:
|
|
virtual bool Item(cPlayer * a_Player) override
|
|
{
|
|
lua_rawgeti( LuaState, LUA_REGISTRYINDEX, FuncRef); // Push function reference
|
|
tolua_pushusertype( LuaState, a_Player, "cPlayer" );
|
|
if( TableRef != LUA_REFNIL )
|
|
{
|
|
lua_rawgeti( LuaState, LUA_REGISTRYINDEX, TableRef); // Push table reference
|
|
}
|
|
|
|
int s = lua_pcall( LuaState, (TableRef==LUA_REFNIL?1:2), 1, 0);
|
|
if( report_errors( LuaState, s ) )
|
|
{
|
|
return false; // Maybe we should return true?
|
|
}
|
|
|
|
if( lua_isboolean( LuaState, -1 ) )
|
|
{
|
|
return (tolua_toboolean( LuaState, -1, 0) > 0);
|
|
}
|
|
return false;
|
|
}
|
|
lua_State* LuaState;
|
|
int FuncRef;
|
|
int TableRef;
|
|
} Callback( tolua_S, FuncRef, TableRef );
|
|
|
|
bool bRetVal = self->ForEachPlayer( Callback );
|
|
|
|
// Unreference the values again, so the LUA_REGISTRYINDEX can make place for other references
|
|
luaL_unref( tolua_S, LUA_REGISTRYINDEX, TableRef );
|
|
luaL_unref( tolua_S, LUA_REGISTRYINDEX, FuncRef );
|
|
|
|
// Push return value on stack
|
|
tolua_pushboolean( tolua_S, bRetVal );
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_cPlugin_GetCommands(lua_State* tolua_S)
|
|
{
|
|
cPlugin* self = (cPlugin*) tolua_tousertype(tolua_S,1,0);
|
|
|
|
const std::vector< cPlugin::CommandStruct > & AllCommands = self->GetCommands();
|
|
|
|
lua_createtable(tolua_S, AllCommands.size(), 0);
|
|
int newTable = lua_gettop(tolua_S);
|
|
int index = 1;
|
|
std::vector< cPlugin::CommandStruct >::const_iterator iter = AllCommands.begin();
|
|
while(iter != AllCommands.end())
|
|
{
|
|
const cPlugin::CommandStruct & CS = *iter;
|
|
tolua_pushusertype( tolua_S, (void*)&CS, "const cPlugin::CommandStruct" );
|
|
lua_rawseti(tolua_S, newTable, index);
|
|
++iter;
|
|
++index;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_cPluginManager_GetAllPlugins(lua_State* tolua_S)
|
|
{
|
|
cPluginManager* self = (cPluginManager*) tolua_tousertype(tolua_S,1,0);
|
|
|
|
const cPluginManager::PluginList & AllPlugins = self->GetAllPlugins();
|
|
|
|
lua_createtable(tolua_S, AllPlugins.size(), 0);
|
|
int newTable = lua_gettop(tolua_S);
|
|
int index = 1;
|
|
cPluginManager::PluginList::const_iterator iter = AllPlugins.begin();
|
|
while(iter != AllPlugins.end())
|
|
{
|
|
const cPlugin* Plugin = *iter;
|
|
tolua_pushusertype( tolua_S, (void*)Plugin, "const cPlugin" );
|
|
lua_rawseti(tolua_S, newTable, index);
|
|
++iter;
|
|
++index;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_cPlayer_GetGroups(lua_State* tolua_S)
|
|
{
|
|
cPlayer* self = (cPlayer*) tolua_tousertype(tolua_S,1,0);
|
|
|
|
const cPlayer::GroupList & AllGroups = self->GetGroups();
|
|
|
|
lua_createtable(tolua_S, AllGroups.size(), 0);
|
|
int newTable = lua_gettop(tolua_S);
|
|
int index = 1;
|
|
cPlayer::GroupList::const_iterator iter = AllGroups.begin();
|
|
while(iter != AllGroups.end())
|
|
{
|
|
const cGroup* Group = *iter;
|
|
tolua_pushusertype( tolua_S, (void*)Group, "const cGroup" );
|
|
lua_rawseti(tolua_S, newTable, index);
|
|
++iter;
|
|
++index;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_cPlayer_GetResolvedPermissions(lua_State* tolua_S)
|
|
{
|
|
cPlayer* self = (cPlayer*) tolua_tousertype(tolua_S,1,0);
|
|
|
|
cPlayer::StringList AllPermissions = self->GetResolvedPermissions();
|
|
|
|
lua_createtable(tolua_S, AllPermissions.size(), 0);
|
|
int newTable = lua_gettop(tolua_S);
|
|
int index = 1;
|
|
cPlayer::StringList::iterator iter = AllPermissions.begin();
|
|
while(iter != AllPermissions.end())
|
|
{
|
|
std::string& Permission = *iter;
|
|
tolua_pushstring( tolua_S, Permission.c_str() );
|
|
lua_rawseti(tolua_S, newTable, index);
|
|
++iter;
|
|
++index;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_cPlugin_BindCommand(lua_State* tolua_S)
|
|
{
|
|
cPlugin* self = (cPlugin*) tolua_tousertype(tolua_S,1,0);
|
|
cPluginManager* PluginManager = cRoot::Get()->GetPluginManager();
|
|
cLuaCommandBinder* CommandBinder = PluginManager->GetLuaCommandBinder();
|
|
|
|
tolua_Error tolua_err;
|
|
tolua_err.array = 0;
|
|
tolua_err.index = 0;
|
|
tolua_err.type = 0;
|
|
|
|
std::string Permission = "";
|
|
std::string Command = "";
|
|
int Reference = LUA_REFNIL;
|
|
|
|
if( tolua_isstring( tolua_S, 2, 0, &tolua_err ) &&
|
|
lua_isfunction( tolua_S, 3 ) )
|
|
{
|
|
Reference = luaL_ref(tolua_S, LUA_REGISTRYINDEX);
|
|
Command = ((std::string) tolua_tocppstring(tolua_S,2,0));
|
|
}
|
|
else if( tolua_isstring( tolua_S, 2, 0, &tolua_err ) &&
|
|
tolua_isstring( tolua_S, 3, 0, &tolua_err ) &&
|
|
lua_isfunction( tolua_S, 4 ) )
|
|
{
|
|
Reference = luaL_ref(tolua_S, LUA_REGISTRYINDEX);
|
|
Command = ((std::string) tolua_tocppstring(tolua_S,2,0));
|
|
Permission = ((std::string) tolua_tocppstring(tolua_S,3,0));
|
|
}
|
|
else
|
|
{
|
|
if( tolua_err.type == 0 )
|
|
{
|
|
tolua_err.type = "function";
|
|
}
|
|
tolua_error(tolua_S,"#ferror in function 'BindCommand'.",&tolua_err);
|
|
return 0;
|
|
}
|
|
|
|
if( Reference != LUA_REFNIL )
|
|
{
|
|
if( !CommandBinder->BindCommand( Command, Permission, self, tolua_S, Reference ) )
|
|
{
|
|
luaL_unref( tolua_S, LUA_REGISTRYINDEX, Reference );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
LOGERROR("ERROR: cPlugin:BindCommand invalid function reference in 2nd argument (Command: \"%s\")", Command.c_str() );
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_cWebPlugin_Lua_AddTab(lua_State* tolua_S)
|
|
{
|
|
cWebPlugin_Lua* self = (cWebPlugin_Lua*) tolua_tousertype(tolua_S,1,0);
|
|
|
|
tolua_Error tolua_err;
|
|
tolua_err.array = 0;
|
|
tolua_err.index = 0;
|
|
tolua_err.type = 0;
|
|
|
|
std::string Title = "";
|
|
int Reference = LUA_REFNIL;
|
|
|
|
if( tolua_isstring( tolua_S, 2, 0, &tolua_err ) &&
|
|
lua_isfunction( tolua_S, 3 ) )
|
|
{
|
|
Reference = luaL_ref(tolua_S, LUA_REGISTRYINDEX);
|
|
Title = ((std::string) tolua_tocppstring(tolua_S,2,0));
|
|
}
|
|
else
|
|
{
|
|
if( tolua_err.type == 0 )
|
|
{
|
|
tolua_err.type = "function";
|
|
}
|
|
tolua_error(tolua_S,"#ferror in function 'AddTab'.",&tolua_err);
|
|
return 0;
|
|
}
|
|
|
|
if( Reference != LUA_REFNIL )
|
|
{
|
|
if( !self->AddTab( Title.c_str(), tolua_S, Reference ) )
|
|
{
|
|
luaL_unref( tolua_S, LUA_REGISTRYINDEX, Reference );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
LOGERROR("ERROR: cWebPlugin_Lua:AddTab invalid function reference in 2nd argument (Title: \"%s\")", Title.c_str() );
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_md5(lua_State* tolua_S)
|
|
{
|
|
std::string SourceString = tolua_tostring(tolua_S, 1, 0);
|
|
std::string CryptedString = md5( SourceString );
|
|
tolua_pushstring( tolua_S, CryptedString.c_str() );
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_push_StringStringMap(lua_State* tolua_S, std::map< std::string, std::string >& a_StringStringMap )
|
|
{
|
|
lua_newtable(tolua_S);
|
|
int top = lua_gettop(tolua_S);
|
|
|
|
for( std::map< std::string, std::string >::iterator it = a_StringStringMap.begin(); it != a_StringStringMap.end(); ++it )
|
|
{
|
|
const char* key = it->first.c_str();
|
|
const char* value = it->second.c_str();
|
|
lua_pushstring(tolua_S, key);
|
|
lua_pushstring(tolua_S, value);
|
|
lua_settable(tolua_S, top);
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_get_HTTPRequest_Params(lua_State* tolua_S)
|
|
{
|
|
HTTPRequest* self = (HTTPRequest*) tolua_tousertype(tolua_S,1,0);
|
|
return tolua_push_StringStringMap(tolua_S, self->Params);
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_get_HTTPRequest_PostParams(lua_State* tolua_S)
|
|
{
|
|
HTTPRequest* self = (HTTPRequest*) tolua_tousertype(tolua_S,1,0);
|
|
return tolua_push_StringStringMap(tolua_S, self->PostParams);
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int tolua_get_HTTPRequest_FormData(lua_State* tolua_S)
|
|
{
|
|
HTTPRequest* self = (HTTPRequest*) tolua_tousertype(tolua_S,1,0);
|
|
std::map< std::string, HTTPFormData >& FormData = self->FormData;
|
|
|
|
lua_newtable(tolua_S);
|
|
int top = lua_gettop(tolua_S);
|
|
|
|
for( std::map< std::string, HTTPFormData >::iterator it = FormData.begin(); it != FormData.end(); ++it )
|
|
{
|
|
lua_pushstring(tolua_S, it->first.c_str() );
|
|
tolua_pushusertype(tolua_S, &(it->second), "HTTPFormData" );
|
|
//lua_pushlstring(tolua_S, it->second.Value.c_str(), it->second.Value.size() ); // Might contain binary data
|
|
lua_settable(tolua_S, top);
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void ManualBindings::Bind( lua_State* tolua_S )
|
|
{
|
|
tolua_beginmodule(tolua_S,NULL);
|
|
tolua_function(tolua_S,"StringSplit",tolua_StringSplit);
|
|
tolua_function(tolua_S,"LOG",tolua_LOG);
|
|
tolua_function(tolua_S,"LOGINFO",tolua_LOGINFO);
|
|
tolua_function(tolua_S,"LOGWARN",tolua_LOGWARN);
|
|
tolua_function(tolua_S,"LOGERROR",tolua_LOGERROR);
|
|
tolua_function(tolua_S,"Log",tolua_LOG); // Deprecated
|
|
|
|
tolua_beginmodule(tolua_S,"cWorld");
|
|
tolua_function(tolua_S,"ForEachPlayer",tolua_cWorld_ForEachPlayer);
|
|
tolua_endmodule(tolua_S);
|
|
tolua_beginmodule(tolua_S,"cPlugin");
|
|
tolua_function(tolua_S,"GetCommands",tolua_cPlugin_GetCommands);
|
|
tolua_function(tolua_S,"BindCommand",tolua_cPlugin_BindCommand);
|
|
tolua_endmodule(tolua_S);
|
|
tolua_beginmodule(tolua_S,"cPluginManager");
|
|
tolua_function(tolua_S,"GetAllPlugins",tolua_cPluginManager_GetAllPlugins);
|
|
tolua_endmodule(tolua_S);
|
|
tolua_beginmodule(tolua_S,"cPlayer");
|
|
tolua_function(tolua_S,"GetGroups",tolua_cPlayer_GetGroups);
|
|
tolua_function(tolua_S,"GetResolvedPermissions",tolua_cPlayer_GetResolvedPermissions);
|
|
tolua_endmodule(tolua_S);
|
|
tolua_beginmodule(tolua_S,"cWebPlugin_Lua");
|
|
tolua_function(tolua_S,"AddTab",tolua_cWebPlugin_Lua_AddTab);
|
|
tolua_endmodule(tolua_S);
|
|
|
|
tolua_cclass(tolua_S,"HTTPRequest","HTTPRequest","",NULL);
|
|
tolua_beginmodule(tolua_S,"HTTPRequest");
|
|
//tolua_variable(tolua_S,"Method",tolua_get_HTTPRequest_Method,tolua_set_HTTPRequest_Method);
|
|
//tolua_variable(tolua_S,"Path",tolua_get_HTTPRequest_Path,tolua_set_HTTPRequest_Path);
|
|
tolua_variable(tolua_S,"Params",tolua_get_HTTPRequest_Params,0);
|
|
tolua_variable(tolua_S,"PostParams",tolua_get_HTTPRequest_PostParams,0);
|
|
tolua_variable(tolua_S,"FormData",tolua_get_HTTPRequest_FormData,0);
|
|
tolua_endmodule(tolua_S);
|
|
|
|
tolua_function(tolua_S,"md5",tolua_md5);
|
|
|
|
tolua_endmodule(tolua_S);
|
|
}
|
|
|
|
|
|
|
|
|