#include "stdafx.h"

int hLangpack;

static TTBButton* MakeTBButton(lua_State *L)
{
	TTBButton *tbb = (TTBButton*)mir_calloc(sizeof(TTBButton));
	tbb->dwFlags = TTBBF_ISLBUTTON;

	lua_getfield(L, -1, "Name");
	tbb->name = mir_utf8decodeA(luaL_checkstring(L, -1));
	lua_pop(L, 1);

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

	lua_getfield(L, -1, "Flags");
	tbb->dwFlags = lua_tointeger(L, -1);
	lua_pop(L, 1);

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

	lua_getfield(L, -1, "TooltipUp");
	tbb->pszTooltipUp = mir_utf8decodeA(lua_tostring(L, -1));
	lua_pop(L, 1);

	lua_getfield(L, -1, "wParamUp");
	tbb->wParamUp = (WPARAM)luaM_tomparam(L, -1);
	lua_pop(L, 1);

	lua_getfield(L, -1, "lParamUp");
	tbb->lParamUp = (LPARAM)luaM_tomparam(L, -1);
	lua_pop(L, 1);

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

	lua_getfield(L, -1, "TooltipDown");
	tbb->pszTooltipDn = mir_utf8decodeA(lua_tostring(L, -1));
	lua_pop(L, 1);

	lua_getfield(L, -1, "wParamDown");
	tbb->wParamDown = (WPARAM)luaM_tomparam(L, -1);
	lua_pop(L, 1);

	lua_getfield(L, -1, "lParamDown");
	tbb->lParamDown = (LPARAM)luaM_tomparam(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, nullptr);
		return 1;
	}

	TTBButton *tbb = MakeTBButton(L);

	HANDLE res = TopToolbar_AddButton(tbb);
	if (res == (HANDLE)-1)
	{
		lua_pushnil(L);
		return 1;
	}

	lua_pushlightuserdata(L, res);

	mir_free(tbb->name);
	mir_free(tbb->pszTooltipUp);
	mir_free(tbb->pszTooltipDn);
	mir_free(tbb);

	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 luaL_Reg toptoolbarApi[] =
{
	{ "AddButton", lua_AddButton },
	{ "RemoveButton", lua_RemoveButton },

	{ nullptr, nullptr }
};

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

	return 1;
}