TnbRapiFileFinder.h

[詳解]

#pragma once
#include "TnbFileFinder.h"
#include "TnbStr.h"
#include "TnbRapi.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CRapiFileFinder : public CAbstractFileFinder
{
    DEFSUPER(CAbstractFileFinder);
public:
 
    CRapiFileFinder(void) : _super(), m_hFind(INVALID_HANDLE_VALUE)
    {
        CRapi::Startup();
    }
 
    virtual ~CRapiFileFinder(void)
    {
        Finish();
    }
 
protected:
 
    virtual bool OnStart(WIN32_FIND_DATA& _data, LPCTSTR lpszName)
    {
        if ( ! CRapi::IsSupported() ) { return false; }
        Finish();
        CE_FIND_DATA ceFd;
        m_hFind = ::CeFindFirstFile(CUnicode(lpszName), &ceFd);
        bool r = (m_hFind != INVALID_HANDLE_VALUE);
        if ( r )
        {
            m_CeFindDataToWin32(_data, ceFd);
        }
        return r;
    }
 
    virtual bool OnNext(WIN32_FIND_DATA& _data)
    {
        if ( ! CRapi::IsSupported() ) { return false; }
        if ( m_hFind != INVALID_HANDLE_VALUE )
        {
            CE_FIND_DATA ceFd;
            bool r = !! ::CeFindNextFile(m_hFind, &ceFd);
            if ( r )
            {
                m_CeFindDataToWin32(_data, ceFd);
            }
            return r;
        }
        return false;
    }
 
    virtual void OnFinish(void)
    {
        if ( CRapi::IsSupported() && m_hFind != INVALID_HANDLE_VALUE )
        {
            ::CeFindClose(m_hFind);
            m_hFind = INVALID_HANDLE_VALUE;
        }
    }
 
private:
    HANDLE              m_hFind;    
    void m_CeFindDataToWin32(WIN32_FIND_DATA& _winFd, const CE_FIND_DATA& ceFd)
    {
        Zero(_winFd);
        _winFd.dwFileAttributes = ceFd.dwFileAttributes;
        _winFd.ftCreationTime = ceFd.ftCreationTime;
        _winFd.ftLastAccessTime = ceFd.ftLastAccessTime;
        _winFd.ftLastWriteTime = ceFd.ftLastWriteTime;
        _winFd.nFileSizeHigh = ceFd.nFileSizeHigh;
        _winFd.nFileSizeLow = ceFd.nFileSizeLow;
        CStr s = ceFd.cFileName;
        STRLIB::Copy(_winFd.cFileName, s);
    }
    friend class CCeFileFinderTest;
};
 
 
 
}; //TNB