#!/bin/bash

script_dir=$(cd "$(dirname "$0")" && pwd)
# shellcheck source=ci/shared.bash
. "$script_dir/shared.bash"

set -e
# Disable cause it makes shared script not to work properly
#set -x

TARGET="$1"

RUST_REPO="https://github.com/rust-lang/rust"
ARTIFACTS_BASE_URL="https://ci-artifacts.rust-lang.org/rustc-builds"

# A `Dockerfile` under `rustup`'s `ci/docker` directory may start with `FROM rust-$TARGET`.
# This means it is using a Docker image fetched from a container registry provided by `rustc`'s CI.
LOCAL_DOCKER_TAG="rust-$TARGET"
# The following is a mapping from `$TARGET`s to cached Docker images built from `Dockerfile`s under
# <https://github.com/rust-lang/rust/blob/HEAD/src/ci/docker/host-x86_64/>,
# e.g. `FROM rust-aarch64-unknown-linux-musl` means the base `Dockerfile` to look at is located under
# <https://github.com/rust-lang/rust/blob/HEAD/src/ci/docker/host-x86_64/dist-arm-linux-musl>.
case "$TARGET" in
  aarch64-unknown-linux-gnu)       image=dist-aarch64-linux ;;
  aarch64-unknown-linux-musl)      image=dist-arm-linux-musl ;;
  arm-unknown-linux-gnueabi)       image=dist-arm-linux-gnueabi ;;
  arm-unknown-linux-gnueabihf)     image=dist-armhf-linux ;;
  armv7-unknown-linux-gnueabihf)   image=dist-armv7-linux ;;
  i686-unknown-linux-gnu)          image=dist-i686-linux ;;
  *-linux-android*)                image=dist-android; LOCAL_DOCKER_TAG=rust-android ;;
  mips-unknown-linux-gnu)          image=dist-mips-linux ;;
  mips64-unknown-linux-gnuabi64)   image=dist-mips64-linux ;;
  mips64el-unknown-linux-gnuabi64) image=dist-mips64el-linux ;;
  mipsel-unknown-linux-gnu)        image=dist-mipsel-linux ;;
  powerpc-unknown-linux-gnu)       image=dist-powerpc-linux ;;
  powerpc64-unknown-linux-gnu)     image=dist-powerpc64-linux-gnu ;;
  powerpc64-unknown-linux-musl)    image=dist-powerpc64-linux-musl ;;
  powerpc64le-unknown-linux-gnu)   image=dist-powerpc64le-linux-gnu ;;
  powerpc64le-unknown-linux-musl)  image=dist-powerpc64le-linux-musl ;;
  s390x-unknown-linux-gnu)         image=dist-s390x-linux ;;
  sparcv9-sun-solaris)             image=dist-sparcv9-solaris ;;
  x86_64-pc-solaris)               image=dist-x86_64-solaris ;;
  x86_64-unknown-freebsd)          image=dist-x86_64-freebsd ;;
  x86_64-unknown-illumos)          image=dist-x86_64-illumos ;;
  x86_64-unknown-linux-gnu)        image=dist-x86_64-linux ;;
  x86_64-unknown-netbsd)           image=dist-x86_64-netbsd ;;
  riscv64gc-unknown-linux-gnu)     image=dist-riscv64-linux ;;
  loongarch64-unknown-linux-gnu)   image=dist-loongarch64-linux ;;
  loongarch64-unknown-linux-musl)  image=dist-loongarch64-musl ;;
  *) exit ;;
esac

head=$(git ls-remote "$RUST_REPO" HEAD | cut -f1)
image_url="$ARTIFACTS_BASE_URL/$head/image-$image.txt"
info="/tmp/image-$image.txt"

rm -f "$info"
curl -o "$info" "$image_url"

if [ -z "$(docker images -q "${LOCAL_DOCKER_TAG}")" ]; then
  set -e
  image_tag=$(cat $info)
  echo "Attempting to pull image ${image_tag}"
  docker pull "${image_tag}"
  docker tag "${image_tag}" "${LOCAL_DOCKER_TAG}"
fi