#!/usr/bin/env bash
# ==============================================================================
# Jellyfin 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 "   JELLYFIN MIGRATIE NAAR KUBERNETES CLUSTER (NAMESPACE: ${HTPC_NS})"
echo "========================================================================"
echo "Bron:       jellyfin.service op ${CLUSTER_HOST}"
echo "Config:     /etc/jellyfin & /var/lib/jellyfin"
echo "Doel:       K8s Pod in namespace '${HTPC_NS}' op ${CLUSTER_HOST}"
echo "Poort:      8096 (ClusterIP & HostPort 8096)"
echo "Transcode:  AMD VA-API (/dev/dri/renderD128)"
echo "========================================================================"
echo ""

read -p "Weet je zeker dat je de Jellyfin 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 JELLYFIN CONFIG & DATA
# ------------------------------------------------------------------------------
log_step "2. Reservekopie maken van /etc/jellyfin en /var/lib/jellyfin/data op ${CLUSTER_HOST}..."
ssh -t "${SSH_USER}@${CLUSTER_HOST}" "
    sudo mkdir -p '${BACKUP_DIR}'
    if [ -d '/etc/jellyfin' ] || [ -d '/var/lib/jellyfin/data' ]; then
        echo 'Inpakken van Jellyfin configuratie en database naar ${BACKUP_DIR}/jellyfin_backup_${DATE_STR}.tar.gz...'
        sudo tar -czf '${BACKUP_DIR}/jellyfin_backup_${DATE_STR}.tar.gz' \
            -C / etc/jellyfin \
            -C /var/lib/jellyfin data
        echo 'Backup gereed:'
        ls -lh '${BACKUP_DIR}/jellyfin_backup_${DATE_STR}.tar.gz'
    else
        echo 'Geen bestaande Jellyfin data gevonden om te back-uppen.'
    fi
"

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

# ------------------------------------------------------------------------------
# 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 Jellyfin pod (eerste start kan enkele minuten duren voor datamigratie)..."
KUBECONFIG="${KUBECONFIG_PATH}" kubectl rollout status deployment/jellyfin -n "${HTPC_NS}" --timeout=300s

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

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

log_info "Cluster IP service testen..."
CLUSTER_IP=$(KUBECONFIG="${KUBECONFIG_PATH}" kubectl get svc jellyfin -n "${HTPC_NS}" -o jsonpath='{.spec.clusterIP}')
ssh -o BatchMode=yes "${SSH_USER}@${CLUSTER_HOST}" "curl -I -s http://${CLUSTER_IP}:8096/jellyfin/health" | head -n 1

log_info "Controleer of render apparaten (/dev/dri) toegankelijk zijn in container..."
POD_NAME=$(KUBECONFIG="${KUBECONFIG_PATH}" kubectl get pod -n "${HTPC_NS}" -l app=jellyfin -o jsonpath='{.items[0].metadata.name}')
KUBECONFIG="${KUBECONFIG_PATH}" kubectl exec -n "${HTPC_NS}" "${POD_NAME}" -- ls -la /dev/dri || true

echo ""
echo "========================================================================"
log_info "JELLYFIN MIGRATIE SUCCESVOL AFGEROND!"
echo "De Jellyfin media server draait nu in Kubernetes."
echo "Apache proxy: https://thuus.bnhmeijer.nl/jellyfin"
echo "========================================================================"