#!/usr/bin/env bash
# =============================================================
# Dragon Wind — First-Run Setup
# Generates a .env file with randomized ports if one doesn't
# already exist, then optionally starts the stack.
# =============================================================

set -euo pipefail

ENV_FILE="$(dirname "$0")/.env"

# ---- Random port helper (avoids well-known ports) -----------
rand_port() {
  # Pick a random port in range 10000–59999
  echo $(( RANDOM % 50000 + 10000 ))
}

# ---- Only generate .env if it doesn't exist -----------------
if [ -f "$ENV_FILE" ]; then
  echo "✅  .env already exists — skipping generation."
  echo "    Edit $ENV_FILE to change ports or credentials."
else
  WEB_PORT=$(rand_port)
  RELAY_TCP_PORT=$(rand_port)
  RELAY_UDP_PORT=$(rand_port)

  # Ensure all three are distinct
  while [ "$RELAY_TCP_PORT" -eq "$WEB_PORT" ]; do RELAY_TCP_PORT=$(rand_port); done
  while [ "$RELAY_UDP_PORT" -eq "$WEB_PORT" ] || [ "$RELAY_UDP_PORT" -eq "$RELAY_TCP_PORT" ]; do
    RELAY_UDP_PORT=$(rand_port)
  done

  cat > "$ENV_FILE" <<EOF
# Dragon Wind — auto-generated by setup.sh
# Ports are randomized on first run. Edit freely.

# ---- Ports --------------------------------------------------
WEB_PORT=$WEB_PORT
RELAY_TCP_PORT=$RELAY_TCP_PORT
RELAY_UDP_PORT=$RELAY_UDP_PORT

# ---- Auth (change these!) -----------------------------------
AUTH_USER=admin
AUTH_PASS=DragonWind2026!

# ---- S3 (set via Admin UI or here) --------------------------
S3_ENDPOINT=
S3_REGION=us-east-1
S3_BUCKET=
S3_ACCESS_KEY=
S3_SECRET_KEY=

# ---- AMPP (set via Admin UI or here) ------------------------
AMPP_BASE_URL=https://us-east-1.gvampp.com
AMPP_API_KEY=
EOF

  echo ""
  echo "🐉  Dragon Wind setup complete!"
  echo ""
  echo "    Ports assigned:"
  echo "      Web UI       →  http://localhost:$WEB_PORT"
  echo "      Relay TCP    →  :$RELAY_TCP_PORT"
  echo "      UDP Data     →  :$RELAY_UDP_PORT/udp"
  echo ""
  echo "    ⚠️  Change AUTH_PASS in .env before going live!"
  echo ""
fi

# ---- Optionally start the stack -----------------------------
if [ "${1:-}" = "--start" ] || [ "${1:-}" = "-s" ]; then
  echo "🚀  Starting Dragon Wind..."
  docker compose up -d --build
  echo ""
  # Re-read the port in case .env was pre-existing
  WEB_PORT_LIVE=$(grep '^WEB_PORT=' "$ENV_FILE" | cut -d= -f2)
  echo "✅  Running at http://localhost:${WEB_PORT_LIVE}"
else
  echo "    To start: docker compose up -d --build"
  echo "    Or run:   bash setup.sh --start"
fi