src/TortoiseProc/RebaseDlg.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2008-2019 - 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 #pragma once
  21 #include "StandAloneDlg.h"
  22 #include "GitStatusListCtrl.h"
  23 #include "SciEdit.h"
  24 #include "SplitterControl.h"
  25 #include "HistoryCombo.h"
  26 #include "GitLogList.h"
  27 #include "MenuButton.h"
  28 #include "ProjectProperties.h"
  29 #include "AppUtils.h"
  30
  31 // CRebaseDlg dialog
  32 #define IDC_REBASE_TAB 0x1000000
  33
  34 #define REBASE_TAB_CONFLICT     0
  35 #define REBASE_TAB_MESSAGE      1
  36 #define REBASE_TAB_LOG          2
  37
  38 #define MSG_REBASE_UPDATE_UI    (WM_USER+151)
  39
  40 class CRebaseDlg : public CResizableStandAloneDialog
  41 {
  42         DECLARE_DYNAMIC(CRebaseDlg)
  43
  44 public:
  45         CRebaseDlg(CWnd* pParent = nullptr); // standard constructor
  46         virtual ~CRebaseDlg();
  47
  48 // Dialog Data
  49         enum { IDD = IDD_REBASE };
  50
  51         enum REBASE_STAGE
  52         {
  53                 CHOOSE_BRANCH,
  54                 CHOOSE_COMMIT_PICK_MODE,
  55                 REBASE_START,
  56                 REBASE_CONTINUE,
  57                 REBASE_ABORT,
  58                 REBASE_FINISH,
  59                 REBASE_CONFLICT,
  60                 REBASE_EDIT,
  61                 REBASE_SQUASH_EDIT,
  62                 REBASE_SQUASH_CONFLICT,
  63                 REBASE_DONE,
  64                 REBASE_ERROR,
  65         };
  66
  67 protected:
  68         virtual void DoDataExchange(CDataExchange* pDX) override; // DDX/DDV support
  69         virtual BOOL OnInitDialog() override;
  70         DECLARE_MESSAGE_MAP()
  71         virtual LRESULT DefWindowProc(UINT message, WPARAM wParam, LPARAM lParam) override;
  72         afx_msg HBRUSH OnCtlColor(CDC* pDC, CWnd* pWnd, UINT nCtlColor);
  73         afx_msg LRESULT OnRebaseUpdateUI(WPARAM wParam, LPARAM lParam);
  74         void DoSize(int delta);
  75         void AddRebaseAnchor();
  76
  77         void SetSplitterRange();
  78         void SaveSplitterPos();
  79
  80         void LoadBranchInfo();
  81         void FetchLogList();
  82         void SetAllRebaseAction(int action);
  83         void OnCancel();
  84
  85         CRect m_DlgOrigRect;
  86         CRect m_CommitListOrigRect;
  87         CString m_sStatusText;
  88         bool m_bStatusWarning;
  89         virtual BOOL PreTranslateMessage(MSG* pMsg) override;
  90         bool LogListHasFocus(HWND hwnd);
  91         bool LogListHasMenuItem(int i);
  92
  93         CSciEdit m_wndOutputRebase;
  94         void SetContinueButtonText();
  95         void SetControlEnable();
  96         void UpdateProgress();
  97         void UpdateCurrentStatus();
  98         void ListConflictFile(bool noStoreScrollPosition);
  99         int     RunGitCmdRetryOrAbort(const CString& cmd);
 100         int  DoRebase();
 101         afx_msg LRESULT OnGitStatusListCtrlNeedsRefresh(WPARAM, LPARAM);
 102         void Refresh();
 103         volatile LONG m_bThreadRunning;
 104         volatile LONG m_bAbort;
 105         int  RebaseThread();
 106         static UINT RebaseThreadEntry(LPVOID pVoid) { return reinterpret_cast<CRebaseDlg*>(pVoid)->RebaseThread(); };
 107         BOOL IsEnd();
 108
 109         int IsCommitEmpty(const CGitHash& hash);
 110
 111         BOOL m_IsFastForward;
 112
 113         CGitHash m_OrigBranchHash;
 114         CGitHash m_OrigUpstreamHash;
 115         CString m_OrigHEADBranch;
 116         CGitHash m_OrigHEADHash;
 117
 118         ProjectProperties m_ProjectProperties;
 119
 120         int VerifyNoConflict();
 121
 122         CString m_SquashMessage;
 123         bool m_CurrentCommitEmpty;
 124         struct SquashFirstMetaData
 125         {
 126                 CString name;
 127                 CString email;
 128                 CTime time;
 129                 bool set = false;
 130
 131                 SquashFirstMetaData()
 132                         : set(false)
 133                 {
 134                 }
 135
 136                 SquashFirstMetaData(GitRev* rev)
 137                         : set(true)
 138                         , name(rev->GetAuthorName())
 139                         , email(rev->GetAuthorEmail())
 140                         , time(rev->GetAuthorDate())
 141                 {
 142                 }
 143
 144                 void UpdateDate(GitRev* rev)
 145                 {
 146                         ATLASSERT(set);
 147                         time = rev->GetAuthorDate();
 148                 }
 149
 150                 void Empty()
 151                 {
 152                         set = false;
 153                         name.Empty();
 154                         email.Empty();
 155                         time = 0;
 156                 }
 157
 158                 CString GetAuthor() const
 159                 {
 160                         if (!set)
 161                                 return CString();
 162                         CString temp;
 163                         temp.Format(L"%s <%s>", (LPCTSTR)name, (LPCTSTR)email);
 164                         return temp;
 165                 }
 166
 167                 CString GetAsParam(bool now) const
 168                 {
 169                         if (!set)
 170                                 return CString();
 171
 172                         CString date = time.Format(L"%Y-%m-%dT%H:%M:%S");
 173                         if (now)
 174                                 date = L"\"now\"";
 175
 176                         CString temp;
 177                         temp.Format(L"--date=%s --author=\"%s\" ", (LPCTSTR)date, (LPCTSTR)GetAuthor());
 178                         return temp;
 179                 }
 180         } m_SquashFirstMetaData;
 181         int m_iSquashdate;
 182
 183         int CheckNextCommitIsSquash();
 184         int GetCurrentCommitID();
 185         int FinishRebase();
 186         void RewriteNotes();
 187
 188         CMenuButton m_PostButton;
 189
 190         afx_msg void OnBnClickedRebaseSplit();
 191         afx_msg void OnSize(UINT nType, int cx, int cy);
 192         afx_msg void OnCbnSelchangeBranch();
 193         afx_msg void OnCbnSelchangeUpstream();
 194         afx_msg void OnBnClickedContinue();
 195         afx_msg void OnBnClickedAbort();
 196         afx_msg void OnLvnItemchangedLoglist(NMHDR *pNMHDR, LRESULT *pResult);
 197         afx_msg void OnSysColorChange();
 198         afx_msg LRESULT OnThemeChanged();
 199         void FillLogMessageCtrl();
 200
 201         CProgressCtrl           m_ProgressBar;
 202         CStatic                         m_CtrlStatusText;
 203
 204         BOOL                            m_bForce;
 205         BOOL                            m_bAddCherryPickedFrom;
 206         BOOL                            m_bAutoSkipFailedCommit;
 207         bool                            m_bRebaseAutoEnd;
 208
 209 public:
 210         CStringArray            m_PostButtonTexts;
 211         CGitLogList                     m_CommitList;
 212
 213         CString                         m_Upstream;
 214         CString                         m_Branch;
 215         CString                         m_Onto;
 216
 217         BOOL                            m_IsCherryPick;
 218         bool                            m_bRebaseAutoStart;
 219         BOOL                            m_bPreserveMerges;
 220 protected:
 221         CSplitterControl        m_wndSplitter;
 222         CMFCTabCtrl                     m_ctrlTabCtrl;
 223         CGitStatusListCtrl      m_FileListCtrl;
 224         CSciEdit                        m_LogMessageCtrl;
 225
 226         CHistoryCombo           m_BranchCtrl;
 227         CHistoryCombo           m_UpstreamCtrl;
 228
 229         CMenuButton                     m_SplitAllOptions;
 230
 231         BOOL                            m_bSplitCommit;
 232
 233         REBASE_STAGE            m_RebaseStage;
 234         bool                            m_bFinishedRebase;
 235         bool                            m_bStashed;
 236
 237         std::unordered_map<CGitHash, CGitHash> m_rewrittenCommitsMap;
 238         std::vector<CGitHash> m_forRewrite;
 239         std::unordered_map<CGitHash, GIT_REV_LIST> m_droppedCommitsMap;
 240         std::vector<CGitHash> m_currentCommits;
 241
 242         void AddBranchToolTips(CHistoryCombo& pBranch);
 243         void AddLogString(const CString& str);
 244         int StartRebase();
 245         int CheckRebaseCondition();
 246         void CheckRestoreStash();
 247         int m_CurrentRebaseIndex;
 248         int GoNext();
 249         void ResetParentForSquash(const CString& commitMessage);
 250         void CleanUpRebaseActiveFolder();
 251         afx_msg void OnBnClickedButtonReverse();
 252         afx_msg void OnBnClickedButtonBrowse();
 253         afx_msg void OnBnClickedRebaseCheckForce();
 254         afx_msg void OnBnClickedCheckCherryPickedFrom();
 255         afx_msg void OnBnClickedRebasePostButton();
 256         afx_msg void OnBnClickedSplitAllOptions();
 257         afx_msg void OnBnClickedButtonUp();
 258         afx_msg void OnBnClickedButtonDown();
 259         afx_msg void OnHelp();
 260
 261         afx_msg LRESULT OnTaskbarBtnCreated(WPARAM wParam, LPARAM lParam);
 262         CComPtr<ITaskbarList3>  m_pTaskbarList;
 263
 264         afx_msg LRESULT OnRebaseActionMessage(WPARAM wParam, LPARAM lParam);
 265         afx_msg LRESULT OnCommitsReordered(WPARAM wParam, LPARAM lParam);
 266         afx_msg void OnBnClickedRebaseSplitCommit();
 267         afx_msg void OnBnClickedButtonOnto();
 268         afx_msg void OnBnClickedButtonAdd();
 269 };