1 files changed, 8 insertions, 5 deletions
diff --git a/include/User.hpp b/include/User.hpp
index ab5872a..fb37876 100644
--- a/include/User.hpp
+++ b/include/User.hpp
@@ -3,9 +3,12 @@
 #include "Signature.hpp"
 #include <string>
 #include <stdexcept>
+#include <vector>
 
 namespace odhtdb
 {
+    class Group;
+    
     class UserNameTooLongException : public std::runtime_error
     {
     public:
@@ -21,15 +24,15 @@ namespace odhtdb
     public:
         virtual ~User(){}
         
+        void addToGroup(Group *group);
+        
         const std::string& getName() const { return name; }
+        const std::vector<Group*>& getGroups() const { return groups; }
         virtual const Signature::PublicKey& getPublicKey() const = 0;
     protected:
-        User(const std::string &_name) : name(_name)
-        {
-            if(name.size() > 255)
-                throw UserNameTooLongException(name);
-        }
+        User(const std::string &name, Group *group);
     private:
         std::string name;
+        std::vector<Group*> groups;
     };
 }