#!/bin/bash

# tipi CLI macos & linux installer script
# Copyright 2020 - () tipi technologies Ltd (Zürich)
#
# Environment variables
#
# - TIPI_INSTALL_VERSION:   overrides release to install - must match release tag 
# - TIPI_INSTALL_SOURCE:    override the source path of the release zip
#                           -> a file:// url can be used to install from a locally present zip

###
# Helper functions
###

abort() {
  printf " \e[91m $1 \n"
  exit 1
}

info() {
  printf "\e[1;32m -> \e[0m $1 \n"
}

should_install() {
 if [[ $(command -v $1) ]]; then
    return 1
  fi
  return 0
}

request_install_permission() {
  local cmd="$1"
  local reason="$2"

  info "The '$cmd' command is required ${reason}, we are installing $cmd with your package manager"
  echo "Could you validate with your password ? 😇 "
}

###
# impl.
### 

VERSION="${TIPI_INSTALL_VERSION:-v0.0.87}"
INSTALL_FOLDER=/usr/local

if [[ -z "${TIPI_INSTALL_SOURCE}" ]]; then
  if [ "$(uname)" == "Linux" ]; then
    TIPI_URL="https://github.com/tipi-build/cli/releases/download/$VERSION/tipi-$VERSION-linux-x86_64.zip" 
  elif [ "$(uname)" == "Darwin" ]; then
    TIPI_URL="https://github.com/tipi-build/cli/releases/download/$VERSION/tipi-$VERSION-macOS.zip"

    # test for M1/M2 
    if [[ $(uname -m) == 'arm64' ]]; then
      info "Tipi requires rosetta to run on Apple ARM Silicon. Launching installation, if you disagree with the Apple Rosetta license press Ctrl-C." 
      $PRIV_ELEV_CMD softwareupdate --install-rosetta --agree-to-license
    fi

  fi
else
  TIPI_URL="${TIPI_INSTALL_SOURCE}"
fi

PRIV_ELEV_CMD=''
if  command -v sudo &> /dev/null; then
  info "Note: using 'sudo' as priviledge elevation method if required during installation"
  PRIV_ELEV_CMD='sudo'
elif command -v doas &> /dev/null; then
  info "Note: using 'doas' as priviledge elevation method if required during installation"
  PRIV_ELEV_CMD='doas'
else
  info "Note: no supported priviledge elevation method (sudo or doas) found, trying to install without asking for elevation"
fi


if [ -f "/etc/redhat-release" ]; then
  if should_install unzip; then
    request_install_permission  "unzip" "to extract the downloaded file"
    $PRIV_ELEV_CMD yum install unzip -y || abort "Error while installing unzip"
  fi

  if should_install xz; then
    request_install_permission "xz" "for the installation of some of our tools"
    $PRIV_ELEV_CMD yum install xz -y || abort "Error while installing xz"
  fi

  if should_install bzip2; then
    request_install_permission "bzip2" "for the installation of some of our tools"
    $PRIV_ELEV_CMD yum install bzip2 -y || abort "Error while installing bzip2"
  fi

  if should_install which; then
    request_install_permission "which" "for the installation of some of our tools"
    $PRIV_ELEV_CMD yum install which -y || abort "Error while installing which"
  fi
fi

if [ -f "/etc/arch-release" ]; then
  pacman -Syu --noconfirm
  pacman -S --needed --noconfirm python
  pacman -S --needed --noconfirm unzip
  pacman -S --needed --noconfirm base-devel
  pacman -S --needed --noconfirm openssh
fi

if [ -f "/etc/lsb-release" ]; then
  if should_install unzip; then
      request_install_permission  "unzip" "to extract the downloaded file"
      $PRIV_ELEV_CMD apt-get update -y && apt-get install unzip -y || abort "Error while installing unzip"
  fi
fi

TMP_DOWNLOAD_PATH=~/tipi-$VERSION.zip

info "Downloading tipi from: $TIPI_URL"
info "Saving archive to: $TMP_DOWNLOAD_PATH"
curl -fSL $TIPI_URL --output $TMP_DOWNLOAD_PATH || wget -q $TIPI_URL -O $TMP_DOWNLOAD_PATH || abort "Could not download tipi"

info "Installing tipi in $INSTALL_FOLDER"
$PRIV_ELEV_CMD unzip -o $TMP_DOWNLOAD_PATH -d $INSTALL_FOLDER 'bin/*'

if [ $? -eq 0 ]; then
    tipi_full_path=$INSTALL_FOLDER/bin/tipi
    cmake_full_path=$INSTALL_FOLDER/bin/cmake-re

    for file in "$tipi_full_path" "$cmake_full_path"; do
      if [ -f "$file" ]; then
        $PRIV_ELEV_CMD chown "${USER:=$(id -run)}" "$file"
        $PRIV_ELEV_CMD chmod a+x,u+w "$file"
      fi
    done

    info "Cleaning up temporary download"
    rm $TMP_DOWNLOAD_PATH

    TIPI_DATA_DIR=/usr/local/share/.tipi
    info "Setting up tipi's local storage in $TIPI_DATA_DIR"
    $PRIV_ELEV_CMD mkdir -p $TIPI_DATA_DIR
    $PRIV_ELEV_CMD chown -R ${USER:=$(/usr/bin/id -run)} $TIPI_DATA_DIR

    info "Provisioning included tools."
    $INSTALL_FOLDER/bin/tipi -v --dont-upgrade run echo "[INFO] Shipping tools done"
    if [ $? -eq 0 ]; then
        info "tipi has been installed in $INSTALL_FOLDER/bin and can be used now."
        info "----------------------------"
        info "If you are new to tipi you can explore how to use it at: https://tipi.build/explore. If you are currently following the onboarding guide it is now time to get back to your browser: https://tipi.build/onboarding/step4"
    else 
        abort "Error while installing the dependencies"
    fi
else
    abort "Installation failed, please contact us on https://tipi.build : We are happy to help."
fi