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