You've already forked hackingtool
mirror of
https://github.com/Z4nzu/hackingtool.git
synced 2026-06-09 00:16:18 +02:00
Hardik Zinzuvadiya
26e38aea6e
core.py: - Add is_installed property to HackingTool class Checks: (1) shutil.which() for binary from first RUN_COMMAND, (2) os.path.isdir() for git clone target directory Handles "cd foo && binary" and "sudo binary" patterns - Add status column to HackingToolsCollection.show_options() table ✔ (green) = installed, ✘ (dim) = not installed - Archived/back rows updated for extra column
372 lines
13 KiB
Python
372 lines
13 KiB
Python
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"),
|
|
(" 99 ", "bold cyan"), ("go back\n", "white"),
|
|
(" 98 ", "bold cyan"), ("open project page\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
|
|
|
|
# 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)
|
|
|
|
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 == 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]")
|