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