setup.sh

#! /bin/env bash

# THEJAS32 SDK setup script
# Author - Arif B <arif.dev@pm.me>

if [ ! -f $PWD/config ];then
    echo "Configuration file absent, aborting installation!"
    exit 1
fi

source $PWD/config

set -e

# Check if running as root
if [[ "$EUID" != 0 ]]
then 
    echo "${MAGENTA}Please run the script with administrator privileges!${NORMAL}"
    echo "Try sudo $0"
    exit
fi

# Install dependencies
if command -v apt &> /dev/null; then
    apt-get -y install make tar xz-utils git minicom gcc
elif command -v yum &> /dev/null; then
    yum install -y make tar xz git minicom gcc ncurses
elif command -v pacman &> /dev/null; then
    pacman -Sy make tar xz git minicom gcc
else
    printf "${RED}Unsupported Distro! Please make sure the following packages are installed:${NORMAL}"
    printf "make\ntar\ngit\nxz\nminicom\ngcc\n"
    printf "${YELLOW}Continue? (Y/n) :${NORMAL}"
    read input
    if [ ${input^^} = "N" ]; then
       exit
    fi
fi

# Copy header files
mkdir -p "$HEADER_PATH"
cp "$SETUP_PATH"/hal/include/* "$HEADER_PATH"

# Copy libraries
mkdir -p "$SDK_PATH"
cp hal/thejas32/* "$SDK_PATH"
cp -r "$SETUP_PATH"/hal/drivers "$SDK_PATH"
cp -r "$SETUP_PATH"/hal/libs "$SDK_PATH"
gcc "$SETUP_PATH"/tools/xmodem.c -o "$SDK_PATH"/xmodem

# Add sdk directory to PATH variable
case "$DEFAULT_SHELL" in
  "bash")
    RC_FILE=/home/$SUDO_USER/.bashrc
    ;;
  "zsh")
    RC_FILE=/home/$SUDO_USER/.zshrc
    ;;
  *)
    printf "Unsupported shell: %s
    Manually export the environment variable VEGA_SDK_PATH in your shell profile
    export VEGA_SDK_PATH=%s",$DEFAULT_SHELL,$VEGA_SDK_PATH
    ;;
esac

echo "export VEGA_SDK_PATH=$SDK_PATH" >> $RC_FILE


# Download RISC-V cross-toolchain 
# WIP, Downloading a fixed release for Linux currently
printf "\nIf this is your first time running the script, you might want to download pre-compiled RISC-V toolchain\n"
printf "\n${YELLOW}Download RISC-V Toolchain? ([Y]/n) :${NORMAL} "
read -r input

if [ -z "$input" ] || [ "${input^^}" = "Y" ]; then

    if [ "$HOST_ARCH" = "x86_64" ];then
        TOOLCHAIN_URL=https://github.com/xpack-dev-tools/riscv-none-elf-gcc-xpack/releases/download/v13.2.0-2/xpack-riscv-none-elf-gcc-13.2.0-2-linux-x64.tar.gz
        flag=1
    elif [ "$HOST_ARCH" = "aarch64" ];then
        TOOLCHAIN_URL=https://github.com/xpack-dev-tools/riscv-none-elf-gcc-xpack/releases/download/v13.2.0-2/xpack-riscv-none-elf-gcc-13.2.0-2-linux-arm64.tar.gz
        flag=1
    else
        printf "${MAGENTA}Unsupported host architecture! Please create a pull request${NORMAL}"
        flag=0
    fi

    if [ "$flag" = 1 ];then
        mkdir -p $RISCV_TOOLCHAIN_PATH
        pushd $RISCV_TOOLCHAIN_PATH
        curl -OL "$TOOLCHAIN_URL"
        tar xzf *.tar.gz
        rm *.tar.gz
        extracted_dir=$(find . -maxdepth 1 -type d -name "xpack-*" -print -quit)

        if [ -n "$extracted_dir" ]; then
            bin_path="$(realpath "$extracted_dir")/bin"
        
            echo "export PATH=\$PATH:$bin_path" >> $RC_FILE
            popd
        fi
    fi
fi


#  Create minicom config

printf "
# THEJAS32 Minicom Configuration
pu port             /dev/ttyUSB0
pu baudrate         115200
pu bits             8
pu parity           N
pu stopbits         1
pu rtscts           No
" > $MINICOM_CONFIG

# Add user to dialout group
usermod -aG dialout $SUDO_USER

# Add udev rules

echo '
# CP2102 on ARIESv3
SUBSYSTEM=="usb", SYSFS{idVendor}=="0x10c4", SYSFS{idProduct}=="0xea60", ACTION=="add", GROUP="dialout", MODE="0664"

# FT230X on ARIESv2
SUBSYSTEM=="usb", SYSFS{idVendor}=="0404", SYSFS{idProduct} =="6015", ACTION=="add", GROUP="dialout", MODE="0664"

' > $UDEV_CONFIG

# Reload rules
udevadm control --reload-rules && udevadm trigger

echo "${GREEN}
------------------------------
Successfully Installed the SDK
------------------------------
${NORMAL}"