imput handling fix
This commit is contained in:
@@ -15,30 +15,21 @@ ICON_SYNC="🔄"
|
||||
ICON_GEAR="⚙️"
|
||||
|
||||
# ----------------------------- Output Helpers -----------------------------
|
||||
info() {
|
||||
echo "$ICON_INFO [INFO] $*"
|
||||
}
|
||||
|
||||
warn() {
|
||||
echo "$ICON_WARN [WARN] $*"
|
||||
}
|
||||
|
||||
error() {
|
||||
echo "$ICON_ERROR [ERROR] $*"
|
||||
}
|
||||
|
||||
success() {
|
||||
echo "$ICON_SUCCESS [OK] $*"
|
||||
}
|
||||
info() { echo "$ICON_INFO [INFO] $*"; }
|
||||
warn() { echo "$ICON_WARN [WARN] $*"; }
|
||||
error() { echo "$ICON_ERROR [ERROR] $*"; }
|
||||
success() { echo "$ICON_SUCCESS [OK] $*"; }
|
||||
|
||||
log() {
|
||||
[[ "$ENABLE_LOGGING" == true ]] && echo "[LOG] $*"
|
||||
[[ "${ENABLE_LOGGING:-false}" == true ]] && echo "[LOG] $*"
|
||||
}
|
||||
|
||||
# ----------------------------- Argument Parsing -----------------------------
|
||||
parse_args() {
|
||||
DRY_RUN=${DRY_RUN:-false}
|
||||
ENABLE_LOGGING=${ENABLE_LOGGING:-false}
|
||||
DRY_RUN=${DRY_RUN:-false}
|
||||
|
||||
local CLEAN_ARGS=()
|
||||
|
||||
for ARG in "$@"; do
|
||||
if [[ "$ARG" == *=* ]]; then
|
||||
@@ -47,21 +38,15 @@ parse_args() {
|
||||
|
||||
case "$VAR_NAME" in
|
||||
LOG)
|
||||
if [[ "$VAR_VALUE" == "true" ]]; then
|
||||
ENABLE_LOGGING=true
|
||||
info "Verbose logging enabled"
|
||||
elif [[ "$VAR_VALUE" == "false" ]]; then
|
||||
ENABLE_LOGGING=false
|
||||
else
|
||||
warn "Invalid LOG value: $VAR_VALUE"
|
||||
fi
|
||||
[[ "$VAR_VALUE" == "true" ]] && ENABLE_LOGGING=true
|
||||
[[ "$VAR_VALUE" == "false" ]] && ENABLE_LOGGING=false
|
||||
;;
|
||||
*)
|
||||
if declare -p "$VAR_NAME" &>/dev/null; then
|
||||
printf -v "$VAR_NAME" '%s' "$VAR_VALUE"
|
||||
log "Overriding $VAR_NAME -> $VAR_VALUE"
|
||||
log "Set $VAR_NAME=$VAR_VALUE"
|
||||
else
|
||||
warn "Unknown variable $VAR_NAME"
|
||||
warn "Unknown variable: $VAR_NAME"
|
||||
fi
|
||||
;;
|
||||
esac
|
||||
@@ -70,11 +55,9 @@ parse_args() {
|
||||
case "$ARG" in
|
||||
--dry-run|-n)
|
||||
DRY_RUN=true
|
||||
info "Dry-run enabled"
|
||||
;;
|
||||
--log)
|
||||
ENABLE_LOGGING=true
|
||||
info "Logging enabled"
|
||||
;;
|
||||
--no-log)
|
||||
ENABLE_LOGGING=false
|
||||
@@ -83,31 +66,29 @@ parse_args() {
|
||||
SHOW_STATUS=true
|
||||
;;
|
||||
--help|-h)
|
||||
echo "Usage: script <dir> [--dry-run|-n] [--log] [VAR=value ...]"
|
||||
echo "Usage: script <dir> [--dry-run] [--log] [VAR=value]"
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
warn "Unknown argument: $ARG"
|
||||
CLEAN_ARGS+=("$ARG")
|
||||
;;
|
||||
esac
|
||||
fi
|
||||
done
|
||||
|
||||
PARSED_ARGS=("${CLEAN_ARGS[@]}")
|
||||
}
|
||||
|
||||
# ----------------------------- Validation -----------------------------
|
||||
require_var() {
|
||||
local var="$1"
|
||||
[[ -z "${!var:-}" ]] && error "Missing required variable: $var" && exit 1
|
||||
[[ -z "${!1:-}" ]] && error "Missing required: $1" && exit 1
|
||||
}
|
||||
|
||||
validate_int() {
|
||||
local name="$1"
|
||||
local value="$2"
|
||||
|
||||
[[ ! "${value:-}" =~ ^[0-9]+$ ]] && error "$name must be integer" && exit 1
|
||||
[[ ! "${2:-}" =~ ^[0-9]+$ ]] && error "$1 must be integer" && exit 1
|
||||
}
|
||||
|
||||
# ----------------------------- HOST DETECTION -----------------------------
|
||||
# ----------------------------- Host Detection -----------------------------
|
||||
detect_hosts() {
|
||||
LOCAL_HOSTNAME="$(hostname)"
|
||||
|
||||
@@ -119,6 +100,7 @@ detect_hosts() {
|
||||
REMOTE_SERVER_NAME="$HOST1"
|
||||
else
|
||||
error "Unknown host: $LOCAL_HOSTNAME"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
declare -A SSH_KEYS
|
||||
@@ -127,24 +109,22 @@ detect_hosts() {
|
||||
|
||||
SSH_KEY="${SSH_KEYS[$LOCAL_SERVER_NAME|$REMOTE_SERVER_NAME]}"
|
||||
|
||||
[[ -z "$SSH_KEY" ]] && error "Missing SSH key mapping"
|
||||
[[ -z "$SSH_KEY" ]] && error "Missing SSH key mapping" && exit 1
|
||||
|
||||
info "Host: $LOCAL_SERVER_NAME → $REMOTE_SERVER_NAME"
|
||||
}
|
||||
|
||||
# ----------------------------- REMOTE -----------------------------
|
||||
# ----------------------------- 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"
|
||||
[[ -z "$REMOTE_SERVER" ]] && error "Failed to resolve remote IP" && exit 1
|
||||
|
||||
info "Remote IP: $REMOTE_SERVER"
|
||||
}
|
||||
|
||||
check_remote_online() {
|
||||
ping -c 1 "$REMOTE_SERVER" &>/dev/null
|
||||
}
|
||||
|
||||
# ----------------------------- CONTAINERS -----------------------------
|
||||
# ----------------------------- Containers -----------------------------
|
||||
declare -a RUNNING_CONTAINERS=()
|
||||
|
||||
stop_containers() {
|
||||
@@ -176,27 +156,23 @@ start_containers() {
|
||||
done
|
||||
}
|
||||
|
||||
# ----------------------------- RSYNC -----------------------------
|
||||
# ----------------------------- RSYNC OPTIONS -----------------------------
|
||||
get_rsync_opts() {
|
||||
if [[ -n "${PROFILE_RSYNC_OPTS[$PROFILE_NAME]:-}" ]]; then
|
||||
read -r -a RSYNC_OPTS <<< "${PROFILE_RSYNC_OPTS[$PROFILE_NAME]}"
|
||||
log "Profile rsync opts loaded"
|
||||
else
|
||||
RSYNC_OPTS=("${DEFAULT_RSYNC_OPTS[@]}")
|
||||
log "Default rsync opts loaded"
|
||||
fi
|
||||
}
|
||||
|
||||
# ----------------------------- STATUS -----------------------------
|
||||
show_status() {
|
||||
echo
|
||||
echo "================= STATUS ================="
|
||||
echo "Local: $LOCAL_SERVER_NAME"
|
||||
echo "Remote: $REMOTE_SERVER_NAME"
|
||||
echo "IP: $REMOTE_SERVER"
|
||||
echo "Profile: $PROFILE_NAME"
|
||||
echo "Dry Run: $DRY_RUN"
|
||||
echo "Logging: $ENABLE_LOGGING"
|
||||
echo "Containers:${CRITICAL_CONTAINER_NAMES[*]:-(none)}"
|
||||
echo "========================================="
|
||||
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 "=================="
|
||||
}
|
||||
Reference in New Issue
Block a user