#!/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

#-----------------------------------------------------------------------------------------------
#----------------------------- 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["shows"]=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["shows"]=8000
# Retry logic per profile
declare -A PROFILE_RETRY_COUNT
PROFILE_RETRY_COUNT["arrs_stack"]=3
PROFILE_RETRY_COUNT["movies"]=3
PROFILE_RETRY_COUNT["shows"]=3
# Sleep between retries per profile (seconds)
declare -A PROFILE_SLEEP
PROFILE_SLEEP["arrs_stack"]=300
PROFILE_SLEEP["critical-data"]=300
PROFILE_SLEEP["shows"]=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"]=""
PROFILE_CRITICAL_CONTAINER_NAMES["shows"]=""
# Delayed containers per profile (space-separated strings)
declare -A PROFILE_DELAYED_CONTAINERS
PROFILE_DELAYED_CONTAINERS["arrs_stack"]="Mariadb-Authelia Redis-Authelia Lldap-Gmer4Lfe NginxProxyManager Authelia"
PROFILE_DELAYED_CONTAINERS["critical-data"]="Authelia"
PROFILE_DELAYED_CONTAINERS["shows"]=""
# 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["shows"]=0

#-----------------------------------------------------------------------------------------------
#----------------------------- End of Master Config ---------------------------------------------
#-----------------------------------------------------------------------------------------------