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