summaryrefslogtreecommitdiff
path: root/plugins/MirLua/src/m_toptoolbar.cpp
blob: 0a17afa9cd0edf4b1dd84f93489b857fc256aa25 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#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;
}