1 // TortoiseGit - a Windows shell extension for easy version control
3 // Copyright (C) 2009-2012 - 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.
21 #include "TortoiseProc.h"
22 #include "messagebox.h"
23 #include "Revertdlg.h"
26 #include ".\revertdlg.h"
29 #define REFRESHTIMER 100
31 IMPLEMENT_DYNAMIC(CRevertDlg
, CResizableStandAloneDialog
)
32 CRevertDlg::CRevertDlg(CWnd
* pParent
/*=NULL*/)
33 : CResizableStandAloneDialog(CRevertDlg::IDD
, pParent
)
35 , m_bThreadRunning(FALSE
)
40 CRevertDlg::~CRevertDlg()
44 void CRevertDlg::DoDataExchange(CDataExchange
* pDX
)
46 CResizableStandAloneDialog::DoDataExchange(pDX
);
47 DDX_Control(pDX
, IDC_REVERTLIST
, m_RevertList
);
48 DDX_Check(pDX
, IDC_SELECTALL
, m_bSelectAll
);
49 DDX_Control(pDX
, IDC_SELECTALL
, m_SelectAll
);
53 BEGIN_MESSAGE_MAP(CRevertDlg
, CResizableStandAloneDialog
)
54 ON_BN_CLICKED(IDHELP
, OnBnClickedHelp
)
55 ON_BN_CLICKED(IDC_SELECTALL
, OnBnClickedSelectall
)
56 ON_REGISTERED_MESSAGE(CGitStatusListCtrl::GITSLNM_NEEDSREFRESH
, OnSVNStatusListCtrlNeedsRefresh
)
57 ON_REGISTERED_MESSAGE(CGitStatusListCtrl::GITSLNM_ADDFILE
, OnFileDropped
)
63 BOOL
CRevertDlg::OnInitDialog()
65 CResizableStandAloneDialog::OnInitDialog();
66 CAppUtils::MarkWindowAsUnpinnable(m_hWnd
);
68 m_RevertList
.Init(GITSLC_COLEXT
| GITSLC_COLSTATUS
| GITSLC_COLADD
| GITSLC_COLDEL
, _T("RevertDlg"));
69 m_RevertList
.SetConfirmButton((CButton
*)GetDlgItem(IDOK
));
70 m_RevertList
.SetSelectButton(&m_SelectAll
);
71 m_RevertList
.SetCancelBool(&m_bCancelled
);
72 m_RevertList
.SetBackgroundImage(IDI_REVERT_BKG
);
73 m_RevertList
.EnableFileDrop();
76 GetWindowText(sWindowTitle
);
77 if (m_pathList
.GetCount() == 1)
78 CAppUtils::SetWindowTitle(m_hWnd
, (g_Git
.m_CurrentDir
+ _T("\\") + m_pathList
[0].GetUIPathString()).TrimRight('\\'), sWindowTitle
);
80 CAppUtils::SetWindowTitle(m_hWnd
, m_pathList
.GetCommonRoot().GetUIPathString(), sWindowTitle
);
82 AdjustControlSize(IDC_SELECTALL
);
84 AddAnchor(IDC_REVERTLIST
, TOP_LEFT
, BOTTOM_RIGHT
);
85 AddAnchor(IDC_SELECTALL
, BOTTOM_LEFT
);
86 AddAnchor(IDC_UNVERSIONEDITEMS
, BOTTOM_RIGHT
);
87 AddAnchor(IDOK
, BOTTOM_RIGHT
);
88 AddAnchor(IDCANCEL
, BOTTOM_RIGHT
);
89 AddAnchor(IDHELP
, BOTTOM_RIGHT
);
91 CenterWindow(CWnd::FromHandle(hWndExplorer
));
92 EnableSaveRestore(_T("RevertDlg"));
94 // first start a thread to obtain the file list with the status without
95 // blocking the dialog
96 if (AfxBeginThread(RevertThreadEntry
, this)==0)
98 CMessageBox::Show(this->m_hWnd
, IDS_ERR_THREADSTARTFAILED
, IDS_APPNAME
, MB_OK
| MB_ICONERROR
);
100 InterlockedExchange(&m_bThreadRunning
, TRUE
);
105 UINT
CRevertDlg::RevertThreadEntry(LPVOID pVoid
)
107 return ((CRevertDlg
*)pVoid
)->RevertThread();
110 UINT
CRevertDlg::RevertThread()
112 // get the status of all selected file/folders recursively
113 // and show the ones which can be reverted to the user
114 // in a list control.
115 DialogEnableWindow(IDOK
, false);
116 m_bCancelled
= false;
118 m_RevertList
.Clear();
120 if (!m_RevertList
.GetStatus(&m_pathList
))
122 m_RevertList
.SetEmptyString(m_RevertList
.GetLastErrorMessage());
124 m_RevertList
.Show(GITSLC_SHOWVERSIONEDBUTNORMALANDEXTERNALSFROMDIFFERENTREPOS
| GITSLC_SHOWDIRECTFILES
| GITSLC_SHOWEXTERNALFROMDIFFERENTREPO
,
125 // do not select all files, only the ones the user has selected directly
126 GITSLC_SHOWDIRECTFILES
|GITSLC_SHOWADDED
);
128 if (m_RevertList
.HasUnversionedItems())
130 if (DWORD(CRegStdDWORD(_T("Software\\TortoiseGit\\UnversionedAsModified"), FALSE
)))
132 GetDlgItem(IDC_UNVERSIONEDITEMS
)->ShowWindow(SW_SHOW
);
135 GetDlgItem(IDC_UNVERSIONEDITEMS
)->ShowWindow(SW_HIDE
);
138 GetDlgItem(IDC_UNVERSIONEDITEMS
)->ShowWindow(SW_HIDE
);
140 InterlockedExchange(&m_bThreadRunning
, FALSE
);
146 void CRevertDlg::OnOK()
148 if (m_bThreadRunning
)
150 // save only the files the user has selected into the temporary file
152 for (int i
=0; i
<m_RevertList
.GetItemCount(); ++i
)
154 if (!m_RevertList
.GetCheck(i
))
156 m_bRecursive
= FALSE
;
160 m_selectedPathList
.AddPath(*(CTGitPath
*)m_RevertList
.GetItemData(i
));
162 CGitStatusListCtrl::FileEntry
* entry
= m_RevertList
.GetListEntry(i
);
163 // add all selected entries to the list, except the ones with 'added'
164 // status: we later *delete* all the entries in the list before
165 // the actual revert is done (so the user has the reverted files
166 // still in the trash bin to recover from), but it's not good to
167 // delete added files because they're not restored by the revert.
168 if (entry
->status
!= svn_wc_status_added
)
169 m_selectedPathList
.AddPath(entry
->GetPath());
170 // if an entry inside an external is selected, we can't revert
171 // recursively anymore because the recursive revert stops at the
172 // external boundaries.
173 if (entry
->IsInExternal())
174 m_bRecursive
= FALSE
;
180 m_RevertList
.WriteCheckedNamesToPathList(m_pathList
);
182 m_selectedPathList
.SortByPathname();
184 CResizableStandAloneDialog::OnOK();
187 void CRevertDlg::OnCancel()
190 if (m_bThreadRunning
)
193 CResizableStandAloneDialog::OnCancel();
196 void CRevertDlg::OnBnClickedHelp()
201 void CRevertDlg::OnBnClickedSelectall()
203 UINT state
= (m_SelectAll
.GetState() & 0x0003);
204 if (state
== BST_INDETERMINATE
)
206 // It is not at all useful to manually place the checkbox into the indeterminate state...
207 // We will force this on to the unchecked state
208 state
= BST_UNCHECKED
;
209 m_SelectAll
.SetCheck(state
);
211 theApp
.DoWaitCursor(1);
212 m_RevertList
.SelectAll(state
== BST_CHECKED
);
213 theApp
.DoWaitCursor(-1);
216 BOOL
CRevertDlg::PreTranslateMessage(MSG
* pMsg
)
218 if (pMsg
->message
== WM_KEYDOWN
)
220 switch (pMsg
->wParam
)
224 if (GetAsyncKeyState(VK_CONTROL
)&0x8000)
226 if ( GetDlgItem(IDOK
)->IsWindowEnabled() )
228 PostMessage(WM_COMMAND
, IDOK
);
236 if (!m_bThreadRunning
)
238 if (AfxBeginThread(RevertThreadEntry
, this)==0)
240 CMessageBox::Show(this->m_hWnd
, IDS_ERR_THREADSTARTFAILED
, IDS_APPNAME
, MB_OK
| MB_ICONERROR
);
243 InterlockedExchange(&m_bThreadRunning
, TRUE
);
250 return CResizableStandAloneDialog::PreTranslateMessage(pMsg
);
253 LRESULT
CRevertDlg::OnSVNStatusListCtrlNeedsRefresh(WPARAM
, LPARAM
)
255 if (AfxBeginThread(RevertThreadEntry
, this)==0)
257 CMessageBox::Show(this->m_hWnd
, IDS_ERR_THREADSTARTFAILED
, IDS_APPNAME
, MB_OK
| MB_ICONERROR
);
262 LRESULT
CRevertDlg::OnFileDropped(WPARAM
, LPARAM lParam
)
265 SetForegroundWindow();
267 // if multiple files/folders are dropped
268 // this handler is called for every single item
270 // To avoid creating multiple refresh threads and
271 // causing crashes, we only add the items to the
272 // list control and start a timer.
273 // When the timer expires, we start the refresh thread,
274 // but only if it isn't already running - otherwise we
275 // restart the timer.
277 path
.SetFromWin((LPCTSTR
)lParam
);
279 if (!m_RevertList
.HasPath(path
))
281 if (m_pathList
.AreAllPathsFiles())
283 m_pathList
.AddPath(path
);
284 m_pathList
.RemoveDuplicates();
288 // if the path list contains folders, we have to check whether
289 // our just (maybe) added path is a child of one of those. If it is
290 // a child of a folder already in the list, we must not add it. Otherwise
291 // that path could show up twice in the list.
292 bool bHasParentInList
= false;
293 for (int i
=0; i
<m_pathList
.GetCount(); ++i
)
295 if (m_pathList
[i
].IsAncestorOf(path
))
297 bHasParentInList
= true;
301 if (!bHasParentInList
)
303 m_pathList
.AddPath(path
);
304 m_pathList
.RemoveDuplicates();
309 // Always start the timer, since the status of an existing item might have changed
310 SetTimer(REFRESHTIMER
, 200, NULL
);
311 ATLTRACE(_T("Item %s dropped, timer started\n"), path
.GetWinPath());
315 void CRevertDlg::OnTimer(UINT_PTR nIDEvent
)
320 if (m_bThreadRunning
)
322 SetTimer(REFRESHTIMER
, 200, NULL
);
323 ATLTRACE("Wait some more before refreshing\n");
327 KillTimer(REFRESHTIMER
);
328 ATLTRACE("Refreshing after items dropped\n");
329 OnSVNStatusListCtrlNeedsRefresh(0, 0);
333 __super::OnTimer(nIDEvent
);