7 #include "debugtools.h"
12 #include "wine/obj_base.h"
13 #include "wine/obj_enumidlist.h"
14 #include "wine/obj_shellfolder.h"
15 #include "wine/undocshell.h"
18 #include "shell32_main.h"
22 DEFAULT_DEBUG_CHANNEL(shell
);
24 #define IDD_TREEVIEW 99
26 static HWND hwndTreeView
;
27 static LPBROWSEINFOA lpBrowseInfo
;
28 static LPITEMIDLIST pidlRet
;
30 static void FillTreeView(LPSHELLFOLDER lpsf
, LPITEMIDLIST lpifq
, HTREEITEM hParent
);
32 static void InitializeTreeView(HWND hwndParent
)
34 HIMAGELIST hImageList
;
38 hwndTreeView
= GetDlgItem (hwndParent
, IDD_TREEVIEW
);
39 Shell_GetImageList(NULL
, &hImageList
);
41 TRACE("dlg=%x tree=%x\n", hwndParent
, hwndTreeView
);
43 if (hImageList
&& hwndTreeView
)
44 { TreeView_SetImageList(hwndTreeView
, hImageList
, 0);
47 hr
= SHGetDesktopFolder(&lpsf
);
49 if (SUCCEEDED(hr
) && hwndTreeView
)
50 { TreeView_DeleteAllItems(hwndTreeView
);
51 FillTreeView(lpsf
, NULL
, TVI_ROOT
);
55 { IShellFolder_Release(lpsf
);
59 static int GetIcon(LPITEMIDLIST lpi
, UINT uFlags
)
61 SHGetFileInfoA((LPCSTR
)lpi
,0,&sfi
, sizeof(SHFILEINFOA
), uFlags
);
65 static void GetNormalAndSelectedIcons(LPITEMIDLIST lpifq
,LPTVITEMA lpTV_ITEM
)
66 { TRACE("%p %p\n",lpifq
, lpTV_ITEM
);
68 lpTV_ITEM
->iImage
= GetIcon(lpifq
, SHGFI_PIDL
| SHGFI_SYSICONINDEX
| SHGFI_SMALLICON
);
69 lpTV_ITEM
->iSelectedImage
= GetIcon(lpifq
, SHGFI_PIDL
| SHGFI_SYSICONINDEX
| SHGFI_SMALLICON
| SHGFI_OPENICON
);
76 LPSHELLFOLDER lpsfParent
;
79 } TV_ITEMDATA
, *LPTV_ITEMDATA
;
81 static BOOL
GetName(LPSHELLFOLDER lpsf
, LPITEMIDLIST lpi
, DWORD dwFlags
, LPSTR lpFriendlyName
)
86 TRACE("%p %p %lx %p\n", lpsf
, lpi
, dwFlags
, lpFriendlyName
);
87 if (SUCCEEDED(IShellFolder_GetDisplayNameOf(lpsf
, lpi
, dwFlags
, &str
)))
89 if(FAILED(StrRetToStrNA (lpFriendlyName
, MAX_PATH
, &str
, lpi
)))
97 TRACE("-- %s\n",lpFriendlyName
);
101 static void FillTreeView(IShellFolder
* lpsf
, LPITEMIDLIST pidl
, HTREEITEM hParent
)
104 TVINSERTSTRUCTA tvins
;
107 LPITEMIDLIST pidlTemp
=0;
108 LPTV_ITEMDATA lptvid
=0;
112 HWND hwnd
=GetParent(hwndTreeView
);
114 TRACE("%p %p %x\n",lpsf
, pidl
, (INT
)hParent
);
115 SetCapture(GetParent(hwndTreeView
));
116 SetCursor(LoadCursorA(0, IDC_WAITA
));
118 hr
=IShellFolder_EnumObjects(lpsf
,hwnd
, SHCONTF_FOLDERS
| SHCONTF_NONFOLDERS
,&lpe
);
121 { while (NOERROR
== IEnumIDList_Next(lpe
,1,&pidlTemp
,&ulFetched
))
122 { ULONG ulAttrs
= SFGAO_HASSUBFOLDER
| SFGAO_FOLDER
;
123 IShellFolder_GetAttributesOf(lpsf
, 1, &pidlTemp
, &ulAttrs
);
124 if (ulAttrs
& (SFGAO_HASSUBFOLDER
| SFGAO_FOLDER
))
125 { if (ulAttrs
& SFGAO_FOLDER
)
126 { tvi
.mask
= TVIF_TEXT
| TVIF_IMAGE
| TVIF_SELECTEDIMAGE
| TVIF_PARAM
;
128 if (ulAttrs
& SFGAO_HASSUBFOLDER
)
130 tvi
.mask
|= TVIF_CHILDREN
;
133 if (!( lptvid
= (LPTV_ITEMDATA
)SHAlloc(sizeof(TV_ITEMDATA
))))
136 if (!GetName(lpsf
, pidlTemp
, SHGDN_NORMAL
, szBuff
))
139 tvi
.pszText
= szBuff
;
140 tvi
.cchTextMax
= MAX_PATH
;
141 tvi
.lParam
= (LPARAM
)lptvid
;
143 IShellFolder_AddRef(lpsf
);
144 lptvid
->lpsfParent
= lpsf
;
145 lptvid
->lpi
= ILClone(pidlTemp
);
146 lptvid
->lpifq
= ILCombine(pidl
, pidlTemp
);
147 GetNormalAndSelectedIcons(lptvid
->lpifq
, &tvi
);
149 tvins
.DUMMYUNIONNAME
.item
= tvi
;
150 tvins
.hInsertAfter
= hPrev
;
151 tvins
.hParent
= hParent
;
153 hPrev
= (HTREEITEM
)TreeView_InsertItemA (hwndTreeView
, &tvins
);
157 SHFree(pidlTemp
); /* Finally, free the pidl that the shell gave us... */
164 SetCursor(LoadCursorA(0, IDC_ARROWA
));
167 IEnumIDList_Release(lpe
);
172 static LRESULT
MsgNotify(HWND hWnd
, UINT CtlID
, LPNMHDR lpnmh
)
174 NMTREEVIEWA
*pnmtv
= (NMTREEVIEWA
*)lpnmh
;
175 LPTV_ITEMDATA lptvid
; /* Long pointer to TreeView item data */
176 IShellFolder
* lpsf2
=0;
179 TRACE("%x %x %p msg=%x\n", hWnd
, CtlID
, lpnmh
, pnmtv
->hdr
.code
);
181 switch (pnmtv
->hdr
.idFrom
)
183 switch (pnmtv
->hdr
.code
)
184 { case TVN_DELETEITEMA
:
185 { FIXME("TVN_DELETEITEMA\n");
186 lptvid
=(LPTV_ITEMDATA
)pnmtv
->itemOld
.lParam
;
187 IShellFolder_Release(lptvid
->lpsfParent
);
189 SHFree(lptvid
->lpifq
);
194 case TVN_ITEMEXPANDINGA
:
195 { FIXME("TVN_ITEMEXPANDINGA\n");
196 if ((pnmtv
->itemNew
.state
& TVIS_EXPANDEDONCE
))
199 lptvid
=(LPTV_ITEMDATA
)pnmtv
->itemNew
.lParam
;
200 if (SUCCEEDED(IShellFolder_BindToObject(lptvid
->lpsfParent
, lptvid
->lpi
,0,(REFIID
)&IID_IShellFolder
,(LPVOID
*)&lpsf2
)))
201 { FillTreeView( lpsf2
, lptvid
->lpifq
, pnmtv
->itemNew
.hItem
);
203 TreeView_SortChildren(hwndTreeView
, pnmtv
->itemNew
.hItem
, FALSE
);
206 case TVN_SELCHANGEDA
:
207 lptvid
=(LPTV_ITEMDATA
)pnmtv
->itemNew
.lParam
;
208 pidlRet
= lptvid
->lpifq
;
212 FIXME("unhandled\n");
225 /*************************************************************************
226 * BrsFolderDlgProc32 (not an exported API function)
228 static BOOL WINAPI
BrsFolderDlgProc( HWND hWnd
, UINT msg
, WPARAM wParam
,
230 { TRACE("hwnd=%i msg=%i 0x%08x 0x%08lx\n", hWnd
, msg
, wParam
, lParam
);
233 { case WM_INITDIALOG
:
235 lpBrowseInfo
= (LPBROWSEINFOA
) lParam
;
236 if (lpBrowseInfo
->lpfn
)
237 FIXME("Callbacks not implemented\n");
238 if (lpBrowseInfo
->ulFlags
)
239 FIXME("flag %x not implemented\n", lpBrowseInfo
->ulFlags
);
240 if (lpBrowseInfo
->lpszTitle
)
241 FIXME("title %s not displayed\n", lpBrowseInfo
->lpszTitle
);
242 if ( lpBrowseInfo
->pidlRoot
)
243 FIXME("root is desktop\n");
245 InitializeTreeView ( hWnd
);
249 MsgNotify( hWnd
, (UINT
)wParam
, (LPNMHDR
)lParam
);
256 SHGetPathFromIDListA(pidlRet
, lpBrowseInfo
->pszDisplayName
);
257 EndDialog(hWnd
, (DWORD
) ILClone(pidlRet
));
269 /*************************************************************************
270 * SHBrowseForFolderA [SHELL32.209]
273 LPITEMIDLIST WINAPI
SHBrowseForFolderA (LPBROWSEINFOA lpbi
)
275 TRACE("(%p{lpszTitle=%s})\n", lpbi
, debugstr_a(lpbi
->lpszTitle
));
277 return (LPITEMIDLIST
) DialogBoxParamA( shell32_hInstance
,
278 "SHBRSFORFOLDER_MSGBOX", lpbi
->hwndOwner
,
279 BrsFolderDlgProc
, (INT
)lpbi
);