src/TortoiseProc/ResolveDlg.cpp

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