VirtualFileSystem.cpp

Go to the documentation of this file.

 
 
#include <entt/locator/locator.hpp>
#include <mimalloc.h>
#include <physfs.h>
#include <tinyfiledialogs.h>
#include <zip.h>
 
#include "galaxy/core/Settings.hpp"
#include "galaxy/logging/Log.hpp"
#include "galaxy/logging/PhysFSError.hpp"
#include "galaxy/platform/Pragma.hpp"
 
#include "VirtualFileSystem.hpp"
 
#ifdef GALAXY_WIN_PLATFORM
GALAXY_DISABLE_WARNING_PUSH
GALAXY_DISABLE_WARNING(26493)
GALAXY_DISABLE_WARNING(4244)
#endif
 
namespace galaxy
{
    VirtualFileSystem::VirtualFileSystem() noexcept
    {
        // Create data directories.
        std::filesystem::create_directories(settings::root_dir() / settings::assets_dir());
        std::filesystem::create_directories(settings::root_dir() / settings::editor_dir());
 
        // Set up allocators for physfs.
        PHYSFS_Allocator a = {};
        a.Init             = nullptr;
        a.Deinit           = nullptr;
        a.Free             = mi_free;
        a.Malloc           = mi_malloc;
        a.Realloc          = mi_realloc;
        log::physfs_check(PHYSFS_setAllocator(&a));
 
        // Set up vfs.
        if (log::physfs_check(PHYSFS_init(nullptr)))
        {
            PHYSFS_permitSymbolicLinks(false);
 
            auto write_dir = settings::root_dir() / settings::assets_dir();
            write_dir.make_preferred();
            log::physfs_check(PHYSFS_setWriteDir(write_dir.string().c_str()));
 
            auto read_dir = settings::root_dir() / settings::asset_pack();
            read_dir.make_preferred();
 
            if (settings::use_loose_assets())
            {
                mkdir(settings::assets_dir_music());
                mkdir(settings::assets_dir_sfx());
                mkdir(settings::assets_dir_voice());
                mkdir(settings::assets_dir_font());
                mkdir(settings::assets_dir_script());
                mkdir(settings::assets_dir_shaders());
                mkdir(settings::assets_dir_animation());
                mkdir(settings::assets_dir_texture());
                mkdir(settings::assets_dir_prefabs());
                mkdir(settings::assets_dir_maps());
                mkdir(settings::assets_dir_video());
                mkdir(settings::assets_dir_ui());
 
                read_dir = write_dir;
            }
 
            log::physfs_check(PHYSFS_mount(read_dir.string().c_str(), nullptr, true));
 
            const auto merged = settings::root_dir() / settings::editor_dir();
            log::physfs_check(PHYSFS_mount(merged.string().c_str(), nullptr, true));
        }
    }
 
    VirtualFileSystem::~VirtualFileSystem() noexcept
    {
        log::physfs_check(PHYSFS_deinit());
    }
 
    std::string VirtualFileSystem::read(const std::string& file)
    {
        PHYSFS_File* f = PHYSFS_openRead(file.c_str());
        if (log::physfs_check(f))
        {
            const auto len = PHYSFS_fileLength(f);
 
            if (log::physfs_check(len))
            {
                std::string buffer;
                buffer.resize(len);
 
                if (log::physfs_check(PHYSFS_readBytes(f, buffer.data(), buffer.size())))
                {
                    return buffer;
                }
                else
                {
                    GALAXY_LOG(GALAXY_ERROR, "Failed to read '{0}'.", file);
                }
            }
            else
            {
                GALAXY_LOG(GALAXY_ERROR, "Failed to check file length for '{0}'.", file);
            }
 
            log::physfs_check(PHYSFS_close(f));
        }
        else
        {
            GALAXY_LOG(GALAXY_ERROR, "Failed to read '{0}'.", file);
        }
 
        return {};
    }
 
    std::vector<std::uint8_t> VirtualFileSystem::read_binary(const std::string& file)
    {
        PHYSFS_File* f = PHYSFS_openRead(file.c_str());
        if (log::physfs_check(f))
        {
            const auto len = PHYSFS_fileLength(f);
 
            if (log::physfs_check(len))
            {
                std::vector<std::uint8_t> buffer;
                buffer.resize(len);
 
                if (log::physfs_check(PHYSFS_readBytes(f, &buffer[0], buffer.size())))
                {
                    return buffer;
                }
                else
                {
                    GALAXY_LOG(GALAXY_ERROR, "Failed to read '{0}'.", file);
                }
            }
            else
            {
                GALAXY_LOG(GALAXY_ERROR, "Failed to check file length for '{0}'.", file);
            }
 
            log::physfs_check(PHYSFS_close(f));
        }
        else
        {
            GALAXY_LOG(GALAXY_ERROR, "Failed to read '{0}'.", file);
        }
 
        return {};
    }
 
    bool VirtualFileSystem::write(const std::string& data, const std::string& file)
    {
        return write_raw(data.data(), data.size() * sizeof(char), file);
    }
 
    bool VirtualFileSystem::write_binary(std::span<std::uint8_t> data, const std::string& file)
    {
        return write_raw(data.data(), data.size_bytes(), file);
    }
 
    bool VirtualFileSystem::write_raw(const void* data, const std::size_t size, const std::string& file)
    {
        auto path = std::filesystem::path(file);
        if (path.is_absolute())
        {
            path = path.filename();
        }
 
        PHYSFS_File* f = PHYSFS_openWrite(path.string().c_str());
        if (log::physfs_check(f))
        {
            const auto len = PHYSFS_fileLength(f);
 
            if (log::physfs_check(PHYSFS_writeBytes(f, data, size)))
            {
                return true;
            }
            else
            {
                GALAXY_LOG(GALAXY_ERROR, "Failed to read '{0}'.", path.string());
            }
 
            log::physfs_check(PHYSFS_close(f));
        }
        else
        {
            GALAXY_LOG(GALAXY_ERROR, "Failed to read '{0}'.", path.string());
        }
 
        return false;
    }
 
    void VirtualFileSystem::mkdir(const std::string& dir) noexcept
    {
        if (!exists(dir))
        {
            log::physfs_check(PHYSFS_mkdir(dir.c_str()));
        }
    }
 
    void VirtualFileSystem::remove(const std::string& path) noexcept
    {
        log::physfs_check(PHYSFS_delete(path.c_str()));
    }
 
    bool VirtualFileSystem::exists(const std::string& file) noexcept
    {
        return PHYSFS_exists(file.c_str());
    }
 
    bool VirtualFileSystem::is_dir(const std::string& path) noexcept
    {
        return PHYSFS_isDirectory(path.c_str());
    }
 
    std::vector<std::string> VirtualFileSystem::list(const std::string& dir)
    {
        std::vector<std::string> list;
 
        log::physfs_check(PHYSFS_enumerate(
            dir.c_str(),
            [](void* data, const char* origdir, const char* fname) -> PHYSFS_EnumerateCallbackResult {
                if (data != nullptr && fname != nullptr)
                {
                    std::string o = origdir;
                    std::string f = fname;
 
                    auto* my_list = static_cast<std::vector<std::string>*>(data);
                    my_list->emplace_back(o + f);
                }
 
                return PHYSFS_ENUM_OK;
            },
            &list
        ));
 
        return list;
    }
 
    void VirtualFileSystem::alert() noexcept
    {
        tinyfd_beep();
    }
 
    void VirtualFileSystem::notification(const std::string& title, const std::string& msg, const fs::DialogIcon icon) noexcept
    {
        std::string tinyfd_icon {magic_enum::enum_name(icon)};
        tinyfd_notifyPopup(title.c_str(), msg.c_str(), tinyfd_icon.c_str());
    }
 
    int VirtualFileSystem::message_box(const std::string& title, const std::string& msg, const fs::DialogType type, const fs::DialogIcon icon, const fs::DialogButton btn) noexcept
    {
        std::string tinyfd_type {magic_enum::enum_name(type)};
        std::string tinyfd_icon {magic_enum::enum_name(icon)};
 
        return tinyfd_messageBox(title.c_str(), msg.c_str(), tinyfd_type.c_str(), tinyfd_icon.c_str(), static_cast<int>(btn));
    }
 
    std::string VirtualFileSystem::input_box(const std::string& title, const std::string& msg, const std::string& default_text, const bool password) noexcept
    {
        const char* dt = password ? nullptr : default_text.c_str();
        return tinyfd_inputBox(title.c_str(), msg.c_str(), dt);
    }
 
    std::string VirtualFileSystem::open_save_dialog(const std::string& default_filename, const std::vector<const char*>& filters)
    {
        const char* const* filter_patterns = (filters.size() > 0) ? filters.data() : nullptr;
        const char*        result          = tinyfd_saveFileDialog("Save file", default_filename.c_str(), static_cast<int>(filters.size()), filter_patterns, nullptr);
 
        if (result != nullptr)
        {
            return {result};
        }
        else
        {
            return {};
        }
    }
 
    std::string VirtualFileSystem::open_file_dialog(const std::vector<const char*>& filters, const std::string& def_path)
    {
        const auto default_path = (settings::root_dir() / def_path).string();
 
        const char* const* filter_patterns = (filters.size() > 0) ? filters.data() : nullptr;
        const char*        result          = tinyfd_openFileDialog("Open file", default_path.c_str(), static_cast<int>(filters.size()), filter_patterns, "Select a file", false);
 
        if (result != nullptr)
        {
            return {result};
        }
        else
        {
            return {};
        }
    }
 
    std::string VirtualFileSystem::select_folder_dialog(const std::string& def_path)
    {
        const auto  default_path = (settings::root_dir() / def_path).string();
        const char* result       = tinyfd_selectFolderDialog("Select folder", default_path.c_str());
 
        if (result != nullptr)
        {
            return {result};
        }
        else
        {
            return {};
        }
    }
} // namespace galaxy
 
#ifdef GALAXY_WIN_PLATFORM
GALAXY_DISABLE_WARNING_POP
#endif