#!/usr/bin/env sh

# Small script to run tests for a target (or all targets) inside all the
# respective docker images.

set -ex

# Default to assuming the CARGO_HOME is one directory up (to account for a `bin`
# subdir) from where the `cargo` binary in `$PATH` lives.
DEFAULT_CARGO_HOME="$(dirname "$(dirname "$(command -v cargo)")")"
# If the CARGO_HOME env var is already set, use that. If it isn't set use the
# default.
CARGO_HOME="${CARGO_HOME:-$DEFAULT_CARGO_HOME}"

echo "${HOME}"
pwd

# Avoid "no space left on device" failure.
if [ "${1}" = "aarch64-linux-android" ] ; then
  docker system prune -af
  docker system df
fi

run() {
    echo "Building docker container for target ${1}"

    # use -f so we can use ci/ as build context
    docker build -t libc -f "ci/docker/${1}/Dockerfile" ci/
    mkdir -p target
    if [ -w /dev/kvm ]; then
        kvm="--volume /dev/kvm:/dev/kvm"
    else
        kvm=""
    fi

    docker run \
      --rm \
      --user "$(id -u)":"$(id -g)" \
      --env LIBC_CI \
      --env CARGO_HOME=/cargo \
      --env CARGO_TARGET_DIR=/checkout/target \
      --volume "$CARGO_HOME":/cargo \
      --volume "$(rustc --print sysroot)":/rust:ro \
      --volume "$(pwd)":/checkout:ro \
      --volume "$(pwd)"/target:/checkout/target \
      $kvm \
      --init \
      --workdir /checkout \
      libc \
      sh -c "HOME=/tmp PATH=\$PATH:/rust/bin exec ci/run.sh ${1}"
}

build_switch() {
    echo "Building docker container for target switch"

    # use -f so we can use ci/ as build context
    docker build -t libc -f "ci/docker/switch/Dockerfile" ci/
    mkdir -p target
    if [ -w /dev/kvm ]; then
        kvm="--volume /dev/kvm:/dev/kvm"
    else
        kvm=""
    fi

    cp "$(which rustup)" "$(rustc --print sysroot)/bin"

    docker run \
      --rm \
      --user "$(id -u)":"$(id -g)" \
      --env LIBC_CI \
      --env CARGO_HOME=/cargo \
      --env CARGO_TARGET_DIR=/checkout/target \
      --volume "$CARGO_HOME":/cargo \
      --volume "$(rustc --print sysroot)":/rust:ro \
      --volume "$(pwd)":/checkout:ro \
      --volume "$(pwd)"/target:/checkout/target \
      --volume ~/.rustup:/.rustup:Z \
      $kvm \
      --init \
      --workdir /checkout \
      libc \
      sh -c "HOME=/tmp RUSTUP_HOME=/tmp PATH=\$PATH:/rust/bin rustup default nightly \
        && rustup component add rust-src --target ci/switch.json \
        && cargo build -Z build-std=core,alloc --target ci/switch.json"
}

if [ -z "${1}" ]; then
  for d in ci/docker/*; do
    run "${d}"
  done
else
  if [ "${1}" != "switch" ]; then
    run "${1}"
  else
    build_switch
  fi
fi
