Make sure buffer is large enough for the nul terminator
[TortoiseGit.git] / src / TortoiseProc / ChangedDlg.cpp
blob8c77bde732c73107cf4dcc35b10a9d3e9e5518a2
1 // TortoiseGit - a Windows shell extension for easy version control
3 // Copyright (C) 2008-2015 - TortoiseGit
4 // Copyright (C) 2003-2008 - TortoiseSVN
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.
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.
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.
20 #include "stdafx.h"
21 #include "TortoiseProc.h"
22 #include "ChangedDlg.h"
23 #include "MessageBox.h"
24 #include "cursor.h"
25 #include "AppUtils.h"
26 #include "ChangedDlg.h"
27 #include "IconMenu.h"
28 #include "RefLogDlg.h"
30 #include "GitStatusListCtrl.h"
32 IMPLEMENT_DYNAMIC(CChangedDlg, CResizableStandAloneDialog)
33 CChangedDlg::CChangedDlg(CWnd* pParent /*=NULL*/)
34 : CResizableStandAloneDialog(CChangedDlg::IDD, pParent)
35 , m_bShowUnversioned(FALSE)
36 , m_iShowUnmodified(0)
37 , m_bBlock(FALSE)
38 , m_bCanceled(false)
39 , m_bShowIgnored(FALSE)
40 , m_bShowLocalChangesIgnored(FALSE)
41 , m_bWholeProject(FALSE)
43 m_bRemote = FALSE;
46 CChangedDlg::~CChangedDlg()
50 void CChangedDlg::DoDataExchange(CDataExchange* pDX)
52 CResizableStandAloneDialog::DoDataExchange(pDX);
53 DDX_Control(pDX, IDC_CHANGEDLIST, m_FileListCtrl);
54 DDX_Control(pDX, IDC_BUTTON_STASH, m_ctrlStash);
55 DDX_Check(pDX, IDC_SHOWUNVERSIONED, m_bShowUnversioned);
56 DDX_Check(pDX, IDC_SHOWUNMODIFIED, m_iShowUnmodified);
57 DDX_Check(pDX, IDC_SHOWIGNORED, m_bShowIgnored);
58 DDX_Check(pDX, IDC_SHOWLOCALCHANGESIGNORED, m_bShowLocalChangesIgnored);
59 DDX_Check(pDX, IDC_WHOLE_PROJECT, m_bWholeProject);
63 BEGIN_MESSAGE_MAP(CChangedDlg, CResizableStandAloneDialog)
64 ON_BN_CLICKED(IDC_SHOWUNVERSIONED, OnBnClickedShowunversioned)
65 ON_BN_CLICKED(IDC_SHOWUNMODIFIED, OnBnClickedShowUnmodified)
66 ON_REGISTERED_MESSAGE(CGitStatusListCtrl::GITSLNM_NEEDSREFRESH, OnSVNStatusListCtrlNeedsRefresh)
67 ON_REGISTERED_MESSAGE(CGitStatusListCtrl::GITSLNM_ITEMCOUNTCHANGED, OnSVNStatusListCtrlItemCountChanged)
68 ON_BN_CLICKED(IDC_SHOWIGNORED, &CChangedDlg::OnBnClickedShowignored)
69 ON_BN_CLICKED(IDC_REFRESH, &CChangedDlg::OnBnClickedRefresh)
70 ON_BN_CLICKED(IDC_COMMIT, &CChangedDlg::OnBnClickedCommit)
71 ON_BN_CLICKED(IDC_BUTTON_STASH, &CChangedDlg::OnBnClickedStash)
72 ON_BN_CLICKED(IDC_BUTTON_UNIFIEDDIFF, &CChangedDlg::OnBnClickedButtonUnifieddiff)
73 ON_BN_CLICKED(IDC_SHOWLOCALCHANGESIGNORED, &CChangedDlg::OnBnClickedShowlocalchangesignored)
74 ON_BN_CLICKED(IDC_WHOLE_PROJECT, OnBnClickedWholeProject)
75 END_MESSAGE_MAP()
77 BOOL CChangedDlg::OnInitDialog()
79 CResizableStandAloneDialog::OnInitDialog();
81 m_regAddBeforeCommit = CRegDWORD(_T("Software\\TortoiseGit\\AddBeforeCommit"), TRUE);
82 m_bShowUnversioned = m_regAddBeforeCommit;
84 CString regPath(g_Git.m_CurrentDir);
85 regPath.Replace(_T(":"), _T("_"));
86 m_regShowWholeProject = CRegDWORD(_T("Software\\TortoiseGit\\TortoiseProc\\ShowWholeProject\\") + regPath, FALSE);
87 m_bWholeProject = m_regShowWholeProject;
88 SetDlgTitle();
90 UpdateData(FALSE);
92 m_FileListCtrl.Init(GITSLC_COLEXT | GITSLC_COLSTATUS | GITSLC_COLADD| GITSLC_COLDEL | GITSLC_COLMODIFICATIONDATE, _T("ChangedDlg"), (GITSLC_POPALL ^ (GITSLC_POPSAVEAS | GITSLC_POPRESTORE | GITSLC_PREPAREDIFF)), false);
93 m_FileListCtrl.SetCancelBool(&m_bCanceled);
94 m_FileListCtrl.SetBackgroundImage(IDI_CFM_BKG);
95 m_FileListCtrl.SetEmptyString(IDS_REPOSTATUS_EMPTYFILELIST);
97 AdjustControlSize(IDC_SHOWUNVERSIONED);
98 AdjustControlSize(IDC_SHOWUNMODIFIED);
99 AdjustControlSize(IDC_SHOWLOCALCHANGESIGNORED);
100 AdjustControlSize(IDC_SHOWIGNORED);
101 AdjustControlSize(IDC_WHOLE_PROJECT);
103 AddAnchor(IDC_CHANGEDLIST, TOP_LEFT, BOTTOM_RIGHT);
104 AddAnchor(IDC_SUMMARYTEXT, BOTTOM_LEFT, BOTTOM_RIGHT);
105 AddAnchor(IDC_SHOWUNVERSIONED, BOTTOM_LEFT);
106 AddAnchor(IDC_SHOWUNMODIFIED, BOTTOM_LEFT);
107 AddAnchor(IDC_SHOWLOCALCHANGESIGNORED, BOTTOM_LEFT);
108 AddAnchor(IDC_SHOWIGNORED, BOTTOM_LEFT);
109 AddAnchor(IDC_WHOLE_PROJECT, BOTTOM_LEFT);
110 AddAnchor(IDC_INFOLABEL, BOTTOM_RIGHT);
111 AddAnchor(IDC_BUTTON_STASH, BOTTOM_RIGHT);
112 AddAnchor(IDC_BUTTON_UNIFIEDDIFF, BOTTOM_RIGHT);
113 AddAnchor(IDC_COMMIT, BOTTOM_RIGHT);
114 AddAnchor(IDC_REFRESH, BOTTOM_RIGHT);
115 AddAnchor(IDOK, BOTTOM_RIGHT);
116 // SetPromptParentWindow(m_hWnd);
117 if (hWndExplorer)
118 CenterWindow(CWnd::FromHandle(hWndExplorer));
119 EnableSaveRestore(_T("ChangedDlg"));
121 m_bRemote = !!(DWORD)CRegDWORD(_T("Software\\TortoiseGit\\CheckRepo"), FALSE);
123 // first start a thread to obtain the status without
124 // blocking the dialog
125 if (AfxBeginThread(ChangedStatusThreadEntry, this)==NULL)
127 CMessageBox::Show(NULL, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
130 return TRUE;
133 UINT CChangedDlg::ChangedStatusThreadEntry(LPVOID pVoid)
135 return ((CChangedDlg*)pVoid)->ChangedStatusThread();
138 UINT CChangedDlg::ChangedStatusThread()
140 InterlockedExchange(&m_bBlock, TRUE);
142 g_Git.RefreshGitIndex();
144 m_bCanceled = false;
145 SetDlgItemText(IDOK, CString(MAKEINTRESOURCE(IDS_MSGBOX_CANCEL)));
146 DialogEnableWindow(IDC_REFRESH, FALSE);
147 DialogEnableWindow(IDC_SHOWUNVERSIONED, FALSE);
148 DialogEnableWindow(IDC_SHOWUNMODIFIED, FALSE);
149 DialogEnableWindow(IDC_SHOWIGNORED, FALSE);
150 DialogEnableWindow(IDC_SHOWLOCALCHANGESIGNORED, FALSE);
151 CString temp;
152 m_FileListCtrl.Clear();
153 if (!m_FileListCtrl.GetStatus(m_bWholeProject ? nullptr : &m_pathList, m_bRemote, m_bShowIgnored != FALSE, m_bShowUnversioned != FALSE, m_bShowLocalChangesIgnored != FALSE))
155 if (!m_FileListCtrl.GetLastErrorMessage().IsEmpty())
156 m_FileListCtrl.SetEmptyString(m_FileListCtrl.GetLastErrorMessage());
158 unsigned int dwShow = GITSLC_SHOWVERSIONEDBUTNORMALANDEXTERNALS | GITSLC_SHOWLOCKS | GITSLC_SHOWSWITCHED | GITSLC_SHOWINCHANGELIST;
159 dwShow |= m_bShowUnversioned ? GITSLC_SHOWUNVERSIONED : 0;
160 dwShow |= m_iShowUnmodified ? GITSLC_SHOWNORMAL : 0;
161 dwShow |= m_bShowIgnored ? GITSLC_SHOWIGNORED : 0;
162 dwShow |= m_bShowLocalChangesIgnored ? GITSLC_SHOWASSUMEVALID | GITSLC_SHOWSKIPWORKTREE : 0;
163 m_FileListCtrl.Show(dwShow);
164 UpdateStatistics();
166 SetDlgTitle();
167 bool bIsDirectory = false;
168 if (m_pathList.GetCount() == 1)
169 bIsDirectory = m_pathList[0].IsDirectory() || m_pathList[0].IsEmpty(); // if it is empty it is g_Git.m_CurrentDir which is a directory
171 SetDlgItemText(IDOK, CString(MAKEINTRESOURCE(IDS_MSGBOX_OK)));
172 DialogEnableWindow(IDC_REFRESH, TRUE);
173 DialogEnableWindow(IDC_SHOWUNVERSIONED, bIsDirectory);
174 //DialogEnableWindow(IDC_SHOWUNMODIFIED, bIsDirectory);
175 DialogEnableWindow(IDC_SHOWIGNORED, bIsDirectory);
176 DialogEnableWindow(IDC_SHOWLOCALCHANGESIGNORED, TRUE);
177 InterlockedExchange(&m_bBlock, FALSE);
178 // revert the remote flag back to the default
179 m_bRemote = !!(DWORD)CRegDWORD(_T("Software\\TortoiseGit\\CheckRepo"), FALSE);
180 RefreshCursor();
181 return 0;
184 void CChangedDlg::OnOK()
186 if (m_bBlock)
188 m_bCanceled = true;
189 return;
191 __super::OnOK();
194 void CChangedDlg::OnCancel()
196 if (m_bBlock)
198 m_bCanceled = true;
199 return;
201 __super::OnCancel();
204 DWORD CChangedDlg::UpdateShowFlags()
206 DWORD dwShow = m_FileListCtrl.GetShowFlags();
207 if (m_bShowUnversioned)
208 dwShow |= GITSLC_SHOWUNVERSIONED;
209 else
210 dwShow &= ~GITSLC_SHOWUNVERSIONED;
211 if (m_iShowUnmodified)
212 dwShow |= GITSLC_SHOWNORMAL;
213 else
214 dwShow &= ~GITSLC_SHOWNORMAL;
215 if (m_bShowIgnored)
216 dwShow |= GITSLC_SHOWIGNORED;
217 else
218 dwShow &= ~GITSLC_SHOWIGNORED;
219 if (m_bShowLocalChangesIgnored)
220 dwShow |= GITSLC_SHOWASSUMEVALID|GITSLC_SHOWSKIPWORKTREE;
221 else
222 dwShow &= ~(GITSLC_SHOWASSUMEVALID|GITSLC_SHOWSKIPWORKTREE);
224 // old bShowExternals:
225 dwShow &= ~(GITSLC_SHOWEXTERNAL | GITSLC_SHOWINEXTERNALS | GITSLC_SHOWEXTERNALFROMDIFFERENTREPO);
227 return dwShow;
230 void CChangedDlg::SetDlgTitle()
232 if (m_sTitle.IsEmpty())
233 GetWindowText(m_sTitle);
235 if (m_bWholeProject)
236 CAppUtils::SetWindowTitle(m_hWnd, g_Git.m_CurrentDir, m_sTitle);
237 else
239 if (m_pathList.GetCount() == 1)
240 CAppUtils::SetWindowTitle(m_hWnd, g_Git.CombinePath(m_pathList[0].GetUIPathString()), m_sTitle);
241 else
242 CAppUtils::SetWindowTitle(m_hWnd, g_Git.CombinePath(m_FileListCtrl.GetCommonDirectory(false)), m_sTitle);
246 void CChangedDlg::OnBnClickedShowunversioned()
248 UpdateData();
249 m_regAddBeforeCommit = m_bShowUnversioned;
250 if(m_FileListCtrl.m_FileLoaded & CGitStatusListCtrl::FILELIST_UNVER)
252 m_FileListCtrl.Show(UpdateShowFlags());
254 else
256 if(m_bShowUnversioned)
258 if (AfxBeginThread(ChangedStatusThreadEntry, this)==NULL)
260 CMessageBox::Show(NULL, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
264 UpdateStatistics();
267 void CChangedDlg::OnBnClickedShowUnmodified()
269 UpdateData();
270 m_FileListCtrl.Show(UpdateShowFlags());
271 m_regAddBeforeCommit = m_bShowUnversioned;
272 UpdateStatistics();
275 void CChangedDlg::OnBnClickedShowignored()
277 UpdateData();
278 if (m_FileListCtrl.m_FileLoaded & CGitStatusListCtrl::FILELIST_IGNORE)
279 m_FileListCtrl.Show(UpdateShowFlags());
280 else if (m_bShowIgnored)
282 if (AfxBeginThread(ChangedStatusThreadEntry, this) == nullptr)
283 CMessageBox::Show(NULL, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
285 UpdateStatistics();
288 void CChangedDlg::OnBnClickedShowlocalchangesignored()
290 UpdateData();
291 if (m_FileListCtrl.m_FileLoaded & CGitStatusListCtrl::FILELIST_LOCALCHANGESIGNORED)
292 m_FileListCtrl.Show(UpdateShowFlags());
293 else if (m_bShowLocalChangesIgnored)
295 if (AfxBeginThread(ChangedStatusThreadEntry, this) == nullptr)
296 CMessageBox::Show(nullptr, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
298 UpdateStatistics();
301 LRESULT CChangedDlg::OnSVNStatusListCtrlNeedsRefresh(WPARAM, LPARAM)
303 if (AfxBeginThread(ChangedStatusThreadEntry, this)==NULL)
305 CMessageBox::Show(NULL, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
307 return 0;
310 LRESULT CChangedDlg::OnSVNStatusListCtrlItemCountChanged(WPARAM, LPARAM)
312 UpdateStatistics();
313 return 0;
316 BOOL CChangedDlg::PreTranslateMessage(MSG* pMsg)
318 if (pMsg->message == WM_KEYDOWN)
320 switch (pMsg->wParam)
322 case VK_F5:
324 if (m_bBlock)
325 return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
326 if (AfxBeginThread(ChangedStatusThreadEntry, this)==NULL)
328 CMessageBox::Show(NULL, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
331 break;
335 return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
338 void CChangedDlg::OnBnClickedRefresh()
340 if (!m_bBlock)
342 if (AfxBeginThread(ChangedStatusThreadEntry, this)==NULL)
344 CMessageBox::Show(NULL, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
349 void CChangedDlg::UpdateStatistics()
351 CString temp;
352 #if 0
353 LONG lMin, lMax;
355 m_FileListCtrl.GetMinMaxRevisions(lMin, lMax, true, false);
356 if (LONG(m_FileListCtrl.m_HeadRev) >= 0)
358 temp.Format(IDS_REPOSTATUS_HEADREV, lMin, lMax, LONG(m_FileListCtrl.m_HeadRev));
359 SetDlgItemText(IDC_SUMMARYTEXT, temp);
361 else
363 temp.Format(IDS_REPOSTATUS_WCINFO, lMin, lMax);
364 SetDlgItemText(IDC_SUMMARYTEXT, temp);
366 #endif
367 temp = m_FileListCtrl.GetStatisticsString();
368 temp.Replace(_T(" = "), _T("="));
369 temp.Replace(_T("\n"), _T(", "));
370 SetDlgItemText(IDC_INFOLABEL, temp);
374 void CChangedDlg::OnBnClickedCommit()
376 CString cmd = _T("/command:commit /path:\"");
377 bool bSingleFile = ((m_pathList.GetCount()==1)&&(!m_pathList[0].IsEmpty())&&(!m_pathList[0].IsDirectory()));
378 if (bSingleFile)
379 cmd += m_pathList[0].GetWinPathString();
380 else
381 cmd += g_Git.CombinePath(m_pathList.GetCommonRoot().GetDirectory());
382 cmd += _T("\"");
383 CAppUtils::RunTortoiseGitProc(cmd);
386 void CChangedDlg::OnBnClickedStash()
388 CIconMenu popup;
390 if (popup.CreatePopupMenu())
392 popup.AppendMenuIcon(ID_STASH_SAVE, IDS_MENUSTASHSAVE, IDI_COMMIT);
394 CTGitPath root = g_Git.m_CurrentDir;
395 if (root.HasStashDir())
397 popup.AppendMenuIcon(ID_STASH_POP, IDS_MENUSTASHPOP, IDI_RELOCATE);
398 popup.AppendMenuIcon(ID_STASH_APPLY, IDS_MENUSTASHAPPLY, IDI_RELOCATE);
399 popup.AppendMenuIcon(ID_STASH_LIST, IDS_MENUSTASHLIST, IDI_LOG);
402 POINT cursorPos;
403 GetCursorPos(&cursorPos);
404 int cmd = popup.TrackPopupMenu(TPM_RETURNCMD | TPM_LEFTALIGN | TPM_NONOTIFY, cursorPos.x, cursorPos.y, this, 0);
406 switch (cmd & 0xFFFF)
408 case ID_STASH_SAVE:
409 CAppUtils::StashSave();
410 break;
411 case ID_STASH_POP:
412 CAppUtils::StashPop(false);
413 break;
414 case ID_STASH_APPLY:
415 CAppUtils::StashApply(_T(""), false);
416 break;
417 case ID_STASH_LIST:
419 CRefLogDlg dlg;
420 dlg.m_CurrentBranch = _T("refs/stash");
421 dlg.DoModal();
423 break;
424 default:
425 return;
427 OnBnClickedRefresh();
431 void CChangedDlg::OnBnClickedButtonUnifieddiff()
433 CTGitPath commonDirectory = m_bWholeProject ? g_Git.m_CurrentDir : m_pathList.GetCommonRoot().GetDirectory();
434 bool bSingleFile = ((m_pathList.GetCount()==1)&&(!m_pathList[0].IsEmpty())&&(!m_pathList[0].IsDirectory()));
435 if (bSingleFile)
436 commonDirectory = m_pathList[0];
437 CAppUtils::StartShowUnifiedDiff(m_hWnd, commonDirectory, GitRev::GetHead(), commonDirectory, GitRev::GetWorkingCopy());
440 void CChangedDlg::OnBnClickedWholeProject()
442 UpdateData();
443 m_regShowWholeProject = m_bWholeProject;
444 OnBnClickedRefresh();