86 lines
1.4 KiB
C++
86 lines
1.4 KiB
C++
|
#include "DirIterator.h"
|
||
|
|
|
||
|
|
#include "Log.h"
|
||
|
|
#include "StringUtils.h"
|
||
|
|
|
||
|
|
#ifdef PLATFORM_UNIX
|
||
|
|
#include <dirent.h>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
DirIterator::DirIterator(const char* path)
|
||
|
|
{
|
||
|
|
m_path = path;
|
||
|
|
|
||
|
|
#ifdef PLATFORM_UNIX
|
||
|
|
m_dir = opendir(path);
|
||
|
|
m_entry = 0;
|
||
|
|
#else
|
||
|
|
// to list the contents of a directory, the first
|
||
|
|
// argument to FindFirstFile needs to be a wildcard
|
||
|
|
// of the form: C:\path\to\dir\*
|
||
|
|
std::string searchPath = m_path;
|
||
|
|
if (!endsWith(searchPath,"/"))
|
||
|
|
{
|
||
|
|
searchPath.append("/");
|
||
|
|
}
|
||
|
|
searchPath.append("*");
|
||
|
|
m_findHandle = FindFirstFile(searchPath.c_str(),&m_findData);
|
||
|
|
m_firstEntry = true;
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
DirIterator::~DirIterator()
|
||
|
|
{
|
||
|
|
#ifdef PLATFORM_UNIX
|
||
|
|
closedir(m_dir);
|
||
|
|
#else
|
||
|
|
FindClose(m_findHandle);
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
bool DirIterator::next()
|
||
|
|
{
|
||
|
|
#ifdef PLATFORM_UNIX
|
||
|
|
m_entry = readdir(m_dir);
|
||
|
|
return m_entry != 0;
|
||
|
|
#else
|
||
|
|
bool result;
|
||
|
|
if (m_firstEntry)
|
||
|
|
{
|
||
|
|
m_firstEntry = false;
|
||
|
|
return m_findHandle != INVALID_HANDLE_VALUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
result = FindNextFile(m_findHandle,&m_findData);
|
||
|
|
}
|
||
|
|
return result;
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string DirIterator::fileName() const
|
||
|
|
{
|
||
|
|
#ifdef PLATFORM_UNIX
|
||
|
|
return m_entry->d_name;
|
||
|
|
#else
|
||
|
|
return m_findData.cFileName;
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string DirIterator::filePath() const
|
||
|
|
{
|
||
|
|
return m_path + '/' + fileName();
|
||
|
|
}
|
||
|
|
|
||
|
|
bool DirIterator::isDir() const
|
||
|
|
{
|
||
|
|
#ifdef PLATFORM_UNIX
|
||
|
|
return m_entry->d_type == DT_DIR;
|
||
|
|
#else
|
||
|
|
return (m_findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|