src/TortoiseGitBlame/TortoiseGitBlameDoc.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2008-2017 - 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
  20
  21 // TortoiseGitBlameDoc.cpp : implementation of the CTortoiseGitBlameDoc class
  22 //
  23
  24 #include "stdafx.h"
  25 #include "TortoiseGitBlame.h"
  26
  27 #include "TortoiseGitBlameDoc.h"
  28 #include "GitAdminDir.h"
  29 #include "Git.h"
  30 #include "MainFrm.h"
  31 #include "TGitPath.h"
  32 #include "TortoiseGitBlameView.h"
  33 #include "CmdLineParser.h"
  34 #include "CommonAppUtils.h"
  35 #include "BlameDetectMovedOrCopiedLines.h"
  36 #include "TempFile.h"
  37
  38 #ifdef _DEBUG
  39 #define new DEBUG_NEW
  40 #endif
  41
  42
  43 // CTortoiseGitBlameDoc
  44
  45 IMPLEMENT_DYNCREATE(CTortoiseGitBlameDoc, CDocument)
  46
  47 BEGIN_MESSAGE_MAP(CTortoiseGitBlameDoc, CDocument)
  48 END_MESSAGE_MAP()
  49
  50 typedef CComCritSecLock<CComCriticalSection> CAutoLocker;
  51
  52 // CTortoiseGitBlameDoc construction/destruction
  53
  54 CTortoiseGitBlameDoc::CTortoiseGitBlameDoc()
  55 : m_bFirstStartup(true)
  56 , m_IsGitFile(FALSE)
  57 , m_lLine(1)
  58 {
  59 }
  60
  61 CTortoiseGitBlameDoc::~CTortoiseGitBlameDoc()
  62 {
  63 }
  64
  65 BOOL CTortoiseGitBlameDoc::OnNewDocument()
  66 {
  67         return TRUE;
  68 }
  69 BOOL CTortoiseGitBlameDoc::OnOpenDocument(LPCTSTR lpszPathName)
  70 {
  71         CCmdLineParser parser(AfxGetApp()->m_lpCmdLine);
  72         if (m_bFirstStartup)
  73         {
  74                 m_Rev = parser.GetVal(L"rev");
  75                 m_lLine = (int)parser.GetLongVal(L"line");
  76                 m_bFirstStartup = false;
  77         }
  78         else
  79         {
  80                 m_Rev.Empty();
  81                 m_lLine = 1;
  82         }
  83
  84         return OnOpenDocument(lpszPathName,m_Rev);
  85 }
  86
  87 BOOL CTortoiseGitBlameDoc::OnOpenDocument(LPCTSTR lpszPathName,CString Rev)
  88 {
  89         if(Rev.IsEmpty())
  90                 Rev = L"HEAD";
  91
  92         // enable blame for files which do not exist in current working tree
  93         if (!PathFileExists(lpszPathName) && Rev != L"HEAD")
  94         {
  95                 if (!CDocument::OnOpenDocument(CTempFiles::Instance().GetTempFilePath(true).GetWinPathString()))
  96                         return FALSE;
  97         }
  98         else
  99         {
 100                 if (!CDocument::OnOpenDocument(lpszPathName))
 101                         return FALSE;
 102         }
 103
 104         m_CurrentFileName = lpszPathName;
 105
 106         m_Rev=Rev;
 107
 108         // (SDI documents will reuse this document)
 109         if(!g_Git.CheckMsysGitDir())
 110         {
 111                 CCommonAppUtils::RunTortoiseGitProc(L" /command:settings");
 112                 return FALSE;
 113         }
 114         CString topdir;
 115         if (!GitAdminDir::HasAdminDir(m_CurrentFileName, &topdir))
 116         {
 117                 CString temp;
 118                 temp.Format(IDS_CANNOTBLAMENOGIT, (LPCTSTR)m_CurrentFileName);
 119                 MessageBox(nullptr, temp, L"TortoiseGitBlame", MB_OK | MB_ICONERROR);
 120                 return FALSE;
 121         }
 122         else
 123         {
 124                 m_IsGitFile=TRUE;
 125                 sOrigCWD = g_Git.m_CurrentDir = topdir;
 126
 127                 CString PathName = m_CurrentFileName;
 128                 if (topdir[topdir.GetLength() - 1] == L'\\' || topdir[topdir.GetLength() - 1] == L'/')
 129                         PathName=PathName.Right(PathName.GetLength()-g_Git.m_CurrentDir.GetLength());
 130                 else
 131                         PathName=PathName.Right(PathName.GetLength()-g_Git.m_CurrentDir.GetLength()-1);
 132
 133                 CTGitPath path;
 134                 path.SetFromWin(PathName);
 135
 136                 if(!g_Git.m_CurrentDir.IsEmpty())
 137                         SetCurrentDirectory(g_Git.m_CurrentDir);
 138
 139                 try
 140                 {
 141                         // make sure all config files are read in order to check that none contains an error
 142                         g_Git.GetConfigValue(L"doesnot.exist");
 143
 144                         // make sure git_init() works and that .git-dir is ok
 145                         CAutoLocker lock(g_Git.m_critGitDllSec);
 146                         g_Git.CheckAndInitDll();
 147                 }
 148                 catch (char * libgiterr)
 149                 {
 150                         MessageBox(nullptr, CString(libgiterr), L"TortoiseGitBlame", MB_ICONERROR);
 151                         return FALSE;
 152                 }
 153
 154                 CString cmd, option;
 155                 int dwDetectMovedOrCopiedLines = theApp.GetInt(L"DetectMovedOrCopiedLines", BLAME_DETECT_MOVED_OR_COPIED_LINES_DISABLED);
 156                 int dwDetectMovedOrCopiedLinesNumCharactersWithinFile = theApp.GetInt(L"DetectMovedOrCopiedLinesNumCharactersWithinFile", BLAME_DETECT_MOVED_OR_COPIED_LINES_NUM_CHARACTERS_WITHIN_FILE_DEFAULT);
 157                 int dwDetectMovedOrCopiedLinesNumCharactersFromFiles = theApp.GetInt(L"DetectMovedOrCopiedLinesNumCharactersFromFiles", BLAME_DETECT_MOVED_OR_COPIED_LINES_NUM_CHARACTERS_FROM_FILES_DEFAULT);
 158                 switch(dwDetectMovedOrCopiedLines)
 159                 {
 160                 default:
 161                 case BLAME_DETECT_MOVED_OR_COPIED_LINES_DISABLED:
 162                         option.Empty();
 163                         break;
 164                 case BLAME_DETECT_MOVED_OR_COPIED_LINES_WITHIN_FILE:
 165                         option.Format(L"-M%d", dwDetectMovedOrCopiedLinesNumCharactersWithinFile);
 166                         break;
 167                 case BLAME_DETECT_MOVED_OR_COPIED_LINES_FROM_MODIFIED_FILES:
 168                         option.Format(L"-C%d", dwDetectMovedOrCopiedLinesNumCharactersFromFiles);
 169                         break;
 170                 case BLAME_DETECT_MOVED_OR_COPIED_LINES_FROM_EXISTING_FILES_AT_FILE_CREATION:
 171                         option.Format(L"-C -C%d", dwDetectMovedOrCopiedLinesNumCharactersFromFiles);
 172                         break;
 173                 case BLAME_DETECT_MOVED_OR_COPIED_LINES_FROM_EXISTING_FILES:
 174                         option.Format(L"-C -C -C%d", dwDetectMovedOrCopiedLinesNumCharactersFromFiles);
 175                         break;
 176                 }
 177
 178                 if (theApp.GetInt(L"IgnoreWhitespace", 0) == 1)
 179                         option += L" -w";
 180
 181                 bool onlyFirstParent = theApp.GetInt(L"OnlyFirstParent", 0) == 1;
 182                 if (onlyFirstParent)
 183                 {
 184                         CString tmpfile = CTempFiles::Instance().GetTempFilePath(true).GetWinPathString();
 185                         cmd.Format(L"git rev-list --first-parent %s", (LPCTSTR)Rev);
 186                         CString err;
 187                         CAutoFILE file = _wfsopen(tmpfile, L"wb", SH_DENYWR);
 188                         if (!file)
 189                         {
 190                                 MessageBox(nullptr, CString(MAKEINTRESOURCE(IDS_BLAMEERROR)) + L"\n\nCould not create temp file!", L"TortoiseGitBlame", MB_OK | MB_ICONERROR);
 191                                 return FALSE;
 192                         }
 193
 194                         CStringA lastline;
 195                         if (g_Git.Run(cmd, [&](const CStringA& line)
 196                         {
 197                                 fwrite(lastline + ' ' + line + '\n', sizeof(char), lastline.GetLength() + 1 + line.GetLength() + 1, file);
 198                                 lastline = line;
 199                         }
 200                         , &err))
 201                         {
 202                                 MessageBox(nullptr, CString(MAKEINTRESOURCE(IDS_BLAMEERROR)) + L"\n\n" + err, L"TortoiseGitBlame", MB_OK | MB_ICONERROR);
 203                                 return FALSE;
 204                         }
 205                         option.AppendFormat(L" -S \"%s\"", (LPCTSTR)tmpfile);
 206                 }
 207
 208                 cmd.Format(L"git.exe blame -p %s %s -- \"%s\"", (LPCTSTR)option, (LPCTSTR)Rev, (LPCTSTR)path.GetGitPathString());
 209                 m_BlameData.clear();
 210                 BYTE_VECTOR err;
 211                 if(g_Git.Run(cmd, &m_BlameData, &err))
 212                 {
 213                         CString str;
 214                         if (!m_BlameData.empty())
 215                                 CGit::StringAppend(&str, m_BlameData.data(), CP_UTF8, (int)m_BlameData.size());
 216                         if (!err.empty())
 217                                 CGit::StringAppend(&str, err.data(), CP_UTF8, (int)err.size());
 218                         MessageBox(nullptr, CString(MAKEINTRESOURCE(IDS_BLAMEERROR)) + L"\n\n" + str, L"TortoiseGitBlame", MB_OK | MB_ICONERROR);
 219
 220                         return FALSE;
 221                 }
 222
 223 #ifdef USE_TEMPFILENAME
 224                 m_TempFileName = CTempFiles::Instance().GetTempFilePath(true).GetWinPathString();
 225
 226                 cmd.Format(L"git.exe cat-file blob %s:\"%s\"", (LPCTSTR)Rev, (LPCTSTR)path.GetGitPathString());
 227
 228                 if(g_Git.RunLogFile(cmd, m_TempFileName))
 229                 {
 230                         CString str;
 231                         str.Format(IDS_CHECKOUTFAILED, (LPCTSTR)path.GetGitPathString());
 232                         MessageBox(nullptr, CString(MAKEINTRESOURCE(IDS_BLAMEERROR)) + L"\n\n" + str, L"TortoiseGitBlame", MB_OK | MB_ICONERROR);
 233                         return FALSE;
 234                 }
 235 #endif
 236                 m_GitPath = path;
 237
 238                 CTortoiseGitBlameView *pView=DYNAMIC_DOWNCAST(CTortoiseGitBlameView,GetMainFrame()->GetActiveView());
 239                 if (!pView)
 240                 {
 241                         CWnd* pWnd = GetMainFrame()->GetDescendantWindow(AFX_IDW_PANE_FIRST, TRUE);
 242                         if (pWnd && pWnd->IsKindOf(RUNTIME_CLASS(CTortoiseGitBlameView)))
 243                                 pView = static_cast<CTortoiseGitBlameView*>(pWnd);
 244                         else
 245                                 return FALSE;
 246                 }
 247                 pView->ParseBlame();
 248
 249                 BOOL bShowCompleteLog = (theApp.GetInt(L"ShowCompleteLog", 1) == 1);
 250                 if (bShowCompleteLog && BlameIsLimitedToOneFilename(dwDetectMovedOrCopiedLines) && !onlyFirstParent)
 251                 {
 252                         if (GetMainFrame()->m_wndOutput.LoadHistory(path.GetGitPathString(), m_Rev, (theApp.GetInt(L"FollowRenames", 0) == 1)))
 253                                 return FALSE;
 254                 }
 255                 else
 256                 {
 257                         std::unordered_set<CGitHash> hashes;
 258                         pView->m_data.GetHashes(hashes);
 259                         if (GetMainFrame()->m_wndOutput.LoadHistory(hashes))
 260                                 return FALSE;
 261                 }
 262
 263                 pView->MapLineToLogIndex();
 264                 pView->UpdateInfo();
 265                 if (m_lLine > 0)
 266                         pView->GotoLine(m_lLine);
 267
 268                 SetPathName(m_CurrentFileName, FALSE);
 269         }
 270
 271         return TRUE;
 272 }
 273
 274 void CTortoiseGitBlameDoc::SetPathName(LPCTSTR lpszPathName, BOOL bAddToMRU)
 275 {
 276         CDocument::SetPathName(lpszPathName, bAddToMRU && (m_Rev == L"HEAD"));
 277
 278         this->SetTitle(CString(lpszPathName) + L':' + m_Rev);
 279 }
 280
 281 // CTortoiseGitBlameDoc diagnostics
 282
 283 #ifdef _DEBUG
 284 void CTortoiseGitBlameDoc::AssertValid() const
 285 {
 286         CDocument::AssertValid();
 287 }
 288
 289 void CTortoiseGitBlameDoc::Dump(CDumpContext& dc) const
 290 {
 291         CDocument::Dump(dc);
 292 }
 293 #endif //_DEBUG
 294
 295
 296 // CTortoiseGitBlameDoc commands