shell32: Use the standard DEFINE_GUID macro to define GUIDs.
[wine] / dlls / shell32 / brsfolder.c
1 /*
2  * Copyright 1999 Juergen Schmied
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  *
18  * FIXME:
19  *  - many memory leaks
20  *  - many flags unimplemented
21  */
22
23 #include <stdlib.h>
24 #include <string.h>
25
26 #define COBJMACROS
27 #define NONAMELESSUNION
28 #define NONAMELESSSTRUCT
29
30 #include "wine/debug.h"
31 #include "undocshell.h"
32 #include "shlguid.h"
33 #include "pidl.h"
34 #include "shell32_main.h"
35 #include "shellapi.h"
36 #include "shresdef.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(shell);
39
40 typedef struct tagbrowse_info
41 {
42     HWND          hWnd;
43     HWND          hwndTreeView;
44     LPBROWSEINFOW lpBrowseInfo;
45     LPITEMIDLIST  pidlRet;
46 } browse_info;
47
48 typedef struct tagTV_ITEMDATA
49 {
50    LPSHELLFOLDER lpsfParent; /* IShellFolder of the parent */
51    LPITEMIDLIST  lpi;        /* PIDL relativ to parent */
52    LPITEMIDLIST  lpifq;      /* Fully qualified PIDL */
53    IEnumIDList*  pEnumIL;    /* Children iterator */ 
54 } TV_ITEMDATA, *LPTV_ITEMDATA;
55
56 #define SUPPORTEDFLAGS (BIF_STATUSTEXT | \
57                         BIF_BROWSEFORCOMPUTER | \
58                         BIF_RETURNFSANCESTORS | \
59                         BIF_RETURNONLYFSDIRS | \
60                         BIF_BROWSEINCLUDEFILES)
61
62 static void FillTreeView(browse_info*, LPSHELLFOLDER,
63                LPITEMIDLIST, HTREEITEM, IEnumIDList*);
64 static HTREEITEM InsertTreeViewItem( browse_info*, IShellFolder *,
65                LPCITEMIDLIST, LPCITEMIDLIST, IEnumIDList*, HTREEITEM);
66
67 const WCHAR szBrowseFolderInfo[] = {
68     '_','_','W','I','N','E','_',
69     'B','R','S','F','O','L','D','E','R','D','L','G','_',
70     'I','N','F','O',0
71 };
72
73 static inline DWORD BrowseFlagsToSHCONTF(UINT ulFlags)
74 {
75     return SHCONTF_FOLDERS | (ulFlags & BIF_BROWSEINCLUDEFILES ? SHCONTF_NONFOLDERS : 0);
76 }
77
78 static void browsefolder_callback( LPBROWSEINFOW lpBrowseInfo, HWND hWnd,
79                                    UINT msg, LPARAM param )
80 {
81     if (!lpBrowseInfo->lpfn)
82         return;
83     lpBrowseInfo->lpfn( hWnd, msg, param, lpBrowseInfo->lParam );
84 }
85
86 /******************************************************************************
87  * InitializeTreeView [Internal]
88  *
89  * Called from WM_INITDIALOG handler.
90  * 
91  * PARAMS
92  *  hwndParent [I] The BrowseForFolder dialog
93  *  root       [I] ITEMIDLIST of the root shell folder
94  */
95 static void InitializeTreeView( browse_info *info )
96 {
97     LPITEMIDLIST pidlParent, pidlChild;
98     HIMAGELIST hImageList;
99     HRESULT hr;
100     IShellFolder *lpsfParent, *lpsfRoot;
101     IEnumIDList * pEnumChildren = NULL;
102     HTREEITEM item;
103     DWORD flags;
104     LPCITEMIDLIST root = info->lpBrowseInfo->pidlRoot;
105
106     TRACE("%p\n", info );
107     
108     Shell_GetImageList(NULL, &hImageList);
109
110     if (hImageList)
111         SendMessageW( info->hwndTreeView, TVM_SETIMAGELIST, 0, (LPARAM)hImageList );
112
113     /* We want to call InsertTreeViewItem down the code, in order to insert
114      * the root item of the treeview. Due to InsertTreeViewItem's signature, 
115      * we need the following to do this:
116      *
117      * + An ITEMIDLIST corresponding to _the parent_ of root. 
118      * + An ITEMIDLIST, which is a relative path from root's parent to root 
119      *   (containing a single SHITEMID).
120      * + An IShellFolder interface pointer of root's parent folder.
121      *
122      * If root is 'Desktop', then root's parent is also 'Desktop'.
123      */
124
125     pidlParent = ILClone(root);
126     ILRemoveLastID(pidlParent);
127     pidlChild = ILClone(ILFindLastID(root));
128     
129     if (_ILIsDesktop(pidlParent)) {
130         hr = SHGetDesktopFolder(&lpsfParent);
131     } else {
132         IShellFolder *lpsfDesktop;
133         hr = SHGetDesktopFolder(&lpsfDesktop);
134         if (!SUCCEEDED(hr)) {
135             WARN("SHGetDesktopFolder failed! hr = %08lx\n", hr);
136             return;
137         }
138         hr = IShellFolder_BindToObject(lpsfDesktop, pidlParent, 0, &IID_IShellFolder, (LPVOID*)&lpsfParent);
139         IShellFolder_Release(lpsfDesktop);
140     }
141     
142     if (!SUCCEEDED(hr)) {
143         WARN("Could not bind to parent shell folder! hr = %08lx\n", hr);
144         return;
145     }
146
147     if (pidlChild && pidlChild->mkid.cb) {
148         hr = IShellFolder_BindToObject(lpsfParent, pidlChild, 0, &IID_IShellFolder, (LPVOID*)&lpsfRoot);
149     } else {
150         lpsfRoot = lpsfParent;
151         hr = IShellFolder_AddRef(lpsfParent);
152     }
153     
154     if (!SUCCEEDED(hr)) {
155         WARN("Could not bind to root shell folder! hr = %08lx\n", hr);
156         IShellFolder_Release(lpsfParent);
157         return;
158     }
159
160     flags = BrowseFlagsToSHCONTF( info->lpBrowseInfo->ulFlags );
161     hr = IShellFolder_EnumObjects( lpsfRoot, info->hWnd, flags, &pEnumChildren );
162     if (!SUCCEEDED(hr)) {
163         WARN("Could not get child iterator! hr = %08lx\n", hr);
164         IShellFolder_Release(lpsfParent);
165         IShellFolder_Release(lpsfRoot);
166         return;
167     }
168
169     SendMessageW( info->hwndTreeView, TVM_DELETEITEM, 0, (LPARAM)TVI_ROOT );
170     item = InsertTreeViewItem( info, lpsfParent, pidlChild,
171                                pidlParent, pEnumChildren, TVI_ROOT );
172     SendMessageW( info->hwndTreeView, TVM_EXPAND, TVE_EXPAND, (LPARAM)item );
173
174     IShellFolder_Release(lpsfRoot);
175     IShellFolder_Release(lpsfParent);
176 }
177
178 static int GetIcon(LPITEMIDLIST lpi, UINT uFlags)
179 {
180     SHFILEINFOW sfi;
181     SHGetFileInfoW((LPCWSTR)lpi, 0 ,&sfi, sizeof(SHFILEINFOW), uFlags);
182     return sfi.iIcon;
183 }
184
185 static void GetNormalAndSelectedIcons(LPITEMIDLIST lpifq, LPTVITEMW lpTV_ITEM)
186 {
187     LPITEMIDLIST pidlDesktop = NULL;
188     DWORD flags;
189
190     TRACE("%p %p\n",lpifq, lpTV_ITEM);
191
192     if (!lpifq)
193     {
194         pidlDesktop = _ILCreateDesktop();
195         lpifq = pidlDesktop;
196     }
197
198     flags = SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON;
199     lpTV_ITEM->iImage = GetIcon( lpifq, flags );
200
201     flags = SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_OPENICON;
202     lpTV_ITEM->iSelectedImage = GetIcon( lpifq, flags );
203
204     if (pidlDesktop)
205         ILFree( pidlDesktop );
206 }
207
208 /******************************************************************************
209  * GetName [Internal]
210  *
211  * Query a shell folder for the display name of one of it's children
212  *
213  * PARAMS
214  *  lpsf           [I] IShellFolder interface of the folder to be queried.
215  *  lpi            [I] ITEMIDLIST of the child, relative to parent
216  *  dwFlags        [I] as in IShellFolder::GetDisplayNameOf
217  *  lpFriendlyName [O] The desired display name in unicode
218  *
219  * RETURNS
220  *  Success: TRUE
221  *  Failure: FALSE
222  */
223 static BOOL GetName(LPSHELLFOLDER lpsf, LPCITEMIDLIST lpi, DWORD dwFlags, LPWSTR lpFriendlyName)
224 {
225         BOOL   bSuccess=TRUE;
226         STRRET str;
227
228         TRACE("%p %p %lx %p\n", lpsf, lpi, dwFlags, lpFriendlyName);
229         if (SUCCEEDED(IShellFolder_GetDisplayNameOf(lpsf, lpi, dwFlags, &str)))
230           bSuccess = StrRetToStrNW(lpFriendlyName, MAX_PATH, &str, lpi);
231         else
232           bSuccess = FALSE;
233
234         TRACE("-- %s\n", debugstr_w(lpFriendlyName));
235         return bSuccess;
236 }
237
238 /******************************************************************************
239  * InsertTreeViewItem [Internal]
240  *
241  * PARAMS
242  *  info       [I] data for the dialog
243  *  lpsf       [I] IShellFolder interface of the item's parent shell folder 
244  *  pidl       [I] ITEMIDLIST of the child to insert, relativ to parent 
245  *  pidlParent [I] ITEMIDLIST of the parent shell folder
246  *  pEnumIL    [I] Iterator for the children of the item to be inserted
247  *  hParent    [I] The treeview-item that represents the parent shell folder
248  *
249  * RETURNS
250  *  Success: Handle to the created and inserted treeview-item
251  *  Failure: NULL
252  */
253 static HTREEITEM InsertTreeViewItem( browse_info *info, IShellFolder * lpsf,
254     LPCITEMIDLIST pidl, LPCITEMIDLIST pidlParent, IEnumIDList* pEnumIL,
255     HTREEITEM hParent)
256 {
257         TVITEMW         tvi;
258         TVINSERTSTRUCTW tvins;
259         WCHAR           szBuff[MAX_PATH];
260         LPTV_ITEMDATA   lptvid=0;
261
262         tvi.mask  = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM;
263
264         tvi.cChildren= pEnumIL ? 1 : 0;
265         tvi.mask |= TVIF_CHILDREN;
266
267         lptvid = SHAlloc( sizeof(TV_ITEMDATA) );
268         if (!lptvid)
269             return NULL;
270
271         if (!GetName(lpsf, pidl, SHGDN_NORMAL, szBuff))
272             return NULL;
273
274         tvi.pszText    = szBuff;
275         tvi.cchTextMax = MAX_PATH;
276         tvi.lParam = (LPARAM)lptvid;
277
278         IShellFolder_AddRef(lpsf);
279         lptvid->lpsfParent = lpsf;
280         lptvid->lpi     = ILClone(pidl);
281         lptvid->lpifq   = pidlParent ? ILCombine(pidlParent, pidl) : ILClone(pidl);
282         lptvid->pEnumIL = pEnumIL;
283         GetNormalAndSelectedIcons(lptvid->lpifq, &tvi);
284
285         tvins.u.item       = tvi;
286         tvins.hInsertAfter = NULL;
287         tvins.hParent      = hParent;
288
289         return (HTREEITEM)TreeView_InsertItemW( info->hwndTreeView, &tvins );
290 }
291
292 /******************************************************************************
293  * FillTreeView [Internal]
294  *
295  * For each child (given by lpe) of the parent shell folder, which is given by 
296  * lpsf and whose PIDL is pidl, insert a treeview-item right under hParent
297  *
298  * PARAMS
299  *  info    [I] data for the dialog
300  *  lpsf    [I] IShellFolder interface of the parent shell folder
301  *  pidl    [I] ITEMIDLIST of the parent shell folder
302  *  hParent [I] The treeview item that represents the parent shell folder
303  *  lpe     [I] An iterator for the children of the parent shell folder
304  */
305 static void FillTreeView( browse_info *info, IShellFolder * lpsf,
306                  LPITEMIDLIST  pidl, HTREEITEM hParent, IEnumIDList* lpe)
307 {
308         HTREEITEM       hPrev = 0;
309         LPITEMIDLIST    pidlTemp = 0;
310         ULONG           ulFetched;
311         HRESULT         hr;
312         HWND            hwnd = GetParent( info->hwndTreeView );
313
314         TRACE("%p %p %p %p\n",lpsf, pidl, hParent, lpe);
315
316         /* No IEnumIDList -> No children */
317         if (!lpe) return;
318         
319         SetCapture( hwnd );
320         SetCursor( LoadCursorA( 0, (LPSTR)IDC_WAIT ) );
321
322         while (NOERROR == IEnumIDList_Next(lpe,1,&pidlTemp,&ulFetched))
323         {
324             ULONG ulAttrs = SFGAO_HASSUBFOLDER | SFGAO_FOLDER;
325             IEnumIDList* pEnumIL = NULL;
326             IShellFolder* pSFChild = NULL;
327             IShellFolder_GetAttributesOf(lpsf, 1, (LPCITEMIDLIST*)&pidlTemp, &ulAttrs);
328             if (ulAttrs & SFGAO_FOLDER)
329             {
330                 hr = IShellFolder_BindToObject(lpsf,pidlTemp,NULL,&IID_IShellFolder,(LPVOID*)&pSFChild);
331                 if (SUCCEEDED(hr))
332                 {
333                     DWORD flags = BrowseFlagsToSHCONTF(info->lpBrowseInfo->ulFlags);
334                     hr = IShellFolder_EnumObjects(pSFChild, hwnd, flags, &pEnumIL);
335                     if (hr == S_OK)
336                     {
337                         if ((IEnumIDList_Skip(pEnumIL, 1) != S_OK) ||
338                              FAILED(IEnumIDList_Reset(pEnumIL)))
339                         {
340                             IEnumIDList_Release(pEnumIL);
341                             pEnumIL = NULL;
342                         }
343                     }
344                     IShellFolder_Release(pSFChild);
345                 }
346             }
347
348             if (!(hPrev = InsertTreeViewItem(info, lpsf, pidlTemp, pidl, pEnumIL, hParent)))
349                 goto done;
350             SHFree(pidlTemp);  /* Finally, free the pidl that the shell gave us... */
351             pidlTemp=NULL;
352         }
353
354 done:
355         ReleaseCapture();
356         SetCursor(LoadCursorW(0, (LPWSTR)IDC_ARROW));
357
358         if (pidlTemp)
359           SHFree(pidlTemp);
360 }
361
362 static inline BOOL PIDLIsType(LPCITEMIDLIST pidl, PIDLTYPE type)
363 {
364     LPPIDLDATA data = _ILGetDataPointer(pidl);
365     if (!data)
366         return FALSE;
367     return (data->type == type);
368 }
369
370 static void BrsFolder_CheckValidSelection( browse_info *info, LPTV_ITEMDATA lptvid )
371 {
372     LPBROWSEINFOW lpBrowseInfo = info->lpBrowseInfo;
373     LPCITEMIDLIST pidl = lptvid->lpi;
374     BOOL bEnabled = TRUE;
375     DWORD dwAttributes;
376     HRESULT r;
377
378     if ((lpBrowseInfo->ulFlags & BIF_BROWSEFORCOMPUTER) &&
379         !PIDLIsType(pidl, PT_COMP))
380         bEnabled = FALSE;
381     if (lpBrowseInfo->ulFlags & BIF_RETURNFSANCESTORS)
382     {
383         dwAttributes = SFGAO_FILESYSANCESTOR | SFGAO_FILESYSTEM;
384         r = IShellFolder_GetAttributesOf(lptvid->lpsfParent, 1,
385                                 (LPCITEMIDLIST*)&lptvid->lpi, &dwAttributes);
386         if (FAILED(r) || !(dwAttributes & (SFGAO_FILESYSANCESTOR|SFGAO_FILESYSTEM)))
387             bEnabled = FALSE;
388     }
389     if (lpBrowseInfo->ulFlags & BIF_RETURNONLYFSDIRS)
390     {
391         dwAttributes = SFGAO_FOLDER | SFGAO_FILESYSTEM;
392         r = IShellFolder_GetAttributesOf(lptvid->lpsfParent, 1,
393                                 (LPCITEMIDLIST*)&lptvid->lpi, &dwAttributes);
394         if (FAILED(r) || 
395             ((dwAttributes & (SFGAO_FOLDER|SFGAO_FILESYSTEM)) != (SFGAO_FOLDER|SFGAO_FILESYSTEM)))
396         {
397             bEnabled = FALSE;
398         }
399     }
400     SendMessageW(info->hWnd, BFFM_ENABLEOK, 0, (LPARAM)bEnabled);
401 }
402
403 static LRESULT BrsFolder_Treeview_Delete( browse_info *info, NMTREEVIEWW *pnmtv )
404 {
405     LPTV_ITEMDATA lptvid = (LPTV_ITEMDATA)pnmtv->itemOld.lParam;
406
407     TRACE("TVN_DELETEITEMA/W %p\n", lptvid);
408
409     IShellFolder_Release(lptvid->lpsfParent);
410     if (lptvid->pEnumIL)
411         IEnumIDList_Release(lptvid->pEnumIL);
412     SHFree(lptvid->lpi);
413     SHFree(lptvid->lpifq);
414     SHFree(lptvid);
415     return 0;
416 }
417
418 static LRESULT BrsFolder_Treeview_Expand( browse_info *info, NMTREEVIEWW *pnmtv )
419 {
420     IShellFolder *lpsf2 = NULL;
421     LPTV_ITEMDATA lptvid = (LPTV_ITEMDATA) pnmtv->itemNew.lParam;
422     HRESULT r;
423
424     TRACE("TVN_ITEMEXPANDINGA/W\n");
425
426     if ((pnmtv->itemNew.state & TVIS_EXPANDEDONCE))
427         return 0;
428
429     if (lptvid->lpi && lptvid->lpi->mkid.cb) {
430         r = IShellFolder_BindToObject( lptvid->lpsfParent, lptvid->lpi, 0,
431                                       (REFIID)&IID_IShellFolder, (LPVOID *)&lpsf2 );
432     } else {
433         lpsf2 = lptvid->lpsfParent;
434         r = IShellFolder_AddRef(lpsf2);
435     }
436
437     if (SUCCEEDED(r))
438         FillTreeView( info, lpsf2, lptvid->lpifq, pnmtv->itemNew.hItem, lptvid->pEnumIL);
439
440     /* My Computer is already sorted and trying to do a simple text
441      * sort will only mess things up */
442     if (!_ILIsMyComputer(lptvid->lpi))
443         SendMessageW( info->hwndTreeView, TVM_SORTCHILDREN,
444                       FALSE, (LPARAM)pnmtv->itemNew.hItem );
445
446     return 0;
447 }
448
449 static HRESULT BrsFolder_Treeview_Changed( browse_info *info, NMTREEVIEWW *pnmtv )
450 {
451     LPTV_ITEMDATA lptvid = (LPTV_ITEMDATA) pnmtv->itemNew.lParam;
452
453     lptvid = (LPTV_ITEMDATA) pnmtv->itemNew.lParam;
454     info->pidlRet = lptvid->lpifq;
455     browsefolder_callback( info->lpBrowseInfo, info->hWnd, BFFM_SELCHANGED,
456                            (LPARAM)info->pidlRet );
457     BrsFolder_CheckValidSelection( info, lptvid );
458     return 0;
459 }
460
461 static LRESULT BrsFolder_OnNotify( browse_info *info, UINT CtlID, LPNMHDR lpnmh )
462 {
463     NMTREEVIEWW *pnmtv = (NMTREEVIEWW *)lpnmh;
464
465     TRACE("%p %x %p msg=%x\n", info, CtlID, lpnmh, pnmtv->hdr.code);
466
467     if (pnmtv->hdr.idFrom != IDD_TREEVIEW)
468         return 0;
469
470     switch (pnmtv->hdr.code)
471     {
472     case TVN_DELETEITEMA:
473     case TVN_DELETEITEMW:
474         return BrsFolder_Treeview_Delete( info, pnmtv );
475
476     case TVN_ITEMEXPANDINGA:
477     case TVN_ITEMEXPANDINGW:
478         return BrsFolder_Treeview_Expand( info, pnmtv );
479
480     case TVN_SELCHANGEDA:
481     case TVN_SELCHANGEDW:
482         return BrsFolder_Treeview_Changed( info, pnmtv );
483
484     default:
485         WARN("unhandled (%d)\n", pnmtv->hdr.code);
486         break;
487     }
488
489     return 0;
490 }
491
492
493 static BOOL BrsFolder_OnCreate( HWND hWnd, browse_info *info )
494 {
495     LPBROWSEINFOW lpBrowseInfo = info->lpBrowseInfo;
496
497     info->hWnd = hWnd;
498     SetPropW( hWnd, szBrowseFolderInfo, info );
499
500     if (lpBrowseInfo->ulFlags & ~SUPPORTEDFLAGS)
501         FIXME("flags %x not implemented\n", lpBrowseInfo->ulFlags & ~SUPPORTEDFLAGS);
502
503     if (lpBrowseInfo->lpszTitle)
504         SetWindowTextW( GetDlgItem(hWnd, IDD_TITLE), lpBrowseInfo->lpszTitle );
505     else
506         ShowWindow( GetDlgItem(hWnd, IDD_TITLE), SW_HIDE );
507
508     if (!(lpBrowseInfo->ulFlags & BIF_STATUSTEXT))
509         ShowWindow( GetDlgItem(hWnd, IDD_STATUS), SW_HIDE );
510
511     info->hwndTreeView = GetDlgItem( hWnd, IDD_TREEVIEW );
512     if (info->hwndTreeView)
513         InitializeTreeView( info );
514     else
515         ERR("treeview control missing!\n");
516
517     browsefolder_callback( info->lpBrowseInfo, hWnd, BFFM_INITIALIZED, 0 );
518
519     return TRUE;
520 }
521
522 static BOOL BrsFolder_OnCommand( browse_info *info, UINT id )
523 {
524     LPBROWSEINFOW lpBrowseInfo = info->lpBrowseInfo;
525
526     switch (id)
527     {
528     case IDOK:
529         info->pidlRet = ILClone(info->pidlRet); /* The original pidl will be free'd. */
530         pdump( info->pidlRet );
531         if (lpBrowseInfo->pszDisplayName)
532             SHGetPathFromIDListW( info->pidlRet, lpBrowseInfo->pszDisplayName );
533         EndDialog( info->hWnd, 1 );
534         return TRUE;
535
536     case IDCANCEL:
537         EndDialog( info->hWnd, 0 );
538         return TRUE;
539     }
540     return FALSE;
541 }
542
543 static BOOL BrsFolder_OnSetExpanded(browse_info *info, LPVOID selection, 
544     BOOL is_str, HTREEITEM *pItem)
545 {
546     LPITEMIDLIST pidlSelection = (LPITEMIDLIST)selection;
547     LPCITEMIDLIST pidlCurrent, pidlRoot;
548     TVITEMEXW item;
549     BOOL bResult = FALSE;
550     
551     /* If 'selection' is a string, convert to a Shell ID List. */ 
552     if (is_str) {
553         IShellFolder *psfDesktop;
554         HRESULT hr;
555
556         hr = SHGetDesktopFolder(&psfDesktop);
557         if (FAILED(hr))
558             goto done;
559
560         hr = IShellFolder_ParseDisplayName(psfDesktop, NULL, NULL, 
561                      (LPOLESTR)selection, NULL, &pidlSelection, NULL);
562         IShellFolder_Release(psfDesktop);
563         if (FAILED(hr)) 
564             goto done;
565     }
566
567     /* Move pidlCurrent behind the SHITEMIDs in pidlSelection, which are the root of
568      * the sub-tree currently displayed. */
569     pidlRoot = info->lpBrowseInfo->pidlRoot;
570     pidlCurrent = pidlSelection;
571     while (!_ILIsEmpty(pidlRoot) && _ILIsEqualSimple(pidlRoot, pidlCurrent)) {
572         pidlRoot = ILGetNext(pidlRoot);
573         pidlCurrent = ILGetNext(pidlCurrent);
574     }
575
576     /* The given ID List is not part of the SHBrowseForFolder's current sub-tree. */
577     if (!_ILIsEmpty(pidlRoot))
578         goto done;
579
580     /* Initialize item to point to the first child of the root folder. */
581     memset(&item, 0, sizeof(item));
582     item.mask = TVIF_PARAM;
583     item.hItem = TreeView_GetRoot(info->hwndTreeView);
584     if (item.hItem) 
585         item.hItem = TreeView_GetChild(info->hwndTreeView, item.hItem);
586
587     /* Walk the tree along the nodes corresponding to the remaining ITEMIDLIST */
588     while (item.hItem && !_ILIsEmpty(pidlCurrent)) {
589         LPTV_ITEMDATA pItemData;
590
591         TreeView_GetItemW(info->hwndTreeView, &item);
592         pItemData = (LPTV_ITEMDATA)item.lParam;
593
594         if (_ILIsEqualSimple(pItemData->lpi, pidlCurrent)) {
595             pidlCurrent = ILGetNext(pidlCurrent);
596             if (!_ILIsEmpty(pidlCurrent)) {
597                 /* Only expand current node and move on to it's first child,
598                  * if we didn't already reach the last SHITEMID */
599                 TreeView_Expand(info->hwndTreeView, item.hItem, TVE_EXPAND);
600                 item.hItem = TreeView_GetChild(info->hwndTreeView, item.hItem);
601             }
602         } else {
603             item.hItem = TreeView_GetNextSibling(info->hwndTreeView, item.hItem);
604         }
605     }
606
607     if (_ILIsEmpty(pidlCurrent) && item.hItem) 
608         bResult = TRUE;
609
610 done:
611     if (pidlSelection && pidlSelection != (LPITEMIDLIST)selection)
612         ILFree(pidlSelection);
613
614     if (pItem) 
615         *pItem = item.hItem;
616     
617     return bResult;
618 }
619
620 static BOOL BrsFolder_OnSetSelectionW(browse_info *info, LPVOID selection, BOOL is_str) {
621     HTREEITEM hItem;
622     BOOL bResult;
623
624     bResult = BrsFolder_OnSetExpanded(info, selection, is_str, &hItem);
625     if (bResult)
626         TreeView_Select(info->hwndTreeView, hItem, TVGN_CARET);
627     return bResult;
628 }
629
630 static BOOL BrsFolder_OnSetSelectionA(browse_info *info, LPVOID selection, BOOL is_str) {
631     LPWSTR selectionW = NULL;
632     BOOL result = FALSE;
633     int length;
634     
635     if (!is_str)
636         return BrsFolder_OnSetSelectionW(info, selection, is_str);
637     
638     if ((length = MultiByteToWideChar(CP_ACP, 0, (LPCSTR)selection, -1, NULL, 0)) &&
639         (selectionW = HeapAlloc(GetProcessHeap(), 0, length)) &&
640         MultiByteToWideChar(CP_ACP, 0, (LPCSTR)selection, -1, selectionW, length))
641     {
642         result = BrsFolder_OnSetSelectionW(info, selectionW, is_str);
643     }
644
645     HeapFree(GetProcessHeap(), 0, selectionW);
646     return result;
647 }
648
649 /*************************************************************************
650  *             BrsFolderDlgProc32  (not an exported API function)
651  */
652 static INT_PTR CALLBACK BrsFolderDlgProc( HWND hWnd, UINT msg, WPARAM wParam,
653                                           LPARAM lParam )
654 {
655     browse_info *info;
656
657     TRACE("hwnd=%p msg=%04x 0x%08x 0x%08lx\n", hWnd,  msg, wParam, lParam );
658
659     if (msg == WM_INITDIALOG)
660         return BrsFolder_OnCreate( hWnd, (browse_info*) lParam );
661
662     info = (browse_info*) GetPropW( hWnd, szBrowseFolderInfo );
663
664     switch (msg)
665     {
666     case WM_NOTIFY:
667         return BrsFolder_OnNotify( info, (UINT)wParam, (LPNMHDR)lParam);
668
669     case WM_COMMAND:
670         return BrsFolder_OnCommand( info, wParam );
671
672     case BFFM_SETSTATUSTEXTA:
673         TRACE("Set status %s\n", debugstr_a((LPSTR)lParam));
674         SetWindowTextA(GetDlgItem(hWnd, IDD_STATUS), (LPSTR)lParam);
675         break;
676
677     case BFFM_SETSTATUSTEXTW:
678         TRACE("Set status %s\n", debugstr_w((LPWSTR)lParam));
679         SetWindowTextW(GetDlgItem(hWnd, IDD_STATUS), (LPWSTR)lParam);
680         break;
681
682     case BFFM_ENABLEOK:
683         TRACE("Enable %ld\n", lParam);
684         EnableWindow(GetDlgItem(hWnd, 1), (lParam)?TRUE:FALSE);
685         break;
686
687     case BFFM_SETOKTEXT: /* unicode only */
688         TRACE("Set OK text %s\n", debugstr_w((LPWSTR)wParam));
689         SetWindowTextW(GetDlgItem(hWnd, 1), (LPWSTR)wParam);
690         break;
691
692     case BFFM_SETSELECTIONA:
693         return BrsFolder_OnSetSelectionA(info, (LPVOID)lParam, (BOOL)wParam);
694
695     case BFFM_SETSELECTIONW:
696         return BrsFolder_OnSetSelectionW(info, (LPVOID)lParam, (BOOL)wParam);
697
698     case BFFM_SETEXPANDED: /* unicode only */
699         return BrsFolder_OnSetExpanded(info, (LPVOID)lParam, (BOOL)wParam, NULL);
700     }
701     return FALSE;
702 }
703
704 static const WCHAR swBrowseTemplateName[] = {
705     'S','H','B','R','S','F','O','R','F','O','L','D','E','R','_','M','S','G','B','O','X',0};
706
707 /*************************************************************************
708  * SHBrowseForFolderA [SHELL32.@]
709  * SHBrowseForFolder  [SHELL32.@]
710  */
711 LPITEMIDLIST WINAPI SHBrowseForFolderA (LPBROWSEINFOA lpbi)
712 {
713     BROWSEINFOW bi;
714     LPITEMIDLIST lpid;
715     INT len;
716     
717     TRACE("%p\n", lpbi);
718
719     bi.hwndOwner = lpbi->hwndOwner;
720     bi.pidlRoot = lpbi->pidlRoot;
721     if (lpbi->pszDisplayName)
722     {
723         bi.pszDisplayName = HeapAlloc( GetProcessHeap(), 0, MAX_PATH * sizeof(WCHAR) );
724         MultiByteToWideChar( CP_ACP, 0, lpbi->pszDisplayName, -1, bi.pszDisplayName, MAX_PATH );
725     }
726     else
727         bi.pszDisplayName = NULL;
728
729     if (lpbi->lpszTitle)
730     {
731         len = MultiByteToWideChar( CP_ACP, 0, lpbi->lpszTitle, -1, NULL, 0 );
732         bi.lpszTitle = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
733         MultiByteToWideChar( CP_ACP, 0, lpbi->lpszTitle, -1, (LPWSTR)bi.lpszTitle, len );
734     }
735     else
736         bi.lpszTitle = NULL;
737
738     bi.ulFlags = lpbi->ulFlags;
739     bi.lpfn = lpbi->lpfn;
740     bi.lParam = lpbi->lParam;
741     bi.iImage = lpbi->iImage;
742     lpid = SHBrowseForFolderW( &bi );
743     if (bi.pszDisplayName)
744     {
745         WideCharToMultiByte( CP_ACP, 0, bi.pszDisplayName, -1,
746                              lpbi->pszDisplayName, MAX_PATH, 0, NULL);
747         HeapFree( GetProcessHeap(), 0, bi.pszDisplayName );
748     }
749     HeapFree(GetProcessHeap(), 0, (LPVOID)bi.lpszTitle);
750     lpbi->iImage = bi.iImage;
751     return lpid;
752 }
753
754
755 /*************************************************************************
756  * SHBrowseForFolderW [SHELL32.@]
757  *
758  * NOTES
759  *  crashes when passed a null pointer
760  */
761 LPITEMIDLIST WINAPI SHBrowseForFolderW (LPBROWSEINFOW lpbi)
762 {
763     browse_info info;
764     DWORD r;
765     HRESULT hr;
766
767     info.hWnd = 0;
768     info.pidlRet = NULL;
769     info.lpBrowseInfo = lpbi;
770     info.hwndTreeView = NULL;
771
772     hr = OleInitialize(NULL);
773     r = DialogBoxParamW( shell32_hInstance, swBrowseTemplateName, lpbi->hwndOwner,
774                          BrsFolderDlgProc, (LPARAM)&info );
775     if (SUCCEEDED(hr)) 
776         OleUninitialize();
777     if (!r)
778         return NULL;
779
780     return info.pidlRet;
781 }