TnbArchiver.h

[詳解]

#pragma once
#include "TnbArchiverDllHandler.h"
#include "TnbStrVector.h"
#include "TnbTemporaryFile.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CArchiver
{
public:
 
    CArchiver(CArchiverDllHandler* pDll) : m_pDll(pDll)
    {
        ASSERT0( pDll != NULL, "CArchiver()", "正しいアーカイバ管理クラスが指定されていません。" );
        if ( pDll == NULL ) 
        {
            throw CNullPointerException();
        }
        ASSERT0( pDll->IsReady(), "CArchiver()", "使用可能なアーカイバ管理クラスが指定されていません。" );
        if ( ! pDll->IsReady() ) 
        {
            throw CNullPointerException(); 
        }
        ::GetTempPathA(MAX_PATH, m_strWorkDir.GetBuffer(MAX_PATH));
        m_strWorkDir.ReleaseBuffer();
    }
    
    CStr GetTargetArchiveFile(void) const
    {
        CStr s = m_strArcFile;
        return s;
    }
 
    bool SetTargetArchiveFile(LPCTSTR lpszArcFile)
    {
        bool r = false;
        if ( ToInt(::GetFileAttributes(lpszArcFile)) == -1 )
        {
            r = (::GetLastError() == ERROR_FILE_NOT_FOUND);
        }
        else if ( m_pDll->CheckArchive(CAscii(lpszArcFile), 0) )
        {
            r = true;
        }
        if ( r )
        {
            m_strArcFile = lpszArcFile;
        }
        return r;
    }
 
    void SetWorkDirectory(LPCTSTR lpszWorkDir)
    {
        m_strWorkDir = lpszWorkDir;
    }
 
    int Delete(LPCTSTR lpszName, LPCTSTR lpszOption = NULL)
    {
        if ( m_strArcFile.IsEmpty() )
        {
            return -1; 
        }
        return m_Exec(m_pDll->EC_Delete, lpszName, lpszOption);
    }
 
    int Delete(const CStrVector& vs, LPCTSTR lpszOption = NULL)
    {
        if ( m_strArcFile.IsEmpty() ) 
        {
            return -1; 
        }
        return m_ExecList(m_pDll->EC_Delete, vs, lpszOption);
    }
 
    int Extract(LPCTSTR lpszName, LPCTSTR lpszOption = NULL)
    {
        if ( ! m_IsReady() )
        {
            return -1; 
        }
        return m_Exec(m_pDll->EC_Extract, lpszName, lpszOption);
    }
 
    int Extract(const CStrVector& vs, LPCTSTR lpszOption = NULL)
    {
        if ( ! m_IsReady() ) 
        {
            return -1; 
        }
        return m_ExecList(m_pDll->EC_Extract, vs, lpszOption);
    }
 
    int ExtractMem(LPCTSTR lpszName, CByteVector& _vb, LPCTSTR lpszOption = NULL)
    {
        int r = -1;
        if ( ! m_strArcFile.IsEmpty() ) 
        {
            size_t dwSize = _vb.GetSize();
            #ifdef _UNICODE
                int r = m_pDll->ExtractMem(m_strArcFile, CAscii(lpszName), _vb.GetBuffer(), dwSize, CAscii(lpszOption));
            #else
                int r = m_pDll->ExtractMem(m_strArcFile, lpszName, _vb.GetBuffer(), dwSize, lpszOption);
            #endif
            _vb.ReleaseBuffer();
            _vb.SetSize((r == 0) ? dwSize : 0);
        }
        return r;
    }
 
    int ExtractWithDirectory(LPCTSTR lpszName, LPCTSTR lpszOption = NULL)
    {
        if ( ! m_IsReady() ) 
        {
            return -1; 
        }
        return m_Exec(m_pDll->EC_ExtractDir, lpszName, lpszOption);
    }
 
    int ExtractWithDirectory(const CStrVector& vs, LPCTSTR lpszOption = NULL)
    {
        if ( ! m_IsReady() ) 
        {
            return -1; 
        }
        return m_ExecList(m_pDll->EC_ExtractDir, vs, lpszOption);
    }
 
    int Attract(LPCTSTR lpszName, LPCTSTR lpszOption = NULL)
    {
        if ( ! m_IsReady() )
        {
            return -1; 
        }
        return m_Exec(m_pDll->EC_Attract, lpszName, lpszOption);
    }
 
    int Attract(const CStrVector& vs, LPCTSTR lpszOption = NULL)
    {
        if ( ! m_IsReady() )
        {
            return -1;
        }
        return m_ExecList(m_pDll->EC_Attract, vs, lpszOption);
    }
 
    int AttractMem(LPCTSTR lpszName, time_t time, WORD wAttr, const CByteVector& vb, LPCTSTR lpszOption = NULL)
    {
        int r = -1;
        if ( ! m_strArcFile.IsEmpty() )
        {
            DWORD dwSize = ToDword(vb.GetSize());
            #ifdef _UNICODE
                r = m_pDll->CompressMem(m_strArcFile, CAscii(lpszName), time, wAttr, vb.ReferBuffer(), dwSize, CAscii(lpszOption));
            #else
                r = m_pDll->CompressMem(m_strArcFile, lpszName, time, wAttr, vb.ReferBuffer(), dwSize, lpszOption);
            #endif
        }
        return r;
    }
private:
    CArchiverDllHandler*    m_pDll;         
    CAscii                  m_strArcFile;   
    CAscii                  m_strWorkDir;   
 
    bool m_IsReady(void) const
    {
        return (! m_strArcFile.IsEmpty()) && (! m_strWorkDir.IsEmpty());
    }
 
    int m_Exec(CArchiverDllHandler::ECommand cmd, LPCTSTR lpszName, LPCTSTR lpszOption)
    {
        #ifdef _UNICODE
            return m_pDll->Execute(cmd, m_strArcFile, m_strWorkDir, CAscii(lpszName), CAscii(lpszOption));
        #else
            return m_pDll->Execute(cmd, m_strArcFile, m_strWorkDir, lpszName, lpszOption);
        #endif
    }
 
    int m_ExecList(CArchiverDllHandler::ECommand cmd, const CStrVector& vs, LPCTSTR lpszOption)
    {
        CAscii strContent;
        if ( lpszOption != NULL )
        {
            strContent += lpszOption;
            strContent += "\r\n";
        }
        loop ( i, vs.GetSize() )
        {
            strContent += CAscii(vs[i]);
            strContent += "\r\n";
        }
        CTemporaryFile tmp;
        CStr strName = tmp.Create(strContent.GetLength(), strContent.begin());
        int r = m_Exec(cmd, "@" + strName, NULL);
        tmp.Free();
        return r;
    }
};
 
 
 
}; // TNB