[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 /* load "My Computer" string */
329 LoadStringW(hInst, IDS_MY_COMPUTER, buffer, COUNT_OF(buffer));
330
331 g_pChildWnd = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ChildWnd));
332 if (!g_pChildWnd) return 0;
333
334 wcsncpy(g_pChildWnd->szPath, buffer, MAX_PATH);
335 g_pChildWnd->nSplitPos = 250;
336 g_pChildWnd->hWnd = hWnd;
337 g_pChildWnd->hAddressBarWnd = CreateWindowExW(WS_EX_CLIENTEDGE, L"Edit", NULL, WS_CHILD | WS_VISIBLE | WS_CHILDWINDOW | WS_TABSTOP,
338 CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
339 hWnd, (HMENU)0, hInst, 0);
340 g_pChildWnd->hAddressBtnWnd = CreateWindowExW(WS_EX_CLIENTEDGE, L"Button", L"ยป", WS_CHILD | WS_VISIBLE | WS_CHILDWINDOW | WS_TABSTOP | BS_DEFPUSHBUTTON,
341 CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
342 hWnd, (HMENU)0, hInst, 0);
343 g_pChildWnd->hTreeWnd = CreateTreeView(hWnd, g_pChildWnd->szPath, (HMENU) TREE_WINDOW);
344 g_pChildWnd->hListWnd = CreateListView(hWnd, (HMENU) LIST_WINDOW/*, g_pChildWnd->szPath*/);
345 SetFocus(g_pChildWnd->hTreeWnd);
346
347 /* set the address bar and button font */
348 if ((g_pChildWnd->hAddressBarWnd) && (g_pChildWnd->hAddressBtnWnd))
349 {
350 hFont = (HFONT)GetStockObject(DEFAULT_GUI_FONT);
351 SendMessageW(g_pChildWnd->hAddressBarWnd,
352 WM_SETFONT,
353 (WPARAM)hFont,
354 0);
355 SendMessageW(g_pChildWnd->hAddressBtnWnd,
356 WM_SETFONT,
357 (WPARAM)hFont,
358 0);
359 }
360 /* Subclass the AddressBar */
361 oldproc = (WNDPROC)(LONG_PTR)GetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWL_WNDPROC);
362 SetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWL_USERDATA, (DWORD_PTR)oldproc);
363 SetWindowLongPtr(g_pChildWnd->hAddressBarWnd, GWL_WNDPROC, (DWORD_PTR)AddressBarProc);
364 break;
365 }
366 case WM_COMMAND:
367 if(HIWORD(wParam) == BN_CLICKED)
368 {
369 PostMessageW(g_pChildWnd->hAddressBarWnd, WM_KEYUP, VK_RETURN, 0);
370 }
371
372 if (!_CmdWndProc(hWnd, message, wParam, lParam))
373 {
374 goto def;
375 }
376 break;
377 case WM_PAINT:
378 OnPaint(hWnd);
379 return 0;
380 case WM_SETCURSOR:
381 if (LOWORD(lParam) == HTCLIENT)
382 {
383 POINT pt;
384 GetCursorPos(&pt);
385 ScreenToClient(hWnd, &pt);
386 if (pt.x>=g_pChildWnd->nSplitPos-SPLIT_WIDTH/2 && pt.x<g_pChildWnd->nSplitPos+SPLIT_WIDTH/2+1)
387 {
388 SetCursor(LoadCursorW(0, IDC_SIZEWE));
389 return TRUE;
390 }
391 }
392 goto def;
393 case WM_DESTROY:
394 DestroyTreeView();
395 DestroyListView(g_pChildWnd->hListWnd);
396 DestroyMainMenu();
397 HeapFree(GetProcessHeap(), 0, g_pChildWnd);
398 g_pChildWnd = NULL;
399 PostQuitMessage(0);
400 break;
401 case WM_LBUTTONDOWN:
402 {
403 RECT rt;
404 int x = (short)LOWORD(lParam);
405 GetClientRect(hWnd, &rt);
406 if (x>=g_pChildWnd->nSplitPos-SPLIT_WIDTH/2 && x<g_pChildWnd->nSplitPos+SPLIT_WIDTH/2+1)
407 {
408 last_split = g_pChildWnd->nSplitPos;
409 draw_splitbar(hWnd, last_split);
410 SetCapture(hWnd);
411 }
412 break;
413 }
414
415 case WM_LBUTTONUP:
416 case WM_RBUTTONDOWN:
417 if (GetCapture() == hWnd)
418 {
419 finish_splitbar(hWnd, LOWORD(lParam));
420 }
421 break;
422
423 case WM_CAPTURECHANGED:
424 if (GetCapture()==hWnd && last_split>=0)
425 draw_splitbar(hWnd, last_split);
426 break;
427
428 case WM_KEYDOWN:
429 if (wParam == VK_ESCAPE)
430 if (GetCapture() == hWnd)
431 {
432 RECT rt;
433 draw_splitbar(hWnd, last_split);
434 GetClientRect(hWnd, &rt);
435 ResizeWnd(rt.right, rt.bottom);
436 last_split = -1;
437 ReleaseCapture();
438 SetCursor(LoadCursorW(0, IDC_ARROW));
439 }
440 break;
441
442 case WM_MOUSEMOVE:
443 if (GetCapture() == hWnd)
444 {
445 HDC hdc;
446 RECT rt;
447 HGDIOBJ OldObj;
448 int x = LOWORD(lParam);
449 if(!SizingPattern)
450 {
451 const DWORD Pattern[4] = {0x5555AAAA, 0x5555AAAA, 0x5555AAAA, 0x5555AAAA};
452 SizingPattern = CreateBitmap(8, 8, 1, 1, Pattern);
453 }
454 if(!SizingBrush)
455 {
456 SizingBrush = CreatePatternBrush(SizingPattern);
457 }
458
459 GetClientRect(hWnd, &rt);
460 x = (SHORT) min(max(x, SPLIT_MIN), rt.right - SPLIT_MIN);
461 if(last_split != x)
462 {
463 rt.left = last_split-SPLIT_WIDTH/2;
464 rt.right = last_split+SPLIT_WIDTH/2+1;
465 hdc = GetDC(hWnd);
466 OldObj = SelectObject(hdc, SizingBrush);
467 PatBlt(hdc, rt.left, rt.top, rt.right - rt.left, rt.bottom - rt.top, PATINVERT);
468 last_split = x;
469 rt.left = x-SPLIT_WIDTH/2;
470 rt.right = x+SPLIT_WIDTH/2+1;
471 PatBlt(hdc, rt.left, rt.top, rt.right - rt.left, rt.bottom - rt.top, PATINVERT);
472 SelectObject(hdc, OldObj);
473 ReleaseDC(hWnd, hdc);
474 }
475 }
476 break;
477
478 case WM_SETFOCUS:
479 if (g_pChildWnd != NULL)
480 {
481 SetFocus(g_pChildWnd->nFocusPanel? g_pChildWnd->hListWnd: g_pChildWnd->hTreeWnd);
482 }
483 break;
484
485 case WM_TIMER:
486 break;
487
488 case WM_NOTIFY:
489 if ((int)wParam == TREE_WINDOW && g_pChildWnd != NULL)
490 {
491 switch (((LPNMHDR)lParam)->code)
492 {
493 case TVN_ITEMEXPANDING:
494 return !OnTreeExpanding(g_pChildWnd->hTreeWnd, (NMTREEVIEW*)lParam);
495 case TVN_SELCHANGED:
496 {
497 LPCWSTR keyPath, rootName;
498 LPWSTR fullPath;
499 HKEY hRootKey;
500
501 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, ((NMTREEVIEW*)lParam)->itemNew.hItem, &hRootKey);
502 if (keyPath)
503 {
504 RefreshListView(g_pChildWnd->hListWnd, hRootKey, keyPath);
505 rootName = get_root_key_name(hRootKey);
506 fullPath = HeapAlloc(GetProcessHeap(), 0, (wcslen(rootName) + 1 + wcslen(keyPath) + 1) * sizeof(WCHAR));
507 if (fullPath)
508 {
509 /* set (correct) the address bar text */
510 if(keyPath[0] != L'\0')
511 swprintf(fullPath, L"%s\\%s", rootName, keyPath);
512 else
513 fullPath = wcscpy(fullPath, rootName);
514 SendMessageW(hStatusBar, SB_SETTEXTW, 0, (LPARAM)fullPath);
515 SendMessageW(g_pChildWnd->hAddressBarWnd, WM_SETTEXT, 0, (LPARAM)fullPath);
516 HeapFree(GetProcessHeap(), 0, fullPath);
517 /* disable hive manipulation items temporarily (enable only if necessary) */
518 EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_LOADHIVE, MF_BYCOMMAND | MF_GRAYED);
519 EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_UNLOADHIVE, MF_BYCOMMAND | MF_GRAYED);
520 /* compare the strings to see if we should enable/disable the "Load Hive" menus accordingly */
521 if (!(wcsicmp(rootName, L"HKEY_LOCAL_MACHINE") &&
522 wcsicmp(rootName, L"HKEY_USERS")))
523 {
524 // enable the unload menu item if at the root
525 // otherwise enable the load menu item if there is no slash in keyPath (ie. immediate child selected)
526 if(keyPath[0] == L'\0')
527 EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_LOADHIVE, MF_BYCOMMAND | MF_ENABLED);
528 else if(!wcschr(keyPath, L'\\'))
529 EnableMenuItem(GetSubMenu(hMenuFrame,0), ID_REGISTRY_UNLOADHIVE, MF_BYCOMMAND | MF_ENABLED);
530 }
531 }
532 }
533 }
534 break;
535 case NM_SETFOCUS:
536 g_pChildWnd->nFocusPanel = 0;
537 break;
538 case TVN_BEGINLABELEDIT:
539 {
540 LPNMTVDISPINFO ptvdi;
541 /* cancel label edit for rootkeys */
542 ptvdi = (LPNMTVDISPINFO) lParam;
543 if (!TreeView_GetParent(g_pChildWnd->hTreeWnd, ptvdi->item.hItem) ||
544 !TreeView_GetParent(g_pChildWnd->hTreeWnd, TreeView_GetParent(g_pChildWnd->hTreeWnd, ptvdi->item.hItem)))
545 return TRUE;
546 break;
547 }
548 case TVN_ENDLABELEDIT:
549 {
550 LPCWSTR keyPath;
551 HKEY hRootKey;
552 HKEY hKey = NULL;
553 LPNMTVDISPINFO ptvdi;
554 LONG lResult = TRUE;
555 WCHAR szBuffer[MAX_PATH];
556
557 ptvdi = (LPNMTVDISPINFO) lParam;
558 if (ptvdi->item.pszText)
559 {
560 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, TreeView_GetParent(g_pChildWnd->hTreeWnd, ptvdi->item.hItem), &hRootKey);
561 _snwprintf(szBuffer, COUNT_OF(szBuffer), L"%s\\%s", keyPath, ptvdi->item.pszText);
562 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, ptvdi->item.hItem, &hRootKey);
563 if (RegOpenKeyExW(hRootKey, szBuffer, 0, KEY_READ, &hKey) == ERROR_SUCCESS)
564 {
565 lResult = FALSE;
566 RegCloseKey(hKey);
567 (void)TreeView_EditLabel(g_pChildWnd->hTreeWnd, ptvdi->item.hItem);
568 }
569 else
570 {
571 if (RenameKey(hRootKey, keyPath, ptvdi->item.pszText) != ERROR_SUCCESS)
572 lResult = FALSE;
573 }
574 return lResult;
575 }
576 }
577 default:
578 return 0;
579 }
580 }
581 else
582 {
583 if ((int)wParam == LIST_WINDOW && g_pChildWnd != NULL)
584 {
585 switch (((LPNMHDR)lParam)->code)
586 {
587 case NM_SETFOCUS:
588 g_pChildWnd->nFocusPanel = 1;
589 break;
590 default:
591 if(!ListWndNotifyProc(g_pChildWnd->hListWnd, wParam, lParam, &Result))
592 {
593 goto def;
594 }
595 return Result;
596 break;
597 }
598 }
599 }
600 break;
601
602 case WM_CONTEXTMENU:
603 {
604 POINT pt;
605 if((HWND)wParam == g_pChildWnd->hListWnd)
606 {
607 int i, cnt;
608 BOOL IsDefault;
609 pt.x = (short) LOWORD(lParam);
610 pt.y = (short) HIWORD(lParam);
611 cnt = ListView_GetSelectedCount(g_pChildWnd->hListWnd);
612 i = ListView_GetNextItem(g_pChildWnd->hListWnd, -1, LVNI_FOCUSED | LVNI_SELECTED);
613 if (pt.x == -1 && pt.y == -1)
614 {
615 RECT rc;
616 if (i != -1)
617 {
618 rc.left = LVIR_BOUNDS;
619 SendMessageW(g_pChildWnd->hListWnd, LVM_GETITEMRECT, i, (LPARAM) &rc);
620 pt.x = rc.left + 8;
621 pt.y = rc.top + 8;
622 }
623 else
624 pt.x = pt.y = 0;
625 ClientToScreen(g_pChildWnd->hListWnd, &pt);
626 }
627 if(i == -1)
628 {
629 TrackPopupMenu(GetSubMenu(hPopupMenus, PM_NEW), TPM_RIGHTBUTTON, pt.x, pt.y, 0, hFrameWnd, NULL);
630 }
631 else
632 {
633 HMENU mnu = GetSubMenu(hPopupMenus, PM_MODIFYVALUE);
634 SetMenuDefaultItem(mnu, ID_EDIT_MODIFY, MF_BYCOMMAND);
635 IsDefault = IsDefaultValue(g_pChildWnd->hListWnd, i);
636 if(cnt == 1)
637 EnableMenuItem(mnu, ID_EDIT_RENAME, MF_BYCOMMAND | (IsDefault ? MF_DISABLED | MF_GRAYED : MF_ENABLED));
638 else
639 EnableMenuItem(mnu, ID_EDIT_RENAME, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
640 EnableMenuItem(mnu, ID_EDIT_MODIFY, MF_BYCOMMAND | (cnt == 1 ? MF_ENABLED : MF_DISABLED | MF_GRAYED));
641 EnableMenuItem(mnu, ID_EDIT_MODIFY_BIN, MF_BYCOMMAND | (cnt == 1 ? MF_ENABLED : MF_DISABLED | MF_GRAYED));
642
643 TrackPopupMenu(mnu, TPM_RIGHTBUTTON, pt.x, pt.y, 0, hFrameWnd, NULL);
644 }
645 }
646 else if ((HWND)wParam == g_pChildWnd->hTreeWnd)
647 {
648 TVHITTESTINFO hti;
649 HMENU hContextMenu;
650 TVITEM item;
651 MENUITEMINFO mii;
652 WCHAR resource[256];
653 WCHAR buffer[256];
654 LPWSTR s;
655 LPCWSTR keyPath;
656 HKEY hRootKey;
657 int iLastPos;
658 WORD wID;
659
660 pt.x = (short) LOWORD(lParam);
661 pt.y = (short) HIWORD(lParam);
662
663 if (pt.x == -1 && pt.y == -1)
664 {
665 RECT rc;
666 hti.hItem = TreeView_GetSelection(g_pChildWnd->hTreeWnd);
667 if (hti.hItem != NULL)
668 {
669 TreeView_GetItemRect(g_pChildWnd->hTreeWnd, hti.hItem, &rc, TRUE);
670 pt.x = rc.left + 8;
671 pt.y = rc.top + 8;
672 ClientToScreen(g_pChildWnd->hTreeWnd, &pt);
673 hti.flags = TVHT_ONITEM;
674 }
675 else
676 hti.flags = 0;
677 }
678 else
679 {
680 hti.pt.x = pt.x;
681 hti.pt.y = pt.y;
682 ScreenToClient(g_pChildWnd->hTreeWnd, &hti.pt);
683 (void)TreeView_HitTest(g_pChildWnd->hTreeWnd, &hti);
684 }
685
686 if (hti.flags & TVHT_ONITEM)
687 {
688 hContextMenu = GetSubMenu(hPopupMenus, PM_TREECONTEXT);
689 (void)TreeView_SelectItem(g_pChildWnd->hTreeWnd, hti.hItem);
690
691 memset(&item, 0, sizeof(item));
692 item.mask = TVIF_STATE | TVIF_CHILDREN;
693 item.hItem = hti.hItem;
694 (void)TreeView_GetItem(g_pChildWnd->hTreeWnd, &item);
695
696 /* Set the Expand/Collapse menu item appropriately */
697 LoadStringW(hInst, (item.state & TVIS_EXPANDED) ? IDS_COLLAPSE : IDS_EXPAND, buffer, COUNT_OF(buffer));
698 memset(&mii, 0, sizeof(mii));
699 mii.cbSize = sizeof(mii);
700 mii.fMask = MIIM_STRING | MIIM_STATE | MIIM_ID;
701 mii.fState = (item.cChildren > 0) ? MFS_DEFAULT : MFS_GRAYED;
702 mii.wID = (item.state & TVIS_EXPANDED) ? ID_TREE_COLLAPSEBRANCH : ID_TREE_EXPANDBRANCH;
703 mii.dwTypeData = (LPWSTR) buffer;
704 SetMenuItemInfo(hContextMenu, 0, TRUE, &mii);
705
706 /* Remove any existing suggestions */
707 memset(&mii, 0, sizeof(mii));
708 mii.cbSize = sizeof(mii);
709 mii.fMask = MIIM_ID;
710 GetMenuItemInfo(hContextMenu, GetMenuItemCount(hContextMenu) - 1, TRUE, &mii);
711 if ((mii.wID >= ID_TREE_SUGGESTION_MIN) && (mii.wID <= ID_TREE_SUGGESTION_MAX))
712 {
713 do
714 {
715 iLastPos = GetMenuItemCount(hContextMenu) - 1;
716 GetMenuItemInfo(hContextMenu, iLastPos, TRUE, &mii);
717 RemoveMenu(hContextMenu, iLastPos, MF_BYPOSITION);
718 }
719 while((mii.wID >= ID_TREE_SUGGESTION_MIN) && (mii.wID <= ID_TREE_SUGGESTION_MAX));
720 }
721
722 /* Come up with suggestions */
723 keyPath = GetItemPath(g_pChildWnd->hTreeWnd, NULL, &hRootKey);
724 SuggestKeys(hRootKey, keyPath, Suggestions, COUNT_OF(Suggestions));
725 if (Suggestions[0])
726 {
727 AppendMenu(hContextMenu, MF_SEPARATOR, 0, NULL);
728
729 LoadStringW(hInst, IDS_GOTO_SUGGESTED_KEY, resource, COUNT_OF(resource));
730
731 s = Suggestions;
732 wID = ID_TREE_SUGGESTION_MIN;
733 while(*s && (wID <= ID_TREE_SUGGESTION_MAX))
734 {
735 _snwprintf(buffer, COUNT_OF(buffer), resource, s);
736
737 memset(&mii, 0, sizeof(mii));
738 mii.cbSize = sizeof(mii);
739 mii.fMask = MIIM_STRING | MIIM_ID;
740 mii.wID = wID++;
741 mii.dwTypeData = buffer;
742 InsertMenuItem(hContextMenu, GetMenuItemCount(hContextMenu), TRUE, &mii);
743
744 s += wcslen(s) + 1;
745 }
746 }
747 TrackPopupMenu(hContextMenu, TPM_RIGHTBUTTON, pt.x, pt.y, 0, g_pChildWnd->hWnd, NULL);
748 }
749 }
750 break;
751 }
752
753 case WM_SIZE:
754 if (wParam != SIZE_MINIMIZED && g_pChildWnd != NULL)
755 {
756 ResizeWnd(LOWORD(lParam), HIWORD(lParam));
757 }
758 /* fall through */
759 default:
760 def:
761 return DefWindowProcW(hWnd, message, wParam, lParam);
762 }
763 return 0;
764 }