From b3221b09d6e12cda47a4ebc8443cac43bb71d29c Mon Sep 17 00:00:00 2001
From: Nicolas Werner <nicolas.werner@hotmail.de>
Date: Sat, 5 Feb 2022 08:40:56 +0100
Subject: Add /reset-state command

---
 src/Cache.cpp | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

(limited to 'src/Cache.cpp')

diff --git a/src/Cache.cpp b/src/Cache.cpp
index b55d53a6..9f3ada58 100644
--- a/src/Cache.cpp
+++ b/src/Cache.cpp
@@ -1460,6 +1460,44 @@ Cache::calculateRoomReadStatus(const std::string &room_id)
     return getEventIndex(room_id, last_event_id_) > getEventIndex(room_id, fullyReadEventId_);
 }
 
+void
+Cache::updateState(const std::string &room, const mtx::responses::StateEvents &state)
+{
+    auto txn         = lmdb::txn::begin(env_);
+    auto statesdb    = getStatesDb(txn, room);
+    auto stateskeydb = getStatesKeyDb(txn, room);
+    auto membersdb   = getMembersDb(txn, room);
+    auto eventsDb    = getEventsDb(txn, room);
+
+    saveStateEvents(txn, statesdb, stateskeydb, membersdb, eventsDb, room, state.events);
+
+    RoomInfo updatedInfo;
+    updatedInfo.name       = getRoomName(txn, statesdb, membersdb).toStdString();
+    updatedInfo.topic      = getRoomTopic(txn, statesdb).toStdString();
+    updatedInfo.avatar_url = getRoomAvatarUrl(txn, statesdb, membersdb).toStdString();
+    updatedInfo.version    = getRoomVersion(txn, statesdb).toStdString();
+    updatedInfo.is_space   = getRoomIsSpace(txn, statesdb);
+
+    {
+        std::string_view data;
+        if (roomsDb_.get(txn, room, data)) {
+            try {
+                RoomInfo tmp     = json::parse(std::string_view(data.data(), data.size()));
+                updatedInfo.tags = tmp.tags;
+            } catch (const json::exception &e) {
+                nhlog::db()->warn("failed to parse room info: room_id ({}), {}: {}",
+                                  room,
+                                  std::string(data.data(), data.size()),
+                                  e.what());
+            }
+        }
+    }
+
+    roomsDb_.put(txn, room, json(updatedInfo).dump());
+    updateSpaces(txn, {room}, {room});
+    txn.commit();
+}
+
 void
 Cache::saveState(const mtx::responses::Sync &res)
 {
-- 
cgit 1.5.1