src/TortoiseProc/PushDlg.cpp

   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
  20 // PushDlg.cpp : implementation file
  21 //
  22
  23 #include "stdafx.h"
  24 #include "TortoiseProc.h"
  25 #include "PushDlg.h"
  26
  27 #include "Git.h"
  28 #include "registry.h"
  29 #include "AppUtils.h"
  30 #include "LogDlg.h"
  31 #include "BrowseRefsDlg.h"
  32 #include "RefLogDlg.h"
  33 #include "MessageBox.h"
  34
  35 // CPushDlg dialog
  36
  37 IMPLEMENT_DYNAMIC(CPushDlg, CHorizontalResizableStandAloneDialog)
  38
  39 CPushDlg::CPushDlg(CWnd* pParent /*=NULL*/)
  40         : CHorizontalResizableStandAloneDialog(CPushDlg::IDD, pParent)
  41         , m_bPushAllBranches(FALSE)
  42         , m_bForce(FALSE)
  43         , m_bPack(FALSE)
  44         , m_bTags(FALSE)
  45         , m_bAutoLoad(FALSE)
  46 {
  47         m_bAutoLoad = CAppUtils::IsSSHPutty();
  48 }
  49
  50 CPushDlg::~CPushDlg()
  51 {
  52 }
  53
  54 void CPushDlg::DoDataExchange(CDataExchange* pDX)
  55 {
  56         CHorizontalResizableStandAloneDialog::DoDataExchange(pDX);
  57         DDX_Control(pDX, IDC_BRANCH_REMOTE, this->m_BranchRemote);
  58         DDX_Control(pDX, IDC_BRANCH_SOURCE, this->m_BranchSource);
  59         DDX_Control(pDX, IDC_REMOTE, this->m_Remote);
  60         DDX_Control(pDX, IDC_URL, this->m_RemoteURL);
  61         DDX_Control(pDX, IDC_BUTTON_BROWSE_SOURCE_BRANCH, m_BrowseLocalRef);
  62         DDX_Check(pDX,IDC_FORCE,this->m_bForce);
  63         DDX_Check(pDX,IDC_PUSHALL,this->m_bPushAllBranches);
  64         DDX_Check(pDX,IDC_PACK,this->m_bPack);
  65         DDX_Check(pDX,IDC_TAGS,this->m_bTags);
  66         DDX_Check(pDX,IDC_PUTTYKEY_AUTOLOAD,this->m_bAutoLoad);
  67 }
  68
  69 BEGIN_MESSAGE_MAP(CPushDlg, CHorizontalResizableStandAloneDialog)
  70         ON_BN_CLICKED(IDC_RD_REMOTE, &CPushDlg::OnBnClickedRd)
  71         ON_BN_CLICKED(IDC_RD_URL, &CPushDlg::OnBnClickedRd)
  72         ON_CBN_SELCHANGE(IDC_BRANCH_SOURCE, &CPushDlg::OnCbnSelchangeBranchSource)
  73         ON_BN_CLICKED(IDOK, &CPushDlg::OnBnClickedOk)
  74         ON_BN_CLICKED(IDC_REMOTE_MANAGE, &CPushDlg::OnBnClickedRemoteManage)
  75         ON_BN_CLICKED(IDC_BUTTON_BROWSE_SOURCE_BRANCH, &CPushDlg::OnBnClickedButtonBrowseSourceBranch)
  76         ON_BN_CLICKED(IDC_BUTTON_BROWSE_DEST_BRANCH, &CPushDlg::OnBnClickedButtonBrowseDestBranch)
  77         ON_BN_CLICKED(IDC_PUSHALL, &CPushDlg::OnBnClickedPushall)
  78 END_MESSAGE_MAP()
  79
  80 BOOL CPushDlg::OnInitDialog()
  81 {
  82         CHorizontalResizableStandAloneDialog::OnInitDialog();
  83         CAppUtils::MarkWindowAsUnpinnable(m_hWnd);
  84
  85         AddAnchor(IDOK,BOTTOM_RIGHT);
  86         AddAnchor(IDCANCEL,BOTTOM_RIGHT);
  87         AddAnchor(IDC_BRANCH_GROUP, TOP_LEFT,TOP_RIGHT);
  88         AddAnchor(IDC_STATIC_REMOTE, TOP_LEFT);
  89         AddAnchor(IDC_STATIC_SOURCE, TOP_LEFT);
  90
  91         AddAnchor(IDC_PUSHALL, TOP_LEFT);
  92         AddAnchor(IDC_BRANCH_REMOTE, TOP_LEFT, TOP_RIGHT);
  93         AddAnchor(IDC_BUTTON_BROWSE_DEST_BRANCH, TOP_RIGHT);
  94         AddAnchor(IDC_BRANCH_SOURCE, TOP_LEFT, TOP_RIGHT);
  95         AddAnchor(IDC_BUTTON_BROWSE_SOURCE_BRANCH, TOP_RIGHT);
  96
  97         AddAnchor(IDC_URL_GROUP, TOP_LEFT,TOP_RIGHT);
  98         AddAnchor(IDC_RD_REMOTE, TOP_LEFT);
  99         AddAnchor(IDC_RD_URL, TOP_LEFT);
 100
 101         AddAnchor(IDC_REMOTE, TOP_LEFT, TOP_RIGHT);
 102
 103         AddAnchor(IDC_URL, TOP_LEFT,TOP_RIGHT);
 104
 105         AddAnchor(IDC_OPTION_GROUP, TOP_LEFT,TOP_RIGHT);
 106
 107         AddAnchor(IDC_FORCE, TOP_LEFT);
 108         AddAnchor(IDC_PACK, TOP_LEFT);
 109         AddAnchor(IDC_TAGS, TOP_LEFT);
 110         AddAnchor(IDC_PUTTYKEY_AUTOLOAD,TOP_LEFT);
 111
 112         AddAnchor(IDC_REMOTE_MANAGE,TOP_RIGHT);
 113         AddAnchor(IDHELP, BOTTOM_RIGHT);
 114
 115         AddOthersToAnchor();
 116
 117         AdjustControlSize(IDC_RD_REMOTE);
 118         AdjustControlSize(IDC_RD_URL);
 119         AdjustControlSize(IDC_PUSHALL);
 120         AdjustControlSize(IDC_FORCE);
 121         AdjustControlSize(IDC_PACK);
 122         AdjustControlSize(IDC_TAGS);
 123         AdjustControlSize(IDC_PUTTYKEY_AUTOLOAD);
 124
 125         CString sWindowTitle;
 126         GetWindowText(sWindowTitle);
 127         CAppUtils::SetWindowTitle(m_hWnd, g_Git.m_CurrentDir, sWindowTitle);
 128
 129         this->GetDlgItem(IDC_PUTTYKEY_AUTOLOAD)->EnableWindow(CAppUtils::IsSSHPutty());
 130
 131         EnableSaveRestore(_T("PushDlg"));
 132
 133         m_RemoteURL.SetURLHistory(TRUE);
 134
 135         CString WorkingDir=g_Git.m_CurrentDir;
 136         WorkingDir.Replace(_T(':'),_T('_'));
 137         m_RemoteURL.LoadHistory(CString(_T("Software\\TortoiseGit\\History\\PushURLS\\"))+WorkingDir, _T("url"));
 138         CString clippath=CAppUtils::GetClipboardLink();
 139         if(clippath.IsEmpty())
 140                 m_RemoteURL.SetCurSel(0);
 141         else
 142                 m_RemoteURL.SetWindowText(clippath);
 143
 144         m_RemoteURL.EnableWindow(FALSE);
 145         CheckRadioButton(IDC_RD_REMOTE,IDC_RD_URL,IDC_RD_REMOTE);
 146
 147
 148         this->m_regAutoLoad = CRegDWORD(CString(_T("Software\\TortoiseGit\\History\\PushDlgAutoLoad\\"))+WorkingDir,
 149                                                                         m_bAutoLoad);
 150         m_bAutoLoad = this->m_regAutoLoad;
 151         if(!CAppUtils::IsSSHPutty())
 152                 m_bAutoLoad = false;
 153
 154         m_BrowseLocalRef.m_bRightArrow = TRUE;
 155         m_BrowseLocalRef.m_bDefaultClick = FALSE;
 156         m_BrowseLocalRef.m_bMarkDefault = FALSE;
 157         m_BrowseLocalRef.AddEntry(CString(MAKEINTRESOURCE(IDS_REFBROWSE)));
 158         m_BrowseLocalRef.AddEntry(CString(MAKEINTRESOURCE(IDS_LOG)));
 159         m_BrowseLocalRef.AddEntry(CString(MAKEINTRESOURCE(IDS_REFLOG)));
 160
 161         Refresh();
 162
 163
 164         //m_BranchRemote.SetWindowTextW(m_BranchSource.GetString());
 165
 166         this->UpdateData(false);
 167         return TRUE;
 168 }
 169
 170 void CPushDlg::Refresh()
 171 {
 172         CString WorkingDir=g_Git.m_CurrentDir;
 173         WorkingDir.Replace(_T(':'),_T('_'));
 174
 175         CRegString remote(CString(_T("Software\\TortoiseGit\\History\\PushRemote\\")+WorkingDir));
 176         m_RemoteReg = remote;
 177         int sel=0;
 178
 179         STRING_VECTOR list;
 180         m_Remote.Reset();
 181
 182         if(!g_Git.GetRemoteList(list))
 183         {
 184                 for(unsigned int i=0;i<list.size();i++)
 185                 {
 186                         m_Remote.AddString(list[i]);
 187                         if(list[i] == remote)
 188                                 sel = i;
 189                 }
 190         }
 191         m_Remote.SetCurSel(sel);
 192
 193         int current=0;
 194         list.clear();
 195         m_BranchSource.Reset();
 196         m_BranchSource.AddString(_T(" ")); // empty string does not work, for removal of remote branches/tags
 197         m_BranchSource.SetMaxHistoryItems(0x7FFFFFFF);
 198         if(!g_Git.GetBranchList(list,&current))
 199         {
 200                 for(unsigned int i=0;i<list.size();i++)
 201                         m_BranchSource.AddString(list[i]);
 202                 current++; // shift for " "
 203         }
 204         if (wcsncmp(m_BranchSourceName, _T("refs/"), 5) == 0)
 205                 m_BranchSourceName = m_BranchSourceName.Mid(5);
 206         if (wcsncmp(m_BranchSourceName, _T("heads/"), 6) == 0)
 207         {
 208                 m_BranchSourceName = m_BranchSourceName.Mid(6);
 209                 m_BranchSource.SetCurSel(m_BranchSource.FindStringExact(-1, m_BranchSourceName));
 210         }
 211         else if (wcsncmp(m_BranchSourceName, _T("remotes/"), 8) == 0)
 212                 m_BranchSource.SetCurSel(m_BranchSource.FindStringExact(-1, m_BranchSourceName));
 213         else
 214                 m_BranchSource.SetCurSel(current);
 215
 216         GetRemoteBranch(m_BranchSource.GetString());
 217
 218         this->GetDlgItem(IDOK)->EnableWindow(m_BranchSource.GetCount() != 0);
 219 }
 220
 221 void CPushDlg::GetRemoteBranch(CString currentBranch)
 222 {
 223         CString WorkingDir=g_Git.m_CurrentDir;
 224         WorkingDir.Replace(_T(':'), _T('_'));
 225
 226         if (currentBranch.IsEmpty())
 227                 return;
 228
 229         CString configName;
 230
 231         configName.Format(L"branch.%s.pushremote", currentBranch);
 232         CString pushRemote = g_Git.GetConfigValue(configName);
 233         if( pushRemote.IsEmpty() )
 234         {
 235                 configName.Format(L"branch.%s.remote", currentBranch);
 236                 pushRemote = g_Git.GetConfigValue(configName);
 237         }
 238
 239         CRegString remote(CString(_T("Software\\TortoiseGit\\History\\PushRemote\\")+WorkingDir));
 240
 241         if( !pushRemote.IsEmpty() )
 242                 remote=pushRemote;
 243
 244         //Select pull-branch from current branch
 245         configName.Format(L"branch.%s.pushbranch", currentBranch);
 246         CString pushBranch = CGit::StripRefName(g_Git.GetConfigValue(configName));
 247         if( pushBranch.IsEmpty() )
 248         {
 249                 configName.Format(L"branch.%s.merge", currentBranch);
 250                 pushBranch = CGit::StripRefName(g_Git.GetConfigValue(configName));
 251         }
 252
 253         m_BranchRemote.LoadHistory(CString(_T("Software\\TortoiseGit\\History\\RemoteBranch\\"))+WorkingDir, _T("branch"));
 254         if( !pushBranch.IsEmpty() )
 255                 m_BranchRemote.AddString(pushBranch);
 256 }
 257
 258 // CPushDlg message handlers
 259
 260 void CPushDlg::OnBnClickedRd()
 261 {
 262         if( GetCheckedRadioButton(IDC_RD_REMOTE,IDC_RD_URL) == IDC_RD_REMOTE)
 263         {
 264                 m_Remote.EnableWindow(TRUE);
 265                 GetDlgItem(IDC_REMOTE_MANAGE)->EnableWindow(TRUE);
 266                 m_RemoteURL.EnableWindow(FALSE);
 267         }
 268         if( GetCheckedRadioButton(IDC_RD_REMOTE,IDC_RD_URL) == IDC_RD_URL)
 269         {
 270                 m_Remote.EnableWindow(FALSE);
 271                 GetDlgItem(IDC_REMOTE_MANAGE)->EnableWindow(FALSE);
 272                 m_RemoteURL.EnableWindow(TRUE);
 273         }
 274 }
 275
 276
 277 void CPushDlg::OnCbnSelchangeBranchSource()
 278 {
 279         GetRemoteBranch(m_BranchSource.GetString());
 280 }
 281
 282 void CPushDlg::OnBnClickedOk()
 283 {
 284         CHorizontalResizableStandAloneDialog::UpdateData(TRUE);
 285
 286         if( GetCheckedRadioButton(IDC_RD_REMOTE,IDC_RD_URL) == IDC_RD_REMOTE)
 287         {
 288                 m_URL=m_Remote.GetString();
 289         }
 290         if( GetCheckedRadioButton(IDC_RD_REMOTE,IDC_RD_URL) == IDC_RD_URL)
 291         {
 292                 m_URL=m_RemoteURL.GetString();
 293         }
 294
 295         if (!m_bPushAllBranches)
 296         {
 297                 this->m_BranchRemoteName=m_BranchRemote.GetString().Trim();
 298                 this->m_BranchSourceName=m_BranchSource.GetString().Trim();
 299
 300                 if (m_BranchSourceName.IsEmpty() && m_BranchRemoteName.IsEmpty())
 301                 {
 302                         if (CMessageBox::Show(NULL, IDS_B_T_BOTHEMPTY, IDS_APPNAME, MB_ICONQUESTION | MB_YESNO) == IDNO)
 303                                 return;
 304                 }
 305                 if (m_BranchSourceName.IsEmpty() && !m_BranchRemoteName.IsEmpty())
 306                 {
 307                         if (CMessageBox::Show(NULL, IDS_B_T_LOCALEMPTY, IDS_APPNAME, MB_ICONEXCLAMATION | MB_YESNO) == IDNO)
 308                                 return;
 309                 }
 310                 else if (!m_BranchRemoteName.IsEmpty() && !g_Git.IsBranchNameValid(this->m_BranchRemoteName))
 311                 {
 312                         CMessageBox::Show(NULL, IDS_B_T_INVALID, IDS_APPNAME, MB_OK);
 313                         return;
 314                 }
 315                 else
 316                 {
 317                         // do not store branch names on removal
 318                         this->m_RemoteURL.SaveHistory();
 319                         this->m_BranchRemote.SaveHistory();
 320                         m_RemoteReg = m_Remote.GetString();
 321                 }
 322         }
 323
 324         this->m_regAutoLoad = m_bAutoLoad ;
 325
 326         CHorizontalResizableStandAloneDialog::OnOK();
 327 }
 328
 329 void CPushDlg::OnBnClickedRemoteManage()
 330 {
 331         CAppUtils::LaunchRemoteSetting();
 332         Refresh();
 333 }
 334
 335 void CPushDlg::OnBnClickedButtonBrowseSourceBranch()
 336 {
 337         switch (m_BrowseLocalRef.GetCurrentEntry())
 338         {
 339         case 0: /* Browse Refence*/
 340                 {
 341                         if(CBrowseRefsDlg::PickRefForCombo(&m_BranchSource, gPickRef_Head))
 342                                 OnCbnSelchangeBranchSource();
 343                 }
 344                 break;
 345
 346         case 1: /* Log */
 347                 {
 348                         CLogDlg dlg;
 349                         dlg.SetSelect(true);
 350                         if(dlg.DoModal() == IDOK)
 351                         {
 352                                 if (dlg.GetSelectedHash().IsEmpty())
 353                                         return;
 354
 355                                 m_BranchSource.SetWindowText(dlg.GetSelectedHash());
 356                                 OnCbnSelchangeBranchSource();
 357                         }
 358                 }
 359                 break;
 360
 361         case 2: /*RefLog*/
 362                 {
 363                         CRefLogDlg dlg;
 364                         if(dlg.DoModal() == IDOK)
 365                         {
 366                                 m_BranchSource.SetWindowText(dlg.m_SelectedHash);
 367                                 OnCbnSelchangeBranchSource();
 368                         }
 369                 }
 370                 break;
 371         }
 372 }
 373
 374 void CPushDlg::OnBnClickedButtonBrowseDestBranch()
 375 {
 376         CString remoteBranchName;
 377         CString remoteName;
 378         m_BranchRemote.GetWindowText(remoteBranchName);
 379         remoteName = m_Remote.GetString();
 380         remoteBranchName = remoteName + '/' + remoteBranchName;
 381         remoteBranchName = CBrowseRefsDlg::PickRef(false, remoteBranchName, gPickRef_Remote);
 382         if(remoteBranchName.IsEmpty())
 383                 return; //Canceled
 384         remoteBranchName = remoteBranchName.Mid(13);//Strip 'refs/remotes/'
 385         int slashPlace = remoteBranchName.Find('/');
 386         remoteName = remoteBranchName.Left(slashPlace);
 387         remoteBranchName = remoteBranchName.Mid(slashPlace + 1); //Strip remote name (for example 'origin/')
 388
 389         //Select remote
 390         int remoteSel = m_Remote.FindStringExact(0,remoteName);
 391         if(remoteSel >= 0)
 392                 m_Remote.SetCurSel(remoteSel);
 393
 394         //Select branch
 395         m_BranchRemote.AddString(remoteBranchName);
 396 }
 397
 398 BOOL CPushDlg::PreTranslateMessage(MSG* pMsg)
 399 {
 400         if (pMsg->message == WM_KEYDOWN)
 401         {
 402                 switch (pMsg->wParam)
 403                 {
 404                 case VK_F5:
 405                         {
 406                                 Refresh();
 407                         }
 408                         break;
 409                 }
 410         }
 411
 412         return CHorizontalResizableStandAloneDialog::PreTranslateMessage(pMsg);
 413 }
 414
 415 void CPushDlg::OnBnClickedPushall()
 416 {
 417         this->UpdateData();
 418         this->GetDlgItem(IDC_BRANCH_REMOTE)->EnableWindow(!m_bPushAllBranches);
 419         this->GetDlgItem(IDC_BRANCH_SOURCE)->EnableWindow(!m_bPushAllBranches);
 420         this->GetDlgItem(IDC_BUTTON_BROWSE_SOURCE_BRANCH)->EnableWindow(!m_bPushAllBranches);
 421         this->GetDlgItem(IDC_BUTTON_BROWSE_DEST_BRANCH)->EnableWindow(!m_bPushAllBranches);
 422         this->GetDlgItem(IDC_TAGS)->EnableWindow(!m_bPushAllBranches);
 423         if (m_bTags && m_bPushAllBranches)
 424         {
 425                 m_bTags = FALSE;
 426                 UpdateData(FALSE);
 427         }
 428 }