Connect channels to database, currently only locally

7 files changed, 192 insertions, 29 deletions

diff --git a/src/Cache.cpp b/src/Cache.cpp
index abfa1dd..bfa3df4 100644
--- a/src/Cache.cpp
+++ b/src/Cache.cpp
@@ -55,7 +55,7 @@ namespace dchat
     
     boost::filesystem::path Cache::getDchatDir()
     {
-        boost::filesystem::path dchatHomeDir = getHomeDir() / ".dchat";
+        boost::filesystem::path dchatHomeDir = getHomeDir() / ".local" / "share" / "dchat";
         boost::filesystem::create_directories(dchatHomeDir);
         return dchatHomeDir;
     }
diff --git a/src/Channel.cpp b/src/Channel.cpp
index f02b97a..ef252f6 100644
--- a/src/Channel.cpp
+++ b/src/Channel.cpp
@@ -1,61 +1,75 @@
 #include "../include/Channel.hpp"
 #include "../include/UsersSidePanel.hpp"
 #include "../include/ChannelSidePanel.hpp"
+#include <odhtdb/User.hpp>
+#include <odhtdb/Database.hpp>
 #include <cstring>
 
 using namespace std;
 
 namespace dchat
 {
-    Channel::Channel(const std::string &_name) : 
+    Channel::Channel(const string &_name, const odhtdb::DatabaseNode &_databaseNodeInfo, User *_localUser, odhtdb::Database *_database) : 
+        database(_database),
+        databaseNodeInfo(_databaseNodeInfo),
+        name(_name),
         messageBoard(sf::Vector2u(1.0f, 1.0f)),
-        localOfflineUser("You"),
-        name(_name)
+        localUser(_localUser ? _localUser : new OfflineUser("You"))
     {
+        addUser(localUser);
         {
-            Message *message = new Message(&localOfflineUser, u8"hello, worldåäö1![emoji](https://discordemoji.com/assets/emoji/playtime.png)");
+            Message *message = new Message(&systemUser, u8"hello, worldåäö1![emoji](https://discordemoji.com/assets/emoji/playtime.png)");
             messageBoard.addMessage(message);
         }
         
         {
-            Message *message = new Message(&localOfflineUser, u8"hello, world2![emoji](https://discordemoji.com/assets/emoji/Feels3DMan.gif)");
+            Message *message = new Message(&systemUser, u8"hello, world2![emoji](https://discordemoji.com/assets/emoji/Feels3DMan.gif)");
             messageBoard.addMessage(message);
         }
         
         {
-            Message *message = new Message(&localOfflineUser, u8"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.");
+            Message *message = new Message(&systemUser, u8"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.");
             messageBoard.addMessage(message);
         }
         
         {
-            Message *message = new Message(&localOfflineUser, u8"Lorem ipsumdolorsitamet,consecteturadipiscingelit,seddoeiusmodtemporincididuntutaboreetdoloremagnaaliqua.Utenimadminimveniam");
+            Message *message = new Message(&systemUser, u8"Lorem ipsumdolorsitamet,consecteturadipiscingelit,seddoeiusmodtemporincididuntutaboreetdoloremagnaaliqua.Utenimadminimveniam");
             messageBoard.addMessage(message);
         }
         
         {
-            Message *message = new Message(&localOfflineUser, u8"xddd");
+            Message *message = new Message(&systemUser, u8"xddd");
             messageBoard.addMessage(message);
         }
         
         {
-            Message *message = new Message(&localOfflineUser, u8"[emoji](https://discordemoji.com/assets/emoji/SlowbroDumb.png)");
+            Message *message = new Message(&systemUser, u8"[emoji](https://discordemoji.com/assets/emoji/SlowbroDumb.png)");
             messageBoard.addMessage(message);
         }
         
         {
-            Message *message = new Message(&localOfflineUser, u8"Message after big emoji");
+            Message *message = new Message(&systemUser, u8"Message after big emoji");
             messageBoard.addMessage(message);
         }
+        
+        if(database)
+            database->seed(databaseNodeInfo);
     }
     
     Channel::~Channel()
     {
+        if(database)
+            database->stopSeeding(*databaseNodeInfo.getRequestHash());
         
+        for(User *user : users)
+        {
+            delete user;
+        }
     }
     
     User* Channel::getLocalUser()
     {
-        return &localOfflineUser;
+        return localUser;
     }
     
     MessageBoard& Channel::getMessageBoard()
@@ -68,6 +82,47 @@ namespace dchat
         return name;
     }
     
+    const vector<User*> Channel::getUsers() const
+    {
+        return users;
+    }
+    
+    User* Channel::getUserByPublicKey(const odhtdb::Signature::PublicKey &publicKey)
+    {
+        auto userIt = publicKeyOnlineUsersMap.find(publicKey);
+        if(userIt != publicKeyOnlineUsersMap.end())
+            return userIt->second;
+        return nullptr;
+    }
+    
+    void Channel::addLocalMessage(const std::string &msg, User *owner)
+    {
+        assert(owner);
+        messageBoard.addMessage(new Message(owner, msg));
+    }
+    
+    void Channel::addMessage(const std::string &msg)
+    {
+        messageBoard.addMessage(new Message(localUser, msg));
+        if(database && localUser->type == User::Type::ONLINE)
+        {
+            auto onlineUser = static_cast<OnlineUser*>(localUser);
+            assert(onlineUser->databaseUser->getType() == odhtdb::User::Type::LOCAL);
+            database->addData(databaseNodeInfo, static_cast<const odhtdb::LocalUser*>(onlineUser->databaseUser), odhtdb::DataView((void*)msg.data(), msg.size()));
+            database->commit();
+        }
+    }
+    
+    void Channel::addUser(User *user)
+    {
+        users.push_back(user);
+        if(user->type == User::Type::ONLINE)
+        {
+            auto onlineUser = static_cast<OnlineUser*>(user);
+            publicKeyOnlineUsersMap[onlineUser->databaseUser->getPublicKey()] = onlineUser;
+        }
+    }
+    
     void Channel::processEvent(const sf::Event &event)
     {
         chatbar.processEvent(event, this);
diff --git a/src/ChannelSidePanel.cpp b/src/ChannelSidePanel.cpp
index a85dbdc..eb5c74c 100644
--- a/src/ChannelSidePanel.cpp
+++ b/src/ChannelSidePanel.cpp
@@ -20,6 +20,11 @@ namespace dchat
         channels.push_back(channel);
     }
     
+    void ChannelSidePanel::removeAllChannels()
+    {
+        channels.clear();
+    }
+    
     void ChannelSidePanel::draw(sf::RenderWindow &window)
     {
         auto windowSize = window.getSize();
diff --git a/src/Chatbar.cpp b/src/Chatbar.cpp
index 51ab9f5..241a808 100644
--- a/src/Chatbar.cpp
+++ b/src/Chatbar.cpp
@@ -235,7 +235,7 @@ namespace dchat
                         if(msg[0] == '/')
                             processChatCommand(StringView(msg.data() + 1, msg.size() - 1));
                         else
-                            channel->getMessageBoard().addMessage(new Message(channel->getLocalUser(), msg));
+                            channel->addMessage(msg);
                         clear();
                     }
                 }
diff --git a/src/User.cpp b/src/User.cpp
index 6b07b5b..2908fd4 100644
--- a/src/User.cpp
+++ b/src/User.cpp
@@ -1,8 +1,31 @@
 #include "../include/User.hpp"
+#include <odhtdb/User.hpp>
+#include <cassert>
 
 namespace dchat
 {
+    const static std::string SYSTEM_USER_NAME = "System";
+    
+    User::User(Type _type) : 
+        type(_type)
+    {
+        
+    }
+    
+    OnlineUser::OnlineUser(const odhtdb::User *_databaseUser) :
+        User(Type::ONLINE),
+        databaseUser(_databaseUser)
+    {
+        assert(databaseUser);
+    }
+    
+    const std::string& OnlineUser::getName() const
+    {
+        return databaseUser->getName();
+    }
+    
     OfflineUser::OfflineUser(const std::string &_name) : 
+        User(Type::OFFLINE),
         name(_name)
     {
         
@@ -12,4 +35,15 @@ namespace dchat
     {
         return name;
     }
+    
+    SystemUser::SystemUser() : 
+        User(Type::SYSTEM)
+    {
+        
+    }
+    
+    const std::string& SystemUser::getName() const
+    {
+        return SYSTEM_USER_NAME;
+    }
 }
diff --git a/src/UsersSidePanel.cpp b/src/UsersSidePanel.cpp
index 5af01bc..39c0b86 100644
--- a/src/UsersSidePanel.cpp
+++ b/src/UsersSidePanel.cpp
@@ -1,6 +1,7 @@
 #include "../include/UsersSidePanel.hpp"
 #include "../include/ResourceCache.hpp"
 #include "../include/Settings.hpp"
+#include "../include/Channel.hpp"
 #include <SFML/Graphics/RectangleShape.hpp>
 #include <SFML/Graphics/Text.hpp>
 #include <vector>
@@ -10,13 +11,18 @@ using namespace std;
 
 namespace dchat
 {
-    vector<User*> users;
+    Channel *currentChannel = nullptr;
     const float width = 200.0f;
     const unsigned int FONT_SIZE = 20;
     
-    void UsersSidePanel::addUser(User *user)
+    void UsersSidePanel::setCurrentChannel(Channel *channel)
     {
-        users.push_back(user);
+        currentChannel = channel;
+    }
+    
+    Channel* UsersSidePanel::getCurrentChannel()
+    {
+        return currentChannel;
     }
     
     void UsersSidePanel::draw(sf::RenderWindow &window)
@@ -27,9 +33,11 @@ namespace dchat
         rect.setPosition(windowSize.x - width, 0.0f);
         window.draw(rect);
         
+        if(!currentChannel) return;
+        
         const sf::Font &font = ResourceCache::getFont("fonts/Roboto-Regular.ttf");
         sf::Vector2f position(rect.getPosition().x + 10.0f, 10.0f);
-        for(User *user : users)
+        for(User *user : currentChannel->getUsers())
         {
             // TODO: Remove this shit
             sf::String str = sf::String::fromUtf8(user->getName().begin(), user->getName().end());
diff --git a/src/main.cpp b/src/main.cpp
index 52a19b9..6f54820 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -43,6 +43,24 @@ odhtdb::DatabaseNode createDatabaseNodeFromJoinKey(const string &joinKey)
     return result;
 }
 
+void channelAddStoredMessages(Channel *channel, const odhtdb::DatabaseStorageObjectList *nodeStorage)
+{
+    for(auto nodeStorageAddedObject : nodeStorage->objects)
+    {
+        if(nodeStorageAddedObject->decryptedObject.operation == odhtdb::DatabaseOperation::ADD_DATA)
+        {
+            User *user = channel->getUserByPublicKey(nodeStorageAddedObject->creatorPublicKey);
+            if(!user)
+            {
+                fprintf(stderr, "Missing user? %s\n", nodeStorageAddedObject->creatorPublicKey.toString().c_str());
+                continue;
+            }
+            string msg((const char*)nodeStorageAddedObject->decryptedObject.data.data, nodeStorageAddedObject->decryptedObject.data.size);
+            channel->addLocalMessage(msg, user);
+        }
+    }
+}
+
 int main(int argc, char **argv)
 {
     /*
@@ -77,16 +95,18 @@ int main(int argc, char **argv)
     
     Cache cache;
     
-    Channel channel("latenightshiendjs");
-    ChannelSidePanel::addChannel(&channel);
-    UsersSidePanel::addUser(channel.getLocalUser());
+    Channel offlineChannel("Offline");
+    ChannelSidePanel::addChannel(&offlineChannel);
+    UsersSidePanel::setCurrentChannel(&offlineChannel);
+    
+    vector<Channel*> channels;
     
     odhtdb::Signature::KeyPair *currentUserKeyPair = nullptr;
     vector<odhtdb::NodeLocalUser> localNodeUsers;
     string currentUserName;
     string currentUserPassword;
     
-    Command::add("login", [&currentUserKeyPair, &currentUserName, &currentUserPassword, &localNodeUsers, &database](const vector<string> &args)
+    Command::add("login", [&currentUserKeyPair, &currentUserName, &currentUserPassword, &localNodeUsers, &database, &channels](const vector<string> &args)
     {
         if(args.size() != 2)
         {
@@ -99,10 +119,39 @@ int main(int argc, char **argv)
             odhtdb::Signature::KeyPair keyPair = database.getStorage().decryptLocalEncryptedUser(args[0], args[1]);
             localNodeUsers = database.getStorage().getLocalNodeUsers(keyPair);
             
+            ChannelSidePanel::removeAllChannels();
+            for(Channel *channel : channels)
+            {
+                delete channel;
+            }
+            channels.clear();
+            
             for(auto localNodeUser : localNodeUsers)
             {
-                //const odhtdb::DatabaseStorageObjectList *nodeStorage = database.getStorage().getStorage(localNodeUser.nodeHash);
-                //Channel *channel = new Channel()
+                auto nodeStorage = database.getStorage().getStorage(localNodeUser.nodeHash);
+                if(!nodeStorage) continue;
+                
+                auto nodeDecryptionKeyResult = database.getStorage().getNodeDecryptionKey(localNodeUser.nodeHash);
+                if(!nodeDecryptionKeyResult.first) continue;
+                
+                User *newLocalUser = new OnlineUser(localNodeUser.localUser);
+                odhtdb::DatabaseNode databaseNode(nodeDecryptionKeyResult.second, make_shared<odhtdb::Hash>(localNodeUser.nodeHash));
+                Channel *channel = new Channel(nodeStorage->nodeName, databaseNode, newLocalUser, &database);
+                
+                auto nodeUserMapByPublicKey = database.getStorage().getNodeUsers(localNodeUser.nodeHash);
+                for(auto nodeUserIt : *nodeUserMapByPublicKey)
+                {
+                    if(nodeUserIt.second != localNodeUser.localUser)
+                    {
+                        User *newRemoteUser = new OnlineUser(nodeUserIt.second);
+                        channel->addUser(newRemoteUser);
+                    }
+                }
+                
+                ChannelSidePanel::addChannel(channel);
+                channels.push_back(channel);
+                UsersSidePanel::setCurrentChannel(channel);
+                channelAddStoredMessages(channel, nodeStorage);
             }
             
             printf("Successfully logged into user %s\n", args[0].c_str());
@@ -119,7 +168,7 @@ int main(int argc, char **argv)
         }
     });
     
-    Command::add("register", [&currentUserKeyPair, &currentUserName, &currentUserPassword, &database](const vector<string> &args)
+    Command::add("register", [&currentUserKeyPair, &currentUserName, &currentUserPassword, &localNodeUsers, &database](const vector<string> &args)
     {
         if(args.size() != 2)
         {
@@ -127,6 +176,12 @@ int main(int argc, char **argv)
             return;
         }
         
+        if(currentUserKeyPair)
+        {
+            fprintf(stderr, "You can't register a new account when you are logged in, please logout first\n");
+            return;
+        }
+        
         odhtdb::Signature::KeyPair keyPair;
         if(!database.getStorage().storeLocalUser(args[0], keyPair, args[1]))
         {
@@ -140,12 +195,13 @@ int main(int argc, char **argv)
         if(currentUserKeyPair)
             delete currentUserKeyPair;
         currentUserKeyPair = new odhtdb::Signature::KeyPair(keyPair);
+        localNodeUsers.clear();
         
         currentUserName = args[0];
         currentUserPassword = args[1];
     });
     
-    Command::add("cc", [&currentUserKeyPair, &currentUserName, &database](const vector<string> &args)
+    Command::add("cc", [&currentUserKeyPair, &currentUserName, &database, &channels](const vector<string> &args)
     {
         if(args.size() != 1)
         {
@@ -161,9 +217,14 @@ int main(int argc, char **argv)
         
         auto createResponse = database.create(currentUserName, *currentUserKeyPair, args[0]);
         database.commit();
-        odhtdb::DatabaseNode databaseNode(createResponse->getNodeEncryptionKey(), createResponse->getRequestHash());
-        database.seed(databaseNode);
         printf("Created database '%s', join key: '%s'\n", args[0].c_str(), createChannelJoinKey(createResponse).c_str());
+        
+        User *newLocalUser = new OnlineUser(createResponse->getNodeAdminUser());
+        odhtdb::DatabaseNode databaseNode(createResponse->getNodeEncryptionKey(), createResponse->getRequestHash());
+        Channel *channel = new Channel(args[0], databaseNode, newLocalUser, &database);
+        ChannelSidePanel::addChannel(channel);
+        channels.push_back(channel);
+        UsersSidePanel::setCurrentChannel(channel);
     });
     
     Command::add("jc", [&currentUserKeyPair, &database](const vector<string> &args)
@@ -210,11 +271,11 @@ int main(int argc, char **argv)
                 sf::View view(viewRect);
                 window.setView(view);
             }
-            channel.processEvent(event);
+            UsersSidePanel::getCurrentChannel()->processEvent(event);
         }
 
         window.clear(sf::Color(40, 40, 40));
-        channel.draw(window, cache);
+        UsersSidePanel::getCurrentChannel()->draw(window, cache);
         //video.draw(window);
         window.display();
     }