UserAccountData: optimize helper and clean code.

matrix-sdk-android/src/main/java/im/vector/matrix/android/internal/session/sync/UserAccountDataSyncHandler.kt

@@ -78,13 +78,7 @@ internal class UserAccountDataSyncHandler @Inject constructor(private val monarc
     // If we get some direct chat invites, we synchronize the user account data including those.
     private fun synchronizeWithServerIfNeeded(realm: Realm, invites: Map<String, InvitedRoomSync>?) {
         if (invites.isNullOrEmpty()) return
-
         val directChats = directChatsHelper.getLocalUserAccount()
-        val directChatInvites = HashMap<String, MutableList<String>>().apply {
-            directChats.forEach { (inviterId, roomIds) ->
-                put(inviterId, ArrayList(roomIds))
-            }
-        }
         var hasUpdate = false
         invites.forEach { (roomId, _) ->
             val myUserStateEvent = RoomMembers(realm, roomId).getStateEvent(credentials.userId)
@@ -92,11 +86,13 @@ internal class UserAccountDataSyncHandler @Inject constructor(private val monarc
             val myUserRoomMember: RoomMember? = myUserStateEvent?.let { it.asDomain().content?.toModel() }
             val isDirect = myUserRoomMember?.isDirect
             if (inviterId != null && inviterId != credentials.userId && isDirect == true) {
-                directChatInvites.getOrPut(inviterId, { arrayListOf() }).apply {
+                directChats
+                        .getOrPut(inviterId, { arrayListOf() })
+                        .apply {
                             if (contains(roomId)) {
                                 Timber.v("Direct chats already include room $roomId with user $inviterId")
                             } else {
-                        this.add(roomId)
+                                add(roomId)
                                 hasUpdate = true
                             }
                         }
@@ -104,7 +100,7 @@ internal class UserAccountDataSyncHandler @Inject constructor(private val monarc
         }
         if (hasUpdate) {
             val updateUserAccountParams = UpdateUserAccountDataTask.DirectChatParams(
-                    directMessages = directChatInvites
+                    directMessages = directChats
             )
             updateUserAccountDataTask.configureWith(updateUserAccountParams).executeBy(taskExecutor)
         }

matrix-sdk-android/src/main/java/im/vector/matrix/android/internal/session/user/accountdata/DirectChatsHelper.kt

@@ -30,7 +30,7 @@ internal class DirectChatsHelper @Inject constructor(@SessionDatabase
     /**
      * @return a map of userId <-> list of roomId
      */
-    fun getLocalUserAccount(filterRoomId: String? = null): Map<String, List<String>> {
+    fun getLocalUserAccount(filterRoomId: String? = null): MutableMap<String, MutableList<String>> {
         return Realm.getInstance(realmConfiguration).use { realm ->
             val currentDirectRooms = RoomSummaryEntity.getDirectRooms(realm)
             val directChatsMap = mutableMapOf<String, MutableList<String>>()