run.sh

#!/bin/bash

set -Eeuo pipefail

# Networking & Cybersecurity Automation Toolkit
# Main control script
# /run.sh

# Root check
if [[ $EUID -ne 0 ]]; then
    echo ""
    echo "  [!] This script must be run with sudo or as root."
    echo "  [>] Run: sudo $0"
    echo ""
    exit 1
fi

# Project root
export PROJECT_ROOT="$(cd "$(dirname "$0")" && pwd)"

# Source dependencies
source "$PROJECT_ROOT/config/settings.conf"
source "$PROJECT_ROOT/lib/colors.sh"
source "$PROJECT_ROOT/lib/functions.sh"
source "$PROJECT_ROOT/modules/run_modules.sh"
source "$PROJECT_ROOT/network_lab/network_lab.sh"
source "$PROJECT_ROOT/dashboard/start_dashboard.sh"

# Directory setup
MODULES_DIR="$PROJECT_ROOT/modules"
LOG_DIR="$PROJECT_ROOT/logs"
OUTPUT_DIR="$PROJECT_ROOT/output"
DASHBOARD_DIR="$PROJECT_ROOT/dashboard"

mkdir -p "$LOG_DIR" "$OUTPUT_DIR"
touch "$LOG_DIR/main.log"

# MAIN MENU
show_main_menu() {
    local W=50
    local border
    border=$(printf '%*s' "$W" '' | tr ' ' '=')

    echo -e "${BORDER}${border}${NC}"
    printf "${BORDER}|${NC}  ${TITLE}%-$((W-4))s${NC}  ${BORDER}|${NC}\n" "MAIN MENU"
    echo -e "${BORDER}${border}${NC}"
    echo
    echo -e "  ${LABEL}SETUP${NC}"
    echo -e "  ${GREEN}  1.${NC}  Install / Verify Networking Tools"
    echo
    echo -e "  ${LABEL}SECURITY${NC}"
    echo -e "  ${GREEN}  2.${NC}  Run Security Modules"
    echo
    echo -e "  ${LABEL}MONITORING${NC}"
    echo -e "  ${GREEN}  3.${NC}  View Dashboard"
    echo -e "  ${GREEN}  7.${NC}  Stop Dashboard"
    echo
    echo -e "  ${LABEL}SYSTEM${NC}"
    echo -e "  ${GREEN}  4.${NC}  Clean Logs & Output"
    echo -e "  ${GREEN}  5.${NC}  System Information"
    echo
    echo -e "  ${LABEL}NETWORK LAB${NC}"
    echo -e "  ${GREEN}  6.${NC}  Network Lab"
    echo
    echo -e "  ${RED}  0.${NC}  Exit"
    echo
    echo -e "${BORDER}${border}${NC}"
}

install_dependencies() {
    clear
    show_banner
    echo -e "  ${LABEL}Dependency Installer${NC}"
    echo
    if [[ -f "$PROJECT_ROOT/install.sh" ]]; then
        echo -e "  ${AMBER}[+] Running dependency installer...${NC}"
        echo
        bash "$PROJECT_ROOT/install.sh" || log_error "Installer failed"
        echo
        log_success "Dependency installation completed."
    else
        log_error "install.sh not found in project root."
    fi
    echo
    read -rp "$(echo -e "  ${MUTED}Press Enter to continue...${NC}  ")"
}

# CLEAN DATA
clean_data() {
    clear
    show_banner
    echo -e "  ${FAILURE}[!] WARNING: This will permanently delete all logs and output files!"
    echo
    read -rp "$(echo -e "  ${WARNING}[?] Are you sure? [yes/no]: ${NC}")" confirm

    if [[ "$confirm" == "yes" ]]; then
        rm -rf -- "${LOG_DIR:?}/"* "${OUTPUT_DIR:?}/"*
        log_success "Cleaned successfully."
    else
        log_info "Operation cancelled."
    fi
    echo
    read -rp "$(echo -e "  ${MUTED}Press Enter to continue...${NC}  ")"
}

# SYSTEM INFO
show_system_info() {
    clear
    show_banner

    echo -e "  ${AMBER}Host${NC}"
    kv "  Operating System" "$(detect_os)"
    kv "  Hostname" "$(hostname)"
    kv "  User" "$(whoami)"
    kv "  Date" "$(date '+%Y-%m-%d %H:%M:%S')"
    kv "  Uptime" "$(uptime -p 2>/dev/null || uptime)"
    echo
    echo -e "  ${AMBER}Toolkit${NC}"

    kv "  Log files" "$(find "$LOG_DIR" -type f 2>/dev/null -printf '.' | wc -c)"
    kv "  Output files" "$(find "$OUTPUT_DIR" -type f 2>/dev/null | wc -l)"
    kv "  Modules" "$(find "$MODULES_DIR" -name '*.sh' ! -name 'run_modules.sh' | wc -l)"

    echo

    if [[ -f "$PID_FILE" ]]; then
        pid=$(cat "$PID_FILE")
        if kill -0 "$pid" 2>/dev/null; then
            kv "  Dashboard" "${SUCCESS}Running${NC} (PID: ${pid})"
        else
            kv "  Dashboard" "${FAILURE}Not running (stale PID)"
        fi
    else
        kv "  Dashboard" "${MUTED}Not running${NC}"
    fi
    echo
    read -rp "$(echo -e "  ${MUTED}Press Enter to continue...${NC}  ")"
}

# CLEANUP
cleanup() {
    echo
    log_info "Toolkit shutting down..."
}

trap cleanup EXIT INT TERM

# MAIN LOOP
main() {
    while true; do
        clear
        show_banner
        show_main_menu
        read -rp "$(echo -e "  ${PROMPT}[?] Enter your choice: ${NC}")" choice
        echo
        case $choice in
            1) install_dependencies ;;
            2)
                while true; do
                    show_modules_menu
                    read -rp "$(echo -e "  ${PROMPT}[?] Enter your choice: ${NC}")" module_choice

                    [[ "$module_choice" == "0" ]] && break

                    run_modules "$module_choice"
                done
                ;;
            3) start_dashboard_main ;;
            4) clean_data ;;
            5) show_system_info ;;
            6) network_lab ;;
            7) stop_dashboard ;;
            0)
                clear
                show_banner
                echo -e "  ${SUCCESS}[+] Thank you for using the Networking & Cybersecurity Toolkit!"
                echo
                exit 0
                ;;
            *)
                log_error "Invalid choice. Please try again."
                sleep 1
                ;;
        esac
    done
}

main