#!/bin/sh
set -eu

OUT_DIR="${OUT_DIR:-/data}"
PDB_BASE="${PDB_BASE:-https://www.peeringdb.com}"
INFO_TYPE="${PDB_INFO_TYPE:-Educational/Research}"

if [ -z "${MAXMIND_LICENSE_KEY:-}" ]; then
  echo "[health] MAXMIND_LICENSE_KEY missing" >&2
  exit 1
fi

if [ ! -d "${OUT_DIR}" ]; then
  echo "[health] OUT_DIR missing: ${OUT_DIR}" >&2
  exit 1
fi

if [ ! -s "${OUT_DIR}/GeoLite2-ASN.mmdb" ]; then
  echo "[health] GeoLite2-ASN.mmdb missing in ${OUT_DIR}" >&2
  exit 1
fi

if [ ! -s "${OUT_DIR}/nren_asns.txt" ]; then
  echo "[health] nren_asns.txt missing in ${OUT_DIR}" >&2
  exit 1
fi

mm_url="https://download.maxmind.com/app/geoip_download?edition_id=GeoLite2-ASN&license_key=${MAXMIND_LICENSE_KEY}&suffix=tar.gz"
mm_code="$(curl -fsS -o /dev/null -w "%{http_code}" "${mm_url}" || true)"
if [ "${mm_code}" != "200" ]; then
  echo "[health] MaxMind download not accessible (status ${mm_code})" >&2
  exit 1
fi

pdb_code="000"
pdb_url="${PDB_BASE}/api/net"
pdb_args="--get --data-urlencode info_type=${INFO_TYPE} --data-urlencode limit=1 --data-urlencode skip=0 --data-urlencode fields=asn,status,info_type"
if [ -n "${PDB_API_KEY:-}" ]; then
  pdb_code="$(curl -fsS -o /dev/null -w "%{http_code}" -H "Accept: application/json" -H "Authorization: Api-Key ${PDB_API_KEY}" ${pdb_args} "${pdb_url}" || true)"
else
  pdb_code="$(curl -fsS -o /dev/null -w "%{http_code}" -H "Accept: application/json" ${pdb_args} "${pdb_url}" || true)"
fi

if [ "${pdb_code}" != "200" ] && [ "${pdb_code}" != "429" ]; then
  echo "[health] PeeringDB not accessible (status ${pdb_code})" >&2
  exit 1
fi

exit 0
