#!/bin/bash
set -e

# Install Codex on Linux, macOS, and Windows(msys2)

# Variables
VERSION=${VERSION:-latest}
INSTALL_CIRDL=${INSTALL_CIRDL:-false}
INSTALL_DIR=${INSTALL_DIR:-/usr/local/bin}
CODEX_ARCHIVE_PREFIX="codex"
CIRDL_ARCHIVE_PREFIX="cirdl"
CODEX_BINARY_PREFIX="codex"
CIRDL_BINARY_PREFIX="cirdl"
WINDOWS_LIBS=${WINDOWS_LIBS:-false}
WINDOWS_LIBS_LIST="libstdc++-6.dll libgomp-1.dll libgcc_s_seh-1.dll libwinpthread-1.dll"
BASE_URL="https://github.com/codex-storage/nim-codex"
API_BASE_URL="https://api.github.com/repos/codex-storage/nim-codex"
TEMP_DIR="${TEMP_DIR:-.}"
PROGRESS_MARK="\033[0;36m\u2022\033[0m"
PASS_MARK="\033[0;32m\u2714\033[0m"
FAIL_MARK="\033[0;31m\u2718\033[0m"

# Help
if [[ $1 == *"h"* ]] ; then
  echo -e "
  \e[33mUsage:\e[0m
    curl https://get.codex.storage/`basename $0` | bash
    curl https://get.codex.storage/`basename $0` | VERSION=0.1.7 bash
    curl https://get.codex.storage/`basename $0` | VERSION=0.1.7 INSTALL_CIRDL=true bash
    curl https://get.codex.storage/`basename $0` | bash -s help

  \e[33mOptions:\e[0m
    - help                       - show this help
    - VERSION=0.1.7              - codex and cird version to install
    - INSTALL_CIRDL=true         - install cirdl
    - INSTALL_DIR=/usr/local/bin - directory to install binaries
    - WINDOWS_LIBS=true          - download and install archive with libs for windows
  "
  exit 0
fi

# Show
show_start() {
  echo -e "\n \e[4m${1}\e[0m\n"
}

show_progress() {
  echo -e " ${PROGRESS_MARK} ${1}"
}

show_pass() {
  echo -e "\r\e[1A\e[0K ${PASS_MARK} ${1}"
}

show_fail() {
  echo -e "\r\e[1A\e[0K ${FAIL_MARK} ${1}"
  [[ -n "${2}" ]] && echo -e "\e[31m \n Error: ${2}\e[0m"
  exit 1
}

show_end() {
  echo -e "\n\e[32m ${1}\e[0m\n"
}

# Start
show_start "Installing Codex..."

# Version
message="Compute version"
show_progress "${message}"
[[ "${VERSION}" == "latest" ]] && VERSION=$(curl -s ${API_BASE_URL}/releases/latest | grep tag_name | cut -d '"' -f 4) || VERSION="v${VERSION}"
[[ $? -eq 0 ]] && show_pass "${message}" || show_fail "${message}"

# Archives and binaries
message="Compute archives and binaries names"
show_progress "${message}"
[[ "${INSTALL_CIRDL}" == "true" ]] && ARCHIVES=("${CODEX_ARCHIVE_PREFIX}" "${CIRDL_ARCHIVE_PREFIX}") || ARCHIVES=("${CODEX_ARCHIVE_PREFIX}")
[[ "${INSTALL_CIRDL}" == "true" ]] && BINARIES=("${CODEX_BINARY_PREFIX}" "${CIRDL_BINARY_PREFIX}") || BINARIES=("${CODEX_BINARY_PREFIX}")
show_pass "${message}"

# Get the current OS
message="Checking the current OS"
show_progress "${message}"
case "$(uname -s)" in
  Linux*)               OS="linux"                                          ;;
  Darwin*)              OS="darwin"                                         ;;
  CYGWIN*|MINGW*|MSYS*) OS="windows"                                        ;;
  *)                    show_fail "${message}" "Unsupported OS $(uname -s)" ;;
esac
[[ $? -eq 0 ]] && show_pass "${message}" || show_fail "${message}"

# Get the current architecture
message="Checking the current architecture"
show_progress "${message}"
case "$(uname -m)" in
  x86_64|amd64)  ARCHITECTURE="amd64"                                           ;;
  arm64|aarch64) ARCHITECTURE="arm64"                                           ;;
  *)             show_fail "${message}" "Unsupported architecture: $(uname -m)" ;;
esac
[[ $? -eq 0 ]] && show_pass "${message}" || show_fail "${message}"

# Not supported
if [[ "${OS}" == "windows" && "${ARCHITECTURE}" == "arm64" ]]; then
  show_fail "${message}" "Windows ${ARCHITECTURE} is not supported at the moment"
fi

# Prerequisites
message="Checking prerequisites"
show_progress "${message}"
if [[ ("${OS}" != "windows") ]]; then
  $(command -v tar &> /dev/null) || show_fail "${message}" "Please install tar to continue installation"
fi
show_pass "${message}"

# Archive and binaries names
if [[ "$OS" == "windows" ]]; then
  [[ "${WINDOWS_LIBS}" == "true" ]] && ARCHIVE_SUFFIX="${VERSION}-${OS}-${ARCHITECTURE}-libs.zip" || ARCHIVE_SUFFIX="${VERSION}-${OS}-${ARCHITECTURE}.zip"
  BINARY_SUFFIX="${VERSION}-${OS}-${ARCHITECTURE}"
else
  ARCHIVE_SUFFIX="${VERSION}-${OS}-${ARCHITECTURE}.tar.gz"
  BINARY_SUFFIX="${VERSION}-${OS}-${ARCHITECTURE}"
fi

# Download
for ARCHIVE in "${ARCHIVES[@]}"; do
  FILE_NAME="${ARCHIVE}-${ARCHIVE_SUFFIX}"

  for FILE in "${FILE_NAME}" "${FILE_NAME}.sha256"; do
    DOWNLOAD_URL="${BASE_URL}/releases/download/${VERSION}/${FILE}"

    message="Downloading ${FILE}"
    show_progress "${message}"
    http_code=$(curl --write-out "%{http_code}" --connect-timeout 5 --retry 5 -sL "${DOWNLOAD_URL}" -o "${TEMP_DIR}/${FILE}")
    [[ "${http_code}" -eq 200 ]] && show_pass "${message}" || show_fail "${message}" "Failed to download ${DOWNLOAD_URL}"
  done
done

# Checksum
for ARCHIVE in "${ARCHIVES[@]}"; do
  FILE_NAME="${ARCHIVE}-${ARCHIVE_SUFFIX}"
  message="Verifying checksum for ${FILE_NAME}"
  show_progress "${message}"

  EXPECTED_SHA256=$(cat "${TEMP_DIR}/${FILE_NAME}.sha256" | cut -d' ' -f1)
  if [[ "${OS}" == "darwin" ]]; then
    ACTUAL_SHA256=$(shasum -a 256 "${TEMP_DIR}/${FILE_NAME}" | cut -d ' ' -f 1)
  else
    ACTUAL_SHA256=$(sha256sum "${TEMP_DIR}/${FILE_NAME}" | cut -d ' ' -f 1)
  fi

  if [[ "$ACTUAL_SHA256" == "$EXPECTED_SHA256" ]]; then
    show_pass "${message}"
  else
    show_fail "${message}" "Checksum verification failed for ${FILE_NAME}. Expected: $EXPECTED_SHA256, Got: $ACTUAL_SHA256"
  fi
done

# Extract
for ARCHIVE in "${ARCHIVES[@]}"; do
  FILE_NAME="${ARCHIVE}-${ARCHIVE_SUFFIX}"

  message="Extracting ${FILE_NAME}"
  show_progress "${message}"

  if [[ "${OS}" == "windows" ]]; then
    if unzip -v &> /dev/null; then
      unzip -o "${TEMP_DIR}/${FILE_NAME}" -d "${TEMP_DIR}"
      [[ $? -ne 0 ]] && show_fail "${message}"
    else
      C:/Windows/system32/tar.exe -xzf "${TEMP_DIR}/${FILE_NAME}" -C "${TEMP_DIR}"
      [[ $? -ne 0 ]] && show_fail "${message}"
    fi
  else
    tar -xzf "${TEMP_DIR}/${FILE_NAME}" -C "${TEMP_DIR}"
    [[ $? -ne 0 ]] && show_fail "${message}"
  fi
  show_pass "${message}"
done

# Install
for BINARY in "${BINARIES[@]}"; do
  FILE_NAME="${BINARY}-${BINARY_SUFFIX}"
  INSTALL_PATH="${INSTALL_DIR}/${BINARY}"

  # Install
  message="Installing ${FILE_NAME} to ${INSTALL_PATH}"
  show_progress "${message}"
  if ! (mkdir -p "${INSTALL_DIR}" && install -m 755 "${TEMP_DIR}/${FILE_NAME}" "${INSTALL_PATH}") 2> /dev/null; then
    sudo mkdir -p "${INSTALL_DIR}" && sudo install -m 755 "${TEMP_DIR}/${FILE_NAME}" "${INSTALL_PATH}"
    [[ $? -ne 0 ]] && show_fail "${message}"
  fi
  show_pass "${message}"
done

# Windows libs
if [[ "${OS}" == "windows" && "${WINDOWS_LIBS}" == "true" ]]; then
  message="Copy libs to ${MINGW_PREFIX}/bin"
  show_progress "${message}"
  for LIB in ${WINDOWS_LIBS_LIST}; do
    mv "${TEMP_DIR}/${LIB}" "${MINGW_PREFIX}/bin"
  done
  [[ $? -eq 0 ]] && show_pass "${message}" || show_fail "${message}"
fi

# Cleanup
message="Cleanup"
show_progress "${message}"
for BINARY in "${BINARIES[@]}"; do
  FILE_NAME="${BINARY}-${BINARY_SUFFIX}"
  rm -f "${TEMP_DIR}/${FILE_NAME}"*
  [[ $? -ne 0 ]] && show_fail "${message}"
done
show_pass "${message}"

# End
show_end "Setup completed successfully!"

# Dependencies
dependencies=()
for BINARY in "${BINARIES[@]}"; do
  LOCATION="${INSTALL_DIR}/${BINARY}"
  case "${OS}" in
    linux)  dependencies+=($(ldd "${LOCATION}" | awk '/not found/ {print $1}'))      ;;
    darwin) dependencies+=($(otool -L "${LOCATION}" | awk '/not found/ {print $1}')) ;;
  esac
done

if [[ ${#dependencies[@]} -ne 0 ]]; then
  echo -e " Please inatall the following dpependencies:
  ${dependencies[@]}\n"
fi

# Path
[[ "${INSTALL_DIR}" == "." ]] && INSTALL_DIR=$(pwd)
if [[ $PATH != *"${INSTALL_DIR}"* ]]; then
  echo -e " Note: Please add install directory '"${INSTALL_DIR}"' to your PATH\n"
fi