#!/usr/bin/env sh
# ═══════════════════════════════════════════════════════════════════════════
#  BitWell Reporter — one-line installer for any Linux host.
#
#  Usage:
#    curl -fsSL https://bitwell.dev/install/reporter.sh | sh -s -- \
#         --url https://bitwell.dev --token <ENROLL_TOKEN> [--region <label>]
#
#  Idempotent: re-running updates config and restarts the container.
# ═══════════════════════════════════════════════════════════════════════════
set -eu

INSTALL_DIR="/opt/bitwell-reporter"
IMAGE="bitwell.dev/reporter:latest"
GATHERER_URL=""
ENROLL_TOKEN=""
REGION=""

log()  { printf '\033[0;32m[reporter]\033[0m %s\n' "$*"; }
warn() { printf '\033[1;33m[reporter]\033[0m %s\n' "$*"; }
die()  { printf '\033[0;31m[reporter]\033[0m %s\n' "$*" >&2; exit 1; }

# ── Parse flags ─────────────────────────────────────────────────────────────
while [ $# -gt 0 ]; do
  case "$1" in
    --url)    GATHERER_URL="$2"; shift 2 ;;
    --token)  ENROLL_TOKEN="$2"; shift 2 ;;
    --region) REGION="$2"; shift 2 ;;
    *) die "unknown argument: $1" ;;
  esac
done

# ── Fall back to env / prompt ───────────────────────────────────────────────
[ -n "$GATHERER_URL" ] || GATHERER_URL="${GATHERER_URL_ENV:-}"
[ -n "$ENROLL_TOKEN" ] || ENROLL_TOKEN="${REPORTER_ENROLL_TOKEN:-}"

if [ -z "$GATHERER_URL" ]; then
  printf 'Gatherer URL [https://bitwell.dev]: '; read -r GATHERER_URL || true
  [ -n "$GATHERER_URL" ] || GATHERER_URL="https://bitwell.dev"
fi
if [ -z "$ENROLL_TOKEN" ] && [ ! -f "$INSTALL_DIR/data/agent.json" ]; then
  printf 'Enrollment token: '; read -r ENROLL_TOKEN || true
  [ -n "$ENROLL_TOKEN" ] || die "enrollment token is required for first install"
fi

# ── Ensure Docker ───────────────────────────────────────────────────────────
if ! command -v docker >/dev/null 2>&1; then
  log "Docker not found — installing via get.docker.com"
  curl -fsSL https://get.docker.com | sh || die "Docker install failed"
fi

if docker compose version >/dev/null 2>&1; then
  COMPOSE="docker compose"
elif command -v docker-compose >/dev/null 2>&1; then
  COMPOSE="docker-compose"
else
  die "docker compose plugin not available"
fi

# ── Write install dir + compose file ────────────────────────────────────────
log "Installing to $INSTALL_DIR"
mkdir -p "$INSTALL_DIR/data"

cat > "$INSTALL_DIR/docker-compose.yml" <<YAML
services:
  reporter:
    image: ${IMAGE}
    container_name: bitwell-reporter
    restart: unless-stopped
    env_file: .env
    volumes:
      - ./data:/data
    logging:
      driver: "json-file"
      options:
        max-size: "10m"
        max-file: "3"
YAML

# .env holds the token only until enrollment completes; harmless to keep (token is one-time).
cat > "$INSTALL_DIR/.env" <<ENV
GATHERER_URL=${GATHERER_URL}
REPORTER_ENROLL_TOKEN=${ENROLL_TOKEN}
REPORTER_REGION=${REGION}
ENV
chmod 600 "$INSTALL_DIR/.env"

# ── Pull + (re)start ────────────────────────────────────────────────────────
cd "$INSTALL_DIR"
log "Pulling image $IMAGE"
docker pull "$IMAGE" || warn "pull failed (will use local image if present)"
log "Starting reporter"
$COMPOSE up -d

log "Done. Logs: $COMPOSE -f $INSTALL_DIR/docker-compose.yml logs -f"