#include "ElementStyleProxy.h"
#include <RmlUi/Core/Element.h>
#include <RmlUi/Core/PropertiesIteratorView.h>
#include <RmlUi/Core/Property.h>
#include <RmlUi/Core/PropertyDefinition.h>
#include <RmlUi/Lua/IncludeLua.h>
#include <RmlUi/Lua/LuaType.h>

namespace Rml {
namespace Lua {
template <>
void ExtraInit<ElementStyleProxy>(lua_State* L, int metatable_index)
{
	lua_pushcfunction(L, ElementStyleProxy__index);
	lua_setfield(L, metatable_index, "__index");

	lua_pushcfunction(L, ElementStyleProxy__newindex);
	lua_setfield(L, metatable_index, "__newindex");

	lua_pushcfunction(L, ElementStyleProxy__pairs);
	lua_setfield(L, metatable_index, "__pairs");
}

int ElementStyleProxy__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 keytype = lua_type(L, 2);
	if (keytype == LUA_TSTRING) // if we are trying to access a string, then we will assume that it is a property
	{
		ElementStyleProxy* es = LuaType<ElementStyleProxy>::check(L, 1);
		RMLUI_CHECK_OBJ(es);
		const Property* prop = es->owner->GetProperty(lua_tostring(L, 2));
		RMLUI_CHECK_OBJ(prop)
		lua_pushstring(L, prop->ToString().c_str());
		return 1;
	}
	else // if it wasn't trying to get a string
	{
		lua_settop(L, 2);
		return LuaType<ElementStyleProxy>::index(L);
	}
}

int ElementStyleProxy__newindex(lua_State* L)
{
	//[1] = obj, [2] = key, [3] = value
	ElementStyleProxy* es = LuaType<ElementStyleProxy>::check(L, 1);
	RMLUI_CHECK_OBJ(es);
	int keytype = lua_type(L, 2);
	int valuetype = lua_type(L, 3);
	if (keytype == LUA_TSTRING)
	{
		const char* key = lua_tostring(L, 2);
		if (valuetype == LUA_TSTRING)
		{
			const char* value = lua_tostring(L, 3);
			lua_pushboolean(L, es->owner->SetProperty(key, value));
			return 1;
		}
		else if (valuetype == LUA_TNIL)
		{
			es->owner->RemoveProperty(key);
			return 0;
		}
	}
	// everything else returns when it needs to, so we are safe to pass it
	// on if needed

	lua_settop(L, 3);
	return LuaType<ElementStyleProxy>::newindex(L);
}

struct ElementStyleProxyPairs {
	static int next(lua_State* L)
	{
		ElementStyleProxyPairs* self = static_cast<ElementStyleProxyPairs*>(lua_touserdata(L, lua_upvalueindex(1)));
		if (self->m_view.AtEnd())
		{
			return 0;
		}
		const String& key = self->m_view.GetName();
		const Property& property = self->m_view.GetProperty();
		String val;
		property.definition->GetValue(val, property);
		lua_pushlstring(L, key.c_str(), key.size());
		lua_pushlstring(L, val.c_str(), val.size());
		++self->m_view;
		return 2;
	}
	static int destroy(lua_State* L)
	{
		static_cast<ElementStyleProxyPairs*>(lua_touserdata(L, 1))->~ElementStyleProxyPairs();
		return 0;
	}
	static int constructor(lua_State* L, ElementStyleProxy* obj)
	{
		void* storage = lua_newuserdata(L, sizeof(ElementStyleProxyPairs));
		if (luaL_newmetatable(L, "RmlUi::Lua::ElementStyleProxyPairs"))
		{
			static luaL_Reg mt[] = {
				{"__gc", destroy},
				{NULL, NULL},
			};
			luaL_setfuncs(L, mt, 0);
		}
		lua_setmetatable(L, -2);
		lua_pushcclosure(L, next, 1);
		new (storage) ElementStyleProxyPairs(obj);
		return 1;
	}
	ElementStyleProxyPairs(ElementStyleProxy* obj) : m_view(obj->owner->IterateLocalProperties()) {}
	PropertiesIteratorView m_view;
};

int ElementStyleProxy__pairs(lua_State* L)
{
	ElementStyleProxy* obj = LuaType<ElementStyleProxy>::check(L, 1);
	RMLUI_CHECK_OBJ(obj);
	ElementStyleProxyPairs::constructor(L, obj);
	return 1;
}

RegType<ElementStyleProxy> ElementStyleProxyMethods[] = {
	{nullptr, nullptr},
};

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

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

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