#!/bin/bash
set -e

# ============================================================
# Container Spawner - Installationsskript
# https://gitea.iotxs.de/RainerWieland/spawner
# ============================================================

REPO_URL="https://gitea.iotxs.de/RainerWieland/spawner.git"
RAW_URL="https://gitea.iotxs.de/RainerWieland/spawner/raw/branch/main"
INSTALL_DIR="${PWD}"
VERSION="0.1.0"

# Farben fuer Output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Mindestversionen
MIN_DOCKER_VERSION="20.10"
MIN_COMPOSE_VERSION="2.0"

# ============================================================
# Hilfsfunktion: Versionen vergleichen (BusyBox/Synology kompatibel)
# Gibt 0 zurueck wenn $1 >= $2, sonst 1
# ============================================================
version_gte() {
    # Vergleiche zwei Versionen (z.B. "20.10.21" >= "20.10")
    # Kompatibel mit BusyBox (kein sort -V)
    local ver1="$1"
    local ver2="$2"

    # Extrahiere Major.Minor.Patch (fuege .0 hinzu falls noetig)
    local v1_major v1_minor v1_patch
    local v2_major v2_minor v2_patch

    v1_major=$(echo "$ver1" | cut -d. -f1)
    v1_minor=$(echo "$ver1" | cut -d. -f2)
    v1_patch=$(echo "$ver1" | cut -d. -f3)
    v1_minor=${v1_minor:-0}
    v1_patch=${v1_patch:-0}

    v2_major=$(echo "$ver2" | cut -d. -f1)
    v2_minor=$(echo "$ver2" | cut -d. -f2)
    v2_patch=$(echo "$ver2" | cut -d. -f3)
    v2_minor=${v2_minor:-0}
    v2_patch=${v2_patch:-0}

    # Vergleiche Major
    if [ "$v1_major" -gt "$v2_major" ] 2>/dev/null; then
        return 0
    elif [ "$v1_major" -lt "$v2_major" ] 2>/dev/null; then
        return 1
    fi

    # Major gleich, vergleiche Minor
    if [ "$v1_minor" -gt "$v2_minor" ] 2>/dev/null; then
        return 0
    elif [ "$v1_minor" -lt "$v2_minor" ] 2>/dev/null; then
        return 1
    fi

    # Minor gleich, vergleiche Patch
    if [ "$v1_patch" -ge "$v2_patch" ] 2>/dev/null; then
        return 0
    else
        return 1
    fi
}

echo ""
echo "============================================================"
echo "  Container Spawner Installation v${VERSION}"
echo "============================================================"
echo ""

# ============================================================
# 1. Pruefe .env
# ============================================================
if [ ! -f "${INSTALL_DIR}/.env" ]; then
    echo -e "${YELLOW}HINWEIS: Keine .env-Datei gefunden!${NC}"
    echo ""

    # Erstelle .env.example aus Repository
    echo "Lade .env.example herunter..."
    if command -v curl >/dev/null 2>&1; then
        curl -sSL "${RAW_URL}/.env.example" -o "${INSTALL_DIR}/.env.example"
    elif command -v wget >/dev/null 2>&1; then
        wget -q "${RAW_URL}/.env.example" -O "${INSTALL_DIR}/.env.example"
    else
        echo -e "${RED}Fehler: Weder curl noch wget gefunden!${NC}"
        exit 1
    fi

    echo -e "${GREEN}Vorlage erstellt: .env.example${NC}"
    echo ""
    echo "Naechste Schritte:"
    echo "  1. Kopiere die Vorlage:  cp .env.example .env"
    echo "  2. Passe die Werte an:   nano .env"
    echo "     - SECRET_KEY generieren (siehe Kommentar in .env)"
    echo "     - BASE_DOMAIN setzen"
    echo "     - TRAEFIK_NETWORK pruefen"
    echo "  3. Fuehre erneut aus:    bash install.sh"
    echo ""
    exit 0
fi

echo -e "${GREEN}.env-Datei gefunden${NC}"

# Lade .env Variablen
set -a
source "${INSTALL_DIR}/.env"
set +a

# ============================================================
# 2. Pruefe Voraussetzungen
# ============================================================
echo ""
echo "Pruefe Voraussetzungen..."

# Docker - Existenz und Version pruefen
if ! command -v docker >/dev/null 2>&1; then
    echo -e "${RED}Fehler: Docker nicht gefunden!${NC}"
    echo "Installiere Docker: https://docs.docker.com/get-docker/"
    exit 1
fi

# Docker Version extrahieren (z.B. "20.10.21" aus "Docker version 20.10.21, build baeda1f")
# BusyBox-kompatibel (kein grep -P)
DOCKER_VERSION=$(docker version --format '{{.Server.Version}}' 2>/dev/null || \
    docker version 2>/dev/null | grep -i "version" | head -1 | sed 's/.*version[: ]*\([0-9.]*\).*/\1/')
if [ -z "$DOCKER_VERSION" ]; then
    echo -e "${YELLOW}Warnung: Docker-Version konnte nicht ermittelt werden${NC}"
    echo -e "  Docker:         ${YELLOW}OK${NC} (Version unbekannt, min. ${MIN_DOCKER_VERSION} empfohlen)"
elif version_gte "$DOCKER_VERSION" "$MIN_DOCKER_VERSION"; then
    echo -e "  Docker:         ${GREEN}OK${NC} (v${DOCKER_VERSION})"
else
    echo -e "${RED}Fehler: Docker-Version ${DOCKER_VERSION} ist zu alt!${NC}"
    echo "Mindestversion: ${MIN_DOCKER_VERSION}"
    echo "Aktualisiere Docker: https://docs.docker.com/get-docker/"
    exit 1
fi

# Docker Compose - Existenz und Version pruefen
# BusyBox-kompatibel (kein grep -P)
COMPOSE_VERSION=""
if docker compose version >/dev/null 2>&1; then
    COMPOSE_CMD="docker compose"
    # Version extrahieren (z.B. "2.21.0" aus "Docker Compose version v2.21.0")
    COMPOSE_VERSION=$(docker compose version --short 2>/dev/null || \
        docker compose version 2>/dev/null | sed 's/[^0-9.]*\([0-9][0-9.]*\).*/\1/' | head -1)
elif command -v docker-compose >/dev/null 2>&1; then
    COMPOSE_CMD="docker-compose"
    COMPOSE_VERSION=$(docker-compose version --short 2>/dev/null || \
        docker-compose version 2>/dev/null | sed 's/[^0-9.]*\([0-9][0-9.]*\).*/\1/' | head -1)
else
    echo -e "${RED}Fehler: Docker Compose nicht gefunden!${NC}"
    echo "Docker Compose ist Teil von Docker Desktop oder kann separat installiert werden."
    exit 1
fi

if [ -z "$COMPOSE_VERSION" ]; then
    echo -e "  Docker Compose: ${YELLOW}OK${NC} (${COMPOSE_CMD}, Version unbekannt)"
elif version_gte "$COMPOSE_VERSION" "$MIN_COMPOSE_VERSION"; then
    echo -e "  Docker Compose: ${GREEN}OK${NC} (v${COMPOSE_VERSION})"
else
    echo -e "${RED}Fehler: Docker Compose Version ${COMPOSE_VERSION} ist zu alt!${NC}"
    echo "Mindestversion: ${MIN_COMPOSE_VERSION}"
    echo "Aktualisiere Docker Compose: https://docs.docker.com/compose/install/"
    exit 1
fi

# Git
if ! command -v git >/dev/null 2>&1; then
    echo -e "${RED}Fehler: Git nicht gefunden!${NC}"
    exit 1
fi
echo -e "  Git:            ${GREEN}OK${NC}"

# ============================================================
# 3. Pruefe ob bereits installiert (Update vs. Neuinstallation)
# ============================================================
echo ""

# Git safe.directory setzen (fuer NAS/Container-Umgebungen)
git config --global --add safe.directory "${INSTALL_DIR}" 2>/dev/null || true

if [ -d "${INSTALL_DIR}/.git" ]; then
    echo -e "${YELLOW}Update erkannt - hole neueste Aenderungen...${NC}"

    cd "${INSTALL_DIR}"

    # Sichere lokale Aenderungen (ignoriere Fehler)
    if git diff --quiet 2>/dev/null; then
        : # Keine Aenderungen
    else
        echo "Lokale Aenderungen gefunden, erstelle Stash..."
        git stash 2>/dev/null || true
    fi

    # Update durchfuehren
    if git fetch origin 2>/dev/null && git pull origin main 2>/dev/null; then
        echo -e "${GREEN}Repository aktualisiert${NC}"
    else
        echo -e "${YELLOW}Git-Update fehlgeschlagen, fahre mit lokalen Dateien fort...${NC}"
    fi
else
    echo "Neuinstallation - klone Repository..."

    # Temporaeres Verzeichnis fuer Clone
    TEMP_DIR=$(mktemp -d)

    if git clone "${REPO_URL}" "${TEMP_DIR}"; then
        # Kopiere Dateien (ueberschreibt nicht .env)
        shopt -s dotglob 2>/dev/null || true
        for item in "${TEMP_DIR}"/*; do
            basename_item=$(basename "$item")
            if [ "$basename_item" != ".env" ] && [ "$basename_item" != ".git" ]; then
                cp -r "$item" "${INSTALL_DIR}/"
            fi
        done

        # .git Verzeichnis kopieren fuer Updates
        cp -r "${TEMP_DIR}/.git" "${INSTALL_DIR}/"

        rm -rf "${TEMP_DIR}"
        echo -e "${GREEN}Repository geklont${NC}"
    else
        echo -e "${RED}Fehler: Repository konnte nicht geklont werden!${NC}"
        echo "URL: ${REPO_URL}"
        rm -rf "${TEMP_DIR}"
        exit 1
    fi
fi

# ============================================================
# 4. Verzeichnisse und Rechte setzen
# ============================================================
echo ""
echo "Setze Verzeichnisse und Berechtigungen..."

# Verzeichnisse erstellen falls nicht vorhanden
mkdir -p "${INSTALL_DIR}/data"
mkdir -p "${INSTALL_DIR}/logs"

# Berechtigungen setzen (rwx fuer Owner, rx fuer Group/Other)
chmod 755 "${INSTALL_DIR}/data"
chmod 755 "${INSTALL_DIR}/logs"

# Fuer Docker: Verzeichnisse muessen vom Container beschreibbar sein
# Option 1: Wenn Container als root laeuft (Standard) - 755 reicht
# Option 2: Wenn Container als non-root laeuft - 777 oder chown noetig

# Pruefen ob wir als root laufen (fuer chown)
if [ "$(id -u)" = "0" ]; then
    # Als root: Owner auf aktuellen User setzen (oder Docker-User)
    # Standard: belassen wie es ist (root kann alles)
    echo -e "  data/:  ${GREEN}OK${NC} (755, root)"
    echo -e "  logs/:  ${GREEN}OK${NC} (755, root)"
else
    # Als normaler User: Verzeichnisse muessen beschreibbar sein
    # Docker-Container laeuft meist als root, daher 755 ausreichend
    # Falls Container als non-root laeuft, auf 777 setzen:
    # chmod 777 "${INSTALL_DIR}/data" "${INSTALL_DIR}/logs"
    echo -e "  data/:  ${GREEN}OK${NC} (755)"
    echo -e "  logs/:  ${GREEN}OK${NC} (755)"
fi

# .env Datei schuetzen (nur Owner kann lesen/schreiben)
if [ -f "${INSTALL_DIR}/.env" ]; then
    chmod 600 "${INSTALL_DIR}/.env"
    echo -e "  .env:   ${GREEN}OK${NC} (600, nur Owner)"
fi

# install.sh ausfuehrbar machen
if [ -f "${INSTALL_DIR}/install.sh" ]; then
    chmod +x "${INSTALL_DIR}/install.sh"
fi

# ============================================================
# 5. Docker-Netzwerk pruefen/erstellen
# ============================================================
echo ""
NETWORK="${TRAEFIK_NETWORK:-web}"
echo "Pruefe Docker-Netzwerk: ${NETWORK}"

if docker network inspect "${NETWORK}" >/dev/null 2>&1; then
    echo -e "  Netzwerk '${NETWORK}': ${GREEN}existiert${NC}"
else
    echo -e "${RED}Fehler: Docker-Netzwerk '${NETWORK}' existiert nicht!${NC}"
    echo ""
    echo "Das Netzwerk muss von Traefik erstellt werden oder bereits existieren."
    echo "Stelle sicher, dass Traefik laeuft und das Netzwerk '${NETWORK}' verwendet."
    echo ""
    echo "Optionen:"
    echo "  1. Starte Traefik zuerst (empfohlen)"
    echo "  2. Erstelle das Netzwerk manuell: docker network create ${NETWORK}"
    echo ""
    read -p "Netzwerk jetzt manuell erstellen? (j/N): " create_network
    if [ "$create_network" = "j" ] || [ "$create_network" = "J" ]; then
        docker network create "${NETWORK}"
        echo -e "  Netzwerk '${NETWORK}': ${GREEN}erstellt${NC}"
    else
        exit 1
    fi
fi

# ============================================================
# 6. Pruefe ob Traefik laeuft
# ============================================================
echo ""
echo "Pruefe Traefik..."

# Suche nach laufenden Traefik-Containern
TRAEFIK_CONTAINER=$(docker ps --filter "name=traefik" --filter "status=running" --format "{{.Names}}" 2>/dev/null | head -1)

# Falls nicht nach Name gefunden, suche nach Image
if [ -z "$TRAEFIK_CONTAINER" ]; then
    TRAEFIK_CONTAINER=$(docker ps --filter "ancestor=traefik" --filter "status=running" --format "{{.Names}}" 2>/dev/null | head -1)
fi

# Falls immer noch nicht gefunden, suche nach Label (traefik.enable=true auf sich selbst)
if [ -z "$TRAEFIK_CONTAINER" ]; then
    TRAEFIK_CONTAINER=$(docker ps --filter "status=running" --format "{{.Names}}" 2>/dev/null | while read name; do
        if docker inspect "$name" 2>/dev/null | grep -q '"traefik.http.routers\|"com.docker.compose.service": "traefik"'; then
            echo "$name"
            break
        fi
    done)
fi

if [ -n "$TRAEFIK_CONTAINER" ]; then
    # Pruefe Traefik-Version (BusyBox-kompatibel)
    TRAEFIK_VERSION=$(docker exec "$TRAEFIK_CONTAINER" traefik version 2>/dev/null | \
        grep -i "version" | head -1 | sed 's/.*Version[: ]*\([0-9.]*\).*/\1/' || echo "unbekannt")
    [ -z "$TRAEFIK_VERSION" ] && TRAEFIK_VERSION="unbekannt"
    echo -e "  Traefik:        ${GREEN}laeuft${NC} (Container: ${TRAEFIK_CONTAINER}, v${TRAEFIK_VERSION})"

    # Pruefe ob Traefik am gleichen Netzwerk haengt
    TRAEFIK_NETWORKS=$(docker inspect "$TRAEFIK_CONTAINER" --format '{{range $k, $v := .NetworkSettings.Networks}}{{$k}} {{end}}' 2>/dev/null)
    if echo "$TRAEFIK_NETWORKS" | grep -q "$NETWORK"; then
        echo -e "  Traefik-Netzwerk: ${GREEN}OK${NC} (verbunden mit '${NETWORK}')"
    else
        echo -e "  ${YELLOW}Warnung: Traefik ist nicht mit Netzwerk '${NETWORK}' verbunden${NC}"
        echo "  Traefik-Netzwerke: ${TRAEFIK_NETWORKS}"
        echo "  Stelle sicher, dass TRAEFIK_NETWORK in .env korrekt konfiguriert ist."
    fi
else
    echo -e "  ${YELLOW}Warnung: Kein laufender Traefik-Container gefunden!${NC}"
    echo ""
    echo "  Traefik wird fuer das Routing der User-Container benoetigt."
    echo "  Der Spawner kann ohne Traefik gestartet werden, aber:"
    echo "    - User-Container sind nur lokal erreichbar"
    echo "    - Kein automatisches HTTPS"
    echo "    - Kein Subdomain-Routing"
    echo ""
    read -p "  Trotzdem fortfahren? (j/N): " continue_without_traefik
    if [ "$continue_without_traefik" != "j" ] && [ "$continue_without_traefik" != "J" ]; then
        echo ""
        echo "Installation abgebrochen."
        echo "Starte zuerst Traefik und fuehre dann erneut 'bash install.sh' aus."
        exit 1
    fi
    echo -e "  ${YELLOW}Fortfahren ohne Traefik...${NC}"
fi

# ============================================================
# 7. Docker-Images bauen
# ============================================================
echo ""
echo "Baue Docker-Images..."

# Stoppe laufende Container
${COMPOSE_CMD} down 2>/dev/null || true

# User-Template Image bauen (fuer User-Container)
if [ -d "${INSTALL_DIR}/user-template" ]; then
    echo "  [1/4] Baue user-service-template (User-Container)..."
    echo ""

    # Build ausfuehren und Output in Datei speichern
    BUILD_LOG="/tmp/build-user-template.log"
    docker build --no-cache -t user-service-template:latest "${INSTALL_DIR}/user-template/" > "${BUILD_LOG}" 2>&1
    BUILD_EXIT=$?

    # Gefilterten Output anzeigen
    grep -E "(Step |#[0-9]+ |Successfully|ERROR|error:|COPY|RUN|FROM)" "${BUILD_LOG}" 2>/dev/null | sed 's/^/        /' || true

    # Pruefe ob Build erfolgreich UND Image existiert
    if [ $BUILD_EXIT -eq 0 ] && docker image inspect user-service-template:latest >/dev/null 2>&1; then
        echo ""
        echo -e "  user-service-template: ${GREEN}OK${NC}"
    else
        echo ""
        echo -e "  user-service-template: ${RED}FEHLER${NC}"
        echo "  Build-Log:"
        cat "${BUILD_LOG}"
        exit 1
    fi
fi

# User-Template-Next Image bauen (alternatives Template, optional)
if [ -d "${INSTALL_DIR}/user-template-next" ]; then
    echo "  [2/4] Baue user-template-next (Next.js Template)..."
    echo -e "        ${BLUE}Dies kann 2-5 Minuten dauern (npm install + build)...${NC}"
    echo ""

    BUILD_LOG="/tmp/build-user-template-next.log"
    docker build --no-cache -t user-template-next:latest "${INSTALL_DIR}/user-template-next/" > "${BUILD_LOG}" 2>&1
    BUILD_EXIT=$?

    grep -E "(Step |#[0-9]+ |Successfully|ERROR|error:|COPY|RUN|FROM)" "${BUILD_LOG}" 2>/dev/null | sed 's/^/        /' || true

    if [ $BUILD_EXIT -eq 0 ] && docker image inspect user-template-next:latest >/dev/null 2>&1; then
        echo ""
        echo -e "  user-template-next: ${GREEN}OK${NC}"
    else
        echo ""
        echo -e "  user-template-next: ${YELLOW}WARNUNG - Build fehlgeschlagen (optional)${NC}"
    fi
fi

# Spawner Backend Image bauen
echo "  [3/4] Baue Spawner API (Flask Backend)..."
echo ""

BUILD_LOG="/tmp/build-spawner.log"
docker build --no-cache -t spawner:latest "${INSTALL_DIR}/" > "${BUILD_LOG}" 2>&1
BUILD_EXIT=$?

grep -E "(Step |#[0-9]+ |Successfully|ERROR|error:|COPY|RUN|FROM)" "${BUILD_LOG}" 2>/dev/null | sed 's/^/        /' || true

if [ $BUILD_EXIT -eq 0 ] && docker image inspect spawner:latest >/dev/null 2>&1; then
    echo ""
    echo -e "  spawner-api: ${GREEN}OK${NC}"
else
    echo ""
    echo -e "  spawner-api: ${RED}FEHLER${NC}"
    echo "  Build-Log:"
    cat "${BUILD_LOG}"
    exit 1
fi

# Frontend Image bauen
if [ -d "${INSTALL_DIR}/frontend" ]; then
    echo "  [4/4] Baue Frontend (Next.js)..."
    echo -e "        ${BLUE}Dies kann 2-5 Minuten dauern (npm install + build)...${NC}"
    echo ""

    BUILD_LOG="/tmp/build-frontend.log"
    docker build --no-cache -t spawner-frontend:latest "${INSTALL_DIR}/frontend/" > "${BUILD_LOG}" 2>&1
    BUILD_EXIT=$?

    grep -E "(Step |#[0-9]+ |Successfully|ERROR|error:|COPY|RUN|FROM)" "${BUILD_LOG}" 2>/dev/null | sed 's/^/        /' || true

    if [ $BUILD_EXIT -eq 0 ] && docker image inspect spawner-frontend:latest >/dev/null 2>&1; then
        echo ""
        echo -e "  spawner-frontend: ${GREEN}OK${NC}"
    else
        echo ""
        echo -e "  spawner-frontend: ${RED}FEHLER${NC}"
        echo "  Build-Log (letzte 50 Zeilen):"
        tail -50 "${BUILD_LOG}"
        exit 1
    fi
fi

echo ""
echo "Alle Images erfolgreich gebaut."

# ============================================================
# 8. Container starten
# ============================================================
echo ""
echo "Starte Container..."
${COMPOSE_CMD} up -d

# Warte auf Health-Check
echo ""
echo "Warte auf Spawner-Start..."
sleep 5

# Health-Check fuer API
SPAWNER_URL="http://localhost:${SPAWNER_PORT:-5000}/health"
if curl -sf "${SPAWNER_URL}" >/dev/null 2>&1; then
    echo -e "  API Health-Check:      ${GREEN}OK${NC}"
else
    echo -e "  API Health-Check:      ${YELLOW}Noch nicht bereit (normal beim ersten Start)${NC}"
fi

# Health-Check fuer Frontend
if curl -sf "http://localhost:3000/" >/dev/null 2>&1; then
    echo -e "  Frontend Health-Check: ${GREEN}OK${NC}"
else
    echo -e "  Frontend Health-Check: ${YELLOW}Noch nicht bereit (normal beim ersten Start)${NC}"
fi

# ============================================================
# 9. Fertig
# ============================================================
echo ""
echo "============================================================"
echo -e "  ${GREEN}Installation abgeschlossen!${NC}"
echo "============================================================"
echo ""

# URLs anzeigen
SCHEME="https"
if [ "${BASE_DOMAIN:-localhost}" = "localhost" ]; then
    SCHEME="http"
fi
FULL_URL="${SCHEME}://${SPAWNER_SUBDOMAIN:-coder}.${BASE_DOMAIN:-localhost}"

echo "Zugriff:"
echo "  Frontend:    ${FULL_URL}"
echo "  API:         ${FULL_URL}/api"
echo "  Health:      ${FULL_URL}/health"
echo ""
echo "Lokaler Zugriff (ohne Traefik):"
echo "  API:         http://localhost:${SPAWNER_PORT:-5000}"
echo "  Frontend:    http://localhost:3000"
echo ""
echo "Nuetzliche Befehle:"
echo "  Status:      ${COMPOSE_CMD} ps"
echo "  Logs API:    ${COMPOSE_CMD} logs -f spawner"
echo "  Logs FE:     ${COMPOSE_CMD} logs -f frontend"
echo "  Neustart:    ${COMPOSE_CMD} restart"
echo "  Stoppen:     ${COMPOSE_CMD} down"
echo ""