Repo changes from SWG Source

src/stationapi/Node.hpp

@@ -8,6 +8,7 @@
 #include <memory>
 #include <string>
 #include <vector>
+#include <stdexcept>
 
 template <typename NodeT, typename ClientT>
 class Node : public UdpManagerHandler

src/stationchat/CMakeLists.txt

@@ -34,6 +34,7 @@ add_executable(
   protocol/SetApiVersion.hpp
   protocol/SetAvatarAttributes.hpp
   protocol/UpdatePersistentMessage.hpp
+  protocol/UpdatePersistentMessages.hpp
   ChatAvatar.cpp
   ChatAvatar.hpp
   ChatAvatarService.cpp
@@ -60,7 +61,7 @@ add_executable(
   StationChatApp.cpp
   StationChatApp.hpp
   StationChatConfig.hpp)
-
+  
 # cmake-format: off
 target_link_libraries(stationchat
     stationapi

src/stationchat/ChatAvatarService.cpp

@@ -72,6 +72,7 @@ void ChatAvatarService::LoginAvatar(ChatAvatar* avatar) {
 }
 
 void ChatAvatarService::LogoutAvatar(ChatAvatar* avatar) {
+	if(!avatar->isOnline_) return;
     avatar->isOnline_ = false;
 
     onlineAvatars_.erase(std::remove_if(

src/stationchat/GatewayClient.cpp

@@ -42,6 +42,7 @@
 #include "protocol/SetApiVersion.hpp"
 #include "protocol/SetAvatarAttributes.hpp"
 #include "protocol/UpdatePersistentMessage.hpp"
+#include "protocol/UpdatePersistentMessages.hpp"
 
 #include "easylogging++.h"
 
@@ -138,6 +139,9 @@ void GatewayClient::OnIncoming(std::istringstream& istream) {
     case ChatRequestType::UPDATEPERSISTENTMESSAGE:
         HandleIncomingMessage<UpdatePersistentMessage>(istream);
         break;
+    case ChatRequestType::UPDATEPERSISTENTMESSAGES:
+        HandleIncomingMessage<UpdatePersistentMessages>(istream);
+        break;
     case ChatRequestType::IGNORESTATUS:
         HandleIncomingMessage<IgnoreStatus>(istream);
         break;

src/stationchat/PersistentMessageService.cpp

@@ -208,3 +208,32 @@ void PersistentMessageService::UpdateMessageStatus(
 
     sqlite3_finalize(stmt);
 }
+
+void PersistentMessageService::BulkUpdateMessageStatus(
+    uint32_t avatarId, const std::u16string& category, PersistentState newStatus)
+{
+    sqlite3_stmt* stmt;
+
+    char sql[] = "UPDATE persistent_message SET status = @status WHERE avatar_id = @avatar_id AND "
+             "category = @category";
+
+    auto result = sqlite3_prepare_v2(db_, sql, -1, &stmt, 0);
+    if (result != SQLITE_OK) {
+        throw SQLite3Exception{result, sqlite3_errmsg(db_)};
+    }
+
+    int statusIdx = sqlite3_bind_parameter_index(stmt, "@status");
+    int avatarIdIdx = sqlite3_bind_parameter_index(stmt, "@avatar_id");
+    int categoryIdx = sqlite3_bind_parameter_index(stmt, "@category");
+
+    sqlite3_bind_int(stmt, statusIdx, static_cast<uint32_t>(newStatus));
+    sqlite3_bind_int(stmt, avatarIdIdx, avatarId);
+    std::string cat = FromWideString(category);
+    sqlite3_bind_text(stmt, categoryIdx, cat.c_str(), -1, nullptr);
+
+    result = sqlite3_step(stmt);
+    if (result != SQLITE_DONE) {
+        throw SQLite3Exception{result, sqlite3_errmsg(db_)};
+    }
+    sqlite3_finalize(stmt);
+}

src/stationchat/PersistentMessageService.hpp

@@ -25,6 +25,9 @@ class PersistentMessageService {
     void UpdateMessageStatus(
         uint32_t avatarId, uint32_t messageId, PersistentState status);
 
+    void BulkUpdateMessageStatus(
+        uint32_t avatarId, const std::u16string& category, PersistentState newStatus);
+
 private:
     sqlite3* db_;
 };