#!/bin/bash
#
# Script to install Vaxtor Software
#
# Usage:
# Run as root or insert `sudo -E` before `bash`:
#
# curl -sL https://nexus.vaxtor.com/setup_vaxtor.sh | sudo -E bash -s -- [options]
#   or
# wget -qO- https://nexus.vaxtor.com/setup_vaxtor.sh | sudo -E bash -s -- [options]
#
# Options:
#   -t, --test
#       Use the test repository (https://nexus.vaxtor.com/repository/test-...)
#       instead of the standard production repository. Useful for QA and testing.
#
#   -c, --component, -e, --engine <name>
#       Specify the component or engine to install (default: ml600).
#       Available options typically include: nosimd, ml300, ml600.
#
# Examples:
#   1. Install with the default component (ml601) from the production repo:
#      curl -sL https://nexus.vaxtor.com/setup_vaxtor.sh | sudo -E bash -
#
#   2. Install the ml300 engine from the production repo:
#      curl -sL https://nexus.vaxtor.com/setup_vaxtor.sh | sudo -E bash -s -- -e ml300
#
#   3. Install the default engine from the TEST repo:
#      curl -sL https://nexus.vaxtor.com/setup_vaxtor.sh | sudo -E bash -s -- --test
#
#   4. Install the ml600 engine from the TEST repo:
#      curl -sL https://nexus.vaxtor.com/setup_vaxtor.sh | sudo -E bash -s -- -t -e ml600
#
#
# Need support? Contact us at support@vaxtor.com
#
#

REPOSITORY_BASE="https://nexus.vaxtor.com/repository/"
COMPONENT="ml600"
ARCH=`uname -m`

POSITIONAL_ARGS=()

while [[ $# -gt 0 ]]; do
  case $1 in
    -t|--test)
      REPOSITORY_BASE="https://nexus.vaxtor.com/repository/test-"
      shift
    ;;
    -c|--component|-e|--engine)
      COMPONENT="$2"
      shift
      shift
      ;;
    -*|--*)
      echo "Unknown option $1"
      exit 1
      ;;
    *)
      POSITIONAL_ARGS+=("$1")
      shift
      ;;
  esac
done

set -- "${POSITIONAL_ARGS[@]}"


export DEBIAN_FRONTEND=noninteractive

if test -t 1; then # if terminal
    ncolors=$(which tput > /dev/null && tput colors) # supports color
    if test -n "$ncolors" && test $ncolors -ge 8; then
        termcols=$(tput cols)
        bold="$(tput bold)"
        underline="$(tput smul)"
        standout="$(tput smso)"
        normal="$(tput sgr0)"
        black="$(tput setaf 0)"
        red="$(tput setaf 1)"
        green="$(tput setaf 2)"
        yellow="$(tput setaf 3)"
        blue="$(tput setaf 4)"
        magenta="$(tput setaf 5)"
        cyan="$(tput setaf 6)"
        white="$(tput setaf 7)"
    fi
fi

print_status() {
    echo
    echo "${yellow}## $1${normal}"
    echo
}

print_logo() {

    echo
    echo
    echo
    echo "${cyan}================================================================================${normal}"
    echo "${cyan}================================================================================${normal}"
    echo

    echo "
                              ${blue}^?JJYJ.  ${cyan}.^~~!~.                                                                              
                            ${blue}:7JJYYJ^.${cyan}.^~~~!!77!:                                                                            
                          ${blue}:7JJJJY^ ${cyan}.:^^^~~!!77?J7:                                                                          
                         ${blue}^7????JJJ ${cyan}.^^^^^~~!!77??JY?^     
                      ${blue}^7?77???JJJ  ${cyan}::::^^~!!77??JJYYJ^   
                    ${blue}.7?7777???JJJ:...:::: ${cyan}.777??JJYYYY?. 
                      ${blue}^7?77???JJJY555PPGG.${cyan}.777??JJYYJ^   
                        ${blue}:7????JJJYY55PPGG.${cyan}.777??JY?^     
                          ${blue}:7JJJJJYY55PPY^ ${cyan}:!77?J7:                                                                          
                            ${blue}:7JJYYY55J: ${cyan}.~!!77!.                                                                            
                              ${blue}:?JJYJ:  ${cyan}.~~~!~.                                                                              

    "
}

print_installers() {

    print_bold \
    "          Your System is Ready to install Vaxtor Software ($ARCH $COMPONENT)           " "\
                                      
      * Need support? ${green}support@vaxtor.com${normal}

    "
    echo
    echo


}

print_bold() {
    title="$1"
    text="$2"

    echo
    echo "${cyan}================================================================================${normal}"
    echo "${cyan}================================================================================${normal}"
    echo
    echo -e "  ${bold}${yellow}${title}${normal}"
    echo
    echo -en "  ${text}"
    echo
    echo "${cyan}================================================================================${normal}"
    echo "${cyan}================================================================================${normal}"
}

print_warm() {
    title="$1"
    text="$2"

    echo
    echo "${yellow}================================================================================${normal}"
    echo -e "                                   ${bold}${yellow}WARNING !${normal}"
    echo "${yellow}================================================================================${normal}"
    echo
    echo -e "  ${bold}${yellow}${title}${normal}"
    echo
    echo -en "  ${text}"
    echo
    echo "${yellow}================================================================================${normal}"
    echo "${yellow}================================================================================${normal}"
}

bail() {
    echo 'Error executing command, exiting'
    exit 1
}

exec_cmd_nobail() {
    echo "+ $1"
    bash -c "$1"
}

exec_cmd() {
    exec_cmd_nobail "$1" || bail
}

unsupported() {
    echo
    echo
    echo "   ${red}! Your distribution is not supported, exiting${normal}"
    echo
    exit 1

}


script_welcome() {

print_logo

print_bold \
"                            VAXTOR INSTALLER                            " "\
                             
                               
  * ${bold}Welcome to Vaxtor Linux Installer${normal}

  * Your system will prepare to install Vaxtor's software.
  
  * Need support? ${green}support@vaxtor.com${normal}

"
echo
echo

if [[ "$ARCH" == "x86_64" ]]; then
  if ! grep -q avx2 /proc/cpuinfo ; then
    print_warm \
"                            NO AVX2 SUPPORT                            " "\

  * Your system do not support AVX2.
  
  * Component ${green}$COMPONENT${normal} was selected but will be replaced by ${green}nosimd${normal}

"
    COMPONENT="nosimd"
  fi
fi


}

setup() {

script_welcome

PRE_INSTALL_PKGS=""

# Check that HTTPS transport is available to APT
# (Check snaked from: https://get.docker.io/ubuntu/)

if [ ! -e /usr/lib/apt/methods/https ]; then
    PRE_INSTALL_PKGS="${PRE_INSTALL_PKGS} apt-transport-https"
fi

if [ ! -x /usr/bin/lsb_release ]; then
    PRE_INSTALL_PKGS="${PRE_INSTALL_PKGS} lsb-release"
fi

if [ ! -x /usr/bin/curl ] && [ ! -x /usr/bin/wget ]; then
    PRE_INSTALL_PKGS="${PRE_INSTALL_PKGS} curl"
fi

# Used by apt-key to add new keys

if [ ! -x /usr/bin/gpg ]; then
    PRE_INSTALL_PKGS="${PRE_INSTALL_PKGS} gnupg"
fi

# Populating Cache
print_status "Populating apt-get cache..."
exec_cmd 'apt-get update'

if [ "X${PRE_INSTALL_PKGS}" != "X" ]; then
    print_status "Installing packages required for setup:${PRE_INSTALL_PKGS}..."
    # This next command needs to be redirected to /dev/null or the script will bork
    # in some environments
    exec_cmd "apt-get install -y${PRE_INSTALL_PKGS} > /dev/null 2>&1"
fi

DISTRO=$(lsb_release -c -s)

check_alt() {
    if [ "X${DISTRO}" == "X${2}" ]; then
        echo
        echo "## You seem to be using ${1} version ${DISTRO}."
        echo "## This maps to ${3} \"${4}\"... Adjusting for you..."
        DISTRO="${4}"
    fi
}


check_alt "Astra Linux"    "orel"            "Debian"        "stretch"
check_alt "BOSS"           "anokha"          "Debian"        "wheezy"
check_alt "BOSS"           "anoop"           "Debian"        "jessie"
check_alt "BOSS"           "drishti"         "Debian"        "stretch"
check_alt "BOSS"           "unnati"          "Debian"        "buster"
check_alt "BOSS"           "urja"            "Debian"        "bullseye"
check_alt "bunsenlabs"     "bunsen-hydrogen" "Debian"        "jessie"
check_alt "bunsenlabs"     "helium"          "Debian"        "stretch"
check_alt "bunsenlabs"     "lithium"         "Debian"        "buster"
check_alt "Devuan"         "jessie"          "Debian"        "jessie"
check_alt "Devuan"         "ascii"           "Debian"        "stretch"
check_alt "Devuan"         "beowulf"         "Debian"        "buster"
check_alt "Devuan"         "chimaera"        "Debian"        "bullseye"
check_alt "Devuan"         "ceres"           "Debian"        "sid"
check_alt "Deepin"         "panda"           "Debian"        "sid"
check_alt "Deepin"         "unstable"        "Debian"        "sid"
check_alt "Deepin"         "stable"          "Debian"        "buster"
check_alt "Deepin"         "apricot"         "Debian"        "buster"
check_alt "elementaryOS"   "luna"            "Ubuntu"        "precise"
check_alt "elementaryOS"   "freya"           "Ubuntu"        "trusty"
check_alt "elementaryOS"   "loki"            "Ubuntu"        "xenial"
check_alt "elementaryOS"   "juno"            "Ubuntu"        "bionic"
check_alt "elementaryOS"   "hera"            "Ubuntu"        "bionic"
check_alt "elementaryOS"   "odin"            "Ubuntu"        "focal"
check_alt "elementaryOS"   "jolnir"          "Ubuntu"        "focal"
check_alt "Kali"           "sana"            "Debian"        "jessie"
check_alt "Kali"           "kali-rolling"    "Debian"        "bullseye"
check_alt "Linux Mint"     "maya"            "Ubuntu"        "precise"
check_alt "Linux Mint"     "qiana"           "Ubuntu"        "trusty"
check_alt "Linux Mint"     "rafaela"         "Ubuntu"        "trusty"
check_alt "Linux Mint"     "rebecca"         "Ubuntu"        "trusty"
check_alt "Linux Mint"     "rosa"            "Ubuntu"        "trusty"
check_alt "Linux Mint"     "sarah"           "Ubuntu"        "xenial"
check_alt "Linux Mint"     "serena"          "Ubuntu"        "xenial"
check_alt "Linux Mint"     "sonya"           "Ubuntu"        "xenial"
check_alt "Linux Mint"     "sylvia"          "Ubuntu"        "xenial"
check_alt "Linux Mint"     "tara"            "Ubuntu"        "bionic"
check_alt "Linux Mint"     "tessa"           "Ubuntu"        "bionic"
check_alt "Linux Mint"     "tina"            "Ubuntu"        "bionic"
check_alt "Linux Mint"     "tricia"          "Ubuntu"        "bionic"
check_alt "Linux Mint"     "ulyana"          "Ubuntu"        "focal"
check_alt "Linux Mint"     "ulyssa"          "Ubuntu"        "focal"
check_alt "Linux Mint"     "uma"             "Ubuntu"        "focal"
check_alt "Linux Mint"     "una"             "Ubuntu"        "focal"
check_alt "Linux Mint"     "vanessa"         "Ubuntu"        "jammy"
check_alt "Linux Mint"     "vera"            "Ubuntu"        "jammy"
check_alt "Liquid Lemur"   "lemur-3"         "Debian"        "stretch"
check_alt "LMDE"           "betsy"           "Debian"        "jessie"
check_alt "LMDE"           "cindy"           "Debian"        "stretch"
check_alt "LMDE"           "debbie"          "Debian"        "buster"
check_alt "LMDE"           "elsie"           "Debian"        "bullseye"
check_alt "MX Linux 17"    "Horizon"         "Debian"        "stretch"
check_alt "MX Linux 18"    "Continuum"       "Debian"        "stretch"
check_alt "MX Linux 19"    "patito feo"      "Debian"        "buster"
check_alt "MX Linux 21"    "wildflower"      "Debian"        "bullseye"
check_alt "Pardus"         "onyedi"          "Debian"        "stretch"
check_alt "Parrot"         "ara"             "Debian"        "bullseye"
check_alt "PureOS"         "green"           "Debian"        "sid"
check_alt "PureOS"         "amber"           "Debian"        "buster"
check_alt "PureOS"         "byzantium"       "Debian"        "bullseye"
check_alt "SolydXK"        "solydxk-9"       "Debian"        "stretch"
check_alt "Sparky Linux"   "Tyche"           "Debian"        "stretch"
check_alt "Sparky Linux"   "Nibiru"          "Debian"        "buster"
check_alt "Sparky Linux"   "Po-Tolo"         "Debian"        "bullseye"
check_alt "Tanglu"         "chromodoris"     "Debian"        "jessie"
check_alt "Trisquel"       "toutatis"        "Ubuntu"        "precise"
check_alt "Trisquel"       "belenos"         "Ubuntu"        "trusty"
check_alt "Trisquel"       "flidas"          "Ubuntu"        "xenial"
check_alt "Trisquel"       "etiona"          "Ubuntu"        "bionic"
check_alt "Ubilinux"       "dolcetto"        "Debian"        "stretch"
check_alt "Uruk GNU/Linux" "lugalbanda"      "Ubuntu"        "xenial"


print_status "Confirming \"${DISTRO}\" is supported..."

REPO_URL="${REPOSITORY_BASE}${DISTRO}-${COMPONENT}"

if [ -x /usr/bin/curl ]; then
    exec_cmd_nobail "curl -sLf -o /dev/null '${REPO_URL}/dists/${DISTRO}/Release'"
    RC=$?
else
    exec_cmd_nobail "wget -qO /dev/null -o /dev/null '${REPO_URL}/dists/${DISTRO}/Release'"
    RC=$?
fi

if [[ $RC != 0 ]]; then
    print_status "Your distribution, identified as \"${DISTRO}\", is not currently supported, please contact Vaxtor Support at support@vaxtor.com if you think this is incorrect or would like your distribution to be considered for support"
    exit 1
fi



print_status "Adding Vaxtor keyring..."
keyring='/etc/apt/trusted.gpg.d'
key_url="https://nexus.vaxtor.com/repository/keys/keyring.gpg"
local_key="$keyring/vaxtor-keyring.gpg"

if [ -x /usr/bin/curl ]; then
    exec_cmd "curl -s $key_url | tee $local_key >/dev/null"
else
    exec_cmd "wget -q -O - $key_url | tee $local_key >/dev/null"
fi

if [ -f "/etc/apt/sources.list.d/vaxtor-${DISTRO}.list" ]; then
    print_status 'Removing Vaxtor Repository...'

    exec_cmd "rm -f /etc/apt/sources.list.d/vaxtor-${DISTRO}.list"
fi

print_status "Creating Vaxtor apt source list repo..."

exec_cmd "echo deb [signed-by=${local_key}] ${REPOSITORY_BASE}${DISTRO}-${COMPONENT} ${DISTRO} main > /etc/apt/sources.list.d/vaxtor-${DISTRO}.list"

print_status 'Running `apt-get update`...'

exec_cmd 'apt-get update'


print_status "Done !"

print_installers

}


## Defer setup until we have the complete script
setup