src/Utils/TempFile.cpp

   1 // TortoiseSVN - a Windows shell extension for easy version control\r
   2 \r
   3 // Copyright (C) 2003-2008 - TortoiseSVN\r
   4 \r
   5 // This program is free software; you can redistribute it and/or\r
   6 // modify it under the terms of the GNU General Public License\r
   7 // as published by the Free Software Foundation; either version 2\r
   8 // of the License, or (at your option) any later version.\r
   9 \r
  10 // This program is distributed in the hope that it will be useful,\r
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  13 // GNU General Public License for more details.\r
  14 \r
  15 // You should have received a copy of the GNU General Public License\r
  16 // along with this program; if not, write to the Free Software Foundation,\r
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.\r
  18 //\r
  19 #include "StdAfx.h"\r
  20 #include "Registry.h"\r
  21 #include "TempFile.h"\r
  22 \r
  23 CTempFiles::CTempFiles(void)\r
  24 {\r
  25 }\r
  26 \r
  27 CTempFiles::~CTempFiles(void)\r
  28 {\r
  29         m_TempFileList.DeleteAllFiles(false);\r
  30 }\r
  31 \r
  32 CTempFiles& CTempFiles::Instance()\r
  33 {\r
  34         static CTempFiles instance;\r
  35         return instance;\r
  36 }\r
  37 \r
  38 CTSVNPath CTempFiles::GetTempFilePath(bool bRemoveAtEnd, const CTSVNPath& path /* = CTSVNPath() */, const SVNRev revision /* = SVNRev() */)\r
  39 {\r
  40         DWORD len = ::GetTempPath(0, NULL);\r
  41         TCHAR * temppath = new TCHAR[len+1];\r
  42         TCHAR * tempF = new TCHAR[len+50];\r
  43         ::GetTempPath (len+1, temppath);\r
  44         CTSVNPath tempfile;\r
  45         CString possibletempfile;\r
  46         if (path.IsEmpty())\r
  47         {\r
  48                 ::GetTempFileName (temppath, TEXT("svn"), 0, tempF);\r
  49                 tempfile = CTSVNPath(tempF);\r
  50         }\r
  51         else\r
  52         {\r
  53                 int i=0;\r
  54                 do\r
  55                 {\r
  56                         if (revision.IsValid())\r
  57                         {\r
  58                                 possibletempfile.Format(_T("%s%s-rev%s.svn%3.3x.tmp%s"), temppath, (LPCTSTR)path.GetFileOrDirectoryName(), (LPCTSTR)revision.ToString(), i, (LPCTSTR)path.GetFileExtension());\r
  59                         }\r
  60                         else\r
  61                         {\r
  62                                 possibletempfile.Format(_T("%s%s.svn%3.3x.tmp%s"), temppath, (LPCTSTR)path.GetFileOrDirectoryName(), i, (LPCTSTR)path.GetFileExtension());\r
  63                         }\r
  64                         tempfile.SetFromWin(possibletempfile);\r
  65                         i++;\r
  66                 } while (PathFileExists(tempfile.GetWinPath()));\r
  67         }\r
  68         //now create the temp file, so that subsequent calls to GetTempFile() return\r
  69         //different filenames.\r
  70         HANDLE hFile = CreateFile(tempfile.GetWinPath(), GENERIC_READ, FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_TEMPORARY, NULL);\r
  71         CloseHandle(hFile);\r
  72         delete [] temppath;\r
  73         delete [] tempF;\r
  74         if (bRemoveAtEnd)\r
  75                 m_TempFileList.AddPath(tempfile);\r
  76         return tempfile;\r
  77 }\r