TnbArchiveFileFinder.h

[詳解]

#pragma once
#include "TnbArchiverDllHandler.h"
#include "TnbFileFinder.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CArchiveFileFinder : public CAbstractFileFinder
{
    DEFSUPER(CAbstractFileFinder);
    CArchiverDllHandler*    m_pDll;         
    CAscii                  m_strArcFile;   
    DWORD                   m_dwMode;       
    HARC                    m_hArc;         
    TIndividualInfo         m_foundInfo;
 
    static DWORD ms_ArcAttrToWin32Attr(int aa)
    {
        DWORD r = 0;
        if ( (aa & FA_RDONLY) != 0 )    { r |= FILE_ATTRIBUTE_READONLY;     }
        if ( (aa & FA_HIDDEN) != 0 )    { r |= FILE_ATTRIBUTE_HIDDEN;       }
        if ( (aa & FA_SYSTEM) != 0 )    { r |= FILE_ATTRIBUTE_SYSTEM;       }
        if ( (aa & FA_LABEL) != 0 )     { r |= FILE_ATTRIBUTE_TEMPORARY;    }
        if ( (aa & FA_DIREC) != 0 )     { r |= FILE_ATTRIBUTE_DIRECTORY;    }
        if ( (aa & FA_ARCH) != 0 )      { r |= FILE_ATTRIBUTE_ARCHIVE;      }
        if ( (aa & FA_ENCRYPTED) != 0 ) { r |= FILE_ATTRIBUTE_ENCRYPTED;    }
        if ( r == 0 )                   { r |= FILE_ATTRIBUTE_NORMAL;       }
        return r;
    }
 
    bool m_MakeWin32Info(WIN32_FIND_DATA& _win32Info)
    {
        Zero(_win32Info);
        try
        {
            //-- 属性変換
            int arcAttr = 0;
            m_pDll->Func(arcAttr, "GetAttribute", m_hArc);
            _win32Info.dwFileAttributes = ms_ArcAttrToWin32Attr(arcAttr);
            //-- 書込時間
            BOOL r;
            if ( m_pDll->QueryFunctionList(ISARC_GET_WRITE_TIME_EX) )
            {
                m_pDll->Func(r, "GetWriteTimeEx", m_hArc, &(_win32Info.ftLastWriteTime));
            }
            else
            {
                //_win32Info.ftLastWriteTime
                //m_foundInfo.wDate , m_foundInfo.wTime;
            }
            _win32Info.ftCreationTime = _win32Info.ftLastWriteTime;
            _win32Info.ftLastAccessTime = _win32Info.ftLastWriteTime;
            if ( m_pDll->QueryFunctionList(ISARC_GET_CREATE_TIME_EX) )
            {
                m_pDll->Func(r, "GetCreateTimeEx", m_hArc, &(_win32Info.ftCreationTime));
            }
            if ( m_pDll->QueryFunctionList(ISARC_GET_ACCESS_TIME_EX) )
            {
                m_pDll->Func(r, "GetAccessTimeEx", m_hArc, &(_win32Info.ftLastAccessTime));
            }
            //-- サイズ
            _win32Info.nFileSizeHigh = 0;
            _win32Info.nFileSizeLow = m_foundInfo.dwOriginalSize;
            //-- 予備 
            _win32Info.dwReserved0 = 0;
            _win32Info.dwReserved1 = 0;
            //-- 名前
            CStr s = m_foundInfo.szFileName;
            STRLIB::Copy(_win32Info.cFileName, s.Left(MAX_PATH));
            _win32Info.cAlternateFileName[0] = 0;
            return true;
        }
        catch(CNullPointerException& e)
        {
            e.OnCatch();
        }
        catch(...)
        {
            ASSERTLIB(false);
            throw;
        }
        return false;
    }
 
public:
 
    CArchiveFileFinder(void) : _super(), m_pDll(NULL), m_hArc(NULL), m_dwMode(0)
    {
    }
 
    virtual ~CArchiveFileFinder(void)
    {
        OnFinish();
    }
 
    bool Initialize(CArchiverDllHandler* pDll, LPCTSTR lpszArcFileName, DWORD dwMode = 0)
    {
        Finish();
        if ( pDll == NULL ) 
        {
            ASSERT0( false, "CArchiveFileFinder()", "正しいアーカイバ管理クラスが指定されていません。" );
            return false;
        }
        if ( ! pDll->IsReady() ) 
        {
            ASSERT0( false, "CArchiveFileFinder()", "使用可能なアーカイバ管理クラスが指定されていません。" );
            return false;
        }
        int r = static_cast<int>(::GetFileAttributes(lpszArcFileName));
        if ( r < 0 )
        {
            _GetLastError("GetFileAttributes");
            return false;
        }
        m_pDll = pDll;
        m_strArcFile = lpszArcFileName;
        m_dwMode = dwMode;
        return true;
    }
    
    const TIndividualInfo* RefPtr(void) const
    {
        if ( m_hArc != NULL )
        {
            return &m_foundInfo;
        }
        return NULL;
    }
 
    CAscii GetTargetArchiveFileName(void) const
    {
        return m_strArcFile;
    }
 
    CArchiverDllHandler* GetDllHandler(void) const
    {
        return m_pDll;
    }
 
protected:
 
    virtual bool OnStart(WIN32_FIND_DATA& _data, LPCTSTR lpszName)
    {
        if ( m_pDll != NULL )
        {
            Finish();
            m_hArc = m_pDll->OpenArchive(m_strArcFile, m_dwMode);
            if ( m_hArc != NULL )
            {
                int r;
                m_pDll->Func(r, "FindFirst", m_hArc, lpszName, &m_foundInfo);
                if ( r == 0 )
                {
                    return m_MakeWin32Info(_data);
                }
            }
        }
        return false;
    }
 
    virtual bool OnNext(WIN32_FIND_DATA& _data)
    {
        if ( m_pDll != NULL && m_hArc != NULL )
        {
            int r;
            m_pDll->Func(r, "FindNext", m_hArc, &m_foundInfo);
            if ( r == 0 )
            {
                return m_MakeWin32Info(_data);
            }
        }
        return false;
    }
 
    virtual void OnFinish(void)
    {
        if ( m_pDll != NULL && m_hArc != NULL )
        {
            m_pDll->CloseArchive(m_hArc);
            m_hArc = NULL;
        }
    }
};
 
 
 
}; // TNB