#!/bin/bash
#
# Copyright (c) 2019 Ant Financial
#
# SPDX-License-Identifier: Apache-2.0

set -o errexit
set -o nounset
set -o pipefail

script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
script_name="$(basename "${BASH_SOURCE[0]}")"

source "${script_dir}/common.bash"

rustarch=$(arch_to_rust)

version="${1:-""}"
if [ -z "${version}" ]; then
	version=$(get_from_kata_deps ".languages.rust.meta.newest-version")
fi

echo "Install rust ${version}"

if ! command -v rustup > /dev/null; then
	curl https://sh.rustup.rs -sSf | sh -s -- -y --default-toolchain ${version}
fi

export PATH="${PATH}:${HOME}/.cargo/bin"

## Still try to install the target version of toolchain,
## in case that the rustup has been installed but
## with a different version toolchain.
## Even though the target version toolchain has been installed,
## this command will not take too long to run.
rustup toolchain install ${version}
rustup default ${version}
if [ "${rustarch}" == "x86_64" ] || [ "${rustarch}" == "aarch64" ] ; then
	rustup target add ${rustarch}-unknown-linux-musl
	$([ "$(whoami)" != "root" ] && echo sudo) ln -sf /usr/bin/g++ /bin/musl-g++
else
	rustup target add ${rustarch}-unknown-linux-gnu
fi
rustup component add rustfmt
rustup component add clippy