oleaut32/tests: Replace some macros by functions to reduce code size and compile...
[wine] / 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 }