src/TortoiseProc/ChooseVersion.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2008-2014 - 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 "afxwin.h"
  21 #include "LogDlg.h"
  22 #include "BrowseRefsDlg.h"
  23 #include "MessageBox.h"
  24
  25 static UINT WM_GUIUPDATES = RegisterWindowMessage(_T("TORTOISEGIT_CHOOSEVERSION_GUIUPDATES"));
  26
  27 class CChooseVersion
  28 {
  29 public:
  30         CString m_initialRefName;
  31
  32 private:
  33         CWnd *  m_pWin;
  34         CWinThread*                     m_pLoadingThread;
  35         static UINT LoadingThreadEntry(LPVOID pVoid)
  36         {
  37                 return ((CChooseVersion*)pVoid)->LoadingThread();
  38         };
  39         volatile LONG           m_bLoadingThreadRunning;
  40
  41 protected:
  42         CHistoryCombo   m_ChooseVersioinBranch;
  43         CHistoryCombo   m_ChooseVersioinTags;
  44         CHistoryCombo   m_ChooseVersioinVersion;
  45         CButton                 m_RadioBranch;
  46         CButton                 m_RadioTag;
  47         CString                 m_pendingRefName;
  48         bool                    m_bNotFullName;
  49
  50         //Notification when version changed. Can be implemented in derived classes.
  51         virtual void OnVersionChanged(){}
  52
  53         afx_msg void OnBnClickedChooseRadio()
  54         {
  55                 this->m_ChooseVersioinTags.EnableWindow(FALSE);
  56                 this->m_ChooseVersioinBranch.EnableWindow(FALSE);
  57                 this->m_ChooseVersioinVersion.EnableWindow(FALSE);
  58                 m_pWin->GetDlgItem(IDC_BUTTON_BROWSE_REF)->EnableWindow(FALSE);
  59                 m_pWin->GetDlgItem(IDC_BUTTON_SHOW)->EnableWindow(FALSE);
  60                 m_bIsBranch = false;
  61                 int radio=m_pWin->GetCheckedRadioButton(IDC_RADIO_HEAD,IDC_RADIO_VERSION);
  62                 switch (radio)
  63                 {
  64                 case IDC_RADIO_HEAD:
  65                         break;
  66                 case IDC_RADIO_BRANCH:
  67                         this->m_ChooseVersioinBranch.EnableWindow(TRUE);
  68                         m_pWin->GetDlgItem(IDC_BUTTON_BROWSE_REF)->EnableWindow(TRUE);
  69                         m_bIsBranch = true;
  70                         break;
  71                 case IDC_RADIO_TAGS:
  72                         this->m_ChooseVersioinTags.EnableWindow(TRUE);
  73                         break;
  74                 case IDC_RADIO_VERSION:
  75                         this->m_ChooseVersioinVersion.EnableWindow(TRUE);
  76                         m_pWin->GetDlgItem(IDC_BUTTON_SHOW)->EnableWindow(TRUE);
  77                 break;
  78                 }
  79                 // enable version browse button if Version is selected
  80                 m_pWin->GetDlgItem(IDC_BUTTON_SHOW)->EnableWindow(radio == IDC_RADIO_VERSION);
  81                 OnVersionChanged();
  82         }
  83
  84         void OnBnClickedChooseVersion()
  85         {
  86                 // use the git log to allow selection of a version
  87                 CLogDlg dlg;
  88                 // tell the dialog to use mode for selecting revisions
  89                 dlg.SetSelect(true);
  90                 // only one revision must be selected however
  91                 dlg.SingleSelection(true);
  92                 if ( dlg.DoModal() == IDOK )
  93                 {
  94                         // get selected hash if any
  95                         CString selectedHash = dlg.GetSelectedHash();
  96                         // load into window, do this even if empty so that it is clear that nothing has been selected
  97                         m_ChooseVersioinVersion.SetWindowText( selectedHash );
  98                         OnVersionChanged();
  99                 }
 100         }
 101
 102         void UpdateRevsionName()
 103         {
 104                 int radio=m_pWin->GetCheckedRadioButton(IDC_RADIO_HEAD,IDC_RADIO_VERSION);
 105                 switch (radio)
 106                 {
 107                 case IDC_RADIO_HEAD:
 108                         this->m_VersionName=_T("HEAD");
 109                         break;
 110                 case IDC_RADIO_BRANCH:
 111                         this->m_VersionName=m_ChooseVersioinBranch.GetString();
 112                         if (!m_VersionName.IsEmpty() && !g_Git.IsBranchTagNameUnique(this->m_VersionName))
 113                                 this->m_VersionName = L"refs/heads/" + this->m_VersionName;
 114                         break;
 115                 case IDC_RADIO_TAGS:
 116                         this->m_VersionName = m_ChooseVersioinTags.GetString();
 117                         if (!m_VersionName.IsEmpty() && !g_Git.IsBranchTagNameUnique(this->m_VersionName))
 118                                 this->m_VersionName = L"refs/tags/" + m_ChooseVersioinTags.GetString();
 119                         break;
 120                 case IDC_RADIO_VERSION:
 121                         this->m_VersionName=m_ChooseVersioinVersion.GetString();
 122                         break;
 123                 }
 124         }
 125         void SetDefaultChoose(int id)
 126         {
 127                 m_pWin->CheckRadioButton(IDC_RADIO_HEAD,IDC_RADIO_VERSION,id);
 128                 OnBnClickedChooseRadio();
 129         }
 130
 131         void OnBnClickedButtonBrowseRef()
 132         {
 133                 CString origRef;
 134                 UpdateRevsionName();
 135                 CString resultRef = CBrowseRefsDlg::PickRef(false, m_VersionName, gPickRef_All);
 136                 if(resultRef.IsEmpty())
 137                 {
 138                         m_pendingRefName = m_VersionName;
 139                         m_bNotFullName = true;
 140                         InitChooseVersion(false, true);
 141                         return;
 142                 }
 143                 m_pendingRefName = resultRef;
 144                 m_bNotFullName = false;
 145                 InitChooseVersion(false, true);
 146         }
 147
 148         void SelectRef(CString refName, bool bRefNameIsPossiblyNotFullName = true)
 149         {
 150                 if(bRefNameIsPossiblyNotFullName)
 151                 {
 152                         //Make sure refName is a full ref name first
 153                         CString fullRefName = g_Git.GetFullRefName(refName);
 154                         if(!fullRefName.IsEmpty())
 155                                 refName = fullRefName;
 156                 }
 157
 158                 if(wcsncmp(refName,L"refs/",5)==0)
 159                         refName = refName.Mid(5);
 160                 if(wcsncmp(refName,L"heads/",6)==0)
 161                 {
 162                         refName = refName.Mid(6);
 163                         SetDefaultChoose(IDC_RADIO_BRANCH);
 164                         m_ChooseVersioinBranch.SetCurSel(
 165                                 m_ChooseVersioinBranch.FindStringExact(-1, refName));
 166                 }
 167                 else if(wcsncmp(refName,L"remotes/",8)==0)
 168                 {
 169                         SetDefaultChoose(IDC_RADIO_BRANCH);
 170                         m_ChooseVersioinBranch.SetCurSel(
 171                                 m_ChooseVersioinBranch.FindStringExact(-1, refName));
 172                 }
 173                 else if(wcsncmp(refName,L"tags/",5)==0)
 174                 {
 175                         refName = refName.Mid(5);
 176                         refName.Replace(_T("^{}"), _T(""));
 177                         SetDefaultChoose(IDC_RADIO_TAGS);
 178                         m_ChooseVersioinTags.SetCurSel(
 179                                 m_ChooseVersioinTags.FindStringExact(-1, refName));
 180                 }
 181                 else
 182                 {
 183                         SetDefaultChoose(IDC_RADIO_VERSION);
 184                         m_ChooseVersioinVersion.AddString(refName);
 185                 }
 186                 OnVersionChanged();
 187         }
 188
 189         UINT LoadingThread()
 190         {
 191                 STRING_VECTOR list;
 192
 193                 int current = -1;
 194                 g_Git.GetBranchList(list,&current,CGit::BRANCH_ALL_F);
 195                 for (int i = m_ChooseVersioinBranch.GetCount(); i >= 0; --i)
 196                         m_ChooseVersioinBranch.DeleteString(i);
 197                 m_ChooseVersioinBranch.AddString(list, false);
 198                 m_ChooseVersioinBranch.SetCurSel(current);
 199
 200                 list.clear();
 201                 g_Git.GetTagList(list);
 202                 for (int i = m_ChooseVersioinTags.GetCount(); i >= 0; --i)
 203                         m_ChooseVersioinTags.DeleteString(i);
 204                 m_ChooseVersioinTags.AddString(list, false);
 205                 m_ChooseVersioinTags.SetCurSel(0);
 206
 207                 m_pWin->SendMessage(WM_GUIUPDATES);
 208
 209                 InterlockedExchange(&m_bLoadingThreadRunning, FALSE);
 210                 return 0;
 211         }
 212         void UpdateGUI()
 213         {
 214                 m_RadioBranch.EnableWindow(TRUE);
 215                 m_RadioTag.EnableWindow(TRUE);
 216
 217                 if (m_pendingRefName.IsEmpty())
 218                         OnVersionChanged();
 219                 else
 220                         SelectRef(m_pendingRefName, m_bNotFullName);
 221
 222                 if (m_bIsFirstTimeToSetFocus && m_pWin->GetDlgItem(IDC_COMBOBOXEX_BRANCH)->IsWindowEnabled())
 223                         m_pWin->GetDlgItem(IDC_COMBOBOXEX_BRANCH)->SetFocus();
 224                 m_bIsFirstTimeToSetFocus = false;
 225         }
 226         void InitChooseVersion(bool setFocusToBranchComboBox = false, bool bReInit = false)
 227         {
 228                 m_ChooseVersioinBranch.SetMaxHistoryItems(0x7FFFFFFF);
 229                 m_ChooseVersioinTags.SetMaxHistoryItems(0x7FFFFFFF);
 230
 231                 m_bIsBranch = false;
 232                 m_RadioBranch.EnableWindow(FALSE);
 233                 m_RadioTag.EnableWindow(FALSE);
 234
 235                 m_bIsFirstTimeToSetFocus = setFocusToBranchComboBox;
 236                 if (!bReInit)
 237                 {
 238                         m_pendingRefName = m_initialRefName;
 239                         m_bNotFullName = true;
 240                 }
 241
 242                 InterlockedExchange(&m_bLoadingThreadRunning, TRUE);
 243
 244                 if ( (m_pLoadingThread=AfxBeginThread(LoadingThreadEntry, this)) ==NULL)
 245                 {
 246                         InterlockedExchange(&m_bLoadingThreadRunning, FALSE);
 247                         CMessageBox::Show(NULL, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
 248                 }
 249         }
 250         void WaitForFinishLoading()
 251         {
 252                 if(m_bLoadingThreadRunning && m_pLoadingThread)
 253                 {
 254                         DWORD ret =::WaitForSingleObject(m_pLoadingThread->m_hThread,20000);
 255                         if(ret == WAIT_TIMEOUT)
 256                                 ::TerminateThread(m_pLoadingThread,0);
 257                 }
 258         }
 259 public:
 260         CString m_VersionName;
 261         bool    m_bIsBranch;
 262         bool    m_bIsFirstTimeToSetFocus;
 263         CChooseVersion(CWnd *win)
 264         : m_bIsBranch(false)
 265         , m_bIsFirstTimeToSetFocus(false)
 266         , m_pLoadingThread(nullptr)
 267         , m_bLoadingThreadRunning(FALSE)
 268         , m_bNotFullName(true)
 269         {
 270                 m_pWin=win;
 271         };
 272
 273 };
 274
 275 #define CHOOSE_VERSION_DDX \
 276         DDX_Control(pDX, IDC_COMBOBOXEX_BRANCH,         m_ChooseVersioinBranch); \
 277         DDX_Control(pDX, IDC_COMBOBOXEX_TAGS,           m_ChooseVersioinTags); \
 278         DDX_Control(pDX, IDC_COMBOBOXEX_VERSION,        m_ChooseVersioinVersion); \
 279         DDX_Control(pDX, IDC_RADIO_BRANCH, m_RadioBranch);\
 280         DDX_Control(pDX, IDC_RADIO_TAGS, m_RadioTag);
 281
 282 #define CHOOSE_VERSION_EVENT\
 283         ON_REGISTERED_MESSAGE(WM_GUIUPDATES,    OnUpdateGUIHost)\
 284         ON_BN_CLICKED(IDC_RADIO_HEAD,                   OnBnClickedChooseRadioHost)\
 285         ON_BN_CLICKED(IDC_RADIO_BRANCH,                 OnBnClickedChooseRadioHost)\
 286         ON_BN_CLICKED(IDC_RADIO_TAGS,                   OnBnClickedChooseRadioHost)\
 287         ON_BN_CLICKED(IDC_BUTTON_SHOW,                  OnBnClickedShow)\
 288         ON_BN_CLICKED(IDC_RADIO_VERSION,                OnBnClickedChooseRadioHost)\
 289         ON_BN_CLICKED(IDC_BUTTON_BROWSE_REF,    OnBnClickedButtonBrowseRefHost)
 290
 291 #define CHOOSE_VERSION_ADDANCHOR                                                                \
 292         {                                                                                                                       \
 293                 AddAnchor(IDC_COMBOBOXEX_BRANCH, TOP_LEFT, TOP_RIGHT);  \
 294                 AddAnchor(IDC_COMBOBOXEX_TAGS, TOP_LEFT, TOP_RIGHT);    \
 295                 AddAnchor(IDC_COMBOBOXEX_VERSION, TOP_LEFT, TOP_RIGHT); \
 296                 AddAnchor(IDC_GROUP_BASEON, TOP_LEFT, TOP_RIGHT);               \
 297                 AddAnchor(IDC_BUTTON_SHOW,TOP_RIGHT);                                   \
 298                 AddAnchor(IDC_BUTTON_BROWSE_REF,TOP_RIGHT);                             \
 299         }
 300
 301 #define CHOOSE_EVENT_RADIO() \
 302         LRESULT OnUpdateGUIHost(WPARAM, LPARAM) { UpdateGUI(); return 0; } \
 303         afx_msg void OnBnClickedChooseRadioHost(){OnBnClickedChooseRadio();}\
 304         afx_msg void OnBnClickedShow(){OnBnClickedChooseVersion();}\
 305         afx_msg void OnBnClickedButtonBrowseRefHost(){OnBnClickedButtonBrowseRef();}