Skip to content

Commit 711153c

Browse files
Island: Merge AuthenticationFacade._generate_otp into the public method
1 parent c731db3 commit 711153c

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

monkey/monkey_island/cc/services/authentication_service/authentication_facade.py

+5-9
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from . import AccountRole
1414
from .i_otp_repository import IOTPRepository
1515
from .token_parser import ParsedToken, TokenParser
16-
from .types import Token
16+
from .types import OTP, Token
1717
from .user import User
1818

1919
OTP_EXPIRATION_TIME = 2 * 60 # 2 minutes
@@ -79,21 +79,17 @@ def _get_refresh_token_owner(self, refresh_token: ParsedToken) -> User:
7979
raise Exception("Invalid refresh token")
8080
return user
8181

82-
def generate_otp(self) -> str:
82+
def generate_otp(self) -> OTP:
8383
"""
8484
Generates a new OTP
8585
8686
The generated OTP is saved to the `IOTPRepository`
87-
:return: The generated OTP
8887
"""
89-
otp_value = self._generate_otp()
88+
otp = secure_generate_random_string(32, string.ascii_letters + string.digits + "._-")
9089
expiration_time = time.monotonic() + OTP_EXPIRATION_TIME
91-
self._otp_repository.insert_otp(otp_value, expiration_time)
90+
self._otp_repository.insert_otp(otp, expiration_time)
9291

93-
return otp_value
94-
95-
def _generate_otp(self) -> str:
96-
return secure_generate_random_string(32, string.ascii_letters + string.digits + "._-")
92+
return otp
9793

9894
def generate_refresh_token(self, user: User) -> Token:
9995
"""

0 commit comments

Comments
 (0)