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
  24 CTempFiles::CTempFiles(void)
  25 {
  26 }
  27
  28 CTempFiles::~CTempFiles(void)
  29 {
  30         m_TempFileList.DeleteAllFiles(false);
  31 }
  32
  33 CTempFiles& CTempFiles::Instance()
  34 {
  35         static CTempFiles instance;
  36         return instance;
  37 }
  38
  39 CTGitPath CTempFiles::GetTempFilePath(bool bRemoveAtEnd, const CTGitPath& path /* = CTGitPath() */, const GitRev revision /* = GitRev() */)
  40 {
  41         DWORD len = ::GetTempPath(0, NULL);
  42         TCHAR * temppath = new TCHAR[len+1];
  43         TCHAR * tempF = new TCHAR[len+50];
  44         ::GetTempPath (len+1, temppath);
  45         CTGitPath tempfile;
  46         CString possibletempfile;
  47         if (path.IsEmpty())
  48         {
  49                 ::GetTempFileName (temppath, TEXT("git"), 0, tempF);
  50                 tempfile = CTGitPath(tempF);
  51         }
  52         else
  53         {
  54                 int i=0;
  55                 do
  56                 {
  57                         if (!((GitRev&)revision).m_CommitHash.IsEmpty())
  58                         {
  59                                 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());
  60                         }
  61                         else
  62                         {
  63                                 possibletempfile.Format(_T("%s%s.git%3.3x.tmp%s"), temppath, (LPCTSTR)path.GetFileOrDirectoryName(), i, (LPCTSTR)path.GetFileExtension());
  64                         }
  65                         tempfile.SetFromWin(possibletempfile);
  66                         i++;
  67                 } while (PathFileExists(tempfile.GetWinPath()));
  68         }
  69         //now create the temp file, so that subsequent calls to GetTempFile() return
  70         //different filenames.
  71         HANDLE hFile = CreateFile(tempfile.GetWinPath(), GENERIC_READ, FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_TEMPORARY, NULL);
  72         CloseHandle(hFile);
  73         delete [] temppath;
  74         delete [] tempF;
  75         if (bRemoveAtEnd)
  76                 m_TempFileList.AddPath(tempfile);
  77         return tempfile;
  78 }