main.py

import sys
import os
import shutil
import subprocess
from rich.console import Console
from rich.panel import Panel
from rich.text import Text
import questionary

from utils.validator import ToolValidator

console = Console()

BANNER = r"""
           _         _
          /x\       /x\
         /v\x\     /v\/\
         \><\x\   /></x/
          \><\x\ /></x/
  __ __  __|><\x/></x/___
 /##_##\/       \</x/    \__________
|###|###|  \         \    __________\
 \##|##/ \__\____\____\__/          \\
   |_|   |  |  | |  | |              \|
   \*/   \  |  | |  | /              /
  ____  __  __ _   _ ___ ____   ____    _    _   _ 
 / __ \|  \/  | \ | |_ _/ ___| / ___|  / \  | \ | |
| |  | | |\/| |  \| || |\___ \| |     / _ \ |  \| |
| |__| | |  | | |\  || | ___) | |___ / ___ \| |\  |
 \____/|_|  |_|_| \_|___|____/ \____/_/   \_\_| \_|
       > The All-in-One Recon & Wordlist Suite <
"""

MENU_CHOICES = [
    "1. Wordlist Generator",
    "2. Web Analysis & Fuzzing  [ffuf, gobuster, wpscan]",
    "3. Network Scanning        [nmap, netcat, enum4linux]",
    "4. OSINT & Recon           [amass, sherlock, theHarvester]",
    "5. Info Gathering          [whois, nslookup]",
    "6. Exit",
]

# Ordered list of terminal emulators to try when spawning a new window.
# Each entry: (binary, [args_before_exec_flag], exec_flag)
TERMINAL_CANDIDATES = [
    ("kitty",         [],                          None),          # kitty runs cmd directly
    ("alacritty",     [],                          "-e"),
    ("gnome-terminal",["--"],                      None),          # uses -- separator
    ("konsole",       [],                          "-e"),
    ("xfce4-terminal",[],                          "-x"),
    ("tilix",         [],                          "-e"),
    ("terminator",    [],                          "-x"),
    ("urxvt",         [],                          "-e"),
    ("rxvt",          [],                          "-e"),
    ("xterm",         [],                          "-e"),
]


# ---------------------------------------------------------------------------
# New-window launcher
# ---------------------------------------------------------------------------

def _find_terminal() -> tuple[str, list[str], str | None] | None:
    """Return the first available terminal entry, or None if none found."""
    for binary, pre_args, exec_flag in TERMINAL_CANDIDATES:
        if shutil.which(binary):
            return binary, pre_args, exec_flag
    return None


def relaunch_in_new_window() -> None:
    """
    Spawn a new terminal window running OmniScan, then exit this process.
    If no supported terminal is found, continue in the current window with a warning.
    """
    entry = _find_terminal()
    if entry is None:
        console.print(
            "[bold yellow][!] No supported terminal emulator found.[/bold yellow]\n"
            "[dim]    Tried: " + ", ".join(b for b, *_ in TERMINAL_CANDIDATES) + "\n"
            "    Running in current window.[/dim]\n"
        )
        return

    binary, pre_args, exec_flag = entry

    # Reconstruct the command that launched us, preserving venv python
    python = sys.executable
    script  = os.path.abspath(__file__)
    workdir = os.path.dirname(script)

    # Pass a special env var so the relaunched process skips the window-spawn
    env = os.environ.copy()
    env["OMNISCAN_NEW_WINDOW"] = "1"

    if binary == "kitty":
        cmd = ["kitty", "--directory", workdir, python, script]
    elif binary == "gnome-terminal":
        cmd = ["gnome-terminal", "--working-directory", workdir, "--"] + [python, script]
    else:
        cmd = [binary] + pre_args
        if exec_flag:
            cmd += [exec_flag]
        cmd += [python, script]

    try:
        subprocess.Popen(cmd, env=env, cwd=workdir)
        sys.exit(0)
    except Exception as e:
        console.print(f"[bold yellow][!] Could not open new window: {e}[/bold yellow]\n"
                      "[dim]    Running in current window.[/dim]\n")


# ---------------------------------------------------------------------------
# Startup checks
# ---------------------------------------------------------------------------

def display_banner() -> None:
    text = Text(BANNER, style="bold cyan")
    console.print(Panel(text, expand=False, border_style="cyan"))


def validate_environment() -> None:
    """
    Check tool availability at startup.

    Tools resolved via an alias (e.g. theharvester instead of theHarvester)
    are considered available and are NOT listed as missing.
    """
    validator = ToolValidator()
    missing = validator.check_all()

    if missing:
        console.print(
            f"\n[bold yellow][!] Tools not found: {', '.join(missing)}[/bold yellow]"
        )
        console.print(
            "[dim]    Install them or add them to your PATH. "
            "If inside a venv, they may only be available system-wide.[/dim]\n"
        )


# ---------------------------------------------------------------------------
# Main menu
# ---------------------------------------------------------------------------

def main_menu() -> None:
    while True:
        choice = questionary.select(
            "Select a scan category:",
            choices=MENU_CHOICES,
            style=questionary.Style([
                ("selected", "fg:cyan bold"),
                ("pointer",  "fg:cyan bold"),
                ("highlighted", "fg:cyan"),
            ]),
        ).ask()

        if choice is None:
            raise KeyboardInterrupt

        if choice.startswith("1"):
            from core.generator import WordlistGenerator
            WordlistGenerator(console).run()

        elif choice.startswith("2"):
            from modules.web import WebModule
            WebModule(console).run()

        elif choice.startswith("3"):
            from modules.network import NetworkModule
            NetworkModule(console).run()

        elif choice.startswith("4"):
            from modules.osint import OSINTModule
            OSINTModule(console).run()

        elif choice.startswith("5"):
            from modules.info import InfoModule
            InfoModule(console).run()

        elif choice.startswith("6"):
            console.print("\n[bold green]Happy Hacking! Goodbye.[/bold green]\n")
            sys.exit(0)


# ---------------------------------------------------------------------------
# Entry point
# ---------------------------------------------------------------------------

if __name__ == "__main__":
    # Only try to open a new window on the first launch (not when already
    # running inside the spawned window).
    if not os.environ.get("OMNISCAN_NEW_WINDOW"):
        relaunch_in_new_window()

    display_banner()
    validate_environment()

    try:
        main_menu()
    except KeyboardInterrupt:
        console.print("\n[bold red]\n[!] Interrupted by user. Goodbye.[/bold red]\n")
        sys.exit(0)