#!/usr/bin/env bash

set -euo pipefail

CPU_ONLY=false
GPU_ONLY=false

ci_dir=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)

new_args="$("${ci_dir}/util/extract_switches.sh" -cpu-only -gpu-only -- "$@")"
declare -a new_args="(${new_args})"
set -- "${new_args[@]}"
while true; do
  case "$1" in
  -cpu-only)
    ARTIFACT_TAG="test_cpu"
    CPU_ONLY=true
    shift
    ;;
  -gpu-only)
    ARTIFACT_TAG="test_gpu"
    GPU_ONLY=true
    shift
    ;;
  --)
    shift
    break
    ;;
  *)
    echo "Unknown argument: $1"
    exit 1
    ;;
  esac
done

# shellcheck source=ci/build_common.sh
source "${ci_dir}/build_common.sh"

print_environment_details

if [[ -n "${GITHUB_ACTIONS:-}" ]]; then
  if ! $CPU_ONLY && ! $GPU_ONLY; then
    echo "Error: test_thrust.sh requires -cpu-only or -gpu-only in CI" >&2
    exit 1
  fi
  producer_id="$(util/workflow/get_producer_id.sh)"
  run_command "📦  Unpacking test artifacts" \
    "${ci_dir}/util/artifacts/download_packed.sh" \
      "z_thrust-test-artifacts-${DEVCONTAINER_NAME:?}-$producer_id-$ARTIFACT_TAG" \
      /home/coder/cccl/
else
  ./build_thrust.sh "$@"
fi

declare -a PRESET_GPU_PAIRS=()

if $CPU_ONLY; then
  PRESET_GPU_PAIRS+=("thrust-cpu:false")
elif $GPU_ONLY; then
  PRESET_GPU_PAIRS+=("thrust-gpu:true")
else
  PRESET_GPU_PAIRS+=("thrust-cpu:false" "thrust-gpu:true")
fi

for pair in "${PRESET_GPU_PAIRS[@]}"; do
  PRESET="${pair%%:*}"
  GPU_REQUIRED="${pair##*:}"
  test_preset "Thrust (${PRESET})" "${PRESET}" "${GPU_REQUIRED}"
done

print_time_summary