TnbTemporaryDirectory.h

[詳解]

#pragma once
#include "TnbTemporaryFile.h"
#include "TnbFileName.h"
 
 
#ifndef _WIN32_WCE
 #include <SHLWAPI.h>
 #pragma comment(lib, "SHLWAPI.lib")
#endif
 
 
 
//TNB Library
namespace TNB
{
 
 
 
template<DWORD PFIX>
class CTemporaryDirectoryT : CCopyImpossible
{
    CStr m_strDir;      
 
    bool m_Remove(LPCTSTR lpszDir)
    {
        if ( ::RemoveDirectory(lpszDir) )
        {
            return true;
        }
        CFileName fn(lpszDir);
        return fn.Remove(false, true);
    }
 
public:
 
    CTemporaryDirectoryT(void)
    {
    }
 
    ~CTemporaryDirectoryT(void)
    {
        if ( ! m_strDir.IsEmpty() )
        {
            m_Remove(m_strDir);
            m_strDir.Empty();
        }
    }
 
    CStr GetName(void) const
    {
        return m_strDir;
    }
 
    bool HasThat(void) const
    {
        return ! m_strDir.IsEmpty();
    }
 
    void Free(void)
    {
        if ( ! m_strDir.IsEmpty() )
        {
            if ( ! m_Remove(m_strDir) )
            {
                throw CDeleteFailureException(::GetLastError());
            }
            m_strDir.Empty();
        }
    }
 
    void Detach(void)
    {
        m_strDir.Empty();
    }
 
    CStr Create(void)
    {
        Free();
        CTemporaryFileT<PFIX> temp;
        CStr s = temp.Create();
        temp.Free();
        if ( ! ::CreateDirectory(s, NULL) )
        {
            s.Empty();
        }
        else
        {
            m_strDir = s;
            s += _T("\\");
        }
        return s;
    }
};
 
 
 
typedef CTemporaryDirectoryT<'bnt'> CTemporaryDirectory;
 
 
 
}; // TNB