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

# Server configuration
NATS_PORT="${NATS_PORT:=4222}"
NATS_HTTP_PORT="${NATS_HTTP_PORT:=8222}"
NATS_CLUSTER_PORT="${NATS_CLUSTER_PORT:=6222}"
DOCKER_NETWORK="${DOCKER_NETWORK:=aviso-net}"
CONTAINER_NAME="${CONTAINER_NAME:=nats-jetstream}"

# JetStream storage limits
MAX_MEMORY="${MAX_MEMORY:=5GB}"
MAX_STORAGE="${MAX_STORAGE:=10GB}"

# Authentication configuration
ENABLE_AUTH="${ENABLE_AUTH:=false}"
TOKEN="${TOKEN:=aviso_secure_token_$(date +%s)}"

# Create configuration directory
CONFIG_DIR="./nats_config"
mkdir -p "${CONFIG_DIR}"

# Generate NATS server configuration
generate_nats_config() {
    local config_file="${CONFIG_DIR}/nats-server.conf"

    cat > "${config_file}" << EOF
# NATS Server Configuration for Aviso Server
port: 4222
http_port: 8222

# JetStream Configuration
jetstream {
    store_dir: "/data"
    max_memory_store: ${MAX_MEMORY}
    max_file_store: ${MAX_STORAGE}
}

# Logging
debug: true
trace: false
logtime: true

# Health check endpoint
http: "0.0.0.0:8222"

EOF

    if [[ "${ENABLE_AUTH}" == "true" ]]; then
        cat >> "${config_file}" << EOF
# Token Authentication
authorization {
    token: "${TOKEN}"
}
EOF
        echo "Authentication enabled with token: ${TOKEN}"
    else
        echo "# No authentication configured" >> "${config_file}"
        echo "Authentication disabled"
    fi

    echo "Generated NATS configuration: ${config_file}"
}

# Generate the configuration
generate_nats_config

# Launch NATS with JetStream using Docker
if [[ -z "${SKIP_DOCKER:-}" ]]; then
    # Stop and remove existing container if it exists
    docker stop "${CONTAINER_NAME}" 2>/dev/null || true
    docker rm "${CONTAINER_NAME}" 2>/dev/null || true

    # Ensure shared docker network exists for NATS ecosystem tools (e.g. NUI)
    docker network inspect "${DOCKER_NETWORK}" >/dev/null 2>&1 || \
        docker network create "${DOCKER_NETWORK}" >/dev/null

    # Create volumes for persistence
    docker volume create nats-jetstream-data 2>/dev/null || true

    echo "Starting NATS JetStream server..."

    # Launch NATS with custom configuration
    docker run \
        --name "${CONTAINER_NAME}" \
        --network "${DOCKER_NETWORK}" \
        --publish "${NATS_PORT}":4222 \
        --publish "${NATS_HTTP_PORT}":8222 \
        --publish "${NATS_CLUSTER_PORT}":6222 \
        --volume "$(pwd)/${CONFIG_DIR}:/config:ro" \
        --volume nats-jetstream-data:/data \
        --detach \
        nats:2-alpine \
        --config /config/nats-server.conf

    # Wait for NATS to be ready
    echo "Waiting for NATS server to be ready..."
    sleep 3

    # Set connection URL and authentication
    if [[ "${ENABLE_AUTH}" == "true" ]]; then
        export NATS_URL="nats://localhost:${NATS_PORT}"
        export NATS_TOKEN="${TOKEN}"
    else
        export NATS_URL="nats://localhost:${NATS_PORT}"
    fi

    # Test connection with retry logic
    MAX_RETRIES=30
    RETRY_COUNT=0

    while [ $RETRY_COUNT -lt $MAX_RETRIES ]; do
        if nats server check connection 2>/dev/null; then
            echo "NATS server is ready!"
            break
        else
            echo "NATS is still starting up - sleeping (attempt $((RETRY_COUNT + 1))/$MAX_RETRIES)"
            sleep 2
            RETRY_COUNT=$((RETRY_COUNT + 1))
        fi
    done

    if [ $RETRY_COUNT -eq $MAX_RETRIES ]; then
        echo "ERROR: NATS server failed to start within expected time"
        docker logs "${CONTAINER_NAME}"
        exit 1
    fi
fi

echo ""
echo "=== NATS JetStream is ready! ==="
echo "Server URL: ${NATS_URL}"
echo "In-network URL: nats://${CONTAINER_NAME}:4222"
echo "HTTP Monitoring: http://localhost:${NATS_HTTP_PORT}"
echo "Docker network: ${DOCKER_NETWORK}"
echo "JetStream enabled with ${MAX_MEMORY} memory and ${MAX_STORAGE} file storage"

if [[ "${ENABLE_AUTH}" == "true" ]]; then
    echo "Authentication: Token-based"
    echo "Token: ${TOKEN}"
    echo ""
    echo "To connect with token:"
    echo "  export NATS_TOKEN=${TOKEN}"
else
    echo "Authentication: Disabled"
fi

echo ""
echo "=== Configuration ==="
echo "Configuration saved in: ${CONFIG_DIR}/nats-server.conf"
echo ""
echo "=== Useful Commands (If nats cli is installed in the system) ==="
echo "  nats stream ls                                    # List streams (created by your app)"
echo "  nats stream info <STREAM_NAME>                   # Stream details"
echo "  nats consumer ls <STREAM_NAME>                   # List consumers"
echo "  nats sub 'diss.>'                               # Subscribe to all dissemination events"
echo "  nats sub 'mars.>'                               # Subscribe to all MARS events"
echo "  nats sub 'bench.>'                              # Subscribe to benchmark events"
echo ""
echo "=== Management ==="
echo "To stop: docker stop ${CONTAINER_NAME}"
echo "To remove: docker rm ${CONTAINER_NAME} && docker volume rm nats-jetstream-data"
echo "To restart: docker start ${CONTAINER_NAME}"
echo ""
echo "=== Environment Variables for Aviso Server ==="
echo "export AVISOSERVER_NOTIFICATION_BACKEND__KIND=jetstream"
if [[ "${ENABLE_AUTH}" == "true" ]]; then
    echo "export NATS_TOKEN=${TOKEN}"
fi