Add private/public key for users

1 files changed, 5 insertions, 2 deletions

diff --git a/include/User.hpp b/include/User.hpp
index e542434..ab5872a 100644
--- a/include/User.hpp
+++ b/include/User.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include "Signature.hpp"
 #include <string>
 #include <stdexcept>
 
@@ -18,7 +19,10 @@ namespace odhtdb
     class User
     {
     public:
+        virtual ~User(){}
+        
         const std::string& getName() const { return name; }
+        virtual const Signature::PublicKey& getPublicKey() const = 0;
     protected:
         User(const std::string &_name) : name(_name)
         {
@@ -26,7 +30,6 @@ namespace odhtdb
                 throw UserNameTooLongException(name);
         }
     private:
-        // TODO: Add public key
         std::string name;
     };
-}
\ No newline at end of file
+}