main.py

import sys

from error_handler import handle_error

# Hack to fix python 3.10 support
if sys.version_info >= (3, 10):
    from typing import Mapping
    import collections

    # imported in prompt_toolkit 1.0.14 which is strictly
    # relied on by PyInquirer
    collections.Mapping = Mapping  # type: ignore

# Importing 3rd Party Libraries
from pyfiglet import Figlet
from PyInquirer import prompt, Separator
from termcolor import colored

# Importing the local functions
from encrypt_code import encrypt_func
from decrypt_code import decrypt_func
from hash_code import hash_func


def main():
    # Generating the initial output text with pyfiglet
    f = Figlet(font='slant')
    credit = colored('                                                 By Arpan Pandey\n', 'yellow', attrs=['bold'])
    description = colored('  A tool to hash or encrypt your data easily using Fernet Encryption.'
                          ' It is very easy and intuitive to use.'
                          ' You can also use this on any type of file below 1GB.', 'cyan')
    print(colored(f.renderText('Encrypto CLI'), 'green'), credit, description, '\n')

## print(colored('Please enter a password', 'red'))

## print(colored('The encrypted text is: ', 'white') +
##        colored(encrypted_text, 'green'))
    
    # Asking the user about which operation do they want to perform
    operation: str = prompt([
        {
            'type': 'list',
            'qmark': '🔘',
            'name': 'operation',
            'message': 'What do you want to do?',
            'choices': [
                Separator(),
                {
                    'name': 'Hash',
                },
                {
                    'name': 'Encrypt',
                },
                {
                    'name': 'Decrypt',
                },
                {
                    'name': 'Exit',
                }
            ],
        }
    ]).get('operation', None)

    if not operation:
        # user hit Ctrl+C
        return


    try:

        # Calling the hash function if the selected operation is Hashing
        if operation == 'Hash':
            # * Called the hashing function
            hash_func()

        # Calling the Encryption function if the selected operation is encryption
        elif operation == 'Encrypt':
            # * Called the encryption function
            encrypt_func()

        # Calling the decryption function if the selected operation is decryption
        elif operation == 'Decrypt':
            # * Called the decryption function
            decrypt_func()
        elif operation == 'Exit':
            print(colored("goodbye :)", "blue"))
        else:
            raise NotImplementedError("operation not supported yet.")

    except Exception as e:
        handle_error(e)

if __name__ == "__main__":
    main()