#!/usr/bin/env bash
set -euo pipefail

TOKIO_FEATURE=""
if [[ "${TOKIO_MULTI_THREAD:-false}" == "true" ]]; then
  TOKIO_FEATURE="tokio-multi-thread"
fi

allocators=(system jemalloc mimalloc_v2 mimalloc_v3)

if [[ "$RUNNER_OS" == "Linux" ]]; then
  time_cmd=(/usr/bin/time -v)
  parse_peak() {
    awk -F': ' '/Maximum resident set size/ {print $2; exit}'
  }
  parse_major() {
    awk -F': ' '/Major \(requiring I\/O\) page faults/ {print $2; exit}'
  }
  parse_minor() {
    awk -F': ' '/Minor \(reclaiming a frame\) page faults/ {print $2; exit}'
  }
else
  time_cmd=(/usr/bin/time -l)
  parse_peak() {
    awk '/maximum resident set size/ {print $1; exit}'
  }
  parse_major() {
    awk '/page faults/ {print $1; exit}'
  }
  parse_minor() {
    awk '/page reclaims/ {print $1; exit}'
  }
fi

build_features() {
  local allocator="$1"
  local features=""
  if [[ "$allocator" != "system" ]]; then
    features="$allocator"
  fi
  if [[ -n "$TOKIO_FEATURE" ]]; then
    if [[ -n "$features" ]]; then
      features="$features,$TOKIO_FEATURE"
    else
      features="$TOKIO_FEATURE"
    fi
  fi
  echo "$features"
}

run_one() {
  local allocator="$1"
  local features
  features="$(build_features "$allocator")"

  local feature_args=()
  if [[ -n "$features" ]]; then
    feature_args=(--features "$features")
  fi

  cargo build --release --locked "${feature_args[@]}"

  local exe="target/release/proxy-scraper-checker"
  if [[ "$RUNNER_OS" == "Windows" ]]; then
    exe="target/release/proxy-scraper-checker.exe"
  fi

  local output
  output="$(${time_cmd[@]} "$exe" 2>&1 >/dev/null)"

  local peak
  peak="$(echo "$output" | parse_peak)"
  local major
  major="$(echo "$output" | parse_major)"
  local minor
  minor="$(echo "$output" | parse_minor)"
  if [[ -z "$peak" ]]; then
    echo "Failed to parse peak memory for $allocator" >&2
    exit 1
  fi
  if [[ -z "$major" ]]; then
    major=0
  fi
  if [[ -z "$minor" ]]; then
    minor=0
  fi

  if [[ "$RUNNER_OS" != "Linux" ]]; then
    peak=$((peak / 1024))
  fi

  printf "%s\t%s\t%s\t%s\n" "$allocator" "$peak" "$major" "$minor" >> results.tsv
}

if [[ "$RUNNER_OS" == "Windows" ]]; then
  allocators=(system mimalloc_v2 mimalloc_v3)
fi

: > results.tsv
for allocator in "${allocators[@]}"; do
  run_one "$allocator"
done

{
  echo "### ${PLATFORM_LABEL:-unknown} (tokio-multi-thread=${TOKIO_MULTI_THREAD:-false})"
  echo ""
  echo "| Allocator | Peak KB | Major PF | Minor PF |"
  echo "| --- | ---: | ---: | ---: |"
  sort -n -k2,2 -k3,3 -k4,4 results.tsv | while IFS=$'\t' read -r allocator peak major minor; do
    echo "| $allocator | $peak | $major | $minor |"
  done
  best="$(sort -n -k2,2 -k3,3 -k4,4 results.tsv | head -n1)"
  best_allocator="${best%%$'\t'*}"
  best_rest="${best#*$'\t'}"
  best_peak="${best_rest%%$'\t'*}"
  best_rest="${best_rest#*$'\t'}"
  best_major="${best_rest%%$'\t'*}"
  best_minor="${best_rest#*$'\t'}"
  echo ""
  echo "**Best:** $best_allocator ($best_peak KB, $best_major major PF, $best_minor minor PF)"
} >> "$GITHUB_STEP_SUMMARY"