src/TortoiseProc/MassiveGitTask.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2011-2012 - Sven Strickroth <email@cs-ware.de>
   4 // Copyright (C) 2013 - TortoiseGit
   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
  21 #include "stdafx.h"
  22 #include "TortoiseProc.h"
  23 #include "MassiveGitTask.h"
  24 #include "MessageBox.h"
  25
  26 CMassiveGitTask::CMassiveGitTask(CString gitParameters, BOOL isPath)
  27         : m_bUnused(true)
  28         , m_bIsPath(isPath)
  29         , m_NotifyCallbackInstance(NULL)
  30         , m_NotifyCallbackMethod(NULL)
  31         , m_NotifyCallbackAction(git_wc_notify_add)
  32 {
  33         m_sParams = gitParameters;
  34 }
  35
  36 CMassiveGitTask::~CMassiveGitTask(void)
  37 {
  38 }
  39
  40 void CMassiveGitTask::AddFile(CString filename)
  41 {
  42         assert(m_bUnused);
  43         if (m_bIsPath)
  44                 m_pathList.AddPath(filename);
  45         else
  46                 m_itemList.push_back(filename);
  47 }
  48
  49 void CMassiveGitTask::AddFile(CTGitPath filename)
  50 {
  51         assert(m_bUnused);
  52         if (m_bIsPath)
  53                 m_pathList.AddPath(filename);
  54         else
  55                 m_itemList.push_back(filename.GetGitPathString());
  56 }
  57
  58 bool CMassiveGitTask::ExecuteWithNotify(CTGitPathList *pathList, BOOL &cancel, git_wc_notify_action_t action, CGitProgressDlg * instance, NOTIFY_CALLBACK notifyMethod)
  59 {
  60         assert(m_bUnused);
  61         m_bUnused = false;
  62
  63         m_pathList = *pathList;
  64         m_NotifyCallbackInstance = instance;
  65         m_NotifyCallbackMethod = notifyMethod;
  66         m_NotifyCallbackAction = action;
  67         return ExecuteCommands(cancel);
  68 }
  69
  70 bool CMassiveGitTask::Execute(BOOL &cancel)
  71 {
  72         assert(m_bUnused);
  73         m_pathList.RemoveDuplicates();
  74         return ExecuteCommands(cancel);
  75 }
  76
  77 bool CMassiveGitTask::ExecuteCommands(BOOL &cancel)
  78 {
  79         m_bUnused = false;
  80
  81         bool bErrorsOccurred = false;
  82         int maxLength = 0;
  83         int firstCombine = 0;
  84         for(int i = 0; i < GetListCount(); i++)
  85         {
  86                 if (maxLength + GetListItem(i).GetLength() > MAX_COMMANDLINE_LENGTH || i == GetListCount() - 1 || cancel)
  87                 {
  88                         CString add;
  89                         for (int j = firstCombine; j <= i; j++)
  90                                 add += _T(" \"") + GetListItem(j) + _T("\"");
  91
  92                         CString cmd, out;
  93                         cmd.Format(_T("git.exe %s %s%s"), m_sParams, m_bIsPath ? _T("--") : _T(""), add);
  94                         if (g_Git.Run(cmd, &out, CP_UTF8))
  95                         {
  96                                 CMessageBox::Show(NULL, out, _T("TortoiseGit"), MB_OK|MB_ICONERROR);
  97                                 bErrorsOccurred = true;
  98                                 return false;
  99                         }
 100
 101                         if (m_bIsPath)
 102                                 if (m_NotifyCallbackInstance && m_NotifyCallbackMethod)
 103                                         for (int j = firstCombine; j <= i; j++)
 104                                                 (*m_NotifyCallbackInstance.*m_NotifyCallbackMethod)(m_pathList[j], m_NotifyCallbackAction, 0, NULL);
 105
 106                         maxLength = 0;
 107                         firstCombine = i+1;
 108
 109                         if (cancel)
 110                                 break;
 111                 }
 112                 else
 113                 {
 114                         maxLength += 3 + GetListItem(i).GetLength();
 115                 }
 116         }
 117         return !bErrorsOccurred;
 118 }
 119
 120 int CMassiveGitTask::GetListCount()
 121 {
 122         return m_bIsPath ? m_pathList.GetCount() : (int)m_itemList.size();
 123 }
 124
 125 CString CMassiveGitTask::GetListItem(int index)
 126 {
 127         return m_bIsPath ? m_pathList[index].GetGitPathString() : m_itemList[index];
 128 }