mv getInvitationCode from brig to subsystems [WIP].

wireapp · Sep 26, 2024 · cd5dfc6 · cd5dfc6
1 parent 84404a4
commit cd5dfc6
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 12 deletions.
diff --git a/libs/wire-subsystems/src/Wire/TeamInvitationSubsystem.hs b/libs/wire-subsystems/src/Wire/TeamInvitationSubsystem.hs
@@ -14,6 +14,7 @@ import Wire.API.User.EmailAddress
 data TeamInvitationSubsystem m a where
   InviteUser :: Local UserId -> TeamId -> InvitationRequest -> TeamInvitationSubsystem m (Invitation, InvitationLocation)
   AcceptInvitation :: UserId -> InvitationId -> InvitationCode -> TeamInvitationSubsystem m ()
+  GetInvitationCode :: TeamId -> InvitationId -> TeamInvitationSubsystem m InvitationCode
   RevokeInvitation :: TeamId -> InvitationId -> TeamInvitationSubsystem m ()
   GetInvitationByCode :: InvitationCode -> TeamInvitationSubsystem m Invitation
   GetInvitationByEmail :: EmailAddress -> TeamInvitationSubsystem m Invitation

diff --git a/libs/wire-subsystems/src/Wire/TeamInvitationSubsystem/Interpreter.hs b/libs/wire-subsystems/src/Wire/TeamInvitationSubsystem/Interpreter.hs
@@ -61,6 +61,7 @@ runTeamInvitationSubsystem ::
 runTeamInvitationSubsystem cfg = interpret $ \case
   InviteUser luid tid request -> runInputConst cfg $ inviteUserImpl luid tid request
   AcceptInvitation uid invitationId invitationCode -> acceptInvitationImpl uid invitationId invitationCode
+  GetInvitationCode tid iid -> getInvitationCodeImpl tid iid
   RevokeInvitation tid invitationId -> revokeInvitationImpl tid invitationId
   GetInvitationByCode invitationCode -> getInvitationByCodeImpl invitationCode
   GetInvitationByEmail email -> getInvitationByEmailImpl email
@@ -250,6 +251,15 @@ logInvitationRequest context action =
 acceptInvitationImpl :: UserId -> InvitationId -> InvitationCode -> Sem r ()
 acceptInvitationImpl = undefined
 
+getInvitationCodeImpl ::
+  (Member Store.InvitationCodeStore r, Member (Error UserSubsystemError) r) =>
+  TeamId ->
+  InvitationId ->
+  Sem r FoundInvitationCode
+getInvitationCodeImpl t r = do
+  inv <- Store.lookupInvitation t r
+  maybe (throw UserSubsystemInvalidInvitationCode) (pure . FoundInvitationCode . (.code)) inv
+
 revokeInvitationImpl :: TeamId -> InvitationId -> Sem r ()
 revokeInvitationImpl = undefined
 

diff --git a/services/brig/src/Brig/Team/API.hs b/services/brig/src/Brig/Team/API.hs
@@ -18,7 +18,6 @@
 module Brig.Team.API
   ( servantAPI,
     getInvitationByEmail,
-    getInvitationCode,
     suspendTeam,
     unsuspendTeam,
     createInvitationViaScim,
@@ -117,17 +116,6 @@ teamSizePublic uid tid = do
   ensurePermissions uid tid [AddTeamMember]
   getTeamSize tid
 
-getInvitationCode ::
-  ( Member Store.InvitationCodeStore r,
-    Member (Error UserSubsystemError) r
-  ) =>
-  TeamId ->
-  InvitationId ->
-  Sem r FoundInvitationCode
-getInvitationCode t r = do
-  inv <- Store.lookupInvitation t r
-  maybe (throw UserSubsystemInvalidInvitationCode) (pure . FoundInvitationCode . (.code)) inv
-
 data CreateInvitationInviter = CreateInvitationInviter
   { inviterUid :: UserId,
     inviterEmail :: EmailAddress