Fix incoming messages on host side.

dchapyshev · Apr 10, 2022 · 8776f54 · 8776f54
1 parent 96c3b9f
commit 8776f54
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 14 deletions.
diff --git a/source/host/client_session_text_chat.cc b/source/host/client_session_text_chat.cc
@@ -59,15 +59,6 @@ bool ClientSessionTextChat::hasUser() const
 void ClientSessionTextChat::setHasUser(bool enable)
 {
     has_user_ = enable;
-
-    if (has_user_)
-    {
-        sendStatus(proto::TextChatStatus::STATUS_USER_CONNECTED);
-    }
-    else
-    {
-        sendStatus(proto::TextChatStatus::STATUS_USER_DISCONNECTED);
-    }
 }
 
 void ClientSessionTextChat::onMessageReceived(const base::ByteArray& buffer)

diff --git a/source/host/user_session.cc b/source/host/user_session.cc
@@ -167,7 +167,7 @@ void UserSession::start(const proto::internal::RouterState& router_state)
         sendRouterState(FROM_HERE);
         sendCredentials(FROM_HERE);
 
-        setTextChatHasUser(true);
+        onTextChatHasUser(FROM_HERE, true);
     }
     else
     {
@@ -214,7 +214,7 @@ void UserSession::restart(std::unique_ptr<base::IpcChannel> channel)
         sendRouterState(FROM_HERE);
         sendCredentials(FROM_HERE);
 
-        setTextChatHasUser(true);
+        onTextChatHasUser(FROM_HERE, true);
     }
     else
     {
@@ -963,7 +963,7 @@ void UserSession::onSessionDettached(const base::Location& location)
     for (const auto& client : file_transfer_clients_)
         client->stop();
 
-    setTextChatHasUser(false);
+    onTextChatHasUser(FROM_HERE, false);
 
     setState(FROM_HERE, State::DETTACHED);
 
@@ -1217,7 +1217,18 @@ void UserSession::addNewClientSession(std::unique_ptr<ClientSession> client_sess
     sendConnectEvent(*client_session_ptr);
 
     if (client_session_ptr->sessionType() == proto::SESSION_TYPE_TEXT_CHAT)
+    {
         onTextChatSessionStarted(client_session_ptr->id());
+
+        bool has_user = channel_ != nullptr;
+        for (const auto& client : text_chat_clients_)
+        {
+            ClientSessionTextChat* text_chat_client =
+                static_cast<ClientSessionTextChat*>(client.get());
+
+            text_chat_client->setHasUser(has_user);
+        }
+    }
 }
 
 void UserSession::setState(const base::Location& location, State state)
@@ -1227,14 +1238,21 @@ void UserSession::setState(const base::Location& location, State state)
     state_ = state;
 }
 
-void UserSession::setTextChatHasUser(bool has_user)
+void UserSession::onTextChatHasUser(const base::Location& location, bool has_user)
 {
+    LOG(LS_INFO) << "User state changed: " << has_user << " (from: " << location.toString() << ")";
+
     for (const auto& client : text_chat_clients_)
     {
         ClientSessionTextChat* text_chat_client =
             static_cast<ClientSessionTextChat*>(client.get());
 
+        proto::TextChatStatus::Status status = proto::TextChatStatus::STATUS_USER_CONNECTED;
+        if (!has_user)
+            status = proto::TextChatStatus::STATUS_USER_DISCONNECTED;
+
         text_chat_client->setHasUser(has_user);
+        text_chat_client->sendStatus(status);
     }
 }
 

diff --git a/source/host/user_session.h b/source/host/user_session.h
@@ -130,7 +130,7 @@ class UserSession
     void sendHostIdRequest(const base::Location& location);
     void addNewClientSession(std::unique_ptr<ClientSession> client_session);
     void setState(const base::Location& location, State state);
-    void setTextChatHasUser(bool has_user);
+    void onTextChatHasUser(const base::Location& location, bool has_user);
     void onTextChatSessionStarted(uint32_t id);
     void onTextChatSessionFinished(uint32_t id);