diff --git a/common.sh b/common.sh
index 2d84f5d..94542f9 100644
--- a/common.sh
+++ b/common.sh
@@ -1,6 +1,9 @@
#!/bin/bash
+# -----------------------------------------------------------------------------------------------
+# ----------------- UNRAID OPS COMMON LIBRARY (STABLE FRAMEWORK v1) -----------------------
+# -----------------------------------------------------------------------------------------------
-# ---------------- ICONS ----------------
+# ICONS Profiles
ICON_INFO="ℹ️"
ICON_WARN="⚠️"
ICON_ERROR="❌"
@@ -11,13 +14,17 @@ ICON_RETRY="🔁"
ICON_SYNC="🔄"
ICON_GEAR="⚙️"
+# OUTPUT
info() { echo "$ICON_INFO [INFO] $*"; }
warn() { echo "$ICON_WARN [WARN] $*"; }
error() { echo "$ICON_ERROR [ERROR] $*"; }
success() { echo "$ICON_SUCCESS [OK] $*"; }
-log() { [[ "${ENABLE_LOGGING:-false}" == true ]] && echo "[LOG] $*"; }
-# ---------------- ARG PARSER ----------------
+log() {
+ [[ "${ENABLE_LOGGING:-false}" == true ]] && echo "[LOG] $*"
+}
+
+# ARG PARSER
parse_args() {
ENABLE_LOGGING=${ENABLE_LOGGING:-false}
DRY_RUN=${DRY_RUN:-false}
@@ -26,23 +33,48 @@ parse_args() {
CLEAN_ARGS=()
for ARG in "$@"; do
- case "$ARG" in
- --dry-run|-n) DRY_RUN=true ;;
- --log) ENABLE_LOGGING=true ;;
- --no-log) ENABLE_LOGGING=false ;;
- --status|--summary) SHOW_STATUS=true ;;
- *) CLEAN_ARGS+=("$ARG") ;;
- esac
+ if [[ "$ARG" == *=* ]]; then
+ VAR="${ARG%%=*}"
+ VAL="${ARG#*=}"
+
+ case "$VAR" in
+ LOG)
+ [[ "$VAL" == "true" ]] && ENABLE_LOGGING=true
+ [[ "$VAL" == "false" ]] && ENABLE_LOGGING=false
+ ;;
+ *)
+ if declare -p "$VAR" &>/dev/null; then
+ printf -v "$VAR" '%s' "$VAL"
+ log "Set $VAR=$VAL"
+ else
+ warn "Unknown variable: $VAR"
+ fi
+ ;;
+ esac
+ else
+ case "$ARG" in
+ --dry-run|-n) DRY_RUN=true ;;
+ --log) ENABLE_LOGGING=true ;;
+ --no-log) ENABLE_LOGGING=false ;;
+ --status|--summary) SHOW_STATUS=true ;;
+ --help|-h)
+ echo "Usage: script
[--dry-run] [--log]"
+ exit 0
+ ;;
+ *) CLEAN_ARGS+=("$ARG") ;;
+ esac
+ fi
done
PARSED_ARGS=("${CLEAN_ARGS[@]}")
}
+# VALIDATION
require_var() {
[[ -z "${!1:-}" ]] && error "Missing required: $1" && exit 1
}
-# ---------------- HOST DETECTION ----------------
+# HOST DETECTION
detect_hosts() {
LOCAL_HOSTNAME="$(hostname)"
@@ -68,24 +100,27 @@ detect_hosts() {
info "Host: $LOCAL_SERVER_NAME → $REMOTE_SERVER_NAME"
}
+# REMOTE
resolve_remote_ip() {
+ log "Resolving remote IP..."
REMOTE_SERVER=$(tailscale ip -4 "$REMOTE_SERVER_NAME" 2>/dev/null)
+
[[ -z "$REMOTE_SERVER" ]] && error "Failed to resolve remote IP" && exit 1
+
info "Remote IP: $REMOTE_SERVER"
}
-# ---------------- CONTAINERS (FIXED) ----------------
+# CONTAINERS
RUNNING_CONTAINERS=()
stop_containers() {
- [[ -z "${CRITICAL_CONTAINER_NAMES:-}" ]] && return
-
- read -r -a containers <<< "${CRITICAL_CONTAINER_NAMES}"
+ [[ ${#CRITICAL_CONTAINER_NAMES[@]} -eq 0 ]] && return
info "Stopping containers..."
+ RUNNING_CONTAINERS=()
- for c in "${containers[@]}"; do
- [[ -z "$c" ]] && continue
+ for c in "${CRITICAL_CONTAINER_NAMES[@]}"; do
+ info "Checking container: $c"
STATUS=$(ssh -i "$SSH_KEY" root@"$REMOTE_SERVER" \
"docker inspect -f '{{.State.Running}}' $c 2>/dev/null" || echo "false")
@@ -100,7 +135,7 @@ stop_containers() {
error "Failed to stop $c"
fi
else
- success "$c already stopped"
+ log "$c already stopped"
fi
done
}
@@ -111,6 +146,8 @@ start_containers() {
info "Starting containers..."
for c in "${RUNNING_CONTAINERS[@]}"; do
+ info "Starting $c"
+
if ssh -i "$SSH_KEY" root@"$REMOTE_SERVER" "docker start $c" >/dev/null; then
success "$c started"
else
@@ -119,14 +156,19 @@ start_containers() {
done
}
-# ---------------- RSYNC LIMIT ----------------
+# RSYNC LIMITER
wait_for_rsync() {
: "${RETRY_COUNT:=5}"
: "${MAX_RSYNC_PROCS:=2}"
: "${SLEEP:=2}"
for i in $(seq 1 "$RETRY_COUNT"); do
- COUNT=$(pgrep -fc "rsync" || true)
+
+ COUNT=$(ps -eo pid,cmd | grep "[r]sync .*${REMOTE_SERVER:-}" | wc -l || true)
+
+ if [[ -z "$REMOTE_SERVER" ]]; then
+ COUNT=$(pgrep -x rsync | wc -l || true)
+ fi
if [[ "$COUNT" -ge "$MAX_RSYNC_PROCS" ]]; then
warn "Waiting rsync slot ($COUNT/$MAX_RSYNC_PROCS)"
@@ -140,7 +182,7 @@ wait_for_rsync() {
exit 1
}
-# ---------------- RSYNC OPTS ----------------
+# RSYNC OPTIONS
get_rsync_opts() {
if [[ -n "${PROFILE_RSYNC_OPTS[$PROFILE_NAME]:-}" ]]; then
read -r -a RSYNC_OPTS <<< "${PROFILE_RSYNC_OPTS[$PROFILE_NAME]}"
@@ -149,11 +191,14 @@ get_rsync_opts() {
fi
}
+# STATUS
show_status() {
echo "===== STATUS ====="
echo "Local: $LOCAL_SERVER_NAME"
echo "Remote: $REMOTE_SERVER_NAME"
echo "IP: $REMOTE_SERVER"
echo "Profile: $PROFILE_NAME"
+ echo "DryRun: $DRY_RUN"
+ echo "Logging: $ENABLE_LOGGING"
echo "=================="
}
\ No newline at end of file