Support different types of channel data. Add nickname change command

1 files changed, 100 insertions, 21 deletions

diff --git a/src/main.cpp b/src/main.cpp
index 59b89a5..3804f6f 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -56,17 +56,67 @@ odhtdb::DatabaseNode createDatabaseNodeFromJoinKey(const string &joinKey)
     return odhtdb::DatabaseNode(encryptionKey, nodeHash);
 }
 
-void channelAddStoredMessage(Channel *channel, odhtdb::DatabaseStorageObject *nodeStorageObject)
+void channelChangeUserNickname(Channel *channel, const StringView data, const odhtdb::Signature::PublicKey &userPublicKey)
 {
-    User *user = channel->getUserByPublicKey(nodeStorageObject->creatorPublicKey);
+    auto user = channel->getUserByPublicKey(userPublicKey);
     if(!user)
     {
-        fprintf(stderr, "Missing user? %s\n", nodeStorageObject->creatorPublicKey.toString().c_str());
+        fprintf(stderr, "Nickname change: user with public key %s not found in channel %s\n", userPublicKey.toString().c_str(), channel->getName().c_str());
         return;
     }
     
-    string msg((const char*)nodeStorageObject->decryptedObject.data.data, nodeStorageObject->decryptedObject.data.size);
-    channel->addLocalMessage(msg, user, ntp::NtpTimestamp::fromCombined(nodeStorageObject->createdTimestamp).seconds);
+    sibs::SafeDeserializer deserializer((const u8*)data.data, data.size);
+    u8 nameLength = deserializer.extract<u8>();
+    if(nameLength > 0)
+    {
+        user->name.resize(nameLength);
+        deserializer.extract((u8*)&user->name[0], nameLength);
+    }
+    // We dont care if there is more data to read (malicious packet), we already got all the data we need
+}
+
+void channelAddStoredMessage(Channel *channel, const odhtdb::Signature::PublicKey &creatorPublicKey, const StringView decryptedObject, u64 timestamp)
+{
+    User *user = channel->getUserByPublicKey(creatorPublicKey);
+    if(!user)
+    {
+        fprintf(stderr, "Missing user? %s\n", creatorPublicKey.toString().c_str());
+        return;
+    }
+    
+    if(decryptedObject.size > 1)
+    {
+        auto channelDataType = (ChannelDataType)static_cast<const char*>(decryptedObject.data)[0];
+        StringView decryptedData((const char*)decryptedObject.data + 1, decryptedObject.size - 1);
+        switch(channelDataType)
+        {
+            case ChannelDataType::MESSAGE:
+            {
+                string msg(decryptedData.data, decryptedData.size);
+                channel->addLocalMessage(msg, user, ntp::NtpTimestamp::fromCombined(timestamp).seconds);
+                break;
+            }
+            case ChannelDataType::NICKNAME_CHANGE:
+            {
+                try
+                {
+                    channelChangeUserNickname(channel, decryptedData, creatorPublicKey);
+                }
+                catch(sibs::DeserializeException &e)
+                {
+                    fprintf(stderr, "Failed to deserialize nick change\n");
+                }
+                break;
+            }
+            default:
+                fprintf(stderr, "Got unexpected channel data type: %u\n", channelDataType);
+                break;
+        }
+    }
+    else
+    {
+        fprintf(stderr, "ADD_DATA packet too small, ignoring...\n");
+    }
 }
 
 void channelAddStoredMessages(Channel *channel, const odhtdb::DatabaseStorageObjectList *nodeStorage)
@@ -76,7 +126,8 @@ void channelAddStoredMessages(Channel *channel, const odhtdb::DatabaseStorageObj
     {
         if(nodeStorageAddedObject->decryptedObject.operation == odhtdb::DatabaseOperation::ADD_DATA)
         {
-            channelAddStoredMessage(channel, nodeStorageAddedObject);
+            StringView decryptedObject((const char*)nodeStorageAddedObject->decryptedObject.data.data, nodeStorageAddedObject->decryptedObject.data.size);
+            channelAddStoredMessage(channel, nodeStorageAddedObject->creatorPublicKey, decryptedObject, nodeStorageAddedObject->createdTimestamp);
         }
     }
 }
@@ -149,15 +200,7 @@ int main(int argc, char **argv)
         {
             if(*request.nodeHash == *channel->getNodeInfo().getRequestHash())
             {
-                User *user = channel->getUserByPublicKey(request.creatorUser->getPublicKey());
-                if(!user)
-                {
-                    fprintf(stderr, "Missing user? %s\n", request.creatorUser->getPublicKey().toString().c_str());
-                    return;
-                }
-                
-                string msg((const char*)request.decryptedData.data, request.decryptedData.size);
-                channel->addLocalMessage(msg, user, ntp::NtpTimestamp::fromCombined(request.timestamp).seconds);
+                channelAddStoredMessage(channel, request.creatorUser->getPublicKey(), StringView((const char*)request.decryptedData.data, request.decryptedData.size), request.timestamp);
                 return;
             }
         }
@@ -427,9 +470,9 @@ int main(int argc, char **argv)
     {
         if(args.size() != 2)
         {
-            string errMsg = "Expected 2 arguments for command addbind, got %u argument(";
+            string errMsg = "Expected 2 arguments for command addbind, got ";
             errMsg += to_string(args.size());
-            errMsg += ")";
+            errMsg += " argument(s)";
             Channel::getCurrent()->addLocalMessage(errMsg, Channel::getCurrent()->getSystemUser());
             return;
         }
@@ -456,9 +499,9 @@ int main(int argc, char **argv)
     {
         if(args.size() != 1)
         {
-            string errMsg = "Expected 1 argument for command removebind, got %u argument(";
+            string errMsg = "Expected 1 argument for command removebind, got ";
             errMsg += to_string(args.size());
-            errMsg += ")";
+            errMsg += " argument(s)";
             Channel::getCurrent()->addLocalMessage(errMsg, Channel::getCurrent()->getSystemUser());
             return;
         }
@@ -485,9 +528,9 @@ int main(int argc, char **argv)
     {
         if(args.size() != 0)
         {
-            string errMsg = "Expected 0 arguments for command removebind, got %u argument(";
+            string errMsg = "Expected 0 arguments for command removebind, got ";
             errMsg += to_string(args.size());
-            errMsg += ")";
+            errMsg += " argument(s)";
             Channel::getCurrent()->addLocalMessage(errMsg, Channel::getCurrent()->getSystemUser());
             return;
         }
@@ -521,6 +564,42 @@ int main(int argc, char **argv)
         Channel::getCurrent()->addLocalMessage(response, Channel::getCurrent()->getSystemUser());
     });
     
+    // Change nick of current user in current channel
+    Command::add("nick", [&currentUserKeyPair, &offlineChannel](const vector<string> &args)
+    {
+        if(args.size() != 1)
+        {
+            string errMsg = "Expected 1 argument for command nick, got ";
+            errMsg += to_string(args.size());
+            errMsg += " argument(s)";
+            Channel::getCurrent()->addLocalMessage(errMsg, Channel::getCurrent()->getSystemUser());
+            return;
+        }
+        
+        if(!currentUserKeyPair)
+        {
+            Channel::getCurrent()->addLocalMessage("You need to be logged in to change your nickname", Channel::getCurrent()->getSystemUser());
+            return;
+        }
+        
+        if(Channel::getCurrent() == &offlineChannel)
+        {
+            Channel::getCurrent()->addLocalMessage("You need to be in a channel to change your nickname", Channel::getCurrent()->getSystemUser());
+            return;
+        }
+        
+        if(args[0].size() == 0 || args[0].size() > 255)
+        {
+            Channel::getCurrent()->addLocalMessage("Invalid nickname. Nickname has to be between 1 and 255 characters", Channel::getCurrent()->getSystemUser());
+            return;
+        }
+        
+        Channel::getCurrent()->changeNick(args[0]);
+        string msg = "Your nickname was changed to ";
+        msg += args[0];
+        Channel::getCurrent()->addLocalMessage(msg, Channel::getCurrent()->getSystemUser());
+    });
+    
     // Get channel join key
     Command::add("joinkey", [&currentUserKeyPair, &currentUserName, &offlineChannel](const vector<string> &args)
     {