eqemu-server/zone/lua_parser.cpp

#ifdef LUA_EQEMU

#include "lua_parser.h"
#include <ctype.h>
#include <sstream>

#include "lua.hpp"
#include <luabind/luabind.hpp>
#include <boost/any.hpp>

#include "masterentity.h"
#include "lua_entity.h"
#include "lua_mob.h"
#include "lua_client.h"
#include "lua_npc.h"

const char *LuaEvents[_LargestEventID] = {
	"event_say",
	"event_item",
	"event_death",
	"event_spawn",
	"event_attack",
	"event_combat",
	"event_aggro",
	"event_slay",
	"event_npc_slay",
	"event_waypoint_arrive",
	"event_waypoint_depart",
	"event_timer",
	"event_signal",
	"event_hp",
	"event_enter",
	"event_exit",
	"event_enterzone",
	"event_clickdoor",
	"event_loot",
	"event_zone",
	"event_level_up",
	"event_killed_merit",
	"event_cast_on",
	"event_taskaccepted",
	"event_task_stage_complete",
	"event_task_update",
	"event_task_complete",
	"event_task_fail",
	"event_aggro_say",
	"event_player_pickup",
	"event_popup_response",
	"event_proximity_say",
	"event_cast",
	"event_scale_calc",
	"event_item_enterzone",
	"event_target_change",
	"event_hate_list",
	"event_spell_effect_client",
	"event_spell_effect_npc",
	"event_spell_effect_buff_tic_client",
	"event_spell_effect_buff_tic_npc",
	"event_spell_effect_translocate_complete",
	"event_combine_success",
	"event_combine_failure",
	"event_item_click",
	"event_item_click_cast",
	"event_group_change",
	"event_forage_success",
	"event_forage_failure",
	"event_fish_start",
	"event_fish_success",
	"event_fish_failure",
	"event_click_object",
	"event_discover_item",
	"event_disconnect",
	"event_connect",
	"event_item_tick",
	"event_duel_win",
	"event_duel_lose"
};

LuaParser::LuaParser() {
}

LuaParser::~LuaParser() {
	ClearStates();
}

int LuaParser::EventNPC(QuestEventID evt, NPC* npc, Mob *init, std::string data, uint32 extra_data) {
	if(evt >= _LargestEventID) {
		return 0;
	}

	if(!npc) {
		return 0;
	}

	if(evt != EVENT_SPAWN && evt != EVENT_SAY) {
		return 0;
	}


	const char *sub_name = LuaEvents[evt];
	if(!HasQuestSub(npc->GetNPCTypeID(), sub_name)) {
		return 0;
	}

	std::stringstream package_name;
	package_name << "npc_" << npc->GetNPCTypeID();
	
	lua_State *L = nullptr;
	auto iter = states_.find(package_name.str());
	if(iter == states_.end()) {
		return 0;
	}
	L = iter->second;

	try {
		lua_getfield(L, LUA_GLOBALSINDEX, sub_name);

		Lua_NPC l_npc(npc);
		Lua_Client l_client;
		luabind::object l_npc_o = luabind::object(L, l_npc);
		l_npc_o.push(L);

		int arg_count = 1;
		int ret_count = 1;

		if(evt == EVENT_SAY) {
			l_client.d_ = init;
			luabind::object l_client_o = luabind::object(L, l_client);
			l_client_o.push(L);

			lua_pushstring(L, data.c_str());
			lua_pushinteger(L, extra_data);

			arg_count += 3;
		}

		if(lua_pcall(L, arg_count, ret_count, 0)) {
			printf("Error: %s\n", lua_tostring(L, -1));
			return 0;
		}

		if(lua_isnumber(L, -1)) {
			int ret = static_cast<int>(lua_tointeger(L, -1));
			return ret;
		}

	} catch(std::exception &ex) {
		printf("%s\n", ex.what());
		return 0;
	}

	return 0;
}

int LuaParser::EventGlobalNPC(QuestEventID evt, NPC* npc, Mob *init, std::string data, uint32 extra_data) {
	return 0;
}

int LuaParser::EventPlayer(QuestEventID evt, Client *client, std::string data, uint32 extra_data) {
	return 0;
}

int LuaParser::EventGlobalPlayer(QuestEventID evt, Client *client, std::string data, uint32 extra_data) {
	return 0;
}

int LuaParser::EventItem(QuestEventID evt, Client *client, ItemInst *item, uint32 objid, uint32 extra_data) {
	return 0;
}

int LuaParser::EventSpell(QuestEventID evt, NPC* npc, Client *client, uint32 spell_id, uint32 extra_data) {
	return 00;
}

bool LuaParser::HasQuestSub(uint32 npc_id, const char *subname) {
	std::stringstream package_name;
	package_name << "npc_" << npc_id;

	return HasFunction(subname, package_name.str());
}

bool LuaParser::HasGlobalQuestSub(const char *subname) {
	return HasFunction(subname, "global_npc");
}

bool LuaParser::PlayerHasQuestSub(const char *subname) {
	return HasFunction(subname, "player");
}

bool LuaParser::GlobalPlayerHasQuestSub(const char *subname) {
	return HasFunction(subname, "global_player");
}

bool LuaParser::SpellHasQuestSub(uint32 spell_id, const char *subname) {
	std::stringstream package_name;
	package_name << "spell_" << spell_id;

	return HasFunction(subname, package_name.str());
}

bool LuaParser::ItemHasQuestSub(ItemInst *itm, const char *subname) {
	std::stringstream package_name;
	package_name << "item_";

	std::stringstream item_name;
	const Item_Struct* item = itm->GetItem();
	if(strcmp("EVENT_SCALE_CALC", subname) == 0 || strcmp("EVENT_ITEM_ENTERZONE", subname) == 0)
	{
		item_name << item->CharmFile;
	}
	else if(strcmp("EVENT_ITEM_CLICK", subname) == 0 || strcmp("EVENT_ITEM_CLICK_CAST", subname) == 0 )
	{
		item_name << "script_";
		item_name << item->ScriptFileID;
	}
	else
	{
		item_name << "item_";
		item_name << item->ID;
	}

	package_name << item_name;
	return HasFunction(subname, package_name.str());
}

void LuaParser::LoadNPCScript(std::string filename, int npc_id) {
	std::stringstream package_name;
	package_name << "npc_" << npc_id;

	LoadScript(filename, package_name.str());
}

void LuaParser::LoadGlobalNPCScript(std::string filename) {
	LoadScript(filename, "global_npc");
}

void LuaParser::LoadPlayerScript(std::string filename) {
	LoadScript(filename, "player");
}

void LuaParser::LoadGlobalPlayerScript(std::string filename) {
	LoadScript(filename, "global_player");
}

void LuaParser::LoadItemScript(std::string filename, std::string item_script) {
	std::stringstream package_name;
	package_name << "item_" << item_script;

	LoadScript(filename, package_name.str());
}

void LuaParser::LoadSpellScript(std::string filename, uint32 spell_id) {
	std::stringstream package_name;
	package_name << "spell_" << spell_id;

	LoadScript(filename, package_name.str());
}

void LuaParser::AddVar(std::string name, std::string val) {
	vars_[name] = val;
}

std::string LuaParser::GetVar(std::string name) {
	auto iter = vars_.find(name);
	if(iter != vars_.end()) {
		return iter->second;
	}

	return std::string();
}

void LuaParser::ReloadQuests() {
	ClearStates();
}

void LuaParser::LoadScript(std::string filename, std::string package_name) {
	auto iter = states_.find(package_name);
	if(iter != states_.end()) {
		return;
	}

	lua_State *L = luaL_newstate();
	luaL_openlibs(L);

	//lua_pushnil(L);
	//lua_setglobal(L, "os");
	//
	//lua_pushnil(L);
	//lua_setglobal(L, "io");

	MapFunctions(L);

	if(luaL_dofile(L, filename.c_str())) {
		printf("Lua Error: %s\n", lua_tostring(L, -1));
		lua_close(L);
		return;
	}

	states_[package_name] = L;
}

bool LuaParser::HasFunction(std::string subname, std::string package_name) {
	size_t sz = subname.length();
	for(size_t i = 0; i < sz; ++i) {
		char c = subname[i];
		if(65 <= c && c <= 90) {
			c += 32;
		}
		subname[i] = c;
	}

	auto iter = states_.find(package_name);
	if(iter == states_.end()) {
		return false;
	}

	lua_getfield(iter->second, LUA_GLOBALSINDEX, subname.c_str());
	if(lua_isfunction(iter->second, -1)) {
		return true;
	}

	return false;
}

void LuaParser::ClearStates() {
	auto iter = states_.begin();
	while(iter != states_.end()) {
		if(iter->second) {
			lua_close(iter->second);
		}
		++iter;
	}

	states_.clear();
}

void LuaParser::MapFunctions(lua_State *L) {

	try {
		luabind::open(L);

		luabind::module(L)
		[
			luabind::class_<Lua_Entity>("Entity")
				.def(luabind::constructor<>())
				.property("null", &Lua_Entity::Null)
				.property("valid", &Lua_Entity::Valid)
				.def("IsClient", &Lua_Entity::IsClient)
				.def("IsNPC", &Lua_Entity::IsNPC)
				.def("IsMob", &Lua_Entity::IsMob)
				.def("IsMerc", &Lua_Entity::IsMerc)
				.def("IsCorpse", &Lua_Entity::IsCorpse)
				.def("IsPlayerCorpse", &Lua_Entity::IsPlayerCorpse)
				.def("IsNPCCorpse", &Lua_Entity::IsNPCCorpse)
				.def("IsObject", &Lua_Entity::IsObject)
				.def("IsDoor", &Lua_Entity::IsDoor)
				.def("IsTrap", &Lua_Entity::IsTrap)
				.def("IsBeacon", &Lua_Entity::IsBeacon)
				.def("GetID", &Lua_Entity::GetID)
				.def("CastToClient", &Lua_Entity::CastToClient)
				.def("CastToNPC", &Lua_Entity::CastToNPC)
				.def("CastToMob", &Lua_Entity::CastToMob),

			luabind::class_<Lua_Mob, Lua_Entity>("Mob")
				.def(luabind::constructor<>())
				.def("GetName", &Lua_Mob::GetName)
				.def("Depop", (void(Lua_Mob::*)(void))&Lua_Mob::Depop)
				.def("Depop", (void(Lua_Mob::*)(bool))&Lua_Mob::Depop)
				.def("RogueAssassinate", &Lua_Mob::RogueAssassinate)
				.def("BehindMob", (bool(Lua_Mob::*)(void))&Lua_Mob::BehindMob)
				.def("BehindMob", (bool(Lua_Mob::*)(Lua_Mob))&Lua_Mob::BehindMob)
				.def("BehindMob", (bool(Lua_Mob::*)(Lua_Mob,float))&Lua_Mob::BehindMob)
				.def("BehindMob", (bool(Lua_Mob::*)(Lua_Mob,float,float))&Lua_Mob::BehindMob)
				.def("SetLevel", (void(Lua_Mob::*)(int))&Lua_Mob::SetLevel)
				.def("SetLevel", (void(Lua_Mob::*)(int,bool))&Lua_Mob::SetLevel)
				.def("GetEquipment", &Lua_Mob::GetEquipment)
				.def("GetEquipmentMaterial", &Lua_Mob::GetEquipmentMaterial)
				.def("GetEquipmentColor", &Lua_Mob::GetEquipmentColor)
				.def("GetArmorTint", &Lua_Mob::GetArmorTint)
				.def("IsMoving", &Lua_Mob::IsMoving)
				.def("GotoBind", &Lua_Mob::GotoBind)
				.def("Attack", (bool(Lua_Mob::*)(Lua_Mob))&Lua_Mob::Attack)
				.def("Attack", (bool(Lua_Mob::*)(Lua_Mob,int))&Lua_Mob::Attack)
				.def("Attack", (bool(Lua_Mob::*)(Lua_Mob,int,bool))&Lua_Mob::Attack)
				.def("Attack", (bool(Lua_Mob::*)(Lua_Mob,int,bool,bool))&Lua_Mob::Attack)
				.def("Attack", (bool(Lua_Mob::*)(Lua_Mob,int,bool,bool,bool))&Lua_Mob::Attack)
				.def("Damage", (void(Lua_Mob::*)(Lua_Mob,int,int,int))&Lua_Mob::Damage)
				.def("Damage", (void(Lua_Mob::*)(Lua_Mob,int,int,int,bool))&Lua_Mob::Damage)
				.def("Damage", (void(Lua_Mob::*)(Lua_Mob,int,int,int,bool,int))&Lua_Mob::Damage)
				.def("Damage", (void(Lua_Mob::*)(Lua_Mob,int,int,int,bool,int,bool))&Lua_Mob::Damage)
				.def("RangedAttack", &Lua_Mob::RangedAttack)
				.def("ThrowingAttack", &Lua_Mob::ThrowingAttack)
				.def("Heal", &Lua_Mob::Heal)
				.def("HealDamage", (void(Lua_Mob::*)(uint32))&Lua_Mob::HealDamage)
				.def("HealDamage", (void(Lua_Mob::*)(uint32,Lua_Mob))&Lua_Mob::HealDamage)
				.def("GetLevelCon", (uint32(Lua_Mob::*)(int))&Lua_Mob::GetLevelCon)
				.def("GetLevelCon", (uint32(Lua_Mob::*)(int,int))&Lua_Mob::GetLevelCon)
				.def("SetHP", &Lua_Mob::SetHP)
				.def("DoAnim", (void(Lua_Mob::*)(int))&Lua_Mob::DoAnim)
				.def("DoAnim", (void(Lua_Mob::*)(int,int))&Lua_Mob::DoAnim)
				.def("DoAnim", (void(Lua_Mob::*)(int,int,bool))&Lua_Mob::DoAnim)
				.def("DoAnim", (void(Lua_Mob::*)(int,int,bool,int))&Lua_Mob::DoAnim)
				.def("ChangeSize", (void(Lua_Mob::*)(double))&Lua_Mob::ChangeSize)
				.def("ChangeSize", (void(Lua_Mob::*)(double,bool))&Lua_Mob::ChangeSize)
				.def("GMMove", (void(Lua_Mob::*)(double,double,double))&Lua_Mob::GMMove)
				.def("GMMove", (void(Lua_Mob::*)(double,double,double,double))&Lua_Mob::GMMove)
				.def("GMMove", (void(Lua_Mob::*)(double,double,double,double,bool))&Lua_Mob::GMMove)
				.def("SendPosUpdate", (void(Lua_Mob::*)(void))&Lua_Mob::SendPosUpdate)
				.def("SendPosUpdate", (void(Lua_Mob::*)(bool))&Lua_Mob::SendPosUpdate)
				.def("SendPosition", &Lua_Mob::SendPosition)
				.def("HasProcs", &Lua_Mob::HasProcs)
				.def("IsInvisible", (bool(Lua_Mob::*)(void))&Lua_Mob::IsInvisible)
				.def("IsInvisible", (bool(Lua_Mob::*)(Lua_Mob))&Lua_Mob::IsInvisible)
				.def("SetInvisible", &Lua_Mob::SetInvisible)
				.def("FindBuff", &Lua_Mob::FindBuff)
				.def("FindType", (bool(Lua_Mob::*)(int))&Lua_Mob::FindType)
				.def("FindType", (bool(Lua_Mob::*)(int,bool))&Lua_Mob::FindType)
				.def("FindType", (bool(Lua_Mob::*)(int,bool,int))&Lua_Mob::FindType)
				.def("GetBuffSlotFromType", &Lua_Mob::GetBuffSlotFromType)
				.def("MakePet", (void(Lua_Mob::*)(int,const char*))&Lua_Mob::MakePet)
				.def("MakePet", (void(Lua_Mob::*)(int,const char*,const char*))&Lua_Mob::MakePet)
				.def("MakePoweredPet", (void(Lua_Mob::*)(int,const char*,int))&Lua_Mob::MakePoweredPet)
				.def("MakePoweredPet", (void(Lua_Mob::*)(int,const char*,int,const char*))&Lua_Mob::MakePoweredPet)
				.def("GetBaseRace", &Lua_Mob::GetBaseRace)
				.def("GetBaseGender", &Lua_Mob::GetBaseGender)
				.def("GetDeity", &Lua_Mob::GetDeity)
				.def("GetRace", &Lua_Mob::GetRace)
				.def("GetGender", &Lua_Mob::GetGender)
				.def("GetTexture", &Lua_Mob::GetTexture)
				.def("GetHelmTexture", &Lua_Mob::GetHelmTexture)
				.def("GetHairColor", &Lua_Mob::GetHairColor)
				.def("GetBeardColor", &Lua_Mob::GetBeardColor)
				.def("GetEyeColor1", &Lua_Mob::GetEyeColor1)
				.def("GetEyeColor2", &Lua_Mob::GetEyeColor2)
				.def("GetHairStyle", &Lua_Mob::GetHairStyle)
				.def("GetLuclinFace", &Lua_Mob::GetLuclinFace)
				.def("GetBeard", &Lua_Mob::GetBeard)
				.def("GetDrakkinHeritage", &Lua_Mob::GetDrakkinHeritage)
				.def("GetDrakkinTattoo", &Lua_Mob::GetDrakkinTattoo)
				.def("GetDrakkinDetails", &Lua_Mob::GetDrakkinDetails)
				.def("GetClass", &Lua_Mob::GetClass)
				.def("GetLevel", &Lua_Mob::GetLevel)
				.def("GetCleanName", &Lua_Mob::GetCleanName)
				.def("GetTarget", &Lua_Mob::GetTarget)
				.def("SetTarget", &Lua_Mob::SetTarget)
				.def("GetHPRatio", &Lua_Mob::GetHPRatio)
				.def("IsWarriorClass", &Lua_Mob::IsWarriorClass)
				.def("GetHP", &Lua_Mob::GetHP)
				.def("GetMaxHP", &Lua_Mob::GetMaxHP)
				.def("GetItemHPBonuses", &Lua_Mob::GetItemHPBonuses)
				.def("GetSpellHPBonuses", &Lua_Mob::GetSpellHPBonuses)
				.def("GetWalkspeed", &Lua_Mob::GetWalkspeed)
				.def("GetRunspeed", &Lua_Mob::GetRunspeed)
				.def("GetCasterLevel", &Lua_Mob::GetCasterLevel)
				.def("GetMaxMana", &Lua_Mob::GetMaxMana)
				.def("GetMana", &Lua_Mob::GetMana)
				.def("SetMana", &Lua_Mob::SetMana)
				.def("GetManaRatio", &Lua_Mob::GetManaRatio)
				.def("GetAC", &Lua_Mob::GetAC)
				.def("GetATK", &Lua_Mob::GetATK)
				.def("GetSTR", &Lua_Mob::GetSTR)
				.def("GetSTA", &Lua_Mob::GetSTA)
				.def("GetDEX", &Lua_Mob::GetDEX)
				.def("GetAGI", &Lua_Mob::GetAGI)
				.def("GetINT", &Lua_Mob::GetINT)
				.def("GetWIS", &Lua_Mob::GetWIS)
				.def("GetCHA", &Lua_Mob::GetCHA)
				.def("GetMR", &Lua_Mob::GetMR)
				.def("GetFR", &Lua_Mob::GetFR)
				.def("GetDR", &Lua_Mob::GetDR)
				.def("GetPR", &Lua_Mob::GetPR)
				.def("GetCR", &Lua_Mob::GetCR)
				.def("GetCorruption", &Lua_Mob::GetCorruption)
				.def("GetMaxSTR", &Lua_Mob::GetMaxSTR)
				.def("GetMaxSTA", &Lua_Mob::GetMaxSTA)
				.def("GetMaxDEX", &Lua_Mob::GetMaxDEX)
				.def("GetMaxAGI", &Lua_Mob::GetMaxAGI)
				.def("GetMaxINT", &Lua_Mob::GetMaxINT)
				.def("GetMaxWIS", &Lua_Mob::GetMaxWIS)
				.def("GetMaxCHA", &Lua_Mob::GetMaxCHA)
				.def("GetActSpellRange", (double(Lua_Mob::*)(int,double))&Lua_Mob::GetActSpellRange)
				.def("GetActSpellRange", (double(Lua_Mob::*)(int,double,bool))&Lua_Mob::GetActSpellRange)
				.def("GetActSpellDamage", &Lua_Mob::GetActSpellDamage)
				.def("GetActSpellHealing", &Lua_Mob::GetActSpellHealing)
				.def("GetActSpellCost", &Lua_Mob::GetActSpellCost)
				.def("GetActSpellDuration", &Lua_Mob::GetActSpellDuration)
				.def("GetActSpellCasttime", &Lua_Mob::GetActSpellCasttime)
				.def("ResistSpell", (double(Lua_Mob::*)(int,int,Lua_Mob))&Lua_Mob::ResistSpell)
				.def("ResistSpell", (double(Lua_Mob::*)(int,int,Lua_Mob,bool))&Lua_Mob::ResistSpell)
				.def("ResistSpell", (double(Lua_Mob::*)(int,int,Lua_Mob,bool,int))&Lua_Mob::ResistSpell)
				.def("ResistSpell", (double(Lua_Mob::*)(int,int,Lua_Mob,bool,int,bool))&Lua_Mob::ResistSpell)
				.def("GetSpecializeSkillValue", &Lua_Mob::GetSpecializeSkillValue)
				.def("GetNPCTypeID", &Lua_Mob::GetNPCTypeID)
				.def("IsTargeted", &Lua_Mob::IsTargeted)
				.def("GetX", &Lua_Mob::GetX)
				.def("GetY", &Lua_Mob::GetY)
				.def("GetZ", &Lua_Mob::GetZ)
				.def("GetHeading", &Lua_Mob::GetHeading)
				.def("GetWaypointX", &Lua_Mob::GetWaypointX)
				.def("GetWaypointY", &Lua_Mob::GetWaypointY)
				.def("GetWaypointZ", &Lua_Mob::GetWaypointZ)
				.def("GetWaypointH", &Lua_Mob::GetWaypointH)
				.def("GetWaypointPause", &Lua_Mob::GetWaypointPause)
				.def("GetWaypointID", &Lua_Mob::GetWaypointID)
				.def("SetCurrentWP", &Lua_Mob::SetCurrentWP)
				.def("GetSize", &Lua_Mob::GetSize)
				.def("SetFollowID", &Lua_Mob::SetFollowID)
				.def("GetFollowID", &Lua_Mob::GetFollowID)
				.def("Message", &Lua_Mob::Message)
				.def("Message_StringID", &Lua_Mob::Message_StringID)
				.def("Say", &Lua_Mob::Say)
				.def("Shout", &Lua_Mob::Shout)
				.def("Emote", &Lua_Mob::Emote)
				.def("InterruptSpell", (void(Lua_Mob::*)(void))&Lua_Mob::InterruptSpell)
				.def("InterruptSpell", (void(Lua_Mob::*)(int))&Lua_Mob::InterruptSpell)
				,

			luabind::class_<Lua_Client, Lua_Mob>("Client")
				.def(luabind::constructor<>()),

			luabind::class_<Lua_NPC, Lua_Mob>("NPC")
				.def(luabind::constructor<>())
		];
	
	} catch(std::exception &ex) {
		printf("Error: %s\n", ex.what());
	}
}

#endif