Remove user from all user groups

Author: pcapriotti

libs/wire-subsystems/src/Wire/UserGroupStore.hs

@@ -23,6 +23,9 @@ data UserGroupPageRequest = UserGroupPageRequest
 
 data PaginationState = PaginationSortByName (Maybe (UserGroupName, UserGroupId)) | PaginationSortByCreatedAt (Maybe (UTCTimeMillis, UserGroupId))
 
+userGroupCreatedAtPaginationState :: UserGroup_ f -> (UTCTimeMillis, UserGroupId)
+userGroupCreatedAtPaginationState ug = (ug.createdAt, ug.id_)
+
 toSortBy :: PaginationState -> SortBy
 toSortBy = \case
   PaginationSortByName _ -> SortByName

libs/wire-subsystems/src/Wire/UserGroupSubsystem.hs

@@ -32,5 +32,6 @@ data UserGroupSubsystem m a where
   AddUsers :: UserId -> UserGroupId -> Vector UserId -> UserGroupSubsystem m ()
   UpdateUsers :: UserId -> UserGroupId -> Vector UserId -> UserGroupSubsystem m ()
   RemoveUser :: UserId -> UserGroupId -> UserId -> UserGroupSubsystem m ()
+  RemoveUserFromAllGroups :: UserId -> TeamId -> UserGroupSubsystem m ()
 
 makeSem ''UserGroupSubsystem

libs/wire-subsystems/src/Wire/UserGroupSubsystem/Interpreter.hs

@@ -50,6 +50,7 @@ interpretUserGroupSubsystem = interpret $ \case
   AddUsers adder groupId addeeIds -> addUsers adder groupId addeeIds
   UpdateUsers updater groupId uids -> updateUsers updater groupId uids
   RemoveUser remover groupId removeeId -> removeUser remover groupId removeeId
+  RemoveUserFromAllGroups uid tid -> removeUserFromAllGroups uid tid
 
 data UserGroupSubsystemError
   = UserGroupNotATeamAdmin
@@ -328,3 +329,38 @@ removeUser remover groupId removeeId = do
     pushNotifications
       [ mkEvent remover (UserGroupUpdated groupId) admins
       ]
+
+removeUserFromAllGroups ::
+  ( Member Store.UserGroupStore r,
+    Member TeamSubsystem r,
+    Member (Error UserGroupSubsystemError) r
+  ) =>
+  UserId ->
+  TeamId ->
+  Sem r ()
+removeUserFromAllGroups uid tid = do
+  void $ internalGetTeamMember uid tid >>= note UserGroupMemberIsNotInTheSameTeam
+  nextPage Nothing >>= go
+  where
+    go (ug : ugs) = do
+      Store.removeUser ug.id_ uid
+      -- when we get to the last item, get a new page
+      ugs' <- case ugs of
+        [] -> nextPage (Just ug)
+        _ -> pure ugs
+      go ugs'
+    -- no more items, terminate
+    go [] = pure ()
+
+    nextPage mug =
+      Store.getUserGroups $
+        UserGroupPageRequest
+          { pageSize = def,
+            sortOrder = Desc,
+            paginationState =
+              PaginationSortByCreatedAt $
+                fmap Store.userGroupCreatedAtPaginationState mug,
+            team = tid,
+            searchString = Nothing,
+            includeMemberCount = False
+          }