src/TortoiseProc/RepositoryBrowser.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2009-2012 - TortoiseGit
   4 // Copyright (C) 2012 Sven Strickroth <email@cs-ware.de>
   5 // Copyright (C) 2003-2012 - TortoiseSVN
   6
   7 // This program is free software; you can redistribute it and/or
   8 // modify it under the terms of the GNU General Public License
   9 // as published by the Free Software Foundation; either version 2
  10 // of the License, or (at your option) any later version.
  11
  12 // This program is distributed in the hope that it will be useful,
  13 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 // GNU General Public License for more details.
  16
  17 // You should have received a copy of the GNU General Public License
  18 // along with this program; if not, write to the Free Software Foundation,
  19 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  20 //
  21 // RepositoryBrowser.cpp : implementation file
  22 //
  23
  24 #include "stdafx.h"
  25 #include "TortoiseProc.h"
  26 #include "RepositoryBrowser.h"
  27 #include "LogDlg.h"
  28 #include "AppUtils.h"
  29 #include "IconMenu.h"
  30 #include "UnicodeUtils.h"
  31 #include "SysImageList.h"
  32 #include <sys/stat.h>
  33 #include "SysInfo.h"
  34 #include "registry.h"
  35 #include "PathUtils.h"
  36 #include "StringUtils.h"
  37
  38 void SetSortArrowA(CListCtrl * control, int nColumn, bool bAscending)
  39 {
  40         if (control == NULL)
  41                 return;
  42
  43         // set the sort arrow
  44         CHeaderCtrl * pHeader = control->GetHeaderCtrl();
  45         HDITEM HeaderItem = {0};
  46         HeaderItem.mask = HDI_FORMAT;
  47         for (int i = 0; i < pHeader->GetItemCount(); ++i)
  48         {
  49                 pHeader->GetItem(i, &HeaderItem);
  50                 HeaderItem.fmt &= ~(HDF_SORTDOWN | HDF_SORTUP);
  51                 pHeader->SetItem(i, &HeaderItem);
  52         }
  53         if (nColumn >= 0)
  54         {
  55                 pHeader->GetItem(nColumn, &HeaderItem);
  56                 HeaderItem.fmt |= (bAscending ? HDF_SORTUP : HDF_SORTDOWN);
  57                 pHeader->SetItem(nColumn, &HeaderItem);
  58         }
  59 }
  60
  61 class CRepoListCompareFunc
  62 {
  63 public:
  64         CRepoListCompareFunc(CListCtrl* pList, int col, bool desc)
  65         : m_col(col)
  66         , m_desc(desc)
  67         , m_pList(pList)
  68         {}
  69
  70         static int CALLBACK StaticCompare(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)
  71         {
  72                 return ((CRepoListCompareFunc *) lParamSort)->Compare(lParam1, lParam2);
  73         }
  74
  75         int Compare(LPARAM lParam1, LPARAM lParam2)
  76         {
  77                 CShadowFilesTree * pLeft        = (CShadowFilesTree *)m_pList->GetItemData(lParam1);
  78                 CShadowFilesTree * pRight       = (CShadowFilesTree *)m_pList->GetItemData(lParam2);
  79
  80                 int result = 0;
  81                 switch(m_col)
  82                 {
  83                 case CRepositoryBrowser::eCol_Name:
  84                         result = SortStrCmp(pLeft->m_sName, pRight->m_sName);
  85                         if (result != 0)
  86                                 break;
  87                 case CRepositoryBrowser::eCol_Extension:
  88                         result = m_pList->GetItemText(static_cast<int>(lParam1), 1).CompareNoCase(m_pList->GetItemText(static_cast<int>(lParam2), 1));
  89                         if (result == 0)  // if extensions are the same, use the filename to sort
  90                                 result = SortStrCmp(pRight->m_sName, pRight->m_sName);
  91                         if (result != 0)
  92                                 break;
  93                 case CRepositoryBrowser::eCol_FileSize:
  94                         if (pLeft->m_iSize > pRight->m_iSize)
  95                                 result = 1;
  96                         else if (pLeft->m_iSize < pRight->m_iSize)
  97                                 result = -1;
  98                         else // fallback
  99                                 result = SortStrCmp(pLeft->m_sName, pRight->m_sName);
 100                 }
 101
 102                 if (m_desc)
 103                         result = -result;
 104
 105                 if (pLeft->m_bFolder != pRight->m_bFolder)
 106                 {
 107                         if (pRight->m_bFolder)
 108                                 result = 1;
 109                         else
 110                                 result = -1;
 111                 }
 112
 113                 return result;
 114         }
 115         int SortStrCmp(CString &left, CString &right)
 116         {
 117                 if (CRepositoryBrowser::s_bSortLogical)
 118                         return StrCmpLogicalW(left, right);
 119                 return StrCmpI(left, right);
 120         }
 121
 122         int m_col;
 123         bool m_desc;
 124         CListCtrl* m_pList;
 125 };
 126
 127 // CRepositoryBrowser dialog
 128
 129 bool CRepositoryBrowser::s_bSortLogical = true;
 130
 131 IMPLEMENT_DYNAMIC(CRepositoryBrowser, CResizableStandAloneDialog)
 132
 133 CRepositoryBrowser::CRepositoryBrowser(CString rev, CWnd* pParent /*=NULL*/)
 134 : CResizableStandAloneDialog(CRepositoryBrowser::IDD, pParent)
 135 , m_currSortCol(0)
 136 , m_currSortDesc(false)
 137 , m_sRevision(rev)
 138 {
 139 }
 140
 141 CRepositoryBrowser::~CRepositoryBrowser()
 142 {
 143 }
 144
 145 void CRepositoryBrowser::DoDataExchange(CDataExchange* pDX)
 146 {
 147         CDialog::DoDataExchange(pDX);
 148         DDX_Control(pDX, IDC_REPOTREE, m_RepoTree);
 149         DDX_Control(pDX, IDC_REPOLIST, m_RepoList);
 150 }
 151
 152
 153 BEGIN_MESSAGE_MAP(CRepositoryBrowser, CResizableStandAloneDialog)
 154         ON_NOTIFY(TVN_SELCHANGED, IDC_REPOTREE, &CRepositoryBrowser::OnTvnSelchangedRepoTree)
 155         ON_WM_CONTEXTMENU()
 156         ON_NOTIFY(LVN_COLUMNCLICK, IDC_REPOLIST, &CRepositoryBrowser::OnLvnColumnclickRepoList)
 157         ON_NOTIFY(NM_DBLCLK, IDC_REPOLIST, &CRepositoryBrowser::OnNMDblclk_RepoList)
 158         ON_BN_CLICKED(IDC_BUTTON_REVISION, &CRepositoryBrowser::OnBnClickedButtonRevision)
 159         ON_WM_SETCURSOR()
 160         ON_WM_MOUSEMOVE()
 161         ON_WM_LBUTTONDOWN()
 162         ON_WM_LBUTTONUP()
 163 END_MESSAGE_MAP()
 164
 165
 166 // CRepositoryBrowser message handlers
 167
 168 BOOL CRepositoryBrowser::OnInitDialog()
 169 {
 170         CResizableStandAloneDialog::OnInitDialog();
 171         CAppUtils::MarkWindowAsUnpinnable(m_hWnd);
 172
 173         AddAnchor(IDC_STATIC, TOP_LEFT);
 174         AddAnchor(IDC_BUTTON_REVISION, TOP_LEFT, TOP_RIGHT);
 175         AddAnchor(IDC_REPOTREE, TOP_LEFT, BOTTOM_LEFT);
 176         AddAnchor(IDC_REPOLIST, TOP_LEFT, BOTTOM_RIGHT);
 177         AddAnchor(IDHELP, BOTTOM_RIGHT);
 178         AddAnchor(IDOK, BOTTOM_RIGHT);
 179         AddAnchor(IDCANCEL, BOTTOM_RIGHT);
 180
 181         CRepositoryBrowser::s_bSortLogical = !CRegDWORD(L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer\\NoStrCmpLogical", 0, false, HKEY_CURRENT_USER);
 182         if (CRepositoryBrowser::s_bSortLogical)
 183                 CRepositoryBrowser::s_bSortLogical = !CRegDWORD(L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer\\NoStrCmpLogical", 0, false, HKEY_LOCAL_MACHINE);
 184
 185         CString temp;
 186         temp.LoadString(IDS_STATUSLIST_COLFILENAME);
 187         m_RepoList.InsertColumn(eCol_Name, temp, 0, 150);
 188         temp.LoadString(IDS_STATUSLIST_COLEXT);
 189         m_RepoList.InsertColumn(eCol_Extension, temp, 0, 100);
 190         temp.LoadString(IDS_LOG_SIZE);
 191         m_RepoList.InsertColumn(eCol_FileSize, temp, 0, 100);
 192
 193         // set up the list control
 194         // set the extended style of the list control
 195         // the style LVS_EX_FULLROWSELECT interferes with the background watermark image but it's more important to be able to select in the whole row.
 196         CRegDWORD regFullRowSelect(_T("Software\\TortoiseGit\\FullRowSelect"), TRUE);
 197         DWORD exStyle = LVS_EX_HEADERDRAGDROP | LVS_EX_DOUBLEBUFFER | LVS_EX_INFOTIP | LVS_EX_SUBITEMIMAGES;
 198         if (DWORD(regFullRowSelect))
 199                 exStyle |= LVS_EX_FULLROWSELECT;
 200         m_RepoList.SetExtendedStyle(exStyle);
 201         m_RepoList.SetImageList(&SYS_IMAGE_LIST(), LVSIL_SMALL);
 202
 203         m_RepoTree.SetImageList(&SYS_IMAGE_LIST(), TVSIL_NORMAL);
 204         if (SysInfo::Instance().IsVistaOrLater())
 205         {
 206                 DWORD exStyle = TVS_EX_FADEINOUTEXPANDOS | TVS_EX_AUTOHSCROLL | TVS_EX_DOUBLEBUFFER;
 207                 m_RepoTree.SetExtendedStyle(exStyle, exStyle);
 208         }
 209
 210         SetWindowTheme(m_RepoTree.GetSafeHwnd(), L"Explorer", NULL);
 211         SetWindowTheme(m_RepoList.GetSafeHwnd(), L"Explorer", NULL);
 212
 213         m_nIconFolder = SYS_IMAGE_LIST().GetDirIconIndex();
 214         m_nOpenIconFolder = SYS_IMAGE_LIST().GetDirOpenIconIndex();
 215
 216         EnableSaveRestore(L"Reposbrowser");
 217
 218         DWORD xPos = CRegDWORD(_T("Software\\TortoiseGit\\TortoiseProc\\ResizableState\\RepobrowserDivider"), 0);
 219         if (xPos == 0)
 220         {
 221                 RECT rc;
 222                 GetDlgItem(IDC_REPOTREE)->GetClientRect(&rc);
 223                 xPos = rc.right - rc.left;
 224         }
 225         bDragMode = true;
 226         HandleDividerMove(CPoint(xPos + 20, 10), false);
 227
 228         CString sWindowTitle;
 229         GetWindowText(sWindowTitle);
 230         CAppUtils::SetWindowTitle(m_hWnd, g_Git.m_CurrentDir, sWindowTitle);
 231
 232         Refresh();
 233
 234         m_RepoList.SetFocus();
 235
 236         return FALSE;
 237 }
 238
 239 void CRepositoryBrowser::OnOK()
 240 {
 241         SaveDividerPosition();
 242         CResizableStandAloneDialog::OnOK();
 243 }
 244
 245 void CRepositoryBrowser::OnCancel()
 246 {
 247         SaveDividerPosition();
 248         CResizableStandAloneDialog::OnCancel();
 249 }
 250
 251 void CRepositoryBrowser::OnNMDblclk_RepoList(NMHDR *pNMHDR, LRESULT *pResult)
 252 {
 253         UNREFERENCED_PARAMETER(pNMHDR);
 254         *pResult = 0;
 255
 256         LPNMITEMACTIVATE pNmItemActivate = reinterpret_cast<LPNMITEMACTIVATE>(pNMHDR);
 257         if (pNmItemActivate->iItem < 0)
 258                 return;
 259
 260         CShadowFilesTree * pItem = (CShadowFilesTree *)m_RepoList.GetItemData(pNmItemActivate->iItem);
 261         if (pItem == NULL || !pItem->m_bFolder)
 262                 return;
 263
 264         FillListCtrlForShadowTree(pItem);
 265         m_RepoTree.SelectItem(pItem->m_hTree);
 266 }
 267
 268 void CRepositoryBrowser::Refresh()
 269 {
 270         m_RepoTree.DeleteAllItems();
 271         m_RepoList.DeleteAllItems();
 272         m_TreeRoot.m_ShadowTree.clear();
 273         m_TreeRoot.m_sName = "";
 274         m_TreeRoot.m_bFolder = true;
 275
 276         TVINSERTSTRUCT tvinsert = {0};
 277         tvinsert.hParent = TVI_ROOT;
 278         tvinsert.hInsertAfter = TVI_ROOT;
 279         tvinsert.itemex.mask = TVIF_DI_SETITEM | TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_STATE;
 280         tvinsert.itemex.pszText = L"/";
 281         tvinsert.itemex.lParam = (LPARAM)&m_TreeRoot;
 282         tvinsert.itemex.iImage = m_nIconFolder;
 283         tvinsert.itemex.iSelectedImage = m_nOpenIconFolder;
 284         m_TreeRoot.m_hTree= m_RepoTree.InsertItem(&tvinsert);
 285
 286         ReadTree(&m_TreeRoot);
 287         m_RepoTree.Expand(m_TreeRoot.m_hTree, TVE_EXPAND);
 288         FillListCtrlForShadowTree(&m_TreeRoot);
 289         m_RepoTree.SelectItem(m_TreeRoot.m_hTree);
 290 }
 291
 292 int CRepositoryBrowser::ReadTreeRecursive(git_repository &repo, git_tree * tree, CShadowFilesTree * treeroot)
 293 {
 294         size_t count = git_tree_entrycount(tree);
 295
 296         for (int i = 0; i < count; i++)
 297         {
 298                 const git_tree_entry *entry = git_tree_entry_byindex(tree, i);
 299                 if (entry == NULL)
 300                         continue;
 301                 int mode = git_tree_entry_attributes(entry);
 302
 303                 CString base = CUnicodeUtils::GetUnicode(git_tree_entry_name(entry), CP_UTF8);
 304
 305                 git_object *object = NULL;
 306                 git_tree_entry_2object(&object, &repo, entry);
 307                 if (object == NULL)
 308                         continue;
 309
 310                 CShadowFilesTree * pNextTree = &treeroot->m_ShadowTree[base];
 311                 pNextTree->m_sName = base;
 312                 pNextTree->m_pParent = treeroot;
 313
 314                 if (mode & S_IFDIR)
 315                 {
 316                         pNextTree->m_bFolder = true;
 317
 318                         TVINSERTSTRUCT tvinsert = {0};
 319                         tvinsert.hParent = treeroot->m_hTree;
 320                         tvinsert.hInsertAfter = TVI_SORT;
 321                         tvinsert.itemex.mask = TVIF_DI_SETITEM | TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_STATE;
 322                         tvinsert.itemex.pszText = base.GetBuffer(base.GetLength());
 323                         tvinsert.itemex.lParam = (LPARAM)pNextTree;
 324                         tvinsert.itemex.iImage = m_nIconFolder;
 325                         tvinsert.itemex.iSelectedImage = m_nOpenIconFolder;
 326                         pNextTree->m_hTree = m_RepoTree.InsertItem(&tvinsert);
 327                         base.ReleaseBuffer();
 328
 329                         ReadTreeRecursive(repo, (git_tree*)object, pNextTree);
 330                 }
 331                 else
 332                 {
 333                         const git_oid * oid = git_object_id(object);
 334
 335                         git_blob * blob;
 336                         git_blob_lookup(&blob, &repo, oid);
 337                         if (blob == NULL)
 338                                 continue;
 339
 340                         pNextTree->m_iSize = git_blob_rawsize(blob);
 341                         git_blob_free(blob);
 342                 }
 343
 344                 git_object_free(object);
 345         }
 346
 347         return 0;
 348 }
 349
 350 int CRepositoryBrowser::ReadTree(CShadowFilesTree * treeroot)
 351 {
 352         CStringA gitdir = CUnicodeUtils::GetMulti(g_Git.m_CurrentDir, CP_UTF8);
 353         git_repository *repository = NULL;
 354         git_commit *commit = NULL;
 355         git_tree * tree = NULL;
 356         int ret = 0;
 357         do
 358         {
 359                 ret = git_repository_open(&repository, gitdir.GetBuffer());
 360                 if(ret)
 361                         break;
 362
 363                 CGitHash hash = g_Git.GetHash(m_sRevision);
 364                 ret = git_commit_lookup(&commit, repository, (git_oid *) hash.m_hash);
 365                 if(ret)
 366                         break;
 367
 368                 ret = git_commit_tree(&tree, commit);
 369                 if(ret)
 370                         break;
 371
 372                 ret = ReadTreeRecursive(*repository, tree, treeroot);
 373                 if(ret)
 374                         break;
 375
 376                 this->GetDlgItem(IDC_BUTTON_REVISION)->SetWindowText(m_sRevision);
 377         } while(0);
 378
 379         if (tree)
 380                 git_tree_free(tree);
 381
 382         if (commit)
 383                 git_commit_free(commit);
 384
 385         if (repository)
 386                 git_repository_free(repository);
 387
 388         return ret;
 389 }
 390
 391 void CRepositoryBrowser::OnTvnSelchangedRepoTree(NMHDR *pNMHDR, LRESULT *pResult)
 392 {
 393         LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);
 394         *pResult = 0;
 395
 396         FillListCtrlForTreeNode(pNMTreeView->itemNew.hItem);
 397 }
 398
 399 void CRepositoryBrowser::FillListCtrlForTreeNode(HTREEITEM treeNode)
 400 {
 401         m_RepoList.DeleteAllItems();
 402
 403         CShadowFilesTree* pTree = (CShadowFilesTree*)(m_RepoTree.GetItemData(treeNode));
 404         if (pTree == NULL)
 405         {
 406                 ASSERT(FALSE);
 407                 return;
 408         }
 409         FillListCtrlForShadowTree(pTree);
 410 }
 411
 412 void CRepositoryBrowser::FillListCtrlForShadowTree(CShadowFilesTree* pTree)
 413 {
 414         for (TShadowFilesTreeMap::iterator itShadowTree = pTree->m_ShadowTree.begin(); itShadowTree != pTree->m_ShadowTree.end(); ++itShadowTree)
 415         {
 416                 int icon = m_nIconFolder;
 417                 if (!(*itShadowTree).second.m_bFolder)
 418                         icon = SYS_IMAGE_LIST().GetFileIconIndex((*itShadowTree).second.m_sName);
 419
 420                 int indexItem = m_RepoList.InsertItem(m_RepoList.GetItemCount(), (*itShadowTree).second.m_sName, icon);
 421
 422                 m_RepoList.SetItemData(indexItem, (DWORD_PTR)&(*itShadowTree).second);
 423                 if (!(*itShadowTree).second.m_bFolder)
 424                 {
 425                         CString temp;
 426
 427                         temp = CPathUtils::GetFileExtFromPath((*itShadowTree).second.m_sName);
 428                         m_RepoList.SetItemText(indexItem, eCol_Extension, temp);
 429
 430                         StrFormatByteSize((*itShadowTree).second.m_iSize, temp.GetBuffer(20), 20);
 431                         temp.ReleaseBuffer();
 432                         m_RepoList.SetItemText(indexItem, eCol_FileSize, temp);
 433                 }
 434         }
 435
 436         CRepoListCompareFunc compareFunc(&m_RepoList, m_currSortCol, m_currSortDesc);
 437         m_RepoList.SortItemsEx(&CRepoListCompareFunc::StaticCompare, (DWORD_PTR)&compareFunc);
 438
 439         SetSortArrowA(&m_RepoList, m_currSortCol, !m_currSortDesc);
 440 }
 441
 442 void CRepositoryBrowser::OnContextMenu(CWnd* pWndFrom, CPoint point)
 443 {
 444         if (pWndFrom == &m_RepoList)
 445                 OnContextMenu_RepoList(point);
 446 }
 447
 448 void CRepositoryBrowser::OnContextMenu_RepoList(CPoint point)
 449 {
 450         TShadowFilesTreeList selectedLeafs;
 451         selectedLeafs.reserve(m_RepoList.GetSelectedCount());
 452         POSITION pos = m_RepoList.GetFirstSelectedItemPosition();
 453         while (pos)
 454         {
 455                 selectedLeafs.push_back((CShadowFilesTree *)m_RepoList.GetItemData(m_RepoList.GetNextSelectedItem(pos)));
 456         }
 457
 458         CIconMenu popupMenu;
 459         popupMenu.CreatePopupMenu();
 460
 461         bool bAddSeparator = false;
 462
 463         if (selectedLeafs.size() == 1)
 464         {
 465                 CString temp;
 466                 temp.LoadString(IDS_MENULOG);
 467                 popupMenu.AppendMenuIcon(eCmd_ViewLog, temp, IDI_LOG);
 468
 469                 if (!selectedLeafs.at(0)->m_bFolder)
 470                 {
 471                         popupMenu.AppendMenu(MF_SEPARATOR);
 472                         temp.LoadString(IDS_LOG_POPUP_SAVE);
 473                         popupMenu.AppendMenuIcon(eCmd_SaveAs, temp, IDI_SAVEAS);
 474                 }
 475
 476                 bAddSeparator = true;
 477         }
 478
 479         if (bAddSeparator)
 480                 popupMenu.AppendMenu(MF_SEPARATOR);
 481         bAddSeparator = false;
 482
 483         popupMenu.AppendMenuIcon(eCmd_CopyPath, IDS_STATUSLIST_CONTEXT_COPY, IDI_COPYCLIP);
 484
 485         eCmd cmd = (eCmd)popupMenu.TrackPopupMenuEx(TPM_LEFTALIGN|TPM_RETURNCMD, point.x, point.y, this, 0);
 486         switch(cmd)
 487         {
 488         case eCmd_ViewLog:
 489                 {
 490                         CString sCmd;
 491                         sCmd.Format(_T("/command:log /path:\"%s%s\""), g_Git.m_CurrentDir, selectedLeafs.at(0)->GetFullName());
 492                         CAppUtils::RunTortoiseProc(sCmd);
 493                 }
 494                 break;
 495         case eCmd_SaveAs:
 496                 FileSaveAs(selectedLeafs.at(0)->GetFullName());
 497                 break;
 498         case eCmd_CopyPath:
 499                 {
 500                         CString sClipboard;
 501                         for (TShadowFilesTreeList::iterator itShadowTree = selectedLeafs.begin(); itShadowTree != selectedLeafs.end(); ++itShadowTree)
 502                         {
 503                                 sClipboard += (*itShadowTree)->m_sName + _T("\r\n");
 504                         }
 505                         CStringUtils::WriteAsciiStringToClipboard(sClipboard);
 506                 }
 507                 break;
 508         }
 509 }
 510
 511 BOOL CRepositoryBrowser::PreTranslateMessage(MSG* pMsg)
 512 {
 513         if (pMsg->message == WM_KEYDOWN)
 514         {
 515                 switch (pMsg->wParam)
 516                 {
 517                 case VK_F5:
 518                         {
 519                                 Refresh();
 520                         }
 521                         break;
 522                 }
 523         }
 524
 525         return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
 526 }
 527
 528 void CRepositoryBrowser::OnLvnColumnclickRepoList(NMHDR *pNMHDR, LRESULT *pResult)
 529 {
 530         LPNMLISTVIEW pNMLV = reinterpret_cast<LPNMLISTVIEW>(pNMHDR);
 531         if (pResult)
 532                 *pResult = 0;
 533
 534         if (m_currSortCol == pNMLV->iSubItem)
 535                 m_currSortDesc = !m_currSortDesc;
 536         else
 537         {
 538                 m_currSortCol  = pNMLV->iSubItem;
 539                 m_currSortDesc = false;
 540         }
 541
 542         CRepoListCompareFunc compareFunc(&m_RepoList, m_currSortCol, m_currSortDesc);
 543         m_RepoList.SortItemsEx(&CRepoListCompareFunc::StaticCompare, (DWORD_PTR)&compareFunc);
 544
 545         SetSortArrowA(&m_RepoList, m_currSortCol, !m_currSortDesc);
 546 }
 547
 548 void CRepositoryBrowser::OnBnClickedButtonRevision()
 549 {
 550                 // use the git log to allow selection of a version
 551                 CLogDlg dlg;
 552                 // tell the dialog to use mode for selecting revisions
 553                 dlg.SetSelect(true);
 554                 // only one revision must be selected however
 555                 dlg.SingleSelection(true);
 556                 if (dlg.DoModal() == IDOK)
 557                 {
 558                         // get selected hash if any
 559                         m_sRevision = dlg.GetSelectedHash();
 560                         Refresh();
 561                 }
 562 }
 563
 564 void CRepositoryBrowser::SaveDividerPosition()
 565 {
 566         RECT rc;
 567         GetDlgItem(IDC_REPOTREE)->GetClientRect(&rc);
 568         CRegDWORD xPos = CRegDWORD(_T("Software\\TortoiseGit\\TortoiseProc\\ResizableState\\RepobrowserDivider"));
 569         xPos = rc.right - rc.left;
 570 }
 571
 572 void CRepositoryBrowser::HandleDividerMove(CPoint point, bool bDraw)
 573 {
 574         RECT rect, tree, list, treelist, treelistclient;
 575
 576         // create an union of the tree and list control rectangle
 577         GetDlgItem(IDC_REPOLIST)->GetWindowRect(&list);
 578         GetDlgItem(IDC_REPOTREE)->GetWindowRect(&tree);
 579         UnionRect(&treelist, &tree, &list);
 580         treelistclient = treelist;
 581         ScreenToClient(&treelistclient);
 582
 583         ClientToScreen(&point);
 584         GetClientRect(&rect);
 585         ClientToScreen(&rect);
 586
 587         CPoint point2 = point;
 588         if (point2.x < treelist.left + REPOBROWSER_CTRL_MIN_WIDTH)
 589                 point2.x = treelist.left + REPOBROWSER_CTRL_MIN_WIDTH;
 590         if (point2.x > treelist.right - REPOBROWSER_CTRL_MIN_WIDTH)
 591                 point2.x = treelist.right - REPOBROWSER_CTRL_MIN_WIDTH;
 592
 593         point.x -= rect.left;
 594         point.y -= treelist.top;
 595
 596         OffsetRect(&treelist, -treelist.left, -treelist.top);
 597
 598         if (point.x < treelist.left+REPOBROWSER_CTRL_MIN_WIDTH)
 599                 point.x = treelist.left+REPOBROWSER_CTRL_MIN_WIDTH;
 600         if (point.x > treelist.right-REPOBROWSER_CTRL_MIN_WIDTH)
 601                 point.x = treelist.right-REPOBROWSER_CTRL_MIN_WIDTH;
 602
 603         if (bDraw)
 604         {
 605                 CDC * pDC = GetDC();
 606                 DrawXorBar(pDC, oldx + 2, treelistclient.top, 4, treelistclient.bottom - treelistclient.top - 2);
 607                 ReleaseDC(pDC);
 608         }
 609
 610         oldx = point.x;
 611         oldy = point.y;
 612
 613         //position the child controls
 614         GetDlgItem(IDC_REPOTREE)->GetWindowRect(&treelist);
 615         treelist.right = point2.x - 2;
 616         ScreenToClient(&treelist);
 617         RemoveAnchor(IDC_REPOTREE);
 618         GetDlgItem(IDC_REPOTREE)->MoveWindow(&treelist);
 619         GetDlgItem(IDC_REPOLIST)->GetWindowRect(&treelist);
 620         treelist.left = point2.x + 2;
 621         ScreenToClient(&treelist);
 622         RemoveAnchor(IDC_REPOLIST);
 623         GetDlgItem(IDC_REPOLIST)->MoveWindow(&treelist);
 624
 625         AddAnchor(IDC_REPOTREE, TOP_LEFT, BOTTOM_LEFT);
 626         AddAnchor(IDC_REPOLIST, TOP_LEFT, BOTTOM_RIGHT);
 627 }
 628
 629 void CRepositoryBrowser::OnMouseMove(UINT nFlags, CPoint point)
 630 {
 631         if (bDragMode == FALSE)
 632                 return;
 633
 634         RECT rect, tree, list, treelist, treelistclient;
 635         // create an union of the tree and list control rectangle
 636         GetDlgItem(IDC_REPOLIST)->GetWindowRect(&list);
 637         GetDlgItem(IDC_REPOTREE)->GetWindowRect(&tree);
 638         UnionRect(&treelist, &tree, &list);
 639         treelistclient = treelist;
 640         ScreenToClient(&treelistclient);
 641
 642         //convert the mouse coordinates relative to the top-left of
 643         //the window
 644         ClientToScreen(&point);
 645         GetClientRect(&rect);
 646         ClientToScreen(&rect);
 647         point.x -= rect.left;
 648         point.y -= treelist.top;
 649
 650         //same for the window coordinates - make them relative to 0,0
 651         OffsetRect(&treelist, -treelist.left, -treelist.top);
 652
 653         if (point.x < treelist.left + REPOBROWSER_CTRL_MIN_WIDTH)
 654                 point.x = treelist.left + REPOBROWSER_CTRL_MIN_WIDTH;
 655         if (point.x > treelist.right - REPOBROWSER_CTRL_MIN_WIDTH)
 656                 point.x = treelist.right - REPOBROWSER_CTRL_MIN_WIDTH;
 657
 658         if ((nFlags & MK_LBUTTON) && (point.x != oldx))
 659         {
 660                 CDC * pDC = GetDC();
 661
 662                 if (pDC)
 663                 {
 664                         DrawXorBar(pDC, oldx + 2, treelistclient.top, 4, treelistclient.bottom - treelistclient.top - 2);
 665                         DrawXorBar(pDC, point.x + 2, treelistclient.top, 4, treelistclient.bottom - treelistclient.top - 2);
 666
 667                         ReleaseDC(pDC);
 668                 }
 669
 670                 oldx = point.x;
 671                 oldy = point.y;
 672         }
 673
 674         CStandAloneDialogTmpl<CResizableDialog>::OnMouseMove(nFlags, point);
 675 }
 676
 677 void CRepositoryBrowser::OnLButtonDown(UINT nFlags, CPoint point)
 678 {
 679         RECT rect, tree, list, treelist, treelistclient;
 680
 681         // create an union of the tree and list control rectangle
 682         GetDlgItem(IDC_REPOLIST)->GetWindowRect(&list);
 683         GetDlgItem(IDC_REPOTREE)->GetWindowRect(&tree);
 684         UnionRect(&treelist, &tree, &list);
 685         treelistclient = treelist;
 686         ScreenToClient(&treelistclient);
 687
 688         //convert the mouse coordinates relative to the top-left of
 689         //the window
 690         ClientToScreen(&point);
 691         GetClientRect(&rect);
 692         ClientToScreen(&rect);
 693         point.x -= rect.left;
 694         point.y -= treelist.top;
 695
 696         //same for the window coordinates - make them relative to 0,0
 697         OffsetRect(&treelist, -treelist.left, -treelist.top);
 698
 699         if (point.x < treelist.left + REPOBROWSER_CTRL_MIN_WIDTH)
 700                 return CStandAloneDialogTmpl < CResizableDialog>::OnLButtonDown(nFlags, point);
 701         if (point.x > treelist.right - 3)
 702                 return CStandAloneDialogTmpl < CResizableDialog>::OnLButtonDown(nFlags, point);
 703         if (point.x > treelist.right - REPOBROWSER_CTRL_MIN_WIDTH)
 704                 point.x = treelist.right - REPOBROWSER_CTRL_MIN_WIDTH;
 705
 706         if ((point.y < treelist.top + 3) || (point.y > treelist.bottom - 3))
 707                 return CStandAloneDialogTmpl<CResizableDialog>::OnLButtonDown(nFlags, point);
 708
 709         bDragMode = true;
 710
 711         SetCapture();
 712
 713         CDC * pDC = GetDC();
 714         DrawXorBar(pDC, point.x + 2, treelistclient.top, 4, treelistclient.bottom - treelistclient.top - 2);
 715         ReleaseDC(pDC);
 716
 717         oldx = point.x;
 718         oldy = point.y;
 719
 720         CStandAloneDialogTmpl<CResizableDialog>::OnLButtonDown(nFlags, point);
 721 }
 722
 723 void CRepositoryBrowser::OnLButtonUp(UINT nFlags, CPoint point)
 724 {
 725         if (bDragMode == FALSE)
 726                 return;
 727
 728         HandleDividerMove(point, true);
 729
 730         bDragMode = false;
 731         ReleaseCapture();
 732
 733         CStandAloneDialogTmpl<CResizableDialog>::OnLButtonUp(nFlags, point);
 734 }
 735
 736 void CRepositoryBrowser::OnCaptureChanged(CWnd *pWnd)
 737 {
 738         bDragMode = false;
 739
 740         __super::OnCaptureChanged(pWnd);
 741 }
 742
 743 void CRepositoryBrowser::DrawXorBar(CDC * pDC, int x1, int y1, int width, int height)
 744 {
 745         static WORD _dotPatternBmp[8] =
 746         {
 747                 0x0055, 0x00aa, 0x0055, 0x00aa,
 748                 0x0055, 0x00aa, 0x0055, 0x00aa
 749         };
 750
 751         HBITMAP hbm;
 752         HBRUSH  hbr, hbrushOld;
 753
 754         hbm = CreateBitmap(8, 8, 1, 1, _dotPatternBmp);
 755         hbr = CreatePatternBrush(hbm);
 756
 757         pDC->SetBrushOrg(x1, y1);
 758         hbrushOld = (HBRUSH)pDC->SelectObject(hbr);
 759
 760         PatBlt(pDC->GetSafeHdc(), x1, y1, width, height, PATINVERT);
 761
 762         pDC->SelectObject(hbrushOld);
 763
 764         DeleteObject(hbr);
 765         DeleteObject(hbm);
 766 }
 767
 768 BOOL CRepositoryBrowser::OnSetCursor(CWnd* pWnd, UINT nHitTest, UINT message)
 769 {
 770         if (pWnd == this)
 771         {
 772                 RECT rect;
 773                 POINT pt;
 774                 GetClientRect(&rect);
 775                 GetCursorPos(&pt);
 776                 ScreenToClient(&pt);
 777                 if (PtInRect(&rect, pt))
 778                 {
 779                         ClientToScreen(&pt);
 780                         // are we right of the tree control?
 781                         GetDlgItem(IDC_REPOTREE)->GetWindowRect(&rect);
 782                         if ((pt.x > rect.right) && (pt.y >= rect.top + 3) && (pt.y <= rect.bottom - 3))
 783                         {
 784                                 // but left of the list control?
 785                                 GetDlgItem(IDC_REPOLIST)->GetWindowRect(&rect);
 786                                 if (pt.x < rect.left)
 787                                 {
 788                                         HCURSOR hCur = LoadCursor(NULL, MAKEINTRESOURCE(IDC_SIZEWE));
 789                                         SetCursor(hCur);
 790                                         return TRUE;
 791                                 }
 792                         }
 793                 }
 794         }
 795         return CStandAloneDialogTmpl<CResizableDialog>::OnSetCursor(pWnd, nHitTest, message);
 796 }
 797
 798 void CRepositoryBrowser::FileSaveAs(const CString path)
 799 {
 800         CTGitPath gitPath(path);
 801
 802         CString filename;
 803         filename.Format(_T("%s-%s%s"), gitPath.GetBaseFilename(), m_sRevision.Left(g_Git.GetShortHASHLength()), gitPath.GetFileExtension());
 804         CFileDialog dlg(FALSE, NULL, filename, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT, NULL);
 805
 806         CString cmd, out;
 807         if (dlg.DoModal() == IDOK)
 808         {
 809                 filename = dlg.GetPathName();
 810                 if (g_Git.GetOneFile(m_sRevision, gitPath, filename))
 811                 {
 812                         out.Format(IDS_STATUSLIST_CHECKOUTFILEFAILED, gitPath.GetGitPathString(), m_sRevision, filename);
 813                         MessageBox(out, _T("TortoiseGit"), MB_OK);
 814                         return;
 815                 }
 816         }
 817 }