tv_rename/filesystem.hpp

#ifndef FSLIB_H
#define FSLIB_H

#include <string.h>
#include <string>

#ifdef _WIN32

#include <Windows.h>

#else

#include <dirent.h>

#endif

// set apropriate data types for each operating system
#ifdef _WIN32

using string = std::wstring;
using char_t = wchar_t;

#else

using string = std::string;
using char_t = char;

#endif

// windows version stolen from
// http://www.martinbroadhurst.com/list-the-files-in-a-directory-in-c.html

namespace FSLib {

bool exists( const string &path );
bool isDirectory( const string &path );
bool rename( const string &file_a, const string &file_b );
string canonical( const string &path );

class Directory {
public:
    Directory( const string &path_ );
    Directory() = delete;
    Directory( const Directory &d ) = default;
    Directory( Directory &&d ) = default;

    class Iterator {
    public:
        Iterator( const Directory &d_ );
        ~Iterator();

#ifdef _WIN32
        Iterator( bool ended_ );
#else
        Iterator( const Directory &d_, const struct dirent *current_entry_ );
#endif

        Iterator() = delete;

        Iterator( const Iterator &i ) = default;

        Iterator( Iterator &&i ) = default;

        char_t const *operator*() const;

        Iterator &operator++();

        bool operator==( const Iterator &i_other ) const;

        Iterator operator++( int ) {
            Iterator ret( *this );
            operator++();
            return ret;
        }

        bool operator!=( const Iterator &i_other ) const {
            return !( i_other == *this );
        }

    private:
#ifndef _WIN32
        DIR *d;
        const struct dirent *current_entry;
#else
        HANDLE hFind;
        WIN32_FIND_DATA data;
        bool ended{ false };
#endif
    };

    using iterator = Iterator;
    using const_iterator = Iterator;

    iterator end();

    const_iterator end() const;

    iterator begin() {
        return Iterator( *this );
    }

    const_iterator begin() const {
        return Iterator( *this );
    }

    const_iterator cbegin() const {
        return begin();
    }

    const_iterator cend() const {
        return end();
    }

    const char_t *path() const {
        return dir_path.c_str();
    }

private:
    string dir_path;
};
} // namespace FSLib

#endif