1 // BrowseRefsDlg.cpp : implementation file
\r
5 #include "TortoiseProc.h"
\r
6 #include "BrowseRefsDlg.h"
\r
8 #include "AddRemoteDlg.h"
\r
9 #include "AppUtils.h"
\r
10 #include "Settings\SettingGitRemote.h"
\r
11 #include "SinglePropSheetDlg.h"
\r
12 #include "MessageBox.h"
\r
14 // CBrowseRefsDlg dialog
\r
16 IMPLEMENT_DYNAMIC(CBrowseRefsDlg, CResizableStandAloneDialog)
\r
18 CBrowseRefsDlg::CBrowseRefsDlg(CString cmdPath, CWnd* pParent /*=NULL*/)
\r
19 : CResizableStandAloneDialog(CBrowseRefsDlg::IDD, pParent),
\r
22 m_currSortDesc(false)
\r
27 CBrowseRefsDlg::~CBrowseRefsDlg()
\r
31 void CBrowseRefsDlg::DoDataExchange(CDataExchange* pDX)
\r
33 CDialog::DoDataExchange(pDX);
\r
34 DDX_Control(pDX, IDC_TREE_REF, m_RefTreeCtrl);
\r
35 DDX_Control(pDX, IDC_LIST_REF_LEAFS, m_ListRefLeafs);
\r
39 BEGIN_MESSAGE_MAP(CBrowseRefsDlg, CResizableStandAloneDialog)
\r
40 ON_BN_CLICKED(IDOK, &CBrowseRefsDlg::OnBnClickedOk)
\r
41 ON_NOTIFY(TVN_SELCHANGED, IDC_TREE_REF, &CBrowseRefsDlg::OnTvnSelchangedTreeRef)
\r
43 ON_NOTIFY(LVN_COLUMNCLICK, IDC_LIST_REF_LEAFS, &CBrowseRefsDlg::OnLvnColumnclickListRefLeafs)
\r
47 // CBrowseRefsDlg message handlers
\r
49 void CBrowseRefsDlg::OnBnClickedOk()
\r
54 BOOL CBrowseRefsDlg::OnInitDialog()
\r
56 CResizableStandAloneDialog::OnInitDialog();
\r
58 AddAnchor(IDC_TREE_REF, TOP_LEFT, BOTTOM_LEFT);
\r
59 AddAnchor(IDC_LIST_REF_LEAFS, TOP_LEFT, BOTTOM_RIGHT);
\r
61 m_ListRefLeafs.SetExtendedStyle(m_ListRefLeafs.GetExtendedStyle()|LVS_EX_FULLROWSELECT);
\r
62 m_ListRefLeafs.InsertColumn(eCol_Name, L"Name",0,150);
\r
63 m_ListRefLeafs.InsertColumn(eCol_Date, L"Date Last Commit",0,100);
\r
64 m_ListRefLeafs.InsertColumn(eCol_Msg, L"Last Commit",0,300);
\r
65 m_ListRefLeafs.InsertColumn(eCol_Hash, L"Hash",0,80);
\r
67 AddAnchor(IDOK,BOTTOM_RIGHT);
\r
68 AddAnchor(IDCANCEL,BOTTOM_RIGHT);
\r
76 CShadowTree* CShadowTree::GetNextSub(CString& nameLeft, bool bCreateIfNotExist)
\r
78 int posSlash=nameLeft.Find('/');
\r
83 nameLeft.Empty();//Nothing left
\r
87 nameSub=nameLeft.Left(posSlash);
\r
88 nameLeft=nameLeft.Mid(posSlash+1);
\r
90 if(nameSub.IsEmpty())
\r
93 if(!bCreateIfNotExist && m_ShadowTree.find(nameSub)==m_ShadowTree.end())
\r
96 CShadowTree& nextNode=m_ShadowTree[nameSub];
\r
97 nextNode.m_csRefName=nameSub;
\r
98 nextNode.m_pParent=this;
\r
102 typedef std::map<CString,CString> MAP_STRING_STRING;
\r
104 void CBrowseRefsDlg::Refresh(bool bSelectCurHead)
\r
106 // m_RefMap.clear();
\r
107 // g_Git.GetMapHashToFriendName(m_RefMap);
\r
112 g_Git.Run(L"git symbolic-ref HEAD",&selectRef,CP_UTF8);
\r
113 selectRef.Trim(L"\r\n\t ");
\r
117 POSITION pos=m_ListRefLeafs.GetFirstSelectedItemPosition();
\r
118 //List ctrl selection?
\r
121 CShadowTree* pTree=(CShadowTree*)m_ListRefLeafs.GetItemData(
\r
122 m_ListRefLeafs.GetNextSelectedItem(pos));
\r
123 selectRef=pTree->GetRefName();
\r
127 //Tree ctrl selection?
\r
128 HTREEITEM hTree=m_RefTreeCtrl.GetSelectedItem();
\r
131 CShadowTree* pTree=(CShadowTree*)m_RefTreeCtrl.GetItemData(hTree);
\r
132 selectRef=pTree->GetRefName();
\r
137 m_RefTreeCtrl.DeleteAllItems();
\r
138 m_ListRefLeafs.DeleteAllItems();
\r
139 m_TreeRoot.m_ShadowTree.clear();
\r
140 m_TreeRoot.m_csRefName="refs";
\r
141 // m_TreeRoot.m_csShowName="Refs";
\r
142 m_TreeRoot.m_hTree=m_RefTreeCtrl.InsertItem(L"Refs",NULL,NULL);
\r
143 m_RefTreeCtrl.SetItemData(m_TreeRoot.m_hTree,(DWORD_PTR)&m_TreeRoot);
\r
146 g_Git.Run(L"git for-each-ref --format="
\r
148 L"%(objectname)%04"
\r
149 L"%(authordate:relative)%04"
\r
151 L"%(authorname)%04"
\r
152 L"%(authordate:iso8601)",
\r
158 MAP_STRING_STRING refMap;
\r
160 //First sort on ref name
\r
161 while(!(singleRef=allRefs.Tokenize(L"\r\n",linePos)).IsEmpty())
\r
164 CString refName=singleRef.Tokenize(L"\04",valuePos);
\r
165 CString refRest=singleRef.Mid(valuePos);
\r
166 refMap[refName]=refRest;
\r
171 // for(MAP_HASH_NAME::iterator iterRef=m_RefMap.begin();iterRef!=m_RefMap.end();++iterRef)
\r
172 // for(STRING_VECTOR::iterator iterRefName=iterRef->second.begin();iterRefName!=iterRef->second.end();++iterRefName)
\r
173 // refName[*iterRefName]=iterRef->first;
\r
175 //Populate ref tree
\r
176 for(MAP_STRING_STRING::iterator iterRefMap=refMap.begin();iterRefMap!=refMap.end();++iterRefMap)
\r
178 CShadowTree& treeLeaf=GetTreeNode(iterRefMap->first,NULL,true);
\r
179 CString values=iterRefMap->second;
\r
182 treeLeaf.m_csRefHash= values.Tokenize(L"\04",valuePos);
\r
183 treeLeaf.m_csDate= values.Tokenize(L"\04",valuePos);
\r
184 treeLeaf.m_csSubject= values.Tokenize(L"\04",valuePos);
\r
185 treeLeaf.m_csAuthor= values.Tokenize(L"\04",valuePos);
\r
186 treeLeaf.m_csDate_Iso8601= values.Tokenize(L"\04",valuePos);
\r
190 if(selectRef.IsEmpty() || !SelectRef(selectRef))
\r
191 //Probably not on a branch. Select root node.
\r
192 m_RefTreeCtrl.Expand(m_TreeRoot.m_hTree,TVE_EXPAND);
\r
196 bool CBrowseRefsDlg::SelectRef(CString refName)
\r
198 if(wcsnicmp(refName,L"refs/",5)!=0)
\r
199 return false; // Not a ref name
\r
201 CShadowTree& treeLeafHead=GetTreeNode(refName,NULL,false);
\r
202 if(treeLeafHead.m_hTree != NULL)
\r
204 //Not a leaf. Select tree node and return
\r
205 m_RefTreeCtrl.Select(treeLeafHead.m_hTree,TVGN_CARET);
\r
209 if(treeLeafHead.m_pParent==NULL)
\r
210 return false; //Weird... should not occur.
\r
212 //This is the current head.
\r
213 m_RefTreeCtrl.Select(treeLeafHead.m_pParent->m_hTree,TVGN_CARET);
\r
215 for(int indexPos = 0; indexPos < m_ListRefLeafs.GetItemCount(); ++indexPos)
\r
217 CShadowTree* pCurrShadowTree = (CShadowTree*)m_ListRefLeafs.GetItemData(indexPos);
\r
218 if(pCurrShadowTree == &treeLeafHead)
\r
220 m_ListRefLeafs.SetItemState(indexPos,LVIS_SELECTED,LVIS_SELECTED);
\r
221 m_ListRefLeafs.EnsureVisible(indexPos,FALSE);
\r
228 CShadowTree& CBrowseRefsDlg::GetTreeNode(CString refName, CShadowTree* pTreePos, bool bCreateIfNotExist)
\r
232 if(wcsnicmp(refName,L"refs/",5)==0)
\r
233 refName=refName.Mid(5);
\r
234 pTreePos=&m_TreeRoot;
\r
236 if(refName.IsEmpty())
\r
237 return *pTreePos;//Found leaf
\r
239 CShadowTree* pNextTree=pTreePos->GetNextSub(refName,bCreateIfNotExist);
\r
240 if(pNextTree==NULL)
\r
242 //Should not occur when all ref-names are valid and bCreateIfNotExist is true.
\r
243 ASSERT(!bCreateIfNotExist);
\r
247 if(!refName.IsEmpty())
\r
249 //When the refName is not empty, this node is not a leaf, so lets add it to the tree control.
\r
250 //Leafs are for the list control.
\r
251 if(pNextTree->m_hTree==NULL)
\r
253 //New tree. Create node in control.
\r
254 pNextTree->m_hTree=m_RefTreeCtrl.InsertItem(pNextTree->m_csRefName,pTreePos->m_hTree,NULL);
\r
255 m_RefTreeCtrl.SetItemData(pNextTree->m_hTree,(DWORD_PTR)pNextTree);
\r
259 return GetTreeNode(refName, pNextTree, bCreateIfNotExist);
\r
263 void CBrowseRefsDlg::OnTvnSelchangedTreeRef(NMHDR *pNMHDR, LRESULT *pResult)
\r
265 LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);
\r
268 FillListCtrlForTreeNode(pNMTreeView->itemNew.hItem);
\r
271 void CBrowseRefsDlg::FillListCtrlForTreeNode(HTREEITEM treeNode)
\r
273 m_ListRefLeafs.DeleteAllItems();
\r
275 CShadowTree* pTree=(CShadowTree*)(m_RefTreeCtrl.GetItemData(treeNode));
\r
281 FillListCtrlForShadowTree(pTree,L"",true);
\r
284 void CBrowseRefsDlg::FillListCtrlForShadowTree(CShadowTree* pTree, CString refNamePrefix, bool isFirstLevel)
\r
286 if(pTree->IsLeaf())
\r
288 int indexItem=m_ListRefLeafs.InsertItem(m_ListRefLeafs.GetItemCount(),L"");
\r
290 m_ListRefLeafs.SetItemData(indexItem,(DWORD_PTR)pTree);
\r
291 m_ListRefLeafs.SetItemText(indexItem,eCol_Name, refNamePrefix+pTree->m_csRefName);
\r
292 m_ListRefLeafs.SetItemText(indexItem,eCol_Date, pTree->m_csDate);
\r
293 m_ListRefLeafs.SetItemText(indexItem,eCol_Msg, pTree->m_csSubject);
\r
294 m_ListRefLeafs.SetItemText(indexItem,eCol_Hash, pTree->m_csRefHash);
\r
299 CString csThisName;
\r
301 csThisName=refNamePrefix+pTree->m_csRefName+L"/";
\r
302 for(CShadowTree::TShadowTreeMap::iterator itSubTree=pTree->m_ShadowTree.begin(); itSubTree!=pTree->m_ShadowTree.end(); ++itSubTree)
\r
304 FillListCtrlForShadowTree(&itSubTree->second,csThisName,false);
\r
309 bool CBrowseRefsDlg::ConfirmDeleteRef(CString completeRefName)
\r
314 UINT mbIcon=MB_ICONQUESTION;
\r
315 csMessage=L"Are you sure you want to delete the ";
\r
316 if(wcsncmp(completeRefName,L"refs/heads",10)==0)
\r
318 CString branchToDelete = completeRefName.Mid(11);
\r
319 csTitle.Format(L"Confirm deletion of branch %s", branchToDelete);
\r
320 csMessage += "branch:\r\n\r\n<b>";
\r
321 csMessage += branchToDelete;
\r
322 csMessage += "</b>";
\r
324 //Check if branch is fully merged in HEAD
\r
325 CString branchHash = g_Git.GetHash(completeRefName);
\r
326 CString commonAncestor;
\r
328 cmd.Format(L"git.exe merge-base HEAD %s",completeRefName);
\r
329 g_Git.Run(cmd,&commonAncestor,CP_UTF8);
\r
331 branchHash=branchHash.Left(40);
\r
332 commonAncestor=commonAncestor.Left(40);
\r
334 if(commonAncestor != branchHash)
\r
336 csMessage += L"\r\n\r\n<b>Warning:\r\nThis branch is not fully merged into HEAD.</b>";
\r
337 mbIcon=MB_ICONWARNING;
\r
340 else if(wcsncmp(completeRefName,L"refs/tags",9)==0)
\r
342 CString tagToDelete = completeRefName.Mid(10);
\r
343 csTitle.Format(L"Confirm deletion of tag %s", tagToDelete);
\r
344 csMessage += "tag:\r\n\r\n<b>";
\r
345 csMessage += tagToDelete;
\r
346 csMessage += "</b>";
\r
349 return CMessageBox::Show(m_hWnd,csMessage,csTitle,MB_YESNO|mbIcon)==IDYES;
\r
354 bool CBrowseRefsDlg::DoDeleteRef(CString completeRefName, bool bForce)
\r
356 if(wcsncmp(completeRefName,L"refs/heads",10)==0)
\r
358 CString branchToDelete = completeRefName.Mid(11);
\r
360 cmd.Format(L"git.exe branch -%c %s",bForce?L'D':L'd',branchToDelete);
\r
361 CString resultDummy;
\r
362 if(g_Git.Run(cmd,&resultDummy,CP_UTF8)!=0)
\r
365 errorMsg.Format(L"Could not delete branch %s. Message from git:\r\n\r\n%s",branchToDelete,resultDummy);
\r
366 CMessageBox::Show(m_hWnd,errorMsg,L"Error deleting branch",MB_OK|MB_ICONERROR);
\r
370 else if(wcsncmp(completeRefName,L"refs/tags",9)==0)
\r
372 CString tagToDelete = completeRefName.Mid(10);
\r
374 cmd.Format(L"git.exe tag -d %s",tagToDelete);
\r
375 CString resultDummy;
\r
376 if(g_Git.Run(cmd,&resultDummy,CP_UTF8)!=0)
\r
379 errorMsg.Format(L"Could not delete tag %s. Message from git:\r\n\r\n%s",tagToDelete,resultDummy);
\r
380 CMessageBox::Show(m_hWnd,errorMsg,L"Error deleting tag",MB_OK|MB_ICONERROR);
\r
387 void CBrowseRefsDlg::OnContextMenu(CWnd* pWndFrom, CPoint point)
\r
389 if(pWndFrom==&m_RefTreeCtrl) OnContextMenu_RefTreeCtrl(point);
\r
390 else if(pWndFrom==&m_ListRefLeafs) OnContextMenu_ListRefLeafs(point);
\r
393 void CBrowseRefsDlg::OnContextMenu_RefTreeCtrl(CPoint point)
\r
395 CPoint clientPoint=point;
\r
396 m_RefTreeCtrl.ScreenToClient(&clientPoint);
\r
398 HTREEITEM hTreeItem=m_RefTreeCtrl.HitTest(clientPoint);
\r
399 if(hTreeItem!=NULL)
\r
400 m_RefTreeCtrl.Select(hTreeItem,TVGN_CARET);
\r
402 ShowContextMenu(point,hTreeItem,VectorPShadowTree());
\r
406 void CBrowseRefsDlg::OnContextMenu_ListRefLeafs(CPoint point)
\r
408 std::vector<CShadowTree*> selectedLeafs;
\r
409 selectedLeafs.reserve(m_ListRefLeafs.GetSelectedCount());
\r
410 POSITION pos=m_ListRefLeafs.GetFirstSelectedItemPosition();
\r
413 selectedLeafs.push_back(
\r
414 (CShadowTree*)m_ListRefLeafs.GetItemData(
\r
415 m_ListRefLeafs.GetNextSelectedItem(pos)));
\r
418 ShowContextMenu(point,m_RefTreeCtrl.GetSelectedItem(),selectedLeafs);
\r
421 void CBrowseRefsDlg::ShowContextMenu(CPoint point, HTREEITEM hTreePos, VectorPShadowTree& selectedLeafs)
\r
424 popupMenu.CreatePopupMenu();
\r
426 if(selectedLeafs.size()==1)
\r
428 popupMenu.AppendMenu(MF_STRING,eCmd_ViewLog,L"View log");
\r
429 if(selectedLeafs[0]->IsFrom(L"refs/heads"))
\r
430 popupMenu.AppendMenu(MF_STRING,eCmd_DeleteBranch,L"Delete Branch");
\r
431 else if(selectedLeafs[0]->IsFrom(L"refs/tags"))
\r
432 popupMenu.AppendMenu(MF_STRING,eCmd_DeleteTag,L"Delete Tag");
\r
434 // CShadowTree* pTree = (CShadowTree*)m_ListRefLeafs.GetItemData(pNMHDR->idFrom);
\r
441 CShadowTree* pTree=(CShadowTree*)m_RefTreeCtrl.GetItemData(hTreePos);
\r
442 if(pTree->IsFrom(L"refs/remotes"))
\r
444 // popupMenu.AppendMenu(MF_STRING,eCmd_AddRemote,L"Add Remote");
\r
445 popupMenu.AppendMenu(MF_STRING,eCmd_ManageRemotes,L"Manage Remotes");
\r
447 else if(pTree->IsFrom(L"refs/heads"))
\r
448 popupMenu.AppendMenu(MF_STRING,eCmd_CreateBranch,L"Create Branch");
\r
449 else if(pTree->IsFrom(L"refs/tags"))
\r
450 popupMenu.AppendMenu(MF_STRING,eCmd_CreateTag,L"Create Tag");
\r
454 eCmd cmd=(eCmd)popupMenu.TrackPopupMenuEx(TPM_LEFTALIGN|TPM_RETURNCMD, point.x, point.y, this, 0);
\r
460 dlg.SetStartRef(selectedLeafs[0]->m_csRefHash);
\r
464 case eCmd_DeleteBranch:
\r
466 if(ConfirmDeleteRef(selectedLeafs[0]->GetRefName()))
\r
467 DoDeleteRef(selectedLeafs[0]->GetRefName(), true);
\r
471 case eCmd_DeleteTag:
\r
473 if(ConfirmDeleteRef(selectedLeafs[0]->GetRefName()))
\r
474 DoDeleteRef(selectedLeafs[0]->GetRefName(), true);
\r
478 case eCmd_AddRemote:
\r
480 CAddRemoteDlg(this).DoModal();
\r
484 case eCmd_ManageRemotes:
\r
486 CSinglePropSheetDlg(L"Git Remote Settings",new CSettingGitRemote(m_cmdPath),this).DoModal();
\r
487 // CSettingGitRemote W_Remotes(m_cmdPath);
\r
488 // W_Remotes.DoModal();
\r
492 case eCmd_CreateBranch:
\r
494 CAppUtils::CreateBranchTag(false);
\r
498 case eCmd_CreateTag:
\r
500 CAppUtils::CreateBranchTag(true);
\r
507 BOOL CBrowseRefsDlg::PreTranslateMessage(MSG* pMsg)
\r
509 if (pMsg->message == WM_KEYDOWN)
\r
511 switch (pMsg->wParam)
\r
515 if (GetAsyncKeyState(VK_CONTROL)&0x8000)
\r
517 if ( GetDlgItem(IDOK)->IsWindowEnabled() )
\r
519 PostMessage(WM_COMMAND, IDOK);
\r
534 return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
\r
537 class CRefLeafListCompareFunc
\r
540 CRefLeafListCompareFunc(CListCtrl* pList, int col, bool desc):m_col(col),m_desc(desc),m_pList(pList){}
\r
542 static int CALLBACK StaticCompare(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)
\r
544 return ((CRefLeafListCompareFunc*)lParamSort)->Compare(lParam1,lParam2);
\r
547 int Compare(LPARAM lParam1, LPARAM lParam2)
\r
550 (CShadowTree*)m_pList->GetItemData(lParam1),
\r
551 (CShadowTree*)m_pList->GetItemData(lParam2));
\r
554 int Compare(CShadowTree* pLeft, CShadowTree* pRight)
\r
556 int result=CompareNoDesc(pLeft,pRight);
\r
562 int CompareNoDesc(CShadowTree* pLeft, CShadowTree* pRight)
\r
566 case CBrowseRefsDlg::eCol_Name: return pLeft->GetRefName().CompareNoCase(pRight->GetRefName());
\r
567 case CBrowseRefsDlg::eCol_Date: return pLeft->m_csDate_Iso8601.CompareNoCase(pRight->m_csDate_Iso8601);
\r
568 case CBrowseRefsDlg::eCol_Msg: return pLeft->m_csSubject.CompareNoCase(pRight->m_csSubject);
\r
569 case CBrowseRefsDlg::eCol_Hash: return pLeft->m_csRefHash.CompareNoCase(pRight->m_csRefHash);
\r
576 CListCtrl* m_pList;
\r
581 void CBrowseRefsDlg::OnLvnColumnclickListRefLeafs(NMHDR *pNMHDR, LRESULT *pResult)
\r
583 LPNMLISTVIEW pNMLV = reinterpret_cast<LPNMLISTVIEW>(pNMHDR);
\r
586 if(m_currSortCol == pNMLV->iSubItem)
\r
587 m_currSortDesc = !m_currSortDesc;
\r
590 m_currSortCol = pNMLV->iSubItem;
\r
591 m_currSortDesc = false;
\r
594 CRefLeafListCompareFunc compareFunc(&m_ListRefLeafs, m_currSortCol, m_currSortDesc);
\r
595 m_ListRefLeafs.SortItemsEx(&CRefLeafListCompareFunc::StaticCompare, (DWORD_PTR)&compareFunc);
\r