/*

Miranda NG: the free IM client for Microsoft* Windows*

Copyright (c) 2012-14 Miranda NG project (http://miranda-ng.org),
Copyright (c) 2000-12 Miranda IM project,
all portions of this codebase are copyrighted to the people
listed in contributors.txt.

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
*/

#include "..\..\core\commonheaders.h"
#include "genmenu.h"

#define STR_SEPARATOR _T("-----------------------------------")

extern bool bIconsDisabled;
extern int DefaultImageListColorDepth;
void RebuildProtoMenus(int);

struct OrderData
{
	int dragging;
	HTREEITEM hDragItem;
	int iInitMenuValue;
};

typedef struct tagMenuItemOptData
{
	TCHAR *name;
	TCHAR *defname;
	char  *uniqname;

	int    pos;
	bool   bShow, bIsSelected;
	int    id;

	PMO_IntMenuItem pimi;
}
	MenuItemOptData, *lpMenuItemOptData;

static BOOL GetCurrentMenuObjectID(HWND hwndDlg, int* result)
{
	HWND hTree = GetDlgItem(hwndDlg, IDC_MENUOBJECTS);
	HTREEITEM hti = TreeView_GetSelection(hTree);
	if (hti == NULL)
		return FALSE;

	TVITEM tvi;
	tvi.mask = TVIF_HANDLE | TVIF_PARAM;
	tvi.hItem = hti;
	TreeView_GetItem(hTree, &tvi);
	*result = (int)tvi.lParam;
	return TRUE;
}

static int SaveTree(HWND hwndDlg)
{
	int MenuObjectId;
	if (!GetCurrentMenuObjectID(hwndDlg, &MenuObjectId))
		return 0;

	HWND hTree = GetDlgItem(hwndDlg, IDC_MENUITEMS);
	TCHAR idstr[100];

	TVITEM tvi;
	tvi.hItem = TreeView_GetRoot(hTree);
	tvi.cchTextMax = 99;
	tvi.mask = TVIF_TEXT | TVIF_PARAM | TVIF_HANDLE;
	tvi.pszText = idstr;

	int count = 0;
	int menupos = GetMenuObjbyId(MenuObjectId);
	if (menupos == -1)
		return -1;

	TIntMenuObject *pimo = g_menus[menupos];

	char MenuNameItems[256];
	mir_snprintf(MenuNameItems, sizeof(MenuNameItems), "%s_Items", pimo->pszName);
	int runtimepos = 100;

	while (tvi.hItem != NULL) {
		TreeView_GetItem(hTree, &tvi);
		MenuItemOptData *iod = (MenuItemOptData*)tvi.lParam;
		if (iod->pimi) {
			char menuItemName[256], DBString[300];
			GetMenuItemName(iod->pimi, menuItemName, sizeof(menuItemName));

			mir_snprintf(DBString, SIZEOF(DBString), "%s_visible", menuItemName);
			db_set_b(NULL, MenuNameItems, DBString, iod->bShow);

			mir_snprintf(DBString, SIZEOF(DBString), "%s_pos", menuItemName);
			db_set_dw(NULL, MenuNameItems, DBString, runtimepos);

			mir_snprintf(DBString, SIZEOF(DBString), "%s_name", menuItemName);
			if (lstrcmp(iod->name, iod->defname) != 0)
				db_set_ts(NULL, MenuNameItems, DBString, iod->name);
			else
				db_unset(NULL, MenuNameItems, DBString);

			runtimepos += 100;
		}

		if (iod->name && !_tcscmp(iod->name, STR_SEPARATOR) && iod->bShow)
			runtimepos += SEPARATORPOSITIONINTERVAL;

		tvi.hItem = TreeView_GetNextSibling(hTree, tvi.hItem);
		count++;
	}
	return 1;
}

static int BuildMenuObjectsTree(HWND hwndDlg)
{
	TVINSERTSTRUCT tvis;
	HWND hTree = GetDlgItem(hwndDlg, IDC_MENUOBJECTS);
	int i;

	tvis.hParent = NULL;
	tvis.hInsertAfter = TVI_LAST;
	tvis.item.mask = TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
	TreeView_DeleteAllItems(hTree);
	if (g_menus.getCount() == 0)
		return FALSE;

	for (i=0; i < g_menus.getCount(); i++) {
		TIntMenuObject *p = g_menus[i];
		if (p->id == (int)hStatusMenuObject  || !p->m_bUseUserDefinedItems)
			continue;

		tvis.item.lParam = (LPARAM)p->id;
		tvis.item.pszText = TranslateTS(p->ptszDisplayName);
		tvis.item.iImage = tvis.item.iSelectedImage = TRUE;
		TreeView_InsertItem(hTree, &tvis);
	}
	return 1;
}

static int sortfunc(const void *a, const void *b)
{
	lpMenuItemOptData *sd1, *sd2;
	sd1 = (lpMenuItemOptData *)a;
	sd2 = (lpMenuItemOptData *)b;
	if ((*sd1)->pos > (*sd2)->pos)
		return 1;

	if ((*sd1)->pos < (*sd2)->pos)
		return -1;

	return 0;
}

static int InsertSeparator(HWND hwndDlg)
{
	HWND hMenuTree = GetDlgItem(hwndDlg, IDC_MENUITEMS);
	HTREEITEM hti = TreeView_GetSelection(hMenuTree);
	if (hti == NULL)
		return 1;

	TVITEM tvi = {0};
	tvi.mask = TVIF_HANDLE | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM | TVIF_TEXT;
	tvi.hItem = hti;
	if (TreeView_GetItem(hMenuTree, &tvi) == FALSE)
		return 1;

	MenuItemOptData *PD = (MenuItemOptData*)mir_calloc(sizeof(MenuItemOptData));
	PD->id = -1;
	PD->name = mir_tstrdup(STR_SEPARATOR);
	PD->bShow = true;
	PD->pos = ((MenuItemOptData *)tvi.lParam)->pos-1;

	TVINSERTSTRUCT tvis = {0};
	tvis.item.lParam = (LPARAM)(PD);
	tvis.item.pszText = PD->name;
	tvis.item.iImage = tvis.item.iSelectedImage = PD->bShow;
	tvis.hInsertAfter = hti;
	tvis.item.mask = TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
	TreeView_InsertItem(hMenuTree, &tvis);
	return 1;
}

static void FreeTreeData(HWND hwndDlg)
{
	HTREEITEM hItem = TreeView_GetRoot( GetDlgItem(hwndDlg, IDC_MENUITEMS));
	while (hItem != NULL) {
		TVITEM tvi;
		tvi.mask = TVIF_HANDLE | TVIF_PARAM;
		tvi.hItem = hItem;
		TreeView_GetItem( GetDlgItem(hwndDlg, IDC_MENUITEMS), &tvi);
		{
			MenuItemOptData* O = (MenuItemOptData *)tvi.lParam;
			if (O->name) mir_free(O->name);
			if (O->defname) mir_free(O->defname);
			if (O->uniqname) mir_free(O->uniqname);
			mir_free(O);
		}

		tvi.lParam = 0;
		TreeView_SetItem( GetDlgItem(hwndDlg, IDC_MENUITEMS), &tvi);

		hItem = TreeView_GetNextSibling( GetDlgItem(hwndDlg, IDC_MENUITEMS), hItem);
	}
}

static int BuildTree(HWND hwndDlg, int MenuObjectId, BOOL bReread)
{
	char menuItemName[256], MenuNameItems[256];
	char buf[256];

	FreeTreeData(hwndDlg);
	TreeView_DeleteAllItems( GetDlgItem(hwndDlg, IDC_MENUITEMS));

	int menupos = GetMenuObjbyId(MenuObjectId);
	if (menupos == -1)
		return FALSE;

	TIntMenuObject* pimo = g_menus[menupos];
	if (pimo->m_items.first == NULL)
		return FALSE;

	mir_snprintf(MenuNameItems, sizeof(MenuNameItems), "%s_Items", pimo->pszName);

	int count = 0;
	{
		for (PMO_IntMenuItem p = pimo->m_items.first; p != NULL; p = p->next)
			if (p->mi.root == (HGENMENU)-1 || p->mi.root == NULL)
				count++;
	}

	lpMenuItemOptData *PDar = (lpMenuItemOptData*)mir_alloc(sizeof(lpMenuItemOptData)*count);

	count = 0;
	{
		for (PMO_IntMenuItem p = pimo->m_items.first; p != NULL; p = p->next) {
			if (p->mi.root != (HGENMENU)-1 && p->mi.root != NULL)
				continue;

			MenuItemOptData *PD = (MenuItemOptData*)mir_calloc(sizeof(MenuItemOptData));
			GetMenuItemName(p, menuItemName, sizeof(menuItemName));
			{
				DBVARIANT dbv;
				mir_snprintf(buf, SIZEOF(buf), "%s_name", menuItemName);

				if (!db_get_ts(NULL, MenuNameItems, buf, &dbv)) {
					PD->name = mir_tstrdup(dbv.ptszVal);
					db_free(&dbv);
				}
				else PD->name = mir_tstrdup(GetMenuItemText(p));
			}

			PD->pimi = p;
			PD->defname = mir_tstrdup(GetMenuItemText(p));

			mir_snprintf(buf, SIZEOF(buf), "%s_visible", menuItemName);
			PD->bShow = db_get_b(NULL, MenuNameItems, buf, 1) != 0;

			if (bReread) {
				mir_snprintf(buf, SIZEOF(buf), "%s_pos", menuItemName);
				PD->pos = db_get_dw(NULL, MenuNameItems, buf, 1);
			}
			else PD->pos = (PD->pimi) ? PD->pimi->originalPosition : 0;

			PD->id = p->iCommand;

			if (p->UniqName)
				PD->uniqname = mir_strdup(p->UniqName);

			PDar[ count ] = PD;
			count++;
		}
	}

	qsort(PDar, count, sizeof(lpMenuItemOptData), sortfunc);

	SendDlgItemMessage(hwndDlg, IDC_MENUITEMS, WM_SETREDRAW, FALSE, 0);
	int lastpos = 0;
	bool first = TRUE;

	TVINSERTSTRUCT tvis;
	tvis.hParent = NULL;
	tvis.hInsertAfter = TVI_LAST;
	tvis.item.mask = TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
	for (int i=0; i < count; i++) {
		if (PDar[i]->pos - lastpos >= SEPARATORPOSITIONINTERVAL) {
			MenuItemOptData *PD = (MenuItemOptData*)mir_calloc(sizeof(MenuItemOptData));
			PD->id = -1;
			PD->name = mir_tstrdup(STR_SEPARATOR);
			PD->pos = PDar[i]->pos - 1;
			PD->bShow = true;

			tvis.item.lParam = (LPARAM)PD;
			tvis.item.pszText = PD->name;
			tvis.item.iImage = tvis.item.iSelectedImage = PD->bShow;
			SendDlgItemMessage(hwndDlg, IDC_MENUITEMS, TVM_INSERTITEM, 0, (LPARAM)&tvis);
		}

		tvis.item.lParam = (LPARAM)PDar[i];
		tvis.item.pszText = PDar[i]->name;
		tvis.item.iImage = tvis.item.iSelectedImage = PDar[i]->bShow;

		HTREEITEM hti = (HTREEITEM)SendDlgItemMessage(hwndDlg, IDC_MENUITEMS, TVM_INSERTITEM, 0, (LPARAM)&tvis);
		if (first) {
			TreeView_SelectItem( GetDlgItem(hwndDlg, IDC_MENUITEMS), hti);
			first = FALSE;
		}

		lastpos = PDar[i]->pos;
	}

	SendDlgItemMessage(hwndDlg, IDC_MENUITEMS, WM_SETREDRAW, TRUE, 0);
	mir_free(PDar);
	ShowWindow( GetDlgItem(hwndDlg, IDC_NOTSUPPORTWARNING), (pimo->m_bUseUserDefinedItems) ? SW_HIDE : SW_SHOW);
	EnableWindow( GetDlgItem(hwndDlg, IDC_MENUITEMS), pimo->m_bUseUserDefinedItems);
	EnableWindow( GetDlgItem(hwndDlg, IDC_INSERTSEPARATOR), pimo->m_bUseUserDefinedItems);
	return 1;
}

static void RebuildCurrent(HWND hwndDlg)
{
	int MenuObjectID;
	if (GetCurrentMenuObjectID(hwndDlg, &MenuObjectID))
		BuildTree(hwndDlg, MenuObjectID, TRUE);
}

static void ResetMenuItems(HWND hwndDlg)
{
	int MenuObjectID;
	if (GetCurrentMenuObjectID(hwndDlg, &MenuObjectID))
		BuildTree(hwndDlg, MenuObjectID, FALSE);
}

static HTREEITEM MoveItemAbove(HWND hTreeWnd, HTREEITEM hItem, HTREEITEM hInsertAfter)
{
	TVITEM tvi = { 0 };
	tvi.mask = TVIF_HANDLE | TVIF_PARAM;
	tvi.hItem = hItem;
	if (!SendMessage(hTreeWnd, TVM_GETITEM, 0, (LPARAM)&tvi))
		return NULL;
	if (hItem && hInsertAfter) {
		TVINSERTSTRUCT tvis;
		TCHAR name[128];
		if (hItem == hInsertAfter)
			return hItem;

		tvis.item.mask = TVIF_HANDLE | TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
		tvis.item.stateMask = 0xFFFFFFFF;
		tvis.item.pszText = name;
		tvis.item.cchTextMax = sizeof(name);
		tvis.item.hItem = hItem;
		tvis.item.iImage = tvis.item.iSelectedImage = ((MenuItemOptData*)tvi.lParam)->bShow;
		if (!SendMessage(hTreeWnd, TVM_GETITEM, 0, (LPARAM)&tvis.item))
			return NULL;
		if (!TreeView_DeleteItem(hTreeWnd, hItem))
			return NULL;
		tvis.hParent = NULL;
		tvis.hInsertAfter = hInsertAfter;
		return TreeView_InsertItem(hTreeWnd, &tvis);
	}
	return NULL;
}

LRESULT CALLBACK LBTNDOWNProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	if (uMsg == WM_LBUTTONDOWN && !(GetKeyState(VK_CONTROL)&0x8000)) {

		TVHITTESTINFO hti;
		hti.pt.x = (short)LOWORD(lParam);
		hti.pt.y = (short)HIWORD(lParam);
		//		ClientToScreen(hwndDlg, &hti.pt);
		//		ScreenToClient( GetDlgItem(hwndDlg, IDC_MENUITEMS), &hti.pt);
		TreeView_HitTest(hwnd, &hti);
		if (hti.flags&TVHT_ONITEMLABEL) {
			/// LabelClicked Set/unset selection
			TVITEM tvi;
			HWND tvw = hwnd;
			tvi.mask = TVIF_HANDLE|TVIF_PARAM;
			tvi.hItem = hti.hItem;
			TreeView_GetItem(tvw, &tvi);

			if (!((MenuItemOptData *)tvi.lParam)->bIsSelected) { /* is not Selected*/
				// reset all selection except current
				HTREEITEM hit;
				hit = TreeView_GetRoot(tvw);
				if (hit)
					do {
						TVITEM tvi = {0};
						tvi.mask = TVIF_HANDLE|TVIF_PARAM;
						tvi.hItem = hit;
						TreeView_GetItem(tvw, &tvi);

						if (hti.hItem != hit)
							((MenuItemOptData *)tvi.lParam)->bIsSelected = false;
						else
							((MenuItemOptData *)tvi.lParam)->bIsSelected = true;
						TreeView_SetItem(tvw, &tvi);
					}
						while (hit = TreeView_GetNextSibling(tvw, hit));
			}
		}
	}

	return mir_callNextSubclass(hwnd, LBTNDOWNProc, uMsg, wParam, lParam);
}

static int handleCustomDraw(HWND hWndTreeView, LPNMTVCUSTOMDRAW pNMTVCD)
{
	if (pNMTVCD == NULL)
		return -1;

	switch (pNMTVCD->nmcd.dwDrawStage) {
	case CDDS_PREPAINT:
		return CDRF_NOTIFYITEMDRAW;

	case CDDS_ITEMPREPAINT:
		{
			HTREEITEM hItem = (HTREEITEM) pNMTVCD->nmcd.dwItemSpec;
			TCHAR buf[255];

			TVITEM tvi = {0};
			tvi.mask = TVIF_HANDLE |TVIF_PARAM|TVIS_SELECTED|TVIF_TEXT|TVIF_IMAGE;
			tvi.stateMask = TVIS_SELECTED;
			tvi.hItem = hItem;
			tvi.pszText = buf;
			tvi.cchTextMax = SIZEOF(buf);
			TreeView_GetItem(hWndTreeView, &tvi);
			if (((MenuItemOptData *)tvi.lParam)->bIsSelected) {
				pNMTVCD->clrTextBk = GetSysColor(COLOR_HIGHLIGHT);
				pNMTVCD->clrText = GetSysColor(COLOR_HIGHLIGHTTEXT);
			}
			else {
				pNMTVCD->clrTextBk = GetSysColor(COLOR_WINDOW);
				pNMTVCD->clrText = GetSysColor(COLOR_WINDOWTEXT);
			}

			/* At this point, you can change the background colors for the item
			and any subitems and return CDRF_NEWFONT. If the list-view control
			is in report mode, you can simply return CDRF_NOTIFYSUBITEMREDRAW
			to customize the item's subitems individually */
			int retVal = CDRF_NEWFONT;
			if (tvi.iImage == -1) {
				SIZE sz;
				GetTextExtentPoint32(pNMTVCD->nmcd.hdc, tvi.pszText, lstrlen(tvi.pszText), &sz);

				RECT rc;
				if (sz.cx+3 > pNMTVCD->nmcd.rc.right - pNMTVCD->nmcd.rc.left)
					rc = pNMTVCD->nmcd.rc;
				else
					SetRect(&rc, pNMTVCD->nmcd.rc.left, pNMTVCD->nmcd.rc.top, pNMTVCD->nmcd.rc.left+sz.cx+3, pNMTVCD->nmcd.rc.bottom);

				HBRUSH br = CreateSolidBrush(pNMTVCD->clrTextBk);
				SetTextColor(pNMTVCD->nmcd.hdc, pNMTVCD->clrText);
				SetBkColor(pNMTVCD->nmcd.hdc, pNMTVCD->clrTextBk);
				FillRect(pNMTVCD->nmcd.hdc, &rc, br);
				DeleteObject(br);
				DrawText(pNMTVCD->nmcd.hdc, tvi.pszText, lstrlen(tvi.pszText), &pNMTVCD->nmcd.rc, DT_LEFT|DT_VCENTER|DT_NOPREFIX);

				retVal |= CDRF_SKIPDEFAULT;
			}

			return retVal;
		}
	}
	return 0;
}

static void OnClickCheckbox(HWND hwndDlg, HWND hwndTree, HTREEITEM hItem)
{
	TVITEM tvi;
	tvi.mask = TVIF_HANDLE | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM;
	tvi.hItem = hItem;
	TreeView_GetItem(hwndTree, &tvi);

	tvi.iImage = tvi.iSelectedImage = !tvi.iImage;
	((MenuItemOptData *)tvi.lParam)->bShow = tvi.iImage != 0;
	TreeView_SetItem(hwndTree, &tvi);
	SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
}

static INT_PTR CALLBACK GenMenuOpts(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
	struct OrderData *dat = (struct OrderData*)GetWindowLongPtr( GetDlgItem(hwndDlg, IDC_MENUITEMS), GWLP_USERDATA);
	LPNMHDR hdr;

	switch (msg) {
	case WM_INITDIALOG:
		TranslateDialogDefault(hwndDlg);
		dat = (struct OrderData*)mir_alloc(sizeof(struct OrderData));
		SetWindowLongPtr( GetDlgItem(hwndDlg, IDC_MENUITEMS), GWLP_USERDATA, (LONG_PTR)dat);
		dat->dragging = 0;
		dat->iInitMenuValue = db_get_b(NULL, "CList", "MoveProtoMenus", TRUE);
		mir_subclassWindow( GetDlgItem(hwndDlg, IDC_MENUITEMS), LBTNDOWNProc);
		{
			HIMAGELIST himlCheckBoxes;
			himlCheckBoxes = ImageList_Create(GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON), ILC_COLOR32 | ILC_MASK, 2, 2);

			ImageList_AddIcon_IconLibLoaded(himlCheckBoxes, SKINICON_OTHER_NOTICK);
			ImageList_AddIcon_IconLibLoaded(himlCheckBoxes, SKINICON_OTHER_TICK);

			TreeView_SetImageList( GetDlgItem(hwndDlg, IDC_MENUOBJECTS), himlCheckBoxes, TVSIL_NORMAL);
			TreeView_SetImageList( GetDlgItem(hwndDlg, IDC_MENUITEMS), himlCheckBoxes, TVSIL_NORMAL);
		}
		CheckDlgButton(hwndDlg, dat->iInitMenuValue ? IDC_RADIO2 : IDC_RADIO1, TRUE);
		CheckDlgButton(hwndDlg, IDC_DISABLEMENUICONS, bIconsDisabled);
		BuildMenuObjectsTree(hwndDlg);
		return TRUE;

	case WM_COMMAND:
		if (HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DBLCLK) {
			switch (LOWORD(wParam)) {
			case IDC_INSERTSEPARATOR:
				InsertSeparator(hwndDlg);
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				break;

			case IDC_RESETMENU:
				ResetMenuItems(hwndDlg);
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				break;

			case IDC_DISABLEMENUICONS:
			case IDC_RADIO1:
			case IDC_RADIO2:
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				break;

			case IDC_GENMENU_DEFAULT:
				{
					TVITEM tvi;
					HTREEITEM hti;
					MenuItemOptData *iod;

					hti = TreeView_GetSelection( GetDlgItem(hwndDlg, IDC_MENUITEMS));
					if (hti == NULL)
						break;

					tvi.mask = TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
					tvi.hItem = hti;
					TreeView_GetItem( GetDlgItem(hwndDlg, IDC_MENUITEMS), &tvi);
					iod = (MenuItemOptData *)tvi.lParam;

					if (iod->name && _tcsstr(iod->name, STR_SEPARATOR))
						break;

					if (iod->name)
						mir_free(iod->name);
					iod->name = mir_tstrdup(iod->defname);

					SaveTree(hwndDlg);
					RebuildCurrent(hwndDlg);
					SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				}
				break;

			case IDC_GENMENU_SET:
				{
					TVITEM tvi;
					TCHAR buf[256];
					MenuItemOptData *iod;

					HTREEITEM hti = TreeView_GetSelection( GetDlgItem(hwndDlg, IDC_MENUITEMS));
					if (hti == NULL)
						break;

					tvi.mask = TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;
					tvi.hItem = hti;
					SendDlgItemMessage(hwndDlg, IDC_MENUITEMS, TVM_GETITEM, 0, (LPARAM)&tvi);
					iod = (MenuItemOptData *)tvi.lParam;

					if (iod->name && _tcsstr(iod->name, STR_SEPARATOR))
						break;

					ZeroMemory(buf, sizeof(buf));
					GetDlgItemText(hwndDlg, IDC_GENMENU_CUSTOMNAME, buf, SIZEOF(buf));
					if (iod->name)
						mir_free(iod->name);

					iod->name = mir_tstrdup(buf);

					SaveTree(hwndDlg);
					RebuildCurrent(hwndDlg);
					SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				}
				break;
			}
		}
		break;

	case WM_NOTIFY:
		hdr = (LPNMHDR)lParam;
		switch(hdr->idFrom) {
		case 0:
			if (hdr->code == PSN_APPLY) {
				bIconsDisabled = IsDlgButtonChecked(hwndDlg, IDC_DISABLEMENUICONS) != 0;
				db_set_b(NULL, "CList", "DisableMenuIcons", bIconsDisabled);
				SaveTree(hwndDlg);
				int iNewMenuValue = IsDlgButtonChecked(hwndDlg, IDC_RADIO1) ? 0 : 1;
				if (iNewMenuValue != dat->iInitMenuValue) {
					RebuildProtoMenus(iNewMenuValue);
					dat->iInitMenuValue = iNewMenuValue;
				}
				RebuildCurrent(hwndDlg);
			}
			break;

		case IDC_MENUOBJECTS:
			if (hdr->code == TVN_SELCHANGEDA)
				RebuildCurrent(hwndDlg);
			break;

		case IDC_MENUITEMS:
			switch (hdr->code) {
			case NM_CUSTOMDRAW:
				SetWindowLongPtr(hwndDlg, DWLP_MSGRESULT, handleCustomDraw(GetDlgItem(hwndDlg, IDC_MENUITEMS), (LPNMTVCUSTOMDRAW)lParam));
				return TRUE;

			case TVN_BEGINDRAGA:
				SetCapture(hwndDlg);
				dat->dragging = 1;
				dat->hDragItem = ((LPNMTREEVIEW)lParam)->itemNew.hItem;
				TreeView_SelectItem(GetDlgItem(hwndDlg, IDC_MENUITEMS), dat->hDragItem);
				break;

			case TVN_KEYDOWN:
				if (((LPNMLVKEYDOWN)lParam)->wVKey == VK_SPACE)
					OnClickCheckbox(hwndDlg, hdr->hwndFrom, TreeView_GetSelection(hdr->hwndFrom));
				break;

			case NM_CLICK:
				{
					TVHITTESTINFO hti;
					hti.pt.x = (short)LOWORD(GetMessagePos());
					hti.pt.y = (short)HIWORD(GetMessagePos());
					ScreenToClient(hdr->hwndFrom, &hti.pt);
					if (TreeView_HitTest(hdr->hwndFrom, &hti)) {
						if (hti.flags & TVHT_ONITEMICON)
							OnClickCheckbox(hwndDlg, hdr->hwndFrom, hti.hItem);

						/*--------MultiSelection----------*/
						if (hti.flags & TVHT_ONITEMLABEL) {
							/// LabelClicked Set/unset selection
							TVITEM tvi;
							HWND tvw = hdr->hwndFrom;
							tvi.mask = TVIF_HANDLE|TVIF_PARAM;
							tvi.hItem = hti.hItem;
							TreeView_GetItem(tvw, &tvi);
							if (GetKeyState(VK_CONTROL) & 0x8000) {
								MenuItemOptData *iod = (MenuItemOptData*)tvi.lParam;
								iod->bIsSelected = !iod->bIsSelected;
								TreeView_SetItem(tvw, &tvi);
							}
							else if (GetKeyState(VK_SHIFT)&0x8000) {
								;  // shifted click
							}
							else {
								// reset all selection except current
								HTREEITEM hit = TreeView_GetRoot(tvw);
								if (!hit)
									break;

								do {
									TVITEM tvi = {0};
									tvi.mask = TVIF_HANDLE|TVIF_PARAM;
									tvi.hItem = hit;
									TreeView_GetItem(tvw, &tvi);

									MenuItemOptData *iod = (MenuItemOptData*)tvi.lParam;
									iod->bIsSelected = (hti.hItem == hit);
									TreeView_SetItem(tvw, &tvi);
								}
									while (hit = TreeView_GetNextSibling(tvw, hit));
							}
						}
					}
				}
				break;

			case TVN_SELCHANGEDA:
				SetDlgItemTextA(hwndDlg, IDC_GENMENU_CUSTOMNAME, "");
				SetDlgItemTextA(hwndDlg, IDC_GENMENU_SERVICE, "");

				EnableWindow( GetDlgItem(hwndDlg, IDC_GENMENU_CUSTOMNAME), FALSE);
				EnableWindow( GetDlgItem(hwndDlg, IDC_GENMENU_DEFAULT), FALSE);
				EnableWindow( GetDlgItem(hwndDlg, IDC_GENMENU_SET), FALSE);

				HTREEITEM hti = TreeView_GetSelection(GetDlgItem(hwndDlg, IDC_MENUITEMS));
				if (hti == NULL)
					break;

				TVITEM tvi;
				tvi.mask = TVIF_HANDLE | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM;
				tvi.hItem = hti;
				TreeView_GetItem( GetDlgItem(hwndDlg, IDC_MENUITEMS), &tvi);

				if (tvi.lParam == 0)
					break;

				MenuItemOptData *iod = (MenuItemOptData *)tvi.lParam;
				if (iod->name && _tcsstr(iod->name, STR_SEPARATOR))
					break;

				SetDlgItemText(hwndDlg, IDC_GENMENU_CUSTOMNAME, iod->name);

				if (iod->pimi->submenu.first == NULL && iod->uniqname)
					SetDlgItemTextA(hwndDlg, IDC_GENMENU_SERVICE, iod->uniqname);

				EnableWindow( GetDlgItem(hwndDlg, IDC_GENMENU_DEFAULT), lstrcmp(iod->name, iod->defname) != 0);
				EnableWindow( GetDlgItem(hwndDlg, IDC_GENMENU_SET), TRUE);
				EnableWindow( GetDlgItem(hwndDlg, IDC_GENMENU_CUSTOMNAME), TRUE);
				break;
			}
		}
		break;

	case WM_MOUSEMOVE:
		if (!dat || !dat->dragging) break;
		{
			TVHITTESTINFO hti;

			hti.pt.x = (short)LOWORD(lParam);
			hti.pt.y = (short)HIWORD(lParam);
			ClientToScreen(hwndDlg, &hti.pt);
			ScreenToClient( GetDlgItem(hwndDlg, IDC_MENUITEMS), &hti.pt);
			TreeView_HitTest( GetDlgItem(hwndDlg, IDC_MENUITEMS), &hti);
			if (hti.flags&(TVHT_ONITEM|TVHT_ONITEMRIGHT)) {
				HTREEITEM it = hti.hItem;
				hti.pt.y -= TreeView_GetItemHeight( GetDlgItem(hwndDlg, IDC_MENUITEMS))/2;
				TreeView_HitTest( GetDlgItem(hwndDlg, IDC_MENUITEMS), &hti);
				if (!(hti.flags&TVHT_ABOVE))
					TreeView_SetInsertMark( GetDlgItem(hwndDlg, IDC_MENUITEMS), hti.hItem, 1);
				else
					TreeView_SetInsertMark( GetDlgItem(hwndDlg, IDC_MENUITEMS), it, 0);
			}
			else {
				if (hti.flags&TVHT_ABOVE) SendDlgItemMessage(hwndDlg, IDC_MENUITEMS, WM_VSCROLL, MAKEWPARAM(SB_LINEUP, 0), 0);
				if (hti.flags&TVHT_BELOW) SendDlgItemMessage(hwndDlg, IDC_MENUITEMS, WM_VSCROLL, MAKEWPARAM(SB_LINEDOWN, 0), 0);
				TreeView_SetInsertMark( GetDlgItem(hwndDlg, IDC_MENUITEMS), NULL, 0);
			}
		}
		break;

	case WM_LBUTTONUP:
		if (!dat->dragging)
			break;

		TreeView_SetInsertMark( GetDlgItem(hwndDlg, IDC_MENUITEMS), NULL, 0);
		dat->dragging = 0;
		ReleaseCapture();
		{
			TVHITTESTINFO hti;
			hti.pt.x = (short)LOWORD(lParam);
			hti.pt.y = (short)HIWORD(lParam);
			ClientToScreen(hwndDlg, &hti.pt);
			ScreenToClient( GetDlgItem(hwndDlg, IDC_MENUITEMS), &hti.pt);
			hti.pt.y-=TreeView_GetItemHeight( GetDlgItem(hwndDlg, IDC_MENUITEMS))/2;
			TreeView_HitTest( GetDlgItem(hwndDlg, IDC_MENUITEMS), &hti);
			if (hti.flags&TVHT_ABOVE) hti.hItem = TVI_FIRST;
			if (dat->hDragItem == hti.hItem) break;
			dat->hDragItem = NULL;
			if (hti.flags&(TVHT_ONITEM|TVHT_ONITEMRIGHT) || (hti.hItem == TVI_FIRST)) {
				HWND tvw;
				HTREEITEM * pSIT;
				HTREEITEM FirstItem = NULL;
				UINT uITCnt, uSic;
				tvw = GetDlgItem(hwndDlg, IDC_MENUITEMS);
				uITCnt = TreeView_GetCount(tvw);
				uSic = 0;
				if (uITCnt) {
					pSIT = (HTREEITEM *)mir_alloc(sizeof(HTREEITEM)*uITCnt);
					if (pSIT) {
						HTREEITEM hit;
						hit = TreeView_GetRoot(tvw);
						if (hit)
							do {
								TVITEM tvi = {0};
								tvi.mask = TVIF_HANDLE|TVIF_PARAM;
								tvi.hItem = hit;
								TreeView_GetItem(tvw, &tvi);
								if (((MenuItemOptData *)tvi.lParam)->bIsSelected) {
									pSIT[uSic] = tvi.hItem;
									uSic++;
								}
							}while (hit = TreeView_GetNextSibling(tvw, hit));
						// Proceed moving
						{
							UINT i;
							HTREEITEM insertAfter;
							insertAfter = hti.hItem;
							for (i=0; i<uSic; i++) {
								if (insertAfter) insertAfter = MoveItemAbove(tvw, pSIT[i], insertAfter);
								else break;
								if (!i) FirstItem = insertAfter;
							}
						}
						// free pointers...
						mir_free(pSIT);
					}
				}

				if (FirstItem) TreeView_SelectItem(tvw, FirstItem);
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				SaveTree(hwndDlg);
			}
		}
		break;

	case WM_DESTROY:
		if (dat)
			mir_free(dat);

		ImageList_Destroy(TreeView_SetImageList( GetDlgItem(hwndDlg, IDC_MENUOBJECTS), NULL, TVSIL_NORMAL));
		FreeTreeData(hwndDlg);
		break;

	}
	return FALSE;
}

INT_PTR CALLBACK ProtocolOrderOpts(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam);

int GenMenuOptInit(WPARAM wParam, LPARAM)
{
	OPTIONSDIALOGPAGE odp = { sizeof(odp) };
	odp.hInstance = hInst;

	odp.position = -1000000000;
	odp.pszTemplate = MAKEINTRESOURCEA(IDD_OPT_GENMENU);
	odp.pszTitle = LPGEN("Menus");
	odp.pszGroup = LPGEN("Customize");
	odp.pfnDlgProc = GenMenuOpts;
	odp.flags = ODPF_BOLDGROUPS;
	Options_AddPage(wParam, &odp);

	odp.position = -10000000;
	odp.groupPosition = 1000000;
	odp.pszTemplate = MAKEINTRESOURCEA(IDD_OPT_PROTOCOLORDER);
	odp.pszTitle = LPGEN("Accounts");
	odp.pszGroup = LPGEN("Contact list");
	odp.pfnDlgProc = ProtocolOrderOpts;
	odp.flags = ODPF_BOLDGROUPS;
	Options_AddPage(wParam, &odp);
	return 0;
}