src/TortoiseProc/PullFetchDlg.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2008-2011 - 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
  20 // PullFetchDlg.cpp : implementation file
  21 //
  22
  23 #include "stdafx.h"
  24 #include "TortoiseProc.h"
  25 #include "PullFetchDlg.h"
  26 #include "Git.h"
  27 #include "AppUtils.h"
  28 #include "BrowseRefsDlg.h"
  29 // CPullFetchDlg dialog
  30
  31 IMPLEMENT_DYNAMIC(CPullFetchDlg, CResizableStandAloneDialog)
  32
  33 CPullFetchDlg::CPullFetchDlg(CWnd* pParent /*=NULL*/)
  34         : CResizableStandAloneDialog(CPullFetchDlg::IDD, pParent)
  35 {
  36         m_IsPull=TRUE;
  37         m_bAutoLoad = CAppUtils::IsSSHPutty();
  38         m_bAutoLoadEnable=CAppUtils::IsSSHPutty();;
  39         m_regRebase = false;
  40 }
  41
  42 CPullFetchDlg::~CPullFetchDlg()
  43 {
  44 }
  45
  46 void CPullFetchDlg::DoDataExchange(CDataExchange* pDX)
  47 {
  48         CDialog::DoDataExchange(pDX);
  49         DDX_Control(pDX, IDC_REMOTE_COMBO, this->m_Remote);
  50         DDX_Control(pDX, IDC_OTHER, this->m_Other);
  51         DDX_Control(pDX, IDC_REMOTE_BRANCH, this->m_RemoteBranch);
  52         DDX_Control(pDX,IDC_REMOTE_MANAGE, this->m_RemoteManage);
  53         DDX_Check(pDX,IDC_PUTTYKEY_AUTOLOAD,m_bAutoLoad);
  54         DDX_Check(pDX,IDC_CHECK_REBASE,m_bRebase);
  55         DDX_Check(pDX,IDC_CHECK_PRUNE,m_bPrune);
  56 }
  57
  58
  59 BEGIN_MESSAGE_MAP(CPullFetchDlg,CResizableStandAloneDialog )
  60         ON_BN_CLICKED(IDC_REMOTE_RD, &CPullFetchDlg::OnBnClickedRd)
  61         ON_BN_CLICKED(IDC_OTHER_RD, &CPullFetchDlg::OnBnClickedRd)
  62         ON_BN_CLICKED(IDOK, &CPullFetchDlg::OnBnClickedOk)
  63         ON_STN_CLICKED(IDC_REMOTE_MANAGE, &CPullFetchDlg::OnStnClickedRemoteManage)
  64         ON_BN_CLICKED(IDC_BUTTON_BROWSE_REF, &CPullFetchDlg::OnBnClickedButtonBrowseRef)
  65 END_MESSAGE_MAP()
  66
  67 BOOL CPullFetchDlg::OnInitDialog()
  68 {
  69         CResizableStandAloneDialog::OnInitDialog();
  70         AddAnchor(IDC_REMOTE_COMBO, TOP_LEFT, TOP_RIGHT);
  71         AddAnchor(IDC_OTHER, TOP_LEFT,TOP_RIGHT);
  72
  73         AddAnchor(IDC_REMOTE_BRANCH, TOP_LEFT,TOP_RIGHT);
  74         AddAnchor(IDC_BUTTON_BROWSE_REF,TOP_RIGHT);
  75
  76         AddAnchor(IDOK,BOTTOM_RIGHT);
  77         AddAnchor(IDCANCEL,BOTTOM_RIGHT);
  78         AddAnchor(IDC_GROUPT_REMOTE,TOP_LEFT,BOTTOM_RIGHT);
  79         AddAnchor(IDC_PUTTYKEY_AUTOLOAD,BOTTOM_LEFT);
  80         AddAnchor(IDC_CHECK_PRUNE,BOTTOM_LEFT);
  81         AddAnchor(IDC_CHECK_REBASE,BOTTOM_LEFT);
  82         AddAnchor(IDC_REMOTE_MANAGE,BOTTOM_LEFT);
  83         AddAnchor(IDHELP, BOTTOM_RIGHT);
  84
  85         CString WorkingDir=g_Git.m_CurrentDir;
  86         WorkingDir.Replace(_T(':'),_T('_'));
  87
  88         m_RemoteReg = CRegString(CString(_T("Software\\TortoiseGit\\History\\PullRemote\\")+WorkingDir));
  89         CString regkey;
  90         regkey.Format(_T("Software\\TortoiseGit\\TortoiseProc\\PullFetch\\%s_%d\\rebase"),WorkingDir,this->m_IsPull);
  91         m_regRebase=CRegDWORD(regkey,false);
  92         regkey.Format(_T("Software\\TortoiseGit\\TortoiseProc\\PullFetch\\%s_%d\\autoload"),WorkingDir,this->m_IsPull);
  93
  94         m_regAutoLoadPutty = CRegDWORD(regkey,this->m_bAutoLoad);
  95         m_bAutoLoad = m_regAutoLoadPutty;
  96
  97         if(!CAppUtils::IsSSHPutty())
  98                 m_bAutoLoad = false;
  99
 100         if(m_bAllowRebase)
 101         {
 102                 this->m_bRebase = m_regRebase;
 103         }
 104         else
 105         {
 106                 GetDlgItem(IDC_CHECK_REBASE)->ShowWindow(SW_HIDE);
 107                 this->m_bRebase = FALSE;
 108         }
 109
 110         this->UpdateData(FALSE);
 111
 112     this->AddOthersToAnchor();
 113
 114         this->GetDlgItem(IDC_PUTTYKEY_AUTOLOAD)->EnableWindow(m_bAutoLoadEnable);
 115
 116         CheckRadioButton(IDC_REMOTE_RD,IDC_OTHER_RD,IDC_REMOTE_RD);
 117         m_Remote.EnableWindow(TRUE);
 118         m_Other.EnableWindow(FALSE);
 119         if(!m_IsPull)
 120                 m_RemoteBranch.EnableWindow(FALSE);
 121
 122         if(m_IsPull) {
 123                 GetDlgItem(IDC_CHECK_REBASE)->ShowWindow(SW_HIDE);
 124                 GetDlgItem(IDC_CHECK_PRUNE)->ShowWindow(SW_HIDE);
 125         }
 126
 127         m_Other.SetURLHistory(TRUE);
 128         m_Other.LoadHistory(_T("Software\\TortoiseGit\\History\\PullURLS"), _T("url"));
 129         CString clippath=CAppUtils::GetClipboardLink();
 130         if(clippath.IsEmpty())
 131                 m_Other.SetCurSel(0);
 132         else
 133                 m_Other.SetWindowText(clippath);
 134
 135         m_RemoteBranch.LoadHistory(_T("Software\\TortoiseGit\\History\\PullRemoteBranch"), _T("br"));
 136         m_RemoteBranch.SetCurSel(0);
 137
 138         if(m_IsPull)
 139                 this->SetWindowTextW(CString(_T("Pull - ")) + g_Git.m_CurrentDir);
 140         else
 141                 this->SetWindowTextW(CString(_T("Fetch - ")) + g_Git.m_CurrentDir);
 142
 143         Refresh();
 144
 145         EnableSaveRestore(_T("PullFetchDlg"));
 146     this->m_RemoteManage.SetURL(CString());
 147         return TRUE;
 148 }
 149
 150 void CPullFetchDlg::Refresh()
 151 {
 152         //Select pull-remote from current branch
 153         CString currentBranch = g_Git.GetSymbolicRef();
 154         CString configName;
 155         configName.Format(L"branch.%s.remote", currentBranch);
 156         CString pullRemote = this->m_configPullRemote = g_Git.GetConfigValue(configName);
 157
 158         //Select pull-branch from current branch
 159         configName.Format(L"branch.%s.merge", currentBranch);
 160         CString pullBranch = m_configPullBranch = CGit::StripRefName(g_Git.GetConfigValue(configName));
 161         m_RemoteBranch.AddString(pullBranch);
 162
 163         if(pullRemote.IsEmpty())
 164                 pullRemote = m_RemoteReg;
 165
 166         if (!m_PreSelectRemote.IsEmpty())
 167                 pullRemote = m_PreSelectRemote;
 168
 169         STRING_VECTOR list;
 170         int sel=0;
 171         if(!g_Git.GetRemoteList(list))
 172         {
 173                 for(unsigned int i=0;i<list.size();i++)
 174                 {
 175                         m_Remote.AddString(list[i]);
 176                         if(list[i] == pullRemote)
 177                                 sel = i;
 178                 }
 179         }
 180         m_Remote.SetCurSel(sel);
 181 }
 182 // CPullFetchDlg message handlers
 183
 184 void CPullFetchDlg::OnBnClickedRd()
 185 {
 186         if( GetCheckedRadioButton(IDC_REMOTE_RD,IDC_OTHER_RD) == IDC_REMOTE_RD)
 187         {
 188                 m_Remote.EnableWindow(TRUE);
 189                 m_Other.EnableWindow(FALSE);
 190                 if(!m_IsPull)
 191                         m_RemoteBranch.EnableWindow(FALSE);
 192         }
 193         if( GetCheckedRadioButton(IDC_REMOTE_RD,IDC_OTHER_RD) == IDC_OTHER_RD)
 194         {
 195                 m_Remote.EnableWindow(FALSE);
 196                 m_Other.EnableWindow(TRUE);;
 197                 if(!m_IsPull)
 198                         m_RemoteBranch.EnableWindow(TRUE);
 199         }
 200 }
 201
 202 void CPullFetchDlg::OnBnClickedOk()
 203 {
 204         this->UpdateData();
 205         if( GetCheckedRadioButton(IDC_REMOTE_RD,IDC_OTHER_RD) == IDC_REMOTE_RD)
 206         {
 207                 m_RemoteURL=m_Remote.GetString();
 208                 if( !m_IsPull ||
 209                         (m_configPullRemote == m_RemoteURL && m_configPullBranch == m_RemoteBranch.GetString() ))
 210                         //When fetching or when pulling from the configured tracking branch, dont explicitly set the remote branch name,
 211                         //because otherwise git will not update the remote tracking branches.
 212                         m_RemoteBranchName.Empty();
 213                 else
 214                         m_RemoteBranchName=m_RemoteBranch.GetString();
 215         }
 216         if( GetCheckedRadioButton(IDC_REMOTE_RD,IDC_OTHER_RD) == IDC_OTHER_RD)
 217         {
 218                 m_Other.GetWindowTextW(m_RemoteURL);
 219                 m_RemoteBranchName=m_RemoteBranch.GetString();
 220         }
 221
 222         m_RemoteReg = m_Remote.GetString();
 223
 224         m_Other.SaveHistory();
 225         m_RemoteBranch.SaveHistory();
 226         this->m_regRebase=this->m_bRebase;
 227
 228         m_regAutoLoadPutty = m_bAutoLoad;
 229
 230         this->OnOK();
 231 }
 232
 233 void CPullFetchDlg::OnStnClickedRemoteManage()
 234 {
 235         CAppUtils::LaunchRemoteSetting();
 236         Refresh();
 237 }
 238
 239 void CPullFetchDlg::OnBnClickedButtonBrowseRef()
 240 {
 241         CString initialRef;
 242         initialRef.Format(L"refs/remotes/%s/%s", m_Remote.GetString(), m_RemoteBranch.GetString());
 243         CString selectedRef = CBrowseRefsDlg::PickRef(false, initialRef, gPickRef_Remote);
 244         if(selectedRef.Left(13) != "refs/remotes/")
 245                 return;
 246
 247         selectedRef = selectedRef.Mid(13);
 248         int ixSlash = selectedRef.Find('/');
 249
 250         CString remoteName   = selectedRef.Left(ixSlash);
 251         CString remoteBranch = selectedRef.Mid(ixSlash + 1);
 252
 253         int ixFound = m_Remote.FindStringExact(0, remoteName);
 254         if(ixFound >= 0)
 255                 m_Remote.SetCurSel(ixFound);
 256         m_RemoteBranch.AddString(remoteBranch);
 257
 258         CheckRadioButton(IDC_REMOTE_RD,IDC_OTHER_RD,IDC_REMOTE_RD);
 259 }