From 1e4dac83952e1f707cdb837f647e6c80c3414154 Mon Sep 17 00:00:00 2001
From: dec05eba <dec05eba@protonmail.com>
Date: Sat, 28 Apr 2018 15:30:21 +0200
Subject: Add stop seeding function

---
 src/DatabaseStorage.cpp | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'src/DatabaseStorage.cpp')

diff --git a/src/DatabaseStorage.cpp b/src/DatabaseStorage.cpp
index 0e9fa32..384ef5f 100644
--- a/src/DatabaseStorage.cpp
+++ b/src/DatabaseStorage.cpp
@@ -692,6 +692,14 @@ namespace odhtdb
         return nullptr;
     }
     
+    const DataViewMap<Group*>* DatabaseStorage::getNodeGroups(const Hash &nodeHash)
+    {
+        auto groupByIdMapIt = nodeGroupByIdMap.find(nodeHash);
+        if(groupByIdMapIt != nodeGroupByIdMap.end())
+            return groupByIdMapIt->second;
+        return nullptr;
+    }
+    
     Group* DatabaseStorage::getGroupById(const Hash &nodeHash, uint8_t groupId[GROUP_ID_LENGTH]) const
     {
         auto groupByIdMapIt = nodeGroupByIdMap.find(nodeHash);
@@ -704,6 +712,14 @@ namespace odhtdb
         return nullptr;
     }
     
+    const Signature::MapPublicKey<User*>* DatabaseStorage::getNodeUsers(const Hash &nodeHash)
+    {
+        auto publicKeyUserDataMapIt = nodePublicKeyUserDataMap.find(nodeHash);
+        if(publicKeyUserDataMapIt != nodePublicKeyUserDataMap.end())
+            return publicKeyUserDataMapIt->second;
+        return nullptr;
+    }
+    
     User* DatabaseStorage::getUserByPublicKey(const Hash &nodeHash, const Signature::PublicKey &userPublicKey) const
     {
         auto publicKeyUserDataMapIt = nodePublicKeyUserDataMap.find(nodeHash);
@@ -806,6 +822,14 @@ namespace odhtdb
         return localUsers;
     }
     
+    std::pair<bool, std::shared_ptr<OwnedMemory>> DatabaseStorage::getNodeDecryptionKey(const Hash &nodeHash)
+    {
+        auto nodeDecryptionKeyIt = nodeDecryptionKeyMap.find(nodeHash);
+        if(nodeDecryptionKeyIt != nodeDecryptionKeyMap.end())
+            return make_pair(true, nodeDecryptionKeyIt->second);
+        return make_pair(false, make_shared<OwnedMemory>());
+    }
+    
     void DatabaseStorage::setNodeDecryptionKey(const Hash &nodeHash, const DataView &decryptionKeyView)
     {  
         bool nodeHasExistingEncryptionKey = nodeDecryptionKeyMap.find(nodeHash) != nodeDecryptionKeyMap.end();
-- 
cgit v1.2.3