#!/usr/bin/env python3
# install_hackingtool.py  (rich-based installer UI)
import os
import sys
import shutil
import subprocess
from pathlib import Path

from rich.console import Console
from rich.panel import Panel
from rich.prompt import Prompt, Confirm, IntPrompt
from rich.table import Table
from rich.align import Align
from rich.progress import Progress, SpinnerColumn, TextColumn
from rich.text import Text
from rich import box
from random import choice

console = Console()

REPO_URL = "https://github.com/Z4nzu/hackingtool.git"
INSTALL_DIR = Path("/usr/share/hackingtool")
BIN_PATH = Path("/usr/bin/hackingtool")
VENV_DIR_NAME = "venv"
REQUIREMENTS = "requirements.txt"


def check_root():
    if os.geteuid() != 0:
        console.print(Panel("[red]This installer must be run as root. Use: sudo python3 install_hackingtool.py[/red]"))
        sys.exit(1)


def run_cmd(cmd, check=True, capture=False, env=None):
    return subprocess.run(cmd, shell=True, check=check, capture_output=capture, text=True, env=env)


def colorful_logo():
    logos = ["magenta", "bright_magenta", "cyan", "blue", "green", "yellow"]
    style = choice(logos)
    logo_lines = r"""
   ▄█    █▄       ▄████████  ▄████████    ▄█   ▄█▄  ▄█  ███▄▄▄▄      ▄██████▄           ███      ▄██████▄   ▄██████▄   ▄█       
  ███    ███     ███    ███ ███    ███   ███ ▄███▀ ███  ███▀▀▀██▄   ███    ███      ▀█████████▄ ███    ███ ███    ███ ███       
  ███    ███     ███    ███ ███    █▀    ███▐██▀   ███▌ ███   ███   ███    █▀          ▀███▀▀██ ███    ███ ███    ███ ███       
 ▄███▄▄▄▄███▄▄   ███    ███ ███         ▄█████▀    ███▌ ███   ███  ▄███                 ███   ▀ ███    ███ ███    ███ ███       
▀▀███▀▀▀▀███▀  ▀███████████ ███        ▀▀█████▄    ███▌ ███   ███ ▀▀███ ████▄           ███     ███    ███ ███    ███ ███       
  ███    ███     ███    ███ ███    █▄    ███▐██▄   ███  ███   ███   ███    ███          ███     ███    ███ ███    ███ ███       
  ███    ███     ███    ███ ███    ███   ███ ▀███▄ ███  ███   ███   ███    ███          ███     ███    ███ ███    ███ ███▌    ▄ 
  ███    █▀      ███    █▀  ████████▀    ███   ▀█▀ █▀    ▀█   █▀    ████████▀          ▄████▀    ▀██████▀   ▀██████▀  █████▄▄██ 
                                         ▀                                                                            ▀                   
"""
    panel = Panel(Text(logo_lines, style=style), box=box.DOUBLE, border_style=style)
    console.print(panel)
    console.print(f"[bold {style}]https://github.com/Z4nzu/hackingtool[/bold {style}]\n")


def choose_distro():
    console.print(Panel("[bold magenta]Select installation target[/bold magenta]\n\n[1] Kali / Parrot (apt)\n[2] Arch (pacman)\n[0] Exit", border_style="bright_magenta"))
    choice = IntPrompt.ask("Choice", choices=["0", "1", "2"], default=1)
    return choice


def check_internet():
    console.print("[yellow]* Checking internet connectivity...[/yellow]")
    try:
        run_cmd("curl -sSf --max-time 10 https://www.google.com > /dev/null", check=True)
        console.print("[green][✔] Internet connection OK[/green]")
        return True
    except Exception:
        try:
            run_cmd("curl -sSf --max-time 10 https://github.com > /dev/null", check=True)
            console.print("[green][✔] Internet connection OK[/green]")
            return True
        except Exception:
            console.print("[red][✘] Internet connection not available[/red]")
            return False


def system_update_and_install(choice):
    if choice == 1:
        console.print("[yellow]* Running apt update/upgrade...[/yellow]")
        try:
            run_cmd("apt update -y && apt upgrade -y")
        except subprocess.CalledProcessError as e:
            console.print(f"[red][!][/red] apt update/upgrade failed (non-fatal). Continuing installation. Error: {e}")
        console.print("[yellow]* Installing required packages (apt)...[/yellow]")
        try:
            run_cmd("apt-get install -y git python3-pip python3-venv figlet boxes php curl xdotool wget")
        except subprocess.CalledProcessError as e:
            console.print(f"[red][!][/red] apt-get install failed (non-fatal). You may need to install some packages manually. Error: {e}")
    elif choice == 2:
        console.print("[yellow]* Running pacman update...[/yellow]")
        try:
            run_cmd("pacman -Syu --noconfirm")
        except subprocess.CalledProcessError as e:
            console.print(f"[red][!][/red] pacman update failed (non-fatal). Continuing installation. Error: {e}")
        console.print("[yellow]* Installing required packages (pacman)...[/yellow]")
        try:
            run_cmd("pacman -S --noconfirm git python-pip")
        except subprocess.CalledProcessError as e:
            console.print(f"[red][!][/red] pacman install failed (non-fatal). You may need to install some packages manually. Error: {e}")
    else:
        console.print("[red]Invalid package manager choice[/red]")


def prepare_install_dir():
    if INSTALL_DIR.exists():
        console.print(f"[red]The directory {INSTALL_DIR} already exists.[/red]")
        if Confirm.ask("Replace it? This will remove the existing directory", default=False):
            run_cmd(f"rm -rf {str(INSTALL_DIR)}")
        else:
            console.print("[red]Installation aborted by user.[/red]")
            sys.exit(1)
    INSTALL_DIR.mkdir(parents=True, exist_ok=True)


def git_clone():
    console.print("[yellow]* Cloning hackingtool repository...[/yellow]")
    try:
        run_cmd(f"git clone {REPO_URL} {str(INSTALL_DIR)}")
        console.print("[green][✔] Repository cloned[/green]")
        return True
    except Exception as e:
        console.print(f"[red][✘] Failed to clone repository: {e}[/red]")
        return False


def create_venv_and_install(choice):
    venv_path = INSTALL_DIR / VENV_DIR_NAME
    console.print("[yellow]* Creating virtual environment...[/yellow]")
    run_cmd(f"python3 -m venv {str(venv_path)}")
    activate = venv_path / "bin" / "activate"
    pip = str(venv_path / "bin" / "pip")
    if (INSTALL_DIR / REQUIREMENTS).exists():
        console.print("[yellow]* Installing Python requirements...[/yellow]")
        run_cmd(f"{pip} install -r {str(INSTALL_DIR / REQUIREMENTS)}")
    else:
        console.print("[yellow]requirements.txt not found, skipping pip install.[/yellow]")
    if choice == 1:
        run_cmd("apt install figlet -y")
    elif choice == 2:
        # try pacman and fallback to AUR instructions
        try:
            run_cmd("pacman -S --noconfirm figlet")
        except Exception:
            console.print("[yellow]figlet not available in pacman automatically. Consider installing from AUR.[/yellow]")


def create_launcher():
    console.print("[yellow]* Creating launcher script...[/yellow]")
    launcher = INSTALL_DIR / "hackingtool.sh"
    with open(launcher, "w") as f:
        f.write("#!/bin/bash\n")
        f.write(f"source {str(INSTALL_DIR / VENV_DIR_NAME)}/bin/activate\n")
        f.write(f"python3 {str(INSTALL_DIR / 'hackingtool.py')} \"$@\"\n")
    os.chmod(launcher, 0o755)
    # move to /usr/bin/hackingtool
    if BIN_PATH.exists():
        BIN_PATH.unlink()
    shutil.move(str(launcher), str(BIN_PATH))
    console.print(f"[green][✔] Launcher installed at {str(BIN_PATH)}[/green]")


def final_messages():
    panel = Panel(
        "[bold magenta]Installation complete[/bold magenta]\n\nType [bold cyan]hackingtool[/bold cyan] in terminal to start.",
        border_style="magenta",
    )
    console.print(panel)

def main():
    check_root()
    console.clear()
    colorful_logo()
    choice = choose_distro()
    if choice == 0:
        console.print("[red]Exiting...[/red]")
        sys.exit(0)
    if not check_internet():
        sys.exit(1)

    with Progress(SpinnerColumn(), TextColumn("[progress.description]{task.description}")) as progress:
        progress.add_task(description="Preparing system...", total=None)
        system_update_and_install(choice)

    prepare_install_dir()
    ok = git_clone()
    if not ok:
        sys.exit(1)

    with Progress(SpinnerColumn(), TextColumn("[progress.description]{task.description}")) as progress:
        progress.add_task(description="Setting up virtualenv & requirements...", total=None)
        create_venv_and_install(choice)

    create_launcher()
    final_messages()


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        console.print("\n[red]Installation interrupted by user[/red]")
        sys.exit(1)
    except subprocess.CalledProcessError as e:
        console.print(f"[red]Command failed: {e}[/red]")
        sys.exit(1)