src/TortoiseProc/ChooseVersion.h

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