425 lines
12 KiB
Plaintext
425 lines
12 KiB
Plaintext
# Copyright (c) 2021-2025 community-scripts ORG
|
|
# License: MIT | https://raw.githubusercontent.com/community-scripts/ProxmoxVED/main/LICENSE
|
|
|
|
set -euo pipefail
|
|
SPINNER_PID=""
|
|
SPINNER_ACTIVE=0
|
|
SPINNER_MSG=""
|
|
declare -A MSG_INFO_SHOWN
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Loads core utility groups once (colors, formatting, icons, defaults).
|
|
# ------------------------------------------------------------------------------
|
|
|
|
[[ -n "${_CORE_FUNC_LOADED:-}" ]] && return
|
|
_CORE_FUNC_LOADED=1
|
|
|
|
load_functions() {
|
|
[[ -n "${__FUNCTIONS_LOADED:-}" ]] && return
|
|
__FUNCTIONS_LOADED=1
|
|
color
|
|
formatting
|
|
icons
|
|
default_vars
|
|
set_std_mode
|
|
# add more
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Sets ANSI color codes used for styled terminal output.
|
|
# ------------------------------------------------------------------------------
|
|
color() {
|
|
YW=$(echo "\033[33m")
|
|
YWB=$'\e[93m'
|
|
BL=$(echo "\033[36m")
|
|
RD=$(echo "\033[01;31m")
|
|
BGN=$(echo "\033[4;92m")
|
|
GN=$(echo "\033[1;92m")
|
|
DGN=$(echo "\033[32m")
|
|
CL=$(echo "\033[m")
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Defines formatting helpers like tab, bold, and line reset sequences.
|
|
# ------------------------------------------------------------------------------
|
|
formatting() {
|
|
BFR="\\r\\033[K"
|
|
BOLD=$(echo "\033[1m")
|
|
HOLD=" "
|
|
TAB=" "
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Sets symbolic icons used throughout user feedback and prompts.
|
|
# ------------------------------------------------------------------------------
|
|
icons() {
|
|
CM="${TAB}✔️${TAB}"
|
|
CROSS="${TAB}✖️${TAB}"
|
|
INFO="${TAB}💡${TAB}${CL}"
|
|
OS="${TAB}🖥️${TAB}${CL}"
|
|
OSVERSION="${TAB}🌟${TAB}${CL}"
|
|
CONTAINERTYPE="${TAB}📦${TAB}${CL}"
|
|
DISKSIZE="${TAB}💾${TAB}${CL}"
|
|
CPUCORE="${TAB}🧠${TAB}${CL}"
|
|
RAMSIZE="${TAB}🛠️${TAB}${CL}"
|
|
SEARCH="${TAB}🔍${TAB}${CL}"
|
|
VERBOSE_CROPPED="🔍${TAB}"
|
|
VERIFYPW="${TAB}🔐${TAB}${CL}"
|
|
CONTAINERID="${TAB}🆔${TAB}${CL}"
|
|
HOSTNAME="${TAB}🏠${TAB}${CL}"
|
|
BRIDGE="${TAB}🌉${TAB}${CL}"
|
|
NETWORK="${TAB}📡${TAB}${CL}"
|
|
GATEWAY="${TAB}🌐${TAB}${CL}"
|
|
DISABLEIPV6="${TAB}🚫${TAB}${CL}"
|
|
DEFAULT="${TAB}⚙️${TAB}${CL}"
|
|
MACADDRESS="${TAB}🔗${TAB}${CL}"
|
|
VLANTAG="${TAB}🏷️${TAB}${CL}"
|
|
ROOTSSH="${TAB}🔑${TAB}${CL}"
|
|
CREATING="${TAB}🚀${TAB}${CL}"
|
|
ADVANCED="${TAB}🧩${TAB}${CL}"
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Sets default retry and wait variables used for system actions.
|
|
# ------------------------------------------------------------------------------
|
|
default_vars() {
|
|
RETRY_NUM=10
|
|
RETRY_EVERY=3
|
|
i=$RETRY_NUM
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Sets default verbose mode for script and os execution.
|
|
# ------------------------------------------------------------------------------
|
|
set_std_mode() {
|
|
if [ "${VERBOSE:-no}" = "yes" ]; then
|
|
STD=""
|
|
else
|
|
STD="silent"
|
|
fi
|
|
}
|
|
|
|
# Silent execution function
|
|
silent() {
|
|
"$@" >/dev/null 2>&1
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Performs a curl request with retry logic and inline feedback.
|
|
# ------------------------------------------------------------------------------
|
|
|
|
run_curl() {
|
|
if [ "$VERBOSE" = "no" ]; then
|
|
$STD curl "$@"
|
|
else
|
|
curl "$@"
|
|
fi
|
|
}
|
|
|
|
curl_handler() {
|
|
set +e
|
|
trap 'set -e' RETURN
|
|
local args=()
|
|
local url=""
|
|
local max_retries=3
|
|
local delay=2
|
|
local attempt=1
|
|
local exit_code
|
|
local has_output_file=false
|
|
local result=""
|
|
|
|
# Parse arguments
|
|
for arg in "$@"; do
|
|
if [[ "$arg" != -* && -z "$url" ]]; then
|
|
url="$arg"
|
|
fi
|
|
[[ "$arg" == "-o" || "$arg" == --output ]] && has_output_file=true
|
|
args+=("$arg")
|
|
done
|
|
|
|
if [[ -z "$url" ]]; then
|
|
msg_error "No valid URL or option entered for curl_handler"
|
|
return 1
|
|
fi
|
|
|
|
$STD msg_info "Fetching: $url"
|
|
|
|
while [[ $attempt -le $max_retries ]]; do
|
|
if $has_output_file; then
|
|
$STD run_curl "${args[@]}"
|
|
exit_code=$?
|
|
else
|
|
result=$(run_curl "${args[@]}")
|
|
exit_code=$?
|
|
fi
|
|
|
|
if [[ $exit_code -eq 0 ]]; then
|
|
$STD msg_ok "Fetched: $url"
|
|
$has_output_file || printf '%s' "$result"
|
|
return 0
|
|
fi
|
|
|
|
if ((attempt >= max_retries)); then
|
|
# Read error log if it exists
|
|
if [ -s /tmp/curl_error.log ]; then
|
|
local curl_stderr
|
|
curl_stderr=$(</tmp/curl_error.log)
|
|
rm -f /tmp/curl_error.log
|
|
fi
|
|
__curl_err_handler "$exit_code" "$url" "${curl_stderr:-}"
|
|
exit
|
|
fi
|
|
|
|
$STD printf "\r\033[K${INFO}${YW}Retry $attempt/$max_retries in ${delay}s...${CL}" >&2
|
|
sleep "$delay"
|
|
((attempt++))
|
|
done
|
|
set -e
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Handles specific curl error codes and displays descriptive messages.
|
|
# ------------------------------------------------------------------------------
|
|
__curl_err_handler() {
|
|
local exit_code="$1"
|
|
local target="$2"
|
|
local curl_msg="$3"
|
|
|
|
case $exit_code in
|
|
1) msg_error "Unsupported protocol: $target" ;;
|
|
2) msg_error "Curl init failed: $target" ;;
|
|
3) msg_error "Malformed URL: $target" ;;
|
|
5) msg_error "Proxy resolution failed: $target" ;;
|
|
6) msg_error "Host resolution failed: $target" ;;
|
|
7) msg_error "Connection failed: $target" ;;
|
|
9) msg_error "Access denied: $target" ;;
|
|
18) msg_error "Partial file transfer: $target" ;;
|
|
22) msg_error "HTTP error (e.g. 400/404): $target" ;;
|
|
23) msg_error "Write error on local system: $target" ;;
|
|
26) msg_error "Read error from local file: $target" ;;
|
|
28) msg_error "Timeout: $target" ;;
|
|
35) msg_error "SSL connect error: $target" ;;
|
|
47) msg_error "Too many redirects: $target" ;;
|
|
51) msg_error "SSL cert verify failed: $target" ;;
|
|
52) msg_error "Empty server response: $target" ;;
|
|
55) msg_error "Send error: $target" ;;
|
|
56) msg_error "Receive error: $target" ;;
|
|
60) msg_error "SSL CA not trusted: $target" ;;
|
|
67) msg_error "Login denied by server: $target" ;;
|
|
78) msg_error "Remote file not found (404): $target" ;;
|
|
*) msg_error "Curl failed with code $exit_code: $target" ;;
|
|
esac
|
|
|
|
[[ -n "$curl_msg" ]] && printf "%s\n" "$curl_msg" >&2
|
|
exit 1
|
|
}
|
|
|
|
### dev spinner ###
|
|
SPINNER_ACTIVE=0
|
|
SPINNER_PID=""
|
|
SPINNER_MSG=""
|
|
declare -A MSG_INFO_SHOWN=()
|
|
|
|
# Trap cleanup on various signals
|
|
trap 'cleanup_spinner' EXIT INT TERM HUP
|
|
|
|
# Cleans up spinner process on exit
|
|
cleanup_spinner() {
|
|
stop_spinner
|
|
# Additional cleanup if needed
|
|
}
|
|
|
|
start_spinner() {
|
|
local msg="${1:-Processing...}"
|
|
local frames=(⠋ ⠙ ⠹ ⠸ ⠼ ⠴ ⠦ ⠧ ⠇ ⠏)
|
|
local spin_i=0
|
|
local interval=0.1
|
|
|
|
# Set message and clear current line
|
|
SPINNER_MSG="$msg"
|
|
printf "\r\e[2K" >&2
|
|
|
|
# Stop any existing spinner
|
|
stop_spinner
|
|
|
|
# Set active flag
|
|
SPINNER_ACTIVE=1
|
|
|
|
# Start spinner in background
|
|
{
|
|
while [[ "$SPINNER_ACTIVE" -eq 1 ]]; do
|
|
printf "\r\e[2K%s %b" "${TAB}${frames[spin_i]}${TAB}" "${YW}${SPINNER_MSG}${CL}" >&2
|
|
spin_i=$(((spin_i + 1) % ${#frames[@]}))
|
|
sleep "$interval"
|
|
done
|
|
} &
|
|
|
|
SPINNER_PID=$!
|
|
|
|
# Disown to prevent getting "Terminated" messages
|
|
disown "$SPINNER_PID" 2>/dev/null || true
|
|
}
|
|
|
|
stop_spinner() {
|
|
# Check if spinner is active and PID exists
|
|
if [[ "$SPINNER_ACTIVE" -eq 1 ]] && [[ -n "${SPINNER_PID}" ]]; then
|
|
SPINNER_ACTIVE=0
|
|
|
|
if kill -0 "$SPINNER_PID" 2>/dev/null; then
|
|
kill "$SPINNER_PID" 2>/dev/null
|
|
# Give it a moment to terminate
|
|
sleep 0.1
|
|
# Force kill if still running
|
|
if kill -0 "$SPINNER_PID" 2>/dev/null; then
|
|
kill -9 "$SPINNER_PID" 2>/dev/null
|
|
fi
|
|
# Wait for process but ignore errors
|
|
wait "$SPINNER_PID" 2>/dev/null || true
|
|
fi
|
|
|
|
# Clear spinner line
|
|
printf "\r\e[2K" >&2
|
|
SPINNER_PID=""
|
|
fi
|
|
}
|
|
|
|
spinner_guard() {
|
|
# Safely stop spinner if it's running
|
|
if [[ "$SPINNER_ACTIVE" -eq 1 ]] && [[ -n "${SPINNER_PID}" ]]; then
|
|
stop_spinner
|
|
fi
|
|
}
|
|
|
|
msg_info() {
|
|
local msg="${1:-Information message}"
|
|
|
|
# Only show each message once unless reset
|
|
if [[ -n "${MSG_INFO_SHOWN["$msg"]+x}" ]]; then
|
|
return
|
|
fi
|
|
MSG_INFO_SHOWN["$msg"]=1
|
|
|
|
spinner_guard
|
|
start_spinner "$msg"
|
|
}
|
|
|
|
msg_ok() {
|
|
local msg="${1:-Operation completed successfully}"
|
|
stop_spinner
|
|
printf "\r\e[2K%s %b\n" "${CM}" "${GN}${msg}${CL}" >&2
|
|
|
|
# Remove from shown messages to allow it to be shown again
|
|
unset MSG_INFO_SHOWN["$msg"]
|
|
}
|
|
|
|
msg_error() {
|
|
local msg="${1:-An error occurred}"
|
|
stop_spinner
|
|
printf "\r\e[2K%s %b\n" "${CROSS}" "${RD}${msg}${CL}" >&2
|
|
}
|
|
|
|
msg_warn() {
|
|
local msg="${1:-Warning}"
|
|
stop_spinner
|
|
printf "\r\e[2K%s %b\n" "${INFO}" "${YWB}${msg}" >&2
|
|
|
|
# Remove from shown messages to allow it to be shown again
|
|
unset MSG_INFO_SHOWN["$msg"]
|
|
}
|
|
|
|
# Helper function to display a message with custom symbol and color
|
|
msg_custom() {
|
|
local symbol="${1:-*}"
|
|
local color="${2:-$CL}"
|
|
local msg="${3:-Custom message}"
|
|
stop_spinner
|
|
printf "\r\e[2K%s %b\n" "$symbol" "${color}${msg}${CL}" >&2
|
|
}
|
|
|
|
# # ------------------------------------------------------------------------------
|
|
# # Spinner trap: ensures spinner is stopped on termination signals.
|
|
# # ------------------------------------------------------------------------------
|
|
# trap 'stop_spinner' EXIT INT TERM HUP
|
|
|
|
# # ------------------------------------------------------------------------------
|
|
# # Starts a spinner animation for ongoing async operations.
|
|
# # ------------------------------------------------------------------------------
|
|
# start_spinner() {
|
|
# local msg="$1"
|
|
# local frames=(⠋ ⠙ ⠹ ⠸ ⠼ ⠴ ⠦ ⠧ ⠇ ⠏)
|
|
# local spin_i=0
|
|
# local interval=0.1
|
|
|
|
# SPINNER_MSG="$msg"
|
|
# printf "\r\e[2K" >&2
|
|
|
|
# {
|
|
# while [[ "$SPINNER_ACTIVE" -eq 1 ]]; do
|
|
# printf "\r\e[2K%s %b" "${TAB}${frames[spin_i]}${TAB}" "${YW}${SPINNER_MSG}${CL}" >&2
|
|
# spin_i=$(((spin_i + 1) % ${#frames[@]}))
|
|
# sleep "$interval"
|
|
# done
|
|
# } &
|
|
|
|
# SPINNER_PID=$!
|
|
# disown "$SPINNER_PID"
|
|
# }
|
|
|
|
# # ------------------------------------------------------------------------------
|
|
# # Stops the spinner animation and resets its state.
|
|
# # ------------------------------------------------------------------------------
|
|
# stop_spinner() {
|
|
# if [[ -n "${SPINNER_PID:-}" ]] && kill -0 "$SPINNER_PID" 2>/dev/null; then
|
|
# kill "$SPINNER_PID" 2>/dev/null
|
|
# sleep 0.1
|
|
# kill -0 "$SPINNER_PID" 2>/dev/null && kill -9 "$SPINNER_PID" 2>/dev/null
|
|
# wait "$SPINNER_PID" 2>/dev/null || true
|
|
# fi
|
|
# SPINNER_ACTIVE=0
|
|
# unset SPINNER_PID
|
|
# }
|
|
|
|
# # ------------------------------------------------------------------------------
|
|
# # Stops the spinner if active, used to avoid multiple spinners.
|
|
# # ------------------------------------------------------------------------------
|
|
# spinner_guard() {
|
|
# if [[ "$SPINNER_ACTIVE" -eq 1 ]] && [[ -n "${SPINNER_PID:-}" ]]; then
|
|
# kill "$SPINNER_PID" 2>/dev/null
|
|
# wait "$SPINNER_PID" 2>/dev/null || true
|
|
# SPINNER_ACTIVE=0
|
|
# unset SPINNER_PID
|
|
# fi
|
|
# }
|
|
|
|
# # ------------------------------------------------------------------------------
|
|
# # Displays an informational spinner once per message.
|
|
# # ------------------------------------------------------------------------------
|
|
# msg_info() {
|
|
# local msg="$1"
|
|
# [[ -n "${MSG_INFO_SHOWN["$msg"]+x}" ]] && return
|
|
# MSG_INFO_SHOWN["$msg"]=1
|
|
|
|
# spinner_guard
|
|
# SPINNER_ACTIVE=1
|
|
# start_spinner "$msg"
|
|
# }
|
|
|
|
# # ------------------------------------------------------------------------------
|
|
# # Displays a success message and stops spinner.
|
|
# # ------------------------------------------------------------------------------
|
|
# msg_ok() {
|
|
# local msg="$1"
|
|
# stop_spinner
|
|
# printf "\r\e[2K%s %b\n" "${CM}" "${GN}${msg}${CL}" >&2
|
|
# unset MSG_INFO_SHOWN["$msg"]
|
|
# }
|
|
|
|
# # ------------------------------------------------------------------------------
|
|
# # Displays an error message and stops spinner.
|
|
# # ------------------------------------------------------------------------------
|
|
# msg_error() {
|
|
# stop_spinner
|
|
# local msg="$1"
|
|
# printf "\r\e[2K%s %b\n" "${CROSS}" "${RD}${msg}${CL}" >&2
|
|
# }
|