src/TortoiseProc/BranchCombox.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2009-2015 - TortoiseGit
   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 #pragma once
  20 #include "LogDlg.h"
  21 #include "BrowseRefsDlg.h"
  22 #include "HistoryCombo.h"
  23 #include "LoglistUtils.h"
  24 #include "UnicodeUtils.h"
  25 #include "Tooltip.h"
  26
  27 class CBranchCombox
  28 {
  29 public:
  30         CBranchCombox()
  31         {
  32                 m_LocalBranchFilter = gPickRef_Head;
  33                 m_RemoteBranchFilter = gPickRef_Remote;
  34                 m_DialogName=_T("sync");
  35                 m_pTooltip=NULL;
  36         }
  37 protected:
  38         CHistoryCombo m_ctrlLocalBranch;
  39         CHistoryCombo m_ctrlRemoteBranch;
  40         int m_LocalBranchFilter;
  41         int m_RemoteBranchFilter;
  42
  43         CToolTips *m_pTooltip;
  44
  45         CString m_DialogName;
  46
  47         CString m_RegKeyRemoteBranch;
  48
  49         void  CbnSelchangeLocalBranch()
  50         {
  51                 //Select pull-remote from current branch
  52                 CString pullRemote, pullBranch;
  53                 g_Git.GetRemoteTrackedBranch(m_ctrlLocalBranch.GetString(), pullRemote, pullBranch);
  54
  55                 this->SetRemote(pullRemote);
  56
  57                 CString defaultUpstream;
  58                 defaultUpstream.Format(L"remotes/%s/%s", pullRemote, pullBranch);
  59                 int found = m_ctrlRemoteBranch.FindStringExact(0, defaultUpstream);
  60                 if(found >= 0)
  61                 {
  62                         m_ctrlRemoteBranch.SetCurSel(found);
  63                 }
  64                 else if(!pullBranch.IsEmpty())
  65                 {
  66                         int index=m_ctrlRemoteBranch.FindStringExact(0,pullBranch);
  67                         if( index<0 )
  68                                 m_ctrlRemoteBranch.AddString(pullBranch);
  69                         else
  70                                 m_ctrlRemoteBranch.SetCurSel(index);
  71                 }
  72                 else
  73                 {
  74                         m_ctrlRemoteBranch.SetCurSel(-1);
  75                 }
  76
  77                 this->AddBranchToolTips(&this->m_ctrlLocalBranch,this->m_pTooltip);
  78
  79                 LocalBranchChange();
  80         };
  81         void  CbnSelchangeRemoteBranch()
  82         {
  83                 if(this->m_RegKeyRemoteBranch.IsEmpty())
  84                         this->AddBranchToolTips(&this->m_ctrlRemoteBranch,this->m_pTooltip);
  85
  86                 RemoteBranchChange();
  87         }
  88         void  BnClickedButtonBrowseLocalBranch()
  89         {
  90                 if(CBrowseRefsDlg::PickRefForCombo(&m_ctrlLocalBranch, m_LocalBranchFilter))
  91                         CbnSelchangeLocalBranch();
  92         }
  93         void  BnClickedButtonBrowseRemoteBranch()
  94         {
  95                 if(!this->m_RegKeyRemoteBranch.IsEmpty())
  96                 {
  97                         CString remoteBranchName;
  98                         CString remoteName;
  99                         this->m_ctrlRemoteBranch.GetWindowText(remoteBranchName);
 100                         //remoteName = m_Remote.GetString();
 101                         //remoteBranchName = remoteName + '/' + remoteBranchName;
 102                         remoteBranchName = CBrowseRefsDlg::PickRef(false, remoteBranchName, gPickRef_Remote);
 103                         if(remoteBranchName.IsEmpty())
 104                                 return; //Canceled
 105
 106                         remoteBranchName = remoteBranchName.Mid(13);//Strip 'refs/remotes/'
 107                         int slashPlace = remoteBranchName.Find('/');
 108                         remoteName = remoteBranchName.Left(slashPlace);
 109                         remoteBranchName = remoteBranchName.Mid(slashPlace + 1); //Strip remote name (for example 'origin/')
 110
 111                         //Select remote
 112                         //int remoteSel = m_Remote.FindStringExact(0,remoteName);
 113                         //if(remoteSel >= 0)
 114                         //      m_Remote.SetCurSel(remoteSel);
 115                         this->SetRemote(remoteName);
 116
 117                         //Select branch
 118                         m_ctrlRemoteBranch.AddString(remoteBranchName);
 119                         CbnSelchangeRemoteBranch();
 120
 121                 }
 122                 else
 123                 {
 124                         if(CBrowseRefsDlg::PickRefForCombo(&m_ctrlRemoteBranch, m_RemoteBranchFilter))
 125                                 CbnSelchangeRemoteBranch();
 126                 }
 127         }
 128
 129         virtual void LocalBranchChange(){};
 130         virtual void RemoteBranchChange(){};
 131         virtual void SetRemote(CString remote){};
 132
 133         void AddBranchToolTips(CHistoryCombo *pBranch, CToolTips *tip)
 134         {
 135                 if(pBranch&&tip)
 136                 {
 137                         CString text=pBranch->GetString();
 138                         CString tooltip;
 139
 140                         if (text.IsEmpty())
 141                         {
 142                                 pBranch->DisableTooltip();
 143                                 return;
 144                         }
 145
 146                         GitRev rev;
 147                         if (rev.GetCommit(text))
 148                         {
 149                                 pBranch->DisableTooltip();
 150                                 MessageBox(nullptr, rev.GetLastErr(), _T("TortoiseGit"), MB_ICONERROR);
 151                                 return;
 152                         }
 153
 154                         tooltip.Format(_T("%s: %s\n%s: %s <%s>\n%s: %s\n%s:\n%s\n%s"),
 155                                 CString(MAKEINTRESOURCE(IDS_LOG_REVISION)),
 156                                 rev.m_CommitHash.ToString(),
 157                                 CString(MAKEINTRESOURCE(IDS_LOG_AUTHOR)),
 158                                 rev.GetAuthorName(),
 159                                 rev.GetAuthorEmail(),
 160                                 CString(MAKEINTRESOURCE(IDS_LOG_DATE)),
 161                                 CLoglistUtils::FormatDateAndTime(rev.GetAuthorDate(), DATE_LONGDATE),
 162                                 CString(MAKEINTRESOURCE(IDS_LOG_MESSAGE)),
 163                                 rev.GetSubject(),
 164                                 rev.GetBody());
 165
 166                         pBranch->DisableTooltip();
 167
 168                         tip->AddTool(pBranch->GetComboBoxCtrl(),tooltip);
 169                 }
 170         }
 171
 172         void LoadBranchInfo()
 173         {
 174                 m_ctrlLocalBranch.SetMaxHistoryItems(0x0FFFFFFF);
 175                 m_ctrlRemoteBranch.SetMaxHistoryItems(0x0FFFFFFF);
 176
 177                 STRING_VECTOR list;
 178                 list.clear();
 179                 int current=0;
 180
 181                 g_Git.GetBranchList(list,&current,CGit::BRANCH_LOCAL_F);
 182
 183                 m_ctrlLocalBranch.SetList(list);
 184
 185                 if(this->m_RegKeyRemoteBranch.IsEmpty())
 186                 {
 187                         list.clear();
 188                         g_Git.GetBranchList(list, NULL, CGit::BRANCH_REMOTE);
 189
 190                         m_ctrlRemoteBranch.SetList(list);
 191
 192                 }
 193                 else
 194                 {
 195                         m_ctrlRemoteBranch.Reset();
 196                         m_ctrlRemoteBranch.LoadHistory(m_RegKeyRemoteBranch,_T("sync"));
 197                 }
 198
 199                 if(!this->m_strLocalBranch.IsEmpty())
 200                 {
 201                         m_ctrlLocalBranch.AddString(m_strLocalBranch);
 202                 }
 203                 else
 204                 {
 205                         m_ctrlLocalBranch.SetCurSel(current);
 206                 }
 207
 208                 if(!m_strRemoteBranch.IsEmpty())
 209                 {
 210                         m_ctrlRemoteBranch.AddString(m_strRemoteBranch);
 211                         m_ctrlRemoteBranch.SetCurSel(m_ctrlRemoteBranch.GetCount()-1);
 212                 }
 213                 else
 214                 {
 215                         CbnSelchangeLocalBranch();
 216                 }
 217
 218                 this->LocalBranchChange();
 219                 this->RemoteBranchChange();
 220         }
 221
 222 public:
 223         CString m_strLocalBranch;
 224         CString m_strRemoteBranch;
 225
 226         void SaveHistory()
 227         {
 228                 if(!this->m_RegKeyRemoteBranch.IsEmpty())
 229                 {
 230                         m_ctrlRemoteBranch.AddString(m_strRemoteBranch);
 231                         m_ctrlRemoteBranch.SaveHistory();
 232                 }
 233         }
 234 };
 235
 236 #define BRANCH_COMBOX_DDX \
 237         DDX_Control(pDX, IDC_COMBOBOXEX_LOCAL_BRANCH,           m_ctrlLocalBranch);      \
 238         DDX_Control(pDX, IDC_COMBOBOXEX_REMOTE_BRANCH,          m_ctrlRemoteBranch);     \
 239
 240 #define BRANCH_COMBOX_EVENT \
 241         ON_CBN_SELCHANGE(IDC_COMBOBOXEX_LOCAL_BRANCH,   OnCbnSelchangeLocalBranch)      \
 242         ON_CBN_SELCHANGE(IDC_COMBOBOXEX_REMOTE_BRANCH,  OnCbnSelchangeRemoteBranch)     \
 243         ON_BN_CLICKED(IDC_BUTTON_LOCAL_BRANCH,                  OnBnClickedButtonBrowseLocalBranch) \
 244         ON_BN_CLICKED(IDC_BUTTON_REMOTE_BRANCH,                 OnBnClickedButtonBrowseRemoteBranch) \
 245
 246 #define BRANCH_COMBOX_ADD_ANCHOR() \
 247         AddAnchor(IDC_COMBOBOXEX_LOCAL_BRANCH,TOP_LEFT);\
 248         AddAnchor(IDC_COMBOBOXEX_REMOTE_BRANCH,TOP_RIGHT);\
 249         AddAnchor(IDC_BUTTON_LOCAL_BRANCH,TOP_LEFT);\
 250         AddAnchor(IDC_BUTTON_REMOTE_BRANCH,TOP_RIGHT);\
 251         AddAnchor(IDC_STATIC_REMOTE_BRANCH,TOP_RIGHT);
 252
 253 #define BRANCH_COMBOX_EVENT_HANDLE() \
 254         afx_msg void OnCbnSelchangeLocalBranch(){CbnSelchangeLocalBranch();} \
 255         afx_msg void OnCbnSelchangeRemoteBranch(){CbnSelchangeRemoteBranch();}\
 256         afx_msg void OnBnClickedButtonBrowseLocalBranch(){BnClickedButtonBrowseLocalBranch();}\
 257         afx_msg void OnBnClickedButtonBrowseRemoteBranch(){BnClickedButtonBrowseRemoteBranch();}