#!/bin/bash
# Copyright 2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

set -e

# Navigate to the repository root
# KOKORO_ARTIFACTS_DIR is set to /tmpfs/src by default in the Kokoro Ubuntu image.
export KOKORO_ARTIFACTS_DIR="${KOKORO_ARTIFACTS_DIR:-/tmpfs/src}"
cd "${KOKORO_ARTIFACTS_DIR}/git/staging"

# Use the Bazel caches pre-warmed in the container image, if any.
bazel_flags=()
if [[ -n "${BAZEL_REPOSITORY_CACHE:-}" ]]; then
  bazel_flags+=("--repository_cache=${BAZEL_REPOSITORY_CACHE}")
fi
if [[ -n "${BAZEL_DISK_CACHE:-}" ]]; then
  bazel_flags+=("--disk_cache=${BAZEL_DISK_CACHE}")
fi

# Run Bazel build and test all targets
bazel build "${bazel_flags[@]}" ...
bazel test "${bazel_flags[@]}" ...

function cleanup() {
  set +e
  if command -v bazel >/dev/null 2>&1; then
    bazel shutdown >/dev/null 2>&1 || true
  fi
}
trap cleanup EXIT