test
This commit is contained in:
+91
-108
@@ -18,129 +18,112 @@ set -e
|
|||||||
# -------------- End Of User Variables, Please adjust in Master.conf as needed -----------------
|
# -------------- End Of User Variables, Please adjust in Master.conf as needed -----------------
|
||||||
# ----------------------------------------------------------------------------------------------
|
# ----------------------------------------------------------------------------------------------
|
||||||
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
BASE_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
|
||||||
# shellcheck source=common.sh
|
# -----------------------------------------------------------------------------
|
||||||
source "$SCRIPT_DIR/common.sh"
|
# Self-healing config loader (OPTION C)
|
||||||
|
# -----------------------------------------------------------------------------
|
||||||
#############################################
|
if [[ -f "$BASE_DIR/../common.sh" ]]; then
|
||||||
# ARG PARSING (UNRAID SAFE)
|
# normal layout: Rsync/ + shared root files
|
||||||
#############################################
|
source "$BASE_DIR/../Master.conf"
|
||||||
|
source "$BASE_DIR/../common.sh"
|
||||||
SOURCE=""
|
elif [[ -f "$BASE_DIR/common.sh" ]]; then
|
||||||
DEST=""
|
# flat layout: everything inside Rsync/
|
||||||
ARGS=()
|
source "$BASE_DIR/Master.conf"
|
||||||
|
source "$BASE_DIR/common.sh"
|
||||||
for arg in "$@"; do
|
else
|
||||||
if [[ "$arg" == -* ]]; then
|
echo "ℹ️ [INFO] Host: $(hostname)"
|
||||||
ARGS+=("$arg")
|
echo "❌ [ERROR] common.sh and Master.conf not found in expected locations"
|
||||||
elif [[ -z "$SOURCE" ]]; then
|
echo "ℹ️ [INFO] BASE_DIR = $BASE_DIR"
|
||||||
SOURCE="$arg"
|
|
||||||
elif [[ -z "$DEST" ]]; then
|
|
||||||
DEST="$arg"
|
|
||||||
else
|
|
||||||
warn "Unknown argument: $arg"
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
#############################################
|
|
||||||
# LOAD CONFIG
|
|
||||||
#############################################
|
|
||||||
|
|
||||||
PROFILE_NAME=${PROFILE_NAME:-"default"}
|
|
||||||
|
|
||||||
MAX_RSYNC_PROCS=${PROFILE_MAX_PROCS:-2}
|
|
||||||
MAX_RSYNC_PROCS=${PROFILE_MAX_PROCS[$PROFILE_NAME]:-$MAX_RSYNC_PROCS}
|
|
||||||
|
|
||||||
#############################################
|
|
||||||
# HEADER
|
|
||||||
#############################################
|
|
||||||
|
|
||||||
info "Host: $(hostname) → ${REMOTE_HOST:-unknown}"
|
|
||||||
info "Sync Starting"
|
|
||||||
info "Source: $SOURCE"
|
|
||||||
info "Remote: ${REMOTE_HOST:-$DEST}"
|
|
||||||
info "Profile: $PROFILE_NAME"
|
|
||||||
|
|
||||||
#############################################
|
|
||||||
# REMOTE RESOLVE
|
|
||||||
#############################################
|
|
||||||
|
|
||||||
info "Resolving remote IP..."
|
|
||||||
REMOTE_IP=$(resolve_remote_ip "$REMOTE_HOST")
|
|
||||||
info "Remote IP: $REMOTE_IP"
|
|
||||||
|
|
||||||
REMOTE_TARGET="${REMOTE_IP}:${DEST}"
|
|
||||||
|
|
||||||
#############################################
|
|
||||||
# CHECK REMOTE
|
|
||||||
#############################################
|
|
||||||
|
|
||||||
info "Checking remote..."
|
|
||||||
if ! ping -c 1 -W 1 "$REMOTE_IP" >/dev/null 2>&1; then
|
|
||||||
error "Remote unreachable"
|
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
ok "Remote reachable"
|
|
||||||
|
|
||||||
#############################################
|
# -----------------------------------------------------------------------------
|
||||||
# CONTAINER STOP (SAFE BARRIER)
|
# Load shared helpers (if your common.sh provides these)
|
||||||
#############################################
|
# -----------------------------------------------------------------------------
|
||||||
|
log_info "Sync script initialized"
|
||||||
|
|
||||||
info "Stopping containers..."
|
# -----------------------------------------------------------------------------
|
||||||
|
# Host resolution (safe fallback)
|
||||||
|
# -----------------------------------------------------------------------------
|
||||||
|
LOCAL_HOST="$(hostname)"
|
||||||
|
REMOTE_HOST="${REMOTE_HOST:-jayred365}"
|
||||||
|
|
||||||
stop_containers || true
|
log_info "Host: ${LOCAL_HOST} → ${REMOTE_HOST}"
|
||||||
|
|
||||||
# ensure docker settle (prevents race condition)
|
# -----------------------------------------------------------------------------
|
||||||
sleep 3
|
# Resolve remote IP
|
||||||
|
# -----------------------------------------------------------------------------
|
||||||
|
log_info "Resolving remote IP..."
|
||||||
|
|
||||||
wait_for_containers_to_stop || true
|
REMOTE_IP="$(getent hosts "$REMOTE_HOST" | awk '{print $1}' || true)"
|
||||||
|
|
||||||
#############################################
|
if [[ -z "$REMOTE_IP" ]]; then
|
||||||
# RSYNC SLOT LIMITER (FIXED)
|
log_error "Failed to resolve remote host: $REMOTE_HOST"
|
||||||
#############################################
|
|
||||||
|
|
||||||
wait_for_rsync() {
|
|
||||||
: "${RETRY_COUNT:=5}"
|
|
||||||
: "${MAX_RSYNC_PROCS:=2}"
|
|
||||||
: "${SLEEP:=2}"
|
|
||||||
|
|
||||||
for _ in $(seq 1 "$RETRY_COUNT"); do
|
|
||||||
|
|
||||||
# ONLY COUNT TOP-LEVEL RSYNC COMMANDS
|
|
||||||
COUNT=$(pgrep -x rsync | wc -l || true)
|
|
||||||
|
|
||||||
if [[ "$COUNT" -ge "$MAX_RSYNC_PROCS" ]]; then
|
|
||||||
warn "Waiting rsync slot ($COUNT/$MAX_RSYNC_PROCS)"
|
|
||||||
sleep "$SLEEP"
|
|
||||||
else
|
|
||||||
return 0
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
error "Too many rsync processes"
|
|
||||||
exit 1
|
exit 1
|
||||||
}
|
fi
|
||||||
|
|
||||||
wait_for_rsync
|
log_info "Remote IP: $REMOTE_IP"
|
||||||
|
|
||||||
#############################################
|
# -----------------------------------------------------------------------------
|
||||||
# RSYNC OPTIONS
|
# Config validation
|
||||||
#############################################
|
# -----------------------------------------------------------------------------
|
||||||
|
PROFILE="${1:-arrs_stack}"
|
||||||
|
|
||||||
info "Building rsync opts..."
|
SOURCE_PATH="${SOURCE_PATH:-/mnt/user/appdata-Failover/Arrs_Stack}"
|
||||||
RSYNC_OPTS=(-aHAX --numeric-ids --delete)
|
REMOTE_PATH="${REMOTE_PATH:-/mnt/user/appdata-Failover/Arrs_Stack}"
|
||||||
|
REMOTE_USER="${REMOTE_USER:-root}"
|
||||||
|
|
||||||
|
log_info "Sync Starting"
|
||||||
|
log_info "Source: $SOURCE_PATH"
|
||||||
|
log_info "Remote: $REMOTE_IP:$REMOTE_PATH"
|
||||||
|
log_info "Profile: $PROFILE"
|
||||||
|
log_info "Dry Run: ${DRY_RUN:-false}"
|
||||||
|
|
||||||
|
# -----------------------------------------------------------------------------
|
||||||
|
# Remote check
|
||||||
|
# -----------------------------------------------------------------------------
|
||||||
|
log_info "Checking remote..."
|
||||||
|
|
||||||
|
if ! ssh -o BatchMode=yes -o ConnectTimeout=5 "${REMOTE_USER}@${REMOTE_IP}" "echo ok" >/dev/null 2>&1; then
|
||||||
|
log_error "Remote unreachable"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_ok "Remote reachable"
|
||||||
|
|
||||||
|
# -----------------------------------------------------------------------------
|
||||||
|
# Stop containers (if defined in common.sh)
|
||||||
|
# -----------------------------------------------------------------------------
|
||||||
|
if declare -f stop_containers >/dev/null 2>&1; then
|
||||||
|
log_info "Stopping containers..."
|
||||||
|
stop_containers "$PROFILE" || true
|
||||||
|
fi
|
||||||
|
|
||||||
|
# -----------------------------------------------------------------------------
|
||||||
|
# RSYNC EXECUTION
|
||||||
|
# -----------------------------------------------------------------------------
|
||||||
|
log_info "Running rsync..."
|
||||||
|
|
||||||
|
RSYNC_OPTS="-aHAX --delete --numeric-ids"
|
||||||
|
|
||||||
if [[ "${DRY_RUN:-false}" == "true" ]]; then
|
if [[ "${DRY_RUN:-false}" == "true" ]]; then
|
||||||
RSYNC_OPTS+=("--dry-run")
|
RSYNC_OPTS="$RSYNC_OPTS -n"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
#############################################
|
rsync $RSYNC_OPTS \
|
||||||
# EXECUTE RSYNC
|
-e "ssh" \
|
||||||
#############################################
|
"$SOURCE_PATH/" \
|
||||||
|
"${REMOTE_USER}@${REMOTE_IP}:${REMOTE_PATH}/"
|
||||||
|
|
||||||
info "Running rsync..."
|
log_ok "Rsync complete"
|
||||||
rsync "${RSYNC_OPTS[@]}" \
|
|
||||||
-e "ssh -o StrictHostKeyChecking=no" \
|
# -----------------------------------------------------------------------------
|
||||||
"$SOURCE" \
|
# Restart containers (if defined)
|
||||||
"root@$REMOTE_TARGET"
|
# -----------------------------------------------------------------------------
|
||||||
|
if declare -f start_containers >/dev/null 2>&1; then
|
||||||
|
log_info "Starting containers..."
|
||||||
|
start_containers "$PROFILE" || true
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_ok "Sync finished successfully"
|
||||||
Reference in New Issue
Block a user