From 7778f01194d84fae18827fd338299dd7482ef7c5 Mon Sep 17 00:00:00 2001 From: Vaxry Date: Tue, 23 Apr 2024 01:27:08 +0100 Subject: [PATCH] managers: Add a TokenManager --- CMakeLists.txt | 1 + src/Compositor.cpp | 4 ++++ src/managers/TokenManager.cpp | 43 +++++++++++++++++++++++++++++++++++ src/managers/TokenManager.hpp | 36 +++++++++++++++++++++++++++++ 4 files changed, 84 insertions(+) create mode 100644 src/managers/TokenManager.cpp create mode 100644 src/managers/TokenManager.hpp diff --git a/CMakeLists.txt b/CMakeLists.txt index d8aed5f2..d45600cf 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -247,6 +247,7 @@ target_link_libraries(Hyprland OpenGL::GL Threads::Threads libudis86 + uuid ) protocol("protocols/idle.xml" "idle" true) diff --git a/src/Compositor.cpp b/src/Compositor.cpp index a6c97bf7..8df52ab9 100644 --- a/src/Compositor.cpp +++ b/src/Compositor.cpp @@ -2,6 +2,7 @@ #include "helpers/Splashes.hpp" #include "config/ConfigValue.hpp" #include "managers/CursorManager.hpp" +#include "managers/TokenManager.hpp" #include "managers/eventLoop/EventLoopManager.hpp" #include #include @@ -456,6 +457,9 @@ void CCompositor::initManagers(eManagersInitStage stage) { Debug::log(LOG, "Creating the LayoutManager!"); g_pLayoutManager = std::make_unique(); + Debug::log(LOG, "Creating the TokenManager!"); + g_pTokenManager = std::make_unique(); + g_pConfigManager->init(); g_pWatchdog = std::make_unique(); // requires config } break; diff --git a/src/managers/TokenManager.cpp b/src/managers/TokenManager.cpp new file mode 100644 index 00000000..e48addd0 --- /dev/null +++ b/src/managers/TokenManager.cpp @@ -0,0 +1,43 @@ +#include "TokenManager.hpp" +#include +#include + +CUUIDToken::CUUIDToken(const std::string& uuid_, std::any data_, std::chrono::system_clock::duration expires) : data(data_), uuid(uuid_) { + expiresAt = std::chrono::system_clock::now() + expires; +} + +std::string CUUIDToken::getUUID() { + return uuid; +} + +std::string CTokenManager::registerNewToken(std::any data, std::chrono::system_clock::duration expires) { + std::string uuid; + do { + uuid_t uuid_; + uuid_generate_random(uuid_); + uuid = std::format("{:02x}{:02x}{:02x}{:02x}-{:02x}{:02x}-{:02x}{:02x}-{:02x}{:02x}-{:02x}{:02x}{:02x}{:02x}{:02x}{:02x}", (uint16_t)uuid_[0], (uint16_t)uuid_[1], (uint16_t)uuid_[2], + (uint16_t)uuid_[3], (uint16_t)uuid_[4], (uint16_t)uuid_[5], (uint16_t)uuid_[6], (uint16_t)uuid_[7], (uint16_t)uuid_[8], (uint16_t)uuid_[9], + (uint16_t)uuid_[10], (uint16_t)uuid_[11], (uint16_t)uuid_[12], (uint16_t)uuid_[13], (uint16_t)uuid_[14], (uint16_t)uuid_[15]); + } while (m_mTokens.contains(uuid)); + + m_mTokens[uuid] = std::make_shared(uuid, data, expires); + return uuid; +} + +std::shared_ptr CTokenManager::getToken(const std::string& uuid) { + + // cleanup expired tokens + const auto NOW = std::chrono::system_clock::now(); + std::erase_if(m_mTokens, [this, &NOW](const auto& el) { return el.second->expiresAt < NOW; }); + + if (!m_mTokens.contains(uuid)) + return {}; + + return m_mTokens.at(uuid); +} + +void CTokenManager::removeToken(std::shared_ptr token) { + if (!token) + return; + m_mTokens.erase(token->uuid); +} \ No newline at end of file diff --git a/src/managers/TokenManager.hpp b/src/managers/TokenManager.hpp new file mode 100644 index 00000000..45b0679a --- /dev/null +++ b/src/managers/TokenManager.hpp @@ -0,0 +1,36 @@ +#pragma once + +#include +#include +#include +#include +#include + +class CUUIDToken { + public: + CUUIDToken(const std::string& uuid_, std::any data_, std::chrono::system_clock::duration expires); + + std::string getUUID(); + + std::any data; + + private: + std::string uuid; + + std::chrono::system_clock::time_point expiresAt; + + friend class CTokenManager; +}; + +class CTokenManager { + public: + std::string registerNewToken(std::any data, std::chrono::system_clock::duration expires); + + std::shared_ptr getToken(const std::string& uuid); + void removeToken(std::shared_ptr token); + + private: + std::unordered_map> m_mTokens; +}; + +inline std::unique_ptr g_pTokenManager; \ No newline at end of file