zencryptCLI.py

# ****************************************************************************************
# Title: Hash Generator CLI        *******************************************************
# Developed by: Ryan Hatch         *******************************************************
# Date: August 10th 2023           *******************************************************
# Last Updated: October 13th 2023  *******************************************************
# Version: B1.5                    *******************************************************
# ****************************************************************************************

import hashlib
import getpass
import os
import pyperclip  # For cross-platform clipboard operations
from cryptography.fernet import Fernet

KEY_FILE = "CLIencryption_keyCLI.key"

def save_key_to_file(key):
    with open(KEY_FILE, "wb") as key_file:
        key_file.write(key)

def load_key_from_file():
    with open(KEY_FILE, "rb") as key_file:
        return key_file.read()

# Check if the key file exists. If not, generate a new key and save it.
if not os.path.exists(KEY_FILE):
    key = Fernet.generate_key()
    save_key_to_file(key)
else:
    key = load_key_from_file()

cipher_suite = Fernet(key)

def clear_clipboard():
    pyperclip.copy('')  # Clearing the clipboard
    print("\n\nClipboard cleared.")

def copy_to_clipboard(text):
    pyperclip.copy(text)
    print("\n\nOutput copied to clipboard.")

def decrypt_text():
    try:
        encrypted_text = input("\nEnter the encrypted text to decrypt: ")
        decrypted_text = cipher_suite.decrypt(encrypted_text.encode()).decode()
        print(f"\nDecrypted Text: {decrypted_text}")
    except Exception as e:
        print(f"\nError during decryption: {e}")
def encrypt_text():
    try:
        text_to_encrypt = input("\nEnter the text to encrypt: ")
        encrypted_text = cipher_suite.encrypt(text_to_encrypt.encode()).decode()
        print(f"\nEncrypted Text: {encrypted_text}")
        return encrypted_text
    except Exception as e:
        print(f"\nError during encryption: {e}")
        return None

def encryption_manager():
    while True:
        print("********************************************************************************")
        print("************************|  ENCRYPTION MAIN MENU  |******************************")
        print("********************************************************************************\n\n")

        print("********************************************************************************")
        print("* 1 | Clear Clipboard   |*******************************************************")
        # print("* 2 | Copy Output       |*******************************************************")
        print("********************************************************************************")
        print("* 2 | Encrypt           |*******************************************************")
        print("* 3 | Decrypt           |*******************************************************")
        print("********************************************************************************")
        print("=><=" * 20)
        print("=><=" * 20)
        print("********************************************************************************")
        print("* 4 | ******************|  HashGenerator Manager  |*****************************")
        print("********************************************************************************")
        print("\n\n")

        choice = input("Enter Option: ")
        if choice == "1":
            clear_clipboard()
        # elif choice == "2":
        #     # Here, we don't have a specific output to copy, so we might want to inform the user.
        #     print("\nNo specific output to copy. Use after encryption or decryption.")
        elif choice == "2":
            encrypted_text = encrypt_text()
            if encrypted_text:
                copy_to_clipboard(encrypted_text)
        elif choice == "3":
            decrypt_text()
        elif choice == "4":
            break
        else:
            print("\nInvalid Input.")

def print_menu(sha256_hash):
    print("\n\n\n")
    print("********************************************************************************")
    print("************************|  HASHGEN MAIN MENU  |*********************************")
    print("********************************************************************************\n\n")

    print("********************************************************************************")
    print("* 1 | Clear Clipboard   |*******************************************************")
    print("* 2 | Copy Output       |*******************************************************")
    print("********************************************************************************")
    print("* 3 | Generate New Hash |*******************************************************")
    print("* 4 | Verify Hash       |*******************************************************")
    print("********************************************************************************")
    print("=><=" * 20)
    print("=><=" * 20)
    print("********************************************************************************")
    print("* 5 | ******************|  Encryption Manager  |********************************")
    # print("* 5 | Encryption Manager|*******************************************************")
    print("* 6 | Exit              |*******************************************************")
    print("********************************************************************************")

    answer = input("\n\nEnter Option: ")
    if answer == "1":
        clear_clipboard()
    elif answer == "2":
        copy_to_clipboard(sha256_hash)
    elif answer == "3":
        main_loop()  # Restart the main loop
    elif answer == "4":
        verify_hash()  # Call the verify_hash function
    elif answer == "5":
        encryption_manager()
    elif answer == "6":
        exit()  # Exit program
    else:
        print("\nInvalid Input.")

# def print_menu(sha256_hash):
#     print("\n\n\n")
#     print("********************************************************************************")
#     print("* 1 | Clear Clipboard   |*******************************************************")
#     print("* 2 | Copy Output       |*******************************************************")
#     print("* 3 | Generate New Hash |*******************************************************")
#     print("* 4 | Verify Hash       |*******************************************************")
#     print("* 5 | Encrypt           |*******************************************************")
#     print("* 6 | Decrypt           |*******************************************************")
#     print("* 7 | Exit              |*******************************************************")
#     print("********************************************************************************")
#
#     answer = input("\n\nEnter Option: ")
#     if answer == "1":
#         clear_clipboard()
#     elif answer == "2":
#         copy_to_clipboard(sha256_hash)
#     elif answer == "3":
#         main_loop()  # Restart the main loop
#     elif answer == "4":
#         verify_hash()  # Call the verify_hash function
#     elif answer == "5":
#         encrypt_text()
#     elif answer == "6":
#         decrypt_text()
#     elif answer == "7":
#         exit()  # Exit program
#     else:
#         print("\nInvalid Input.")

def verify_hash():
    try:
        input_hash = input("\nEnter the hash to verify: ")
        original_text = input("\nEnter the original text to verify against the hash: ")
        salt = input("Enter the salt value used during hashing: ")  # Ask for the salt
        computed_hash = hashlib.sha256((original_text + salt).encode()).hexdigest()  # Apply the salt

        if computed_hash == input_hash:
            print("\n\nHash successfully verified.")
        else:
            print("\n\nVerification unsuccessful. Hash does not match.")
    except Exception as e:
        print(f"\n\nError during verification: {e}")

def main_loop():
    counter = 0

    while True:
        text = getpass.getpass(prompt="\n\n. . . . . .")
        if text == "exit":
            break
        counter += 1
        print(f"\n--- session {counter} ---")
        try:
            # Adding a salt for security
            salt = input("Enter salt value:")
            sha256_hash = hashlib.sha256((text + salt).encode()).hexdigest()
            print("\n\nOutput:\n")
            print("-" * 80)
            print(sha256_hash)
            print("-" * 80)

            print_menu(sha256_hash)
        except Exception as e:
            print(f"\nError: {e}")

    input("\nPress Enter To Exit.")

# ASCII Art
print("\n")
print("                           /\\")
print("                          /__\\")
print("                         /\\  /\\")
print("                        /__\\/__\\")
print("                       /\\      /\\")
print("                      /__\\    /__\\")
print("                     /\\  /\\  /\\  /\\")
print("                    /__\\/__\\/__\\/__\\")

main_loop()