#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
REPO_DIR="/opt/NextWks"
TARGET_DIR="/opt/nextworkspace"
BACKUP_DIR="/opt/backup"
NETWORK_NAME="nextwks-net"
HEALTH_CHECK_RETRIES=10
HEALTH_CHECK_INTERVAL=2

# --- Load .env ---
if [ -f "$BACKUP_DIR/.env" ]; then
    set -a; source "$BACKUP_DIR/.env"; set +a
elif [ -f "$TARGET_DIR/.env" ]; then
    set -a; source "$TARGET_DIR/.env"; set +a
fi
DOMAIN="${DOMAIN:-nextwks.eu}"

# --- Mode ---
GREENFIELD=false
if [ "${1:-}" = "--destroy" ]; then GREENFIELD=true; echo "[MODE] Greenfield deploy (--destroy)"
elif [ ! -d "$TARGET_DIR" ]; then GREENFIELD=true; echo "[MODE] Greenfield deploy (target missing)"
else echo "[MODE] Smart update (target exists)"; fi

# --- Pull + build ---
cd "$REPO_DIR"
echo "[1/5] Pulling latest code..."
git pull 2>/dev/null || true

echo "[2/5] Building binary (static)..."
export PATH=$PATH:/usr/local/go/bin
CGO_ENABLED=0 go build -o nextworkspace .

# --- Greenfield ---
if [ "$GREENFIELD" = true ]; then
    echo "[3/5] Full teardown..."
    systemctl stop nextworkspace 2>/dev/null || true
    systemctl disable nextworkspace 2>/dev/null || true
    rm -f /etc/systemd/system/nextworkspace.service
    systemctl daemon-reload
    # Kill any lingering Caddy processes on port 80/443
    pkill -f "caddy" 2>/dev/null || true
    # Force clean ALL old pods, containers, and networks
    for p in $(podman pod ls -q 2>/dev/null); do podman pod rm -f "$p" 2>/dev/null || true; done
    podman rm -f caddy authelia launcher 2>/dev/null || true
    podman network rm "$NETWORK_NAME" 2>&1 || true

    if [ -d "$TARGET_DIR" ]; then
        chattr -R -i "$TARGET_DIR" 2>/dev/null || true
        rm -rf "$TARGET_DIR"
    fi

    echo "[4/5] Building directories..."
    mkdir -p "$TARGET_DIR/config/caddy"
    mkdir -p "$TARGET_DIR/config/authelia"
    mkdir -p "$TARGET_DIR/data/caddy"
    mkdir -p "$TARGET_DIR/data/authelia"
    mkdir -p "$TARGET_DIR/compose"
    mkdir -p "$TARGET_DIR/www"
    mkdir -p "$TARGET_DIR/config/nextworkspace"
    mkdir -p "$TARGET_DIR/logs"
    chown -R master:master "$TARGET_DIR/data/caddy" 2>/dev/null || true
    chown -R master:master "$TARGET_DIR/logs" 2>/dev/null || true

    if [ -f "$BACKUP_DIR/.env" ]; then
        cp "$BACKUP_DIR/.env" "$TARGET_DIR/.env"
        chmod 644 "$TARGET_DIR/.env"
    fi

    # Generate Caddyfile
    sed -e "s|{DOMAIN}|$DOMAIN|g" -e "s|{TLS_EMAIL}|${TLS_EMAIL:-admin@$DOMAIN}|g" \
        "$SCRIPT_DIR/config/caddy/Caddyfile" > "$TARGET_DIR/config/caddy/Caddyfile"

    # Generate Authelia config
    JWT_SECRET="${JWT_SECRET:-$(openssl rand -hex 32)}"
    SESSION_SECRET="${SESSION_SECRET:-$(openssl rand -hex 32)}"
    sed -e "s|{DOMAIN}|$DOMAIN|g" -e "s|{JWT_SECRET}|$JWT_SECRET|g" \
        -e "s|{SESSION_SECRET}|$SESSION_SECRET|g" \
        -e "s|{SMTP_HOST}|${SMTP_HOST:-smtp.openxchange.eu}|g" \
        -e "s|{SMTP_PORT}|${SMTP_PORT:-587}|g" \
        -e "s|{SMTP_USER}|${SMTP_USER:-post@nextwks.eu}|g" -e "s|{SMTP_PASS}|$SMTP_PASS|g" \
        "$SCRIPT_DIR/config/authelia/configuration.yml" > "$TARGET_DIR/config/authelia/configuration.yml"

    ADMIN_PASSWORD_HASH="${ADMIN_PASSWORD_HASH:-}"
    if [ -z "$ADMIN_PASSWORD_HASH" ] && [ -n "${ADMIN_PASSWORD:-}" ] && [ -f "$SCRIPT_DIR/tools/hash-password/main.go" ]; then
        ADMIN_PASSWORD_HASH=$(cd "$SCRIPT_DIR" && go run ./tools/hash-password/ "$ADMIN_PASSWORD" 2>/dev/null || echo "")
    fi
    sed -e "s|{ADMIN_PASSWORD_HASH}|$ADMIN_PASSWORD_HASH|g" -e "s|{TLS_EMAIL}|${TLS_EMAIL:-admin@$DOMAIN}|g" \
        "$SCRIPT_DIR/config/authelia/users_database.yml" > "$TARGET_DIR/config/authelia/users_database.yml"

    # Copy files
    cp "$SCRIPT_DIR/compose/stack.yaml" "$TARGET_DIR/compose/stack.yaml"
    cp nextworkspace "$TARGET_DIR/nextworkspace"
    if [ -f "$REPO_DIR/VERSION" ]; then cp "$REPO_DIR/VERSION" "$TARGET_DIR/VERSION"; fi
    if [ -d "$SCRIPT_DIR/config/www" ]; then cp -r "$SCRIPT_DIR/config/www"/* "$TARGET_DIR/www/"; fi
    cp -r "$SCRIPT_DIR/config/nextworkspace"/* "$TARGET_DIR/config/nextworkspace/" 2>/dev/null || true

    # Podman network + deploy stack
    echo "[5/5] Deploying containers on $NETWORK_NAME..."
    podman network create "$NETWORK_NAME" 2>&1 || true
    podman-compose -f "$TARGET_DIR/compose/stack.yaml" down 2>/dev/null || true
    podman-compose -f "$TARGET_DIR/compose/stack.yaml" up -d 2>&1 || echo "[WARN] Stack deploy had issues"

    sleep 3
    AUTHELIA_SECRET=$(grep -oP 'session_secret: \K.*' "$TARGET_DIR/config/authelia/configuration.yml" 2>/dev/null || echo "")
    if [ -n "$AUTHELIA_SECRET" ] && ! grep -q "AUTHELIA_SECRET" "$BACKUP_DIR/.env" 2>/dev/null; then
        echo "AUTHELIA_SECRET=$AUTHELIA_SECRET" >> "$BACKUP_DIR/.env"
    fi

# --- Smart update ---
else
    echo "[3/5] Redeploying containers..."
    podman-compose -f "$TARGET_DIR/compose/stack.yaml" up -d 2>&1 || true
fi

# --- Health check ---
echo "[*] Running health check..."
for i in $(seq 1 $HEALTH_CHECK_RETRIES); do
    HEALTH=$(podman exec launcher wget -qO- http://127.0.0.1:9000/health 2>/dev/null || echo "")
    if [ "$HEALTH" = "OK" ]; then
        echo "[OK] NextWorkspace launcher is healthy"
        exit 0
    fi
    echo "  Attempt $i/$HEALTH_CHECK_RETRIES — not ready yet..."
    sleep $HEALTH_CHECK_INTERVAL
done

echo "[FAIL] Health check failed — launcher did not respond"
exit 1