src/TortoiseProc/RevisionGraph/RevGraphFilterDlg.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2006 - Stefan Kueng
   4 // Copyright (C) 2012-2016, 2018 - TortoiseGit
   5
   6 // This program is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU General Public License
   8 // as published by the Free Software Foundation; either version 2
   9 // of the License, or (at your option) any later version.
  10
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License
  17 // along with this program; if not, write to the Free Software Foundation,
  18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 //
  20 #include "stdafx.h"
  21 #include "TortoiseProc.h"
  22 #include "RevGraphFilterDlg.h"
  23 #include "gittype.h"
  24 #include "Git.h"
  25 #include "BrowseRefsDlg.h"
  26 #include "StringUtils.h"
  27
  28 IMPLEMENT_DYNAMIC(CRevGraphFilterDlg, CDialog)
  29
  30 CRevGraphFilterDlg::CRevGraphFilterDlg(CWnd* pParent /*=nullptr*/)
  31         : CDialog(CRevGraphFilterDlg::IDD, pParent)
  32         , m_bCurrentBranch(FALSE)
  33         , m_bLocalBranches(FALSE)
  34 {
  35 }
  36
  37 CRevGraphFilterDlg::~CRevGraphFilterDlg()
  38 {
  39 }
  40
  41 void CRevGraphFilterDlg::DoDataExchange(CDataExchange* pDX)
  42 {
  43         CDialog::DoDataExchange(pDX);
  44         DDX_Text(pDX, IDC_FROMREV, m_sFromRev);
  45         DDX_Text(pDX, IDC_TOREV, m_sToRev);
  46         DDX_Check(pDX, IDC_CURRENT_BRANCH, m_bCurrentBranch);
  47         DDX_Check(pDX, IDC_LOCAL_BRANCHES, m_bLocalBranches);
  48         DDX_Control(pDX, IDC_FROMREV, m_ctrlFromRev);
  49         DDX_Control(pDX, IDC_TOREV, m_ctrlToRev);
  50 }
  51
  52 BEGIN_MESSAGE_MAP(CRevGraphFilterDlg, CDialog)
  53         ON_BN_CLICKED(IDC_REV1BTN1, &CRevGraphFilterDlg::OnBnClickedRev1btn1)
  54         ON_BN_CLICKED(IDC_REV1BTN2, &CRevGraphFilterDlg::OnBnClickedRev1btn2)
  55         ON_BN_CLICKED(IDC_RESETFILTER, &CRevGraphFilterDlg::OnBnClickedResetfilter)
  56         ON_BN_CLICKED(IDC_CURRENT_BRANCH, &CRevGraphFilterDlg::OnBnClickedCurrentBranch)
  57         ON_BN_CLICKED(IDC_LOCAL_BRANCHES, &CRevGraphFilterDlg::OnBnClickedLocalBranches)
  58 END_MESSAGE_MAP()
  59
  60 BOOL CRevGraphFilterDlg::OnInitDialog()
  61 {
  62         CDialog::OnInitDialog();
  63         this->m_ctrlFromRev.Init();
  64         this->m_ctrlToRev.Init();
  65
  66         STRING_VECTOR list;
  67         if (g_Git.GetRefList(list))
  68                 MessageBox(g_Git.GetGitLastErr(L"Could not get all refs."), L"TortoiseGit", MB_ICONERROR);
  69
  70         m_ctrlFromRev.AddSearchString(L"HEAD");
  71         m_ctrlToRev.AddSearchString(L"HEAD");
  72
  73         for (size_t i = 0; i < list.size(); ++i)
  74         {
  75                 CString str=list[i];
  76
  77                 m_ctrlFromRev.AddSearchString(list[i]);
  78                 m_ctrlToRev.AddSearchString(list[i]);
  79
  80                 if (CStringUtils::StartsWith(str, L"refs/"))
  81                 {
  82                         m_ctrlFromRev.AddSearchString(list[i].Mid((int)wcslen(L"refs/")));
  83                         m_ctrlToRev.AddSearchString(list[i].Mid((int)wcslen(L"refs/")));
  84                 }
  85
  86                 if (CStringUtils::StartsWith(str, L"refs/heads/"))
  87                 {
  88                         m_ctrlFromRev.AddSearchString(list[i].Mid((int)wcslen(L"refs/heads/")));
  89                         m_ctrlToRev.AddSearchString(list[i].Mid((int)wcslen(L"refs/heads/")));
  90                 }
  91
  92                 if (CStringUtils::StartsWith(str, L"refs/remotes/"))
  93                 {
  94                         m_ctrlFromRev.AddSearchString(list[i].Mid((int)wcslen(L"refs/remotes/")));
  95                         m_ctrlToRev.AddSearchString(list[i].Mid((int)wcslen(L"refs/remotes/")));
  96                 }
  97
  98                 if (CStringUtils::StartsWith(str, L"refs/tags/"))
  99                 {
 100                         m_ctrlFromRev.AddSearchString(list[i].Mid((int)wcslen(L"refs/tags/")));
 101                         m_ctrlToRev.AddSearchString(list[i].Mid((int)wcslen(L"refs/tags/")));
 102                 }
 103         }
 104
 105         OnBnClickedCurrentBranch();
 106         OnBnClickedLocalBranches();
 107
 108         return TRUE;
 109 }
 110
 111
 112 void CRevGraphFilterDlg::GetRevisionRange(CString& minrev, CString& maxrev)
 113 {
 114         minrev = m_sFromRev;
 115         maxrev = m_sToRev;
 116 }
 117
 118 void CRevGraphFilterDlg::SetRevisionRange (CString minrev, CString maxrev)
 119 {
 120         m_sFromRev = minrev;
 121         m_sToRev = maxrev;
 122 }
 123
 124 void CRevGraphFilterDlg::OnOK()
 125 {
 126         CDialog::OnOK();
 127 }
 128
 129 void CRevGraphFilterDlg::OnBnClickedRev1btn1()
 130 {
 131         CString str = CBrowseRefsDlg::PickRef();
 132         if(str.IsEmpty())
 133                 return;
 134
 135         m_ctrlFromRev.SetWindowText(str);
 136 }
 137
 138 void CRevGraphFilterDlg::OnBnClickedRev1btn2()
 139 {
 140         CString str = CBrowseRefsDlg::PickRef();
 141         if(str.IsEmpty())
 142                 return;
 143
 144         m_ctrlToRev.SetWindowText(str);
 145 }
 146
 147 void CRevGraphFilterDlg::OnBnClickedResetfilter()
 148 {
 149         m_sFromRev.Empty();
 150         m_sToRev.Empty();
 151         m_bCurrentBranch = FALSE;
 152         m_bLocalBranches = FALSE;
 153         UpdateData(FALSE);
 154         CDialog::OnOK();
 155 }
 156
 157 void CRevGraphFilterDlg::OnBnClickedCurrentBranch()
 158 {
 159         UpdateData();
 160         if (m_bCurrentBranch)
 161         {
 162                 m_bLocalBranches = FALSE;
 163                 GetDlgItem(IDC_LOCAL_BRANCHES)->EnableWindow(FALSE);
 164                 UpdateData(FALSE);
 165         }
 166         else
 167                 GetDlgItem(IDC_LOCAL_BRANCHES)->EnableWindow(TRUE);
 168 }
 169
 170 void CRevGraphFilterDlg::OnBnClickedLocalBranches()
 171 {
 172         UpdateData();
 173         if (m_bLocalBranches)
 174         {
 175                 m_bCurrentBranch = FALSE;
 176                 GetDlgItem(IDC_CURRENT_BRANCH)->EnableWindow(FALSE);
 177                 UpdateData(FALSE);
 178         }
 179         else
 180                 GetDlgItem(IDC_CURRENT_BRANCH)->EnableWindow(TRUE);
 181 }