#!/bin/bash
cd /docker/keepalived

# 1. SOURCING
# Force export of all variables in .env so sed can access them
if [ -f .env ]; then
    echo "Sourcing variables from .env..."
    set -a
    source .env
    set +a
else
    echo "ERROR: .env file not found in $(pwd)"
    exit 1
fi

# 2. QUOTE CLEANING
# Remove literal double quotes from the start/end of the string if they exist
SLOT_DEFINITIONS=$(echo "$SLOT_DEFINITIONS" | sed 's/^"//;s/"$//')

# 3. PATHS
CONFIG_DIR="./config"
CHECKS_DIR="./checks"
HEADER_TPL="./keepalived_header.tpl"
SLOT_TPL="./keepalived_slot.tpl"
CONFIG_FILE="${CONFIG_DIR}/keepalived.conf"

mkdir -p "$CONFIG_DIR" "$CHECKS_DIR"

# 4. BUILD HEADER
if [ -z "$LXC_NUM" ]; then echo "ERROR: LXC_NUM not found in env"; exit 1; fi

cp "$HEADER_TPL" "$CONFIG_FILE"
sed -i "s/{{LXC_NUM}}/${LXC_NUM}/g" "$CONFIG_FILE"

# 5. PARSE SLOTS
# Using echo | while to ensure the cleaned string is parsed line-by-line
echo "$SLOT_DEFINITIONS" | while IFS=',' read -r s_num s_name s_dep s_state s_prio s_peer s_vip junk
do
    [[ -z "$s_num" ]] && continue
    [[ -z "$s_vip" ]] && continue

    echo "Processing Slot $s_num: $s_name"
    TEMP_SLOT="/tmp/slot_${s_num}.conf"
    cp "$SLOT_TPL" "$TEMP_SLOT"
    
    # Password logic (8 char limit)
    FULL_PASS="${BASE_PASSWORD}-${s_num}"
    SHORT_PASS="${FULL_PASS:0:8}"

    sed -i "s/{{SLOT_NUM}}/${s_num}/g" "$TEMP_SLOT"
    sed -i "s/{{LXC_NUM}}/${LXC_NUM}/g" "$TEMP_SLOT"
    sed -i "s/{{SVC_NAME}}/${s_name}/g" "$TEMP_SLOT"
    sed -i "s/{{SVC_DEP}}/${s_dep}/g" "$TEMP_SLOT"
    sed -i "s/{{STATE}}/${s_state}/g" "$TEMP_SLOT"
    sed -i "s/{{PRIORITY}}/${s_prio}/g" "$TEMP_SLOT"
    sed -i "s/{{PEER}}/${s_peer}/g" "$TEMP_SLOT"
    sed -i "s/{{SVC_VIP}}/${s_vip}/g" "$TEMP_SLOT"
    sed -i "s/{{AUTH_PASS}}/${SHORT_PASS}/g" "$TEMP_SLOT"

    cat "$TEMP_SLOT" >> "$CONFIG_FILE"
    rm "$TEMP_SLOT"
done

# 6. PERMS
cp ./check_services.sh "${CHECKS_DIR}/check_services.sh"
chmod +x "${CHECKS_DIR}/check_services.sh"

echo "Success: keepalived.conf generated."