From 63998a217acd4cdec6058b00f4e63d53df2b945c Mon Sep 17 00:00:00 2001
From: Nicolas Werner <nicolas.werner@hotmail.de>
Date: Sun, 22 Aug 2021 14:45:57 +0200
Subject: [PATCH] Add db migration that clears the cache

This fixes spaces or stickers not showing up for old databases as well
as the wrong format of the state_keys db.
---
 src/Cache.cpp | 56 ++++++++++++++++++++++++++++++++++++++++++---------
 src/Cache_p.h |  1 -
 2 files changed, 47 insertions(+), 10 deletions(-)

diff --git a/src/Cache.cpp b/src/Cache.cpp
index 6b28ca91..1adf8eee 100644
--- a/src/Cache.cpp
+++ b/src/Cache.cpp
@@ -35,7 +35,7 @@
 
 //! Should be changed when a breaking change occurs in the cache format.
 //! This will reset client's data.
-static const std::string CURRENT_CACHE_FORMAT_VERSION("2020.10.20");
+static const std::string CURRENT_CACHE_FORMAT_VERSION("2021.08.22");
 static const std::string SECRET("secret");
 
 //! Keys used for the DB
@@ -891,12 +891,6 @@ Cache::setNextBatchToken(lmdb::txn &txn, const std::string &token)
         syncStateDb_.put(txn, NEXT_BATCH_KEY, token);
 }
 
-void
-Cache::setNextBatchToken(lmdb::txn &txn, const QString &token)
-{
-        setNextBatchToken(txn, token.toStdString());
-}
-
 bool
 Cache::isInitialized()
 {
@@ -1137,6 +1131,51 @@ Cache::runMigrations()
                    nhlog::db()->info("Successfully migrated olm sessions.");
                    return true;
            }},
+          {"2021.08.22",
+           [this]() {
+                   try {
+                           auto txn      = lmdb::txn::begin(env_, nullptr);
+                           auto try_drop = [this, &txn](const std::string &dbName) {
+                                   try {
+                                           auto db = lmdb::dbi::open(txn, dbName.c_str());
+                                           db.drop(txn, true);
+                                   } catch (std::exception &e) {
+                                           nhlog::db()->warning(
+                                             "Failed to drop '{}': {}", dbName, e.what());
+                                   }
+                           };
+
+                           auto room_ids = getRoomIds(txn);
+
+                           for (const auto &room : room_ids) {
+                                   try_drop(room + "/state");
+                                   try_drop(room + "/state_by_key");
+                                   try_drop(room + "/account_data");
+                                   try_drop(room + "/members");
+                                   try_drop(room + "/mentions");
+                                   try_drop(room + "/events");
+                                   try_drop(room + "/event_order");
+                                   try_drop(room + "/event2order");
+                                   try_drop(room + "/msg2order");
+                                   try_drop(room + "/order2msg");
+                                   try_drop(room + "/pending");
+                                   try_drop(room + "/related");
+                           }
+
+                           // clear db, don't delete
+                           roomsDb_.drop(txn, false);
+                           setNextBatchToken(txn, "");
+
+                           txn.commit();
+                   } catch (const lmdb::error &) {
+                           nhlog::db()->critical("Failed to clear cache!");
+                           return false;
+                   }
+
+                   nhlog::db()->info(
+                     "Successfully cleared the cache. Will do a clean sync after startup.");
+                   return true;
+           }},
         };
 
         nhlog::db()->info("Running migrations, this may take a while!");
@@ -3230,8 +3269,7 @@ Cache::isNotificationSent(const std::string &event_id)
 std::vector<std::string>
 Cache::getRoomIds(lmdb::txn &txn)
 {
-        auto db     = lmdb::dbi::open(txn, ROOMS_DB, MDB_CREATE);
-        auto cursor = lmdb::cursor::open(txn, db);
+        auto cursor = lmdb::cursor::open(txn, roomsDb_);
 
         std::vector<std::string> rooms;
 
diff --git a/src/Cache_p.h b/src/Cache_p.h
index 8322a6af..d97eb531 100644
--- a/src/Cache_p.h
+++ b/src/Cache_p.h
@@ -692,7 +692,6 @@ private:
         std::optional<UserKeyCache> userKeys_(const std::string &user_id, lmdb::txn &txn);
 
         void setNextBatchToken(lmdb::txn &txn, const std::string &token);
-        void setNextBatchToken(lmdb::txn &txn, const QString &token);
 
         lmdb::env env_;
         lmdb::dbi syncStateDb_;