src/TortoiseProc/BrowseRefsDlg.cpp

   1 // BrowseRefsDlg.cpp : implementation file\r
   2 //\r
   3 \r
   4 #include "stdafx.h"\r
   5 #include "TortoiseProc.h"\r
   6 #include "BrowseRefsDlg.h"\r
   7 #include "LogDlg.h"\r
   8 #include "AddRemoteDlg.h"\r
   9 #include "CreateBranchTagDlg.h"\r
  10 #include "Settings\SettingGitRemote.h"\r
  11 #include "SinglePropSheetDlg.h"\r
  12 #include "ConfirmDelRefDlg.h"\r
  13 \r
  14 \r
  15 // CBrowseRefsDlg dialog\r
  16 \r
  17 IMPLEMENT_DYNAMIC(CBrowseRefsDlg, CResizableStandAloneDialog)\r
  18 \r
  19 CBrowseRefsDlg::CBrowseRefsDlg(CString cmdPath, CWnd* pParent /*=NULL*/)\r
  20 :       CResizableStandAloneDialog(CBrowseRefsDlg::IDD, pParent),\r
  21         m_cmdPath(cmdPath)\r
  22 {\r
  23 \r
  24 }\r
  25 \r
  26 CBrowseRefsDlg::~CBrowseRefsDlg()\r
  27 {\r
  28 }\r
  29 \r
  30 void CBrowseRefsDlg::DoDataExchange(CDataExchange* pDX)\r
  31 {\r
  32         CDialog::DoDataExchange(pDX);\r
  33         DDX_Control(pDX, IDC_TREE_REF,                  m_RefTreeCtrl);\r
  34         DDX_Control(pDX, IDC_LIST_REF_LEAFS,    m_ListRefLeafs);\r
  35 }\r
  36 \r
  37 \r
  38 BEGIN_MESSAGE_MAP(CBrowseRefsDlg, CResizableStandAloneDialog)\r
  39         ON_BN_CLICKED(IDOK, &CBrowseRefsDlg::OnBnClickedOk)\r
  40         ON_NOTIFY(TVN_SELCHANGED, IDC_TREE_REF, &CBrowseRefsDlg::OnTvnSelchangedTreeRef)\r
  41         ON_WM_CONTEXTMENU()\r
  42 END_MESSAGE_MAP()\r
  43 \r
  44 \r
  45 // CBrowseRefsDlg message handlers\r
  46 \r
  47 void CBrowseRefsDlg::OnBnClickedOk()\r
  48 {\r
  49         OnOK();\r
  50 }\r
  51 \r
  52 BOOL CBrowseRefsDlg::OnInitDialog()\r
  53 {\r
  54         CResizableStandAloneDialog::OnInitDialog();\r
  55 \r
  56         AddAnchor(IDC_TREE_REF, TOP_LEFT, BOTTOM_LEFT);\r
  57         AddAnchor(IDC_LIST_REF_LEAFS, TOP_LEFT, BOTTOM_RIGHT);\r
  58 \r
  59         m_ListRefLeafs.SetExtendedStyle(m_ListRefLeafs.GetExtendedStyle()|LVS_EX_FULLROWSELECT);\r
  60         m_ListRefLeafs.InsertColumn(0,L"Name",0,150);\r
  61         m_ListRefLeafs.InsertColumn(1,L"Date Last Commit",0,100);\r
  62         m_ListRefLeafs.InsertColumn(2,L"Last Commit",0,300);\r
  63         m_ListRefLeafs.InsertColumn(3,L"Hash",0,80);\r
  64 \r
  65         AddAnchor(IDOK,BOTTOM_RIGHT);\r
  66         AddAnchor(IDCANCEL,BOTTOM_RIGHT);\r
  67 \r
  68         Refresh();\r
  69 \r
  70 \r
  71         return TRUE;\r
  72 }\r
  73 \r
  74 CShadowTree* CShadowTree::GetNextSub(CString& nameLeft)\r
  75 {\r
  76         int posSlash=nameLeft.Find('/');\r
  77         CString nameSub;\r
  78         if(posSlash<0)\r
  79         {\r
  80                 nameSub=nameLeft;\r
  81                 nameLeft.Empty();//Nothing left\r
  82         }\r
  83         else\r
  84         {\r
  85                 nameSub=nameLeft.Left(posSlash);\r
  86                 nameLeft=nameLeft.Mid(posSlash+1);\r
  87         }\r
  88         if(nameSub.IsEmpty())\r
  89                 return NULL;\r
  90 \r
  91         CShadowTree& nextNode=m_ShadowTree[nameSub];\r
  92         nextNode.m_csRefName=nameSub;\r
  93         nextNode.m_pParent=this;\r
  94         return &nextNode;\r
  95 }\r
  96 \r
  97 typedef std::map<CString,CString> MAP_STRING_STRING;\r
  98 \r
  99 void CBrowseRefsDlg::Refresh()\r
 100 {\r
 101 //      m_RefMap.clear();\r
 102 //      g_Git.GetMapHashToFriendName(m_RefMap);\r
 103                 \r
 104 \r
 105         m_RefTreeCtrl.DeleteAllItems();\r
 106         m_TreeRoot.m_ShadowTree.clear();\r
 107         m_TreeRoot.m_csRefName="refs";\r
 108 //      m_TreeRoot.m_csShowName="Refs";\r
 109         m_TreeRoot.m_hTree=m_RefTreeCtrl.InsertItem(L"Refs",NULL,NULL);\r
 110         m_RefTreeCtrl.SetItemData(m_TreeRoot.m_hTree,(DWORD_PTR)&m_TreeRoot);\r
 111 \r
 112         CString allRefs;\r
 113         g_Git.Run(L"git for-each-ref --format="\r
 114                           L"%(refname)%04"\r
 115                           L"%(objectname)%04"\r
 116                           L"%(authordate:relative)%04"\r
 117                           L"%(subject)%04"\r
 118                           L"%(authorname)",\r
 119                           &allRefs,CP_UTF8);\r
 120 \r
 121         int linePos=0;\r
 122         CString singleRef;\r
 123 \r
 124         MAP_STRING_STRING refMap;\r
 125 \r
 126         //First sort on ref name\r
 127         while(!(singleRef=allRefs.Tokenize(L"\r\n",linePos)).IsEmpty())\r
 128         {\r
 129                 int valuePos=0;\r
 130                 CString refName=singleRef.Tokenize(L"\04",valuePos);\r
 131                 CString refRest=singleRef.Mid(valuePos);\r
 132                 refMap[refName]=refRest;\r
 133         }\r
 134 \r
 135 \r
 136 \r
 137 //      for(MAP_HASH_NAME::iterator iterRef=m_RefMap.begin();iterRef!=m_RefMap.end();++iterRef)\r
 138 //              for(STRING_VECTOR::iterator iterRefName=iterRef->second.begin();iterRefName!=iterRef->second.end();++iterRefName)\r
 139 //                      refName[*iterRefName]=iterRef->first;\r
 140 \r
 141         //Populate ref tree\r
 142         for(MAP_STRING_STRING::iterator iterRefMap=refMap.begin();iterRefMap!=refMap.end();++iterRefMap)\r
 143         {\r
 144                 CShadowTree& treeLeaf=GetTreeNode(iterRefMap->first);\r
 145                 CString values=iterRefMap->second;\r
 146 \r
 147                 int valuePos=0;\r
 148                 treeLeaf.m_csRefHash= values.Tokenize(L"\04",valuePos);\r
 149                 treeLeaf.m_csDate=    values.Tokenize(L"\04",valuePos);\r
 150                 treeLeaf.m_csSubject= values.Tokenize(L"\04",valuePos);\r
 151                 treeLeaf.m_csAuthor=  values.Tokenize(L"\04",valuePos);\r
 152         }\r
 153 \r
 154         CString currHead;\r
 155         g_Git.Run(L"git symbolic-ref HEAD",&currHead,CP_UTF8);\r
 156 \r
 157         currHead.Trim(L"\r\n\t ");\r
 158 \r
 159         if(!SelectRef(currHead))\r
 160                 //Probably not on a branch. Select root node.\r
 161                 m_RefTreeCtrl.Expand(m_TreeRoot.m_hTree,TVE_EXPAND);\r
 162 \r
 163 }\r
 164 \r
 165 bool CBrowseRefsDlg::SelectRef(CString refName)\r
 166 {\r
 167         if(wcsnicmp(refName,L"refs/",5)!=0)\r
 168                 return false; // Not a ref name\r
 169 \r
 170         CShadowTree& treeLeafHead=GetTreeNode(refName);\r
 171         if(treeLeafHead.m_pParent==NULL)\r
 172                 return false; //Weird... should not occur.\r
 173 \r
 174         //This is the current head.\r
 175         m_RefTreeCtrl.Select(treeLeafHead.m_pParent->m_hTree,TVGN_CARET);\r
 176 \r
 177         for(int indexPos = 0; indexPos < m_ListRefLeafs.GetItemCount(); ++indexPos)\r
 178         {\r
 179                 CShadowTree* pCurrShadowTree = (CShadowTree*)m_ListRefLeafs.GetItemData(indexPos);\r
 180                 if(pCurrShadowTree == &treeLeafHead)\r
 181                 {\r
 182                         m_ListRefLeafs.SetItemState(indexPos,LVIS_SELECTED,LVIS_SELECTED);\r
 183                 }\r
 184         }\r
 185 \r
 186         return true;\r
 187 }\r
 188 \r
 189 CShadowTree& CBrowseRefsDlg::GetTreeNode(CString refName, CShadowTree* pTreePos)\r
 190 {\r
 191         if(pTreePos==NULL)\r
 192         {\r
 193                 if(wcsnicmp(refName,L"refs/",5)==0)\r
 194                         refName=refName.Mid(5);\r
 195                 pTreePos=&m_TreeRoot;\r
 196         }\r
 197         if(refName.IsEmpty())\r
 198                 return *pTreePos;//Found leaf\r
 199 \r
 200         CShadowTree* pNextTree=pTreePos->GetNextSub(refName);\r
 201         if(pNextTree==NULL)\r
 202         {\r
 203                 //Should not occur when all ref-names are valid.\r
 204                 ASSERT(FALSE);\r
 205                 return *pTreePos;\r
 206         }\r
 207 \r
 208         if(!refName.IsEmpty())\r
 209         {\r
 210                 //When the refName is not empty, this node is not a leaf, so lets add it to the tree control.\r
 211                 //Leafs are for the list control.\r
 212                 if(pNextTree->m_hTree==NULL)\r
 213                 {\r
 214                         //New tree. Create node in control.\r
 215                         pNextTree->m_hTree=m_RefTreeCtrl.InsertItem(pNextTree->m_csRefName,pTreePos->m_hTree,NULL);\r
 216                         m_RefTreeCtrl.SetItemData(pNextTree->m_hTree,(DWORD_PTR)pNextTree);\r
 217                 }\r
 218         }\r
 219 \r
 220         return GetTreeNode(refName,pNextTree);\r
 221 }\r
 222 \r
 223 \r
 224 void CBrowseRefsDlg::OnTvnSelchangedTreeRef(NMHDR *pNMHDR, LRESULT *pResult)\r
 225 {\r
 226         LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);\r
 227         *pResult = 0;\r
 228 \r
 229         FillListCtrlForTreeNode(pNMTreeView->itemNew.hItem);\r
 230 }\r
 231 \r
 232 void CBrowseRefsDlg::FillListCtrlForTreeNode(HTREEITEM treeNode)\r
 233 {\r
 234         m_ListRefLeafs.DeleteAllItems();\r
 235 \r
 236         CShadowTree* pTree=(CShadowTree*)(m_RefTreeCtrl.GetItemData(treeNode));\r
 237         if(pTree==NULL)\r
 238         {\r
 239                 ASSERT(FALSE);\r
 240                 return;\r
 241         }\r
 242         FillListCtrlForShadowTree(pTree,L"",true);\r
 243 }\r
 244 \r
 245 void CBrowseRefsDlg::FillListCtrlForShadowTree(CShadowTree* pTree, CString refNamePrefix, bool isFirstLevel)\r
 246 {\r
 247         if(pTree->IsLeaf())\r
 248         {\r
 249                 int indexItem=m_ListRefLeafs.InsertItem(m_ListRefLeafs.GetItemCount(),L"");\r
 250 \r
 251                 m_ListRefLeafs.SetItemData(indexItem,(DWORD_PTR)pTree);\r
 252                 m_ListRefLeafs.SetItemText(indexItem,0,refNamePrefix+pTree->m_csRefName);\r
 253                 m_ListRefLeafs.SetItemText(indexItem,1,pTree->m_csDate);\r
 254                 m_ListRefLeafs.SetItemText(indexItem,2,pTree->m_csSubject);\r
 255                 m_ListRefLeafs.SetItemText(indexItem,3,pTree->m_csRefHash);\r
 256         }\r
 257         else\r
 258         {\r
 259 \r
 260                 CString csThisName;\r
 261                 if(!isFirstLevel)\r
 262                         csThisName=refNamePrefix+pTree->m_csRefName+L"/";\r
 263                 for(CShadowTree::TShadowTreeMap::iterator itSubTree=pTree->m_ShadowTree.begin(); itSubTree!=pTree->m_ShadowTree.end(); ++itSubTree)\r
 264                 {\r
 265                         FillListCtrlForShadowTree(&itSubTree->second,csThisName,false);\r
 266                 }\r
 267         }\r
 268 }\r
 269 \r
 270 void CBrowseRefsDlg::OnContextMenu_ListRefLeafs(CPoint point)\r
 271 {\r
 272 \r
 273         CPoint clientPoint=point;\r
 274         m_RefTreeCtrl.ScreenToClient(&clientPoint);\r
 275 \r
 276 \r
 277         std::vector<CShadowTree*> selectedTrees;\r
 278         selectedTrees.reserve(m_ListRefLeafs.GetSelectedCount());\r
 279         POSITION pos=m_ListRefLeafs.GetFirstSelectedItemPosition();\r
 280         while(pos)\r
 281         {\r
 282                 selectedTrees.push_back(\r
 283                         (CShadowTree*)m_ListRefLeafs.GetItemData(\r
 284                                 m_ListRefLeafs.GetNextSelectedItem(pos)));\r
 285         }\r
 286 \r
 287         CMenu popupMenu;\r
 288         popupMenu.CreatePopupMenu();\r
 289 \r
 290         if(selectedTrees.size()==1)\r
 291         {\r
 292                 popupMenu.AppendMenu(MF_STRING,eCmd_ViewLog,L"View log");\r
 293                 if(selectedTrees[0]->IsFrom(L"refs/heads"))\r
 294                         popupMenu.AppendMenu(MF_STRING,eCmd_DeleteBranch,L"Delete Branch");\r
 295                 else if(selectedTrees[0]->IsFrom(L"refs/tags"))\r
 296                         popupMenu.AppendMenu(MF_STRING,eCmd_DeleteTag,L"Delete Tag");\r
 297 \r
 298 //              CShadowTree* pTree = (CShadowTree*)m_ListRefLeafs.GetItemData(pNMHDR->idFrom);\r
 299 //              if(pTree==NULL)\r
 300 //                      return;\r
 301         }\r
 302 \r
 303 \r
 304         eCmd cmd=(eCmd)popupMenu.TrackPopupMenuEx(TPM_LEFTALIGN|TPM_RETURNCMD, point.x, point.y, this, 0);\r
 305         switch(cmd)\r
 306         {\r
 307         case eCmd_ViewLog:\r
 308                 {\r
 309                         CLogDlg dlg;\r
 310                         dlg.SetStartRef(selectedTrees[0]->m_csRefHash);\r
 311                         dlg.DoModal();\r
 312                 }\r
 313                 break;\r
 314         case eCmd_DeleteBranch:\r
 315                 {\r
 316                         CConfirmDelRefDlg(selectedTrees[0]->GetRefName(),this).DoModal();\r
 317                         Refresh();\r
 318                 }\r
 319                 break;\r
 320         case eCmd_DeleteTag:\r
 321                 {\r
 322                         CConfirmDelRefDlg(selectedTrees[0]->GetRefName(),this).DoModal();\r
 323                         Refresh();\r
 324                 }\r
 325                 break;\r
 326         }\r
 327 }\r
 328 \r
 329 void CBrowseRefsDlg::OnContextMenu(CWnd* pWndFrom, CPoint point)\r
 330 {\r
 331         if(pWndFrom==&m_RefTreeCtrl)       OnContextMenu_RefTreeCtrl(point);\r
 332         else if(pWndFrom==&m_ListRefLeafs) OnContextMenu_ListRefLeafs(point);\r
 333 }\r
 334 \r
 335 void CBrowseRefsDlg::OnContextMenu_RefTreeCtrl(CPoint point)\r
 336 {\r
 337         CMenu popupMenu;\r
 338         popupMenu.CreatePopupMenu();\r
 339 \r
 340         CPoint clientPoint=point;\r
 341         m_RefTreeCtrl.ScreenToClient(&clientPoint);\r
 342 \r
 343         HTREEITEM hTreeItem=m_RefTreeCtrl.HitTest(clientPoint);\r
 344         if(hTreeItem!=NULL)\r
 345         {\r
 346                 m_RefTreeCtrl.Select(hTreeItem,TVGN_CARET);\r
 347                 CShadowTree* pTree=(CShadowTree*)m_RefTreeCtrl.GetItemData(hTreeItem);\r
 348                 if(pTree->IsFrom(L"refs/remotes"))\r
 349                 {\r
 350 //                      popupMenu.AppendMenu(MF_STRING,eCmd_AddRemote,L"Add Remote");\r
 351                         popupMenu.AppendMenu(MF_STRING,eCmd_ManageRemotes,L"Manage Remotes");\r
 352                 }\r
 353                 else if(pTree->IsFrom(L"refs/heads"))\r
 354                         popupMenu.AppendMenu(MF_STRING,eCmd_CreateBranch,L"Create Branch");\r
 355                 else if(pTree->IsFrom(L"refs/tags"))\r
 356                         popupMenu.AppendMenu(MF_STRING,eCmd_CreateTag,L"Create Tag");\r
 357         }\r
 358 \r
 359         eCmd cmd=(eCmd)popupMenu.TrackPopupMenuEx(TPM_LEFTALIGN|TPM_RETURNCMD, point.x, point.y, this, 0);\r
 360         switch(cmd)\r
 361         {\r
 362         case eCmd_AddRemote:\r
 363                 {\r
 364                         CAddRemoteDlg(this).DoModal();\r
 365                         Refresh();\r
 366                 }\r
 367                 break;\r
 368         case eCmd_ManageRemotes:\r
 369                 {\r
 370                         CSinglePropSheetDlg(L"Git Remote Settings",new CSettingGitRemote(m_cmdPath),this).DoModal();\r
 371 //                      CSettingGitRemote W_Remotes(m_cmdPath);\r
 372 //                      W_Remotes.DoModal();\r
 373                         Refresh();\r
 374                 }\r
 375                 break;\r
 376         case eCmd_CreateBranch:\r
 377                 {\r
 378                         CCreateBranchTagDlg dlg(this);\r
 379                         dlg.m_bIsTag=false;\r
 380                         dlg.DoModal();\r
 381                         Refresh();\r
 382                 }\r
 383                 break;\r
 384         case eCmd_CreateTag:\r
 385                 {\r
 386                         CCreateBranchTagDlg dlg(this);\r
 387                         dlg.m_bIsTag=true;\r
 388                         dlg.DoModal();\r
 389                         Refresh();\r
 390                 }\r
 391                 break;\r
 392         }\r
 393 }\r
 394 \r
 395 BOOL CBrowseRefsDlg::PreTranslateMessage(MSG* pMsg)\r
 396 {\r
 397         if (pMsg->message == WM_KEYDOWN)\r
 398         {\r
 399                 switch (pMsg->wParam)\r
 400                 {\r
 401 /*              case VK_RETURN:\r
 402                         {\r
 403                                 if (GetAsyncKeyState(VK_CONTROL)&0x8000)\r
 404                                 {\r
 405                                         if ( GetDlgItem(IDOK)->IsWindowEnabled() )\r
 406                                         {\r
 407                                                 PostMessage(WM_COMMAND, IDOK);\r
 408                                         }\r
 409                                         return TRUE;\r
 410                                 }\r
 411                         }\r
 412                         break;\r
 413 */              case VK_F5:\r
 414                         {\r
 415                                 Refresh();\r
 416                         }\r
 417                         break;\r
 418                 }\r
 419         }\r
 420 \r
 421 \r
 422         return CResizableStandAloneDialog::PreTranslateMessage(pMsg);\r
 423 }\r