programs/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 St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #define WIN32_LEAN_AND_MEAN     /* Exclude rarely-used stuff from Windows headers */
  22 #include <windows.h>
  23 #include <commctrl.h>
  24 #include <stdio.h>
  25
  26 #include "main.h"
  27 #include "regproc.h"
  28
  29 #include "wine/debug.h"
  30 #include "wine/unicode.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(regedit);
  33
  34 ChildWnd* g_pChildWnd;
  35 static int last_split;
  36
  37 static const WCHAR wszLastKey[] = {'L','a','s','t','K','e','y',0};
  38 static const WCHAR wszKeyName[] = {'S','o','f','t','w','a','r','e','\\',
  39                                    'M','i','c','r','o','s','o','f','t','\\',
  40                                    'W','i','n','d','o','w','s','\\',
  41                                    'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
  42                                    'A','p','p','l','e','t','s','\\','R','e','g','e','d','i','t',0};
  43
  44 /*******************************************************************************
  45  * Local module support methods
  46  */
  47
  48 static LPCWSTR GetRootKeyName(HKEY hRootKey)
  49 {
  50     if(hRootKey == HKEY_CLASSES_ROOT)
  51         return reg_class_namesW[INDEX_HKEY_CLASSES_ROOT];
  52     if(hRootKey == HKEY_CURRENT_USER)
  53         return reg_class_namesW[INDEX_HKEY_CURRENT_USER];
  54     if(hRootKey == HKEY_LOCAL_MACHINE)
  55         return reg_class_namesW[INDEX_HKEY_LOCAL_MACHINE];
  56     if(hRootKey == HKEY_USERS)
  57         return reg_class_namesW[INDEX_HKEY_USERS];
  58     if(hRootKey == HKEY_CURRENT_CONFIG)
  59         return reg_class_namesW[INDEX_HKEY_CURRENT_CONFIG];
  60     if(hRootKey == HKEY_DYN_DATA)
  61         return reg_class_namesW[INDEX_HKEY_DYN_DATA];
  62     else
  63     {
  64         static const WCHAR unknown_key[] = {'U','N','K','N','O','W','N',' ','H','K','E','Y',',',' ',
  65                                             'P','L','E','A','S','E',' ','R','E','P','O','R','T',0};
  66         return unknown_key;
  67     }
  68 }
  69
  70 static void draw_splitbar(HWND hWnd, int x)
  71 {
  72     RECT rt;
  73     HDC hdc = GetDC(hWnd);
  74
  75     GetClientRect(hWnd, &rt);
  76     rt.left = x - SPLIT_WIDTH/2;
  77     rt.right = x + SPLIT_WIDTH/2+1;
  78     InvertRect(hdc, &rt);
  79     ReleaseDC(hWnd, hdc);
  80 }
  81
  82 static void ResizeWnd(int cx, int cy)
  83 {
  84     HDWP hdwp = BeginDeferWindowPos(2);
  85     RECT rt = {0, 0, cx, cy};
  86
  87     cx = g_pChildWnd->nSplitPos + SPLIT_WIDTH/2;
  88     DeferWindowPos(hdwp, g_pChildWnd->hTreeWnd, 0, rt.left, rt.top, g_pChildWnd->nSplitPos-SPLIT_WIDTH/2-rt.left, rt.bottom-rt.top, SWP_NOZORDER|SWP_NOACTIVATE);
  89     DeferWindowPos(hdwp, g_pChildWnd->hListWnd, 0, rt.left+cx  , rt.top, rt.right-cx, rt.bottom-rt.top, SWP_NOZORDER|SWP_NOACTIVATE);
  90     EndDeferWindowPos(hdwp);
  91 }
  92
  93 static void OnPaint(HWND hWnd)
  94 {
  95     PAINTSTRUCT ps;
  96     RECT rt;
  97
  98     GetClientRect(hWnd, &rt);
  99     BeginPaint(hWnd, &ps);
 100     FillRect(ps.hdc, &rt, GetSysColorBrush(COLOR_BTNFACE));
 101     EndPaint(hWnd, &ps);
 102 }
 103
 104 static LPWSTR CombinePaths(LPCWSTR pPaths[], int nPaths) {
 105     int i, len, pos;
 106     LPWSTR combined;
 107     for (i=0, len=0; i<nPaths; i++) {
 108         if (pPaths[i] && *pPaths[i]) {
 109             len += lstrlenW(pPaths[i])+1;
 110         }
 111     }
 112     combined = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 113     *combined = '\0';
 114     for (i=0, pos=0; i<nPaths; i++) {
 115         if (pPaths[i] && *pPaths[i]) {
 116             int llen = lstrlenW(pPaths[i]);
 117             if (!*combined)
 118                 lstrcpyW(combined, pPaths[i]);
 119             else {
 120                 combined[pos++] = '\\';
 121                 lstrcpyW(combined+pos, pPaths[i]);
 122             }
 123             pos += llen;
 124         }
 125     }
 126     return combined;
 127 }
 128
 129 static LPWSTR GetPathRoot(HWND hwndTV, HTREEITEM hItem, BOOL bFull) {
 130     LPCWSTR parts[2] = {0,0};
 131     WCHAR text[260];
 132     HKEY hRootKey = NULL;
 133     if (!hItem)
 134         hItem = (HTREEITEM)SendMessageW(hwndTV, TVM_GETNEXTITEM, TVGN_CARET, 0);
 135     HeapFree(GetProcessHeap(), 0, GetItemPath(hwndTV, hItem, &hRootKey));
 136     if (!bFull && !hRootKey)
 137         return NULL;
 138     if (hRootKey)
 139         parts[1] = GetRootKeyName(hRootKey);
 140     if (bFull) {
 141         DWORD dwSize = sizeof(text)/sizeof(WCHAR);
 142         GetComputerNameW(text, &dwSize);
 143         parts[0] = text;
 144     }
 145     return CombinePaths(parts, 2);
 146 }
 147
 148 LPWSTR GetItemFullPath(HWND hwndTV, HTREEITEM hItem, BOOL bFull) {
 149     LPWSTR parts[2];
 150     LPWSTR ret;
 151     HKEY hRootKey = NULL;
 152
 153     parts[0] = GetPathRoot(hwndTV, hItem, bFull);
 154     parts[1] = GetItemPath(hwndTV, hItem, &hRootKey);
 155     ret = CombinePaths((LPCWSTR *)parts, 2);
 156     HeapFree(GetProcessHeap(), 0, parts[0]);
 157     HeapFree(GetProcessHeap(), 0, parts[1]);
 158     return ret;
 159 }
 160
 161 static LPWSTR GetPathFullPath(HWND hwndTV, LPWSTR path) {
 162     LPWSTR parts[2];
 163     LPWSTR ret;
 164
 165     parts[0] = GetPathRoot(hwndTV, 0, TRUE);
 166     parts[1] = path;
 167     ret = CombinePaths((LPCWSTR*)parts, 2);
 168     HeapFree(GetProcessHeap(), 0, parts[0]);
 169     return ret;
 170 }
 171
 172 static void OnTreeSelectionChanged(HWND hwndTV, HWND hwndLV, HTREEITEM hItem, BOOL bRefreshLV)
 173 {
 174     if (bRefreshLV) {
 175         LPWSTR keyPath;
 176         HKEY hRootKey = NULL;
 177         HTREEITEM rootitem;
 178
 179         rootitem = (HTREEITEM)SendMessageW(hwndTV, TVM_GETNEXTITEM, TVGN_ROOT, 0);
 180         if (rootitem == hItem)
 181         {
 182             SendMessageW(hwndLV, LVM_DELETEALLITEMS, 0, 0);
 183             UpdateStatusBar();
 184             return;
 185         }
 186
 187         keyPath = GetItemPath(hwndTV, hItem, &hRootKey);
 188         RefreshListView(hwndLV, hRootKey, keyPath, NULL);
 189         HeapFree(GetProcessHeap(), 0, keyPath);
 190     }
 191     UpdateStatusBar();
 192 }
 193
 194 /*******************************************************************************
 195  * finish_splitbar [internal]
 196  *
 197  * make the splitbar invisible and resize the windows
 198  * (helper for ChildWndProc)
 199  */
 200 static void finish_splitbar(HWND hWnd, int x)
 201 {
 202     RECT rt;
 203
 204     draw_splitbar(hWnd, last_split);
 205     last_split = -1;
 206     GetClientRect(hWnd, &rt);
 207     g_pChildWnd->nSplitPos = x;
 208     ResizeWnd(rt.right, rt.bottom);
 209     ReleaseCapture();
 210 }
 211
 212 /*******************************************************************************
 213  *
 214  *  FUNCTION: _CmdWndProc(HWND, unsigned, WORD, LONG)
 215  *
 216  *  PURPOSE:  Processes WM_COMMAND messages for the main frame window.
 217  *
 218  */
 219
 220 static BOOL _CmdWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
 221 {
 222     switch (LOWORD(wParam)) {
 223         /* Parse the menu selections: */
 224     case ID_REGISTRY_EXIT:
 225         DestroyWindow(hWnd);
 226         break;
 227     case ID_VIEW_REFRESH:
 228         WINE_TRACE("Is this ever called or is it just dead code?\n");
 229         /* TODO */
 230         break;
 231     case ID_SWITCH_PANELS:
 232         g_pChildWnd->nFocusPanel = !g_pChildWnd->nFocusPanel;
 233         SetFocus(g_pChildWnd->nFocusPanel? g_pChildWnd->hListWnd: g_pChildWnd->hTreeWnd);
 234         break;
 235     default:
 236         return FALSE;
 237     }
 238     return TRUE;
 239 }
 240
 241 /*******************************************************************************
 242  * get_last_key [internal]
 243  *
 244  * open last key
 245  *
 246  */
 247 static void get_last_key(HWND hwndTV)
 248 {
 249     HKEY hkey;
 250     WCHAR wszVal[KEY_MAX_LEN];
 251     DWORD dwSize = sizeof(wszVal);
 252
 253     if (RegCreateKeyExW(HKEY_CURRENT_USER, wszKeyName, 0, NULL, 0, KEY_READ, NULL, &hkey, NULL) == ERROR_SUCCESS)
 254     {
 255         if (RegQueryValueExW(hkey, wszLastKey, NULL, NULL, (LPBYTE)wszVal, &dwSize) == ERROR_SUCCESS)
 256         {
 257             HTREEITEM selection;
 258             if (!strcmpW(wszVal, g_pChildWnd->szPath))
 259                 selection = (HTREEITEM)SendMessageW(g_pChildWnd->hTreeWnd, TVM_GETNEXTITEM, TVGN_ROOT, 0);
 260             else
 261                 selection = FindPathInTree(hwndTV, wszVal);
 262             SendMessageW(hwndTV, TVM_SELECTITEM, TVGN_CARET, (LPARAM)selection);
 263         }
 264
 265         RegCloseKey(hkey);
 266     }
 267 }
 268
 269 /*******************************************************************************
 270  * set_last_key [internal]
 271  *
 272  * save last key
 273  *
 274  */
 275 static void set_last_key(HWND hwndTV)
 276 {
 277     HKEY hkey;
 278
 279     if (RegCreateKeyExW(HKEY_CURRENT_USER, wszKeyName, 0, NULL, 0, KEY_WRITE, NULL, &hkey, NULL) == ERROR_SUCCESS)
 280     {
 281         HTREEITEM selection = (HTREEITEM)SendMessageW(g_pChildWnd->hTreeWnd, TVM_GETNEXTITEM, TVGN_CARET, 0);
 282         HTREEITEM root = (HTREEITEM)SendMessageW(g_pChildWnd->hTreeWnd, TVM_GETNEXTITEM, TVGN_ROOT, 0);
 283         WCHAR *value;
 284
 285         if (selection == root)
 286             value = g_pChildWnd->szPath;
 287         else
 288             value = GetItemFullPath(g_pChildWnd->hTreeWnd, selection, FALSE);
 289         RegSetValueExW(hkey, wszLastKey, 0, REG_SZ, (LPBYTE)value, (lstrlenW(value) + 1) * sizeof(WCHAR));
 290         if (selection != root)
 291             HeapFree(GetProcessHeap(), 0, value);
 292         RegCloseKey(hkey);
 293     }
 294 }
 295
 296 #define GET_X_LPARAM(lp) ((int)(short)LOWORD(lp))
 297 #define GET_Y_LPARAM(lp) ((int)(short)HIWORD(lp))
 298
 299 /*******************************************************************************
 300  *
 301  *  FUNCTION: ChildWndProc(HWND, unsigned, WORD, LONG)
 302  *
 303  *  PURPOSE:  Processes messages for the child windows.
 304  *
 305  *  WM_COMMAND  - process the application menu
 306  *  WM_PAINT    - Paint the main window
 307  *  WM_DESTROY  - post a quit message and return
 308  *
 309  */
 310 LRESULT CALLBACK ChildWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
 311 {
 312     switch (message) {
 313     case WM_CREATE:
 314         g_pChildWnd = HeapAlloc(GetProcessHeap(), 0, sizeof(ChildWnd));
 315         if (!g_pChildWnd) return 0;
 316         LoadStringW(hInst, IDS_REGISTRY_ROOT_NAME, g_pChildWnd->szPath, MAX_PATH);
 317         g_pChildWnd->nSplitPos = 250;
 318         g_pChildWnd->hWnd = hWnd;
 319         g_pChildWnd->hTreeWnd = CreateTreeView(hWnd, g_pChildWnd->szPath, TREE_WINDOW);
 320         g_pChildWnd->hListWnd = CreateListView(hWnd, LIST_WINDOW/*, g_pChildWnd->szPath*/);
 321         g_pChildWnd->nFocusPanel = 1;
 322         SetFocus(g_pChildWnd->hTreeWnd);
 323         get_last_key(g_pChildWnd->hTreeWnd);
 324         break;
 325     case WM_COMMAND:
 326         if (!_CmdWndProc(hWnd, message, wParam, lParam)) {
 327             goto def;
 328         }
 329         break;
 330     case WM_PAINT:
 331         OnPaint(hWnd);
 332         return 0;
 333     case WM_SETCURSOR:
 334         if (LOWORD(lParam) == HTCLIENT) {
 335             POINT pt;
 336             GetCursorPos(&pt);
 337             ScreenToClient(hWnd, &pt);
 338             if (pt.x>=g_pChildWnd->nSplitPos-SPLIT_WIDTH/2 && pt.x<g_pChildWnd->nSplitPos+SPLIT_WIDTH/2+1) {
 339                 SetCursor(LoadCursorW(0, (LPCWSTR)IDC_SIZEWE));
 340                 return TRUE;
 341             }
 342         }
 343         goto def;
 344     case WM_DESTROY:
 345         set_last_key(g_pChildWnd->hTreeWnd);
 346         HeapFree(GetProcessHeap(), 0, g_pChildWnd);
 347         g_pChildWnd = NULL;
 348         PostQuitMessage(0);
 349         break;
 350     case WM_LBUTTONDOWN: {
 351             RECT rt;
 352             int x = (short)LOWORD(lParam);
 353             GetClientRect(hWnd, &rt);
 354             if (x>=g_pChildWnd->nSplitPos-SPLIT_WIDTH/2 && x<g_pChildWnd->nSplitPos+SPLIT_WIDTH/2+1) {
 355                 last_split = g_pChildWnd->nSplitPos;
 356                 draw_splitbar(hWnd, last_split);
 357                 SetCapture(hWnd);
 358             }
 359             break;
 360         }
 361
 362     /* WM_RBUTTONDOWN sets the splitbar the same way as WM_LBUTTONUP */
 363     case WM_LBUTTONUP:
 364     case WM_RBUTTONDOWN:
 365         if (GetCapture() == hWnd) {
 366             finish_splitbar(hWnd, LOWORD(lParam));
 367         }
 368         break;
 369
 370     case WM_CAPTURECHANGED:
 371         if (GetCapture()==hWnd && last_split>=0)
 372             draw_splitbar(hWnd, last_split);
 373         break;
 374
 375     case WM_CONTEXTMENU: {
 376         POINT pt = {GET_X_LPARAM(lParam), GET_Y_LPARAM(lParam)};
 377         TVHITTESTINFO ht;
 378         ht.pt = pt;
 379         ScreenToClient(g_pChildWnd->hTreeWnd, &ht.pt);
 380         if (SendMessageW(g_pChildWnd->hTreeWnd, TVM_HITTEST, 0, (LPARAM)&ht)) {
 381             HTREEITEM root;
 382             SendMessageW(g_pChildWnd->hTreeWnd, TVM_SELECTITEM, TVGN_CARET, (LPARAM)ht.hItem);
 383             root = (HTREEITEM)SendMessageW(g_pChildWnd->hTreeWnd, TVM_GETNEXTITEM, TVGN_ROOT, 0);
 384             TrackPopupMenu(GetSubMenu(hPopupMenus, ht.hItem == root ? PM_COMPUTER : PM_TREEVIEW),
 385                            TPM_RIGHTBUTTON, pt.x, pt.y, 0, hFrameWnd, NULL);
 386         }
 387         break;
 388     }
 389
 390     case WM_KEYDOWN:
 391         if (wParam == VK_ESCAPE)
 392             if (GetCapture() == hWnd) {
 393                 RECT rt;
 394                 draw_splitbar(hWnd, last_split);
 395                 GetClientRect(hWnd, &rt);
 396                 ResizeWnd(rt.right, rt.bottom);
 397                 last_split = -1;
 398                 ReleaseCapture();
 399                 SetCursor(LoadCursorW(0, (LPCWSTR)IDC_ARROW));
 400             }
 401         break;
 402
 403     case WM_MOUSEMOVE:
 404         if (GetCapture() == hWnd) {
 405             RECT rt;
 406             int x = LOWORD(lParam);
 407             HDC hdc = GetDC(hWnd);
 408             GetClientRect(hWnd, &rt);
 409             rt.left = last_split-SPLIT_WIDTH/2;
 410             rt.right = last_split+SPLIT_WIDTH/2+1;
 411             InvertRect(hdc, &rt);
 412             last_split = x;
 413             rt.left = x-SPLIT_WIDTH/2;
 414             rt.right = x+SPLIT_WIDTH/2+1;
 415             InvertRect(hdc, &rt);
 416             ReleaseDC(hWnd, hdc);
 417         }
 418         break;
 419
 420     case WM_SETFOCUS:
 421         if (g_pChildWnd != NULL) {
 422             SetFocus(g_pChildWnd->nFocusPanel? g_pChildWnd->hListWnd: g_pChildWnd->hTreeWnd);
 423         }
 424         break;
 425
 426     case WM_TIMER:
 427         break;
 428
 429     case WM_NOTIFY:
 430         if (((int)wParam == TREE_WINDOW) && (g_pChildWnd != NULL)) {
 431             switch (((LPNMHDR)lParam)->code) {
 432             case TVN_ITEMEXPANDINGW:
 433                 return !OnTreeExpanding(g_pChildWnd->hTreeWnd, (NMTREEVIEWW*)lParam);
 434             case TVN_SELCHANGEDW:
 435                 OnTreeSelectionChanged(g_pChildWnd->hTreeWnd, g_pChildWnd->hListWnd,
 436                     ((NMTREEVIEWW *)lParam)->itemNew.hItem, TRUE);
 437                 break;
 438             case NM_SETFOCUS:
 439                 g_pChildWnd->nFocusPanel = 0;
 440                 break;
 441             case TVN_BEGINLABELEDITW: {
 442                 HKEY hRootKey;
 443                 LPWSTR path;
 444
 445                 if (!GetWindowLongPtrW(g_pChildWnd->hTreeWnd, GWLP_USERDATA))
 446                     return 1;
 447
 448                 path = GetItemPath(g_pChildWnd->hTreeWnd, 0, &hRootKey);
 449                 if (!path || !*path) return 1;
 450                 return 0;
 451             }
 452             case TVN_ENDLABELEDITW: {
 453                 HKEY hRootKey;
 454                 LPNMTVDISPINFOW dispInfo = (LPNMTVDISPINFOW)lParam;
 455                 LPWSTR path = GetItemPath(g_pChildWnd->hTreeWnd, 0, &hRootKey);
 456                 BOOL res = RenameKey(hWnd, hRootKey, path, dispInfo->item.pszText);
 457                 if (res) {
 458                     TVITEMEXW item;
 459                     LPWSTR fullPath = GetPathFullPath(g_pChildWnd->hTreeWnd,
 460                      dispInfo->item.pszText);
 461                     item.mask = TVIF_HANDLE | TVIF_TEXT;
 462                     item.hItem = (HTREEITEM)SendMessageW(g_pChildWnd->hTreeWnd, TVM_GETNEXTITEM, TVGN_CARET, 0);
 463                     item.pszText = dispInfo->item.pszText;
 464                     SendMessageW( g_pChildWnd->hTreeWnd, TVM_SETITEMW, 0, (LPARAM)&item );
 465                     SendMessageW(hStatusBar, SB_SETTEXTW, 0, (LPARAM)fullPath);
 466                     HeapFree(GetProcessHeap(), 0, fullPath);
 467                 }
 468                 HeapFree(GetProcessHeap(), 0, path);
 469                 SetWindowLongPtrW(g_pChildWnd->hTreeWnd, GWLP_USERDATA, 0);
 470                 return res;
 471             }
 472             default:
 473                 return 0; /* goto def; */
 474             }
 475         } else if ((int)wParam == LIST_WINDOW && g_pChildWnd != NULL) {
 476             return SendMessageW(g_pChildWnd->hListWnd, WM_NOTIFY_REFLECT, wParam, lParam);
 477         }
 478         break;
 479
 480     case WM_SIZE:
 481         if (wParam != SIZE_MINIMIZED && g_pChildWnd != NULL) {
 482             ResizeWnd(LOWORD(lParam), HIWORD(lParam));
 483         }
 484         /* fall through */
 485 default: def:
 486         return DefWindowProcW(hWnd, message, wParam, lParam);
 487     }
 488     return 0;
 489 }