steps:
  - label: ":buildkite: Graphify test selection"
    key: "graphify-test-selection"
    env:
      GRAPHIFY_REPO_ID: "YOUR_GRAPHIFY_REPOSITORY_ID"
    command: "bash .buildkite/graphify-select-tests.sh"

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

full_suite() {
  printf '%s\n' '{"steps":[{"label":":test_tube: full test suite","command":"pytest"}]}' \
    | buildkite-agent pipeline upload
}

# Branch builds keep running the full suite.
if [[ "${BUILDKITE_PULL_REQUEST:-false}" == "false" ]]; then
  full_suite
  exit 0
fi

: "${GRAPHIFY_REPO_ID:?Set your Graphify repository ID}"
: "${BUILDKITE_PULL_REQUEST_BASE_BRANCH:?Missing pull request base branch}"
GRAPHIFY_CI_TOKEN="${GRAPHIFY_CI_TOKEN:-$(buildkite-agent secret get GRAPHIFY_CI_TOKEN)}"
: "${GRAPHIFY_CI_TOKEN:?Provide a Graphify CI token}"

head_sha="$(git rev-parse "${BUILDKITE_COMMIT:-HEAD}^{commit}")"
git fetch --no-tags origin "$BUILDKITE_PULL_REQUEST_BASE_BRANCH"
base_sha="$(git merge-base "$head_sha" FETCH_HEAD)"

payload="$(jq -n --arg base "$base_sha" --arg head "$head_sha" \
  '{baseSha: $base, headSha: $head, format: "buildkite", command: "pytest"}')"
response="$(mktemp)"
trap 'rm -f "$response"' EXIT

if ! status="$(curl --silent --show-error --connect-timeout 10 --max-time 180 \
  --output "$response" --write-out '%{http_code}' \
  --request POST \
  --header "Authorization: Bearer $GRAPHIFY_CI_TOKEN" \
  --header "Content-Type: application/json" \
  --data "$payload" \
  "https://api.graphify.com/v1/repos/$GRAPHIFY_REPO_ID/test-impact")"; then
  echo "Graphify could not be reached; running the full suite."
  full_suite
  exit 0
fi

case "$status" in
  200)
    jq -e '.steps | type == "array" and length > 0' "$response" > /dev/null
    buildkite-agent pipeline upload --no-interpolation "$response"
    ;;
  429|5??)
    echo "Graphify returned HTTP $status; running the full suite."
    full_suite
    ;;
  *)
    echo "Graphify returned HTTP $status. Check the CI token, repository ID, and commit range." >&2
    exit 1
    ;;
esac