#include "stdafx.h"

static TTBButton* MakeTBButton(lua_State *L)
{
	TTBButton *tbb = (TTBButton*)mir_calloc(sizeof(TTBButton));

	lua_pushliteral(L, "Name");
	lua_gettable(L, -2);
	tbb->name = mir_utf8decode((char*)luaL_checkstring(L, -1), NULL);
	lua_pop(L, 1);

	lua_pushliteral(L, "Service");
	lua_gettable(L, -2);
	tbb->pszService = (char*)lua_tostring(L, -1);
	lua_pop(L, 1);

	lua_pushliteral(L, "Flags");
	lua_gettable(L, -2);
	tbb->dwFlags = lua_tointeger(L, -1);
	lua_pop(L, 1);

	// up state
	lua_pushliteral(L, "IconUp");
	lua_gettable(L, -2);
	tbb->hIconHandleUp = (HANDLE)lua_touserdata(L, -1);
	lua_pop(L, 1);

	lua_pushliteral(L, "TooltipUp");
	lua_gettable(L, -2);
	tbb->pszTooltipUp = mir_utf8decode((char*)lua_tostring(L, -1), NULL);
	lua_pop(L, 1);

	lua_pushliteral(L, "wParamUp");
	lua_gettable(L, -2);
	tbb->wParamUp = luaM_towparam(L, -1);
	lua_pop(L, 1);

	lua_pushliteral(L, "lParamUp");
	lua_gettable(L, -2);
	tbb->lParamUp = luaM_tolparam(L, -1);
	lua_pop(L, 1);

	// dn state
	lua_pushliteral(L, "IconDown");
	lua_gettable(L, -2);
	tbb->hIconHandleDn = (HANDLE)lua_touserdata(L, -1);
	lua_pop(L, 1);

	lua_pushliteral(L, "TooltipDown");
	lua_gettable(L, -2);
	tbb->pszTooltipDn = mir_utf8decode((char*)lua_tostring(L, -1), NULL);
	lua_pop(L, 1);

	lua_pushliteral(L, "wParamDown");
	lua_gettable(L, -2);
	tbb->wParamDown = luaM_towparam(L, -1);
	lua_pop(L, 1);

	lua_pushliteral(L, "lParamDown");
	lua_gettable(L, -2);
	tbb->lParamDown = luaM_tolparam(L, -1);
	lua_pop(L, 1);

	return tbb;
}

static int lua_AddButton(lua_State *L)
{
	if (lua_type(L, 1) != LUA_TTABLE)
	{
		lua_pushlightuserdata(L, 0);
		return 1;
	}

	mir_ptr<TTBButton> tbb(MakeTBButton(L));

	HANDLE res = ::TopToolbar_AddButton(tbb);
	lua_pushlightuserdata(L, res);

	return 1;
}

static int lua_RemoveButton(lua_State *L)
{
	HANDLE hTTButton = (HANDLE)lua_touserdata(L, 1);

	INT_PTR res = ::CallService(MS_TTB_REMOVEBUTTON, (WPARAM)hTTButton, 0);
	lua_pushinteger(L, res);

	return 1;
}

static int lua_OnTopToolBarLoaded(lua_State *L)
{
	if (!lua_isfunction(L, 1))
	{
		lua_pushlightuserdata(L, NULL);
		return 1;
	}

	lua_pushvalue(L, 1);
	int ref = luaL_ref(L, LUA_REGISTRYINDEX);

	HANDLE res = ::HookEventObjParam(ME_TTB_MODULELOADED, CMLua::HookEventObjParam, L, ref);
	lua_pushlightuserdata(L, res);

	CMLua::Hooks.insert(res);
	CMLua::HookRefs.insert(new HandleRefParam(L, res, ref));

	return 1;
}

static luaL_Reg toptoolbarApi[] =
{
	{ "AddButton", lua_AddButton },
	{ "RemoveButton", lua_RemoveButton },

	{ "OnTopToolBarLoaded", lua_OnTopToolBarLoaded },

	{ NULL, NULL }
};

LUAMOD_API int luaopen_m_toptoolbar(lua_State *L)
{
	luaL_newlib(L, toptoolbarApi);

	return 1;
}