diff --git a/Modules/All_Pages/wrappers.py b/Modules/All_Pages/wrappers.py deleted file mode 100644 index 8b3cc06..0000000 --- a/Modules/All_Pages/wrappers.py +++ /dev/null @@ -1,83 +0,0 @@ -import subprocess -from colorama import Fore, Style - -from Assets.ascii_text_prompts import ascii_art, full_ascii_art, infinitei -from Modules.All_Pages.clear_screen import clear_screen -from Modules.All_Pages.random_tip import get_random_tip_with_color - -from Modules.All_Pages.center_text import * - -from Modules.Login.scan import BACKGROUND_TASK -from Modules.Login.check_ip import get_target_ip - - - -def highlight_ports(port, open_ports): # Highlight a port if it is in the list of open ports. - return f"{Fore.GREEN}{port}{Style.RESET_ALL}" if port in open_ports else f"{port}" - - - -def display_background_task(): - if BACKGROUND_TASK: - print("\nBackground Task:") - for task in BACKGROUND_TASK: - print(f" - {task}") - else: - print("\nBackground Task: None") - - - -def header(target_ip, open_ports): - clear_screen() - print(center_text(ascii_art)) - print(center_text(get_random_tip_with_color()) + "\n") - print(center_text(f"Target IP: {target_ip}\n")) - if open_ports: - print(center_text(f"Open Ports: {', '.join(open_ports)}\n")) - else: - print(center_text("Open Ports: None\n")) - display_background_task() # Include the background tasks section - - - -def display_menu(menu_options, open_ports, target_ip): - while True: - header(target_ip, open_ports) # Pass target_ip and open_ports to the header - print("\nStandard Enumeration") - for key, value in menu_options.items(): - ports = "/".join([highlight_ports(port, open_ports) for port in value["ports"]]) - print(f"[{key}] {value['name'].upper()}: Ports {ports}") - print("[0] Logout\n") - - choice = input("Enter your choice: ").strip().lower() - - # Match the choice to a menu option - if choice in menu_options: - menu_options[choice]["submenu"](target_ip, open_ports) # Pass arguments directly - elif choice in ["0", "q", "exit", "quit"]: - print("Logging out...") - break - else: - print("Invalid choice. Please try again.") - - - -def run_command(title, content, target_ip, open_ports): - header(target_ip, open_ports) # Display the header at the top - print("=" * 40) - print(f"{title.center(40)}") - print("=" * 40) - print("\n") - print("Executing Command:\n") - print(content) - print("\n") - - try: - # Execute the command and display its output in real-time - subprocess.run(content, shell=True, check=True) - except subprocess.CalledProcessError as e: - print(f"Error executing command:\n{e}") - except Exception as ex: - print(f"An unexpected error occurred: {ex}") - finally: - input("\nPress Enter to return...") \ No newline at end of file