#include "Document.h"
#include "Element.h"
#include <RmlUi/Core/Context.h>
#include <RmlUi/Core/ElementDocument.h>
#include <RmlUi/Lua/Utilities.h>

namespace Rml {
namespace Lua {

template <>
void ExtraInit<Document>(lua_State* L, int metatable_index)
{
	// we will inherit from Element
	ExtraInit<Element>(L, metatable_index);
	LuaType<Element>::_regfunctions(L, metatable_index, metatable_index - 1);
	AddTypeToElementAsTable<Document>(L);

	// create the DocumentFocus table
	lua_getglobal(L, "DocumentModal");
	if (lua_isnoneornil(L, -1))
	{
		lua_pop(L, 1);   // pop unsucessful getglobal
		lua_newtable(L); // create a table for holding the enum
		lua_pushinteger(L, (int)ModalFlag::None);
		lua_setfield(L, -2, "None");
		lua_pushinteger(L, (int)ModalFlag::Modal);
		lua_setfield(L, -2, "Modal");
		lua_pushinteger(L, (int)ModalFlag::Keep);
		lua_setfield(L, -2, "Keep");
		lua_setglobal(L, "DocumentModal");
	}

	// create the DocumentFocus table
	lua_getglobal(L, "DocumentFocus");
	if (lua_isnoneornil(L, -1))
	{
		lua_pop(L, 1);   // pop unsucessful getglobal
		lua_newtable(L); // create a table for holding the enum
		lua_pushinteger(L, (int)FocusFlag::None);
		lua_setfield(L, -2, "None");
		lua_pushinteger(L, (int)FocusFlag::Document);
		lua_setfield(L, -2, "Document");
		lua_pushinteger(L, (int)FocusFlag::Keep);
		lua_setfield(L, -2, "Keep");
		lua_pushinteger(L, (int)FocusFlag::Auto);
		lua_setfield(L, -2, "Auto");
		lua_setglobal(L, "DocumentFocus");
	}
}

// methods
int DocumentPullToFront(lua_State* /*L*/, Document* obj)
{
	obj->PullToFront();
	return 0;
}

int DocumentPushToBack(lua_State* /*L*/, Document* obj)
{
	obj->PushToBack();
	return 0;
}

int DocumentShow(lua_State* L, Document* obj)
{
	int top = lua_gettop(L);
	if (top == 0)
		obj->Show();
	else if (top == 1)
	{
		ModalFlag modal = (ModalFlag)luaL_checkinteger(L, 1);
		obj->Show(modal);
	}
	else
	{
		ModalFlag modal = (ModalFlag)luaL_checkinteger(L, 1);
		FocusFlag focus = (FocusFlag)luaL_checkinteger(L, 2);
		obj->Show(modal, focus);
	}
	return 0;
}

int DocumentHide(lua_State* /*L*/, Document* obj)
{
	obj->Hide();
	return 0;
}

int DocumentClose(lua_State* /*L*/, Document* obj)
{
	obj->Close();
	return 0;
}

int DocumentCreateElement(lua_State* L, Document* obj)
{
	const char* tag = luaL_checkstring(L, 1);
	ElementPtr* ele = new ElementPtr(obj->CreateElement(tag));
	LuaType<ElementPtr>::push(L, ele, true);
	return 1;
}

int DocumentCreateTextNode(lua_State* L, Document* obj)
{
	// need ElementText object first
	const char* text = luaL_checkstring(L, 1);
	ElementPtr* et = new ElementPtr(obj->CreateTextNode(text));
	LuaType<ElementPtr>::push(L, et, true);
	return 1;
}

// getters
int DocumentGetAttrtitle(lua_State* L)
{
	Document* doc = LuaType<Document>::check(L, 1);
	RMLUI_CHECK_OBJ(doc);
	lua_pushstring(L, doc->GetTitle().c_str());
	return 1;
}

int DocumentGetAttrcontext(lua_State* L)
{
	Document* doc = LuaType<Document>::check(L, 1);
	RMLUI_CHECK_OBJ(doc);
	LuaType<Context>::push(L, doc->GetContext(), false);
	return 1;
}

// setters
int DocumentSetAttrtitle(lua_State* L)
{
	Document* doc = LuaType<Document>::check(L, 1);
	RMLUI_CHECK_OBJ(doc);
	const char* title = luaL_checkstring(L, 2);
	doc->SetTitle(title);
	return 0;
}

RegType<Document> DocumentMethods[] = {
	RMLUI_LUAMETHOD(Document, PullToFront),
	RMLUI_LUAMETHOD(Document, PushToBack),
	RMLUI_LUAMETHOD(Document, Show),
	RMLUI_LUAMETHOD(Document, Hide),
	RMLUI_LUAMETHOD(Document, Close),
	RMLUI_LUAMETHOD(Document, CreateElement),
	RMLUI_LUAMETHOD(Document, CreateTextNode),
	{nullptr, nullptr},
};

luaL_Reg DocumentGetters[] = {
	RMLUI_LUAGETTER(Document, title),
	RMLUI_LUAGETTER(Document, context),
	{nullptr, nullptr},
};

luaL_Reg DocumentSetters[] = {
	RMLUI_LUASETTER(Document, title),
	{nullptr, nullptr},
};

RMLUI_LUATYPE_DEFINE(Document)
} // namespace Lua
} // namespace Rml