1 // BrowseRefsDlg.cpp : implementation file
\r
5 #include "TortoiseProc.h"
\r
6 #include "BrowseRefsDlg.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 "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
25 CBrowseRefsDlg::~CBrowseRefsDlg()
\r
29 void CBrowseRefsDlg::DoDataExchange(CDataExchange* pDX)
\r
31 CDialog::DoDataExchange(pDX);
\r
32 DDX_Control(pDX, IDC_TREE_REF, m_RefTreeCtrl);
\r
33 DDX_Control(pDX, IDC_LIST_REF_LEAFS, m_ListRefLeafs);
\r
37 BEGIN_MESSAGE_MAP(CBrowseRefsDlg, CResizableStandAloneDialog)
\r
38 ON_BN_CLICKED(IDOK, &CBrowseRefsDlg::OnBnClickedOk)
\r
39 ON_NOTIFY(TVN_SELCHANGED, IDC_TREE_REF, &CBrowseRefsDlg::OnTvnSelchangedTreeRef)
\r
44 // CBrowseRefsDlg message handlers
\r
46 void CBrowseRefsDlg::OnBnClickedOk()
\r
51 BOOL CBrowseRefsDlg::OnInitDialog()
\r
53 CResizableStandAloneDialog::OnInitDialog();
\r
55 AddAnchor(IDC_TREE_REF, TOP_LEFT, BOTTOM_LEFT);
\r
56 AddAnchor(IDC_LIST_REF_LEAFS, TOP_LEFT, BOTTOM_RIGHT);
\r
58 m_ListRefLeafs.SetExtendedStyle(m_ListRefLeafs.GetExtendedStyle()|LVS_EX_FULLROWSELECT);
\r
59 m_ListRefLeafs.InsertColumn(0,L"Name",0,150);
\r
60 m_ListRefLeafs.InsertColumn(1,L"Date Last Commit",0,100);
\r
61 m_ListRefLeafs.InsertColumn(2,L"Last Commit",0,300);
\r
62 m_ListRefLeafs.InsertColumn(3,L"Hash",0,80);
\r
64 AddAnchor(IDOK,BOTTOM_RIGHT);
\r
65 AddAnchor(IDCANCEL,BOTTOM_RIGHT);
\r
73 CShadowTree* CShadowTree::GetNextSub(CString& nameLeft, bool bCreateIfNotExist)
\r
75 int posSlash=nameLeft.Find('/');
\r
80 nameLeft.Empty();//Nothing left
\r
84 nameSub=nameLeft.Left(posSlash);
\r
85 nameLeft=nameLeft.Mid(posSlash+1);
\r
87 if(nameSub.IsEmpty())
\r
90 if(!bCreateIfNotExist && m_ShadowTree.find(nameSub)==m_ShadowTree.end())
\r
93 CShadowTree& nextNode=m_ShadowTree[nameSub];
\r
94 nextNode.m_csRefName=nameSub;
\r
95 nextNode.m_pParent=this;
\r
99 typedef std::map<CString,CString> MAP_STRING_STRING;
\r
101 void CBrowseRefsDlg::Refresh(bool bSelectCurHead)
\r
103 // m_RefMap.clear();
\r
104 // g_Git.GetMapHashToFriendName(m_RefMap);
\r
109 g_Git.Run(L"git symbolic-ref HEAD",&selectRef,CP_UTF8);
\r
110 selectRef.Trim(L"\r\n\t ");
\r
114 POSITION pos=m_ListRefLeafs.GetFirstSelectedItemPosition();
\r
115 //List ctrl selection?
\r
118 CShadowTree* pTree=(CShadowTree*)m_ListRefLeafs.GetItemData(
\r
119 m_ListRefLeafs.GetNextSelectedItem(pos));
\r
120 selectRef=pTree->GetRefName();
\r
124 //Tree ctrl selection?
\r
125 HTREEITEM hTree=m_RefTreeCtrl.GetSelectedItem();
\r
128 CShadowTree* pTree=(CShadowTree*)m_RefTreeCtrl.GetItemData(hTree);
\r
129 selectRef=pTree->GetRefName();
\r
134 m_RefTreeCtrl.DeleteAllItems();
\r
135 m_ListRefLeafs.DeleteAllItems();
\r
136 m_TreeRoot.m_ShadowTree.clear();
\r
137 m_TreeRoot.m_csRefName="refs";
\r
138 // m_TreeRoot.m_csShowName="Refs";
\r
139 m_TreeRoot.m_hTree=m_RefTreeCtrl.InsertItem(L"Refs",NULL,NULL);
\r
140 m_RefTreeCtrl.SetItemData(m_TreeRoot.m_hTree,(DWORD_PTR)&m_TreeRoot);
\r
143 g_Git.Run(L"git for-each-ref --format="
\r
145 L"%(objectname)%04"
\r
146 L"%(authordate:relative)%04"
\r
154 MAP_STRING_STRING refMap;
\r
156 //First sort on ref name
\r
157 while(!(singleRef=allRefs.Tokenize(L"\r\n",linePos)).IsEmpty())
\r
160 CString refName=singleRef.Tokenize(L"\04",valuePos);
\r
161 CString refRest=singleRef.Mid(valuePos);
\r
162 refMap[refName]=refRest;
\r
167 // for(MAP_HASH_NAME::iterator iterRef=m_RefMap.begin();iterRef!=m_RefMap.end();++iterRef)
\r
168 // for(STRING_VECTOR::iterator iterRefName=iterRef->second.begin();iterRefName!=iterRef->second.end();++iterRefName)
\r
169 // refName[*iterRefName]=iterRef->first;
\r
171 //Populate ref tree
\r
172 for(MAP_STRING_STRING::iterator iterRefMap=refMap.begin();iterRefMap!=refMap.end();++iterRefMap)
\r
174 CShadowTree& treeLeaf=GetTreeNode(iterRefMap->first,NULL,true);
\r
175 CString values=iterRefMap->second;
\r
178 treeLeaf.m_csRefHash= values.Tokenize(L"\04",valuePos);
\r
179 treeLeaf.m_csDate= values.Tokenize(L"\04",valuePos);
\r
180 treeLeaf.m_csSubject= values.Tokenize(L"\04",valuePos);
\r
181 treeLeaf.m_csAuthor= values.Tokenize(L"\04",valuePos);
\r
185 if(selectRef.IsEmpty() || !SelectRef(selectRef))
\r
186 //Probably not on a branch. Select root node.
\r
187 m_RefTreeCtrl.Expand(m_TreeRoot.m_hTree,TVE_EXPAND);
\r
191 bool CBrowseRefsDlg::SelectRef(CString refName)
\r
193 if(wcsnicmp(refName,L"refs/",5)!=0)
\r
194 return false; // Not a ref name
\r
196 CShadowTree& treeLeafHead=GetTreeNode(refName,NULL,false);
\r
197 if(treeLeafHead.m_hTree != NULL)
\r
199 //Not a leaf. Select tree node and return
\r
200 m_RefTreeCtrl.Select(treeLeafHead.m_hTree,TVGN_CARET);
\r
204 if(treeLeafHead.m_pParent==NULL)
\r
205 return false; //Weird... should not occur.
\r
207 //This is the current head.
\r
208 m_RefTreeCtrl.Select(treeLeafHead.m_pParent->m_hTree,TVGN_CARET);
\r
210 for(int indexPos = 0; indexPos < m_ListRefLeafs.GetItemCount(); ++indexPos)
\r
212 CShadowTree* pCurrShadowTree = (CShadowTree*)m_ListRefLeafs.GetItemData(indexPos);
\r
213 if(pCurrShadowTree == &treeLeafHead)
\r
215 m_ListRefLeafs.SetItemState(indexPos,LVIS_SELECTED,LVIS_SELECTED);
\r
216 m_ListRefLeafs.EnsureVisible(indexPos,FALSE);
\r
223 CShadowTree& CBrowseRefsDlg::GetTreeNode(CString refName, CShadowTree* pTreePos, bool bCreateIfNotExist)
\r
227 if(wcsnicmp(refName,L"refs/",5)==0)
\r
228 refName=refName.Mid(5);
\r
229 pTreePos=&m_TreeRoot;
\r
231 if(refName.IsEmpty())
\r
232 return *pTreePos;//Found leaf
\r
234 CShadowTree* pNextTree=pTreePos->GetNextSub(refName,bCreateIfNotExist);
\r
235 if(pNextTree==NULL)
\r
237 //Should not occur when all ref-names are valid and bCreateIfNotExist is true.
\r
238 ASSERT(!bCreateIfNotExist);
\r
242 if(!refName.IsEmpty())
\r
244 //When the refName is not empty, this node is not a leaf, so lets add it to the tree control.
\r
245 //Leafs are for the list control.
\r
246 if(pNextTree->m_hTree==NULL)
\r
248 //New tree. Create node in control.
\r
249 pNextTree->m_hTree=m_RefTreeCtrl.InsertItem(pNextTree->m_csRefName,pTreePos->m_hTree,NULL);
\r
250 m_RefTreeCtrl.SetItemData(pNextTree->m_hTree,(DWORD_PTR)pNextTree);
\r
254 return GetTreeNode(refName, pNextTree, bCreateIfNotExist);
\r
258 void CBrowseRefsDlg::OnTvnSelchangedTreeRef(NMHDR *pNMHDR, LRESULT *pResult)
\r
260 LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);
\r
263 FillListCtrlForTreeNode(pNMTreeView->itemNew.hItem);
\r
266 void CBrowseRefsDlg::FillListCtrlForTreeNode(HTREEITEM treeNode)
\r
268 m_ListRefLeafs.DeleteAllItems();
\r
270 CShadowTree* pTree=(CShadowTree*)(m_RefTreeCtrl.GetItemData(treeNode));
\r
276 FillListCtrlForShadowTree(pTree,L"",true);
\r
279 void CBrowseRefsDlg::FillListCtrlForShadowTree(CShadowTree* pTree, CString refNamePrefix, bool isFirstLevel)
\r
281 if(pTree->IsLeaf())
\r
283 int indexItem=m_ListRefLeafs.InsertItem(m_ListRefLeafs.GetItemCount(),L"");
\r
285 m_ListRefLeafs.SetItemData(indexItem,(DWORD_PTR)pTree);
\r
286 m_ListRefLeafs.SetItemText(indexItem,0,refNamePrefix+pTree->m_csRefName);
\r
287 m_ListRefLeafs.SetItemText(indexItem,1,pTree->m_csDate);
\r
288 m_ListRefLeafs.SetItemText(indexItem,2,pTree->m_csSubject);
\r
289 m_ListRefLeafs.SetItemText(indexItem,3,pTree->m_csRefHash);
\r
294 CString csThisName;
\r
296 csThisName=refNamePrefix+pTree->m_csRefName+L"/";
\r
297 for(CShadowTree::TShadowTreeMap::iterator itSubTree=pTree->m_ShadowTree.begin(); itSubTree!=pTree->m_ShadowTree.end(); ++itSubTree)
\r
299 FillListCtrlForShadowTree(&itSubTree->second,csThisName,false);
\r
304 void CBrowseRefsDlg::OnContextMenu_ListRefLeafs(CPoint point)
\r
307 CPoint clientPoint=point;
\r
308 m_RefTreeCtrl.ScreenToClient(&clientPoint);
\r
311 std::vector<CShadowTree*> selectedTrees;
\r
312 selectedTrees.reserve(m_ListRefLeafs.GetSelectedCount());
\r
313 POSITION pos=m_ListRefLeafs.GetFirstSelectedItemPosition();
\r
316 selectedTrees.push_back(
\r
317 (CShadowTree*)m_ListRefLeafs.GetItemData(
\r
318 m_ListRefLeafs.GetNextSelectedItem(pos)));
\r
322 popupMenu.CreatePopupMenu();
\r
324 if(selectedTrees.size()==1)
\r
326 popupMenu.AppendMenu(MF_STRING,eCmd_ViewLog,L"View log");
\r
327 if(selectedTrees[0]->IsFrom(L"refs/heads"))
\r
328 popupMenu.AppendMenu(MF_STRING,eCmd_DeleteBranch,L"Delete Branch");
\r
329 else if(selectedTrees[0]->IsFrom(L"refs/tags"))
\r
330 popupMenu.AppendMenu(MF_STRING,eCmd_DeleteTag,L"Delete Tag");
\r
332 // CShadowTree* pTree = (CShadowTree*)m_ListRefLeafs.GetItemData(pNMHDR->idFrom);
\r
338 eCmd cmd=(eCmd)popupMenu.TrackPopupMenuEx(TPM_LEFTALIGN|TPM_RETURNCMD, point.x, point.y, this, 0);
\r
344 dlg.SetStartRef(selectedTrees[0]->m_csRefHash);
\r
348 case eCmd_DeleteBranch:
\r
350 if(ConfirmDeleteRef(selectedTrees[0]->GetRefName()))
\r
351 DoDeleteRef(selectedTrees[0]->GetRefName(), true);
\r
355 case eCmd_DeleteTag:
\r
357 if(ConfirmDeleteRef(selectedTrees[0]->GetRefName()))
\r
358 DoDeleteRef(selectedTrees[0]->GetRefName(), true);
\r
365 bool CBrowseRefsDlg::ConfirmDeleteRef(CString completeRefName)
\r
370 UINT mbIcon=MB_ICONQUESTION;
\r
371 csMessage=L"Are you sure you want to delete the ";
\r
372 if(wcsncmp(completeRefName,L"refs/heads",10)==0)
\r
374 CString branchToDelete = completeRefName.Mid(11);
\r
375 csTitle.Format(L"Confirm deletion of branch %s", branchToDelete);
\r
376 csMessage += "branch:\r\n\r\n<b>";
\r
377 csMessage += branchToDelete;
\r
378 csMessage += "</b>";
\r
380 //Check if branch is fully merged in HEAD
\r
381 CString branchHash = g_Git.GetHash(completeRefName);
\r
382 CString commonAncestor;
\r
384 cmd.Format(L"git.exe merge-base HEAD %s",completeRefName);
\r
385 g_Git.Run(cmd,&commonAncestor,CP_UTF8);
\r
387 branchHash=branchHash.Left(40);
\r
388 commonAncestor=commonAncestor.Left(40);
\r
390 if(commonAncestor != branchHash)
\r
392 csMessage += L"\r\n\r\n<b>Warning:\r\nThis branch is not fully merged into HEAD.</b>";
\r
393 mbIcon=MB_ICONWARNING;
\r
396 else if(wcsncmp(completeRefName,L"refs/tags",9)==0)
\r
398 CString tagToDelete = completeRefName.Mid(10);
\r
399 csTitle.Format(L"Confirm deletion of tag %s", tagToDelete);
\r
400 csMessage += "tag:\r\n\r\n<b>";
\r
401 csMessage += tagToDelete;
\r
402 csMessage += "</b>";
\r
405 return CMessageBox::Show(m_hWnd,csMessage,csTitle,MB_YESNO|mbIcon)==IDYES;
\r
410 bool CBrowseRefsDlg::DoDeleteRef(CString completeRefName, bool bForce)
\r
412 if(wcsncmp(completeRefName,L"refs/heads",10)==0)
\r
414 CString branchToDelete = completeRefName.Mid(11);
\r
416 cmd.Format(L"git.exe branch -%c %s",bForce?L'D':L'd',branchToDelete);
\r
417 CString resultDummy;
\r
418 if(g_Git.Run(cmd,&resultDummy,CP_UTF8)!=0)
\r
421 errorMsg.Format(L"Could not delete branch %s. Message from git:\r\n\r\n%s",branchToDelete,resultDummy);
\r
422 CMessageBox::Show(m_hWnd,errorMsg,L"Error deleting branch",MB_OK|MB_ICONERROR);
\r
426 else if(wcsncmp(completeRefName,L"refs/tags",9)==0)
\r
428 CString tagToDelete = completeRefName.Mid(10);
\r
430 cmd.Format(L"git.exe tag -d %s",tagToDelete);
\r
431 CString resultDummy;
\r
432 if(g_Git.Run(cmd,&resultDummy,CP_UTF8)!=0)
\r
435 errorMsg.Format(L"Could not delete tag %s. Message from git:\r\n\r\n%s",tagToDelete,resultDummy);
\r
436 CMessageBox::Show(m_hWnd,errorMsg,L"Error deleting tag",MB_OK|MB_ICONERROR);
\r
443 void CBrowseRefsDlg::OnContextMenu(CWnd* pWndFrom, CPoint point)
\r
445 if(pWndFrom==&m_RefTreeCtrl) OnContextMenu_RefTreeCtrl(point);
\r
446 else if(pWndFrom==&m_ListRefLeafs) OnContextMenu_ListRefLeafs(point);
\r
449 void CBrowseRefsDlg::OnContextMenu_RefTreeCtrl(CPoint point)
\r
452 popupMenu.CreatePopupMenu();
\r
454 CPoint clientPoint=point;
\r
455 m_RefTreeCtrl.ScreenToClient(&clientPoint);
\r
457 HTREEITEM hTreeItem=m_RefTreeCtrl.HitTest(clientPoint);
\r
458 if(hTreeItem!=NULL)
\r
460 m_RefTreeCtrl.Select(hTreeItem,TVGN_CARET);
\r
461 CShadowTree* pTree=(CShadowTree*)m_RefTreeCtrl.GetItemData(hTreeItem);
\r
462 if(pTree->IsFrom(L"refs/remotes"))
\r
464 // popupMenu.AppendMenu(MF_STRING,eCmd_AddRemote,L"Add Remote");
\r
465 popupMenu.AppendMenu(MF_STRING,eCmd_ManageRemotes,L"Manage Remotes");
\r
467 else if(pTree->IsFrom(L"refs/heads"))
\r
468 popupMenu.AppendMenu(MF_STRING,eCmd_CreateBranch,L"Create Branch");
\r
469 else if(pTree->IsFrom(L"refs/tags"))
\r
470 popupMenu.AppendMenu(MF_STRING,eCmd_CreateTag,L"Create Tag");
\r
473 eCmd cmd=(eCmd)popupMenu.TrackPopupMenuEx(TPM_LEFTALIGN|TPM_RETURNCMD, point.x, point.y, this, 0);
\r
476 case eCmd_AddRemote:
\r
478 CAddRemoteDlg(this).DoModal();
\r
482 case eCmd_ManageRemotes:
\r
484 CSinglePropSheetDlg(L"Git Remote Settings",new CSettingGitRemote(m_cmdPath),this).DoModal();
\r
485 // CSettingGitRemote W_Remotes(m_cmdPath);
\r
486 // W_Remotes.DoModal();
\r
490 case eCmd_CreateBranch:
\r
492 CCreateBranchTagDlg dlg(this);
\r
493 dlg.m_bIsTag=false;
\r
498 case eCmd_CreateTag:
\r
500 CCreateBranchTagDlg dlg(this);
\r
509 BOOL CBrowseRefsDlg::PreTranslateMessage(MSG* pMsg)
\r
511 if (pMsg->message == WM_KEYDOWN)
\r
513 switch (pMsg->wParam)
\r
517 if (GetAsyncKeyState(VK_CONTROL)&0x8000)
\r
519 if ( GetDlgItem(IDOK)->IsWindowEnabled() )
\r
521 PostMessage(WM_COMMAND, IDOK);
\r
536 return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
\r