[REGEDIT]
[reactos.git] / reactos / base / applications / regedit / childwnd.c
1 /*
2 * Regedit child window
3 *
4 * Copyright (C) 2002 Robert Dickenson <robd@reactos.org>
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include <regedit.h>
22
23 ChildWnd* g_pChildWnd;
24 static int last_split;
25 HBITMAP SizingPattern = 0;
26 HBRUSH SizingBrush = 0;
27 static WCHAR Suggestions[256];
28
29 extern LPCWSTR get_root_key_name(HKEY hRootKey)
30 {
31 if (hRootKey == HKEY_CLASSES_ROOT) return L"HKEY_CLASSES_ROOT";
32 if (hRootKey == HKEY_CURRENT_USER) return L"HKEY_CURRENT_USER";
33 if (hRootKey == HKEY_LOCAL_MACHINE) return L"HKEY_LOCAL_MACHINE";
34 if (hRootKey == HKEY_USERS) return L"HKEY_USERS";
35 if (hRootKey == HKEY_CURRENT_CONFIG) return L"HKEY_CURRENT_CONFIG";
36 if (hRootKey == HKEY_DYN_DATA) return L"HKEY_DYN_DATA";
37
38 return L"UKNOWN HKEY, PLEASE REPORT";
39 }
40
41 extern void ResizeWnd(int cx, int cy)
42 {
43 HDWP hdwp = BeginDeferWindowPos(3);
44 RECT rt, rs, rb;
45 const int tHeight = 18;
46 SetRect(&rt, 0, 0, cx, cy);
47 cy = 0;
48 if (hStatusBar != NULL)
49 {
50 GetWindowRect(hStatusBar, &rs);
51 cy = rs.bottom - rs.top;
52 }
53 GetWindowRect(g_pChildWnd->hAddressBtnWnd, &rb);
54 cx = g_pChildWnd->nSplitPos + SPLIT_WIDTH/2;
55 DeferWindowPos(hdwp, g_pChildWnd->hAddressBarWnd, 0, rt.left, rt.top, rt.right-rt.left - tHeight-2, tHeight, SWP_NOZORDER|SWP_NOACTIVATE);
56 DeferWindowPos(hdwp, g_pChildWnd->hAddressBtnWnd, 0, rt.right - tHeight, rt.top, tHeight, tHeight, SWP_NOZORDER|SWP_NOACTIVATE);
57 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);
58 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);
59 EndDeferWindowPos(hdwp);
60 }
61
62 /*******************************************************************************
63 * Local module support methods
64 */
65
66 static void draw_splitbar(HWND hWnd, int x)
67 {
68 RECT rt;
69 HGDIOBJ OldObj;
70 HDC hdc = GetDC(hWnd);
71
72 if(!SizingPattern)
73 {
74 const DWORD Pattern[4] = {0x5555AAAA, 0x5555AAAA, 0x5555AAAA, 0x5555AAAA};
75 SizingPattern = CreateBitmap(8, 8, 1, 1, Pattern);
76 }
77 if(!SizingBrush)
78 {
79 SizingBrush = CreatePatternBrush(SizingPattern);
80 }
81 GetClientRect(hWnd, &rt);
82 rt.left = x - SPLIT_WIDTH/2;
83 rt.right = x + SPLIT_WIDTH/2+1;
84 OldObj = SelectObject(hdc, SizingBrush);
85 PatBlt(hdc, rt.left, rt.top, rt.right - rt.left, rt.bottom - rt.top, PATINVERT);
86 SelectObject(hdc, OldObj);
87 ReleaseDC(hWnd, hdc);
88 }
89
90 static void OnPaint(HWND hWnd)
91 {
92 PAINTSTRUCT ps;
93 RECT rt;
94
95 GetClientRect(hWnd, &rt);
96 BeginPaint(hWnd, &ps);
97 FillRect(ps.hdc, &rt, GetSysColorBrush(COLOR_BTNFACE));
98 EndPaint(hWnd, &ps);
99 }
100
101 /*******************************************************************************
102 * finish_splitbar [internal]
103 *
104 * make the splitbar invisible and resize the windows
105 * (helper for ChildWndProc)
106 */
107 static void finish_splitbar(HWND hWnd, int x)
108 {
109 RECT rt;
110
111 draw_splitbar(hWnd, last_split);
112 last_split = -1;
113 GetClientRect(hWnd, &rt);
114 g_pChildWnd->nSplitPos = x;
115 ResizeWnd(rt.right, rt.bottom);
116 ReleaseCapture();
117 }
118
119 /*******************************************************************************
120 *
121 * FUNCTION: _CmdWndProc(HWND, unsigned, WORD, LONG)
122 *
123 * PURPOSE: Processes WM_COMMAND messages for the main frame window.
124 *
125 */
126
127 static BOOL _CmdWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
128 {
129 HTREEITEM hSelection;
130 HKEY hRootKey;
131 LPCWSTR keyPath, s;
132 WORD wID = LOWORD(wParam);
133
134 UNREFERENCED_PARAMETER(message);
135
136 switch (wID)
137 {
138 /* Parse the menu selections: */
139 case ID_REGISTRY_EXIT:
140 DestroyWindow(hWnd);
141 break;
142 case ID_VIEW_REFRESH:
143 /* TODO */
144 break;
145 case ID_TREE_EXPANDBRANCH:
146 (void)TreeView_Expand(g_pChildWnd->hTreeWnd, TreeView_GetSelection(g_pChildWnd->hTreeWnd), TVE_EXPAND);
147 break;
148 case ID_TREE_COLLAPSEBRANCH:
149 (void)TreeView_Expand(g_pChildWnd->hTreeWnd, TreeView_GetSelection(g_pChildWnd->hTreeWnd), TVE_COLLAPSE);
150 break;
151 case ID_TREE_RENAME:
152 SetFocus(g_pChildWnd->hTreeWnd);
153 (void)TreeView_EditLabel(g_pChildWnd->hTreeWnd, TreeView_GetSelection(g_pChildWnd->hTreeWnd));
154 break;
155 case ID_TREE_DELETE:
156 hSelection = TreeView_GetSelection(g_pChildWnd->hTreeWnd);
157 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, hSelection, &hRootKey);
158
159 if (keyPath == 0 || *keyPath == 0)
160 {
161 MessageBeep(MB_ICONHAND);
162 }
163 else if (DeleteKey(hWnd, hRootKey, keyPath))
164 DeleteNode(g_pChildWnd->hTreeWnd, 0);
165 break;
166 case ID_TREE_EXPORT:
167 ExportRegistryFile(g_pChildWnd->hTreeWnd);
168 break;
169 case ID_EDIT_FIND:
170 FindDialog(hWnd);
171 break;
172 case ID_EDIT_COPYKEYNAME:
173 hSelection = TreeView_GetSelection(g_pChildWnd->hTreeWnd);
174 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, hSelection, &hRootKey);
175 CopyKeyName(hWnd, hRootKey, keyPath);
176 break;
177 case ID_EDIT_NEW_KEY:
178 CreateNewKey(g_pChildWnd->hTreeWnd, TreeView_GetSelection(g_pChildWnd->hTreeWnd));
179 break;
180 case ID_EDIT_NEW_STRINGVALUE:
181 case ID_EDIT_NEW_BINARYVALUE:
182 case ID_EDIT_NEW_DWORDVALUE:
183 SendMessageW(hFrameWnd, WM_COMMAND, wParam, lParam);
184 break;
185 case ID_SWITCH_PANELS:
186 g_pChildWnd->nFocusPanel = !g_pChildWnd->nFocusPanel;
187 SetFocus(g_pChildWnd->nFocusPanel? g_pChildWnd->hListWnd: g_pChildWnd->hTreeWnd);
188 break;
189 default:
190 if ((wID >= ID_TREE_SUGGESTION_MIN) && (wID <= ID_TREE_SUGGESTION_MAX))
191 {
192 s = Suggestions;
193 while(wID > ID_TREE_SUGGESTION_MIN)
194 {
195 if (*s)
196 s += wcslen(s) + 1;
197 wID--;
198 }
199 SelectNode(g_pChildWnd->hTreeWnd, s);
200 break;
201 }
202 return FALSE;
203 }
204 return TRUE;
205 }
206
207 /*******************************************************************************
208 *
209 * Key suggestion
210 */
211
212 #define MIN(a,b) ((a < b) ? (a) : (b))
213
214 static void SuggestKeys(HKEY hRootKey, LPCWSTR pszKeyPath, LPWSTR pszSuggestions,
215 size_t iSuggestionsLength)
216 {
217 WCHAR szBuffer[256];
218 WCHAR szLastFound[256];
219 size_t i;
220 HKEY hOtherKey, hSubKey;
221 BOOL bFound;
222
223 memset(pszSuggestions, 0, iSuggestionsLength * sizeof(*pszSuggestions));
224 iSuggestionsLength--;
225
226 /* Are we a root key in HKEY_CLASSES_ROOT? */
227 if ((hRootKey == HKEY_CLASSES_ROOT) && pszKeyPath[0] && !wcschr(pszKeyPath, L'\\'))
228 {
229 do
230 {
231 bFound = FALSE;
232
233 /* Check default key */
234 if (QueryStringValue(hRootKey, pszKeyPath, NULL,
235 szBuffer, COUNT_OF(szBuffer)) == ERROR_SUCCESS)
236 {
237 /* Sanity check this key; it cannot be empty, nor can it be a
238 * loop back */
239 if ((szBuffer[0] != L'\0') && wcsicmp(szBuffer, pszKeyPath))
240 {
241 if (RegOpenKeyW(hRootKey, szBuffer, &hOtherKey) == ERROR_SUCCESS)
242 {
243 wcsncpy(pszSuggestions, L"HKCR\\", (int) iSuggestionsLength);
244 i = wcslen(pszSuggestions);
245 pszSuggestions += i;
246 iSuggestionsLength -= i;
247
248 wcsncpy(pszSuggestions, szBuffer, (int) iSuggestionsLength);
249 i = MIN(wcslen(pszSuggestions) + 1, iSuggestionsLength);
250 pszSuggestions += i;
251 iSuggestionsLength -= i;
252 RegCloseKey(hOtherKey);
253
254 bFound = TRUE;
255 wcscpy(szLastFound, szBuffer);
256 pszKeyPath = szLastFound;
257 }
258 }
259 }
260 }
261 while(bFound && (iSuggestionsLength > 0));
262
263 /* Check CLSID key */
264 if (RegOpenKeyW(hRootKey, pszKeyPath, &hSubKey) == ERROR_SUCCESS)
265 {
266 if (QueryStringValue(hSubKey, L"CLSID", NULL, szBuffer,
267 COUNT_OF(szBuffer)) == ERROR_SUCCESS)
268 {
269 wcsncpy(pszSuggestions, L"HKCR\\CLSID\\", (int)iSuggestionsLength);
270 i = wcslen(pszSuggestions);
271 pszSuggestions += i;
272 iSuggestionsLength -= i;
273
274 wcsncpy(pszSuggestions, szBuffer, (int)iSuggestionsLength);
275 i = MIN(wcslen(pszSuggestions) + 1, iSuggestionsLength);
276 pszSuggestions += i;
277 iSuggestionsLength -= i;
278 }
279 RegCloseKey(hSubKey);
280 }
281 }
282 }
283
284
285 LRESULT CALLBACK AddressBarProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
286 {
287 WNDPROC oldwndproc;
288 static WCHAR s_szNode[256];
289 oldwndproc = (WNDPROC)(LONG_PTR)GetWindowLongPtr(hwnd, GWL_USERDATA);
290
291 switch (uMsg)
292 {
293 case WM_KEYUP:
294 if (wParam == VK_RETURN)
295 {
296 GetWindowTextW(hwnd, s_szNode, COUNT_OF(s_szNode));
297 SelectNode(g_pChildWnd->hTreeWnd, s_szNode);
298 }
299 break;
300 default:
301 break;
302 }
303 return CallWindowProc(oldwndproc, hwnd, uMsg, wParam, lParam);
304 }
305
306 /*******************************************************************************
307 *
308 * FUNCTION: ChildWndProc(HWND, unsigned, WORD, LONG)
309 *
310 * PURPOSE: Processes messages for the child windows.
311 *
312 * WM_COMMAND - process the application menu
313 * WM_PAINT - Paint the main window
314 * WM_DESTROY - post a quit message and return
315 *
316 */
317 LRESULT CALLBACK ChildWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
318 {
319 BOOL Result;
320
321 switch (message)
322 {
323 case WM_CREATE:
324 {
325 WNDPROC oldproc;
326 HFONT hFont;
327 WCHAR buffer[MAX_PATH];
328
329 /* Load "My Computer" string */
330 LoadStringW(hInst, IDS_MY_COMPUTER, buffer, COUNT_OF(buffer));
331
332 g_pChildWnd = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ChildWnd));
333 if (!g_pChildWnd) return 0;
334
335 wcsncpy(g_pChildWnd->szPath, buffer, MAX_PATH);
336 g_pChildWnd->nSplitPos = 250;
337 g_pChildWnd->hWnd = hWnd;
338 g_pChildWnd->hAddressBarWnd = CreateWindowExW(WS_EX_CLIENTEDGE, L"Edit", NULL, WS_CHILD | WS_VISIBLE | WS_CHILDWINDOW | WS_TABSTOP,
339 CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
340 hWnd, (HMENU)0, hInst, 0);
341 g_pChildWnd->hAddressBtnWnd = CreateWindowExW(WS_EX_CLIENTEDGE, L"Button", L"ยป", WS_CHILD | WS_VISIBLE | WS_CHILDWINDOW | WS_TABSTOP | BS_DEFPUSHBUTTON,
342 CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
343 hWnd, (HMENU)0, hInst, 0);
344 g_pChildWnd->hTreeWnd = CreateTreeView(hWnd, g_pChildWnd->szPath, (HMENU) TREE_WINDOW);
345 g_pChildWnd->hListWnd = CreateListView(hWnd, (HMENU) LIST_WINDOW/*, g_pChildWnd->szPath*/);
346 SetFocus(g_pChildWnd->hTreeWnd);
347
348 /* set the address bar and button font */
349 if ((g_pChildWnd->hAddressBarWnd) && (g_pChildWnd->hAddressBtnWnd))
350 {
351 hFont = (HFONT)GetStockObject(DEFAULT_GUI_FONT);
352 SendMessageW(g_pChildWnd->hAddressBarWnd,
353 WM_SETFONT,
354 (WPARAM)hFont,
355 0);
356 SendMessageW(g_pChildWnd->hAddressBtnWnd,
357 WM_SETFONT,
358 (WPARAM)hFont,
359 0);
360 }
361 /* Subclass the AddressBar */
362 oldproc = (WNDPROC)(LONG_PTR)GetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWL_WNDPROC);
363 SetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWL_USERDATA, (DWORD_PTR)oldproc);
364 SetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWL_WNDPROC, (DWORD_PTR)AddressBarProc);
365 break;
366 }
367 case WM_COMMAND:
368 if(HIWORD(wParam) == BN_CLICKED)
369 {
370 PostMessageW(g_pChildWnd->hAddressBarWnd, WM_KEYUP, VK_RETURN, 0);
371 }
372
373 if (!_CmdWndProc(hWnd, message, wParam, lParam))
374 {
375 goto def;
376 }
377 break;
378 case WM_PAINT:
379 OnPaint(hWnd);
380 return 0;
381 case WM_SETCURSOR:
382 if (LOWORD(lParam) == HTCLIENT)
383 {
384 POINT pt;
385 GetCursorPos(&pt);
386 ScreenToClient(hWnd, &pt);
387 if (pt.x>=g_pChildWnd->nSplitPos-SPLIT_WIDTH/2 && pt.x<g_pChildWnd->nSplitPos+SPLIT_WIDTH/2+1)
388 {
389 SetCursor(LoadCursorW(0, IDC_SIZEWE));
390 return TRUE;
391 }
392 }
393 goto def;
394 case WM_DESTROY:
395 DestroyTreeView();
396 DestroyListView(g_pChildWnd->hListWnd);
397 DestroyMainMenu();
398 HeapFree(GetProcessHeap(), 0, g_pChildWnd);
399 g_pChildWnd = NULL;
400 PostQuitMessage(0);
401 break;
402 case WM_LBUTTONDOWN:
403 {
404 RECT rt;
405 int x = (short)LOWORD(lParam);
406 GetClientRect(hWnd, &rt);
407 if (x>=g_pChildWnd->nSplitPos-SPLIT_WIDTH/2 && x<g_pChildWnd->nSplitPos+SPLIT_WIDTH/2+1)
408 {
409 last_split = g_pChildWnd->nSplitPos;
410 draw_splitbar(hWnd, last_split);
411 SetCapture(hWnd);
412 }
413 break;
414 }
415
416 case WM_LBUTTONUP:
417 case WM_RBUTTONDOWN:
418 if (GetCapture() == hWnd)
419 {
420 finish_splitbar(hWnd, LOWORD(lParam));
421 }
422 break;
423
424 case WM_CAPTURECHANGED:
425 if (GetCapture()==hWnd && last_split>=0)
426 draw_splitbar(hWnd, last_split);
427 break;
428
429 case WM_KEYDOWN:
430 if (wParam == VK_ESCAPE)
431 if (GetCapture() == hWnd)
432 {
433 RECT rt;
434 draw_splitbar(hWnd, last_split);
435 GetClientRect(hWnd, &rt);
436 ResizeWnd(rt.right, rt.bottom);
437 last_split = -1;
438 ReleaseCapture();
439 SetCursor(LoadCursorW(0, IDC_ARROW));
440 }
441 break;
442
443 case WM_MOUSEMOVE:
444 if (GetCapture() == hWnd)
445 {
446 HDC hdc;
447 RECT rt;
448 HGDIOBJ OldObj;
449 int x = LOWORD(lParam);
450 if(!SizingPattern)
451 {
452 const DWORD Pattern[4] = {0x5555AAAA, 0x5555AAAA, 0x5555AAAA, 0x5555AAAA};
453 SizingPattern = CreateBitmap(8, 8, 1, 1, Pattern);
454 }
455 if(!SizingBrush)
456 {
457 SizingBrush = CreatePatternBrush(SizingPattern);
458 }
459
460 GetClientRect(hWnd, &rt);
461 x = (SHORT) min(max(x, SPLIT_MIN), rt.right - SPLIT_MIN);
462 if(last_split != x)
463 {
464 rt.left = last_split-SPLIT_WIDTH/2;
465 rt.right = last_split+SPLIT_WIDTH/2+1;
466 hdc = GetDC(hWnd);
467 OldObj = SelectObject(hdc, SizingBrush);
468 PatBlt(hdc, rt.left, rt.top, rt.right - rt.left, rt.bottom - rt.top, PATINVERT);
469 last_split = x;
470 rt.left = x-SPLIT_WIDTH/2;
471 rt.right = x+SPLIT_WIDTH/2+1;
472 PatBlt(hdc, rt.left, rt.top, rt.right - rt.left, rt.bottom - rt.top, PATINVERT);
473 SelectObject(hdc, OldObj);
474 ReleaseDC(hWnd, hdc);
475 }
476 }
477 break;
478
479 case WM_SETFOCUS:
480 if (g_pChildWnd != NULL)
481 {
482 SetFocus(g_pChildWnd->nFocusPanel? g_pChildWnd->hListWnd: g_pChildWnd->hTreeWnd);
483 }
484 break;
485
486 case WM_TIMER:
487 break;
488
489 case WM_NOTIFY:
490 if ((int)wParam == TREE_WINDOW && g_pChildWnd != NULL)
491 {
492 switch (((LPNMHDR)lParam)->code)
493 {
494 case TVN_ITEMEXPANDING:
495 return !OnTreeExpanding(g_pChildWnd->hTreeWnd, (NMTREEVIEW*)lParam);
496 case TVN_SELCHANGED:
497 {
498 LPCWSTR keyPath, rootName;
499 LPWSTR fullPath;
500 HKEY hRootKey;
501
502 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, ((NMTREEVIEW*)lParam)->itemNew.hItem, &hRootKey);
503 if (keyPath)
504 {
505 RefreshListView(g_pChildWnd->hListWnd, hRootKey, keyPath);
506 rootName = get_root_key_name(hRootKey);
507 fullPath = HeapAlloc(GetProcessHeap(), 0, (wcslen(rootName) + 1 + wcslen(keyPath) + 1) * sizeof(WCHAR));
508 if (fullPath)
509 {
510 /* set (correct) the address bar text */
511 if(keyPath[0] != L'\0')
512 swprintf(fullPath, L"%s\\%s", rootName, keyPath);
513 else
514 fullPath = wcscpy(fullPath, rootName);
515 SendMessageW(hStatusBar, SB_SETTEXTW, 0, (LPARAM)fullPath);
516 SendMessageW(g_pChildWnd->hAddressBarWnd, WM_SETTEXT, 0, (LPARAM)fullPath);
517 HeapFree(GetProcessHeap(), 0, fullPath);
518 /* disable hive manipulation items temporarily (enable only if necessary) */
519 EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_LOADHIVE, MF_BYCOMMAND | MF_GRAYED);
520 EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_UNLOADHIVE, MF_BYCOMMAND | MF_GRAYED);
521 /* compare the strings to see if we should enable/disable the "Load Hive" menus accordingly */
522 if (!(wcsicmp(rootName, L"HKEY_LOCAL_MACHINE") &&
523 wcsicmp(rootName, L"HKEY_USERS")))
524 {
525 // enable the unload menu item if at the root
526 // otherwise enable the load menu item if there is no slash in keyPath (ie. immediate child selected)
527 if(keyPath[0] == L'\0')
528 EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_LOADHIVE, MF_BYCOMMAND | MF_ENABLED);
529 else if(!wcschr(keyPath, L'\\'))
530 EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_UNLOADHIVE, MF_BYCOMMAND | MF_ENABLED);
531 }
532 }
533 }
534 }
535 break;
536 case NM_SETFOCUS:
537 g_pChildWnd->nFocusPanel = 0;
538 break;
539 case TVN_BEGINLABELEDIT:
540 {
541 LPNMTVDISPINFO ptvdi;
542 /* cancel label edit for rootkeys */
543 ptvdi = (LPNMTVDISPINFO) lParam;
544 if (!TreeView_GetParent(g_pChildWnd->hTreeWnd, ptvdi->item.hItem) ||
545 !TreeView_GetParent(g_pChildWnd->hTreeWnd, TreeView_GetParent(g_pChildWnd->hTreeWnd, ptvdi->item.hItem)))
546 return TRUE;
547 break;
548 }
549 case TVN_ENDLABELEDIT:
550 {
551 LPCWSTR keyPath;
552 HKEY hRootKey;
553 HKEY hKey = NULL;
554 LPNMTVDISPINFO ptvdi;
555 LONG lResult = TRUE;
556 WCHAR szBuffer[MAX_PATH];
557
558 ptvdi = (LPNMTVDISPINFO) lParam;
559 if (ptvdi->item.pszText)
560 {
561 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, TreeView_GetParent(g_pChildWnd->hTreeWnd, ptvdi->item.hItem), &hRootKey);
562 _snwprintf(szBuffer, COUNT_OF(szBuffer), L"%s\\%s", keyPath, ptvdi->item.pszText);
563 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, ptvdi->item.hItem, &hRootKey);
564 if (RegOpenKeyExW(hRootKey, szBuffer, 0, KEY_READ, &hKey) == ERROR_SUCCESS)
565 {
566 lResult = FALSE;
567 RegCloseKey(hKey);
568 (void)TreeView_EditLabel(g_pChildWnd->hTreeWnd, ptvdi->item.hItem);
569 }
570 else
571 {
572 if (RenameKey(hRootKey, keyPath, ptvdi->item.pszText) != ERROR_SUCCESS)
573 lResult = FALSE;
574 }
575 return lResult;
576 }
577 }
578 default:
579 return 0;
580 }
581 }
582 else
583 {
584 if ((int)wParam == LIST_WINDOW && g_pChildWnd != NULL)
585 {
586 switch (((LPNMHDR)lParam)->code)
587 {
588 case NM_SETFOCUS:
589 g_pChildWnd->nFocusPanel = 1;
590 break;
591 default:
592 if(!ListWndNotifyProc(g_pChildWnd->hListWnd, wParam, lParam, &Result))
593 {
594 goto def;
595 }
596 return Result;
597 break;
598 }
599 }
600 }
601 break;
602
603 case WM_CONTEXTMENU:
604 {
605 POINT pt;
606 if((HWND)wParam == g_pChildWnd->hListWnd)
607 {
608 int i, cnt;
609 BOOL IsDefault;
610 pt.x = (short) LOWORD(lParam);
611 pt.y = (short) HIWORD(lParam);
612 cnt = ListView_GetSelectedCount(g_pChildWnd->hListWnd);
613 i = ListView_GetNextItem(g_pChildWnd->hListWnd, -1, LVNI_FOCUSED | LVNI_SELECTED);
614 if (pt.x == -1 && pt.y == -1)
615 {
616 RECT rc;
617 if (i != -1)
618 {
619 rc.left = LVIR_BOUNDS;
620 SendMessageW(g_pChildWnd->hListWnd, LVM_GETITEMRECT, i, (LPARAM) &rc);
621 pt.x = rc.left + 8;
622 pt.y = rc.top + 8;
623 }
624 else
625 pt.x = pt.y = 0;
626 ClientToScreen(g_pChildWnd->hListWnd, &pt);
627 }
628 if(i == -1)
629 {
630 TrackPopupMenu(GetSubMenu(hPopupMenus, PM_NEW), TPM_RIGHTBUTTON, pt.x, pt.y, 0, hFrameWnd, NULL);
631 }
632 else
633 {
634 HMENU mnu = GetSubMenu(hPopupMenus, PM_MODIFYVALUE);
635 SetMenuDefaultItem(mnu, ID_EDIT_MODIFY, MF_BYCOMMAND);
636 IsDefault = IsDefaultValue(g_pChildWnd->hListWnd, i);
637 if(cnt == 1)
638 EnableMenuItem(mnu, ID_EDIT_RENAME, MF_BYCOMMAND | (IsDefault ? MF_DISABLED | MF_GRAYED : MF_ENABLED));
639 else
640 EnableMenuItem(mnu, ID_EDIT_RENAME, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
641 EnableMenuItem(mnu, ID_EDIT_MODIFY, MF_BYCOMMAND | (cnt == 1 ? MF_ENABLED : MF_DISABLED | MF_GRAYED));
642 EnableMenuItem(mnu, ID_EDIT_MODIFY_BIN, MF_BYCOMMAND | (cnt == 1 ? MF_ENABLED : MF_DISABLED | MF_GRAYED));
643
644 TrackPopupMenu(mnu, TPM_RIGHTBUTTON, pt.x, pt.y, 0, hFrameWnd, NULL);
645 }
646 }
647 else if ((HWND)wParam == g_pChildWnd->hTreeWnd)
648 {
649 TVHITTESTINFO hti;
650 HMENU hContextMenu;
651 TVITEM item;
652 MENUITEMINFO mii;
653 WCHAR resource[256];
654 WCHAR buffer[256];
655 LPWSTR s;
656 LPCWSTR keyPath;
657 HKEY hRootKey;
658 int iLastPos;
659 WORD wID;
660
661 pt.x = (short) LOWORD(lParam);
662 pt.y = (short) HIWORD(lParam);
663
664 if (pt.x == -1 && pt.y == -1)
665 {
666 RECT rc;
667 hti.hItem = TreeView_GetSelection(g_pChildWnd->hTreeWnd);
668 if (hti.hItem != NULL)
669 {
670 TreeView_GetItemRect(g_pChildWnd->hTreeWnd, hti.hItem, &rc, TRUE);
671 pt.x = rc.left + 8;
672 pt.y = rc.top + 8;
673 ClientToScreen(g_pChildWnd->hTreeWnd, &pt);
674 hti.flags = TVHT_ONITEM;
675 }
676 else
677 hti.flags = 0;
678 }
679 else
680 {
681 hti.pt.x = pt.x;
682 hti.pt.y = pt.y;
683 ScreenToClient(g_pChildWnd->hTreeWnd, &hti.pt);
684 (void)TreeView_HitTest(g_pChildWnd->hTreeWnd, &hti);
685 }
686
687 if (hti.flags & TVHT_ONITEM)
688 {
689 hContextMenu = GetSubMenu(hPopupMenus, PM_TREECONTEXT);
690 (void)TreeView_SelectItem(g_pChildWnd->hTreeWnd, hti.hItem);
691
692 memset(&item, 0, sizeof(item));
693 item.mask = TVIF_STATE | TVIF_CHILDREN;
694 item.hItem = hti.hItem;
695 (void)TreeView_GetItem(g_pChildWnd->hTreeWnd, &item);
696
697 /* Set the Expand/Collapse menu item appropriately */
698 LoadStringW(hInst, (item.state & TVIS_EXPANDED) ? IDS_COLLAPSE : IDS_EXPAND, buffer, COUNT_OF(buffer));
699 memset(&mii, 0, sizeof(mii));
700 mii.cbSize = sizeof(mii);
701 mii.fMask = MIIM_STRING | MIIM_STATE | MIIM_ID;
702 mii.fState = (item.cChildren > 0) ? MFS_DEFAULT : MFS_GRAYED;
703 mii.wID = (item.state & TVIS_EXPANDED) ? ID_TREE_COLLAPSEBRANCH : ID_TREE_EXPANDBRANCH;
704 mii.dwTypeData = (LPWSTR) buffer;
705 SetMenuItemInfo(hContextMenu, 0, TRUE, &mii);
706
707 /* Remove any existing suggestions */
708 memset(&mii, 0, sizeof(mii));
709 mii.cbSize = sizeof(mii);
710 mii.fMask = MIIM_ID;
711 GetMenuItemInfo(hContextMenu, GetMenuItemCount(hContextMenu) - 1, TRUE, &mii);
712 if ((mii.wID >= ID_TREE_SUGGESTION_MIN) && (mii.wID <= ID_TREE_SUGGESTION_MAX))
713 {
714 do
715 {
716 iLastPos = GetMenuItemCount(hContextMenu) - 1;
717 GetMenuItemInfo(hContextMenu, iLastPos, TRUE, &mii);
718 RemoveMenu(hContextMenu, iLastPos, MF_BYPOSITION);
719 }
720 while((mii.wID >= ID_TREE_SUGGESTION_MIN) && (mii.wID <= ID_TREE_SUGGESTION_MAX));
721 }
722
723 /* Come up with suggestions */
724 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, NULL, &hRootKey);
725 SuggestKeys(hRootKey, keyPath, Suggestions, COUNT_OF(Suggestions));
726 if (Suggestions[0])
727 {
728 AppendMenu(hContextMenu, MF_SEPARATOR, 0, NULL);
729
730 LoadStringW(hInst, IDS_GOTO_SUGGESTED_KEY, resource, COUNT_OF(resource));
731
732 s = Suggestions;
733 wID = ID_TREE_SUGGESTION_MIN;
734 while(*s && (wID <= ID_TREE_SUGGESTION_MAX))
735 {
736 _snwprintf(buffer, COUNT_OF(buffer), resource, s);
737
738 memset(&mii, 0, sizeof(mii));
739 mii.cbSize = sizeof(mii);
740 mii.fMask = MIIM_STRING | MIIM_ID;
741 mii.wID = wID++;
742 mii.dwTypeData = buffer;
743 InsertMenuItem(hContextMenu, GetMenuItemCount(hContextMenu), TRUE, &mii);
744
745 s += wcslen(s) + 1;
746 }
747 }
748 TrackPopupMenu(hContextMenu, TPM_RIGHTBUTTON, pt.x, pt.y, 0, g_pChildWnd->hWnd, NULL);
749 }
750 }
751 break;
752 }
753
754 case WM_SIZE:
755 if (wParam != SIZE_MINIMIZED && g_pChildWnd != NULL)
756 {
757 ResizeWnd(LOWORD(lParam), HIWORD(lParam));
758 }
759 /* fall through */
760 default:
761 def:
762 return DefWindowProcW(hWnd, message, wParam, lParam);
763 }
764 return 0;
765 }