#!/bin/bash

# Configuración
REPO_URL="https://raw.githubusercontent.com/MacRimi/ProxMenux/main"
BASE_DIR="/usr/local/share/proxmenux"
LANG_DIR="$BASE_DIR/lang"
LOCAL_VERSION_FILE="$BASE_DIR/version.txt"
LANGUAGE_FILE="/root/.proxmenux_language"

# Colores para salida
YW="\033[33m"; GN="\033[1;92m"; RD="\033[01;31m"; CL="\033[m"
msg_info() { echo -e " ${YW}[INFO] $1${CL}"; }
msg_ok() { echo -e " ${GN}[OK] $1${CL}"; }
msg_error() { echo -e " ${RD}[ERROR] $1${CL}"; }

# Crear directorios necesarios
mkdir -p "$LANG_DIR"

# Descargar archivo de idioma por defecto (español) para mensajes iniciales
if [ ! -f "$LANG_DIR/es.lang" ]; then
    if ! wget -qO "$LANG_DIR/es.lang" "$REPO_URL/lang/es.lang"; then
        msg_error "Error al descargar el archivo de idioma inicial."
        exit 1
    fi
fi

# Cargar mensajes iniciales
source "$LANG_DIR/es.lang"

# Cargar o seleccionar idioma
load_language() {
    if [ ! -f "$LANGUAGE_FILE" ]; then
        select_language_first_time
    else
        LANGUAGE=$(cat "$LANGUAGE_FILE")
        LANG_FILE="$LANG_DIR/$LANGUAGE.lang"
        if [ ! -f "$LANG_FILE" ]; then
            msg_info "$LANG_DOWNLOAD"
            if ! wget -qO "$LANG_FILE" "$REPO_URL/lang/$LANGUAGE.lang"; then
                msg_error "$LANG_DOWNLOAD_ERROR"
                exit 1
            fi
        fi
        source "$LANG_FILE"
        msg_info "$LANG_LOADED $LANGUAGE"
    fi
}

# Función para la primera selección de idioma
select_language_first_time() {
    LANGUAGE=$(whiptail --title "$INITIAL_LANG_SELECT" --menu "$INITIAL_LANG_PROMPT" 15 60 2 \
        "es" "Español" \
        "en" "English" 3>&1 1>&2 2>&3)

    if [ -z "$LANGUAGE" ]; then
        msg_error "$INITIAL_LANG_ERROR"
        exit 1
    fi

    LANG_FILE="$LANG_DIR/$LANGUAGE.lang"
    if [ ! -f "$LANG_FILE" ]; then
        if ! wget -qO "$LANG_FILE" "$REPO_URL/lang/$LANGUAGE.lang"; then
            msg_error "$LANG_DOWNLOAD_ERROR"
            exit 1
        fi
    fi

    echo "$LANGUAGE" > "$LANGUAGE_FILE"
    source "$LANG_FILE"
    msg_ok "$LANG_SUCCESS $LANGUAGE"
}

# Función para cambiar idioma desde el menú
select_language() {
    LANGUAGE=$(whiptail --title "$LANG_SELECT" --menu "$LANG_PROMPT" 15 60 2 \
        "es" "Español" \
        "en" "English" 3>&1 1>&2 2>&3)

    if [ -z "$LANGUAGE" ]; then
        msg_error "$LANG_ERROR"
        return
    fi

    LANG_FILE="$LANG_DIR/$LANGUAGE.lang"
    if [ ! -f "$LANG_FILE" ]; then
        msg_info "$LANG_DOWNLOAD"
        if ! wget -qO "$LANG_FILE" "$REPO_URL/lang/$LANGUAGE.lang"; then
            msg_error "$LANG_DOWNLOAD_ERROR"
            return
        fi
    fi

    echo "$LANGUAGE" > "$LANGUAGE_FILE"
    msg_ok "$LANG_SUCCESS $LANGUAGE"
    exec "$0"
}


# Función para verificar y realizar actualizaciones
check_updates() {
    msg_info "$UPDATE_CHECKING"
    
    # Descargar versión remota
    REMOTE_VERSION=$(curl -s "$REPO_URL/version.txt?$(date +%s)" | tr -d '\r\n')
    
    if [ -z "$REMOTE_VERSION" ]; then
        msg_error "$UPDATE_ERROR_REMOTE"
        return 1
    fi

    # Leer versión local o establecer como la primera instalación
    LOCAL_VERSION=$(cat "$LOCAL_VERSION_FILE" 2>/dev/null || echo "0.0.0")

    if [ "$LOCAL_VERSION" != "$REMOTE_VERSION" ]; then
        msg_info "$(printf "$UPDATE_NEW_AVAILABLE" "$REMOTE_VERSION" "$LOCAL_VERSION")"
        if whiptail --title "$UPDATE_TITLE" --yesno "$(printf "$UPDATE_PROMPT" "$REMOTE_VERSION")" 10 60; then
            perform_update "$REMOTE_VERSION"
        else
            msg_info "$UPDATE_POSTPONED"
        fi
    else
        msg_info "$(printf "$UPDATE_CURRENT" "$LOCAL_VERSION")"
    fi
}

# Función para realizar la actualización
perform_update() {
    REMOTE_VERSION=$1
    msg_info "$(printf "$UPDATE_PROCESS" "$REMOTE_VERSION")"

    # Descargar el nuevo menú
    if curl -sLo /usr/local/bin/menu.sh "$REPO_URL/menu.sh?$(date +%s)"; then
        chmod +x /usr/local/bin/menu.sh
        echo "$REMOTE_VERSION" > "$LOCAL_VERSION_FILE"
        msg_ok "$(printf "$UPDATE_COMPLETE" "$REMOTE_VERSION")"

        # Reiniciar el menú con la nueva versión
        exec /usr/local/bin/menu.sh
    else
        msg_error "$UPDATE_ERROR_DOWNLOAD"
    fi
}


# Función para desinstalar ProxMenu
uninstall_proxmenu() {
    if whiptail --title "$UNINSTALL_TITLE" --yesno "$UNINSTALL_CONFIRM" 10 60; then
        msg_info "$UNINSTALL_PROCESS"
        rm -rf "$BASE_DIR"
        rm -f "/usr/local/bin/menu.sh"
        rm -f "$LANGUAGE_FILE"
        msg_ok "$UNINSTALL_COMPLETE"
        exit 0
    fi
}

# Función para mostrar información de versión
show_version_info() {
    local version=$(cat "$LOCAL_VERSION_FILE" 2>/dev/null || echo "1.0.0")
    whiptail --title "$VERSION_TITLE" --msgbox "$(printf "$VERSION_INFO" "$version")" 12 60
}

# Mostrar menú de configuración
show_config_menu() {
    while true; do
        OPTION=$(whiptail --title "$CONFIG_TITLE" --menu "$SELECT_OPTION" 15 60 3 \
            "1" "$LANG_OPTION" \
            "2" "$VERSION_OPTION" \
            "3" "$UNINSTALL_OPTION" 3>&1 1>&2 2>&3)

        case $OPTION in
            1)
                select_language
                ;;
            2)
                show_version_info
                ;;
            3)
                uninstall_proxmenu
                ;;
            *)
                return
                ;;
        esac
    done
}


# Mostrar menú principal
show_menu() {
    while true; do
        OPTION=$(whiptail --title "$MAIN_MENU_TITLE" --menu "$SELECT_OPTION" 15 60 4 \
            "1" "$OPTION_1" \
            "2" "$OPTION_2" \
            "3" "$OPTION_3" \
            "4" "$EXIT_MENU" 3>&1 1>&2 2>&3)

        case $OPTION in
            1)
                msg_info "$SCRIPT_RUNNING"
                if wget -qO- "$REPO_URL/scripts/igpu_tpu.sh" | bash; then
                    msg_ok "$SCRIPT_SUCCESS"
                else
                    msg_error "$SCRIPT_ERROR"
                fi
                ;;
            2)
                msg_info "$NETWORK_REPAIR_RUNNING"
                if wget -qO- "$REPO_URL/scripts/repair_network.sh" | bash; then
                    msg_ok "$NETWORK_REPAIR_SUCCESS"
                else
                    msg_error "$NETWORK_REPAIR_ERROR"
                fi
                ;;
            3)
                show_config_menu
                ;;
            4)
                clear
                msg_ok "$EXIT_MESSAGE"
                exit 0
                ;;
            *)
                msg_error "$INVALID_OPTION"
                sleep 2
                ;;
        esac
    done
}

# Verificar dependencias
if ! command -v whiptail &> /dev/null; then
    msg_info "$DEPS_INSTALLING"
    if apt-get update && apt-get install -y whiptail; then
        msg_ok "$DEPS_SUCCESS"
    else
        msg_error "$DEPS_ERROR"
        exit 1
    fi
fi

# Flujo principal
load_language
check_updates
show_menu