c
This commit is contained in:
+40
-32
@@ -1,17 +1,14 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
set -e
|
set -e
|
||||||
|
|
||||||
#-----------------------------------------------------------------------------------------------
|
|
||||||
# --------------------------- Core Rsync Script for Unraid -------------------------------------
|
|
||||||
#-----------------------------------------------------------------------------------------------
|
#-----------------------------------------------------------------------------------------------
|
||||||
#----------------- User Variables, Please adjust in Master.conf as needed ----------------------
|
#----------------- User Variables, Please adjust in Master.conf as needed ----------------------
|
||||||
#-----------------------------------------------------------------------------------------------
|
#-----------------------------------------------------------------------------------------------
|
||||||
|
# This is the main Core Rsync script
|
||||||
#
|
#
|
||||||
# User variables can be adjusted in Master.conf
|
# User variables can be adjusted in Master.conf
|
||||||
#
|
|
||||||
# Scripts now just contain runtime logic
|
# Scripts now just contain runtime logic
|
||||||
#
|
#
|
||||||
# This new setup allows for profiles and arguments to use the core script
|
# This allows for profiles and multiple things to use the core script
|
||||||
# Use arguments in unRAID User Plugin for directory
|
# Use arguments in unRAID User Plugin for directory
|
||||||
# Example: /mnt/user/appdata/unraid_scripts/Rsync/rsync.sh /mnt/user/appdata-test/Arrs/
|
# Example: /mnt/user/appdata/unraid_scripts/Rsync/rsync.sh /mnt/user/appdata-test/Arrs/
|
||||||
#
|
#
|
||||||
@@ -19,31 +16,56 @@ set -e
|
|||||||
# If the last name /Arrs/ matches a profile in Master.conf, then profile defaults are used
|
# If the last name /Arrs/ matches a profile in Master.conf, then profile defaults are used
|
||||||
# Otherwise, global configuration in Master.conf is used
|
# Otherwise, global configuration in Master.conf is used
|
||||||
#-----------------------------------------------------------------------------------------------
|
#-----------------------------------------------------------------------------------------------
|
||||||
#---------------End Of User Variables, Please adjust in Master.conf as needed ------------------
|
#---------------End Of User Variables ----------------------------------------------------------
|
||||||
#-----------------------------------------------------------------------------------------------
|
#-----------------------------------------------------------------------------------------------
|
||||||
|
|
||||||
|
# Detect script directory
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
source "$SCRIPT_DIR/../common.sh"
|
|
||||||
|
|
||||||
# Load central config
|
# Load Master.conf
|
||||||
load_master_conf
|
MASTER_CONF="$SCRIPT_DIR/../Master.conf"
|
||||||
|
if [ -f "$MASTER_CONF" ]; then
|
||||||
|
source "$MASTER_CONF"
|
||||||
|
echo "Loaded Master.conf from $MASTER_CONF"
|
||||||
|
else
|
||||||
|
echo "Error: Master.conf not found at $MASTER_CONF"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
# Input Arguments
|
# Input Arguments
|
||||||
DIRECTORY="$1"
|
DIRECTORY="$1"
|
||||||
if [[ -z "$DIRECTORY" ]]; then
|
if [[ -z "$DIRECTORY" ]]; then
|
||||||
echo "Usage: $0 <directory-to-sync> [--dry-run|-n] [VAR=value ...]"
|
echo "Usage: $0 <directory-to-sync> [--dry-run] [VAR=value ...]"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
shift # remove DIRECTORY
|
|
||||||
|
|
||||||
# Parse CLI args (flags + VAR=value)
|
# Optional dry-run
|
||||||
parse_args "$@"
|
DRY_RUN=false
|
||||||
|
shift 1
|
||||||
|
if [[ "$1" == "--dry-run" ]]; then
|
||||||
|
DRY_RUN=true
|
||||||
|
shift 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Optional variable overrides: VAR=value
|
||||||
|
for ARG in "$@"; do
|
||||||
|
if [[ "$ARG" == *=* ]]; then
|
||||||
|
VAR_NAME="${ARG%%=*}"
|
||||||
|
VAR_VALUE="${ARG#*=}"
|
||||||
|
if declare -p "$VAR_NAME" &>/dev/null; then
|
||||||
|
eval "$VAR_NAME=\"$VAR_VALUE\""
|
||||||
|
echo "Overriding $VAR_NAME -> $VAR_VALUE"
|
||||||
|
else
|
||||||
|
echo "Warning: Unknown variable $VAR_NAME, ignoring."
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
# Determine profile
|
# Determine profile
|
||||||
PROFILE_NAME=$(basename "$DIRECTORY" | tr '[:upper:]' '[:lower:]')
|
PROFILE_NAME=$(basename "$DIRECTORY" | tr '[:upper:]' '[:lower:]')
|
||||||
echo "Detected profile: $PROFILE_NAME"
|
echo "Detected profile: $PROFILE_NAME"
|
||||||
|
|
||||||
# Apply profile defaults or fallback to Master.conf
|
# Apply profile defaults or fallback to master.conf
|
||||||
MAX_RSYNC_PROCS=${PROFILE_MAX_PROCS[$PROFILE_NAME]:-$MAX_RSYNC_PROCS}
|
MAX_RSYNC_PROCS=${PROFILE_MAX_PROCS[$PROFILE_NAME]:-$MAX_RSYNC_PROCS}
|
||||||
BW_LIMIT=${PROFILE_BW_LIMIT[$PROFILE_NAME]:-$BW_LIMIT}
|
BW_LIMIT=${PROFILE_BW_LIMIT[$PROFILE_NAME]:-$BW_LIMIT}
|
||||||
CRITICAL_CONTAINER_NAMES=(${PROFILE_CRITICAL_CONTAINER_NAMES[$PROFILE_NAME]})
|
CRITICAL_CONTAINER_NAMES=(${PROFILE_CRITICAL_CONTAINER_NAMES[$PROFILE_NAME]})
|
||||||
@@ -51,15 +73,9 @@ DELAYED_CONTAINERS=(${PROFILE_DELAYED_CONTAINERS[$PROFILE_NAME]})
|
|||||||
CONTAINER_DELAY=${PROFILE_CONTAINER_DELAY[$PROFILE_NAME]:-$CONTAINER_DELAY}
|
CONTAINER_DELAY=${PROFILE_CONTAINER_DELAY[$PROFILE_NAME]:-$CONTAINER_DELAY}
|
||||||
RETRY_COUNT=${PROFILE_RETRY_COUNT[$PROFILE_NAME]:-$RETRY_COUNT}
|
RETRY_COUNT=${PROFILE_RETRY_COUNT[$PROFILE_NAME]:-$RETRY_COUNT}
|
||||||
SLEEP=${PROFILE_SLEEP[$PROFILE_NAME]:-$SLEEP}
|
SLEEP=${PROFILE_SLEEP[$PROFILE_NAME]:-$SLEEP}
|
||||||
|
# Allow multiple patterns per profile
|
||||||
EXCLUDE_DIRS=(${PROFILE_EXCLUDE_DIRS[$PROFILE_NAME]:-${EXCLUDE_DIRS[@]}})
|
EXCLUDE_DIRS=(${PROFILE_EXCLUDE_DIRS[$PROFILE_NAME]:-${EXCLUDE_DIRS[@]}})
|
||||||
|
|
||||||
# Validate numeric variables
|
|
||||||
validate_int MAX_RSYNC_PROCS "$MAX_RSYNC_PROCS"
|
|
||||||
validate_int BW_LIMIT "$BW_LIMIT"
|
|
||||||
validate_int CONTAINER_DELAY "$CONTAINER_DELAY"
|
|
||||||
validate_int RETRY_COUNT "$RETRY_COUNT"
|
|
||||||
validate_int SLEEP "$SLEEP"
|
|
||||||
|
|
||||||
echo "Settings: MAX_RSYNC_PROCS=$MAX_RSYNC_PROCS, BW_LIMIT=$BW_LIMIT, CONTAINERS=${CRITICAL_CONTAINER_NAMES[*]}, EXCLUDES=${EXCLUDE_DIRS[*]}"
|
echo "Settings: MAX_RSYNC_PROCS=$MAX_RSYNC_PROCS, BW_LIMIT=$BW_LIMIT, CONTAINERS=${CRITICAL_CONTAINER_NAMES[*]}, EXCLUDES=${EXCLUDE_DIRS[*]}"
|
||||||
|
|
||||||
# Dependency check
|
# Dependency check
|
||||||
@@ -99,11 +115,7 @@ stop_containers() {
|
|||||||
if [ "$STATUS" == "true" ]; then
|
if [ "$STATUS" == "true" ]; then
|
||||||
echo "Stopping $container"
|
echo "Stopping $container"
|
||||||
RUNNING_CONTAINERS+=("$container")
|
RUNNING_CONTAINERS+=("$container")
|
||||||
if [ "$DRY_RUN" = true ]; then
|
ssh -i "$SSH_KEY" root@"$REMOTE_SERVER" "docker stop $container"
|
||||||
echo "[DRY-RUN] Would stop container $container"
|
|
||||||
else
|
|
||||||
ssh -i "$SSH_KEY" root@"$REMOTE_SERVER" "docker stop $container"
|
|
||||||
fi
|
|
||||||
fi
|
fi
|
||||||
done
|
done
|
||||||
}
|
}
|
||||||
@@ -113,13 +125,9 @@ start_containers() {
|
|||||||
for container in "${RUNNING_CONTAINERS[@]}"; do
|
for container in "${RUNNING_CONTAINERS[@]}"; do
|
||||||
if [[ " ${DELAYED_CONTAINERS[*]} " == *" $container "* ]]; then
|
if [[ " ${DELAYED_CONTAINERS[*]} " == *" $container "* ]]; then
|
||||||
echo "Delaying start of $container by ${CONTAINER_DELAY}s..."
|
echo "Delaying start of $container by ${CONTAINER_DELAY}s..."
|
||||||
[ "$DRY_RUN" = false ] && sleep "$CONTAINER_DELAY"
|
sleep "$CONTAINER_DELAY"
|
||||||
fi
|
|
||||||
if [ "$DRY_RUN" = true ]; then
|
|
||||||
echo "[DRY-RUN] Would start container $container"
|
|
||||||
else
|
|
||||||
ssh -i "$SSH_KEY" root@"$REMOTE_SERVER" "docker start $container"
|
|
||||||
fi
|
fi
|
||||||
|
ssh -i "$SSH_KEY" root@"$REMOTE_SERVER" "docker start $container"
|
||||||
done
|
done
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,13 +1,14 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
#-----------------------------------------------------------------------------------------------
|
#-----------------------------------------------------------------------------------------------
|
||||||
#--------------------- Common Shared Functions -------------------------------------------------
|
# Common Shared Functions for Unraid Scripts
|
||||||
#-----------------------------------------------------------------------------------------------
|
#-----------------------------------------------------------------------------------------------
|
||||||
|
|
||||||
# Load Master.conf
|
# Load Master.conf relative to calling script
|
||||||
load_master_conf() {
|
load_master_conf() {
|
||||||
local script_dir
|
local script_dir
|
||||||
script_dir="$(cd "$(dirname "${BASH_SOURCE[1]}")" && pwd)"
|
script_dir="$(cd "$(dirname "${BASH_SOURCE[1]}")" && pwd)"
|
||||||
|
|
||||||
local master_conf="$script_dir/../Master.conf"
|
local master_conf="$script_dir/../Master.conf"
|
||||||
|
|
||||||
if [ -f "$master_conf" ]; then
|
if [ -f "$master_conf" ]; then
|
||||||
@@ -19,7 +20,7 @@ load_master_conf() {
|
|||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# Parse CLI arguments
|
# Parse CLI arguments (flags + VAR=value overrides)
|
||||||
parse_args() {
|
parse_args() {
|
||||||
DRY_RUN=${DRY_RUN:-false}
|
DRY_RUN=${DRY_RUN:-false}
|
||||||
|
|
||||||
@@ -29,35 +30,44 @@ parse_args() {
|
|||||||
local VAR_VALUE="${ARG#*=}"
|
local VAR_VALUE="${ARG#*=}"
|
||||||
|
|
||||||
if declare -p "$VAR_NAME" &>/dev/null; then
|
if declare -p "$VAR_NAME" &>/dev/null; then
|
||||||
printf -v "$VAR_NAME" '%s' "$VAR_VALUE"
|
if [[ "$VAR_NAME" =~ ^[A-Za-z_][A-Za-z0-9_]*$ ]]; then
|
||||||
echo "Overriding $VAR_NAME -> $VAR_VALUE"
|
printf -v "$VAR_NAME" '%s' "$VAR_VALUE"
|
||||||
|
echo "Overriding $VAR_NAME -> $VAR_VALUE"
|
||||||
|
else
|
||||||
|
echo "Warning: Invalid variable name $VAR_NAME"
|
||||||
|
fi
|
||||||
else
|
else
|
||||||
echo "Warning: Unknown variable $VAR_NAME, ignoring."
|
echo "Warning: Unknown variable $VAR_NAME, ignoring."
|
||||||
fi
|
fi
|
||||||
else
|
else
|
||||||
case "$ARG" in
|
case "$ARG" in
|
||||||
--dry-run|-n) DRY_RUN=true ;;
|
--dry-run|-n)
|
||||||
|
DRY_RUN=true
|
||||||
|
;;
|
||||||
--help|-h)
|
--help|-h)
|
||||||
echo "Usage: script [--dry-run|-n] [VAR=value ...]"
|
echo "Usage: script [--dry-run|-n] [VAR=value ...]"
|
||||||
exit 0 ;;
|
exit 0
|
||||||
|
;;
|
||||||
*)
|
*)
|
||||||
echo "Warning: Unknown argument $ARG" ;;
|
echo "Warning: Unknown argument $ARG"
|
||||||
|
;;
|
||||||
esac
|
esac
|
||||||
fi
|
fi
|
||||||
done
|
done
|
||||||
}
|
}
|
||||||
|
|
||||||
# Validate integer
|
# Validate integer (non-negative)
|
||||||
validate_int() {
|
validate_int() {
|
||||||
local name="$1"
|
local name="$1"
|
||||||
local value="$2"
|
local value="$2"
|
||||||
|
|
||||||
if ! [[ "${value:-}" =~ ^[0-9]+$ ]]; then
|
if ! [[ "${value:-}" =~ ^[0-9]+$ ]]; then
|
||||||
echo "Error: $name must be a non-negative integer."
|
echo "Error: $name must be a non-negative integer."
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# Require variable
|
# Optional: require variable to be set
|
||||||
require_var() {
|
require_var() {
|
||||||
local var="$1"
|
local var="$1"
|
||||||
if [[ -z "${!var:-}" ]]; then
|
if [[ -z "${!var:-}" ]]; then
|
||||||
@@ -65,4 +75,3 @@ require_var() {
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user