05/19 Enhance codes 5

app/routes/auth.py

@@ -282,6 +282,9 @@ def mfa_enable():
     if not pyotp.TOTP(secret).verify(totp_code, valid_window=1):
         return jsonify({'error': 'Invalid verification code'}), 400
 
+    # Encrypt the secret before replay check so totp_secret_enc/totp_iv are defined.
+    totp_secret_enc, totp_iv = encrypt_totp_secret(secret)
+
     # Prevent replay: reject a code that was already consumed within the valid window.
     # user.id is not yet persisted (MFA not enabled), so use g.current_user_id directly.
     if is_totp_code_used(g.current_user_id, totp_code):

app/routes/webauthn.py

@@ -39,7 +39,6 @@ from webauthn.helpers.structs import (
 from webauthn.helpers.exceptions import (
     InvalidCBORData,
     InvalidRegistrationResponse,
-    InvalidAuthenticationResponse,
 )
 from flask import Blueprint, request, jsonify, g, session, current_app
 
@@ -402,4 +401,4 @@ def delete_credential(cred_id):
         ip_address=client_ip(),
     )
     db.session.commit()
-    return jsonify({'message': 'Passkey removed'}), 200
+    return jsonify({'message': 'Passkey removed'}), 200