dep-ygopro-core/scriptlib.cpp

/*
 * scriptlib.cpp
 *
 *  Created on: 2010-7-29
 *      Author: Argon
 */
#include "scriptlib.h"
#include "duel.h"
#include "lua_obj.h"

int32 scriptlib::check_param(lua_State* L, int32 param_type, int32 index, int32 retfalse, void* retobj) {
	const char* type = nullptr;
	lua_obj* obj = nullptr;
	switch (param_type) {
	case PARAM_TYPE_CARD:
	case PARAM_TYPE_GROUP:
	case PARAM_TYPE_EFFECT:
		if((obj = lua_get<lua_obj*>(L, index)) != nullptr && obj->lua_type == param_type) {
			if(retobj)
				*(lua_obj**)retobj = obj;
			return TRUE;
		} else if(obj && obj->lua_type == PARAM_TYPE_DELETED) {
			luaL_error(L, "Attempting to access deleted object.");
		}
		type = param_type == PARAM_TYPE_CARD ? "Card" : param_type == PARAM_TYPE_GROUP ? "Group" : "Effect";
		break;
	case PARAM_TYPE_FUNCTION:
		if(lua_isfunction(L, index))
			return TRUE;
		type = "Function";
		break;
	case PARAM_TYPE_STRING:
		if(lua_isstring(L, index))
			return TRUE;
		type = "String";
		break;
	case PARAM_TYPE_INT:
		if(lua_isinteger(L, index) || lua_isnumber(L, index))
			return TRUE;
		type = "Int";
		break;
	case PARAM_TYPE_BOOLEAN:
		if(lua_gettop(L) >= index)
			return TRUE;
		type = "boolean";
		break;
	}
	if(retfalse)
		return FALSE;
	if(param_type == PARAM_TYPE_INT) {
		interpreter::print_stacktrace(L);
		const auto pduel = lua_get<duel*>(L);
		pduel->handle_message(pduel->lua->format(R"(Parameter %d should be "%s".)", index, type), OCG_LOG_TYPE_ERROR);
	} else {
		luaL_error(L, R"(Parameter %d should be "%s".)", index, type);
	}
	return FALSE;
}

int32 scriptlib::check_param_count(lua_State* L, int32 count) {
	if (lua_gettop(L) < count)
		luaL_error(L, "%d Parameters are needed.", count);
	return TRUE;
}
int32 scriptlib::check_action_permission(lua_State* L) {
	const auto pduel = lua_get<duel*>(L);
	if(pduel->lua->no_action)
		luaL_error(L, "Action is not allowed here.");
	return TRUE;
}

Function get_lua_function(lua_State * L, int index) {
	lua_pushvalue(L, index);
	int32 ref = luaL_ref(L, LUA_REGISTRYINDEX);
	return std::make_shared<Lua_Function_s>(lua_get<duel*>(L), ref);
}