#!/usr/bin/env bash
# Per-slice + host PSI (pressure stall info) avg60 -> node_exporter textfile.
# Consumed by the existing mopac-harness prometheus; gateway may also read it.
set -u
OUT_DIR=${UKRRS_TEXTFILE_DIR:-/var/lib/node_exporter/textfile}
CGROOT=${UKRRS_CGROOT:-/sys/fs/cgroup}
mkdir -p "$OUT_DIR"
tmp=$(mktemp "$OUT_DIR/.ukrrs.XXXXXX")

some60() { # file -> prints avg60 value or nothing
  awk '$1=="some"{for(i=2;i<=NF;i++) if($i ~ /^avg60=/){sub("avg60=","",$i); print $i; exit}}' "$1" 2>/dev/null
}

emit() { # name cgroup_path
  local res v
  for res in cpu memory io; do
    [ -r "$2/$res.pressure" ] || continue
    v=$(some60 "$2/$res.pressure")
    [ -n "${v:-}" ] && echo "ukrrs_slice_pressure_some60{slice=\"$1\",res=\"$res\"} $v"
  done
}

echo "# HELP ukrrs_slice_pressure_some60 PSI some avg60 (percent) for ukrrs slices" >>"$tmp"
echo "# TYPE ukrrs_slice_pressure_some60 gauge" >>"$tmp"
for cg in "$CGROOT"/system.slice/ukrrs-*.slice; do
  [ -d "$cg" ] && emit "${cg##*/}" "$cg"
done
[ -d "$CGROOT/user.slice" ] && emit "user.slice" "$CGROOT/user.slice"

echo "# HELP ukrrs_host_pressure_some60 PSI some avg60 (percent) host-wide" >>"$tmp"
echo "# TYPE ukrrs_host_pressure_some60 gauge" >>"$tmp"
for res in cpu memory io; do
  [ -r "/proc/pressure/$res" ] || continue
  v=$(some60 "/proc/pressure/$res")
  [ -n "${v:-}" ] && echo "ukrrs_host_pressure_some60{res=\"$res\"} $v"
done

mv "$tmp" "$OUT_DIR/ukrrs_pressure.prom"
chmod 0644 "$OUT_DIR/ukrrs_pressure.prom"