src/Utils/TempFile.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2012-2013 - TortoiseGit
   4 // Copyright (C) 2003-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 #include "stdafx.h"
  21 #include "registry.h"
  22 #include "TempFile.h"
  23 #include "TGitPath.h"
  24 #include "SmartHandle.h"
  25
  26 CTempFiles::CTempFiles(void)
  27 {
  28 }
  29
  30 CTempFiles::~CTempFiles(void)
  31 {
  32         m_TempFileList.DeleteAllFiles(false);
  33 }
  34
  35 CTempFiles& CTempFiles::Instance()
  36 {
  37         static CTempFiles instance;
  38         return instance;
  39 }
  40
  41 CTGitPath CTempFiles::GetTempFilePath(bool bRemoveAtEnd, const CTGitPath& path /* = CTGitPath() */, const GitRev revision /* = GitRev() */)
  42 {
  43         DWORD len = GetTortoiseGitTempPath(0, NULL);
  44
  45         std::unique_ptr<TCHAR[]> temppath(new TCHAR[len + 1]);
  46         std::unique_ptr<TCHAR[]> tempF(new TCHAR[len + 50]);
  47         GetTortoiseGitTempPath(len + 1, temppath.get());
  48         CTGitPath tempfile;
  49         CString possibletempfile;
  50         if (path.IsEmpty())
  51         {
  52                 ::GetTempFileName (temppath.get(), _T("git"), 0, tempF.get());
  53                 tempfile = CTGitPath(tempF.get());
  54         }
  55         else
  56         {
  57                 int i=0;
  58                 do
  59                 {
  60                         if (!((GitRev&)revision).m_CommitHash.IsEmpty())
  61                         {
  62                                 possibletempfile.Format(_T("%s%s-rev%s.git%3.3x.tmp%s"), temppath.get(), (LPCTSTR)path.GetFileOrDirectoryName(), (LPCTSTR)((GitRev&)revision).m_CommitHash.ToString().Left(7), i, (LPCTSTR)path.GetFileExtension());
  63                         }
  64                         else
  65                         {
  66                                 possibletempfile.Format(_T("%s%s.git%3.3x.tmp%s"), temppath.get(), (LPCTSTR)path.GetFileOrDirectoryName(), i, (LPCTSTR)path.GetFileExtension());
  67                         }
  68                         tempfile.SetFromWin(possibletempfile);
  69                         ++i;
  70                 } while (PathFileExists(tempfile.GetWinPath()));
  71         }
  72         //now create the temp file, so that subsequent calls to GetTempFile() return
  73         //different filenames.
  74         CAutoFile hFile = CreateFile(tempfile.GetWinPath(), GENERIC_READ, FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_TEMPORARY, NULL);
  75         if (bRemoveAtEnd)
  76                 m_TempFileList.AddPath(tempfile);
  77         return tempfile;
  78 }