From 7997eff27f5d13c9dba2b9cd05457c3361f1b949 Mon Sep 17 00:00:00 2001
From: Martin Fuchs <fuchs.martin@gmail.com>
Date: Sun, 1 May 2005 23:19:48 +0000
Subject: [PATCH] implement owner drawn context menus for winefile

svn path=/trunk/; revision=14940
---
 reactos/subsys/system/winefile/winefile.c | 88 ++++++++++++++++++++++-
 1 file changed, 87 insertions(+), 1 deletion(-)

diff --git a/reactos/subsys/system/winefile/winefile.c b/reactos/subsys/system/winefile/winefile.c
index 013cdcacdba..78ccee11c09 100644
--- a/reactos/subsys/system/winefile/winefile.c
+++ b/reactos/subsys/system/winefile/winefile.c
@@ -3411,6 +3411,59 @@ static BOOL pane_command(Pane* pane, UINT cmd)
 }
 
 
+static IContextMenu2* s_pctxmenu2 = NULL;
+
+#ifndef __MINGW32__	// IContextMenu3 missing in MinGW (as of 6.2.2005)
+static IContextMenu3* s_pctxmenu3 = NULL;
+#endif
+
+static void CtxMenu_reset()
+{
+	s_pctxmenu2 = NULL;
+
+#ifndef __MINGW32__	// IContextMenu3 missing in MinGW (as of 6.2.2005)
+	s_pctxmenu3 = NULL;
+#endif
+}
+
+IContextMenu* CtxMenu_query_interfaces(IContextMenu* pcm1)
+{
+	IContextMenu* pcm = NULL;
+
+	CtxMenu_reset();
+
+#ifndef __MINGW32__	// IContextMenu3 missing in MinGW (as of 6.2.2005)
+	if ((*pcm1->lpVtbl->QueryInterface)(pcm1, &IID_IContextMenu3, (void**)&pcm) == NOERROR)
+		s_pctxmenu3 = (LPCONTEXTMENU3)pcm;
+	else
+#endif
+	if ((*pcm1->lpVtbl->QueryInterface)(pcm1, &IID_IContextMenu2, (void**)&pcm) == NOERROR)
+		s_pctxmenu2 = (LPCONTEXTMENU2)pcm;
+
+	if (pcm) {
+		(*pcm1->lpVtbl->Release)(pcm1);
+		return pcm;
+	} else
+		return pcm1;
+}
+
+static BOOL CtxMenu_HandleMenuMsg(UINT nmsg, WPARAM wparam, LPARAM lparam)
+{
+#ifndef __MINGW32__	// IContextMenu3 missing in MinGW (as of 6.2.2005)
+	if (s_pctxmenu3) {
+		if (SUCCEEDED((*s_pctxmenu3->lpVtbl->HandleMenuMsg)(s_pctxmenu3, nmsg, wparam, lparam)))
+			return TRUE;
+	}
+#endif
+
+	if (s_pctxmenu2)
+		if (SUCCEEDED((*s_pctxmenu2->lpVtbl->HandleMenuMsg)(s_pctxmenu2, nmsg, wparam, lparam)))
+			return TRUE;
+
+	return FALSE;
+}
+
+
 static HRESULT ShellFolderContextMenu(IShellFolder* shell_folder, HWND hwndParent, int cidl, LPCITEMIDLIST* apidl, int x, int y)
 {
 	IContextMenu* pcm;
@@ -3421,12 +3474,16 @@ static HRESULT ShellFolderContextMenu(IShellFolder* shell_folder, HWND hwndParen
 	if (SUCCEEDED(hr)) {
 		HMENU hmenu = CreatePopupMenu();
 
+		pcm = CtxMenu_query_interfaces(pcm);
+
 		if (hmenu) {
 			hr = (*pcm->lpVtbl->QueryContextMenu)(pcm, hmenu, 0, FCIDM_SHVIEWFIRST, FCIDM_SHVIEWLAST, CMF_NORMAL);
 
 			if (SUCCEEDED(hr)) {
 				UINT idCmd = TrackPopupMenu(hmenu, TPM_LEFTALIGN|TPM_RETURNCMD|TPM_RIGHTBUTTON, x, y, 0, hwndParent, NULL);
 
+				CtxMenu_reset();
+
 				if (idCmd) {
 				  CMINVOKECOMMANDINFO cmi;
 
@@ -3466,8 +3523,10 @@ LRESULT CALLBACK ChildWndProc(HWND hwnd, UINT nmsg, WPARAM wparam, LPARAM lparam
 
 			if (dis->CtlID == IDW_TREE_LEFT)
 				draw_item(&child->left, dis, entry, -1);
-			else
+			else if (dis->CtlID == IDW_TREE_RIGHT)
 				draw_item(&child->right, dis, entry, -1);
+			else
+				goto draw_menu_item;
 
 			return TRUE;}
 
@@ -3746,6 +3805,33 @@ LRESULT CALLBACK ChildWndProc(HWND hwnd, UINT nmsg, WPARAM wparam, LPARAM lparam
 			break;}
 #endif
 
+		  case WM_MEASUREITEM:
+		  draw_menu_item:
+			if (!wparam)	// Is the message menu-related?
+				if (CtxMenu_HandleMenuMsg(nmsg, wparam, lparam))
+					return TRUE;
+
+			break;
+
+		  case WM_INITMENUPOPUP:
+			if (CtxMenu_HandleMenuMsg(nmsg, wparam, lparam))
+				return 0;
+
+			break;
+
+#ifndef __MINGW32__	// IContextMenu3 missing in MinGW (as of 6.2.2005)
+		  case WM_MENUCHAR:	// only supported by IContextMenu3
+		   if (s_pctxmenu3) {
+			   LRESULT lResult = 0;
+
+			   (*s_pctxmenu3->lpVtbl->HandleMenuMsg2)(s_pctxmenu3, nmsg, wparam, lparam, &lResult);
+
+			   return lResult;
+		   }
+
+		   break;
+#endif
+
 		case WM_SIZE:
 			if (wparam != SIZE_MINIMIZED)
 				resize_tree(child, LOWORD(lparam), HIWORD(lparam));