Add files via upload

Author: carlospuenteg

classes/Options.py

@@ -1,8 +1,9 @@
 from colorama import Fore, init; init()
-from .Text import Text
+
+from utils.colortext import colortext
 
 class Options:
-    def __init__(self, options:list, first_idx=0):
+    def __init__(self, options:list, first_idx:int=0):
         self.options = options
         self.first_idx = first_idx
 
@@ -13,7 +14,7 @@ def get_choice(self) -> int:
 
         if self.check_input(choice): return int(choice)
 
-        print(Text("\nInvalid choice.\n", Fore.RED))
+        print(colortext("\nInvalid choice.\n", Fore.RED))
         return self.get_choice()
 
 

main.py

@@ -1,3 +1,3 @@
-from classes.Menu import Menu
+from utils.menu import menu
 
-Menu()
+menu()

utils/colortext.py

@@ -0,0 +1,4 @@
+from colorama import Fore, init; init()
+
+def colortext(text:str, color:str=Fore.WHITE) -> str:
+    return f"{color}{text}{Fore.RESET}"

utils/cryptography.py

@@ -1,8 +1,8 @@
 import hashlib
-import sys; sys.path.append("..")
-from constants.constants import *
 import random
 
+from constants.constants import *
+
 def enc(txt, encryption_code:str) -> str:
     encTxt = ""
     for x in txt:
@@ -11,7 +11,7 @@ def enc(txt, encryption_code:str) -> str:
 
 
 # Decrypt an encrypted text with an encryption code
-def dec(txt, encryption:str) -> str:
+def dec(txt:str, encryption:str) -> str:
     decTxt = ""
     for x in txt:
         decTxt += HEX_SYMB[encryption.index(x)]  # The symbol in the equivalent position

utils/encryption.py

@@ -0,0 +1,88 @@
+import random
+from colorama import Fore, init; init()
+
+from utils.colortext import colortext
+from utils.cryptography import *
+from constants.constants import *
+
+def encrypt_txt(text:str, pwd:str) -> str:
+    hash = get_pwd_hash(pwd)
+    print(colortext("Preparing...", Fore.MAGENTA))
+    if len(text)>TXT_MAX_LEN or len(text)==0: 
+        raise Exception(f"text length ({len(text)}) is invalid. It must be between 1 and {TXT_MAX_LEN} characters.")
+
+    # Seed the random generator with the hash
+    seed = int(hash,16)
+    random.seed(seed) 
+
+    # Get the encryption code
+    encryption_code = get_encryption_code(hash) 
+
+    print(colortext("Creating the random base...", Fore.MAGENTA))
+    # Fill the ciphertext with random characters
+    ciphertext = [random.choice(HEX_SYMB) for _ in range(CIPHER_LEN)]  
+
+    print(colortext("Encrypting the input and its length...", Fore.MAGENTA))
+    # INFO: Encrypt the text and the text length
+    textE = enc(t2h(text), encryption_code)
+    textE_len = len(textE)
+    textE_len_max_digits = len(format(CIPHER_LEN,"x")) # If the position length is 5, the length of the encrypted text will fit in 5 digits
+    textE_len_fixed_len = format(textE_len,"x").rjust(textE_len_max_digits,"0")
+    textE_lenE = enc(textE_len_fixed_len, encryption_code) # Encrypted text length (fixed length)
+
+    print(colortext("Generating the length indexes...", Fore.MAGENTA))
+    # Get the indexes where the INFO will be stored
+    textE_len_idxs = get_textE_len_idxs(seed, CIPHER_LEN, textE_len_max_digits)
+
+    print(colortext("Generating the encrypted text indexes...", Fore.MAGENTA))
+    textE_idxs = get_textE_idxs(seed, CIPHER_LEN, len(textE), textE_len_idxs)
+
+    print(colortext("Creating the ciphertext...", Fore.MAGENTA))
+    # Save the text and text length in toret
+    for i,idx in enumerate(textE_len_idxs):
+        ciphertext[idx] = textE_lenE[i]
+    for i,idx in enumerate(textE_idxs):
+        ciphertext[idx] = textE[i]
+
+    return "".join(ciphertext)
+
+
+def decrypt_cipher(ciphertext:str, pwd:str) -> str:
+    hash = get_pwd_hash(pwd)
+    print(colortext("Preparing...", Fore.MAGENTA))
+    cipher_len = len(ciphertext) # Same as CIPHER_LEN
+
+    if not all(c in HEX_SYMB for c in ciphertext) or len(ciphertext) != cipher_len: 
+        raise Exception("Encrypted text is invalid")
+
+    # Seed the random generator with the hash
+    seed = int(hash,16)
+    random.seed(seed) 
+
+    # Get the encryption code
+    encryption_code = get_encryption_code(hash)  
+
+    print(colortext("Decrypting the text length...", Fore.MAGENTA))
+    # Decrypt the text length
+    textE_len_digits = len(format(cipher_len,"x")) # If the position length is 5, the length of the encrypted text will fit in 5 digits
+    textE_len_idxs = get_textE_len_idxs(seed, cipher_len, textE_len_digits)
+    textE_lenE = "".join([ciphertext[idx] for idx in textE_len_idxs])
+    textE_len = int(dec(textE_lenE, encryption_code), 16)
+
+    print(colortext("Decrypting the text...", Fore.MAGENTA))
+    # Decrypt the text
+    textE_idxs = get_textE_idxs(seed, cipher_len, textE_len, textE_len_idxs)
+    textE = "".join([ciphertext[idx] for idx in textE_idxs])
+    text = h2t(dec(textE, encryption_code))
+
+    return text
+
+
+def get_img_arr(text:str) -> np.ndarray:
+    img_arr = np.array(
+            [[int(text[i:i+2], 16), 
+            int(text[i+2:i+4], 16), 
+            int(text[i+4:i+6], 16)] 
+            for i in range(0, len(text), 6)], 
+        dtype=np.uint8)
+    return img_arr.reshape(IMG_SIZE, IMG_SIZE, 3)

utils/menu.py

@@ -0,0 +1,119 @@
+import os
+from colorama import Fore, init; init()
+from PIL import Image
+import time
+
+from classes.Options import Options
+from utils.colortext import colortext
+from utils.encryption import encrypt_txt, decrypt_cipher, get_img_arr
+from constants.constants import *
+
+def menu():
+    create_folders()
+    print(f'\n{colortext("Max length of the text",Fore.CYAN)}: {colortext(f"{TXT_MAX_LEN:,}",Fore.GREEN)} (you can change it in {colortext("config.py",Fore.LIGHTYELLOW_EX)}')
+    option = Options(["EXIT", "Encrypt", "Decrypt text", "Decrypt image"]).get_choice()
+    
+    if option == 0: 
+        exit()
+
+    elif option == 1: 
+        encrypt()
+
+    elif option == 2:
+        decrypt_colortext()
+
+    elif option == 3:
+        decrypt_image()
+
+
+def create_folders():
+    if not os.path.exists(INPUT_DIR): os.mkdir(INPUT_DIR)
+    if not os.path.exists(OUTPUT_DIR): os.mkdir(OUTPUT_DIR)
+
+
+def encrypt():
+    text = open(get_input_file("text filename: ", "txt")).read()
+    pwd = input("Password: ")
+    enc_text_file = get_file("New text filename: ", "txt")
+    enc_img_file = get_file("New image filename: ", "png") if yes_no("Save image? [y/n]: ") else None
+
+    t1 = time.time()
+    try:
+        encrypted_text = encrypt_txt(text, pwd)
+        save_colortext(encrypted_text, enc_text_file)
+        print(colortext("\ntext encrypted succesfully\n", Fore.GREEN))
+        if enc_img_file:
+            save_img(encrypted_text, enc_img_file)
+            print(colortext("Image saved", Fore.GREEN))
+    except Exception as e:
+        print(colortext(f"Error: {e}", Fore.RED))
+        return
+
+    print(colortext(f"\nDone in {time.time() - t1:.2f} seconds", Fore.LIGHTYELLOW_EX))
+
+
+def decrypt(ciphertext:str) -> str:
+    pwd = input("Password: ")
+    dec_text_file = get_file("New filename: ", "txt")
+
+    t1 = time.time()
+    try:
+        decypted_text = decrypt_cipher(ciphertext, pwd)
+        save_colortext(decypted_text, dec_text_file)
+        print(colortext("\ntext decrypted succesfully\n", Fore.GREEN))
+    except Exception as e:
+        print(colortext(f"Error: {e}", Fore.RED))
+        return
+
+    print(colortext(f"\nDone in {time.time() - t1:.2f} seconds", Fore.LIGHTYELLOW_EX))
+
+
+def decrypt_colortext():
+    ciphertext = open(get_input_file("Ciphertext filename: ", "txt")).read()
+    decrypt(ciphertext)
+
+
+def decrypt_image():
+    ciphertext = img_to_colortext(get_input_file("Cipher image filename: ", "png"))
+    decrypt(ciphertext)
+
+
+def img_to_colortext(img_file:str) -> str:
+    img_arr = np.array(Image.open(img_file)).flatten()
+    img_str = "".join([f'{n:02x}' for n in img_arr])
+    return img_str
+
+
+def yes_no(msg:str="Save? [y/n]: ") -> bool:
+    inp = input(msg)
+    if inp.lower() == "y": return True
+    elif inp.lower() == "n": return False
+    else: return yes_no(msg)
+
+
+def save_colortext(text:str, file:str):
+    with open(f"{OUTPUT_DIR}/{file}", "w") as f:
+        f.write(text)
+
+
+def save_img(text:str, file:str):
+    img_arr = get_img_arr(text)
+    Image.fromarray(img_arr).save(f"{OUTPUT_DIR}/{file}")
+
+
+def get_file(msg:str, ext:str) -> str:
+    while True:
+        filename = input(msg)
+        if "." in filename:
+            if filename.endswith(f".{ext}"): return filename
+            print(colortext("Invalid extension", Fore.RED))
+            continue
+        return f"{filename}.{ext}"
+
+
+def get_input_file(msg:str, ext:str) -> str:
+    while True:
+        file = get_file(msg, ext)
+        path = f"{INPUT_DIR}/{file}"
+        if os.path.exists(path): return path
+        print(colortext("File not found",Fore.RED))