Add private/public key for users

1 files changed, 16 insertions, 4 deletions

diff --git a/include/LocalUser.hpp b/include/LocalUser.hpp
index 200f30f..04f483d 100644
--- a/include/LocalUser.hpp
+++ b/include/LocalUser.hpp
@@ -7,11 +7,23 @@ namespace odhtdb
     class LocalUser : public User
     {
     public:
-        static LocalUser* create(const std::string &name)
+        static LocalUser* create(const Signature::KeyPair &keyPair, const std::string &name)
         {
-            return new LocalUser(name);
+            return new LocalUser(keyPair, name);
         }
+        
+        const Signature::PublicKey& getPublicKey() const override
+        {
+            return keyPair.getPublicKey();
+        }
+        
+        const Signature::PrivateKey& getPrivateKey() const
+        {
+            return keyPair.getPrivateKey();
+        }
+    private:
+        LocalUser(const Signature::KeyPair &_keyPair, const std::string &name) : User(name), keyPair(_keyPair) {}
     private:
-        LocalUser(const std::string &name) : User(name){}
+        Signature::KeyPair keyPair;
     };
-}
\ No newline at end of file
+}