luaAPI.h

#ifndef LUAAPI_H
#define LUAAPI_H

#ifndef LAPI_GDEXTENSION
#include "core/core_bind.h"
#include "core/object/ref_counted.h"
#else
#include <godot_cpp/classes/ref.hpp>
#endif

#include "luaError.h"

#include <luaState.h>
#include <lua/lua.hpp>

#ifdef LAPI_GDEXTENSION
using namespace godot;
#endif

class LuaCoroutine;
class LuaObjectMetatable;

class LuaAPI : public RefCounted {
	GDCLASS(LuaAPI, RefCounted);

protected:
	static void _bind_methods();

public:
	LuaAPI();
	~LuaAPI();

	void setHook(Callable hook, int mask, int count);

	void setUseCallables(bool value);
	bool getUseCallables() const;

	void setObjectMetatable(Ref<LuaObjectMetatable> value);
	Ref<LuaObjectMetatable> getObjectMetatable() const;

	void setMemoryLimit(uint64_t limit);
	uint64_t getMemoryLimit() const;

	int configureGC(int what, int data);
	uint64_t getMemoryUsage() const;

	bool luaFunctionExists(String functionName);

	Variant pullVariant(String name);
	Variant callFunction(String functionName, Array args);
	Variant doFile(String fileName, Array args);
	Variant doString(String code, Array args);
	Variant getRegistryValue(String name);

	Ref<LuaError> setRegistryValue(String name, Variant var);
	Ref<LuaError> bindLibraries(Array libs);
	Ref<LuaError> pushGlobalVariant(String name, Variant var);

	Ref<LuaCoroutine> newCoroutine();
	Ref<LuaCoroutine> getRunningCoroutine();

	lua_State *newThreadState();
	lua_State *getState();

	enum HookMask {
		HOOK_MASK_CALL = LUA_MASKCALL,
		HOOK_MASK_RETURN = LUA_MASKRET,
		HOOK_MASK_LINE = LUA_MASKLINE,
		HOOK_MASK_COUNT = LUA_MASKCOUNT,
	};

	enum GCOption {
		GC_STOP = LUA_GCSTOP,
		GC_RESTART = LUA_GCRESTART,
		GC_COLLECT = LUA_GCCOLLECT,
		GC_COUNT = LUA_GCCOUNT,
		GC_COUNTB = LUA_GCCOUNTB,
		GC_STEP = LUA_GCSTEP,
		GC_SETPAUSE = LUA_GCSETPAUSE,
		GC_SETSTEPMUL = LUA_GCSETSTEPMUL,
	};

private:
	bool useCallables = true;

	LuaState state;
	lua_State *lState = nullptr;

	Ref<LuaObjectMetatable> objectMetatable;

	static void *luaAlloc(void *ud, void *ptr, size_t osize, size_t nsize);

	struct LuaAllocData {
		uint64_t memoryUsed = 0;
		uint64_t memoryLimit = 0;
	};

	LuaAllocData luaAllocData;

	Variant execute(int argc, int handlerIndex);
};

VARIANT_ENUM_CAST(LuaAPI::HookMask)
VARIANT_ENUM_CAST(LuaAPI::GCOption)

#endif