static int last_split;
HBITMAP SizingPattern = 0;
HBRUSH SizingBrush = 0;
-static WCHAR Suggestions[256];
+WCHAR Suggestions[256];
extern LPCWSTR get_root_key_name(HKEY hRootKey)
{
extern void ResizeWnd(int cx, int cy)
{
- HDWP hdwp = BeginDeferWindowPos(3);
+ HDWP hdwp = BeginDeferWindowPos(4);
RECT rt, rs, rb;
- const int tHeight = 22;
+ const int nButtonWidth = 44;
+ const int nButtonHeight = 22;
+ int cyEdge = GetSystemMetrics(SM_CYEDGE);
+ const UINT uFlags = SWP_NOZORDER | SWP_NOACTIVATE;
SetRect(&rt, 0, 0, cx, cy);
cy = 0;
if (hStatusBar != NULL)
cy = rs.bottom - rs.top;
}
GetWindowRect(g_pChildWnd->hAddressBtnWnd, &rb);
- cx = g_pChildWnd->nSplitPos + SPLIT_WIDTH/2;
- DeferWindowPos(hdwp, g_pChildWnd->hAddressBarWnd, 0, rt.left, rt.top, rt.right-rt.left - 2*tHeight, tHeight, SWP_NOZORDER|SWP_NOACTIVATE);
- DeferWindowPos(hdwp, g_pChildWnd->hAddressBtnWnd, 0, rt.right - 2*tHeight, rt.top, 2*tHeight, tHeight, SWP_NOZORDER|SWP_NOACTIVATE);
- DeferWindowPos(hdwp, g_pChildWnd->hTreeWnd, 0, rt.left, rt.top + tHeight+2, g_pChildWnd->nSplitPos-SPLIT_WIDTH/2-rt.left, rt.bottom-rt.top-cy, SWP_NOZORDER|SWP_NOACTIVATE);
- DeferWindowPos(hdwp, g_pChildWnd->hListWnd, 0, rt.left+cx, rt.top + tHeight+2, rt.right-cx, rt.bottom-rt.top-cy, SWP_NOZORDER|SWP_NOACTIVATE);
- EndDeferWindowPos(hdwp);
+ cx = g_pChildWnd->nSplitPos + SPLIT_WIDTH / 2;
+ if (hdwp)
+ hdwp = DeferWindowPos(hdwp, g_pChildWnd->hAddressBarWnd, NULL,
+ rt.left, rt.top,
+ rt.right - rt.left - nButtonWidth, nButtonHeight,
+ uFlags);
+ if (hdwp)
+ hdwp = DeferWindowPos(hdwp, g_pChildWnd->hAddressBtnWnd, NULL,
+ rt.right - nButtonWidth, rt.top,
+ nButtonWidth, nButtonHeight,
+ uFlags);
+ if (hdwp)
+ hdwp = DeferWindowPos(hdwp, g_pChildWnd->hTreeWnd, NULL,
+ rt.left,
+ rt.top + nButtonHeight + cyEdge,
+ g_pChildWnd->nSplitPos - SPLIT_WIDTH/2 - rt.left,
+ rt.bottom - rt.top - cy - 2 * cyEdge,
+ uFlags);
+ if (hdwp)
+ hdwp = DeferWindowPos(hdwp, g_pChildWnd->hListWnd, NULL,
+ rt.left + cx,
+ rt.top + nButtonHeight + cyEdge,
+ rt.right - cx,
+ rt.bottom - rt.top - cy - 2 * cyEdge,
+ uFlags);
+ if (hdwp)
+ EndDeferWindowPos(hdwp);
}
/*******************************************************************************
ReleaseDC(hWnd, hdc);
}
-static void OnPaint(HWND hWnd)
-{
- PAINTSTRUCT ps;
- RECT rt;
-
- GetClientRect(hWnd, &rt);
- BeginPaint(hWnd, &ps);
- FillRect(ps.hdc, &rt, GetSysColorBrush(COLOR_BTNFACE));
- EndPaint(hWnd, &ps);
-}
-
/*******************************************************************************
* finish_splitbar [internal]
*
ReleaseCapture();
}
-/*******************************************************************************
- *
- * FUNCTION: _CmdWndProc(HWND, unsigned, WORD, LONG)
- *
- * PURPOSE: Processes WM_COMMAND messages for the main frame window.
- *
- */
-
-static BOOL _CmdWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
-{
- HTREEITEM hSelection;
- HKEY hRootKey;
- LPCWSTR keyPath, s;
- WORD wID = LOWORD(wParam);
-
- UNREFERENCED_PARAMETER(message);
-
- switch (wID)
- {
- /* Parse the menu selections: */
- case ID_REGISTRY_EXIT:
- DestroyWindow(hWnd);
- break;
- case ID_VIEW_REFRESH:
- /* TODO */
- break;
- case ID_TREE_EXPANDBRANCH:
- (void)TreeView_Expand(g_pChildWnd->hTreeWnd, TreeView_GetSelection(g_pChildWnd->hTreeWnd), TVE_EXPAND);
- break;
- case ID_TREE_COLLAPSEBRANCH:
- (void)TreeView_Expand(g_pChildWnd->hTreeWnd, TreeView_GetSelection(g_pChildWnd->hTreeWnd), TVE_COLLAPSE);
- break;
- case ID_TREE_RENAME:
- SetFocus(g_pChildWnd->hTreeWnd);
- (void)TreeView_EditLabel(g_pChildWnd->hTreeWnd, TreeView_GetSelection(g_pChildWnd->hTreeWnd));
- break;
- case ID_TREE_DELETE:
- hSelection = TreeView_GetSelection(g_pChildWnd->hTreeWnd);
- keyPath = GetItemPath(g_pChildWnd->hTreeWnd, hSelection, &hRootKey);
-
- if (keyPath == 0 || *keyPath == 0)
- {
- MessageBeep(MB_ICONHAND);
- }
- else if (DeleteKey(hWnd, hRootKey, keyPath))
- DeleteNode(g_pChildWnd->hTreeWnd, 0);
- break;
- case ID_TREE_EXPORT:
- ExportRegistryFile(g_pChildWnd->hTreeWnd);
- break;
- case ID_EDIT_FIND:
- FindDialog(hWnd);
- break;
- case ID_EDIT_COPYKEYNAME:
- hSelection = TreeView_GetSelection(g_pChildWnd->hTreeWnd);
- keyPath = GetItemPath(g_pChildWnd->hTreeWnd, hSelection, &hRootKey);
- CopyKeyName(hWnd, hRootKey, keyPath);
- break;
- case ID_EDIT_NEW_KEY:
- CreateNewKey(g_pChildWnd->hTreeWnd, TreeView_GetSelection(g_pChildWnd->hTreeWnd));
- break;
- case ID_EDIT_NEW_STRINGVALUE:
- case ID_EDIT_NEW_BINARYVALUE:
- case ID_EDIT_NEW_DWORDVALUE:
- SendMessageW(hFrameWnd, WM_COMMAND, wParam, lParam);
- break;
- case ID_SWITCH_PANELS:
- g_pChildWnd->nFocusPanel = !g_pChildWnd->nFocusPanel;
- SetFocus(g_pChildWnd->nFocusPanel? g_pChildWnd->hListWnd: g_pChildWnd->hTreeWnd);
- break;
- default:
- if ((wID >= ID_TREE_SUGGESTION_MIN) && (wID <= ID_TREE_SUGGESTION_MAX))
- {
- s = Suggestions;
- while(wID > ID_TREE_SUGGESTION_MIN)
- {
- if (*s)
- s += wcslen(s) + 1;
- wID--;
- }
- SelectNode(g_pChildWnd->hTreeWnd, s);
- break;
- }
- return FALSE;
- }
- return TRUE;
-}
-
/*******************************************************************************
*
* Key suggestion
/* Check default key */
if (QueryStringValue(hRootKey, pszKeyPath, NULL,
- szBuffer, COUNT_OF(szBuffer)) == ERROR_SUCCESS)
+ szBuffer, ARRAY_SIZE(szBuffer)) == ERROR_SUCCESS)
{
/* Sanity check this key; it cannot be empty, nor can it be a
* loop back */
RegCloseKey(hOtherKey);
bFound = TRUE;
- wcscpy(szLastFound, szBuffer);
+ StringCbCopyW(szLastFound, sizeof(szLastFound), szBuffer);
pszKeyPath = szLastFound;
}
}
if (RegOpenKeyW(hRootKey, pszKeyPath, &hSubKey) == ERROR_SUCCESS)
{
if (QueryStringValue(hSubKey, L"CLSID", NULL, szBuffer,
- COUNT_OF(szBuffer)) == ERROR_SUCCESS)
+ ARRAY_SIZE(szBuffer)) == ERROR_SUCCESS)
{
lstrcpynW(pszSuggestions, L"HKCR\\CLSID\\", (int)iSuggestionsLength);
i = wcslen(pszSuggestions);
{
WNDPROC oldwndproc;
static WCHAR s_szNode[256];
- oldwndproc = (WNDPROC)(LONG_PTR)GetWindowLongPtr(hwnd, GWLP_USERDATA);
+ oldwndproc = (WNDPROC)GetWindowLongPtr(hwnd, GWLP_USERDATA);
switch (uMsg)
{
case WM_KEYUP:
if (wParam == VK_RETURN)
{
- GetWindowTextW(hwnd, s_szNode, COUNT_OF(s_szNode));
+ GetWindowTextW(hwnd, s_szNode, ARRAY_SIZE(s_szNode));
SelectNode(g_pChildWnd->hTreeWnd, s_szNode);
}
break;
return CallWindowProcW(oldwndproc, hwnd, uMsg, wParam, lParam);
}
-static VOID
-UpdateAddress(HTREEITEM hItem, HKEY hRootKey, LPCWSTR pszPath)
+VOID
+UpdateAddress(HTREEITEM hItem, HKEY hRootKey, LPCWSTR pszPath, BOOL bSelectNone)
{
LPCWSTR keyPath, rootName;
LPWSTR fullPath;
+ DWORD cbFullPath;
+
+ /* Wipe the listview, the status bar and the address bar if the root key was selected */
+ if (TreeView_GetParent(g_pChildWnd->hTreeWnd, hItem) == NULL)
+ {
+ ListView_DeleteAllItems(g_pChildWnd->hListWnd);
+ SendMessageW(hStatusBar, SB_SETTEXTW, 0, (LPARAM)NULL);
+ SendMessageW(g_pChildWnd->hAddressBarWnd, WM_SETTEXT, 0, (LPARAM)NULL);
+ return;
+ }
if (pszPath == NULL)
keyPath = GetItemPath(g_pChildWnd->hTreeWnd, hItem, &hRootKey);
if (keyPath)
{
- RefreshListView(g_pChildWnd->hListWnd, hRootKey, keyPath);
+ RefreshListView(g_pChildWnd->hListWnd, hRootKey, keyPath, bSelectNone);
rootName = get_root_key_name(hRootKey);
- fullPath = HeapAlloc(GetProcessHeap(), 0, (wcslen(rootName) + 1 + wcslen(keyPath) + 1) * sizeof(WCHAR));
+ cbFullPath = (wcslen(rootName) + 1 + wcslen(keyPath) + 1) * sizeof(WCHAR);
+ fullPath = malloc(cbFullPath);
if (fullPath)
{
/* set (correct) the address bar text */
- if (keyPath[0] != L'\0')
- swprintf(fullPath, L"%s\\%s", rootName, keyPath);
+ if (keyPath[0] != UNICODE_NULL)
+ StringCbPrintfW(fullPath, cbFullPath, L"%s%s%s", rootName,
+ ((keyPath[0] == L'\\') ? L"" : L"\\"), keyPath);
else
- fullPath = wcscpy(fullPath, rootName);
+ StringCbCopyW(fullPath, cbFullPath, rootName);
+
SendMessageW(hStatusBar, SB_SETTEXTW, 0, (LPARAM)fullPath);
SendMessageW(g_pChildWnd->hAddressBarWnd, WM_SETTEXT, 0, (LPARAM)fullPath);
- HeapFree(GetProcessHeap(), 0, fullPath);
+ free(fullPath);
+
/* disable hive manipulation items temporarily (enable only if necessary) */
- EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_LOADHIVE, MF_BYCOMMAND | MF_GRAYED);
- EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_UNLOADHIVE, MF_BYCOMMAND | MF_GRAYED);
+ EnableMenuItem(hMenuFrame, ID_REGISTRY_LOADHIVE, MF_BYCOMMAND | MF_GRAYED);
+ EnableMenuItem(hMenuFrame, ID_REGISTRY_UNLOADHIVE, MF_BYCOMMAND | MF_GRAYED);
/* compare the strings to see if we should enable/disable the "Load Hive" menus accordingly */
- if (!(_wcsicmp(rootName, L"HKEY_LOCAL_MACHINE") &&
- _wcsicmp(rootName, L"HKEY_USERS")))
+ if (_wcsicmp(rootName, L"HKEY_LOCAL_MACHINE") == 0 ||
+ _wcsicmp(rootName, L"HKEY_USERS") == 0)
{
/*
* enable the unload menu item if at the root, otherwise
* enable the load menu item if there is no slash in
* keyPath (ie. immediate child selected)
*/
- if(keyPath[0] == L'\0')
- EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_LOADHIVE, MF_BYCOMMAND | MF_ENABLED);
- else if(!wcschr(keyPath, L'\\'))
- EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_UNLOADHIVE, MF_BYCOMMAND | MF_ENABLED);
+ if (keyPath[0] == UNICODE_NULL)
+ EnableMenuItem(hMenuFrame, ID_REGISTRY_LOADHIVE, MF_BYCOMMAND | MF_ENABLED);
+ else if (!wcschr(keyPath, L'\\'))
+ EnableMenuItem(hMenuFrame, ID_REGISTRY_UNLOADHIVE, MF_BYCOMMAND | MF_ENABLED);
}
}
}
* PURPOSE: Processes messages for the child windows.
*
* WM_COMMAND - process the application menu
- * WM_PAINT - Paint the main window
* WM_DESTROY - post a quit message and return
*
*/
LRESULT CALLBACK ChildWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
BOOL Result;
+ RECT rc;
switch (message)
{
WNDPROC oldproc;
HFONT hFont;
WCHAR buffer[MAX_PATH];
+ DWORD style;
/* Load "My Computer" string */
- LoadStringW(hInst, IDS_MY_COMPUTER, buffer, COUNT_OF(buffer));
+ LoadStringW(hInst, IDS_MY_COMPUTER, buffer, ARRAY_SIZE(buffer));
g_pChildWnd = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ChildWnd));
if (!g_pChildWnd) return 0;
wcsncpy(g_pChildWnd->szPath, buffer, MAX_PATH);
- g_pChildWnd->nSplitPos = 250;
+ g_pChildWnd->nSplitPos = 190;
g_pChildWnd->hWnd = hWnd;
- g_pChildWnd->hAddressBarWnd = CreateWindowExW(WS_EX_CLIENTEDGE, L"Edit", NULL, WS_CHILD | WS_VISIBLE | WS_CHILDWINDOW | WS_TABSTOP,
+
+ style = WS_CHILD | WS_VISIBLE | WS_TABSTOP;
+ g_pChildWnd->hAddressBarWnd = CreateWindowExW(WS_EX_CLIENTEDGE, L"Edit", NULL, style,
CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
hWnd, (HMENU)0, hInst, 0);
- g_pChildWnd->hAddressBtnWnd = CreateWindowExW(0, L"Button", L"ยป", WS_CHILD | WS_VISIBLE | WS_CHILDWINDOW | WS_TABSTOP | BS_TEXT | BS_CENTER | BS_VCENTER | BS_FLAT | BS_DEFPUSHBUTTON,
+
+ style = WS_CHILD | WS_VISIBLE | WS_TABSTOP | BS_ICON | BS_CENTER |
+ BS_VCENTER | BS_FLAT | BS_DEFPUSHBUTTON;
+ g_pChildWnd->hAddressBtnWnd = CreateWindowExW(0, L"Button", L"\x00BB", style,
CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
hWnd, (HMENU)0, hInst, 0);
+ g_pChildWnd->hArrowIcon = (HICON)LoadImageW(hInst, MAKEINTRESOURCEW(IDI_ARROW),
+ IMAGE_ICON, 12, 12, 0);
+ SendMessageW(g_pChildWnd->hAddressBtnWnd, BM_SETIMAGE, IMAGE_ICON, (LPARAM)g_pChildWnd->hArrowIcon);
+
+ GetClientRect(hWnd, &rc);
g_pChildWnd->hTreeWnd = CreateTreeView(hWnd, g_pChildWnd->szPath, (HMENU) TREE_WINDOW);
- g_pChildWnd->hListWnd = CreateListView(hWnd, (HMENU) LIST_WINDOW/*, g_pChildWnd->szPath*/);
+ g_pChildWnd->hListWnd = CreateListView(hWnd, (HMENU) LIST_WINDOW, rc.right - g_pChildWnd->nSplitPos);
SetFocus(g_pChildWnd->hTreeWnd);
/* set the address bar and button font */
0);
}
/* Subclass the AddressBar */
- oldproc = (WNDPROC)(LONG_PTR)GetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWLP_WNDPROC);
+ oldproc = (WNDPROC)GetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWLP_WNDPROC);
SetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWLP_USERDATA, (DWORD_PTR)oldproc);
SetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWLP_WNDPROC, (DWORD_PTR)AddressBarProc);
break;
{
PostMessageW(g_pChildWnd->hAddressBarWnd, WM_KEYUP, VK_RETURN, 0);
}
-
- if (!_CmdWndProc(hWnd, message, wParam, lParam))
- {
- goto def;
- }
- break;
- case WM_PAINT:
- OnPaint(hWnd);
- return 0;
+ break; //goto def;
case WM_SETCURSOR:
if (LOWORD(lParam) == HTCLIENT)
{
DestroyListView(g_pChildWnd->hListWnd);
DestroyTreeView(g_pChildWnd->hTreeWnd);
DestroyMainMenu();
+ DestroyIcon(g_pChildWnd->hArrowIcon);
HeapFree(GetProcessHeap(), 0, g_pChildWnd);
g_pChildWnd = NULL;
PostQuitMessage(0);
break;
case WM_NOTIFY:
- if ((int)wParam == TREE_WINDOW && g_pChildWnd != NULL)
- {
- switch (((LPNMHDR)lParam)->code)
- {
- case TVN_ITEMEXPANDING:
- return !OnTreeExpanding(g_pChildWnd->hTreeWnd, (NMTREEVIEW*)lParam);
- case TVN_SELCHANGED:
- {
- NMTREEVIEW* pnmtv = (NMTREEVIEW*)lParam;
- /* Get the parent of the current item */
- HTREEITEM hParentItem = TreeView_GetParent(g_pChildWnd->hTreeWnd, pnmtv->itemNew.hItem);
-
- UpdateAddress(pnmtv->itemNew.hItem, NULL, NULL);
+ if (g_pChildWnd == NULL) break;
- /*
- * Disable Delete/Rename menu options for 'My Computer' (first item so doesn't have any parent)
- * and HKEY_* keys (their parent is 'My Computer' and the previous remark applies).
- */
- if (!hParentItem || !TreeView_GetParent(g_pChildWnd->hTreeWnd, hParentItem))
- {
- EnableMenuItem(hMenuFrame , ID_EDIT_DELETE, MF_BYCOMMAND | MF_GRAYED);
- EnableMenuItem(hMenuFrame , ID_EDIT_RENAME, MF_BYCOMMAND | MF_GRAYED);
- EnableMenuItem(hPopupMenus, ID_TREE_DELETE, MF_BYCOMMAND | MF_GRAYED);
- EnableMenuItem(hPopupMenus, ID_TREE_RENAME, MF_BYCOMMAND | MF_GRAYED);
- }
- else
- {
- EnableMenuItem(hMenuFrame , ID_EDIT_DELETE, MF_BYCOMMAND | MF_ENABLED);
- EnableMenuItem(hMenuFrame , ID_EDIT_RENAME, MF_BYCOMMAND | MF_ENABLED);
- EnableMenuItem(hPopupMenus, ID_TREE_DELETE, MF_BYCOMMAND | MF_ENABLED);
- EnableMenuItem(hPopupMenus, ID_TREE_RENAME, MF_BYCOMMAND | MF_ENABLED);
- }
-
- break;
- }
- case NM_SETFOCUS:
- g_pChildWnd->nFocusPanel = 0;
- break;
- case TVN_BEGINLABELEDIT:
- {
- LPNMTVDISPINFO ptvdi;
- /* cancel label edit for rootkeys */
- ptvdi = (LPNMTVDISPINFO) lParam;
- if (!TreeView_GetParent(g_pChildWnd->hTreeWnd, ptvdi->item.hItem) ||
- !TreeView_GetParent(g_pChildWnd->hTreeWnd, TreeView_GetParent(g_pChildWnd->hTreeWnd, ptvdi->item.hItem)))
- return TRUE;
- break;
- }
- case TVN_ENDLABELEDIT:
+ if (((LPNMHDR)lParam)->idFrom == TREE_WINDOW)
+ {
+ if (!TreeWndNotifyProc(g_pChildWnd->hListWnd, wParam, lParam, &Result))
{
- LPCWSTR keyPath;
- HKEY hRootKey;
- HKEY hKey = NULL;
- LPNMTVDISPINFO ptvdi;
- LONG lResult = TRUE;
- WCHAR szBuffer[MAX_PATH];
-
- ptvdi = (LPNMTVDISPINFO) lParam;
- if (ptvdi->item.pszText)
- {
- keyPath = GetItemPath(g_pChildWnd->hTreeWnd, TreeView_GetParent(g_pChildWnd->hTreeWnd, ptvdi->item.hItem), &hRootKey);
- _snwprintf(szBuffer, COUNT_OF(szBuffer), L"%s\\%s", keyPath, ptvdi->item.pszText);
- keyPath = GetItemPath(g_pChildWnd->hTreeWnd, ptvdi->item.hItem, &hRootKey);
- if (RegOpenKeyExW(hRootKey, szBuffer, 0, KEY_READ, &hKey) == ERROR_SUCCESS)
- {
- lResult = FALSE;
- RegCloseKey(hKey);
- (void)TreeView_EditLabel(g_pChildWnd->hTreeWnd, ptvdi->item.hItem);
- }
- else
- {
- if (RenameKey(hRootKey, keyPath, ptvdi->item.pszText) != ERROR_SUCCESS)
- lResult = FALSE;
- else
- UpdateAddress(ptvdi->item.hItem, hRootKey, szBuffer);
- }
- return lResult;
- }
- }
- default:
- return 0;
+ goto def;
}
+
+ return Result;
}
else
{
- if ((int)wParam == LIST_WINDOW && g_pChildWnd != NULL)
+ if (((LPNMHDR)lParam)->idFrom == LIST_WINDOW)
{
- switch (((LPNMHDR)lParam)->code)
+ if (!ListWndNotifyProc(g_pChildWnd->hListWnd, wParam, lParam, &Result))
{
- case NM_SETFOCUS:
- g_pChildWnd->nFocusPanel = 1;
- break;
- default:
- if(!ListWndNotifyProc(g_pChildWnd->hListWnd, wParam, lParam, &Result))
- {
- goto def;
- }
- return Result;
- break;
+ goto def;
}
+
+ return Result;
+ }
+ else
+ {
+ goto def;
}
}
break;
HKEY hRootKey;
int iLastPos;
WORD wID;
+ BOOL isRoot;
pt.x = (short) LOWORD(lParam);
pt.y = (short) HIWORD(lParam);
hti.pt.x = pt.x;
hti.pt.y = pt.y;
ScreenToClient(g_pChildWnd->hTreeWnd, &hti.pt);
- (void)TreeView_HitTest(g_pChildWnd->hTreeWnd, &hti);
+ TreeView_HitTest(g_pChildWnd->hTreeWnd, &hti);
}
if (hti.flags & TVHT_ONITEM)
{
- hContextMenu = GetSubMenu(hPopupMenus, PM_TREECONTEXT);
- (void)TreeView_SelectItem(g_pChildWnd->hTreeWnd, hti.hItem);
+ TreeView_SelectItem(g_pChildWnd->hTreeWnd, hti.hItem);
+
+ isRoot = (TreeView_GetParent(g_pChildWnd->hTreeWnd, hti.hItem) == NULL);
+ hContextMenu = GetSubMenu(hPopupMenus, isRoot ? PM_ROOTITEM : PM_TREECONTEXT);
memset(&item, 0, sizeof(item));
item.mask = TVIF_STATE | TVIF_CHILDREN;
item.hItem = hti.hItem;
- (void)TreeView_GetItem(g_pChildWnd->hTreeWnd, &item);
+ TreeView_GetItem(g_pChildWnd->hTreeWnd, &item);
/* Set the Expand/Collapse menu item appropriately */
- LoadStringW(hInst, (item.state & TVIS_EXPANDED) ? IDS_COLLAPSE : IDS_EXPAND, buffer, COUNT_OF(buffer));
+ LoadStringW(hInst, (item.state & TVIS_EXPANDED) ? IDS_COLLAPSE : IDS_EXPAND, buffer, ARRAY_SIZE(buffer));
memset(&mii, 0, sizeof(mii));
mii.cbSize = sizeof(mii);
mii.fMask = MIIM_STRING | MIIM_STATE | MIIM_ID;
mii.dwTypeData = (LPWSTR) buffer;
SetMenuItemInfo(hContextMenu, 0, TRUE, &mii);
- /* Remove any existing suggestions */
- memset(&mii, 0, sizeof(mii));
- mii.cbSize = sizeof(mii);
- mii.fMask = MIIM_ID;
- GetMenuItemInfo(hContextMenu, GetMenuItemCount(hContextMenu) - 1, TRUE, &mii);
- if ((mii.wID >= ID_TREE_SUGGESTION_MIN) && (mii.wID <= ID_TREE_SUGGESTION_MAX))
+ if (isRoot == FALSE)
{
- do
+ /* Remove any existing suggestions */
+ memset(&mii, 0, sizeof(mii));
+ mii.cbSize = sizeof(mii);
+ mii.fMask = MIIM_ID;
+ GetMenuItemInfo(hContextMenu, GetMenuItemCount(hContextMenu) - 1, TRUE, &mii);
+ if ((mii.wID >= ID_TREE_SUGGESTION_MIN) && (mii.wID <= ID_TREE_SUGGESTION_MAX))
{
- iLastPos = GetMenuItemCount(hContextMenu) - 1;
- GetMenuItemInfo(hContextMenu, iLastPos, TRUE, &mii);
- RemoveMenu(hContextMenu, iLastPos, MF_BYPOSITION);
+ do
+ {
+ iLastPos = GetMenuItemCount(hContextMenu) - 1;
+ GetMenuItemInfo(hContextMenu, iLastPos, TRUE, &mii);
+ RemoveMenu(hContextMenu, iLastPos, MF_BYPOSITION);
+ }
+ while((mii.wID >= ID_TREE_SUGGESTION_MIN) && (mii.wID <= ID_TREE_SUGGESTION_MAX));
}
- while((mii.wID >= ID_TREE_SUGGESTION_MIN) && (mii.wID <= ID_TREE_SUGGESTION_MAX));
- }
- /* Come up with suggestions */
- keyPath = GetItemPath(g_pChildWnd->hTreeWnd, NULL, &hRootKey);
- SuggestKeys(hRootKey, keyPath, Suggestions, COUNT_OF(Suggestions));
- if (Suggestions[0])
- {
- AppendMenu(hContextMenu, MF_SEPARATOR, 0, NULL);
+ /* Come up with suggestions */
+ keyPath = GetItemPath(g_pChildWnd->hTreeWnd, NULL, &hRootKey);
+ SuggestKeys(hRootKey, keyPath, Suggestions, ARRAY_SIZE(Suggestions));
+ if (Suggestions[0])
+ {
+ AppendMenu(hContextMenu, MF_SEPARATOR, 0, NULL);
- LoadStringW(hInst, IDS_GOTO_SUGGESTED_KEY, resource, COUNT_OF(resource));
+ LoadStringW(hInst, IDS_GOTO_SUGGESTED_KEY, resource, ARRAY_SIZE(resource));
- s = Suggestions;
- wID = ID_TREE_SUGGESTION_MIN;
- while(*s && (wID <= ID_TREE_SUGGESTION_MAX))
- {
- _snwprintf(buffer, COUNT_OF(buffer), resource, s);
+ s = Suggestions;
+ wID = ID_TREE_SUGGESTION_MIN;
+ while(*s && (wID <= ID_TREE_SUGGESTION_MAX))
+ {
+ _snwprintf(buffer, ARRAY_SIZE(buffer), resource, s);
- memset(&mii, 0, sizeof(mii));
- mii.cbSize = sizeof(mii);
- mii.fMask = MIIM_STRING | MIIM_ID;
- mii.wID = wID++;
- mii.dwTypeData = buffer;
- InsertMenuItem(hContextMenu, GetMenuItemCount(hContextMenu), TRUE, &mii);
+ memset(&mii, 0, sizeof(mii));
+ mii.cbSize = sizeof(mii);
+ mii.fMask = MIIM_STRING | MIIM_ID;
+ mii.wID = wID++;
+ mii.dwTypeData = buffer;
+ InsertMenuItem(hContextMenu, GetMenuItemCount(hContextMenu), TRUE, &mii);
- s += wcslen(s) + 1;
+ s += wcslen(s) + 1;
+ }
}
}
- TrackPopupMenu(hContextMenu, TPM_RIGHTBUTTON, pt.x, pt.y, 0, g_pChildWnd->hWnd, NULL);
+ TrackPopupMenu(hContextMenu, TPM_RIGHTBUTTON, pt.x, pt.y, 0, hFrameWnd, NULL);
}
}
break;