feat: cache contacts size (WPB-14523)

logic/src/commonMain/kotlin/com/wire/kalium/logic/data/id/IdProviders.kt

@@ -25,6 +25,10 @@ fun interface CurrentClientIdProvider {
     suspend operator fun invoke(): Either<CoreFailure, ClientId>
 }
 
+fun interface ContactsSizeProvider {
+    suspend operator fun invoke(): Either<CoreFailure, Int>
+}
+
 internal fun interface SelfTeamIdProvider {
     suspend operator fun invoke(): Either<CoreFailure, TeamId?>
 }

logic/src/commonMain/kotlin/com/wire/kalium/logic/data/user/UserRepository.kt

@@ -156,6 +156,7 @@ interface UserRepository {
     suspend fun migrateUserToTeam(teamName: String): Either<CoreFailure, CreateUserTeam>
     suspend fun updateTeamId(userId: UserId, teamId: TeamId): Either<StorageFailure, Unit>
     suspend fun isClientMlsCapable(userId: UserId, clientId: ClientId): Either<StorageFailure, Boolean>
+    suspend fun getContactsCount(): Either<StorageFailure, Int>
 }
 
 @Suppress("LongParameterList", "TooManyFunctions")
@@ -319,8 +320,8 @@ internal class UserDataSource internal constructor(
                     userProfile = userProfileDTO,
                     connectionState = ConnectionEntity.State.ACCEPTED,
                     userTypeEntity =
-                    if (userProfileDTO.service != null) UserTypeEntity.SERVICE
-                    else userTypeEntityMapper.teamRoleCodeToUserType(mapTeamMemberDTO[userProfileDTO.id.value]?.permissions?.own)
+                        if (userProfileDTO.service != null) UserTypeEntity.SERVICE
+                        else userTypeEntityMapper.teamRoleCodeToUserType(mapTeamMemberDTO[userProfileDTO.id.value]?.permissions?.own)
                 )
             }
         val otherUsers = listUserProfileDTO
@@ -602,6 +603,10 @@ internal class UserDataSource internal constructor(
         clientDAO.isMLSCapable(userId.toDao(), clientId.value)
     }
 
+    override suspend fun getContactsCount(): Either<StorageFailure, Int> = wrapStorageRequest {
+        userDAO.getContactsCount()
+    }
+
     companion object {
 
         internal const val BATCH_SIZE = 500

logic/src/commonMain/kotlin/com/wire/kalium/logic/feature/UserSessionScope.kt

@@ -93,6 +93,7 @@ import com.wire.kalium.logic.data.event.EventDataSource
 import com.wire.kalium.logic.data.event.EventRepository
 import com.wire.kalium.logic.data.featureConfig.FeatureConfigDataSource
 import com.wire.kalium.logic.data.featureConfig.FeatureConfigRepository
+import com.wire.kalium.logic.data.id.ContactsSizeProvider
 import com.wire.kalium.logic.data.id.CurrentClientIdProvider
 import com.wire.kalium.logic.data.id.FederatedIdMapper
 import com.wire.kalium.logic.data.id.GroupID
@@ -498,6 +499,7 @@ class UserSessionScope internal constructor(
     )
 
     private var _clientId: ClientId? = null
+    private var _contactsSize: Int? = null
 
     @OptIn(DelicateKaliumApi::class) // Use the uncached client ID in order to create the cache itself.
     private suspend fun clientId(): Either<CoreFailure, ClientId> =
@@ -507,6 +509,13 @@ class UserSessionScope internal constructor(
             }
         }
 
+    private suspend fun contactsSize(): Either<CoreFailure, Int> =
+        if (_contactsSize != null) Either.Right(_contactsSize!!) else {
+            userRepository.getContactsCount().onSuccess {
+                _contactsSize = it
+            }
+        }
+
     private val userScopedLogger: KaliumLogger = kaliumLogger.withUserDeviceData {
         KaliumLogger.UserClientData(userId.toLogString(), _clientId?.value?.obfuscateId() ?: "")
     }
@@ -527,6 +536,8 @@ class UserSessionScope internal constructor(
         )
 
     val clientIdProvider = CurrentClientIdProvider { clientId() }
+    val contactsSizeProvider = ContactsSizeProvider { contactsSize() }
+
     private val mlsSelfConversationIdProvider: MLSSelfConversationIdProvider by lazy {
         MLSSelfConversationIdProviderImpl(
             conversationRepository
@@ -1204,19 +1215,25 @@ class UserSessionScope internal constructor(
             callRepository
         )
 
-    internal val keyPackageManager: KeyPackageManager = KeyPackageManagerImpl(featureSupport,
+    internal val keyPackageManager: KeyPackageManager = KeyPackageManagerImpl(
+        featureSupport,
         incrementalSyncRepository,
         lazy { clientRepository },
         lazy { client.refillKeyPackages },
         lazy { client.mlsKeyPackageCountUseCase },
-        lazy { users.timestampKeyRepository })
-    internal val keyingMaterialsManager: KeyingMaterialsManager = KeyingMaterialsManagerImpl(featureSupport,
+        lazy { users.timestampKeyRepository }
+    )
+
+    internal val keyingMaterialsManager: KeyingMaterialsManager = KeyingMaterialsManagerImpl(
+        featureSupport,
         incrementalSyncRepository,
         lazy { clientRepository },
         lazy { conversations.updateMLSGroupsKeyingMaterials },
-        lazy { users.timestampKeyRepository })
+        lazy { users.timestampKeyRepository }
+    )
 
-    val mlsClientManager: MLSClientManager = MLSClientManagerImpl(clientIdProvider,
+    val mlsClientManager: MLSClientManager = MLSClientManagerImpl(
+        clientIdProvider,
         isAllowedToRegisterMLSClient,
         incrementalSyncRepository,
         lazy { slowSyncRepository },
@@ -1225,7 +1242,8 @@ class UserSessionScope internal constructor(
             RegisterMLSClientUseCaseImpl(
                 mlsClientProvider, clientRepository, keyPackageRepository, keyPackageLimitsProvider, userConfigRepository
             )
-        })
+        }
+    )
 
     internal val mlsMigrationWorker
         get() =

persistence/src/commonMain/db_user/com/wire/kalium/persistence/Users.sq

@@ -32,6 +32,7 @@ CREATE TABLE User (
 );
 CREATE INDEX user_team_index ON User(team);
 CREATE INDEX user_service_id ON User(bot_service);
+CREATE INDEX idx_user_connection_status_accepted ON User (connection_status) WHERE connection_status = 'ACCEPTED';
 
 deleteUser:
 DELETE FROM User WHERE qualified_id = ?;
@@ -280,3 +281,7 @@ UPDATE User SET team = ? WHERE qualified_id = ?;
 selectNameByMessageId:
 SELECT name FROM User
 WHERE qualified_id = (SELECT Message.sender_user_id FROM Message WHERE Message.id = :messageId AND Message.conversation_id = :conversationId);
+
+getContactsCount:
+SELECT COUNT(*)
+FROM User WHERE connection_status == 'ACCEPTED';

persistence/src/commonMain/db_user/migrations/99.sqm

@@ -0,0 +1 @@
+CREATE INDEX idx_user_connection_status_accepted ON User (connection_status) WHERE connection_status = 'ACCEPTED';

persistence/src/commonMain/kotlin/com/wire/kalium/persistence/dao/UserDAO.kt

@@ -317,4 +317,5 @@ interface UserDAO {
     suspend fun getUsersMinimizedByQualifiedIDs(qualifiedIDs: List<QualifiedIDEntity>): List<UserEntityMinimized>
     suspend fun getNameAndHandle(userId: UserIDEntity): NameAndHandleEntity?
     suspend fun updateTeamId(userId: UserIDEntity, teamId: String)
+    suspend fun getContactsCount(): Int?
 }

persistence/src/commonMain/kotlin/com/wire/kalium/persistence/dao/UserDAOImpl.kt

@@ -482,6 +482,12 @@ class UserDAOImpl internal constructor(
     }
 
     override suspend fun updateTeamId(userId: UserIDEntity, teamId: String) {
-        userQueries.updateTeamId(teamId, userId)
+        withContext(queriesContext) {
+            userQueries.updateTeamId(teamId, userId)
+        }
+    }
+
+    override suspend fun getContactsCount(): Int? = withContext(queriesContext) {
+        userQueries.getContactsCount().executeAsOneOrNull()?.toInt()
     }
 }