src/Utils/MiscUI/BrowseFolder.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2012 - TortoiseSVN
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 #include "stdafx.h"
  20 #include <windowsx.h>
  21 #include "BrowseFolder.h"
  22 #include "SmartHandle.h"
  23 #include <strsafe.h>
  24
  25 BOOL CBrowseFolder::m_bCheck = FALSE;
  26 BOOL CBrowseFolder::m_bCheck2 = FALSE;
  27 WNDPROC CBrowseFolder::CBProc = NULL;
  28 HWND CBrowseFolder::checkbox = NULL;
  29 HWND CBrowseFolder::checkbox2 = NULL;
  30 HWND CBrowseFolder::ListView = NULL;
  31 TCHAR CBrowseFolder::m_CheckText[200];
  32 TCHAR CBrowseFolder::m_CheckText2[200];
  33 CString CBrowseFolder::m_sDefaultPath;
  34 bool CBrowseFolder::m_DisableCheckbox2WhenCheckbox1IsChecked = false;
  35
  36
  37 CBrowseFolder::CBrowseFolder(void)
  38 :       m_style(0),
  39         m_root(NULL)
  40 {
  41         memset(m_displayName, 0, sizeof(m_displayName));
  42         memset(m_title, 0, sizeof(m_title));
  43         memset(m_CheckText, 0, sizeof(m_CheckText));
  44 }
  45
  46 CBrowseFolder::~CBrowseFolder(void)
  47 {
  48 }
  49
  50 //show the dialog
  51 CBrowseFolder::retVal CBrowseFolder::Show(HWND parent, LPTSTR path, size_t pathlen, LPCTSTR szDefaultPath /* = NULL */)
  52 {
  53         CString temp;
  54         temp = path;
  55         CString sDefault;
  56         if (szDefaultPath)
  57                 sDefault = szDefaultPath;
  58         CBrowseFolder::retVal ret = Show(parent, temp, sDefault);
  59         _tcscpy_s(path, pathlen, temp);
  60         return ret;
  61 }
  62 CBrowseFolder::retVal CBrowseFolder::Show(HWND parent, CString& path, const CString& sDefaultPath /* = CString() */)
  63 {
  64         retVal ret = OK;                //assume OK
  65         m_sDefaultPath = sDefaultPath;
  66         if (m_sDefaultPath.IsEmpty() && !path.IsEmpty())
  67         {
  68                 // if the result path already contains a path, use that as the default path
  69                 m_sDefaultPath = path;
  70         }
  71
  72         HRESULT hr;
  73
  74         // Create a new common open file dialog
  75         IFileOpenDialog* pfd = NULL;
  76         hr = CoCreateInstance(CLSID_FileOpenDialog, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&pfd));
  77         if (SUCCEEDED(hr))
  78         {
  79                 // Set the dialog as a folder picker
  80                 DWORD dwOptions;
  81                 if (SUCCEEDED(hr = pfd->GetOptions(&dwOptions)))
  82                 {
  83                         hr = pfd->SetOptions(dwOptions | FOS_PICKFOLDERS | FOS_FORCEFILESYSTEM | FOS_PATHMUSTEXIST);
  84                 }
  85
  86                 // Set a title
  87                 if (SUCCEEDED(hr))
  88                 {
  89                         TCHAR * nl = _tcschr(m_title, '\n');
  90                         if (nl)
  91                                 *nl = 0;
  92                         pfd->SetTitle(m_title);
  93                 }
  94
  95                 // set the default folder
  96                 if (SUCCEEDED(hr))
  97                 {
  98                         typedef HRESULT (WINAPI *SHCIFPN)(PCWSTR pszPath, IBindCtx * pbc, REFIID riid, void ** ppv);
  99
 100                         CAutoLibrary hLib = AtlLoadSystemLibraryUsingFullPath(L"shell32.dll");
 101                         if (hLib)
 102                         {
 103                                 SHCIFPN pSHCIFPN = (SHCIFPN)GetProcAddress(hLib, "SHCreateItemFromParsingName");
 104                                 if (pSHCIFPN)
 105                                 {
 106                                         IShellItem *psiDefault = 0;
 107                                         hr = pSHCIFPN(m_sDefaultPath, NULL, IID_PPV_ARGS(&psiDefault));
 108                                         if (SUCCEEDED(hr))
 109                                         {
 110                                                 hr = pfd->SetFolder(psiDefault);
 111                                                 psiDefault->Release();
 112                                         }
 113                                 }
 114                         }
 115                 }
 116
 117                 if (m_CheckText[0] != 0)
 118                 {
 119                         IFileDialogCustomize* pfdCustomize = 0;
 120                         hr = pfd->QueryInterface(IID_PPV_ARGS(&pfdCustomize));
 121                         if (SUCCEEDED(hr))
 122                         {
 123                                 pfdCustomize->StartVisualGroup(100, L"");
 124                                 pfdCustomize->AddCheckButton(101, m_CheckText, FALSE);
 125                                 if (m_CheckText2[0] != 0)
 126                                 {
 127                                         pfdCustomize->AddCheckButton(102, m_CheckText2, FALSE);
 128                                 }
 129                                 pfdCustomize->EndVisualGroup();
 130                                 pfdCustomize->Release();
 131                         }
 132                 }
 133
 134                 // Show the open file dialog
 135                 if (SUCCEEDED(hr) && SUCCEEDED(hr = pfd->Show(parent)))
 136                 {
 137                         // Get the selection from the user
 138                         IShellItem* psiResult = NULL;
 139                         hr = pfd->GetResult(&psiResult);
 140                         if (SUCCEEDED(hr))
 141                         {
 142                                 PWSTR pszPath = NULL;
 143                                 hr = psiResult->GetDisplayName(SIGDN_FILESYSPATH, &pszPath);
 144                                 if (SUCCEEDED(hr))
 145                                 {
 146                                         path = pszPath;
 147                                         CoTaskMemFree(pszPath);
 148                                 }
 149                                 psiResult->Release();
 150
 151                                 IFileDialogCustomize* pfdCustomize = 0;
 152                                 hr = pfd->QueryInterface(IID_PPV_ARGS(&pfdCustomize));
 153                                 if (SUCCEEDED(hr))
 154                                 {
 155                                         pfdCustomize->GetCheckButtonState(101, &m_bCheck);
 156                                         pfdCustomize->GetCheckButtonState(102, &m_bCheck2);
 157                                         pfdCustomize->Release();
 158                                 }
 159                         }
 160                         else
 161                                 ret = CANCEL;
 162                 }
 163                 else
 164                         ret = CANCEL;
 165
 166                 pfd->Release();
 167         }
 168         else
 169         {
 170                 BROWSEINFO browseInfo           = {};
 171                 browseInfo.hwndOwner            = parent;
 172                 browseInfo.pidlRoot                     = m_root;
 173                 browseInfo.pszDisplayName       = m_displayName;
 174                 browseInfo.lpszTitle            = m_title;
 175                 browseInfo.ulFlags                      = m_style;
 176                 browseInfo.lParam                       = (LPARAM)this;
 177
 178                 if ((_tcslen(m_CheckText) > 0)||(!m_sDefaultPath.IsEmpty()))
 179                 {
 180                         browseInfo.lpfn = BrowseCallBackProc;
 181                 }
 182
 183                 PCIDLIST_ABSOLUTE itemIDList = SHBrowseForFolder(&browseInfo);
 184
 185                 //is the dialog canceled?
 186                 if (!itemIDList)
 187                         ret = CANCEL;
 188
 189                 if (ret != CANCEL)
 190                 {
 191                         if (!SHGetPathFromIDList(itemIDList, path.GetBuffer(MAX_PATH)))         // MAX_PATH ok. Explorer can't handle paths longer than MAX_PATH.
 192                                 ret = NOPATH;
 193
 194                         path.ReleaseBuffer();
 195
 196                         CoTaskMemFree((LPVOID)itemIDList);
 197                 }
 198         }
 199
 200         return ret;
 201 }
 202
 203 void CBrowseFolder::SetInfo(LPCTSTR title)
 204 {
 205         ASSERT(title);
 206
 207         if (title)
 208                 _tcscpy_s(m_title, 200, title);
 209 }
 210
 211 void CBrowseFolder::SetCheckBoxText(LPCTSTR checktext)
 212 {
 213         ASSERT(checktext);
 214
 215         if (checktext)
 216                 _tcscpy_s(m_CheckText, 200, checktext);
 217 }
 218
 219 void CBrowseFolder::SetCheckBoxText2(LPCTSTR checktext)
 220 {
 221         ASSERT(checktext);
 222
 223         if (checktext)
 224                 _tcscpy_s(m_CheckText2, 200, checktext);
 225 }
 226
 227 void CBrowseFolder::SetFont(HWND hwnd,LPTSTR FontName,int FontSize)
 228 {
 229
 230         HFONT hf;
 231         LOGFONT lf={0};
 232         HDC hdc=GetDC(hwnd);
 233
 234         GetObject(GetWindowFont(hwnd),sizeof(lf),&lf);
 235         lf.lfWeight = FW_REGULAR;
 236         lf.lfHeight = (LONG)FontSize;
 237         StringCchCopy( lf.lfFaceName, _countof(lf.lfFaceName), FontName );
 238         hf=CreateFontIndirect(&lf);
 239         SetBkMode(hdc,OPAQUE);
 240         SendMessage(hwnd,WM_SETFONT,(WPARAM)hf,TRUE);
 241         ReleaseDC(hwnd,hdc);
 242
 243 }
 244
 245 int CBrowseFolder::BrowseCallBackProc(HWND hwnd, UINT uMsg, LPARAM lParam, LPARAM /*lpData*/)
 246 {
 247         RECT ListViewRect,Dialog;
 248         //Initialization callback message
 249         if (uMsg == BFFM_INITIALIZED)
 250         {
 251                 if (_tcslen(m_CheckText) > 0)
 252                 {
 253                         bool bSecondCheckbox = (_tcslen(m_CheckText2)!=0);
 254                         //Rectangles for getting the positions
 255                         checkbox = CreateWindowEx(      0,
 256                                 _T("BUTTON"),
 257                                 m_CheckText,
 258                                 WS_CHILD|WS_VISIBLE|WS_CLIPCHILDREN|BS_AUTOCHECKBOX,
 259                                 0,100,100,50,
 260                                 hwnd,
 261                                 0,
 262                                 NULL,
 263                                 NULL);
 264                         if (checkbox == NULL)
 265                                 return 0;
 266
 267                         if (bSecondCheckbox)
 268                         {
 269                                 //Rectangles for getting the positions
 270                                 checkbox2 = CreateWindowEx(     0,
 271                                         _T("BUTTON"),
 272                                         m_CheckText2,
 273                                         WS_CHILD|WS_VISIBLE|WS_CLIPCHILDREN|BS_AUTOCHECKBOX,
 274                                         0,100,100,50,
 275                                         hwnd,
 276                                         0,
 277                                         NULL,
 278                                         NULL);
 279                                 if (checkbox2 == NULL)
 280                                         return 0;
 281                         }
 282
 283                         ListView = FindWindowEx(hwnd,NULL,_T("SysTreeView32"),NULL);
 284                         if (ListView == NULL)
 285                                 ListView = FindWindowEx(hwnd,NULL,_T("SHBrowseForFolder ShellNameSpace Control"),NULL);
 286
 287                         if (ListView == NULL)
 288                                 return 0;
 289
 290                         //Gets the dimensions of the windows
 291                         GetWindowRect(hwnd,&Dialog);
 292                         GetWindowRect(ListView,&ListViewRect);
 293                         POINT pt;
 294                         pt.x = ListViewRect.left;
 295                         pt.y = ListViewRect.top;
 296                         ScreenToClient(hwnd, &pt);
 297                         ListViewRect.top = pt.y;
 298                         ListViewRect.left = pt.x;
 299                         pt.x = ListViewRect.right;
 300                         pt.y = ListViewRect.bottom;
 301                         ScreenToClient(hwnd, &pt);
 302                         ListViewRect.bottom = pt.y;
 303                         ListViewRect.right = pt.x;
 304                         //Sets the list view controls dimensions
 305                         SetWindowPos(ListView,0,ListViewRect.left,
 306                                 bSecondCheckbox ? ListViewRect.top+40 : ListViewRect.top+20,
 307                                 (ListViewRect.right-ListViewRect.left),
 308                                 bSecondCheckbox ? (ListViewRect.bottom - ListViewRect.top)-40 : (ListViewRect.bottom - ListViewRect.top)-20,
 309                                 SWP_NOZORDER);
 310                         //Sets the window positions of checkbox and dialog controls
 311                         SetWindowPos(checkbox,HWND_BOTTOM,ListViewRect.left,
 312                                 ListViewRect.top,
 313                                 (ListViewRect.right-ListViewRect.left),
 314                                 14,
 315                                 SWP_NOZORDER);
 316                         if (bSecondCheckbox)
 317                         {
 318                                 SetWindowPos(checkbox2,HWND_BOTTOM,ListViewRect.left,
 319                                         ListViewRect.top+20,
 320                                         (ListViewRect.right-ListViewRect.left),
 321                                         14,
 322                                         SWP_NOZORDER);
 323                         }
 324                         HWND label = FindWindowEx(hwnd, NULL, _T("STATIC"), NULL);
 325                         if (label)
 326                         {
 327                                 HFONT hFont = (HFONT)::SendMessage(label, WM_GETFONT, 0, 0);
 328                                 LOGFONT lf = {0};
 329                                 GetObject(hFont, sizeof(lf), &lf);
 330                                 HFONT hf2 = CreateFontIndirect(&lf);
 331                                 ::SendMessage(checkbox, WM_SETFONT, (WPARAM)hf2, TRUE);
 332                                 if (bSecondCheckbox)
 333                                         ::SendMessage(checkbox2, WM_SETFONT, (WPARAM)hf2, TRUE);
 334                         }
 335                         else
 336                         {
 337                                 //Sets the fonts of static controls
 338                                 SetFont(checkbox,_T("MS Sans Serif"),12);
 339                                 if (bSecondCheckbox)
 340                                         SetFont(checkbox2,_T("MS Sans Serif"),12);
 341                         }
 342
 343                         // Subclass the checkbox control.
 344                         CBProc = (WNDPROC) SetWindowLongPtr(checkbox,GWLP_WNDPROC, (LONG_PTR) CheckBoxSubclassProc);
 345                         //Sets the checkbox to checked position
 346                         SendMessage(checkbox,BM_SETCHECK,(WPARAM)m_bCheck,0);
 347                         if (bSecondCheckbox)
 348                         {
 349                                 CBProc = (WNDPROC) SetWindowLongPtr(checkbox2,GWLP_WNDPROC, (LONG_PTR) CheckBoxSubclassProc2);
 350                                 SendMessage(checkbox2,BM_SETCHECK,(WPARAM)m_bCheck,0);
 351                         }
 352                         // send a resize message to the resized list view control. Otherwise it won't show
 353                         // up properly until the user resizes the window!
 354                         SendMessage(ListView, WM_SIZE, SIZE_RESTORED, MAKELONG(ListViewRect.right-ListViewRect.left, bSecondCheckbox ? (ListViewRect.bottom - ListViewRect.top)-40 : (ListViewRect.bottom - ListViewRect.top)-20));
 355                 }
 356
 357                 // now set the default directory
 358                 SendMessage(hwnd, BFFM_SETSELECTION, TRUE, (LPARAM)(LPCTSTR)m_sDefaultPath);
 359         }
 360         if (uMsg == BFFM_SELCHANGED)
 361         {
 362                 // Set the status window to the currently selected path.
 363                 TCHAR szDir[MAX_PATH] = {0};
 364                 if (SHGetPathFromIDList((LPITEMIDLIST)lParam, szDir))
 365                 {
 366                         SendMessage(hwnd,BFFM_SETSTATUSTEXT, 0, (LPARAM)szDir);
 367                 }
 368         }
 369
 370         return 0;
 371 }
 372
 373 LRESULT CBrowseFolder::CheckBoxSubclassProc(HWND hwnd,UINT uMsg,WPARAM wParam,LPARAM lParam)
 374 {
 375         if (uMsg == WM_LBUTTONUP)
 376         {
 377                 m_bCheck = (SendMessage(hwnd,BM_GETCHECK,0,0)==BST_UNCHECKED);
 378                 if (m_bCheck && m_DisableCheckbox2WhenCheckbox1IsChecked)
 379                 {
 380                         ::EnableWindow(checkbox2, !m_bCheck);
 381                 }
 382                 else
 383                         ::EnableWindow(checkbox2, true);
 384         }
 385
 386         return CallWindowProc(CBProc, hwnd, uMsg,
 387                 wParam, lParam);
 388 }
 389
 390 LRESULT CBrowseFolder::CheckBoxSubclassProc2(HWND hwnd,UINT uMsg,WPARAM wParam,LPARAM lParam)
 391 {
 392         if (uMsg == WM_LBUTTONUP)
 393         {
 394                 m_bCheck2 = (SendMessage(hwnd,BM_GETCHECK,0,0)==BST_UNCHECKED);
 395         }
 396
 397         return CallWindowProc(CBProc, hwnd, uMsg,
 398                 wParam, lParam);
 399 }