TW-1262: notification-for-multiple-accounts

linagora · Jan 4, 2024 · 0ef73e4 · 0ef73e4
1 parent 11f036d
commit 0ef73e4
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/lib/src/database/hive_collections_database.dart b/lib/src/database/hive_collections_database.dart
@@ -90,6 +90,8 @@ class HiveCollectionsDatabase extends DatabaseApi {
 
   late CollectionBox<String> _seenDeviceKeysBox;
 
+  late CollectionBox<String> _roomIdClientBox;
+
   String get _clientBoxName => 'box_client';
 
   String get _accountDataBoxName => 'box_account_data';
@@ -128,6 +130,8 @@ class HiveCollectionsDatabase extends DatabaseApi {
 
   String get _seenDeviceKeysBoxName => 'box_seen_device_keys';
 
+  String get _roomIdClientBoxName => 'box_room_id_client';
+
   HiveCollectionsDatabase(
     this.name,
     this.path,{
@@ -162,6 +166,7 @@ class HiveCollectionsDatabase extends DatabaseApi {
         _eventsBoxName,
         _seenDeviceIdsBoxName,
         _seenDeviceKeysBoxName,
+        _roomIdClientBoxName,
       },
       key: key,
       path: path,
@@ -229,6 +234,10 @@ class HiveCollectionsDatabase extends DatabaseApi {
       _seenDeviceKeysBoxName,
     );
 
+    _roomIdClientBox = await _collection.openBox(
+      _roomIdClientBoxName,
+    );
+
     // Check version and check if we need a migration
     final currentVersion = int.tryParse(await _clientBox.get('version') ?? '');
     if (currentVersion == null) {
@@ -276,6 +285,7 @@ class HiveCollectionsDatabase extends DatabaseApi {
         await _eventsBox.clear();
         await _seenDeviceIdsBox.clear();
         await _seenDeviceKeysBox.clear();
+        await _roomIdClientBox.clear();
         if (supportDeleteCollections) {
           await _collection.deleteFromDisk();
         }
@@ -338,6 +348,9 @@ class HiveCollectionsDatabase extends DatabaseApi {
           if (multiKey.parts.first != roomId) continue;
           await _roomAccountDataBox.delete(key);
         }
+        for (final key in await _roomIdClientBox.getAllKeys()) {
+          await _roomIdClientBox.delete(key);
+        }
         await _roomsBox.delete(roomId);
       });
 
@@ -604,7 +617,13 @@ class HiveCollectionsDatabase extends DatabaseApi {
 
           // Add to the list and continue.
           rooms[room.id] = room;
+
         }
+
+
+        await Future.wait(rooms.values.map((room) {
+          return _roomIdClientBox.put(room.id, client.clientName);
+        }));
 
         for (final room in rooms.values) {
           // Add states to the room
@@ -1532,6 +1551,7 @@ class HiveCollectionsDatabase extends DatabaseApi {
       _eventsBoxName: await _eventsBox.getAllValues(),
       _seenDeviceIdsBoxName: await _seenDeviceIdsBox.getAllValues(),
       _seenDeviceKeysBoxName: await _seenDeviceKeysBox.getAllValues(),
+      _roomIdClientBoxName: await _roomIdClientBox.getAllValues(),
     };
     final json = jsonEncode(dataMap);
     await clear(supportDeleteCollections: supportDeleteCollections);
@@ -1606,6 +1626,9 @@ class HiveCollectionsDatabase extends DatabaseApi {
       for (final key in json[_seenDeviceKeysBoxName]!.keys) {
         await _seenDeviceKeysBox.put(key, json[_seenDeviceKeysBoxName]![key]);
       }
+      for (final key in json[_roomIdClientBoxName]!.keys) {
+        await _roomIdClientBox.put(key, json[_roomIdClientBoxName]![key]);
+      }
       return true;
     } catch (e, s) {
       Logs().e('Database import error: ', e, s);