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 <tchar.h>
  25 #include <stdio.h>
  26
  27 #include "main.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
  36 /*******************************************************************************
  37  * Local module support methods
  38  */
  39
  40 LPCTSTR GetRootKeyName(HKEY hRootKey)
  41 {
  42     if (hRootKey == HKEY_CLASSES_ROOT) return _T("HKEY_CLASSES_ROOT");
  43     if (hRootKey == HKEY_CURRENT_USER) return _T("HKEY_CURRENT_USER");
  44     if (hRootKey == HKEY_LOCAL_MACHINE) return _T("HKEY_LOCAL_MACHINE");
  45     if (hRootKey == HKEY_USERS) return _T("HKEY_USERS");
  46     if (hRootKey == HKEY_CURRENT_CONFIG) return _T("HKEY_CURRENT_CONFIG");
  47     if (hRootKey == HKEY_DYN_DATA) return _T("HKEY_DYN_DATA");
  48     return _T("UNKNOWN HKEY, PLEASE REPORT");
  49 }
  50
  51 static void draw_splitbar(HWND hWnd, int x)
  52 {
  53     RECT rt;
  54     HDC hdc = GetDC(hWnd);
  55
  56     GetClientRect(hWnd, &rt);
  57     rt.left = x - SPLIT_WIDTH/2;
  58     rt.right = x + SPLIT_WIDTH/2+1;
  59     InvertRect(hdc, &rt);
  60     ReleaseDC(hWnd, hdc);
  61 }
  62
  63 static void ResizeWnd(ChildWnd* pChildWnd, int cx, int cy)
  64 {
  65     HDWP hdwp = BeginDeferWindowPos(2);
  66     RECT rt = {0, 0, cx, cy};
  67
  68     cx = pChildWnd->nSplitPos + SPLIT_WIDTH/2;
  69     DeferWindowPos(hdwp, pChildWnd->hTreeWnd, 0, rt.left, rt.top, pChildWnd->nSplitPos-SPLIT_WIDTH/2-rt.left, rt.bottom-rt.top, SWP_NOZORDER|SWP_NOACTIVATE);
  70     DeferWindowPos(hdwp, pChildWnd->hListWnd, 0, rt.left+cx  , rt.top, rt.right-cx, rt.bottom-rt.top, SWP_NOZORDER|SWP_NOACTIVATE);
  71     EndDeferWindowPos(hdwp);
  72 }
  73
  74 static void OnPaint(HWND hWnd)
  75 {
  76     PAINTSTRUCT ps;
  77     RECT rt;
  78     HDC hdc;
  79
  80     GetClientRect(hWnd, &rt);
  81     hdc = BeginPaint(hWnd, &ps);
  82     FillRect(ps.hdc, &rt, GetSysColorBrush(COLOR_BTNFACE));
  83     EndPaint(hWnd, &ps);
  84 }
  85
  86 static LPTSTR CombinePaths(LPCTSTR pPaths[], int nPaths) {
  87     int i, len, pos;
  88     LPTSTR combined;
  89     for (i=0, len=0; i<nPaths; i++) {
  90         if (pPaths[i] && *pPaths[i]) {
  91             len += lstrlen(pPaths[i])+1;
  92         }
  93     }
  94     combined = HeapAlloc(GetProcessHeap(), 0, len * sizeof(TCHAR));
  95     *combined = '\0';
  96     for (i=0, pos=0; i<nPaths; i++) {
  97         if (pPaths[i] && *pPaths[i]) {
  98             int llen = _tcslen(pPaths[i]);
  99             if (!*combined)
 100                 _tcscpy(combined, pPaths[i]);
 101             else {
 102                 combined[pos++] = (TCHAR)'\\';
 103                 _tcscpy(combined+pos, pPaths[i]);
 104             }
 105             pos += llen;
 106         }
 107     }
 108     return combined;
 109 }
 110
 111 static LPTSTR GetPathRoot(HWND hwndTV, HTREEITEM hItem, BOOL bFull) {
 112     LPCTSTR parts[2] = {_T(""), _T("")};
 113     TCHAR text[260];
 114     HKEY hRootKey = NULL;
 115     if (!hItem)
 116         hItem = TreeView_GetSelection(hwndTV);
 117     GetItemPath(hwndTV, hItem, &hRootKey);
 118     if (!bFull && !hRootKey)
 119         return NULL;
 120     if (hRootKey)
 121         parts[1] = GetRootKeyName(hRootKey);
 122     if (bFull) {
 123         DWORD dwSize = sizeof(text)/sizeof(TCHAR);
 124         GetComputerName(text, &dwSize);
 125         parts[0] = text;
 126     }
 127     return CombinePaths(parts, 2);
 128 }
 129
 130 LPTSTR GetItemFullPath(HWND hwndTV, HTREEITEM hItem, BOOL bFull) {
 131     LPTSTR parts[2];
 132     LPTSTR ret;
 133     HKEY hRootKey = NULL;
 134
 135     parts[0] = GetPathRoot(hwndTV, hItem, bFull);
 136     parts[1] = GetItemPath(hwndTV, hItem, &hRootKey);
 137     ret = CombinePaths((LPCTSTR *)parts, 2);
 138     HeapFree(GetProcessHeap(), 0, parts[0]);
 139     return ret;
 140 }
 141
 142 LPTSTR GetPathFullPath(HWND hwndTV, LPTSTR path) {
 143     LPTSTR parts[2];
 144     LPTSTR ret;
 145
 146     parts[0] = GetPathRoot(hwndTV, 0, TRUE);
 147     parts[1] = path;
 148     ret = CombinePaths((LPCTSTR *)parts, 2);
 149     HeapFree(GetProcessHeap(), 0, parts[0]);
 150     return ret;
 151 }
 152
 153 static void OnTreeSelectionChanged(HWND hwndTV, HWND hwndLV, HTREEITEM hItem, BOOL bRefreshLV)
 154 {
 155     if (bRefreshLV) {
 156         LPCTSTR keyPath;
 157         HKEY hRootKey = NULL;
 158         keyPath = GetItemPath(hwndTV, hItem, &hRootKey);
 159         RefreshListView(hwndLV, hRootKey, keyPath, NULL);
 160     }
 161     UpdateStatusBar();
 162 }
 163
 164 /*******************************************************************************
 165  *
 166  *  FUNCTION: _CmdWndProc(HWND, unsigned, WORD, LONG)
 167  *
 168  *  PURPOSE:  Processes WM_COMMAND messages for the main frame window.
 169  *
 170  */
 171
 172 static BOOL _CmdWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
 173 {
 174     ChildWnd* pChildWnd = g_pChildWnd;
 175     switch (LOWORD(wParam)) {
 176         /* Parse the menu selections: */
 177     case ID_REGISTRY_EXIT:
 178         DestroyWindow(hWnd);
 179         break;
 180     case ID_VIEW_REFRESH:
 181         WINE_TRACE("Is this ever called or is it just dead code?\n");
 182         /* TODO */
 183         break;
 184     case ID_SWITCH_PANELS:
 185         pChildWnd->nFocusPanel = !pChildWnd->nFocusPanel;
 186         SetFocus(pChildWnd->nFocusPanel? pChildWnd->hListWnd: pChildWnd->hTreeWnd);
 187         break;
 188     default:
 189         return FALSE;
 190     }
 191     return TRUE;
 192 }
 193
 194 /*******************************************************************************
 195  *
 196  *  FUNCTION: ChildWndProc(HWND, unsigned, WORD, LONG)
 197  *
 198  *  PURPOSE:  Processes messages for the child windows.
 199  *
 200  *  WM_COMMAND  - process the application menu
 201  *  WM_PAINT    - Paint the main window
 202  *  WM_DESTROY  - post a quit message and return
 203  *
 204  */
 205 LRESULT CALLBACK ChildWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
 206 {
 207     static int last_split;
 208     ChildWnd* pChildWnd = g_pChildWnd;
 209
 210     switch (message) {
 211     case WM_CREATE:
 212         g_pChildWnd = pChildWnd = HeapAlloc(GetProcessHeap(), 0, sizeof(ChildWnd));
 213         if (!pChildWnd) return 0;
 214         LoadString(hInst, IDS_REGISTRY_ROOT_NAME, pChildWnd->szPath, MAX_PATH);
 215         pChildWnd->nSplitPos = 250;
 216         pChildWnd->hWnd = hWnd;
 217         pChildWnd->hTreeWnd = CreateTreeView(hWnd, pChildWnd->szPath, TREE_WINDOW);
 218         pChildWnd->hListWnd = CreateListView(hWnd, LIST_WINDOW/*, pChildWnd->szPath*/);
 219         pChildWnd->nFocusPanel = 1;
 220         SetFocus(pChildWnd->hTreeWnd);
 221         break;
 222     case WM_COMMAND:
 223         if (!_CmdWndProc(hWnd, message, wParam, lParam)) {
 224             goto def;
 225         }
 226         break;
 227     case WM_PAINT:
 228         OnPaint(hWnd);
 229         return 0;
 230     case WM_SETCURSOR:
 231         if (LOWORD(lParam) == HTCLIENT) {
 232             POINT pt;
 233             GetCursorPos(&pt);
 234             ScreenToClient(hWnd, &pt);
 235             if (pt.x>=pChildWnd->nSplitPos-SPLIT_WIDTH/2 && pt.x<pChildWnd->nSplitPos+SPLIT_WIDTH/2+1) {
 236                 SetCursor(LoadCursor(0, IDC_SIZEWE));
 237                 return TRUE;
 238             }
 239         }
 240         goto def;
 241     case WM_DESTROY:
 242         HeapFree(GetProcessHeap(), 0, pChildWnd);
 243         pChildWnd = NULL;
 244         PostQuitMessage(0);
 245         break;
 246     case WM_LBUTTONDOWN: {
 247             RECT rt;
 248             int x = (short)LOWORD(lParam);
 249             GetClientRect(hWnd, &rt);
 250             if (x>=pChildWnd->nSplitPos-SPLIT_WIDTH/2 && x<pChildWnd->nSplitPos+SPLIT_WIDTH/2+1) {
 251                 last_split = pChildWnd->nSplitPos;
 252                 draw_splitbar(hWnd, last_split);
 253                 SetCapture(hWnd);
 254             }
 255             break;
 256         }
 257
 258     case WM_LBUTTONUP:
 259         if (GetCapture() == hWnd) {
 260             RECT rt;
 261             int x = LOWORD(lParam);
 262             draw_splitbar(hWnd, last_split);
 263             last_split = -1;
 264             GetClientRect(hWnd, &rt);
 265             pChildWnd->nSplitPos = x;
 266             ResizeWnd(pChildWnd, rt.right, rt.bottom);
 267             ReleaseCapture();
 268         }
 269         break;
 270
 271     case WM_CAPTURECHANGED:
 272         if (GetCapture()==hWnd && last_split>=0)
 273             draw_splitbar(hWnd, last_split);
 274         break;
 275
 276     case WM_KEYDOWN:
 277         if (wParam == VK_ESCAPE)
 278             if (GetCapture() == hWnd) {
 279                 RECT rt;
 280                 draw_splitbar(hWnd, last_split);
 281                 GetClientRect(hWnd, &rt);
 282                 ResizeWnd(pChildWnd, rt.right, rt.bottom);
 283                 last_split = -1;
 284                 ReleaseCapture();
 285                 SetCursor(LoadCursor(0, IDC_ARROW));
 286             }
 287         break;
 288
 289     case WM_MOUSEMOVE:
 290         if (GetCapture() == hWnd) {
 291             RECT rt;
 292             int x = LOWORD(lParam);
 293             HDC hdc = GetDC(hWnd);
 294             GetClientRect(hWnd, &rt);
 295             rt.left = last_split-SPLIT_WIDTH/2;
 296             rt.right = last_split+SPLIT_WIDTH/2+1;
 297             InvertRect(hdc, &rt);
 298             last_split = x;
 299             rt.left = x-SPLIT_WIDTH/2;
 300             rt.right = x+SPLIT_WIDTH/2+1;
 301             InvertRect(hdc, &rt);
 302             ReleaseDC(hWnd, hdc);
 303         }
 304         break;
 305
 306     case WM_SETFOCUS:
 307         if (pChildWnd != NULL) {
 308             SetFocus(pChildWnd->nFocusPanel? pChildWnd->hListWnd: pChildWnd->hTreeWnd);
 309         }
 310         break;
 311
 312     case WM_TIMER:
 313         break;
 314
 315     case WM_NOTIFY:
 316         if ((int)wParam == TREE_WINDOW) {
 317             switch (((LPNMHDR)lParam)->code) {
 318             case TVN_ITEMEXPANDING:
 319                 return !OnTreeExpanding(pChildWnd->hTreeWnd, (NMTREEVIEW*)lParam);
 320             case TVN_SELCHANGED:
 321                 OnTreeSelectionChanged(pChildWnd->hTreeWnd, pChildWnd->hListWnd,
 322                     ((NMTREEVIEW *)lParam)->itemNew.hItem, TRUE);
 323                 break;
 324             case NM_SETFOCUS:
 325                 pChildWnd->nFocusPanel = 0;
 326                 break;
 327             case NM_RCLICK: {
 328                 POINT pt;
 329                 GetCursorPos(&pt);
 330                 TrackPopupMenu(GetSubMenu(hPopupMenus, PM_NEW),
 331                                TPM_RIGHTBUTTON, pt.x, pt.y, 0, hFrameWnd, NULL);
 332                 break;
 333             }
 334             case TVN_ENDLABELEDIT: {
 335                 HKEY hRootKey;
 336                 LPNMTVDISPINFO dispInfo = (LPNMTVDISPINFO)lParam;
 337                 LPCTSTR path = GetItemPath(pChildWnd->hTreeWnd, 0, &hRootKey);
 338                 BOOL res = RenameKey(hWnd, hRootKey, path, dispInfo->item.pszText);
 339                 if (res) {
 340                     TVITEMEX item;
 341                     LPTSTR fullPath = GetPathFullPath(pChildWnd->hTreeWnd,
 342                      dispInfo->item.pszText);
 343                     item.mask = TVIF_HANDLE | TVIF_TEXT;
 344                     item.hItem = TreeView_GetSelection(pChildWnd->hTreeWnd);
 345                     item.pszText = dispInfo->item.pszText;
 346                     SendMessage( pChildWnd->hTreeWnd, TVM_SETITEMW, 0, (LPARAM)&item );
 347                     SendMessage(hStatusBar, SB_SETTEXT, 0, (LPARAM)fullPath);
 348                     HeapFree(GetProcessHeap(), 0, fullPath);
 349                 }
 350                 return res;
 351             }
 352             default:
 353                 return 0; /* goto def; */
 354             }
 355         } else
 356             if ((int)wParam == LIST_WINDOW) {
 357                 if (((LPNMHDR)lParam)->code == NM_SETFOCUS) {
 358                     pChildWnd->nFocusPanel = 1;
 359                 } else if (!SendMessage(pChildWnd->hListWnd, WM_NOTIFY_REFLECT, wParam, lParam)) {
 360                     goto def;
 361                 }
 362             }
 363         break;
 364
 365     case WM_SIZE:
 366         if (wParam != SIZE_MINIMIZED && pChildWnd != NULL) {
 367             ResizeWnd(pChildWnd, LOWORD(lParam), HIWORD(lParam));
 368         }
 369         /* fall through */
 370 default: def:
 371         return DefWindowProc(hWnd, message, wParam, lParam);
 372     }
 373     return 0;
 374 }