Only allow owner of message to delete it

2 files changed, 4 insertions, 3 deletions

diff --git a/include/Channel.hpp b/include/Channel.hpp
index 7d4cb08..93f3725 100644
--- a/include/Channel.hpp
+++ b/include/Channel.hpp
@@ -47,8 +47,8 @@ namespace dchat
         void addLocalMessage(const std::string &msg, User *owner, u64 timestampSeconds = 0);
         void addLocalMessage(const std::string &msg, User *owner, u64 timestampSeconds, const odhtdb::Hash &id);
         void addMessage(const std::string &msg);
-        void deleteLocalMessage(const odhtdb::Hash &id);
-        void deleteMessage(const odhtdb::Hash &id);
+        void deleteLocalMessage(const odhtdb::Hash &id, const odhtdb::Signature::PublicKey &requestedByUser);
+        void deleteMessage(const odhtdb::Hash &id, const odhtdb::Signature::PublicKey &requestedByUser);
         
         void addUserLocally(User *user);
         bool addUser(const odhtdb::Signature::PublicKey &userId, const std::string &groupId);
diff --git a/include/MessageBoard.hpp b/include/MessageBoard.hpp
index a6867d2..9025e48 100644
--- a/include/MessageBoard.hpp
+++ b/include/MessageBoard.hpp
@@ -7,6 +7,7 @@
 #include <SFML/Graphics/RenderWindow.hpp>
 #include <SFML/Window/Event.hpp>
 #include <odhtdb/Hash.hpp>
+#include <odhtdb/Signature.hpp>
 #include <stdexcept>
 #include <mutex>
 
@@ -26,7 +27,7 @@ namespace dchat
     private:
         void updateStaticContentTexture(const sf::Vector2u &newSize);
         void addMessage(Message *message, const odhtdb::Hash &id);
-        void deleteMessage(const odhtdb::Hash &id);
+        void deleteMessage(const odhtdb::Hash &id, const odhtdb::Signature::PublicKey &requestedByUser);
         
         void drawDefault(sf::RenderWindow &window, Cache &cache);
         void drawSimple(sf::RenderWindow &window, Cache &cache);