import os import shutil import sys import webbrowser from collections.abc import Callable from platform import system from rich import box from rich.console import Console from rich.panel import Panel from rich.prompt import Prompt from rich.table import Table from rich.text import Text from rich.theme import Theme from rich.traceback import install from constants import ( THEME_PRIMARY, THEME_BORDER, THEME_ACCENT, THEME_SUCCESS, THEME_ERROR, THEME_WARNING, THEME_DIM, THEME_ARCHIVED, THEME_URL, ) # Enable rich tracebacks globally install() _theme = Theme({ "purple": "#7B61FF", "success": THEME_SUCCESS, "error": THEME_ERROR, "warning": THEME_WARNING, "archived": THEME_ARCHIVED, "url": THEME_URL, "dim": THEME_DIM, }) # Single shared console — all tool files do: from core import console console = Console(theme=_theme) def clear_screen(): os.system("cls" if system() == "Windows" else "clear") def validate_input(ip, val_range: list) -> int | None: """Return the integer if it is in val_range, else None.""" if not val_range: return None try: ip = int(ip) if ip in val_range: return ip except (TypeError, ValueError): pass return None def _show_inline_help(): """Quick help available from any menu level.""" console.print(Panel( Text.assemble( (" Navigation\n", "bold white"), (" ─────────────────────────────────\n", "dim"), (" 1–N ", "bold cyan"), ("select item\n", "white"), (" 97 ", "bold cyan"), ("install all (in category)\n", "white"), (" 99 ", "bold cyan"), ("go back\n", "white"), (" 98 ", "bold cyan"), ("open project page / archived\n", "white"), (" ? ", "bold cyan"), ("show this help\n", "white"), (" q ", "bold cyan"), ("quit hackingtool\n", "white"), ), title="[bold magenta] ? Quick Help [/bold magenta]", border_style="magenta", box=box.ROUNDED, padding=(0, 2), )) Prompt.ask("[dim]Press Enter to return[/dim]", default="") class HackingTool: TITLE: str = "" DESCRIPTION: str = "" INSTALL_COMMANDS: list[str] = [] UNINSTALL_COMMANDS: list[str] = [] RUN_COMMANDS: list[str] = [] OPTIONS: list[tuple[str, Callable]] = [] PROJECT_URL: str = "" # OS / capability metadata SUPPORTED_OS: list[str] = ["linux", "macos"] REQUIRES_ROOT: bool = False REQUIRES_WIFI: bool = False REQUIRES_GO: bool = False REQUIRES_RUBY: bool = False REQUIRES_JAVA: bool = False REQUIRES_DOCKER: bool = False # Tags for search/filter (e.g. ["osint", "web", "recon", "scanner"]) TAGS: list[str] = [] # Archived tool flags ARCHIVED: bool = False ARCHIVED_REASON: str = "" def __init__(self, options=None, installable=True, runnable=True): options = options or [] if not isinstance(options, list): raise TypeError("options must be a list of (option_name, option_fn) tuples") self.OPTIONS = [] if installable: self.OPTIONS.append(("Install", self.install)) if runnable: self.OPTIONS.append(("Run", self.run)) self.OPTIONS.extend(options) @property def is_installed(self) -> bool: """Check if the tool's binary is on PATH or its clone dir exists.""" if self.RUN_COMMANDS: cmd = self.RUN_COMMANDS[0] # Handle "cd foo && binary --help" pattern if "&&" in cmd: cmd = cmd.split("&&")[-1].strip() if cmd.startswith("sudo "): cmd = cmd[5:].strip() binary = cmd.split()[0] if cmd else "" if binary and binary not in (".", "echo", "cd"): if shutil.which(binary): return True # Check if git clone target dir exists if self.INSTALL_COMMANDS: for ic in self.INSTALL_COMMANDS: if "git clone" in ic: parts = ic.split() repo_url = [p for p in parts if p.startswith("http")] if repo_url: dirname = repo_url[0].rstrip("/").rsplit("/", 1)[-1].replace(".git", "") if os.path.isdir(dirname): return True return False def show_info(self): desc = f"[cyan]{self.DESCRIPTION}[/cyan]" if self.PROJECT_URL: desc += f"\n[url]🔗 {self.PROJECT_URL}[/url]" if self.ARCHIVED: desc += f"\n[archived]⚠ ARCHIVED: {self.ARCHIVED_REASON}[/archived]" console.print(Panel( desc, title=f"[{THEME_PRIMARY}]{self.TITLE}[/{THEME_PRIMARY}]", border_style="purple", box=box.DOUBLE, )) def show_options(self, parent=None): """Iterative menu loop — no recursion, no stack growth.""" while True: clear_screen() self.show_info() table = Table(title="Options", box=box.SIMPLE_HEAVY) table.add_column("No.", style="bold cyan", justify="center") table.add_column("Action", style="bold yellow") for index, option in enumerate(self.OPTIONS): table.add_row(str(index + 1), option[0]) if self.PROJECT_URL: table.add_row("98", "Open Project Page") table.add_row("99", f"Back to {parent.TITLE if parent else 'Main Menu'}") console.print(table) console.print( "[dim] Enter number · [bold cyan]?[/bold cyan] help" " · [bold cyan]q[/bold cyan] quit[/dim]" ) raw = Prompt.ask("\n[bold cyan]>[/bold cyan]", default="").strip().lower() if not raw: continue if raw in ("?", "help"): _show_inline_help() continue if raw in ("q", "quit", "exit"): raise SystemExit(0) try: choice = int(raw) except ValueError: console.print("[error]⚠ Enter a number, ? for help, or q to quit.[/error]") Prompt.ask("[dim]Press Enter to continue[/dim]", default="") continue if choice == 99: return elif choice == 98 and self.PROJECT_URL: self.show_project_page() elif 1 <= choice <= len(self.OPTIONS): try: self.OPTIONS[choice - 1][1]() except Exception: console.print_exception(show_locals=True) Prompt.ask("[dim]Press Enter to continue[/dim]", default="") else: console.print("[error]⚠ Invalid option.[/error]") def before_install(self): pass def install(self): self.before_install() if isinstance(self.INSTALL_COMMANDS, (list, tuple)): for cmd in self.INSTALL_COMMANDS: console.print(f"[warning]→ {cmd}[/warning]") os.system(cmd) self.after_install() def after_install(self): console.print("[success]✔ Successfully installed![/success]") def before_uninstall(self) -> bool: return True def uninstall(self): if self.before_uninstall(): if isinstance(self.UNINSTALL_COMMANDS, (list, tuple)): for cmd in self.UNINSTALL_COMMANDS: console.print(f"[error]→ {cmd}[/error]") os.system(cmd) self.after_uninstall() def after_uninstall(self): pass def before_run(self): pass def run(self): self.before_run() if isinstance(self.RUN_COMMANDS, (list, tuple)): for cmd in self.RUN_COMMANDS: console.print(f"[cyan]⚙ Running:[/cyan] [bold]{cmd}[/bold]") os.system(cmd) self.after_run() def after_run(self): pass def show_project_page(self): console.print(f"[url]🌐 Opening: {self.PROJECT_URL}[/url]") webbrowser.open_new_tab(self.PROJECT_URL) class HackingToolsCollection: TITLE: str = "" DESCRIPTION: str = "" TOOLS: list = [] def __init__(self): pass def show_info(self): console.rule(f"[{THEME_PRIMARY}]{self.TITLE}[/{THEME_PRIMARY}]", style="purple") if self.DESCRIPTION: console.print(f"[italic cyan]{self.DESCRIPTION}[/italic cyan]\n") def _active_tools(self) -> list: """Return tools that are not archived and are OS-compatible.""" from os_detect import CURRENT_OS return [ t for t in self.TOOLS if not getattr(t, "ARCHIVED", False) and CURRENT_OS.system in getattr(t, "SUPPORTED_OS", ["linux", "macos"]) ] def _archived_tools(self) -> list: return [t for t in self.TOOLS if getattr(t, "ARCHIVED", False)] def _incompatible_tools(self) -> list: from os_detect import CURRENT_OS return [ t for t in self.TOOLS if not getattr(t, "ARCHIVED", False) and CURRENT_OS.system not in getattr(t, "SUPPORTED_OS", ["linux", "macos"]) ] def _show_archived_tools(self): """Show archived tools sub-menu (option 98).""" archived = self._archived_tools() if not archived: console.print("[dim]No archived tools in this category.[/dim]") Prompt.ask("[dim]Press Enter to return[/dim]", default="") return while True: clear_screen() console.rule(f"[archived]Archived Tools — {self.TITLE}[/archived]", style="yellow") table = Table(box=box.MINIMAL_DOUBLE_HEAD, show_lines=True) table.add_column("No.", justify="center", style="bold yellow") table.add_column("Tool", style="dim yellow") table.add_column("Reason", style="dim white") for i, tool in enumerate(archived): reason = getattr(tool, "ARCHIVED_REASON", "No reason given") table.add_row(str(i + 1), tool.TITLE, reason) table.add_row("99", "Back", "") console.print(table) raw = Prompt.ask("[bold yellow][?] Select[/bold yellow]", default="99") try: choice = int(raw) except ValueError: continue if choice == 99: return elif 1 <= choice <= len(archived): archived[choice - 1].show_options(parent=self) def show_options(self, parent=None): """Iterative menu loop — no recursion, no stack growth.""" while True: clear_screen() self.show_info() active = self._active_tools() incompatible = self._incompatible_tools() archived = self._archived_tools() table = Table(title="Available Tools", box=box.SIMPLE_HEAD, show_lines=True) table.add_column("No.", justify="center", style="bold cyan", width=6) table.add_column("", width=2) # installed indicator table.add_column("Tool", style="bold yellow", min_width=24) table.add_column("Description", style="white", overflow="fold") for index, tool in enumerate(active, start=1): desc = getattr(tool, "DESCRIPTION", "") or "—" desc = desc.splitlines()[0] if desc != "—" else "—" status = "[green]✔[/green]" if tool.is_installed else "[dim]✘[/dim]" table.add_row(str(index), status, tool.TITLE, desc) # Count not-installed tools for "Install All" label not_installed = [t for t in active if not t.is_installed] if not_installed: table.add_row( "[bold green]97[/bold green]", "", f"[bold green]Install all ({len(not_installed)} not installed)[/bold green]", "", ) if archived: table.add_row("[dim]98[/dim]", "", f"[archived]Archived tools ({len(archived)})[/archived]", "") if incompatible: console.print(f"[dim]({len(incompatible)} tools hidden — not supported on current OS)[/dim]") table.add_row("99", "", f"Back to {parent.TITLE if parent else 'Main Menu'}", "") console.print(table) console.print( "[dim] Enter number · [bold cyan]?[/bold cyan] help" " · [bold cyan]q[/bold cyan] quit[/dim]" ) raw = Prompt.ask("\n[bold cyan]>[/bold cyan]", default="").strip().lower() if not raw: continue if raw in ("?", "help"): _show_inline_help() continue if raw in ("q", "quit", "exit"): raise SystemExit(0) try: choice = int(raw) except ValueError: console.print("[error]⚠ Enter a number, ? for help, or q to quit.[/error]") continue if choice == 99: return elif choice == 97 and not_installed: console.print(Panel( f"[bold]Installing {len(not_installed)} tools...[/bold]", border_style="green", box=box.ROUNDED, )) for i, tool in enumerate(not_installed, start=1): console.print(f"\n[bold cyan]({i}/{len(not_installed)})[/bold cyan] {tool.TITLE}") try: tool.install() except Exception: console.print(f"[error]✘ Failed: {tool.TITLE}[/error]") Prompt.ask("\n[dim]Press Enter to continue[/dim]", default="") elif choice == 98 and archived: self._show_archived_tools() elif 1 <= choice <= len(active): try: active[choice - 1].show_options(parent=self) except Exception: console.print_exception(show_locals=True) Prompt.ask("[dim]Press Enter to continue[/dim]", default="") else: console.print("[error]⚠ Invalid option.[/error]")