RmlUi/Source/Lua/ContextDocumentsProxy.cpp

#include "ContextDocumentsProxy.h"
#include <RmlUi/Core/ElementDocument.h>

namespace Rml {
namespace Lua {
typedef ElementDocument Document;
template <>
void ExtraInit<ContextDocumentsProxy>(lua_State* L, int metatable_index)
{
	lua_pushcfunction(L, ContextDocumentsProxy__index);
	lua_setfield(L, metatable_index, "__index");
	lua_pushcfunction(L, ContextDocumentsProxy__pairs);
	lua_setfield(L, metatable_index, "__pairs");
}

int ContextDocumentsProxy__index(lua_State* L)
{
	/*the table obj and the missing key are currently on the stack(index 1 & 2) as defined by the Lua language*/
	int type = lua_type(L, 2);
	if (type == LUA_TNUMBER || type == LUA_TSTRING) // only valid key types
	{
		ContextDocumentsProxy* proxy = LuaType<ContextDocumentsProxy>::check(L, 1);
		Document* ret = nullptr;
		if (type == LUA_TSTRING)
			ret = proxy->owner->GetDocument(luaL_checkstring(L, 2));
		else
			ret = proxy->owner->GetDocument((int)luaL_checkinteger(L, 2) - 1);
		LuaType<Document>::push(L, ret, false);
		return 1;
	}
	else
		return LuaType<ContextDocumentsProxy>::index(L);
}

struct ContextDocumentsProxyPairs {
	static int next(lua_State* L)
	{
		ContextDocumentsProxy* obj = LuaType<ContextDocumentsProxy>::check(L, 1);
		ContextDocumentsProxyPairs* self = static_cast<ContextDocumentsProxyPairs*>(lua_touserdata(L, lua_upvalueindex(1)));
		Document* doc = nullptr;
		int num_docs = obj->owner->GetNumDocuments();
		// because there can be missing indexes, make sure to continue until there
		// is actually a document at the index
		while (self->m_cur < num_docs)
		{
			doc = obj->owner->GetDocument(self->m_cur++);
			if (doc != nullptr)
				break;
		}
		if (doc == nullptr)
		{
			return 0;
		}
		lua_pushstring(L, doc->GetId().c_str());
		LuaType<Document>::push(L, doc);
		return 2;
	}
	static int destroy(lua_State* L)
	{
		static_cast<ContextDocumentsProxyPairs*>(lua_touserdata(L, 1))->~ContextDocumentsProxyPairs();
		return 0;
	}
	static int constructor(lua_State* L)
	{
		void* storage = lua_newuserdata(L, sizeof(ContextDocumentsProxyPairs));
		if (luaL_newmetatable(L, "RmlUi::Lua::ContextDocumentsProxyPairs"))
		{
			static luaL_Reg mt[] = {
				{"__gc", destroy},
				{NULL, NULL},
			};
			luaL_setfuncs(L, mt, 0);
		}
		lua_setmetatable(L, -2);
		lua_pushcclosure(L, next, 1);
		new (storage) ContextDocumentsProxyPairs();
		return 1;
	}
	ContextDocumentsProxyPairs() : m_cur(0) {}
	int m_cur;
};

int ContextDocumentsProxy__pairs(lua_State* L)
{
	ContextDocumentsProxyPairs::constructor(L);
	lua_pushvalue(L, 1);
	return 2;
}

RegType<ContextDocumentsProxy> ContextDocumentsProxyMethods[] = {
	{nullptr, nullptr},
};

luaL_Reg ContextDocumentsProxyGetters[] = {
	{nullptr, nullptr},
};

luaL_Reg ContextDocumentsProxySetters[] = {
	{nullptr, nullptr},
};

RMLUI_LUATYPE_DEFINE(ContextDocumentsProxy)

} // namespace Lua
} // namespace Rml