src/TortoiseGitBlame/TortoiseGitBlameDoc.cpp

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