TnbTemporaryFile.h

[詳解]

#pragma once
#include "TnbStr.h"
#include "TnbVector.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
template<DWORD PFIX>
class CTemporaryFileT : CCopyImpossible
{
    CStr        m_tempFolder;   
    CStr        m_targetFile;   
    bool m_Create(void)
    {
        Free();
        CStr strPath;
        if ( m_tempFolder.IsEmpty() )
        {
            ::GetTempPath(MAX_PATH, strPath.GetBuffer(MAX_PATH));
            strPath.ReleaseBuffer();
        }
        else
        {
            strPath = m_tempFolder;
        }
        const static TCHAR tc[5] = { (PFIX&0xFF), ((PFIX>>8)&0xFF), ((PFIX>>16)&0xFF), ((PFIX>>24)&0xFF), 0 };
        ::GetTempFileName(strPath, tc, 0, m_targetFile.GetBuffer(MAX_PATH));
        m_targetFile.ReleaseBuffer();
        return ! m_targetFile.IsEmpty();
    }
 
public:
 
    CTemporaryFileT(void)
    {
    }
 
    ~CTemporaryFileT(void)
    {
        if ( ! m_targetFile.IsEmpty() )
        {
            ::DeleteFile(m_targetFile);
            m_targetFile.Empty();
        }
    }
 
    CStr GetName(void) const
    {
        return m_targetFile;
    }
 
    bool HasThat(void) const
    {
        return ! m_targetFile.IsEmpty();
    }
 
    void Free(void)
    {
        if ( ! m_targetFile.IsEmpty() )
        {
            if ( ! ::DeleteFile(m_targetFile) )
            {
                throw CDeleteFailureException(_GetLastError("DeleteFile"));
            }
            m_targetFile.Empty();
        }
    }
 
    void Detach(void)
    {
        m_targetFile.Empty();
    }
 
    CByteVector Release(void)
    {
        CByteVector vb;
        if ( ! m_targetFile.IsEmpty() )
        {
            HANDLE h = ::CreateFile(m_targetFile, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
            if ( h == INVALID_HANDLE_VALUE )
            {
                throw COpenFailureException(_GetLastError("CreateFile"));
            }
            DWORD dw = 0;
            DWORD size = ::GetFileSize(h, &dw);
            if ( size > 0 )
            {
                vb.SetSize(size);
                BOOL boRc = ::ReadFile(h, vb.GetBuffer(size), size, &dw, NULL);
                if ( ! boRc || dw != size )
                {
                    throw CReadFailureException(_GetLastError("ReadFile"));
                }
                vb.ReleaseBuffer();
            }
            ::CloseHandle(h);
            Free();
        }
        return vb;
    }
 
    CStr Create(void)
    {
        m_Create();
        return m_targetFile;
    }
 
    CStr Create(size_t size)
    {
        if ( m_Create() && size > 0 )
        {
            HANDLE h = ::CreateFile(m_targetFile, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL);
            if ( h == INVALID_HANDLE_VALUE )
            {
                throw COpenFailureException(_GetLastError("CreateFile"));
            }
            LONG l = 0;
            ::SetFilePointer(h, ToDword(size), &l, FILE_BEGIN);
            VERIFY( ::SetEndOfFile(h) );
            ::CloseHandle(h);
        }
        return m_targetFile;
    }
 
    CStr Create(size_t size, LPCVOID P)
    {
        if ( m_Create() && size >= 0 )
        {
            HANDLE h = ::CreateFile(m_targetFile, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_TEMPORARY, NULL);
            if ( h == INVALID_HANDLE_VALUE )
            {
                throw COpenFailureException(_GetLastError("CreateFile"));
            }
            DWORD dw = 0;
            VERIFY( ::WriteFile(h, P, ToDword(size), &dw, NULL) );
            ::CloseHandle(h);
        }
        return m_targetFile;
    }
 
    CStr Create(const ISequenceCollectionT<BYTE>& vb)
    {
        return Create(vb.GetSize(), vb.ReferBuffer());
    }
 
    void SetTempraryFolder(LPCTSTR pFolder)
    {
        m_tempFolder = pFolder;
    }
 
    CStr Attach(LPCSTR lpszFile)
    {
        if ( ToInt(::GetFileAttributes(lpszFile)) == -1 )
        {
            throw COpenFailureException(_GetLastError("GetFileAttributes"));
        }
        m_targetFile = lpszFile;
        return m_targetFile;
    }
};
 
 
 
typedef CTemporaryFileT<'bnt'> CTemporaryFile;
 
 
 
}; // TNB