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