adapter.py

"""
this is the adapter to connecting sigma.py and core.py, more like a view model
but i prefer called it adapter
"""
from utils import *
from core import Sigma

sigma = Sigma()

def nuke(path, algo):
    nuked =0
    tokens = {}
    print("\nWARNING THIS FUNCTION STILL UNDER DEVELOPMENT")
    print("\n###### Scanning the targeted Dir ######\n")

    list_of_files = list_files(path)
    if len(list_of_files) <=0:
        print("INFO : nothing to nuke here....\n")
        exit()
        
    print("###### Scanning the targeted Dir ######")
    print()
    print("Let's start the nuke")
    #print("\nsupported files : ", list_of_files)
    for file in list_of_files:
        token = algo.generate_token()
        priv_key = algo.generate_private_key(token)
        the_file = path +file
        content = ""
        try:
            if fileSupported(the_file):
                content = readFileContent(the_file)
        except Exception as e:
            print("\nError: {} {}".format(the_file, e))
            print("WTH are you doing list_files() function, this file is not supported but you included in the list\n")
            continue
        encoded = algo.start_encode(content, token)
        overwrite_file(the_file, encoded)
        print("Succesfully nuke file : " + file , "with token / key : ", token)
        nuked += 1
        tokens.update({file : [str(token), str(priv_key)]})
    token_file = storeTokenIntoFile(str(tokens), "nuked-dir", True)
    print("succesfully nuked {} files from total {} files".format(nuked, len(list_of_files)))
    print("keys is stored in file : ", token_file)
    print("\n###### Nuke Completed ######")

def Encoder():
    print("this is encoder module")
    data = ""
    token = ""
    file_or_text = input("\nFile or Text ? : ")
    if file_or_text.lower() == "file":
        file_path = input("\nEnter the file path : ")
        if fileSupported(file_path):
            data = readFileContent(file_path)
            print("\nChecked if File is supported : " + file_path)
        else:
            print("\nError: {} is not supported".format(file_path))
            print("ctrl + c to cancel")

    elif file_or_text.lower() == "text":
    
        data = input("\nPlease enter the text to encode : ")
        print()

    print("Let's start the encode")
    answer = input("\nDo you have your own token ? (y/n) : ")
    if answer.lower() == "y":
        token = input("\nPlease enter your token : ")
    else:
        print("### generating token ###")
        try:
            token_len = int(input("\nPlease enter the token length : "))
            token = sigma.generate_token(token_len)
        except:
            print("Not detecting any input from user, using default token length")
            token = sigma.generate_token()
    priv_key = sigma.generate_private_key(token)
    encoded = sigma.start_encode(data, token)
    print("succesfully encoded")
    print("token / public key : ", token)
    print("private key : ", priv_key)
    print("encoded : ", encoded)
    print("\n###### Encoding Completed ######")

    save_to_file = input("\nDo you want to save the encoded data to file ? (y/n) : ")
    if save_to_file.lower() == "y":
        new_file_name = input("\nPlease enter the file name : ")
        makeCopyOfFile(new_file_name, encoded)
        storeTokenIntoFile(token, priv_key, new_file_name, False)
    else:
        print("okay")



def Decoder():
    print("this is decoder module")
    data = ""
    token = ""
    file_or_text = input("\nFile or Text ? : ")
    if file_or_text.lower() == "file":
        file_path = input("\nEnter the file path : ")
        
        if fileSupported(file_path):
            print("file supported")
            data = readFileContent(file_path)
            #print("\nChecked if File is supported : " + file_path)
        else:
            print("\nError: {} is not supported".format(file_path))
            print("ctrl + c to cancel")
            exit()

    elif file_or_text.lower() == "text":
    
        data = input("\nPlease enter the text to encode : ")
        print()

    print("Let's start the decode")
    answer = input("\nDo you have your private key ? (y/n) : ")
    if answer.lower() == "y":
        token = input("\nPlease enter your private key : ")
    else:
        print()
        print("### you must have your private key, otherwise you can't decode it  ###")
        return "User Don't have key"
       
    priv_key = token 
    decoded = sigma.start_decode(data, priv_key)
    print("succesfully decoded")
    print("decoded : ", decoded)
    print("\n###### Decoding Completed ######")