src/TortoiseProc/ProgressCommands/AddProgressCommand.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2011-2016 - TortoiseGit
   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 "AddProgressCommand.h"
  21 #include "ShellUpdater.h"
  22 #include "MassiveGitTask.h"
  23 #include "AppUtils.h"
  24
  25 bool AddProgressCommand::Run(CGitProgressList* list, CString& sWindowTitle, int& m_itemCountTotal, int& m_itemCount)
  26 {
  27         list->SetWindowTitle(IDS_PROGRS_TITLE_ADD, g_Git.CombinePath(m_targetPathList.GetCommonRoot().GetUIPathString()), sWindowTitle);
  28         list->SetBackgroundImage(IDI_ADD_BKG);
  29         if (m_bExecutable)
  30                 list->ReportCmd(CString(MAKEINTRESOURCE(IDS_STATUSLIST_CONTEXT_ADD_EXE)));
  31         else if (m_bSymlink)
  32                 list->ReportCmd(CString(MAKEINTRESOURCE(IDS_STATUSLIST_CONTEXT_ADD_LINK)));
  33         else
  34                 list->ReportCmd(CString(MAKEINTRESOURCE(IDS_PROGRS_CMD_ADD)));
  35
  36         m_itemCountTotal = m_targetPathList.GetCount();
  37
  38         if (g_Git.UsingLibGit2(CGit::GIT_CMD_ADD))
  39         {
  40                 CAutoRepository repo(g_Git.GetGitRepository());
  41                 if (!repo)
  42                 {
  43                         list->ReportGitError();
  44                         return false;
  45                 }
  46
  47                 CAutoIndex index;
  48                 if (git_repository_index(index.GetPointer(), repo))
  49                 {
  50                         list->ReportGitError();
  51                         return false;
  52                 }
  53                 if (git_index_read(index, true))
  54                 {
  55                         list->ReportGitError();
  56                         return false;
  57                 }
  58
  59                 for (m_itemCount = 0; m_itemCount < m_itemCountTotal; ++m_itemCount)
  60                 {
  61                         CStringA filePathA = CUnicodeUtils::GetMulti(m_targetPathList[m_itemCount].GetGitPathString(), CP_UTF8).TrimRight(L'/');
  62                         if (git_index_add_bypath(index, filePathA))
  63                         {
  64                                 list->ReportGitError();
  65                                 return false;
  66                         }
  67
  68                         if (!m_targetPathList[m_itemCount].IsDirectory() && (m_bExecutable || m_bSymlink))
  69                         {
  70                                 auto entry = const_cast<git_index_entry*>(git_index_get_bypath(index, filePathA, 0));
  71                                 if (m_bExecutable)
  72                                         entry->mode = GIT_FILEMODE_BLOB_EXECUTABLE;
  73                                 else if (m_bSymlink)
  74                                         entry->mode = GIT_FILEMODE_LINK;
  75                                 if (git_index_add(index, entry))
  76                                 {
  77                                         list->ReportGitError();
  78                                         return false;
  79                                 }
  80                         }
  81
  82                         list->AddNotify(new CGitProgressList::WC_File_NotificationData(m_targetPathList[m_itemCount], CGitProgressList::WC_File_NotificationData::git_wc_notify_add));
  83
  84                         if (list->IsCancelled() == TRUE)
  85                         {
  86                                 list->ReportUserCanceled();
  87                                 return false;
  88                         }
  89                 }
  90
  91                 if (git_index_write(index))
  92                 {
  93                         list->ReportGitError();
  94                         return false;
  95                 }
  96         }
  97         else
  98         {
  99                 CMassiveGitTask mgt(L"add -f");
 100                 if (!mgt.ExecuteWithNotify(&m_targetPathList, list->m_bCancelled, CGitProgressList::WC_File_NotificationData::git_wc_notify_add, list))
 101                         return false;
 102                 if (m_bExecutable)
 103                 {
 104                         if (!SetFileMode(GIT_FILEMODE_BLOB_EXECUTABLE))
 105                                 return false;
 106                 }
 107                 else if (m_bSymlink)
 108                 {
 109                         if (!SetFileMode(GIT_FILEMODE_LINK))
 110                                 return false;
 111                 }
 112         }
 113
 114         CShellUpdater::Instance().AddPathsForUpdate(m_targetPathList);
 115
 116         m_PostCmdCallback = [this](DWORD status, PostCmdList& postCmdList)
 117         {
 118                 if (status)
 119                         return;
 120
 121                 if (m_bShowCommitButtonAfterAdd)
 122                         postCmdList.emplace_back(IDI_COMMIT, IDS_MENUCOMMIT, []
 123                         {
 124                                 CString sCmd;
 125                                 sCmd.Format(L"/command:commit /path:\"%s\"", (LPCTSTR)g_Git.m_CurrentDir);
 126                                 CAppUtils::RunTortoiseGitProc(sCmd);
 127                         });
 128                 postCmdList.emplace_back(IDI_ADD, IDS_STATUSLIST_CONTEXT_ADD_EXE, [this] {
 129                         SetFileMode(GIT_FILEMODE_BLOB_EXECUTABLE);
 130                 });
 131                 postCmdList.emplace_back(IDI_ADD, IDS_STATUSLIST_CONTEXT_ADD_LINK, [this] {
 132                         SetFileMode(GIT_FILEMODE_LINK);
 133                 });
 134         };
 135
 136         return true;
 137 }
 138
 139 bool AddProgressCommand::SetFileMode(uint32_t mode)
 140 {
 141         CAutoRepository repo(g_Git.GetGitRepository());
 142         if (!repo)
 143         {
 144                 MessageBox(nullptr, g_Git.GetLibGit2LastErr(L"Could not open repository."), L"TortoiseGit", MB_ICONERROR);
 145                 return false;
 146         }
 147
 148         CAutoIndex index;
 149         if (git_repository_index(index.GetPointer(), repo))
 150         {
 151                 MessageBox(nullptr, g_Git.GetLibGit2LastErr(L"Could not get index."), L"TortoiseGit", MB_ICONERROR);
 152                 return false;
 153         }
 154         if (git_index_read(index, true))
 155         {
 156                 MessageBox(nullptr, g_Git.GetLibGit2LastErr(L"Could not read index."), L"TortoiseGit", MB_ICONERROR);
 157                 return false;
 158         }
 159
 160         for (int i = 0; i < m_targetPathList.GetCount(); ++i)
 161         {
 162                 if (m_targetPathList[i].IsDirectory())
 163                         continue;
 164                 CStringA filePathA = CUnicodeUtils::GetMulti(m_targetPathList[i].GetGitPathString(), CP_UTF8).TrimRight(L'/');
 165                 auto entry = const_cast<git_index_entry*>(git_index_get_bypath(index, filePathA, 0));
 166                 entry->mode = mode;
 167                 if (git_index_add(index, entry))
 168                 {
 169                         MessageBox(nullptr, g_Git.GetLibGit2LastErr(L"Could not update index."), L"TortoiseGit", MB_ICONERROR);
 170                         return false;
 171                 }
 172         }
 173
 174         if (git_index_write(index))
 175         {
 176                 MessageBox(nullptr, g_Git.GetLibGit2LastErr(L"Could not write index."), L"TortoiseGit", MB_ICONERROR);
 177                 return false;
 178         }
 179
 180         return true;
 181 }