refactor enroll method

VAuthenticator · Jul 21, 2024 · 578c19a · 578c19a
1 parent 528b64c
commit 578c19a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/main/kotlin/com/vauthenticator/server/mfa/domain/MfaMethodsEnrollment.kt b/src/main/kotlin/com/vauthenticator/server/mfa/domain/MfaMethodsEnrollment.kt
@@ -33,15 +33,18 @@ class MfaMethodsEnrollmentAssociation(
 
 class MfaMethodsEnrollment(
     private val verificationTicketFactory: VerificationTicketFactory,
-    mfaSender: OtpMfaSender,
+    private val mfaSender: OtpMfaSender,
 ) {
 
     fun enroll(
         account: Account,
         emailMfaMethod: MfaMethod,
         clientAppId: ClientAppId,
-        sendChallengeCode : Boolean = true
+        sendChallengeCode: Boolean = true
     ): VerificationTicket {
+        if (sendChallengeCode) {
+            mfaSender.sendMfaChallenge(account.email)
+        }
         return verificationTicketFactory.createTicketFor(account, clientAppId)
     }
 }