""" temporary legacy """ from typing import Optional from datetime import datetime from selfprivacy_api.utils import UserDataFiles, WriteUserData, ReadUserData from selfprivacy_api.models.tokens.token import Token from selfprivacy_api.models.tokens.recovery_key import RecoveryKey from selfprivacy_api.models.tokens.new_device_key import NewDeviceKey from selfprivacy_api.repositories.tokens.exceptions import TokenNotFoundError from selfprivacy_api.repositories.tokens.abstract_tokens_repository import ( AbstractTokensRepository, ) from mnemonic import Mnemonic class JsonTokensRepository(AbstractTokensRepository): def get_token_by_token_string(self, token_string: str) -> Optional[Token]: """Get the token by token""" with ReadUserData(UserDataFiles.TOKENS) as tokens_file: for userdata_token in tokens_file["tokens"]: if userdata_token["token"] == token_string: return Token( token=token_string, device_name=userdata_token["name"], created_at=userdata_token["date"], ) raise TokenNotFoundError("Token not found!") def get_token_by_name(self, token_name: str) -> Optional[Token]: """Get the token by name""" with ReadUserData(UserDataFiles.TOKENS) as tokens_file: for userdata_token in tokens_file["tokens"]: if userdata_token["name"] == token_name: return Token( token=userdata_token["token"], device_name=token_name, created_at=userdata_token["date"], ) raise TokenNotFoundError("Token not found!") def get_tokens(self) -> list[Token]: """Get the tokens""" tokens_list = [] with ReadUserData(UserDataFiles.TOKENS) as tokens_file: for userdata_token in tokens_file["tokens"]: tokens_list.append( Token( token=userdata_token.token, device_name=userdata_token.name, created_at=userdata_token.date, ) ) return tokens_list def create_token(self, name: str) -> Token: """Create new token""" new_token = Token.generate(device_name=name) with WriteUserData(UserDataFiles.TOKENS) as tokens_file: tokens_file["tokens"].append( { "token": new_token.token, "name": new_token.device_name, "date": new_token.created_at, } ) return new_token def delete_token(self, input_token: Token) -> None: """Delete the token""" with WriteUserData(UserDataFiles.TOKENS) as tokens_file: for userdata_token in tokens_file["tokens"]: if userdata_token["token"] == input_token: tokens_file["tokens"].remove( userdata_token ) # Allah, i pray it works def refresh_token(self, input_token: Token) -> Token: """Change the token field of the existing token""" new_token = Token.generate(device_name=input_token.device_name) with WriteUserData(UserDataFiles.TOKENS) as tokens_file: for userdata_token in tokens_file["tokens"]: if userdata_token["token"] == input_token.token: userdata_token["token"] = new_token.token userdata_token["data"] = new_token.created_at return new_token raise TokenNotFoundError("Token not found!") def get_recovery_key(self) -> Optional[RecoveryKey]: """Get the recovery key""" with ReadUserData(UserDataFiles.TOKENS) as tokens_file: if tokens_file["recovery_token"] is None: return recovery_key = RecoveryKey( key=tokens_file["recovery_token"].get("token"), created_at=tokens_file["recovery_token"].get("date"), expires_at=tokens_file["recovery_token"].get("expitation"), uses_left=tokens_file["recovery_token"].get("uses_left"), ) return recovery_key def create_recovery_key( self, expiration: Optional[datetime], uses_left: Optional[int], ) -> RecoveryKey: """Create the recovery key""" recovery_key = RecoveryKey.generate(expiration=expiration, uses_left=uses_left) with ReadUserData(UserDataFiles.TOKENS) as tokens_file: tokens_file["recovery_key"] = { "token": recovery_key.key, "date": recovery_key.created_at, "expiration": recovery_key.expires_at, "uses_left": recovery_key.uses_left, } return recovery_key def use_mnemonic_recovery_key(self, mnemonic_phrase: str, name: str) -> Token: """Use the mnemonic recovery key and create a new token with the given name""" recovery_key = self.get_recovery_key() # self ? if recovery_key is None: return None if not recovery_key.is_valid(): return None if recovery_key is None: return None recovery_token = bytes.fromhex(recovery_key.key) if not Mnemonic(language="english").check(mnemonic_phrase): return None phrase_bytes = Mnemonic(language="english").to_entropy(mnemonic_phrase) if phrase_bytes != recovery_token: return None new_recovery_key = RecoveryKey.generate() with WriteUserData(UserDataFiles.TOKENS) as tokens: tokens["tokens"].append( { "token": new_recovery_key.key, # "name": new_recovery_key.name, what???? there is no name "date": str(datetime.now()), } ) if "recovery_token" in tokens: if ( "uses_left" in tokens["recovery_token"] and tokens["recovery_token"]["uses_left"] is not None ): tokens["recovery_token"]["uses_left"] -= 1 return new_recovery_key def get_new_device_key(self) -> NewDeviceKey: """Creates and returns the new device key""" new_device_key = NewDeviceKey.generate() with WriteUserData(UserDataFiles.TOKENS) as tokens_file: tokens_file["new_device"] = { "token": new_device_key.key, "data": new_device_key.created_at, "expiration": new_device_key.expires_at, } return new_device_key def delete_new_device_key(self) -> None: """Delete the new device key""" with WriteUserData(UserDataFiles.TOKENS) as tokens_file: if "new_device" in tokens_file: del tokens_file["new_device"] def use_mnemonic_new_device_key(self, mnemonic_phrase: str, name: str) -> None: """Use the mnemonic new device key""" ...