#!/usr/bin/env bash
# ==============================================================================
# slskd Migration Script: Host Service -> Kubernetes Cluster (htpc namespace)
# ==============================================================================
set -euo pipefail

CLUSTER_HOST="10.2.0.4"
SSH_USER="bart"
HTPC_NS="htpc"
BACKUP_DIR="/storage/backups"
DATE_STR=$(date +%Y-%m-%d)
KUBECONFIG_PATH="${KUBECONFIG:-$HOME/.kube/config_vmserver}"

COLOR_GREEN="\033[0;32m"
COLOR_YELLOW="\033[1;33m"
COLOR_RED="\033[0;31m"
COLOR_BLUE="\033[0;34m"
COLOR_RESET="\033[0m"

log_info() { echo -e "${COLOR_GREEN}[INFO]${COLOR_RESET} $1"; }
log_step() { echo -e "\n${COLOR_BLUE}==>${COLOR_RESET} ${COLOR_GREEN}$1${COLOR_RESET}"; }
log_warn() { echo -e "${COLOR_YELLOW}[WARN]${COLOR_RESET} $1"; }
log_err()  { echo -e "${COLOR_RED}[ERROR]${COLOR_RESET} $1"; }

echo "========================================================================"
echo "   SLSKD MIGRATIE NAAR KUBERNETES CLUSTER (NAMESPACE: ${HTPC_NS})"
echo "========================================================================"
echo "Bron:       slskd.service op ${CLUSTER_HOST}"
echo "Config:     /var/www/.local/share/slskd"
echo "Doel:       K8s Pod in namespace '${HTPC_NS}' op ${CLUSTER_HOST}"
echo "Poorten:    Web (5030/5031), Soulseek Listen (50300)"
echo "========================================================================"
echo ""

read -p "Weet je zeker dat je de slskd migratie naar Kubernetes wilt starten? (j/N): " -r CONFIRM
if [[ ! "$CONFIRM" =~ ^[jJyY]$ ]]; then
    log_info "Migratie geannuleerd."
    exit 0
fi

# ------------------------------------------------------------------------------
# STAP 1: PRE-FLIGHT CHECKS
# ------------------------------------------------------------------------------
log_step "1. Pre-flight checks uitvoeren..."

log_info "Verbinding testen naar host (${CLUSTER_HOST})..."
ssh -o BatchMode=yes -o ConnectTimeout=5 "${SSH_USER}@${CLUSTER_HOST}" "true" || {
    log_err "Kan niet verbinden met host (${CLUSTER_HOST}) via SSH."
    exit 1
}

log_info "Controleren van KUBECONFIG verbinding..."
KUBECONFIG="${KUBECONFIG_PATH}" kubectl get nodes | grep -q "vmserver" || {
    log_err "Kan vmserver node niet bereiken via kubectl (KUBECONFIG=${KUBECONFIG_PATH})!"
    exit 1
}

# ------------------------------------------------------------------------------
# STAP 2: RESERVEKOPIE MAKEN VAN SLSKD DATA & CONFIG
# ------------------------------------------------------------------------------
log_step "2. Reservekopie maken van /var/www/.local/share/slskd op ${CLUSTER_HOST}..."
ssh -t "${SSH_USER}@${CLUSTER_HOST}" "
    sudo mkdir -p '${BACKUP_DIR}'
    if [ -d '/var/www/.local/share/slskd' ]; then
        echo 'Inpakken van slskd data naar ${BACKUP_DIR}/slskd_backup_${DATE_STR}.tar.gz...'
        sudo tar -czf '${BACKUP_DIR}/slskd_backup_${DATE_STR}.tar.gz' -C /var/www/.local/share slskd
        echo 'Backup gereed:'
        ls -lh '${BACKUP_DIR}/slskd_backup_${DATE_STR}.tar.gz'
    else
        echo 'Geen /var/www/.local/share/slskd gevonden, backup overgeslagen.'
    fi
"

# ------------------------------------------------------------------------------
# STAP 3: STANDALONE SLSKD SERVICE STOPPEN OP DE HOST
# ------------------------------------------------------------------------------
log_step "3. Standalone slskd.service stoppen op ${CLUSTER_HOST}..."
ssh -t "${SSH_USER}@${CLUSTER_HOST}" "
    echo '--> Stoppen van slskd.service...'
    sudo systemctl stop slskd
"

# ------------------------------------------------------------------------------
# STAP 4: KUBERNETES MANIFESTS TOEPASSEN
# ------------------------------------------------------------------------------
log_step "4. Kubernetes manifests toepassen..."
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
KUBECONFIG="${KUBECONFIG_PATH}" kubectl apply -k "${SCRIPT_DIR}/k8s/"

log_info "Wachten op succesvolle uitrol van slskd pod..."
KUBECONFIG="${KUBECONFIG_PATH}" kubectl rollout status deployment/slskd -n "${HTPC_NS}" --timeout=180s

# ------------------------------------------------------------------------------
# STAP 5: VERIFICATIE
# ------------------------------------------------------------------------------
log_step "5. Verificatie van endpoints..."
sleep 5

log_info "Health endpoint testen via host (poort 5030)..."
ssh -o BatchMode=yes "${SSH_USER}@${CLUSTER_HOST}" "curl -I -s http://127.0.0.1:5030/health" | head -n 1

log_info "Web UI root testen via host..."
ssh -o BatchMode=yes "${SSH_USER}@${CLUSTER_HOST}" "curl -I -s http://127.0.0.1:5030/" | head -n 1

echo ""
echo "========================================================================"
log_info "SLSKD MIGRATIE SUCCESVOL AFGEROND!"
echo "slskd draait nu als pod in Kubernetes namespace '${HTPC_NS}'."
echo "Configuratie en databases zijn bewaard in PersistentVolume slskd-config."
echo "Web UI bereikbaar op: http://10.2.0.4:5030/"
echo "Host service uitschakelen / opruimen kan via: ./cleanup_old_slskd.sh"
echo "========================================================================"