src/TortoiseProc/LogDataVector.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2009-2013 - TortoiseGit
   4 // Copyright (C) 2007-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 /*
  21         Description: start-up repository opening and reading
  22
  23         Author: Marco Costalba (C) 2005-2007
  24
  25         Copyright: See COPYING file that comes with this distribution
  26
  27 */
  28
  29 #include "stdafx.h"
  30 #include "TortoiseProc.h"
  31 #include "GitLogListBase.h"
  32 #include "UnicodeUtils.h"
  33
  34 typedef CComCritSecLock<CComCriticalSection> CAutoLocker;
  35
  36 void CLogDataVector::ClearAll()
  37 {
  38
  39         clear();
  40         m_HashMap.clear();
  41         m_Lns.clear();
  42
  43         m_FirstFreeLane=0;
  44         m_Lns.clear();
  45         if (m_pLogCache)
  46         {
  47                 m_pLogCache->ClearAllLanes();
  48         }
  49
  50         m_RawlogData.clear();
  51         m_RawLogStart.clear();
  52 }
  53
  54 //CLogDataVector Class
  55 int CLogDataVector::ParserFromLog(CTGitPath *path, int count, int infomask, CString *range)
  56 {
  57         // only enable --follow on files
  58         if ((path == NULL || path->IsDirectory()) && (infomask & CGit::LOG_INFO_FOLLOW))
  59                 infomask = infomask ^ CGit::LOG_INFO_FOLLOW;
  60
  61         CString gitrange = _T("HEAD");
  62         if (range != nullptr)
  63                 gitrange = *range;
  64         CString cmd = g_Git.GetLogCmd(gitrange, path, count, infomask, true);
  65
  66         if (!g_Git.CanParseRev(gitrange))
  67                 return 0;
  68
  69         try
  70         {
  71                 [] { git_init(); } ();
  72         }
  73         catch (const char* msg)
  74         {
  75                 MessageBox(NULL, _T("Could not initialize libgit.\nlibgit reports:\n") + CString(msg), _T("TortoiseGit"), MB_ICONERROR);
  76                 return -1;
  77         }
  78
  79         GIT_LOG handle;
  80         try
  81         {
  82                 CAutoLocker lock(g_Git.m_critGitDllSec);
  83                 if (git_open_log(&handle,CUnicodeUtils::GetMulti(cmd, CP_UTF8).GetBuffer()))
  84                 {
  85                         return -1;
  86                 }
  87         }
  88         catch (char* msg)
  89         {
  90                 MessageBox(NULL, _T("Could not open log.\nlibgit reports:\n") + CString(msg), _T("TortoiseGit"), MB_ICONERROR);
  91                 return -1;
  92         }
  93
  94         try
  95         {
  96                 CAutoLocker lock(g_Git.m_critGitDllSec);
  97                 [&]{ git_get_log_firstcommit(handle); }();
  98         }
  99         catch (char* msg)
 100         {
 101                 MessageBox(NULL, _T("Could not get first commit.\nlibgit reports:\n") + CString(msg), _T("TortoiseGit"), MB_ICONERROR);
 102                 return -1;
 103         }
 104
 105         int ret = 0;
 106         while (ret == 0)
 107         {
 108                 GIT_COMMIT commit;
 109
 110                 try
 111                 {
 112                         CAutoLocker lock(g_Git.m_critGitDllSec);
 113                         [&]{ ret = git_get_log_nextcommit(handle, &commit, infomask & CGit::LOG_INFO_FOLLOW); }();
 114                 }
 115                 catch (char* msg)
 116                 {
 117                         MessageBox(NULL, _T("Could not get next commit.\nlibgit reports:\n") + CString(msg), _T("TortoiseGit"), MB_ICONERROR);
 118                         break;
 119                 }
 120
 121                 if (ret)
 122                 {
 123                         if (ret != -2) // other than end of revision walking
 124                                 MessageBox(nullptr, (_T("Could not get next commit.\nlibgit returns:") + std::to_wstring(ret)).c_str(), _T("TortoiseGit"), MB_ICONERROR);
 125                         break;
 126                 }
 127
 128                 if (commit.m_ignore == 1)
 129                 {
 130                         git_free_commit(&commit);
 131                         continue;
 132                 }
 133
 134                 CGitHash hash = (char*)commit.m_hash ;
 135
 136                 GitRev *pRev = this->m_pLogCache->GetCacheData(hash);
 137
 138                 char *pNote = nullptr;
 139                 {
 140                         CAutoLocker lock(g_Git.m_critGitDllSec);
 141                         git_get_notes(commit.m_hash, &pNote);
 142                 }
 143                 if (pNote)
 144                 {
 145                         pRev->m_Notes.Empty();
 146                         g_Git.StringAppend(&pRev->m_Notes,(BYTE*)pNote);
 147                         free(pNote);
 148                         pNote = nullptr;
 149                 }
 150
 151                 ASSERT(pRev->m_CommitHash == hash);
 152                 pRev->ParserFromCommit(&commit);
 153                 pRev->ParserParentFromCommit(&commit);
 154                 git_free_commit(&commit);
 155                 // Must call free commit before SafeFetchFullInfo, commit parent is rewrite by log.
 156                 // file list will wrong if parent rewrite.
 157
 158                 if (!pRev->m_IsFull && (infomask & CGit::LOG_INFO_FULL_DIFF))
 159                 {
 160                         try
 161                         {
 162                                 pRev->SafeFetchFullInfo(&g_Git);
 163                         }
 164                         catch (char * g_last_error)
 165                         {
 166                                 MessageBox(NULL, _T("Could not fetch full info of a commit.\nlibgit reports:\n") + CString(g_last_error), _T("TortoiseGit"), MB_ICONERROR);
 167                                 return -1;
 168                         }
 169                 }
 170
 171                 this->push_back(pRev->m_CommitHash);
 172
 173                 m_HashMap[pRev->m_CommitHash] = (int)size() - 1;
 174
 175         }
 176
 177         {
 178                 CAutoLocker lock(g_Git.m_critGitDllSec);
 179                 git_close_log(handle);
 180         }
 181
 182         return 0;
 183 }
 184
 185 struct SortByParentDate
 186 {
 187         bool operator()(GitRev *pLhs, GitRev *pRhs)
 188         {
 189                 if (pLhs->m_CommitHash == pRhs->m_CommitHash)
 190                         return false;
 191                 for (auto it = pLhs->m_ParentHash.begin(); it != pLhs->m_ParentHash.end(); ++it)
 192                 {
 193                         if (*it == pRhs->m_CommitHash)
 194                                 return true;
 195                 }
 196                 for (auto it = pRhs->m_ParentHash.begin(); it != pRhs->m_ParentHash.end(); ++it)
 197                 {
 198                         if (*it == pLhs->m_CommitHash)
 199                                 return false;
 200                 }
 201                 return pLhs->GetCommitterDate()>pRhs->GetCommitterDate();
 202         }
 203 };
 204
 205 int CLogDataVector::Fill(std::set<CGitHash>& hashes)
 206 {
 207         try
 208         {
 209                 [] { git_init(); } ();
 210         }
 211         catch (const char* msg)
 212         {
 213                 MessageBox(nullptr, _T("Could not initialize libgit.\nlibgit reports:\n") + CString(msg), _T("TortoiseGit"), MB_ICONERROR);
 214                 return -1;
 215         }
 216
 217         std::set<GitRev*, SortByParentDate> revs;
 218
 219         for (auto it = hashes.begin(); it != hashes.end(); ++it)
 220         {
 221                 CGitHash hash = *it;
 222                 GitRev *pRev = this->m_pLogCache->GetCacheData(hash);
 223                 ASSERT(pRev->m_CommitHash == hash);
 224
 225                 GIT_COMMIT commit;
 226                 try
 227                 {
 228                         CAutoLocker lock(g_Git.m_critGitDllSec);
 229                         if (git_get_commit_from_hash(&commit, hash.m_hash))
 230                         {
 231                                 return -1;
 232                         }
 233                 }
 234                 catch (char * msg)
 235                 {
 236                         MessageBox(nullptr, _T("Could not get commit \"") + hash.ToString() + _T("\".\nlibgit reports:\n") + CString(msg), _T("TortoiseGit"), MB_ICONERROR);
 237                         return -1;
 238                 }
 239
 240                 // right now this code is only used by TortoiseGitBlame,
 241                 // as such git notes are not needed to be loaded
 242
 243                 pRev->ParserFromCommit(&commit);
 244                 pRev->ParserParentFromCommit(&commit);
 245                 git_free_commit(&commit);
 246
 247                 revs.insert(pRev);
 248         }
 249
 250         for (auto it = revs.begin(); it != revs.end(); ++it)
 251         {
 252                 GitRev *pRev = *it;
 253                 this->push_back(pRev->m_CommitHash);
 254                 m_HashMap[pRev->m_CommitHash] = (int)size() - 1;
 255         }
 256
 257         return 0;
 258 }
 259
 260 int AddTolist(unsigned char * /*osha1*/, unsigned char *nsha1, const char * /*name*/, unsigned long /*time*/, int /*sz*/, const char *msg, void *data)
 261 {
 262         CLogDataVector *vector = (CLogDataVector*)data;
 263         GitRev rev;
 264         rev.m_CommitHash = (char*)nsha1;
 265
 266         CString one;
 267         g_Git.StringAppend(&one, (BYTE *)msg);
 268
 269         int message = one.Find(_T(":"), 0);
 270         if (message > 0)
 271         {
 272                 rev.m_RefAction = one.Left(message);
 273                 rev.GetSubject() = one.Mid(message + 1);
 274         }
 275
 276         vector->m_pLogCache->m_HashMap[rev.m_CommitHash] = rev;
 277         vector->insert(vector->begin(),rev.m_CommitHash);
 278
 279         return 0;
 280 }
 281
 282 void CLogDataVector::append(CGitHash& sha, bool storeInVector)
 283 {
 284         if (storeInVector)
 285                 this->push_back(sha);
 286
 287         GitRev* r = &m_pLogCache->m_HashMap[sha];
 288         updateLanes(*r, this->m_Lns, sha);
 289 }
 290
 291 void CLogDataVector::setLane(CGitHash& sha)
 292 {
 293         Lanes* l = &(this->m_Lns);
 294         int i = m_FirstFreeLane;
 295
 296 //      QVector<QByteArray> ba;
 297 //      const ShaString& ss = toPersistentSha(sha, ba);
 298 //      const ShaVect& shaVec(fh->revOrder);
 299
 300         for (int cnt = (int)size(); i < cnt; ++i) {
 301
 302                 GitRev* r = & this->GetGitRevAt(i);
 303                 CGitHash curSha=r->m_CommitHash;
 304
 305                 if (r->m_Lanes.empty())
 306                         updateLanes(*r, *l, curSha);
 307
 308                 if (curSha == sha)
 309                         break;
 310         }
 311         m_FirstFreeLane = ++i;
 312
 313 #if 0
 314         Lanes* l = &(this->m_Lanes);
 315         int i = m_FirstFreeLane;
 316
 317         QVector<QByteArray> ba;
 318         const ShaString& ss = toPersistentSha(sha, ba);
 319         const ShaVect& shaVec(fh->revOrder);
 320
 321         for (uint cnt = shaVec.count(); i < cnt; ++i) {
 322
 323                 const ShaString& curSha = shaVec[i];
 324                 Rev* r = m_HashMap[curSha]const_cast<Rev*>(revLookup(curSha, fh));
 325                 if (r->lanes.count() == 0)
 326                         updateLanes(*r, *l, curSha);
 327
 328                 if (curSha == ss)
 329                         break;
 330         }
 331         fh->firstFreeLane = ++i;
 332 #endif
 333 }
 334
 335
 336 void CLogDataVector::updateLanes(GitRev& c, Lanes& lns, CGitHash &sha)
 337 {
 338 // we could get third argument from c.sha(), but we are in fast path here
 339 // and c.sha() involves a deep copy, so we accept a little redundancy
 340
 341         if (lns.isEmpty())
 342                 lns.init(sha);
 343
 344         bool isDiscontinuity;
 345         bool isFork = lns.isFork(sha, isDiscontinuity);
 346         bool isMerge = (c.ParentsCount() > 1);
 347         bool isInitial = (c.ParentsCount() == 0);
 348
 349         if (isDiscontinuity)
 350                 lns.changeActiveLane(sha); // uses previous isBoundary state
 351
 352         lns.setBoundary(c.IsBoundary() == TRUE); // update must be here
 353         TRACE(_T("%s %d"),c.m_CommitHash.ToString(),c.IsBoundary());
 354
 355         if (isFork)
 356                 lns.setFork(sha);
 357         if (isMerge)
 358                 lns.setMerge(c.m_ParentHash);
 359         //if (c.isApplied)
 360         //      lns.setApplied();
 361         if (isInitial)
 362                 lns.setInitial();
 363
 364         lns.getLanes(c.m_Lanes); // here lanes are snapshotted
 365
 366         CGitHash nextSha;
 367         if( !isInitial)
 368                 nextSha = c.m_ParentHash[0];
 369
 370         lns.nextParent(nextSha);
 371
 372         //if (c.isApplied)
 373         //      lns.afterApplied();
 374         if (isMerge)
 375                 lns.afterMerge();
 376         if (isFork)
 377                 lns.afterFork();
 378         if (lns.isBranch())
 379                 lns.afterBranch();
 380 }