#!/usr/bin/env bash

#this script is based on the firesim build toolchains script

# exit script if any command fails
set -e
set -o pipefail

unamestr=$(uname)
RDIR=$(pwd)
: ${CHIPYARD_DIR:=$(pwd)} #default value is the PWD unless overridden

PRECOMPILED_REPO_HASH=56a40961c98db5e8f904f15dc6efd0870bfefd9e

usage() {
    echo "usage: ${0} [riscv-tools | esp-tools | ec2fast]"
    echo "   riscv: if set, builds the riscv toolchain (this is also the default)"
    echo "   hwacha: if set, builds esp-tools toolchain"
    echo "   ec2fast: if set, pulls in a pre-compiled RISC-V toolchain for an EC2 manager instance"
    exit "$1"
}

error() {
    echo "${0##*/}: ${1}" >&2
}

#taken from riscv-tools to check for open-ocd autoconf versions
check_version() {
    "$1" --version | awk "NR==1 {if (\$NF>$2) {exit 0} exit 1}" || {
        error "${3} requires at least ${1} version ${2}"
        exit 1
    }
}

TOOLCHAIN="riscv-tools"
EC2FASTINSTALL="false"
FASTINSTALL="false"

while getopts 'hH-:' opt ; do
    case $opt in
    h|H)
        usage 3 ;;
    -)
        case $OPTARG in
        help)
            usage 3 ;;
        ec2fast) # Preserve compatibility
            EC2FASTINSTALL=true ;;
        *)
            error "invalid option: --${OPTARG}"
            usage 1 ;;
        esac ;;
    *)
        error "invalid option: -${opt}"
        usage 1 ;;
    esac
done

shift $((OPTIND - 1))

if [ "$1" = ec2fast ] ; then
    EC2FASTINSTALL=true
elif [ -n "$1" ] ; then
    TOOLCHAIN="$1"
fi


if [ "$EC2FASTINSTALL" = "true" ]; then
    if [ "$TOOLCHAIN" = "riscv-tools" ]; then
      cd "$RDIR"
      git clone https://github.com/firesim/firesim-riscv-tools-prebuilt.git
      cd firesim-riscv-tools-prebuilt
      git checkout "$PRECOMPILED_REPO_HASH"
      PREBUILTHASH="$(cat HASH)"
      git -C "${CHIPYARD_DIR}" submodule update --init "toolchains/${TOOLCHAIN}"
      cd "$CHIPYARD_DIR/toolchains/$TOOLCHAIN"
      GITHASH="$(git rev-parse HEAD)"
      cd "$RDIR"
      echo "prebuilt hash: $PREBUILTHASH"
      echo "git      hash: $GITHASH"
      if [[ $PREBUILTHASH == $GITHASH && "$EC2FASTINSTALL" == "true" ]]; then
          FASTINSTALL=true
          echo "Using fast pre-compiled install for riscv-tools"
      else
          error 'error: hash of precompiled toolchain does not match the riscv-tools submodule hash'
          exit -1
      fi
    else
          error "error: unsupported precompiled toolchain: ${TOOLCHAIN}"
          exit -1
    fi
fi

INSTALL_DIR="$TOOLCHAIN-install"

RISCV="$(pwd)/$INSTALL_DIR"

# install risc-v tools
export RISCV="$RISCV"

if [ "$FASTINSTALL" = true ]; then
    cd firesim-riscv-tools-prebuilt
    ./installrelease.sh
    mv distrib "$RISCV"
    # copy HASH in case user wants it later
    cp HASH "$RISCV"
    cd "$RDIR"
    rm -rf firesim-riscv-tools-prebuilt
else
    mkdir -p "$RISCV"
    git -C "${CHIPYARD_DIR}" submodule update --init --recursive "toolchains/${TOOLCHAIN}" #--jobs 8
    cd "$CHIPYARD_DIR/toolchains/$TOOLCHAIN"

    # Scale number of parallel make jobs by hardware thread count
    ncpu="$(getconf _NPROCESSORS_ONLN || # GNU
        getconf NPROCESSORS_ONLN || # *BSD, Solaris
        nproc --all || # Linux
        sysctl -n hw.ncpu || # *BSD, OS X
        :)" 2>/dev/null
    case ${ncpu} in
    ''|*[^0-9]*) ;; # Ignore non-integer values
    *) export MAKEFLAGS="-j ${ncpu}" ;;
    esac

    #build the actual toolchain
    #./build.sh
    source build.common
    echo "Starting RISC-V Toolchain build process"
    build_project riscv-fesvr --prefix="${RISCV}"
    build_project riscv-isa-sim --prefix="${RISCV}" --with-fesvr="${RISCV}"
    build_project riscv-gnu-toolchain --prefix="${RISCV}"
    CC= CXX= build_project riscv-pk --prefix="${RISCV}" --host=riscv64-unknown-elf
    build_project riscv-tests --prefix="${RISCV}/riscv64-unknown-elf"
    echo -e "\\nRISC-V Toolchain installation completed!"

    # build static libfesvr library for linking into firesim driver (or others)
    cd riscv-fesvr/build
    "${CHIPYARD_DIR}/scripts/build-static-libfesvr.sh"
    cd "$RDIR"
    # build linux toolchain
    cd "$CHIPYARD_DIR/toolchains/$TOOLCHAIN/riscv-gnu-toolchain/build"
    make linux
    echo -e "\\nRISC-V Linux GNU Toolchain installation completed!"

fi

cd "$RDIR"

{
    echo "export CHIPYARD_TOOLCHAIN_SOURCED=1"
    echo "export RISCV=$(printf '%q' "$RISCV")"
    echo "export PATH=\${RISCV}/bin:\${PATH}"
    echo "export LD_LIBRARY_PATH=\${RISCV}/lib\${LD_LIBRARY_PATH:+":\${LD_LIBRARY_PATH}"}"
} > env.sh
echo "Toolchain Build Complete!"

if [ "$FASTINSTALL" = "false" ]; then
    # commands that can't run on EC2 (specifically, OpenOCD because of autoconf version_
    # see if the instance info page exists. if not, we are not on ec2.
    # this is one of the few methods that works without sudo
    if wget -T 1 -t 3 -O /dev/null http://169.254.169.254/; then
        echo "Skipping RISC-V OpenOCD"
    else
        echo "Building RISC-V OpenOCD"
        cd "$CHIPYARD_DIR/toolchains/$TOOLCHAIN"
        check_version automake 1.14 "OpenOCD build"
        check_version autoconf 2.64 "OpenOCD build"
        build_project riscv-openocd --prefix="${RISCV}" --enable-remote-bitbang --enable-jtag_vpi --disable-werror
        echo -e "\\nRISC-V OpenOCD installation completed!"
        cd "$RDIR"
    fi
fi