src/crashrpt/Utility.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 //
   3 //  Module: Utility.cpp
   4 //
   5 //    Desc: See Utility.h
   6 //
   7 // Copyright (c) 2007 TortoiseSVN
   8 // Copyright (c) 2003 Michael Carruth
   9 //
  10 ///////////////////////////////////////////////////////////////////////////////
  11
  12 #include "stdafx.h"
  13 #include "Utility.h"
  14 #include "resource.h"
  15
  16 namespace CUtility {
  17
  18 BSTR CUtility::AllocSysString(string s)
  19 {
  20 #if defined(_UNICODE) || defined(OLE2ANSI)
  21         BSTR bstr = ::SysAllocStringLen(s.c_str(), s.size());
  22 #else
  23         int nLen = MultiByteToWideChar(CP_ACP, 0, s.c_str(),
  24                 s.size(), NULL, NULL);
  25         BSTR bstr = ::SysAllocStringLen(NULL, nLen);
  26         if(bstr != NULL)
  27                 MultiByteToWideChar(CP_ACP, 0, s.c_str(), s.size(), bstr, nLen);
  28 #endif
  29         return bstr;
  30 }
  31
  32 FILETIME CUtility::getLastWriteFileTime(string sFile)
  33 {
  34    FILETIME          ftLocal = {0};
  35    HANDLE            hFind;
  36    WIN32_FIND_DATA   ff32;
  37    hFind = FindFirstFile(sFile.c_str(), &ff32);
  38    if (INVALID_HANDLE_VALUE != hFind)
  39    {
  40       FileTimeToLocalFileTime(&(ff32.ftLastWriteTime), &ftLocal);
  41       FindClose(hFind);
  42    }
  43    return ftLocal;
  44 }
  45
  46 FILETIME CUtility::getLastWriteFileTime(WCHAR * wszFile)
  47 {
  48         FILETIME          ftLocal = {0};
  49         HANDLE            hFind;
  50         WIN32_FIND_DATAW  ff32;
  51         hFind = FindFirstFileW(wszFile, &ff32);
  52         if (INVALID_HANDLE_VALUE != hFind)
  53         {
  54                 FileTimeToLocalFileTime(&(ff32.ftLastWriteTime), &ftLocal);
  55                 FindClose(hFind);
  56         }
  57         return ftLocal;
  58 }
  59
  60 string CUtility::getAppName()
  61 {
  62    TCHAR szFileName[_MAX_PATH];
  63    GetModuleFileName(NULL, szFileName, _MAX_FNAME);
  64
  65    string sAppName; // Extract from last '\' to '.'
  66
  67    *_tcsrchr(szFileName, '.') = '\0';
  68    sAppName = (_tcsrchr(szFileName, '\\')+1);
  69
  70    return sAppName;
  71 }
  72
  73
  74 string CUtility::getTempFileName()
  75 {
  76    static int counter = 0;
  77    TCHAR szTempDir[MAX_PATH - 14]   = _T("");
  78    TCHAR szTempFile[MAX_PATH]       = _T("");
  79
  80    if (GetTempPath(MAX_PATH - 14, szTempDir))
  81       GetTempFileName(szTempDir, getAppName().c_str(), ++counter, szTempFile);
  82
  83    return szTempFile;
  84 }
  85
  86
  87 string CUtility::getSaveFileName()
  88 {
  89    string sFilter = _T("Zip Files (*.zip)");
  90
  91    OPENFILENAME ofn = {0};                      // common dialog box structure
  92    TCHAR szFile[MAX_PATH] = {0};        // buffer for file name
  93    // Initialize OPENFILENAME
  94    ofn.lStructSize = sizeof(OPENFILENAME);
  95    ofn.hwndOwner = NULL;
  96    ofn.lpstrFile = szFile;
  97    ofn.nMaxFile = _countof(szFile);
  98    ofn.Flags = OFN_OVERWRITEPROMPT;
  99
 100    ofn.lpstrFilter = sFilter.c_str();
 101    ofn.nFilterIndex = 1;
 102    // Display the Open dialog box.
 103    bool bTargetSelected = !!GetSaveFileName(&ofn);
 104
 105    DeleteFile(ofn.lpstrFile);  // Just in-case it already exist
 106    return ofn.lpstrFile;
 107 }
 108
 109 };