#!/bin/bash
set -euo pipefail

# ============================================
# VPSKIT - Point d'entree unique
# Configure, deploie, surveille et sauvegarde votre VPS
#
# Usage : bash <(curl -sL https://raw.githubusercontent.com/mariusdjen/vpskit/main/vpskit.sh)
# ============================================

# --- Couleurs ---
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
RED='\033[0;31m'
BOLD='\033[1m'
NC='\033[0m'

info()    { echo -e "${BLUE}[INFO] $1${NC}"; }
success() { echo -e "${GREEN}[OK] $1${NC}"; }
warn()    { echo -e "${YELLOW}[WARN] $1${NC}"; }
err()     { echo -e "${RED}[ERR] $1${NC}"; }

# --- URL de base du repo ---
REPO_BASE="https://raw.githubusercontent.com/mariusdjen/vpskit/main"

# --- Chargement de la langue ---
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" 2>/dev/null && pwd)"
if [ -f "${SCRIPT_DIR}/lang.sh" ]; then
    . "${SCRIPT_DIR}/lang.sh"
else
    # Mode curl : telecharger lang.sh dans un fichier temp
    _LANG_TMP=$(mktemp)
    trap 'rm -f "$_LANG_TMP"' EXIT
    # shellcheck disable=SC1090
    curl -fsSL "${REPO_BASE}/lang.sh" -o "$_LANG_TMP" 2>/dev/null && . "$_LANG_TMP"
fi

# --- Banniere ---
show_banner() {
    echo ""
    echo -e "${BOLD}${GREEN}"
    echo "  _   _____  _____ _  _____ _____"
    echo " | | / / _ \\/ ___/ |/ /  _/_  __/"
    echo " | |/ / ___/\\__ \\|   // /  / /   "
    echo " |___/_/   /____/_/\\_/___/ /_/    "
    echo -e "${NC}"
    echo -e "  ${BOLD}${MSG_VPSKIT_TAGLINE:-Set up. Secure. Deploy.}${NC}"
    echo -e "  ${BLUE}https://vpskit.pro${NC}"
    echo ""
}

# --- Menu principal ---
show_menu() {
    echo -e "${BOLD}${MSG_VPSKIT_MENU_TITLE}${NC}"
    echo ""
    echo -e "  ${GREEN}1)${NC} ${MSG_VPSKIT_MENU_1}"
    echo -e "  ${GREEN}2)${NC} ${MSG_VPSKIT_MENU_2}"
    echo -e "  ${GREEN}3)${NC} ${MSG_VPSKIT_MENU_3}"
    echo -e "  ${GREEN}4)${NC} ${MSG_VPSKIT_MENU_4}"
    echo -e "  ${GREEN}5)${NC} ${MSG_VPSKIT_MENU_5}"
    echo -e "  ${GREEN}6)${NC} ${MSG_VPSKIT_MENU_6}"
    echo -e "  ${GREEN}7)${NC} ${MSG_VPSKIT_MENU_7}"
    echo ""
}

# --- Telecharger et executer un script ---
run_script() {
    local script_name="$1"
    local script_url="${REPO_BASE}/${script_name}"
    local tmp_script

    tmp_script=$(mktemp)
    trap 'rm -f "${tmp_script:-}" "${_LANG_TMP:-}"' EXIT

    info "$(printf "$MSG_VPSKIT_DOWNLOADING" "$script_name")"

    if command -v curl &>/dev/null; then
        if ! curl -fsSL "$script_url" -o "$tmp_script"; then
            err "$(printf "$MSG_VPSKIT_DOWNLOAD_FAILED" "$script_name")"
            rm -f "$tmp_script"
            return 1
        fi
    elif command -v wget &>/dev/null; then
        if ! wget -qO "$tmp_script" "$script_url"; then
            err "$(printf "$MSG_VPSKIT_DOWNLOAD_FAILED" "$script_name")"
            rm -f "$tmp_script"
            return 1
        fi
    else
        err "$MSG_VPSKIT_NO_CURL_WGET"
        rm -f "$tmp_script"
        return 1
    fi

    # Verification que le fichier n'est pas vide
    if [ ! -s "$tmp_script" ]; then
        err "$MSG_VPSKIT_EMPTY_FILE"
        rm -f "$tmp_script"
        return 1
    fi

    # Verification syntaxe bash
    if ! bash -n "$tmp_script" 2>/dev/null; then
        err "$MSG_VPSKIT_SYNTAX_ERROR"
        rm -f "$tmp_script"
        return 1
    fi

    success "$MSG_VPSKIT_DOWNLOADED"
    echo ""

    bash "$tmp_script"
    rm -f "$tmp_script"
    if [[ -n "${_LANG_TMP:-}" ]]; then
        trap 'rm -f "${_LANG_TMP}"' EXIT
    else
        trap - EXIT
    fi
}

# --- Detection si le script est execute en local ---
is_local() {
    local script_dir
    script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
    [ -f "${script_dir}/setup.sh" ] && [ -f "${script_dir}/deploy.sh" ]
}

# --- Executer un script local ---
run_local() {
    local script_name="$1"
    local script_dir
    script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
    local script_path="${script_dir}/${script_name}"

    if [ ! -f "$script_path" ]; then
        err "$(printf "$MSG_VPSKIT_SCRIPT_NOT_FOUND" "$script_path")"
        return 1
    fi

    bash "$script_path"
}

# --- Lancer le bon script ---
launch() {
    local script_name="$1"
    if is_local; then
        run_local "$script_name"
    else
        run_script "$script_name"
    fi
}

# --- Main ---
main() {
    show_banner

    # Si des arguments sont passes, les transmettre directement
    if [ $# -gt 0 ]; then
        case "$1" in
            setup)    shift; launch "setup.sh" "$@" ;;
            deploy)   shift; launch "deploy.sh" "$@" ;;
            status)   shift; launch "status.sh" "$@" ;;
            backup)   shift; launch "backup.sh" "$@" ;;
            security) shift; launch "security.sh" "$@" ;;
            settings) shift; launch "settings.sh" "$@" ;;
            *)
                err "$(printf "$MSG_VPSKIT_UNKNOWN_CMD" "$1")"
                echo ""
                echo "  $MSG_VPSKIT_USAGE"
                exit 1
                ;;
        esac
        return
    fi

    # Mode interactif : afficher le menu
    while true; do
        show_menu
        read -p "  $MSG_VPSKIT_CHOICE_PROMPT" choice

        case "$choice" in
            1)
                echo ""
                launch "setup.sh"
                echo ""
                ;;
            2)
                echo ""
                launch "deploy.sh"
                echo ""
                ;;
            3)
                echo ""
                launch "status.sh"
                echo ""
                ;;
            4)
                echo ""
                launch "backup.sh"
                echo ""
                ;;
            5)
                echo ""
                launch "security.sh"
                echo ""
                ;;
            6)
                echo ""
                launch "settings.sh"
                echo ""
                ;;
            7)
                echo ""
                info "$MSG_VPSKIT_BYE"
                exit 0
                ;;
            *)
                echo ""
                warn "$MSG_VPSKIT_INVALID_CHOICE"
                echo ""
                ;;
        esac
    done
}

main "$@"