src/TortoiseProc/ResolveDlg.cpp

   1 // TortoiseSVN - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2008 - TortoiseSVN
   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 #include "stdafx.h"
  20 #include "TortoiseProc.h"
  21 #include "messagebox.h"
  22 #include "ResolveDlg.h"
  23 #include "CommonResource.h"
  24
  25 #define REFRESHTIMER   100
  26
  27 IMPLEMENT_DYNAMIC(CResolveDlg, CResizableStandAloneDialog)
  28 CResolveDlg::CResolveDlg(CWnd* pParent /*=NULL*/)
  29         : CResizableStandAloneDialog(CResolveDlg::IDD, pParent)
  30         , m_bThreadRunning(FALSE)
  31         , m_bCancelled(false)
  32 {
  33 }
  34
  35 CResolveDlg::~CResolveDlg()
  36 {
  37 }
  38
  39 void CResolveDlg::DoDataExchange(CDataExchange* pDX)
  40 {
  41         CResizableStandAloneDialog::DoDataExchange(pDX);
  42         DDX_Control(pDX, IDC_RESOLVELIST, m_resolveListCtrl);
  43         DDX_Control(pDX, IDC_SELECTALL, m_SelectAll);
  44 }
  45
  46
  47 BEGIN_MESSAGE_MAP(CResolveDlg, CResizableStandAloneDialog)
  48         ON_BN_CLICKED(IDC_SELECTALL, OnBnClickedSelectall)
  49         ON_BN_CLICKED(IDHELP, OnBnClickedHelp)
  50         ON_REGISTERED_MESSAGE(CGitStatusListCtrl::SVNSLNM_NEEDSREFRESH, OnSVNStatusListCtrlNeedsRefresh)
  51         ON_REGISTERED_MESSAGE(CGitStatusListCtrl::SVNSLNM_ADDFILE, OnFileDropped)
  52         ON_WM_TIMER()
  53 END_MESSAGE_MAP()
  54
  55 BOOL CResolveDlg::OnInitDialog()
  56 {
  57         CResizableStandAloneDialog::OnInitDialog();
  58
  59         m_resolveListCtrl.Init(SVNSLC_COLEXT | SVNSLC_COLTEXTSTATUS | SVNSLC_COLPROPSTATUS, _T("ResolveDlg"), SVNSLC_POPALL ^ (SVNSLC_POPIGNORE|SVNSLC_POPADD|SVNSLC_POPCOMMIT));
  60         m_resolveListCtrl.SetConfirmButton((CButton*)GetDlgItem(IDOK));
  61         m_resolveListCtrl.SetSelectButton(&m_SelectAll);
  62         m_resolveListCtrl.SetCancelBool(&m_bCancelled);
  63         m_resolveListCtrl.SetBackgroundImage(IDI_RESOLVE_BKG);
  64         m_resolveListCtrl.EnableFileDrop();
  65
  66         AdjustControlSize(IDC_SELECTALL);
  67
  68         AddAnchor(IDC_RESOLVELIST, TOP_LEFT, BOTTOM_RIGHT);
  69         AddAnchor(IDC_SELECTALL, BOTTOM_LEFT);
  70         AddAnchor(IDOK, BOTTOM_RIGHT);
  71         AddAnchor(IDCANCEL, BOTTOM_RIGHT);
  72         AddAnchor(IDHELP, BOTTOM_RIGHT);
  73         if (hWndExplorer)
  74                 CenterWindow(CWnd::FromHandle(hWndExplorer));
  75         EnableSaveRestore(_T("ResolveDlg"));
  76
  77         // first start a thread to obtain the file list with the status without
  78         // blocking the dialog
  79         if(AfxBeginThread(ResolveThreadEntry, this) == NULL)
  80         {
  81                 CMessageBox::Show(this->m_hWnd, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
  82         }
  83         InterlockedExchange(&m_bThreadRunning, TRUE);
  84
  85         return TRUE;
  86 }
  87
  88 void CResolveDlg::OnOK()
  89 {
  90         if (m_bThreadRunning)
  91                 return;
  92
  93         //save only the files the user has selected into the path list
  94         m_resolveListCtrl.WriteCheckedNamesToPathList(m_pathList);
  95
  96         CResizableStandAloneDialog::OnOK();
  97 }
  98
  99 void CResolveDlg::OnCancel()
 100 {
 101         m_bCancelled = true;
 102         if (m_bThreadRunning)
 103                 return;
 104
 105         CResizableStandAloneDialog::OnCancel();
 106 }
 107
 108 void CResolveDlg::OnBnClickedSelectall()
 109 {
 110         UINT state = (m_SelectAll.GetState() & 0x0003);
 111         if (state == BST_INDETERMINATE)
 112         {
 113                 // It is not at all useful to manually place the checkbox into the indeterminate state...
 114                 // We will force this on to the unchecked state
 115                 state = BST_UNCHECKED;
 116                 m_SelectAll.SetCheck(state);
 117         }
 118         theApp.DoWaitCursor(1);
 119         m_resolveListCtrl.SelectAll(state == BST_CHECKED);
 120         theApp.DoWaitCursor(-1);
 121 }
 122
 123 UINT CResolveDlg::ResolveThreadEntry(LPVOID pVoid)
 124 {
 125         return ((CResolveDlg*)pVoid)->ResolveThread();
 126 }
 127 UINT CResolveDlg::ResolveThread()
 128 {
 129         // get the status of all selected file/folders recursively
 130         // and show the ones which are in conflict
 131         DialogEnableWindow(IDOK, false);
 132
 133         m_bCancelled = false;
 134
 135         if (!m_resolveListCtrl.GetStatus(&m_pathList))
 136         {
 137                 m_resolveListCtrl.SetEmptyString(m_resolveListCtrl.GetLastErrorMessage());
 138         }
 139         m_resolveListCtrl.Show(SVNSLC_SHOWCONFLICTED|SVNSLC_SHOWINEXTERNALS, SVNSLC_SHOWCONFLICTED);
 140
 141         InterlockedExchange(&m_bThreadRunning, FALSE);
 142         return 0;
 143 }
 144
 145 void CResolveDlg::OnBnClickedHelp()
 146 {
 147         OnHelp();
 148 }
 149
 150 BOOL CResolveDlg::PreTranslateMessage(MSG* pMsg)
 151 {
 152         if (pMsg->message == WM_KEYDOWN)
 153         {
 154                 switch (pMsg->wParam)
 155                 {
 156                 case VK_RETURN:
 157                         {
 158                                 if (GetAsyncKeyState(VK_CONTROL)&0x8000)
 159                                 {
 160                                         if ( GetDlgItem(IDOK)->IsWindowEnabled() )
 161                                         {
 162                                                 PostMessage(WM_COMMAND, IDOK);
 163                                         }
 164                                         return TRUE;
 165                                 }
 166                         }
 167                         break;
 168                 case VK_F5:
 169                         {
 170                                 if (!m_bThreadRunning)
 171                                 {
 172                                         if(AfxBeginThread(ResolveThreadEntry, this) == NULL)
 173                                         {
 174                                                 CMessageBox::Show(this->m_hWnd, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
 175                                         }
 176                                         else
 177                                                 InterlockedExchange(&m_bThreadRunning, TRUE);
 178                                 }
 179                         }
 180                         break;
 181                 }
 182         }
 183
 184         return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
 185 }
 186
 187 LRESULT CResolveDlg::OnSVNStatusListCtrlNeedsRefresh(WPARAM, LPARAM)
 188 {
 189         if(AfxBeginThread(ResolveThreadEntry, this) == NULL)
 190         {
 191                 CMessageBox::Show(this->m_hWnd, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
 192         }
 193         return 0;
 194 }
 195
 196 LRESULT CResolveDlg::OnFileDropped(WPARAM, LPARAM lParam)
 197 {
 198         BringWindowToTop();
 199         SetForegroundWindow();
 200         SetActiveWindow();
 201         // if multiple files/folders are dropped
 202         // this handler is called for every single item
 203         // separately.
 204         // To avoid creating multiple refresh threads and
 205         // causing crashes, we only add the items to the
 206         // list control and start a timer.
 207         // When the timer expires, we start the refresh thread,
 208         // but only if it isn't already running - otherwise we
 209         // restart the timer.
 210         CTGitPath path;
 211         path.SetFromWin((LPCTSTR)lParam);
 212
 213         if (!m_resolveListCtrl.HasPath(path))
 214         {
 215                 if (m_pathList.AreAllPathsFiles())
 216                 {
 217                         m_pathList.AddPath(path);
 218                         m_pathList.RemoveDuplicates();
 219                 }
 220                 else
 221                 {
 222                         // if the path list contains folders, we have to check whether
 223                         // our just (maybe) added path is a child of one of those. If it is
 224                         // a child of a folder already in the list, we must not add it. Otherwise
 225                         // that path could show up twice in the list.
 226                         bool bHasParentInList = false;
 227                         for (int i=0; i<m_pathList.GetCount(); ++i)
 228                         {
 229                                 if (m_pathList[i].IsAncestorOf(path))
 230                                 {
 231                                         bHasParentInList = true;
 232                                         break;
 233                                 }
 234                         }
 235                         if (!bHasParentInList)
 236                         {
 237                                 m_pathList.AddPath(path);
 238                                 m_pathList.RemoveDuplicates();
 239                         }
 240                 }
 241         }
 242
 243         // Always start the timer, since the status of an existing item might have changed
 244         SetTimer(REFRESHTIMER, 200, NULL);
 245         ATLTRACE(_T("Item %s dropped, timer started\n"), path.GetWinPath());
 246         return 0;
 247 }
 248
 249 void CResolveDlg::OnTimer(UINT_PTR nIDEvent)
 250 {
 251         switch (nIDEvent)
 252         {
 253         case REFRESHTIMER:
 254                 if (m_bThreadRunning)
 255                 {
 256                         SetTimer(REFRESHTIMER, 200, NULL);
 257                         ATLTRACE("Wait some more before refreshing\n");
 258                 }
 259                 else
 260                 {
 261                         KillTimer(REFRESHTIMER);
 262                         ATLTRACE("Refreshing after items dropped\n");
 263                         OnSVNStatusListCtrlNeedsRefresh(0, 0);
 264                 }
 265                 break;
 266         }
 267         __super::OnTimer(nIDEvent);
 268 }