src/TortoiseProc/AddDlg.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2008-2013 - TortoiseGit
   4 // Copyright (C) 2003-2008 - TortoiseSVN
   5
   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.
  10
  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.
  15
  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.
  19 //
  20 #include "stdafx.h"
  21 #include "TortoiseProc.h"
  22 #include "messagebox.h"
  23 #include "DirFileEnum.h"
  24 #include "AddDlg.h"
  25 //#include "SVNConfig.h"
  26 #include "registry.h"
  27 #include "AppUtils.h"
  28
  29 #define REFRESHTIMER   100
  30
  31 IMPLEMENT_DYNAMIC(CAddDlg, CResizableStandAloneDialog)
  32 CAddDlg::CAddDlg(CWnd* pParent /*=NULL*/)
  33         : CResizableStandAloneDialog(CAddDlg::IDD, pParent)
  34         , m_bThreadRunning(FALSE)
  35         , m_bCancelled(false)
  36         , m_bIncludeIgnored(FALSE)
  37 {
  38 }
  39
  40 CAddDlg::~CAddDlg()
  41 {
  42 }
  43
  44 void CAddDlg::DoDataExchange(CDataExchange* pDX)
  45 {
  46         CResizableStandAloneDialog::DoDataExchange(pDX);
  47         DDX_Control(pDX, IDC_ADDLIST, m_addListCtrl);
  48         DDX_Control(pDX, IDC_SELECTALL, m_SelectAll);
  49         DDX_Check(pDX, IDC_INCLUDE_IGNORED, m_bIncludeIgnored);
  50 }
  51
  52 BEGIN_MESSAGE_MAP(CAddDlg, CResizableStandAloneDialog)
  53         ON_BN_CLICKED(IDC_SELECTALL, OnBnClickedSelectall)
  54         ON_REGISTERED_MESSAGE(CGitStatusListCtrl::GITSLNM_NEEDSREFRESH, OnSVNStatusListCtrlNeedsRefresh)
  55         ON_REGISTERED_MESSAGE(CGitStatusListCtrl::GITSLNM_ADDFILE, OnFileDropped)
  56         ON_WM_TIMER()
  57         ON_BN_CLICKED(IDC_INCLUDE_IGNORED, &CAddDlg::OnBnClickedIncludeIgnored)
  58 END_MESSAGE_MAP()
  59
  60
  61 BOOL CAddDlg::OnInitDialog()
  62 {
  63         CResizableStandAloneDialog::OnInitDialog();
  64         CAppUtils::MarkWindowAsUnpinnable(m_hWnd);
  65
  66         // initialize the svn status list control
  67         m_addListCtrl.Init(GITSLC_COLEXT, _T("AddDlg"), GITSLC_POPALL ^ (GITSLC_POPADD|GITSLC_POPCOMMIT|GITSLC_POPCHANGELISTS), true, true, GITSLC_COLEXT | GITSLC_COLMODIFICATIONDATE | GITSLC_COLSIZE); // adding and committing is useless in the add dialog
  68         m_addListCtrl.SetIgnoreRemoveOnly();    // when ignoring, don't add the parent folder since we're in the add dialog
  69         m_addListCtrl.SetSelectButton(&m_SelectAll);
  70         m_addListCtrl.SetConfirmButton((CButton*)GetDlgItem(IDOK));
  71         m_addListCtrl.SetEmptyString(IDS_ERR_NOTHINGTOADD);
  72         m_addListCtrl.SetCancelBool(&m_bCancelled);
  73         m_addListCtrl.SetBackgroundImage(IDI_ADD_BKG);
  74         m_addListCtrl.EnableFileDrop();
  75
  76         CString sWindowTitle;
  77         GetWindowText(sWindowTitle);
  78         CAppUtils::SetWindowTitle(m_hWnd, g_Git.CombinePath(m_pathList.GetCommonRoot().GetUIPathString()), sWindowTitle);
  79
  80         AdjustControlSize(IDC_SELECTALL);
  81         AdjustControlSize(IDC_INCLUDE_IGNORED);
  82
  83         AddAnchor(IDC_ADDLIST, TOP_LEFT, BOTTOM_RIGHT);
  84         AddAnchor(IDC_SELECTALL, BOTTOM_LEFT);
  85         AddAnchor(IDC_INCLUDE_IGNORED, BOTTOM_LEFT);
  86         AddAnchor(IDOK, BOTTOM_RIGHT);
  87         AddAnchor(IDCANCEL, BOTTOM_RIGHT);
  88         AddAnchor(IDHELP, BOTTOM_RIGHT);
  89
  90         if (hWndExplorer)
  91                 CenterWindow(CWnd::FromHandle(hWndExplorer));
  92         EnableSaveRestore(_T("AddDlg"));
  93
  94         //first start a thread to obtain the file list with the status without
  95         //blocking the dialog
  96         if(AfxBeginThread(AddThreadEntry, this) == NULL)
  97         {
  98                 CMessageBox::Show(this->m_hWnd, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
  99         }
 100         InterlockedExchange(&m_bThreadRunning, TRUE);
 101
 102         return TRUE;
 103 }
 104
 105 void CAddDlg::OnOK()
 106 {
 107         if (m_bThreadRunning)
 108                 return;
 109
 110         // save only the files the user has selected into the path list
 111         m_addListCtrl.WriteCheckedNamesToPathList(m_pathList);
 112
 113         CResizableStandAloneDialog::OnOK();
 114 }
 115
 116 void CAddDlg::OnCancel()
 117 {
 118         m_bCancelled = true;
 119         if (m_bThreadRunning)
 120                 return;
 121
 122         CResizableStandAloneDialog::OnCancel();
 123 }
 124
 125 void CAddDlg::OnBnClickedSelectall()
 126 {
 127         UINT state = (m_SelectAll.GetState() & 0x0003);
 128         if (state == BST_INDETERMINATE)
 129         {
 130                 // It is not at all useful to manually place the checkbox into the indeterminate state...
 131                 // We will force this on to the unchecked state
 132                 state = BST_UNCHECKED;
 133                 m_SelectAll.SetCheck(state);
 134         }
 135         theApp.DoWaitCursor(1);
 136         m_addListCtrl.SelectAll(state == BST_CHECKED);
 137         theApp.DoWaitCursor(-1);
 138 }
 139
 140 UINT CAddDlg::AddThreadEntry(LPVOID pVoid)
 141 {
 142         return ((CAddDlg*)pVoid)->AddThread();
 143 }
 144
 145 UINT CAddDlg::AddThread()
 146 {
 147         // get the status of all selected file/folders recursively
 148         // and show the ones which the user can add (i.e. the unversioned ones)
 149         DialogEnableWindow(IDOK, false);
 150         m_bCancelled = false;
 151         m_addListCtrl.Clear();
 152         if (!m_addListCtrl.GetStatus(&m_pathList, false, m_bIncludeIgnored != FALSE, true))
 153         {
 154                 m_addListCtrl.SetEmptyString(m_addListCtrl.GetLastErrorMessage());
 155         }
 156         unsigned int dwShow = GITSLC_SHOWUNVERSIONED | GITSLC_SHOWDIRECTFILES | GITSLC_SHOWREMOVEDANDPRESENT;
 157         if (m_bIncludeIgnored)
 158                 dwShow |= GITSLC_SHOWIGNORED;
 159         m_addListCtrl.Show(dwShow);
 160
 161         InterlockedExchange(&m_bThreadRunning, FALSE);
 162         return 0;
 163 }
 164
 165 BOOL CAddDlg::PreTranslateMessage(MSG* pMsg)
 166 {
 167         if (pMsg->message == WM_KEYDOWN)
 168         {
 169                 switch (pMsg->wParam)
 170                 {
 171                 case VK_RETURN:
 172                         {
 173                                 if (GetAsyncKeyState(VK_CONTROL)&0x8000)
 174                                 {
 175                                         if ( GetDlgItem(IDOK)->IsWindowEnabled() )
 176                                         {
 177                                                 PostMessage(WM_COMMAND, IDOK);
 178                                         }
 179                                         return TRUE;
 180                                 }
 181                         }
 182                         break;
 183                 case VK_F5:
 184                         {
 185                                 Refresh();
 186                         }
 187                         break;
 188                 }
 189         }
 190
 191         return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
 192 }
 193
 194 LRESULT CAddDlg::OnSVNStatusListCtrlNeedsRefresh(WPARAM, LPARAM)
 195 {
 196         if(AfxBeginThread(AddThreadEntry, 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 CAddDlg::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_addListCtrl.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         CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": Item %s dropped, timer started\n"), path.GetWinPath());
 253         return 0;
 254 }
 255
 256 void CAddDlg::Refresh()
 257 {
 258         if (!m_bThreadRunning)
 259         {
 260                 if(AfxBeginThread(AddThreadEntry, this) == NULL)
 261                 {
 262                         CMessageBox::Show(this->m_hWnd, IDS_ERR_THREADSTARTFAILED, IDS_APPNAME, MB_OK | MB_ICONERROR);
 263                 }
 264                 else
 265                         InterlockedExchange(&m_bThreadRunning, TRUE);
 266         }
 267 }
 268
 269 void CAddDlg::OnTimer(UINT_PTR nIDEvent)
 270 {
 271         switch (nIDEvent)
 272         {
 273         case REFRESHTIMER:
 274                 if (m_bThreadRunning)
 275                 {
 276                         SetTimer(REFRESHTIMER, 200, NULL);
 277                         CTraceToOutputDebugString::Instance()(__FUNCTION__ ": Wait some more before refreshing\n");
 278                 }
 279                 else
 280                 {
 281                         KillTimer(REFRESHTIMER);
 282                         CTraceToOutputDebugString::Instance()(__FUNCTION__ ": Refreshing after items dropped\n");
 283                         OnSVNStatusListCtrlNeedsRefresh(0, 0);
 284                 }
 285                 break;
 286         }
 287         __super::OnTimer(nIDEvent);
 288 }
 289
 290 void CAddDlg::OnBnClickedIncludeIgnored()
 291 {
 292         UpdateData();
 293         Refresh();
 294 }