mirror of https://github.com/AxioDL/nod.git
74 lines
1.9 KiB
C++
74 lines
1.9 KiB
C++
#ifndef __NOD_DIRECTORY_ENUMERATOR__
|
|
#define __NOD_DIRECTORY_ENUMERATOR__
|
|
|
|
#include "Util.hpp"
|
|
|
|
namespace nod
|
|
{
|
|
|
|
struct CaseInsensitiveCompare
|
|
{
|
|
bool operator()(const std::string& lhs, const std::string& rhs) const
|
|
{
|
|
#if _WIN32
|
|
if (_stricmp(lhs.c_str(), rhs.c_str()) < 0)
|
|
#else
|
|
if (strcasecmp(lhs.c_str(), rhs.c_str()) < 0)
|
|
#endif
|
|
return true;
|
|
return false;
|
|
}
|
|
|
|
#if _WIN32
|
|
bool operator()(const std::wstring& lhs, const std::wstring& rhs) const
|
|
{
|
|
if (_wcsicmp(lhs.c_str(), rhs.c_str()) < 0)
|
|
return true;
|
|
return false;
|
|
}
|
|
#endif
|
|
};
|
|
|
|
class DirectoryEnumerator
|
|
{
|
|
public:
|
|
enum class Mode
|
|
{
|
|
Native,
|
|
DirsSorted,
|
|
FilesSorted,
|
|
DirsThenFilesSorted
|
|
};
|
|
struct Entry
|
|
{
|
|
SystemString m_path;
|
|
SystemString m_name;
|
|
size_t m_fileSz;
|
|
bool m_isDir;
|
|
|
|
private:
|
|
friend class DirectoryEnumerator;
|
|
Entry(SystemString&& path, const SystemChar* name, size_t sz, bool isDir)
|
|
: m_path(std::move(path)), m_name(name), m_fileSz(sz), m_isDir(isDir) {}
|
|
};
|
|
|
|
private:
|
|
std::vector<Entry> m_entries;
|
|
|
|
public:
|
|
DirectoryEnumerator(const SystemString& path, Mode mode=Mode::DirsThenFilesSorted,
|
|
bool sizeSort=false, bool reverse=false, bool noHidden=false)
|
|
: DirectoryEnumerator(path.c_str(), mode, sizeSort, reverse, noHidden) {}
|
|
DirectoryEnumerator(const SystemChar* path, Mode mode=Mode::DirsThenFilesSorted,
|
|
bool sizeSort=false, bool reverse=false, bool noHidden=false);
|
|
|
|
operator bool() const {return m_entries.size() != 0;}
|
|
size_t size() const {return m_entries.size();}
|
|
std::vector<Entry>::const_iterator begin() const {return m_entries.cbegin();}
|
|
std::vector<Entry>::const_iterator end() const {return m_entries.cend();}
|
|
};
|
|
|
|
}
|
|
|
|
#endif // __NOD_DIRECTORY_ENUMERATOR__
|