DirIterator.hxx

#pragma once
 
// cosmos
#include <cosmos/fs/DirStream.hxx>
 
namespace cosmos {
 
 
class DirIterator {
public: // functions
 
        explicit DirIterator(DirStream &dir, bool at_end) :
                        m_dir{dir} {
                if (!at_end)
                        m_entry = dir.nextEntry();
        }
 
        bool operator==(const DirIterator &other) const {
                if (m_entry.has_value() != other.m_entry.has_value())
                        return false;
                else if (!m_entry.has_value())
                        // both are at the end
                        return true;
 
                return m_entry->inode() == other.m_entry->inode();
        }
 
        bool operator!=(const DirIterator &other) const {
                return !(*this == other);
        }
 
        auto& operator++() {
                m_entry = m_dir.nextEntry();
                return *this;
        }
 
        DirEntry& operator*() {
                return *m_entry;
        }
 
protected: // data
        DirStream &m_dir;
        std::optional<DirEntry> m_entry;
};
 
inline DirIterator end(DirStream &dir) {
        return DirIterator{dir, true};
}
 
 
inline DirIterator begin(DirStream &dir) {
        if (!dir.isOpen())
                return end(dir);
 
        // make sure we really start from the beginning.
        dir.rewind();
 
        return DirIterator{dir, false};
}
 
} // end ns