From 0d6ae19d99600d4e9717985309c617c2ef47b9ba Mon Sep 17 00:00:00 2001
From: Bruno Windels <bruno@windels.cloud>
Date: Mon, 30 Aug 2021 15:05:57 +0200
Subject: [PATCH] use same code to add room to identity in migration as in
 device tracker

---
 src/matrix/e2ee/DeviceTracker.js | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/src/matrix/e2ee/DeviceTracker.js b/src/matrix/e2ee/DeviceTracker.js
index 32e2d00c..e230ea7d 100644
--- a/src/matrix/e2ee/DeviceTracker.js
+++ b/src/matrix/e2ee/DeviceTracker.js
@@ -123,17 +123,9 @@ export class DeviceTracker {
     async _writeMember(member, txn) {
         const {userIdentities} = txn;
         const identity = await userIdentities.get(member.userId);
-        if (!identity) {
-            userIdentities.set({
-                userId: member.userId,
-                roomIds: [member.roomId],
-                deviceTrackingStatus: TRACKING_STATUS_OUTDATED,
-            });
-        } else {
-            if (!identity.roomIds.includes(member.roomId)) {
-                identity.roomIds.push(member.roomId);
-                userIdentities.set(identity);
-            }
+        const updatedIdentity = addRoomToIdentity(identity, member.userId, member.roomId);
+        if (updatedIdentity) {
+            userIdentities.set(updatedIdentity);
         }
     }