ImHex/lib/libimhex/source/api/workspace_manager.cpp

#include <hex/api/workspace_manager.hpp>
#include <hex/api/layout_manager.hpp>

#include <hex/helpers/logger.hpp>
#include <hex/helpers/auto_reset.hpp>

#include <wolv/io/file.hpp>

#include <nlohmann/json.hpp>

#include <imgui.h>
#include <wolv/utils/string.hpp>

namespace hex {

    AutoReset<std::map<std::string, WorkspaceManager::Workspace>> WorkspaceManager::s_workspaces;
    decltype(WorkspaceManager::s_workspaces)::Type::iterator WorkspaceManager::s_currentWorkspace  = s_workspaces->end();
    decltype(WorkspaceManager::s_workspaces)::Type::iterator WorkspaceManager::s_previousWorkspace = s_workspaces->end();
    decltype(WorkspaceManager::s_workspaces)::Type::iterator WorkspaceManager::s_workspaceToRemove = s_workspaces->end();

    void WorkspaceManager::createWorkspace(const std::string& name, const std::string &layout) {
        s_currentWorkspace = s_workspaces->insert_or_assign(name, Workspace {
            .layout  = layout.empty() ? LayoutManager::saveToString() : layout,
            .path    = {},
            .builtin = false
        }).first;

        for (const auto &workspaceFolder : fs::getDefaultPaths(fs::ImHexPath::Workspaces)) {
            const auto workspacePath = workspaceFolder / (name + ".hexws");
            if (exportToFile(workspacePath)) {
                s_currentWorkspace->second.path = workspacePath;
                break;
            }
        }
    }

    void WorkspaceManager::switchWorkspace(const std::string& name) {
        const auto newWorkspace = s_workspaces->find(name);
        if (newWorkspace != s_workspaces->end()) {
            s_currentWorkspace = newWorkspace;
            log::info("Switching to workspace '{}'", name);
        }
    }

    void WorkspaceManager::importFromFile(const std::fs::path& path) {
        if (std::ranges::any_of(*s_workspaces, [path](const auto &pair) { return pair.second.path == path; }))
            return;

        wolv::io::File file(path, wolv::io::File::Mode::Read);
        if (!file.isValid()) {
            log::error("Failed to load workspace from file '{}'", path.string());
            file.remove();
            return;
        }

        auto content = file.readString();
        try {
            auto json = nlohmann::json::parse(content.begin(), content.end());

            const std::string name = json["name"];
            std::string layout = json["layout"];
            const bool builtin = json.value("builtin", false);

            (*s_workspaces)[name] = Workspace {
                .layout = std::move(layout),
                .path = path,
                .builtin = builtin
            };
        } catch (nlohmann::json::exception &e) {
            log::error("Failed to load workspace from file '{}': {}", path.string(), e.what());
            file.remove();
        }
    }

    bool WorkspaceManager::exportToFile(std::fs::path path, std::string workspaceName, bool builtin) {
        if (path.empty()) {
            if (s_currentWorkspace == s_workspaces->end())
                return false;

            path = s_currentWorkspace->second.path;
        }

        if (workspaceName.empty())
            workspaceName = s_currentWorkspace->first;

        wolv::io::File file(path, wolv::io::File::Mode::Create);

        if (!file.isValid())
            return false;

        nlohmann::json json;
        json["name"]    = workspaceName;
        json["layout"]  = LayoutManager::saveToString();
        json["builtin"] = builtin;

        file.writeString(json.dump(4));

        return true;
    }

    void WorkspaceManager::removeWorkspace(const std::string& name) {
        for (const auto &[workspaceName, workspace] : *s_workspaces) {
            if (workspaceName == name) {
                log::info("{}", wolv::util::toUTF8String(workspace.path));
                if (wolv::io::File(workspace.path, wolv::io::File::Mode::Write).remove()) {
                    log::info("Removed workspace '{}'", name);

                    switchWorkspace(s_workspaces->begin()->first);
                    s_workspaces->erase(workspaceName);
                } else {
                    log::error("Failed to remove workspace '{}'", name);
                }
                return;
            }
        }
    }


    void WorkspaceManager::process() {
        if (s_previousWorkspace != s_currentWorkspace) {
            log::info("Updating workspace");
            if (s_previousWorkspace != s_workspaces->end())
                exportToFile(s_previousWorkspace->second.path, s_previousWorkspace->first, s_previousWorkspace->second.builtin);

            LayoutManager::closeAllViews();
            ImGui::LoadIniSettingsFromMemory(s_currentWorkspace->second.layout.c_str());

            s_previousWorkspace = s_currentWorkspace;

            if (s_workspaceToRemove != s_workspaces->end()) {
                s_workspaces->erase(s_workspaceToRemove);
                s_workspaceToRemove = s_workspaces->end();
            }
        }
    }


    void WorkspaceManager::reset() {
        s_workspaces->clear();
        s_currentWorkspace  = s_workspaces->end();
        s_previousWorkspace = s_workspaces->end();
    }

    void WorkspaceManager::reload() {
        WorkspaceManager::reset();

        for (const auto &defaultPath : fs::getDefaultPaths(fs::ImHexPath::Workspaces)) {
            for (const auto &entry : std::fs::directory_iterator(defaultPath)) {
                if (!entry.is_regular_file())
                    continue;

                const auto &path = entry.path();
                if (path.extension() != ".hexws")
                    continue;

                WorkspaceManager::importFromFile(path);
            }
        }
    }




}