#include "stdafx.h"

static int clist_AddMainMenuItem(lua_State *L)
{
	HGENMENU res = NULL;
	
	if (lua_type(L, 1) == LUA_TSTRING)
	{
		const char *name = luaL_checkstring(L, 1);
		int position = lua_tointeger(L, 2);
		HANDLE hIcon = (HANDLE)lua_touserdata(L, 3);
		res = Menu_CreateRoot(MO_MAIN, ptrT(Utf8DecodeT(name)), position, hIcon);
	}
	else if (lua_type(L, 1) == LUA_TTABLE)
	{
		CMenuItem mi;
		MakeMenuItem(L, mi);
		res = Menu_AddMainMenuItem(&mi);
	}

	lua_pushlightuserdata(L, res);

	return 1;
}

static int clist_AddContactMenuItem(lua_State *L)
{
	HGENMENU res = NULL;

	if (lua_type(L, 1) == LUA_TSTRING)
	{
		const char *name = luaL_checkstring(L, 1);
		int position = lua_tointeger(L, 2);
		HANDLE hIcon = (HANDLE)lua_touserdata(L, 3);
		res = Menu_CreateRoot(MO_MAIN, ptrT(Utf8DecodeT(name)), position, hIcon);
	}
	else if (lua_type(L, 1) == LUA_TTABLE)
	{
		CMenuItem mi;
		MakeMenuItem(L, mi);
		ptrA szProto(mir_utf8decode((char*)lua_tostring(L, 2), NULL));
		res = Menu_AddContactMenuItem(&mi, szProto);
	}

	lua_pushlightuserdata(L, res);

	return 1;
}

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

	CMenuItem mi;
	MakeMenuItem(L, mi);

	HGENMENU res = Menu_AddTrayMenuItem(&mi);
	lua_pushlightuserdata(L, res);

	return 1;
}

static luaL_Reg clistApi[] =
{
	{ "AddMainMenuItem", clist_AddMainMenuItem },
	{ "AddContactMenuItem", clist_AddContactMenuItem },
	{ "AddTrayMenuItem", clist_AddTrayMenuItem },

	{ NULL, NULL }
};

LUAMOD_API int luaopen_m_clist(lua_State *L)
{
	luaL_newlib(L, clistApi);

	return 1;
}