Skip to content

Commit

Permalink
Revert "feat: stop handling team member add/removed events (#2273)"
Browse files Browse the repository at this point in the history 
This reverts commit e2f290b
  • Loading branch information
@MohamadJaara
MohamadJaara committed Jan 15, 2024
1 parent fe69d4f commit 610df82
Show file tree
Hide file tree
Showing 11 changed files with 360 additions and 11 deletions.
34 changes: 33 additions & 1 deletion logic/src/commonMain/kotlin/com/wire/kalium/logic/data/event/Event.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@ import com.wire.kalium.logger.obfuscateDomain
import com.wire.kalium.logger.obfuscateId
import com.wire.kalium.logic.data.client.Client
import com.wire.kalium.logic.data.conversation.ClientId
import com.wire.kalium.logic.data.conversation.Conversation.Member
import com.wire.kalium.logic.data.conversation.Conversation.Protocol
import com.wire.kalium.logic.data.conversation.Conversation.Member
import com.wire.kalium.logic.data.conversation.Conversation.ReceiptMode
import com.wire.kalium.logic.data.conversation.Conversation.TypingIndicatorMode
import com.wire.kalium.logic.data.conversation.MutedConversationStatus
Expand Down Expand Up @@ -447,6 +447,38 @@ sealed class Event(open val id: String, open val transient: Boolean, open val li
)
}

data class MemberJoin(
override val id: String,
override val teamId: String,
override val transient: Boolean,
override val live: Boolean,
val memberId: String,
) : Team(id, teamId, transient, live) {
override fun toLogMap(): Map<String, Any?> = mapOf(
typeKey to "Team.MemberJoin",
idKey to id.obfuscateId(),
teamIdKey to teamId.obfuscateId(),
memberIdKey to memberId.obfuscateId(),
)
}

data class MemberLeave(
override val id: String,
override val transient: Boolean,
override val live: Boolean,
override val teamId: String,
val memberId: String,
val timestampIso: String,
) : Team(id, teamId, transient, live) {
override fun toLogMap(): Map<String, Any?> = mapOf(
typeKey to "Team.MemberLeave",
idKey to id.obfuscateId(),
teamIdKey to teamId.obfuscateId(),
timestampIsoKey to timestampIso,
memberIdKey to memberId.obfuscateId(),
)
}

data class MemberUpdate(
override val id: String,
override val teamId: String,
Expand Down
29 changes: 29 additions & 0 deletions logic/src/commonMain/kotlin/com/wire/kalium/logic/data/event/EventMapper.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -96,6 +96,8 @@ class EventMapper(
is EventContentDTO.Conversation.AccessUpdate -> unknown(id, transient, live, eventContentDTO)
is EventContentDTO.Conversation.DeletedConversationDTO -> conversationDeleted(id, eventContentDTO, transient, live)
is EventContentDTO.Conversation.ConversationRenameDTO -> conversationRenamed(id, eventContentDTO, transient, live)
is EventContentDTO.Team.MemberJoin -> teamMemberJoined(id, eventContentDTO, transient, live)
is EventContentDTO.Team.MemberLeave -> teamMemberLeft(id, eventContentDTO, transient, live)
is EventContentDTO.Team.MemberUpdate -> teamMemberUpdate(id, eventContentDTO, transient, live)
is EventContentDTO.Team.Update -> teamUpdate(id, eventContentDTO, transient, live)
is EventContentDTO.User.UpdateDTO -> userUpdate(id, eventContentDTO, transient, live)
Expand Down Expand Up @@ -655,6 +657,33 @@ class EventMapper(
timestampIso = event.time,
)

private fun teamMemberJoined(
id: String,
event: EventContentDTO.Team.MemberJoin,
transient: Boolean,
live: Boolean
) = Event.Team.MemberJoin(
id = id,
teamId = event.teamId,
transient = transient,
live = live,
memberId = event.teamMember.nonQualifiedUserId
)

private fun teamMemberLeft(
id: String,
event: EventContentDTO.Team.MemberLeave,
transient: Boolean,
live: Boolean
) = Event.Team.MemberLeave(
id = id,
teamId = event.teamId,
memberId = event.teamMember.nonQualifiedUserId,
transient = transient,
live = live,
timestampIso = event.time
)

private fun teamMemberUpdate(
id: String,
event: EventContentDTO.Team.MemberUpdate,
Expand Down
36 changes: 36 additions & 0 deletions logic/src/commonMain/kotlin/com/wire/kalium/logic/data/team/TeamRepository.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ import com.wire.kalium.logic.data.id.ConversationId
import com.wire.kalium.logic.data.id.TeamId
import com.wire.kalium.logic.data.service.ServiceMapper
import com.wire.kalium.logic.data.user.UserId
import com.wire.kalium.logic.data.user.UserMapper
import com.wire.kalium.logic.data.user.type.UserEntityTypeMapper
import com.wire.kalium.logic.di.MapperProvider
import com.wire.kalium.logic.functional.Either
Expand All @@ -39,7 +40,10 @@ import com.wire.kalium.logic.wrapApiRequest
import com.wire.kalium.logic.wrapStorageRequest
import com.wire.kalium.network.api.base.authenticated.TeamsApi
import com.wire.kalium.network.api.base.authenticated.notification.EventContentDTO
import com.wire.kalium.network.api.base.authenticated.userDetails.UserDetailsApi
import com.wire.kalium.network.api.base.model.LegalHoldStatusDTO
import com.wire.kalium.network.api.base.model.QualifiedID
import com.wire.kalium.persistence.dao.ConnectionEntity
import com.wire.kalium.persistence.dao.QualifiedIDEntity
import com.wire.kalium.persistence.dao.ServiceDAO
import com.wire.kalium.persistence.dao.TeamDAO
Expand All @@ -55,6 +59,8 @@ interface TeamRepository {
suspend fun getTeam(teamId: TeamId): Flow<Team?>
suspend fun deleteConversation(conversationId: ConversationId, teamId: TeamId): Either<CoreFailure, Unit>
suspend fun updateMemberRole(teamId: String, userId: String, permissionCode: Int?): Either<CoreFailure, Unit>
suspend fun fetchTeamMember(teamId: String, userId: String): Either<CoreFailure, Unit>
suspend fun removeTeamMember(teamId: String, userId: String): Either<CoreFailure, Unit>
suspend fun updateTeam(team: Team): Either<CoreFailure, Unit>
suspend fun syncServices(teamId: TeamId): Either<CoreFailure, Unit>
suspend fun approveLegalHoldRequest(teamId: TeamId, password: String?): Either<CoreFailure, Unit>
Expand All @@ -67,10 +73,12 @@ internal class TeamDataSource(
private val userConfigDAO: UserConfigDAO,
private val teamDAO: TeamDAO,
private val teamsApi: TeamsApi,
private val userDetailsApi: UserDetailsApi,
private val selfUserId: UserId,
private val serviceDAO: ServiceDAO,
private val legalHoldHandler: LegalHoldHandler,
private val legalHoldRequestHandler: LegalHoldRequestHandler,
private val userMapper: UserMapper = MapperProvider.userMapper(),
private val teamMapper: TeamMapper = MapperProvider.teamMapper(),
private val serviceMapper: ServiceMapper = MapperProvider.serviceMapper(),
private val userTypeEntityTypeMapper: UserEntityTypeMapper = MapperProvider.userTypeEntityMapper(),
Expand Down Expand Up @@ -146,6 +154,34 @@ internal class TeamDataSource(
}
}

override suspend fun fetchTeamMember(teamId: String, userId: String): Either<CoreFailure, Unit> {
return wrapApiRequest {
teamsApi.getTeamMember(
teamId = teamId,
userId = userId,
)
}.flatMap { member ->
wrapApiRequest { userDetailsApi.getUserInfo(userId = QualifiedID(userId, selfUserId.domain)) }
.flatMap { userProfileDTO ->
wrapStorageRequest {
val userEntity = userMapper.fromUserProfileDtoToUserEntity(
userProfile = userProfileDTO,
connectionState = ConnectionEntity.State.ACCEPTED,
userTypeEntity = userTypeEntityTypeMapper.teamRoleCodeToUserType(member.permissions?.own)
)
userDAO.upsertUser(userEntity)
userDAO.upsertConnectionStatuses(mapOf(userEntity.id to userEntity.connectionStatus))
}
}
}
}

override suspend fun removeTeamMember(teamId: String, userId: String): Either<CoreFailure, Unit> {
return wrapStorageRequest {
userDAO.markUserAsDeleted(QualifiedIDEntity(userId, selfUserId.domain))
}
}

override suspend fun updateTeam(team: Team): Either<CoreFailure, Unit> {
return wrapStorageRequest {
teamDAO.updateTeam(teamMapper.fromModelToEntity(team))
Expand Down
5 changes: 2 additions & 3 deletions logic/src/commonMain/kotlin/com/wire/kalium/logic/feature/UserSessionScope.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -234,8 +234,6 @@ import com.wire.kalium.logic.feature.legalhold.ObserveLegalHoldChangeNotifiedFor
import com.wire.kalium.logic.feature.legalhold.ObserveLegalHoldChangeNotifiedForSelfUseCaseImpl
import com.wire.kalium.logic.feature.legalhold.ObserveLegalHoldForSelfUserUseCase
import com.wire.kalium.logic.feature.legalhold.ObserveLegalHoldForSelfUserUseCaseImpl
import com.wire.kalium.logic.feature.legalhold.ObserveLegalHoldRequestUseCase
import com.wire.kalium.logic.feature.legalhold.ObserveLegalHoldRequestUseCaseImpl
import com.wire.kalium.logic.feature.legalhold.ObserveLegalHoldStateForUserUseCase
import com.wire.kalium.logic.feature.legalhold.ObserveLegalHoldStateForUserUseCaseImpl
import com.wire.kalium.logic.feature.message.AddSystemMessageToAllConversationsUseCase
Expand Down Expand Up @@ -728,6 +726,7 @@ class UserSessionScope internal constructor(
userStorage.database.userConfigDAO,
userStorage.database.teamDAO,
authenticatedNetworkContainer.teamsApi,
authenticatedNetworkContainer.userDetailsApi,
userId,
userStorage.database.serviceDAO,
legalHoldHandler,
Expand Down Expand Up @@ -1443,7 +1442,7 @@ class UserSessionScope internal constructor(
)

private val teamEventReceiver: TeamEventReceiver
get() = TeamEventReceiverImpl(teamRepository)
get() = TeamEventReceiverImpl(teamRepository, conversationRepository, userRepository, persistMessage, userId)

private val guestRoomConfigHandler
get() = GuestRoomConfigHandler(userConfigRepository, kaliumConfigs)
Expand Down
108 changes: 106 additions & 2 deletions logic/src/commonMain/kotlin/com/wire/kalium/logic/sync/receiver/TeamEventReceiver.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -18,26 +18,39 @@

package com.wire.kalium.logic.sync.receiver

import com.benasher44.uuid.uuid4
import com.wire.kalium.logic.CoreFailure
import com.wire.kalium.logic.data.conversation.ConversationRepository
import com.wire.kalium.logic.data.event.Event
import com.wire.kalium.logic.data.event.EventLoggingStatus
import com.wire.kalium.logic.data.event.logEventProcessing
import com.wire.kalium.logic.data.message.Message
import com.wire.kalium.logic.data.message.MessageContent
import com.wire.kalium.logic.data.message.PersistMessageUseCase
import com.wire.kalium.logic.data.team.Team
import com.wire.kalium.logic.data.team.TeamRepository
import com.wire.kalium.logic.data.user.UserId
import com.wire.kalium.logic.data.user.UserRepository
import com.wire.kalium.logic.functional.Either
import com.wire.kalium.logic.functional.onFailure
import com.wire.kalium.logic.functional.onSuccess
import com.wire.kalium.logic.kaliumLogger
import kotlinx.coroutines.flow.first

internal interface TeamEventReceiver : EventReceiver<Event.Team>

internal class TeamEventReceiverImpl(
private val teamRepository: TeamRepository

private val teamRepository: TeamRepository,
private val conversationRepository: ConversationRepository,
private val userRepository: UserRepository,
private val persistMessage: PersistMessageUseCase,
private val selfUserId: UserId,
) : TeamEventReceiver {

override suspend fun onEvent(event: Event.Team): Either<CoreFailure, Unit> {
when (event) {
is Event.Team.MemberJoin -> handleMemberJoin(event)
is Event.Team.MemberLeave -> handleMemberLeave(event)
is Event.Team.MemberUpdate -> handleMemberUpdate(event)
is Event.Team.Update -> handleUpdate(event)
}
Expand All @@ -48,6 +61,97 @@ internal class TeamEventReceiverImpl(
return Either.Right(Unit)
}

private suspend fun handleMemberJoin(event: Event.Team.MemberJoin) =
teamRepository.fetchTeamMember(
teamId = event.teamId,
userId = event.memberId,
)
.onSuccess {
kaliumLogger
.logEventProcessing(
EventLoggingStatus.SUCCESS,
event
)
}
.onFailure {
kaliumLogger
.logEventProcessing(
EventLoggingStatus.FAILURE,
event,
Pair("errorInfo", "$it")
)
}

@Suppress("LongMethod")
private suspend fun handleMemberLeave(event: Event.Team.MemberLeave) {
val userId = UserId(event.memberId, selfUserId.domain)
teamRepository.removeTeamMember(
teamId = event.teamId,
userId = event.memberId,
)
.onSuccess {
val knownUser = userRepository.getKnownUser(userId).first()
if (knownUser?.name != null) {
conversationRepository.getConversationsByUserId(userId)
.onSuccess {
it.forEach { conversation ->
val message = Message.System(
id = uuid4().toString(), // We generate a random uuid for this new system message
content = MessageContent.TeamMemberRemoved(knownUser.name),
conversationId = conversation.id,
date = event.timestampIso,
senderUserId = userId,
status = Message.Status.Sent,
visibility = Message.Visibility.VISIBLE,
expirationData = null
)
persistMessage(message)
}

conversationRepository.deleteUserFromConversations(userId)
.onSuccess {
kaliumLogger
.logEventProcessing(
EventLoggingStatus.SUCCESS,
event
)
}
.onFailure { deleteFailure ->
kaliumLogger
.logEventProcessing(
EventLoggingStatus.FAILURE,
event,
Pair("errorInfo", "$deleteFailure")
)
}

}.onFailure {
kaliumLogger
.logEventProcessing(
EventLoggingStatus.FAILURE,
event,
Pair("errorInfo", "$it")
)
}
} else {
kaliumLogger
.logEventProcessing(
EventLoggingStatus.SKIPPED,
event,
Pair("info", "User or User name is null")
)
}
}
.onFailure {
kaliumLogger
.logEventProcessing(
EventLoggingStatus.FAILURE,
event,
Pair("errorInfo", "$it")
)
}
}

private suspend fun handleMemberUpdate(event: Event.Team.MemberUpdate) =
teamRepository.updateMemberRole(
teamId = event.teamId,
Expand Down
1 change: 0 additions & 1 deletion logic/src/commonMain/kotlin/com/wire/kalium/logic/sync/receiver/UserEventReceiver.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,6 @@ internal interface UserEventReceiver : EventReceiver<Event.User>
internal class UserEventReceiverImpl internal constructor(
private val clientRepository: ClientRepository,
private val connectionRepository: ConnectionRepository,
private val conversationRepository: ConversationRepository,
private val userRepository: UserRepository,
private val logout: LogoutUseCase,
private val oneOnOneResolver: OneOnOneResolver,
Expand Down
Loading

0 comments on commit 610df82

Please sign in to comment.