AuroraRuntime/Source/IO/FS/FS.hpp

/***
    Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.

    File: FS.hpp
    Date: 2021-6-16
    Author: Reece
***/
#pragma once

#if !defined(NO_FSTREAM) && (!defined(__has_include) || __has_include(<fstream>))
    #include <fstream>
    #if !defined(HAS_FSTREAM)
        #define HAS_FSTREAM
    #endif
#endif

#if !defined(NO_STD_FS) && (!defined(__has_include) || __has_include(<filesystem>))
    #include <filesystem>
    #define HAS_STD_FS

    #if !defined(_HAS_CXX17) && defined(AURORA_COMPILER_MSVC)
        // map experimental::filesystem to filesystem::
        // as in newer compilers it is not experimental anymore
        namespace std
        {
            namespace filesystem = experimental::filesystem;
        }
    #endif

#endif 

namespace Aurora::IO::FS
{
    #if defined(VENDOR_GENERIC_MICROSOFT) || defined(VENDOR_CONSOLE_MICROSOFT)
        static const unsigned char kPathSplitter = '\\';
        static const AuString kDoublePathSplitter = "\\\\";
    #else
        static const unsigned char kPathSplitter = '/';
        static const AuString kDoublePathSplitter = "//";
    #endif 

    void _NormalizePath(AuString &str);
    
    static void NormalizePath(AuString &str)
    {
        _NormalizePath(str);
    }

    static inline AuString NormalizePathRet(const AuString &str)
    {
        AuString ret = str;
        _NormalizePath(ret);
        return ret;
    }

    bool _MkDir(const AuString &str);

    static bool CreateDirectories(const AuString &cpath, bool isFile)
    {
        for (int i = 0; i < cpath.size(); i++)
        {
            bool end = i == cpath.size() - 1;
            if ((cpath[i] == kPathSplitter) || 
                ((!isFile) && (end)))
            {
                auto subpath = end ? cpath : AuString(cpath.begin(), cpath.begin() + i);

                if (DirExists(subpath))
                {
                    continue;
                }

                if (!_MkDir(subpath))
                {
                    if (!DirExists(subpath)) // copium. im worried :(
                    {
                        return false; // ...aw shit
                    }
                }
            }
        }

        return true;
    }

    static bool IterateDirEntriesSTL(const AuString &path, bool filesOnly, AuList<AuString> &patches)
    {
    #if defined(HAS_STD_FS)
        auto normalizedPath = NormalizePathRet(path);
        try
        {
            for (const auto &entry : std::filesystem::directory_iterator(normalizedPath))
            {
                auto cpath = entry.path();

                if (filesOnly && !std::filesystem::is_regular_file(cpath))
                    continue;

                if (!filesOnly && !std::filesystem::is_directory(cpath))
                    continue;

                AuString path = cpath.string();
                patches.push_back(path.substr(path.find_last_of(kPathSplitter) + 1));
            }

            return true;
        }
        catch (...)
        {
            Debug::PrintError();
            Console::Logging::LogWarn("IO Error, couldn't iterate path: {}", normalizedPath);
            return false;
        }
    #endif

        return false;
    }

    void InitResources();
}