managers: Add a TokenManager

1 files changed, 43 insertions, 0 deletions

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 <uuid/uuid.h>
+#include <algorithm>
+
+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<CUUIDToken>(uuid, data, expires);
+    return uuid;
+}
+
+std::shared_ptr<CUUIDToken> 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<CUUIDToken> token) {
+    if (!token)
+        return;
+    m_mTokens.erase(token->uuid);
+}
\ No newline at end of file