src/Utils/TempFile.cpp

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