storm

#!/usr/bin/env python3

import os
import time
import sys
import readline  # noqa: F401
import app.base.config_ui as ui
import lib.smf.core.sf.ldch as ldch

from app.base.config_update import *
from app.utility.colors import C
from app.utility.verify import *
from app.banners.uib import banner

from lib.core import handler as h

def main():
    check_critical_files()
    run_verif()
    try:
        for i in range(6, 0, -1):
            sys.stdout.write(f"\r{C.SUCCESS}[*] Verification Success! Start Storm: [{i}] {C.RESET}")
            sys.stdout.flush()
            time.sleep(1)
    except KeyboardInterrupt: return

    os.system('clear')
    print(banner())
    ui.stormUI()
    check_update()

    current_module = None
    current_module_name = ""

    options = {
        "IP": "",
        "PORT": "",
        "PASS": "",
        "URL": "",
        "EMAIL": "",
        "HASH": "",
        "MESSAGE": "",
        "USER": "",
        "ID": "",
        "COUNT": "",
        "PATH": "",
        "INTERFACE": "",
        "THREAD": "",
        "DOMAIN": "",
        "HOSTNAME": "",
        "MODULE": ""
    }
    options = ldch.session(options)

    while True:
        START_INV = "\001"
        END_INV = "\002"

        BLUE = f"{START_INV}{C.BLUE}{END_INV}"
        CYAN = f"{START_INV}{C.MENU}{END_INV}"
        YELLOW = f"{START_INV}{C.INPUT}{END_INV}"
        RED = f"{START_INV}{C.ERROR}{END_INV}"
        RESET = f"{START_INV}{C.RESET}{END_INV}"

        p_mod = current_module_name if current_module else "~"
        try:
            print(f"{CYAN}┌─({BLUE}storm{YELLOW}<⚡>{BLUE}framework{CYAN}){RESET}-{RED}[{p_mod}]{RESET}")
            promp = f"{CYAN}└─{BLUE}➤ {RESET}"
            user_input = input(promp).strip().split()
        except KeyboardInterrupt: return
        except EOFError: break

        if not user_input: continue

        cmd = user_input[0].lower()
        args = user_input[1:]

        # Wrap all data into Context
        context = {
            "current_module": current_module,
            "current_module_name": current_module_name,
            "options": options,
            "exit": False
        }

        # HAND IT OVER TO THE FOREMAN (HANDLER)
        new_context = h.execute(cmd, args, context)

        if new_context:
            current_module = new_context["current_module"]
            current_module_name = new_context["current_module_name"]
            options = new_context["options"]

            if new_context.get("exit"): break
        else:
            print(f"[-] Unknown Command: {cmd} > Run the {C.SUCCESS}help{C.RESET} command for more details.")

if __name__ == "__main__":
    main()