#!/bin/sh
set -eu

: "${SERVICE_NAME:?need SERVICE_NAME}"
: "${SERVICE_ADDR:?need SERVICE_ADDR}"
: "${SERVICE_PORT:?need SERVICE_PORT}"
CONSUL="${CONSUL_HTTP_ADDR:-http://127.0.0.1:8500}"

echo "[registrar] waiting consul agent at $CONSUL ..."
for i in $(seq 1 60); do
  if wget -qO- "$CONSUL/v1/status/leader" >/dev/null 2>&1; then
    break
  fi
  sleep 1
done

ID="${SERVICE_NAME}-${SERVICE_ADDR}-${SERVICE_PORT}"

cat > /tmp/svc.json <<EOF
{
  "service": {
    "id": "${ID}",
    "name": "${SERVICE_NAME}",
    "address": "${SERVICE_ADDR}",
    "port": ${SERVICE_PORT},
    "check": {
      "http": "http://${SERVICE_ADDR}:${SERVICE_PORT}/",
      "interval": "10s",
      "timeout": "2s"
    }
  }
}
EOF

echo "[registrar] registering ${ID} ..."
consul services register -http-addr="$CONSUL" /tmp/svc.json

term() {
  echo "[registrar] deregister ${ID} ..."
  consul services deregister -http-addr="$CONSUL" /tmp/svc.json || true
  exit 0
}
trap term TERM INT

tail -f /dev/null