mirror of
https://github.com/community-scripts/ProxmoxVE.git
synced 2025-07-08 06:57:37 +00:00
411 lines
11 KiB
Bash
411 lines
11 KiB
Bash
# Copyright (c) 2021-2025 community-scripts ORG
|
||
# License: MIT | https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/LICENSE
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# 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
|
||
}
|
||
|
||
# ============================================================================
|
||
# Error & Signal Handling – robust, universal, subshell-safe
|
||
# ============================================================================
|
||
|
||
_tool_error_hint() {
|
||
local cmd="$1"
|
||
local code="$2"
|
||
case "$cmd" in
|
||
curl)
|
||
case "$code" in
|
||
6) echo "Curl: Could not resolve host (DNS problem)" ;;
|
||
7) echo "Curl: Failed to connect to host (connection refused)" ;;
|
||
22) echo "Curl: HTTP error (404/403 etc)" ;;
|
||
28) echo "Curl: Operation timeout" ;;
|
||
*) echo "Curl: Unknown error ($code)" ;;
|
||
esac
|
||
;;
|
||
wget)
|
||
echo "Wget failed – URL unreachable or permission denied"
|
||
;;
|
||
systemctl)
|
||
echo "Systemd unit failure – check service name and permissions"
|
||
;;
|
||
jq)
|
||
echo "jq parse error – malformed JSON or missing key"
|
||
;;
|
||
mariadb | mysql)
|
||
echo "MySQL/MariaDB command failed – check credentials or DB"
|
||
;;
|
||
unzip)
|
||
echo "unzip failed – corrupt file or missing permission"
|
||
;;
|
||
tar)
|
||
echo "tar failed – invalid format or missing binary"
|
||
;;
|
||
node | npm | pnpm | yarn)
|
||
echo "Node tool failed – check version compatibility or package.json"
|
||
;;
|
||
*) echo "" ;;
|
||
esac
|
||
}
|
||
|
||
catch_errors() {
|
||
set -Eeuo pipefail
|
||
trap 'error_handler $LINENO "$BASH_COMMAND"' ERR
|
||
}
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# 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")
|
||
}
|
||
|
||
# Special for spinner and colorized output via printf
|
||
color_spinner() {
|
||
CS_YW=$'\033[33m'
|
||
CS_YWB=$'\033[93m'
|
||
CS_CL=$'\033[m'
|
||
}
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# Defines formatting helpers like tab, bold, and line reset sequences.
|
||
# ------------------------------------------------------------------------------
|
||
formatting() {
|
||
BFR="\\r\\033[K"
|
||
BOLD=$(echo "\033[1m")
|
||
HOLD=" "
|
||
TAB=" "
|
||
TAB3=" "
|
||
}
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# Sets symbolic icons used throughout user feedback and prompts.
|
||
# ------------------------------------------------------------------------------
|
||
icons() {
|
||
CM="${TAB}✔️${TAB}"
|
||
CROSS="${TAB}✖️${TAB}"
|
||
DNSOK="✔️ "
|
||
DNSFAIL="${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}"
|
||
FUSE="${TAB}🗂️${TAB}${CL}"
|
||
HOURGLASS="${TAB}⏳${TAB}"
|
||
|
||
}
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# Sets default retry and wait variables used for system actions.
|
||
# ------------------------------------------------------------------------------
|
||
default_vars() {
|
||
RETRY_NUM=10
|
||
RETRY_EVERY=3
|
||
i=$RETRY_NUM
|
||
#[[ "${VAR_OS:-}" == "unknown" ]]
|
||
}
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# 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
|
||
}
|
||
|
||
# Function to download & save header files
|
||
get_header() {
|
||
local app_name=$(echo "${APP,,}" | tr -d ' ')
|
||
local app_type=${APP_TYPE:-ct}
|
||
local header_url="https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/${app_type}/headers/${app_name}"
|
||
local local_header_path="/usr/local/community-scripts/headers/${app_type}/${app_name}"
|
||
|
||
mkdir -p "$(dirname "$local_header_path")"
|
||
|
||
if [ ! -s "$local_header_path" ]; then
|
||
if ! curl -fsSL "$header_url" -o "$local_header_path"; then
|
||
return 1
|
||
fi
|
||
fi
|
||
|
||
cat "$local_header_path" 2>/dev/null || true
|
||
}
|
||
|
||
header_info() {
|
||
local app_name=$(echo "${APP,,}" | tr -d ' ')
|
||
local header_content
|
||
|
||
header_content=$(get_header "$app_name") || header_content=""
|
||
|
||
clear
|
||
local term_width
|
||
term_width=$(tput cols 2>/dev/null || echo 120)
|
||
|
||
if [ -n "$header_content" ]; then
|
||
echo "$header_content"
|
||
fi
|
||
}
|
||
|
||
ensure_tput() {
|
||
if ! command -v tput >/dev/null 2>&1; then
|
||
if grep -qi 'alpine' /etc/os-release; then
|
||
apk add --no-cache ncurses >/dev/null 2>&1
|
||
elif command -v apt-get >/dev/null 2>&1; then
|
||
apt-get update -qq >/dev/null
|
||
apt-get install -y -qq ncurses-bin >/dev/null 2>&1
|
||
fi
|
||
fi
|
||
}
|
||
|
||
is_alpine() {
|
||
local os_id="${var_os:-${PCT_OSTYPE:-}}"
|
||
|
||
if [[ -z "$os_id" && -f /etc/os-release ]]; then
|
||
os_id="$(
|
||
. /etc/os-release 2>/dev/null
|
||
echo "${ID:-}"
|
||
)"
|
||
fi
|
||
|
||
[[ "$os_id" == "alpine" ]]
|
||
}
|
||
|
||
is_verbose_mode() {
|
||
local verbose="${VERBOSE:-${var_verbose:-no}}"
|
||
local tty_status
|
||
if [[ -t 2 ]]; then
|
||
tty_status="interactive"
|
||
else
|
||
tty_status="not-a-tty"
|
||
fi
|
||
[[ "$verbose" != "no" || ! -t 2 ]]
|
||
}
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# 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
|
||
}
|
||
|
||
fatal() {
|
||
msg_error "$1"
|
||
kill -INT $$
|
||
}
|
||
|
||
spinner() {
|
||
local chars=(⠋ ⠙ ⠹ ⠸ ⠼ ⠴ ⠦ ⠧ ⠇ ⠏)
|
||
local i=0
|
||
while true; do
|
||
local index=$((i++ % ${#chars[@]}))
|
||
printf "\r\033[2K%s %b" "${CS_YWB}${chars[$index]}${CS_CL}" "${CS_YWB}${SPINNER_MSG:-}${CS_CL}"
|
||
sleep 0.1
|
||
done
|
||
}
|
||
|
||
clear_line() {
|
||
tput cr 2>/dev/null || echo -en "\r"
|
||
tput el 2>/dev/null || echo -en "\033[K"
|
||
}
|
||
|
||
stop_spinner() {
|
||
local pid="${SPINNER_PID:-}"
|
||
[[ -z "$pid" && -f /tmp/.spinner.pid ]] && pid=$(</tmp/.spinner.pid)
|
||
|
||
if [[ -n "$pid" && "$pid" =~ ^[0-9]+$ ]]; then
|
||
if kill "$pid" 2>/dev/null; then
|
||
sleep 0.05
|
||
kill -9 "$pid" 2>/dev/null || true
|
||
wait "$pid" 2>/dev/null || true
|
||
fi
|
||
rm -f /tmp/.spinner.pid
|
||
fi
|
||
|
||
unset SPINNER_PID SPINNER_MSG
|
||
stty sane 2>/dev/null || true
|
||
}
|
||
|
||
msg_info() {
|
||
local msg="$1"
|
||
[[ -z "$msg" ]] && return
|
||
|
||
if ! declare -p MSG_INFO_SHOWN &>/dev/null || ! declare -A MSG_INFO_SHOWN &>/dev/null; then
|
||
declare -gA MSG_INFO_SHOWN=()
|
||
fi
|
||
[[ -n "${MSG_INFO_SHOWN["$msg"]+x}" ]] && return
|
||
MSG_INFO_SHOWN["$msg"]=1
|
||
|
||
stop_spinner
|
||
SPINNER_MSG="$msg"
|
||
|
||
if is_verbose_mode || is_alpine; then
|
||
local HOURGLASS="${TAB}⏳${TAB}"
|
||
printf "\r\e[2K%s %b" "$HOURGLASS" "${YW}${msg}${CL}" >&2
|
||
return
|
||
fi
|
||
|
||
color_spinner
|
||
spinner &
|
||
SPINNER_PID=$!
|
||
echo "$SPINNER_PID" >/tmp/.spinner.pid
|
||
disown "$SPINNER_PID" 2>/dev/null || true
|
||
}
|
||
|
||
msg_ok() {
|
||
local msg="$1"
|
||
[[ -z "$msg" ]] && return
|
||
stop_spinner
|
||
clear_line
|
||
printf "%s %b\n" "$CM" "${GN}${msg}${CL}" >&2
|
||
unset MSG_INFO_SHOWN["$msg"]
|
||
}
|
||
|
||
msg_error() {
|
||
stop_spinner
|
||
local msg="$1"
|
||
echo -e "${BFR:-} ${CROSS:-✖️} ${RD}${msg}${CL}"
|
||
}
|
||
|
||
msg_warn() {
|
||
stop_spinner
|
||
local msg="$1"
|
||
echo -e "${BFR:-} ${INFO:-ℹ️} ${YWB}${msg}${CL}"
|
||
}
|
||
|
||
msg_custom() {
|
||
local symbol="${1:-"[*]"}"
|
||
local color="${2:-"\e[36m"}"
|
||
local msg="${3:-}"
|
||
[[ -z "$msg" ]] && return
|
||
stop_spinner
|
||
echo -e "${BFR:-} ${symbol} ${color}${msg}${CL:-\e[0m}"
|
||
printf "\r\033[K\e[?25h\n"
|
||
}
|
||
|
||
run_container_safe() {
|
||
local ct="$1"
|
||
shift
|
||
local cmd="$*"
|
||
|
||
lxc-attach -n "$ct" -- bash -euo pipefail -c "
|
||
trap 'echo Aborted in container; exit 130' SIGINT SIGTERM
|
||
$cmd
|
||
" || __handle_general_error "lxc-attach to CT $ct"
|
||
}
|
||
|
||
check_or_create_swap() {
|
||
msg_info "Checking for active swap"
|
||
|
||
if swapon --noheadings --show | grep -q 'swap'; then
|
||
msg_ok "Swap is active"
|
||
return 0
|
||
fi
|
||
|
||
msg_error "No active swap detected"
|
||
|
||
read -p "Do you want to create a swap file? [y/N]: " create_swap
|
||
create_swap="${create_swap,,}" # to lowercase
|
||
|
||
if [[ "$create_swap" != "y" && "$create_swap" != "yes" ]]; then
|
||
msg_info "Skipping swap file creation"
|
||
return 1
|
||
fi
|
||
|
||
read -p "Enter swap size in MB (e.g., 2048 for 2GB): " swap_size_mb
|
||
if ! [[ "$swap_size_mb" =~ ^[0-9]+$ ]]; then
|
||
msg_error "Invalid size input. Aborting."
|
||
return 1
|
||
fi
|
||
|
||
local swap_file="/swapfile"
|
||
|
||
msg_info "Creating ${swap_size_mb}MB swap file at $swap_file"
|
||
if dd if=/dev/zero of="$swap_file" bs=1M count="$swap_size_mb" status=progress &&
|
||
chmod 600 "$swap_file" &&
|
||
mkswap "$swap_file" &&
|
||
swapon "$swap_file"; then
|
||
msg_ok "Swap file created and activated successfully"
|
||
else
|
||
msg_error "Failed to create or activate swap"
|
||
return 1
|
||
fi
|
||
}
|
||
|
||
trap 'stop_spinner' EXIT INT TERM
|