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

if [[ $# -ne 3 ]]; then
  echo "usage: $0 TOOLCHAIN_ROOT GCC_VERSION unpatched|patched" >&2
  exit 2
fi

root=$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)
source "$root/scripts/regression-common.sh"
resolve_regression_toolchain "$1"
version=$2
state=$3
[[ "$state" == unpatched || "$state" == patched ]] || {
  echo "error: state must be unpatched or patched" >&2
  exit 2
}

gxx="$REGRESSION_TOOLCHAIN/bin/g++"
[[ -x "$gxx" ]] || {
  echo "error: no g++ in $REGRESSION_TOOLCHAIN" >&2
  exit 1
}

cxx_fixture="$root/bundles/cxx-ada-vector-types/tests/vector-types.C"
ada_fixture="$root/bundles/cxx-ada-vector-types/tests/vector_types_consumer.adb"
work=$(mktemp -d "${TMPDIR:-/tmp}/gnat-cxx-ada-vector-types.XXXXXX")
trap 'rm -rf "$work"' EXIT

for optimization in 0 2; do
  case_dir="$work/O$optimization"
  mkdir -p "$case_dir"
  cp "$cxx_fixture" "$ada_fixture" "$case_dir/"
  (
    cd "$case_dir"
    "${REGRESSION_ENV[@]}" "$gxx" -c "-O$optimization" \
      -fdump-ada-spec-slim vector-types.C
  )

  spec="$case_dir/vector_types_c.ads"
  [[ -f "$spec" ]] || {
    echo "error: g++ did not generate $spec" >&2
    exit 1
  }

  set +e
  (
    cd "$case_dir"
    "${REGRESSION_ENV[@]}" "$REGRESSION_GNATMAKE" -q -f \
      "-O$optimization" vector_types_consumer.adb \
      -largs vector-types.o -lstdc++
    "${REGRESSION_ENV[@]}" ./vector_types_consumer
  ) >"$case_dir/run.log" 2>&1
  run_status=$?
  set -e

  if [[ "$state" == unpatched ]]; then
    [[ $run_status -ne 0 ]] || {
      echo "error: unpatched vector regression unexpectedly passed" >&2
      exit 1
    }
    grep -Eiq 'subtype indication expected|identifier expected' \
      "$case_dir/run.log" || {
      cat "$case_dir/run.log"
      exit 1
    }
    echo "cxx-ada-vector-types -O$optimization: expected invalid Ada (GCC $version)"
    continue
  fi

  [[ $run_status -eq 0 ]] || {
    cat "$case_dir/run.log"
    exit 1
  }
  grep -Eq 'type Int_Vector is array \(0 \.\. 3\) of int;' "$spec"
  grep -Eq 'type Double_Vector is array \(0 \.\. 1\) of double;' "$spec"
  grep -Eq 'pragma Machine_Attribute \(Int_Vector, "vector_type"\);' "$spec"
  [[ $(grep -Fc 'Convention => Ada' "$spec") -eq 3 ]]
  grep -Eq 'function transform' "$spec"
  echo "cxx-ada-vector-types -O$optimization: patched (GCC $version)"
done