



import os
import random
import sys

from . import cryptomath_module as cryptoMath  # noqa: N812
from . import rabin_miller as rabinMiller  # noqa: N812

def  lowerCamelCase_()			->							Any:
						print("Making key files..."     )
						make_key_files("rsa"     , 1_024     )
						print("Key files generation successful."     )

def  lowerCamelCase_(lowerCamelCase_     )			->							Tuple:
						print("Generating prime p..."     )
						UpperCAmelCase								=  rabinMiller.generate_large_prime(a_     )
						print("Generating prime q..."     )
						UpperCAmelCase								=  rabinMiller.generate_large_prime(a_     )
						UpperCAmelCase								=  p * q

						print("Generating e that is relatively prime to (p - 1) * (q - 1)..."     )
						while True:
												UpperCAmelCase								=  random.randrange(2 ** (key_size - 1)     , 2 ** (key_size)     )
												if cryptoMath.gcd(a_     , (p - 1) * (q - 1)     ) == 1:
																		break

						print("Calculating d that is mod inverse of e..."     )
						UpperCAmelCase								=  cryptoMath.find_mod_inverse(a_     , (p - 1) * (q - 1)     )

						UpperCAmelCase								=  (n, e)
						UpperCAmelCase								=  (n, d)
						return (public_key, private_key)

def  lowerCamelCase_(lowerCamelCase_     , lowerCamelCase_     )			->							List[str]:
						if os.path.exists(F'{name}_pubkey.txt'     ) or os.path.exists(F'{name}_privkey.txt'     ):
												print("\nWARNING:"     )
												print(
												    F'"{name}_pubkey.txt" or "{name}_privkey.txt" already exists. \n'
												    "Use a different name or delete these files and re-run this program."     )
												sys.exit()

						UpperCAmelCase							,		UpperCAmelCase								=  generate_key(a_     )
						print(F'\nWriting public key to file {name}_pubkey.txt...'     )
						with open(F'{name}_pubkey.txt'     , "w"     ) as out_file:
												out_file.write(F'{key_size},{public_key[0]},{public_key[1]}'     )

						print(F'Writing private key to file {name}_privkey.txt...'     )
						with open(F'{name}_privkey.txt'     , "w"     ) as out_file:
												out_file.write(F'{key_size},{private_key[0]},{private_key[1]}'     )


if __name__ == "__main__":
		main()











class 					UpperCamelCase_ :



	'''simple docstring'''




	def __init__(			self			,     a   )			->       Tuple:
								snake_case_           =				n
								snake_case_           =				[None] * self.n
								snake_case_           =				0  # index of the first element
								snake_case_           =				0
								snake_case_           =				0




	def __len__(			self   )			->       int:
								return self.size




	def 				_UpperCamelCase     (			self   )			->       bool:
								return self.size == 0




	def 				_UpperCamelCase     (			self   )			->       List[Any]:
								return False if self.is_empty() else self.array[self.front]




	def 				_UpperCamelCase     (			self			,     a   )			->       Dict:
								if self.size >= self.n:
															raise Exception('QUEUE IS FULL'   )

								snake_case_           =				data
								snake_case_           =				(self.rear + 1) % self.n
								self.size += 1
								return self






	def 				_UpperCamelCase     (			self   )			->       List[Any]:
								if self.size == 0:
															raise Exception('UNDERFLOW'   )

								snake_case_           =				self.array[self.front]
								snake_case_           =				None
								snake_case_           =				(self.front + 1) % self.n
								self.size -= 1
								return temp





