#!/bin/bash
##################################################################################################
############################## Master Variables for unRAID scripts ################################
##################################################################################################

#-----------------------------------------------------------------------------------------------
#----------------------------- Host and Remote Detection ---------------------------------------
#-----------------------------------------------------------------------------------------------

# Define the hostnames of both servers
HOST1="unRAID-Gmer4Lfe"
HOST2="unRAID-Jayred365"
# Detect local hostname
LOCAL_HOSTNAME="$(hostname)"
if [[ "$LOCAL_HOSTNAME" == "$HOST1" ]]; then
    LOCAL_SERVER_NAME="$HOST1"
    REMOTE_SERVER_NAME="$HOST2"
elif [[ "$LOCAL_HOSTNAME" == "$HOST2" ]]; then
    LOCAL_SERVER_NAME="$HOST2"
    REMOTE_SERVER_NAME="$HOST1"
else
    echo "Error: Local hostname ($LOCAL_HOSTNAME) not recognized."
    exit 1
fi
echo "Local server: $LOCAL_SERVER_NAME"
echo "Remote server: $REMOTE_SERVER_NAME"

#-----------------------------------------------------------------------------------------------
#--------------------------- Remote Server Ip Resolution ----------------------------------------
#-----------------------------------------------------------------------------------------------
# Resolve Tailscale IP dynamically
REMOTE_SERVER=$(tailscale ip -4 "$REMOTE_SERVER_NAME" 2>/dev/null)
if [[ -z "$REMOTE_SERVER" ]]; then
    echo "Error: Could not resolve Tailscale IP for $REMOTE_SERVER_NAME"
    exit 1
fi

#-----------------------------------------------------------------------------------------------
#----------------------------- SSH Key Mapping --------------------------------------------------
#-----------------------------------------------------------------------------------------------

declare -A SSH_KEYS
SSH_KEYS["$HOST1|$HOST2"]="/root/.ssh/Gmer4Lfe-rsync-key"
SSH_KEYS["$HOST2|$HOST1"]="/root/.ssh/Jayred365-rsync-key"

KEY_ID="$LOCAL_SERVER_NAME|$REMOTE_SERVER_NAME"
if [[ -n "${SSH_KEYS[$KEY_ID]}" ]]; then
    SSH_KEY="${SSH_KEYS[$KEY_ID]}"
else
    echo "Error: No SSH key defined for $LOCAL_SERVER_NAME → $REMOTE_SERVER_NAME"
    exit 1
fi
echo "Using SSH key: $SSH_KEY"

#-----------------------------------------------------------------------------------------------
#----------------------------- Global Defaults --------------------------------------------------
#-----------------------------------------------------------------------------------------------
# Used when a profile does not match

# Network speed limit (KB/s)
BW_LIMIT=12500
# Retry logic
RETRY_COUNT=3
# Sleep between retries (seconds)
SLEEP=300
# Max number of concurrently running rsync processes
MAX_RSYNC_PROCS=1
# Container start/stop/restart before and after rsync
CRITICAL_CONTAINER_NAMES=("")
# Containers that need delayed before starting
DELAYED_CONTAINERS=("")
# Delay in seconds between starting containers, useful for things like Authelia
CONTAINER_DELAY=5
# Not include logs during transfer
EXCLUDE_DIRS=("")
#-----------------------------------------------------------------------------------------------
#----------------------------- Profile System ---------------------------------------------------
#-----------------------------------------------------------------------------------------------
# Profiles are inferred from the directory basename (lowercased)

# Max rsync processes per profile
declare -A PROFILE_MAX_PROCS
PROFILE_MAX_PROCS["arrs_stack"]=2
PROFILE_MAX_PROCS["critical-data"]=2
PROFILE_MAX_PROCS["gmer4lfe"]=1
PROFILE_MAX_PROCS["important-data"]=2
PROFILE_MAX_PROCS["emby"]=1
# Bandwidth limits per profile (KB/s, 0 = unlimited)
declare -A PROFILE_BW_LIMIT
PROFILE_BW_LIMIT["arrs_stack"]=5000
PROFILE_BW_LIMIT["critical-data"]=9500
PROFILE_BW_LIMIT["gmer4lfe"]=8000
PROFILE_BW_LIMIT["important-data"]=9500
PROFILE_BW_LIMIT["emby"]=8000
# Retry logic per profile
declare -A PROFILE_RETRY_COUNT
PROFILE_RETRY_COUNT["arrs_stack"]=3
PROFILE_RETRY_COUNT["critical-data"]=3
PROFILE_RETRY_COUNT["gmer4lfe"]=3
PROFILE_RETRY_COUNT["important-data"]=3
PROFILE_RETRY_COUNT["emby"]=3
# Sleep between retries per profile (seconds)
declare -A PROFILE_SLEEP
PROFILE_SLEEP["arrs_stack"]=300
PROFILE_SLEEP["critical-data"]=300
PROFILE_SLEEP["gmer4lfe"]=300
PROFILE_SLEEP["important-data"]=300
PROFILE_SLEEP["emby"]=300
# Containers to stop/start per profile (space-separated strings)
declare -A PROFILE_CRITICAL_CONTAINER_NAMES
PROFILE_CRITICAL_CONTAINER_NAMES["arrs_stack"]="Sonarr Lidarr Readarr Radarr Prowlarr Bazarr Pinchflat"
PROFILE_CRITICAL_CONTAINER_NAMES["critical-data"]="Mariadb-Authelia Redis-Authelia Lldap-Gmer4Lfe NginxProxyManager Authelia"
PROFILE_CRITICAL_CONTAINER_NAMES["gmer4lfe"]="Organizrv2-Gmer4Lfe UptimeKuma-Gmer4Lfe VaultWarden-Gmer4Lfe"
PROFILE_CRITICAL_CONTAINER_NAMES["important-data"]="Postgres-NextCloud NextCloud"
PROFILE_CRITICAL_CONTAINER_NAMES["emby"]=""
# Delayed containers per profile (space-separated strings)
declare -A PROFILE_DELAYED_CONTAINERS
PROFILE_DELAYED_CONTAINERS["arrs_stack"]=""
PROFILE_DELAYED_CONTAINERS["critical-data"]="Authelia"
PROFILE_DELAYED_CONTAINERS["gmer4lfe"]=""
PROFILE_DELAYED_CONTAINERS["important-data"]="NextCloud"
PROFILE_DELAYED_CONTAINERS["emby"]=""
# Delay in seconds before starting delayed containers per profile (seconds)
declare -A PROFILE_CONTAINER_DELAY
PROFILE_CONTAINER_DELAY["arrs_stack"]=5
PROFILE_CONTAINER_DELAY["critical-data"]=10
PROFILE_CONTAINER_DELAY["gmer4lfe"]=5
PROFILE_CONTAINER_DELAY["important-data"]=10
PROFILE_CONTAINER_DELAY["emby"]=5
# Not include logs or other file types during transfer per profile
declare -A PROFILE_EXCLUDE_DIRS
PROFILE_EXCLUDE_DIRS["arrs_stack"]="logs *.tmp"
PROFILE_EXCLUDE_DIRS["critical-data"]="logs *.tmp"
PROFILE_EXCLUDE_DIRS["gmer4lfe"]="logs *.tmp"
PROFILE_EXCLUDE_DIRS["important-data"]="logs *.tmp"
PROFILE_EXCLUDE_DIRS["emby"]="logs *.tmp"
#-----------------------------------------------------------------------------------------------
#----------------------------- End of Master Config ---------------------------------------------
#-----------------------------------------------------------------------------------------------