import logging
from pyfiglet import Figlet
from stringcolor import *
import sys

# Logger is aangeroepen over het hele project
import registry_keys

logging.basicConfig(filename="logboek.log", format='%(name)s: %(asctime)s %(levelname)s: %(message)s',
                    datefmt='%d/%m/%Y %I:%M:%S %p', level=logging.DEBUG)

# cowsay.daemon("Welcome to RELY, the malware analyser made by team fire breathing rubber duckies")

ascii_banner = Figlet(font='STANDARD')
ascii_banner2 = ascii_banner.renderText("Welcome to RELY")
print(cs(ascii_banner2, "blue"))
ascii_banner = Figlet(font='digital')
ascii_underbanner = ascii_banner.renderText("Team firebreathing rubberduckies")
ascii_underbanner2 = ascii_underbanner.replace("|", " ")
print(cs(ascii_underbanner2, "yellow"))
# ascii_banner = pyfiglet.figlet_format("Welcome  to  RELY")
# ascii_under = pyfiglet.print_figlet("", "big", 33)


# Waarop wil je scannen
Wanna_use_Keys = input("Do you want to scan on Registry keys (Y or N)? ")

if (Wanna_use_Keys == "N"):
    print("")
elif (Wanna_use_Keys == "Y"):

    geefHKEY = ""
    geefPad = ""
    filterVraag = ""
    filterNaam = ""
    filterType = ""

    geefHKEY = input("Please give an HKEY (e.g. HKEY_LOCAL_MACHINE): ")
    geefPad = input(
        "Please give the path you want to be scanned (e.g: " + r"SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\Git_is1" + "): ")

    filterVraag = input("Do you want to filter the registry keys? Y/N: ")
    if filterVraag.upper() == "Y":
        filterNaam = input("Do you want to filter on name? Please give the name else leave blank and press enter: ")
        filterType = input("Do you want to filter on type? Please give the type else leave blank and press enter: ")
    if filterVraag.upper() == "N":
        print()
    elif(filterVraag.upper() != "N" or "Y"):
        print("The input you gave did not correspond Y or N.")




else:
    print("Did not select Y or N, please restart program and try again")

if (Wanna_use_Keys == "Y"):
    registry_keys.main(geefHKEY, geefPad, filterVraag, filterNaam, filterType)