#!/bin/bash
# Deploy Keepalived configuration and scripts to Proxmox hosts

set -euo pipefail

# Load IP configuration
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
source "${PROJECT_ROOT}/config/ip-addresses.conf" 2>/dev/null || true


SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)"

if [ -f "$PROJECT_ROOT/.env" ]; then
    set +euo pipefail
    source "$PROJECT_ROOT/.env" 2>/dev/null || true
    set -euo pipefail
fi

PRIMARY_HOST="${PRIMARY_HOST:-192.168.11.11}"
SECONDARY_HOST="${SECONDARY_HOST:-192.168.11.12}"

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
log_success() { echo -e "${GREEN}[✓]${NC} $1"; }
log_warn() { echo -e "${YELLOW}[⚠]${NC} $1"; }
log_error() { echo -e "${RED}[✗]${NC} $1"; }

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "🚀 Deploy Keepalived Configuration"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

# Deploy to primary
log_info "Deploying to primary host ($PRIMARY_HOST)..."
scp -o StrictHostKeyChecking=no \
    "$SCRIPT_DIR/keepalived/check-npmplus-health.sh" \
    "$SCRIPT_DIR/keepalived/keepalived-notify.sh" \
    root@"$PRIMARY_HOST:/usr/local/bin/" || {
    log_error "Failed to deploy scripts to primary"
    exit 1
}

ssh -o StrictHostKeyChecking=no root@"$PRIMARY_HOST" \
    "chmod +x /usr/local/bin/check-npmplus-health.sh /usr/local/bin/keepalived-notify.sh" || {
    log_error "Failed to make scripts executable on primary"
    exit 1
}

scp -o StrictHostKeyChecking=no \
    "$SCRIPT_DIR/keepalived/keepalived-primary.conf" \
    root@"$PRIMARY_HOST:/etc/keepalived/keepalived.conf" || {
    log_error "Failed to deploy Keepalived config to primary"
    exit 1
}

log_success "Primary host configured"

# Deploy to secondary
log_info "Deploying to secondary host ($SECONDARY_HOST)..."
scp -o StrictHostKeyChecking=no \
    "$SCRIPT_DIR/keepalived/check-npmplus-health.sh" \
    "$SCRIPT_DIR/keepalived/keepalived-notify.sh" \
    root@"$SECONDARY_HOST:/usr/local/bin/" || {
    log_error "Failed to deploy scripts to secondary"
    exit 1
}

ssh -o StrictHostKeyChecking=no root@"$SECONDARY_HOST" \
    "chmod +x /usr/local/bin/check-npmplus-health.sh /usr/local/bin/keepalived-notify.sh" || {
    log_error "Failed to make scripts executable on secondary"
    exit 1
}

scp -o StrictHostKeyChecking=no \
    "$SCRIPT_DIR/keepalived/keepalived-secondary.conf" \
    root@"$SECONDARY_HOST:/etc/keepalived/keepalived.conf" || {
    log_error "Failed to deploy Keepalived config to secondary"
    exit 1
}

log_success "Secondary host configured"

echo ""
log_warn "IMPORTANT: Update auth_pass in /etc/keepalived/keepalived.conf on both hosts"
log_warn "The password must match on both hosts"
log_info ""
log_info "To start Keepalived:"
log_info "  On primary: systemctl enable keepalived && systemctl start keepalived"
log_info "  On secondary: systemctl enable keepalived && systemctl start keepalived"
log_info ""
log_info "To verify:"
log_info "  systemctl status keepalived"
log_info "  ip addr show vmbr0 | grep ${IP_NPMPLUS_ETH0:-192.168.11.166}"