| 1 | /* $Id: brsfolder.c,v 1.1 2000-08-30 13:52:50 sandervl Exp $ */ | 
|---|
| 2 | #ifdef __WIN32OS2__ | 
|---|
| 3 | #define ICOM_CINTERFACE 1 | 
|---|
| 4 | #include <odin.h> | 
|---|
| 5 | #endif | 
|---|
| 6 | #include <stdlib.h> | 
|---|
| 7 | #include <string.h> | 
|---|
| 8 |  | 
|---|
| 9 | #include "debugtools.h" | 
|---|
| 10 | #include "wine/undocshell.h" | 
|---|
| 11 | #include "shlguid.h" | 
|---|
| 12 | #include "pidl.h" | 
|---|
| 13 | #include "shell32_main.h" | 
|---|
| 14 | #include "shellapi.h" | 
|---|
| 15 | #include "shresdef.h" | 
|---|
| 16 |  | 
|---|
| 17 | DEFAULT_DEBUG_CHANNEL(shell) | 
|---|
| 18 |  | 
|---|
| 19 | static HWND             hwndTreeView; | 
|---|
| 20 | static LPBROWSEINFOA    lpBrowseInfo; | 
|---|
| 21 | static LPITEMIDLIST     pidlRet; | 
|---|
| 22 |  | 
|---|
| 23 | static void FillTreeView(LPSHELLFOLDER lpsf, LPITEMIDLIST  lpifq, HTREEITEM hParent); | 
|---|
| 24 |  | 
|---|
| 25 | static void InitializeTreeView(HWND hwndParent, LPCITEMIDLIST root) | 
|---|
| 26 | { | 
|---|
| 27 | HIMAGELIST      hImageList; | 
|---|
| 28 | IShellFolder *  lpsf; | 
|---|
| 29 | HRESULT hr; | 
|---|
| 30 |  | 
|---|
| 31 | hwndTreeView = GetDlgItem (hwndParent, IDD_TREEVIEW); | 
|---|
| 32 | Shell_GetImageList(NULL, &hImageList); | 
|---|
| 33 |  | 
|---|
| 34 | TRACE("dlg=%x tree=%x\n", hwndParent, hwndTreeView ); | 
|---|
| 35 |  | 
|---|
| 36 | if (hImageList && hwndTreeView) | 
|---|
| 37 | { TreeView_SetImageList(hwndTreeView, hImageList, 0); | 
|---|
| 38 | } | 
|---|
| 39 |  | 
|---|
| 40 | /* so far, this method doesn't work (still missing the upper level), keep the old way */ | 
|---|
| 41 | #if 0 | 
|---|
| 42 | if (root == NULL) { | 
|---|
| 43 | hr = SHGetDesktopFolder(&lpsf); | 
|---|
| 44 | } else { | 
|---|
| 45 | IShellFolder *       lpsfdesktop; | 
|---|
| 46 |  | 
|---|
| 47 | hr = SHGetDesktopFolder(&lpsfdesktop); | 
|---|
| 48 | if (SUCCEEDED(hr)) { | 
|---|
| 49 | hr = IShellFolder_BindToObject(lpsfdesktop, root, 0,(REFIID)&IID_IShellFolder,(LPVOID *)&lpsf); | 
|---|
| 50 | IShellFolder_Release(lpsfdesktop); | 
|---|
| 51 | } | 
|---|
| 52 | } | 
|---|
| 53 | #else | 
|---|
| 54 | hr = SHGetDesktopFolder(&lpsf); | 
|---|
| 55 | #endif | 
|---|
| 56 |  | 
|---|
| 57 | if (SUCCEEDED(hr) && hwndTreeView) | 
|---|
| 58 | { TreeView_DeleteAllItems(hwndTreeView); | 
|---|
| 59 | FillTreeView(lpsf, NULL, TVI_ROOT); | 
|---|
| 60 | } | 
|---|
| 61 |  | 
|---|
| 62 | if (SUCCEEDED(hr)) | 
|---|
| 63 | { IShellFolder_Release(lpsf); | 
|---|
| 64 | } | 
|---|
| 65 | TRACE("done\n"); | 
|---|
| 66 | } | 
|---|
| 67 |  | 
|---|
| 68 | static int GetIcon(LPITEMIDLIST lpi, UINT uFlags) | 
|---|
| 69 | {       SHFILEINFOA    sfi; | 
|---|
| 70 | SHGetFileInfoA((LPCSTR)lpi,0,&sfi, sizeof(SHFILEINFOA), uFlags); | 
|---|
| 71 | return sfi.iIcon; | 
|---|
| 72 | } | 
|---|
| 73 |  | 
|---|
| 74 | static void GetNormalAndSelectedIcons(LPITEMIDLIST lpifq,LPTVITEMA lpTV_ITEM) | 
|---|
| 75 | {       TRACE("%p %p\n",lpifq, lpTV_ITEM); | 
|---|
| 76 |  | 
|---|
| 77 | lpTV_ITEM->iImage = GetIcon(lpifq, SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON); | 
|---|
| 78 | lpTV_ITEM->iSelectedImage = GetIcon(lpifq, SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_OPENICON); | 
|---|
| 79 |  | 
|---|
| 80 | return; | 
|---|
| 81 | } | 
|---|
| 82 |  | 
|---|
| 83 | typedef struct tagID | 
|---|
| 84 | { | 
|---|
| 85 | LPSHELLFOLDER lpsfParent; | 
|---|
| 86 | LPITEMIDLIST  lpi; | 
|---|
| 87 | LPITEMIDLIST  lpifq; | 
|---|
| 88 | } TV_ITEMDATA, *LPTV_ITEMDATA; | 
|---|
| 89 |  | 
|---|
| 90 | static BOOL GetName(LPSHELLFOLDER lpsf, LPITEMIDLIST lpi, DWORD dwFlags, LPSTR lpFriendlyName) | 
|---|
| 91 | { | 
|---|
| 92 | BOOL   bSuccess=TRUE; | 
|---|
| 93 | STRRET str; | 
|---|
| 94 |  | 
|---|
| 95 | TRACE("%p %p %lx %p\n", lpsf, lpi, dwFlags, lpFriendlyName); | 
|---|
| 96 | if (SUCCEEDED(IShellFolder_GetDisplayNameOf(lpsf, lpi, dwFlags, &str))) | 
|---|
| 97 | { | 
|---|
| 98 | if(FAILED(StrRetToStrNA (lpFriendlyName, MAX_PATH, &str, lpi))) | 
|---|
| 99 | { | 
|---|
| 100 | bSuccess = FALSE; | 
|---|
| 101 | } | 
|---|
| 102 | } | 
|---|
| 103 | else | 
|---|
| 104 | bSuccess = FALSE; | 
|---|
| 105 |  | 
|---|
| 106 | TRACE("-- %s\n",lpFriendlyName); | 
|---|
| 107 | return bSuccess; | 
|---|
| 108 | } | 
|---|
| 109 |  | 
|---|
| 110 | static void FillTreeView(IShellFolder * lpsf, LPITEMIDLIST  pidl, HTREEITEM hParent) | 
|---|
| 111 | { | 
|---|
| 112 | TVITEMA         tvi; | 
|---|
| 113 | TVINSERTSTRUCTA tvins; | 
|---|
| 114 | HTREEITEM       hPrev = 0; | 
|---|
| 115 | LPENUMIDLIST    lpe=0; | 
|---|
| 116 | LPITEMIDLIST    pidlTemp=0; | 
|---|
| 117 | LPTV_ITEMDATA   lptvid=0; | 
|---|
| 118 | ULONG           ulFetched; | 
|---|
| 119 | HRESULT         hr; | 
|---|
| 120 | char            szBuff[256]; | 
|---|
| 121 | HWND            hwnd=GetParent(hwndTreeView); | 
|---|
| 122 |  | 
|---|
| 123 | TRACE("%p %p %x\n",lpsf, pidl, (INT)hParent); | 
|---|
| 124 | SetCapture(GetParent(hwndTreeView)); | 
|---|
| 125 | SetCursor(LoadCursorA(0, IDC_WAITA)); | 
|---|
| 126 |  | 
|---|
| 127 | hr=IShellFolder_EnumObjects(lpsf,hwnd, SHCONTF_FOLDERS | SHCONTF_NONFOLDERS,&lpe); | 
|---|
| 128 |  | 
|---|
| 129 | if (SUCCEEDED(hr)) | 
|---|
| 130 | { while (NOERROR == IEnumIDList_Next(lpe,1,&pidlTemp,&ulFetched)) | 
|---|
| 131 | { ULONG ulAttrs = SFGAO_HASSUBFOLDER | SFGAO_FOLDER; | 
|---|
| 132 | IShellFolder_GetAttributesOf(lpsf, 1, &pidlTemp, &ulAttrs); | 
|---|
| 133 | if (ulAttrs & (SFGAO_HASSUBFOLDER | SFGAO_FOLDER)) | 
|---|
| 134 | { if (ulAttrs & SFGAO_FOLDER) | 
|---|
| 135 | { tvi.mask  = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM; | 
|---|
| 136 |  | 
|---|
| 137 | if (ulAttrs & SFGAO_HASSUBFOLDER) | 
|---|
| 138 | {  tvi.cChildren=1; | 
|---|
| 139 | tvi.mask |= TVIF_CHILDREN; | 
|---|
| 140 | } | 
|---|
| 141 |  | 
|---|
| 142 | if (!( lptvid = (LPTV_ITEMDATA)SHAlloc(sizeof(TV_ITEMDATA)))) | 
|---|
| 143 | goto Done; | 
|---|
| 144 |  | 
|---|
| 145 | if (!GetName(lpsf, pidlTemp, SHGDN_NORMAL, szBuff)) | 
|---|
| 146 | goto Done; | 
|---|
| 147 |  | 
|---|
| 148 | tvi.pszText    = szBuff; | 
|---|
| 149 | tvi.cchTextMax = MAX_PATH; | 
|---|
| 150 | tvi.lParam = (LPARAM)lptvid; | 
|---|
| 151 |  | 
|---|
| 152 | IShellFolder_AddRef(lpsf); | 
|---|
| 153 | lptvid->lpsfParent = lpsf; | 
|---|
| 154 | lptvid->lpi     = ILClone(pidlTemp); | 
|---|
| 155 | lptvid->lpifq   = ILCombine(pidl, pidlTemp); | 
|---|
| 156 | GetNormalAndSelectedIcons(lptvid->lpifq, &tvi); | 
|---|
| 157 |  | 
|---|
| 158 | tvins.DUMMYUNIONNAME.item         = tvi; | 
|---|
| 159 | tvins.hInsertAfter = hPrev; | 
|---|
| 160 | tvins.hParent      = hParent; | 
|---|
| 161 |  | 
|---|
| 162 | hPrev = (HTREEITEM)TreeView_InsertItemA (hwndTreeView, &tvins); | 
|---|
| 163 |  | 
|---|
| 164 | } | 
|---|
| 165 | } | 
|---|
| 166 | SHFree(pidlTemp);  /* Finally, free the pidl that the shell gave us... */ | 
|---|
| 167 | pidlTemp=0; | 
|---|
| 168 | } | 
|---|
| 169 | } | 
|---|
| 170 |  | 
|---|
| 171 | Done: | 
|---|
| 172 | ReleaseCapture(); | 
|---|
| 173 | SetCursor(LoadCursorA(0, IDC_ARROWA)); | 
|---|
| 174 |  | 
|---|
| 175 | if (lpe) | 
|---|
| 176 | IEnumIDList_Release(lpe); | 
|---|
| 177 | if (pidlTemp ) | 
|---|
| 178 | SHFree(pidlTemp); | 
|---|
| 179 | } | 
|---|
| 180 |  | 
|---|
| 181 | static LRESULT MsgNotify(HWND hWnd,  UINT CtlID, LPNMHDR lpnmh) | 
|---|
| 182 | { | 
|---|
| 183 | NMTREEVIEWA     *pnmtv   = (NMTREEVIEWA *)lpnmh; | 
|---|
| 184 | LPTV_ITEMDATA   lptvid;  /* Long pointer to TreeView item data */ | 
|---|
| 185 | IShellFolder *  lpsf2=0; | 
|---|
| 186 |  | 
|---|
| 187 |  | 
|---|
| 188 | TRACE("%x %x %p msg=%x\n", hWnd,  CtlID, lpnmh, pnmtv->hdr.code); | 
|---|
| 189 |  | 
|---|
| 190 | switch (pnmtv->hdr.idFrom) | 
|---|
| 191 | { case IDD_TREEVIEW: | 
|---|
| 192 | switch (pnmtv->hdr.code) | 
|---|
| 193 | { case TVN_DELETEITEMA: | 
|---|
| 194 | { FIXME("TVN_DELETEITEMA\n"); | 
|---|
| 195 | lptvid=(LPTV_ITEMDATA)pnmtv->itemOld.lParam; | 
|---|
| 196 | IShellFolder_Release(lptvid->lpsfParent); | 
|---|
| 197 | SHFree(lptvid->lpi); | 
|---|
| 198 | SHFree(lptvid->lpifq); | 
|---|
| 199 | SHFree(lptvid); | 
|---|
| 200 | } | 
|---|
| 201 | break; | 
|---|
| 202 |  | 
|---|
| 203 | case TVN_ITEMEXPANDINGA: | 
|---|
| 204 | { FIXME("TVN_ITEMEXPANDINGA\n"); | 
|---|
| 205 | if ((pnmtv->itemNew.state & TVIS_EXPANDEDONCE)) | 
|---|
| 206 | break; | 
|---|
| 207 |  | 
|---|
| 208 | lptvid=(LPTV_ITEMDATA)pnmtv->itemNew.lParam; | 
|---|
| 209 | if (SUCCEEDED(IShellFolder_BindToObject(lptvid->lpsfParent, lptvid->lpi,0,(REFIID)&IID_IShellFolder,(LPVOID *)&lpsf2))) | 
|---|
| 210 | { FillTreeView( lpsf2, lptvid->lpifq, pnmtv->itemNew.hItem ); | 
|---|
| 211 | } | 
|---|
| 212 | TreeView_SortChildren(hwndTreeView, pnmtv->itemNew.hItem, FALSE); | 
|---|
| 213 | } | 
|---|
| 214 | break; | 
|---|
| 215 | case TVN_SELCHANGEDA: | 
|---|
| 216 | lptvid=(LPTV_ITEMDATA)pnmtv->itemNew.lParam; | 
|---|
| 217 | pidlRet = lptvid->lpifq; | 
|---|
| 218 | if (lpBrowseInfo->lpfn) | 
|---|
| 219 | (lpBrowseInfo->lpfn)(hWnd, BFFM_SELCHANGED, (LPARAM)pidlRet, lpBrowseInfo->lParam); | 
|---|
| 220 | break; | 
|---|
| 221 |  | 
|---|
| 222 | default: | 
|---|
| 223 | FIXME("unhandled (%d)\n", pnmtv->hdr.code); | 
|---|
| 224 | break; | 
|---|
| 225 | } | 
|---|
| 226 | break; | 
|---|
| 227 |  | 
|---|
| 228 | default: | 
|---|
| 229 | break; | 
|---|
| 230 | } | 
|---|
| 231 |  | 
|---|
| 232 | return 0; | 
|---|
| 233 | } | 
|---|
| 234 |  | 
|---|
| 235 |  | 
|---|
| 236 | /************************************************************************* | 
|---|
| 237 | *             BrsFolderDlgProc32  (not an exported API function) | 
|---|
| 238 | */ | 
|---|
| 239 | static BOOL WINAPI BrsFolderDlgProc( HWND hWnd, UINT msg, WPARAM wParam, | 
|---|
| 240 | LPARAM lParam ) | 
|---|
| 241 | {       TRACE("hwnd=%i msg=%i 0x%08x 0x%08lx\n", hWnd,  msg, wParam, lParam ); | 
|---|
| 242 |  | 
|---|
| 243 | switch(msg) | 
|---|
| 244 | { case WM_INITDIALOG: | 
|---|
| 245 | pidlRet = NULL; | 
|---|
| 246 | lpBrowseInfo = (LPBROWSEINFOA) lParam; | 
|---|
| 247 | if (lpBrowseInfo->ulFlags & ~(BIF_STATUSTEXT)) | 
|---|
| 248 | FIXME("flags %x not implemented\n", lpBrowseInfo->ulFlags & ~(BIF_STATUSTEXT)); | 
|---|
| 249 | if (lpBrowseInfo->lpszTitle) { | 
|---|
| 250 | SetWindowTextA(GetDlgItem(hWnd, IDD_TITLE), lpBrowseInfo->lpszTitle); | 
|---|
| 251 | } else { | 
|---|
| 252 | ShowWindow(GetDlgItem(hWnd, IDD_TITLE), SW_HIDE); | 
|---|
| 253 | } | 
|---|
| 254 | if (!(lpBrowseInfo->ulFlags & BIF_STATUSTEXT)) | 
|---|
| 255 | ShowWindow(GetDlgItem(hWnd, IDD_STATUS), SW_HIDE); | 
|---|
| 256 |  | 
|---|
| 257 | if ( lpBrowseInfo->pidlRoot ) | 
|---|
| 258 | FIXME("root is desktop\n"); | 
|---|
| 259 |  | 
|---|
| 260 | InitializeTreeView( hWnd, lpBrowseInfo->pidlRoot ); | 
|---|
| 261 |  | 
|---|
| 262 | if (lpBrowseInfo->lpfn) { | 
|---|
| 263 | (lpBrowseInfo->lpfn)(hWnd, BFFM_INITIALIZED, 0, lpBrowseInfo->lParam); | 
|---|
| 264 | (lpBrowseInfo->lpfn)(hWnd, BFFM_SELCHANGED, 0/*FIXME*/, lpBrowseInfo->lParam); | 
|---|
| 265 | } | 
|---|
| 266 |  | 
|---|
| 267 | return TRUE; | 
|---|
| 268 |  | 
|---|
| 269 | case WM_NOTIFY: | 
|---|
| 270 | MsgNotify( hWnd, (UINT)wParam, (LPNMHDR)lParam); | 
|---|
| 271 | break; | 
|---|
| 272 |  | 
|---|
| 273 | case WM_COMMAND: | 
|---|
| 274 | switch (wParam) | 
|---|
| 275 | { case IDOK: | 
|---|
| 276 | pdump ( pidlRet ); | 
|---|
| 277 | SHGetPathFromIDListA(pidlRet, lpBrowseInfo->pszDisplayName); | 
|---|
| 278 | EndDialog(hWnd, (DWORD) ILClone(pidlRet)); | 
|---|
| 279 | return TRUE; | 
|---|
| 280 |  | 
|---|
| 281 | case IDCANCEL: | 
|---|
| 282 | EndDialog(hWnd, 0); | 
|---|
| 283 | return TRUE; | 
|---|
| 284 | break; | 
|---|
| 285 | } | 
|---|
| 286 | break; | 
|---|
| 287 | case BFFM_SETSTATUSTEXTA: | 
|---|
| 288 | TRACE("Set status %s\n", debugstr_a((LPSTR)lParam)); | 
|---|
| 289 | SetWindowTextA(GetDlgItem(hWnd, IDD_STATUS), (LPSTR)lParam); | 
|---|
| 290 | break; | 
|---|
| 291 | case BFFM_SETSTATUSTEXTW: | 
|---|
| 292 | TRACE("Set status %s\n", debugstr_w((LPWSTR)lParam)); | 
|---|
| 293 | SetWindowTextW(GetDlgItem(hWnd, IDD_STATUS), (LPWSTR)lParam); | 
|---|
| 294 | break; | 
|---|
| 295 | case BFFM_ENABLEOK: | 
|---|
| 296 | TRACE("Enable %ld\n", lParam); | 
|---|
| 297 | EnableWindow(GetDlgItem(hWnd, 1), (lParam)?TRUE:FALSE); | 
|---|
| 298 | break; | 
|---|
| 299 | case BFFM_SETSELECTIONA: | 
|---|
| 300 | if (wParam) | 
|---|
| 301 | TRACE("Set selection %s\n", debugstr_a((LPSTR)lParam)); | 
|---|
| 302 | else | 
|---|
| 303 | TRACE("Set selection %p\n", (void*)lParam); | 
|---|
| 304 | break; | 
|---|
| 305 | case BFFM_SETSELECTIONW: | 
|---|
| 306 | if (wParam) | 
|---|
| 307 | TRACE("Set selection %s\n", debugstr_w((LPWSTR)lParam)); | 
|---|
| 308 | else | 
|---|
| 309 | TRACE("Set selection %p\n", (void*)lParam); | 
|---|
| 310 | break; | 
|---|
| 311 | } | 
|---|
| 312 | return FALSE; | 
|---|
| 313 | } | 
|---|
| 314 |  | 
|---|
| 315 | /************************************************************************* | 
|---|
| 316 | * SHBrowseForFolderA [SHELL32.209] | 
|---|
| 317 | * | 
|---|
| 318 | */ | 
|---|
| 319 | LPITEMIDLIST WINAPI SHBrowseForFolderA (LPBROWSEINFOA lpbi) | 
|---|
| 320 | { | 
|---|
| 321 | TRACE("(%p{lpszTitle=%s,owner=%i})\n", | 
|---|
| 322 | lpbi, debugstr_a(lpbi->lpszTitle), lpbi->hwndOwner); | 
|---|
| 323 |  | 
|---|
| 324 | return (LPITEMIDLIST) DialogBoxParamA( shell32_hInstance, | 
|---|
| 325 | "SHBRSFORFOLDER_MSGBOX", | 
|---|
| 326 | lpbi->hwndOwner, | 
|---|
| 327 | BrsFolderDlgProc, (INT)lpbi ); | 
|---|
| 328 | } | 
|---|