#/usr/bin/env bash
#
# jill.sh
# Copyright (C) 2017-2020 Abel Soares Siqueira <abel.s.siqueira@gmail.com>
#
# Distributed under terms of the GPLv3 license.
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# Skip confirm if -y is used.
SKIP_CONFIRM=0
# Copy over the old environment to the new one if -u is used.
UPGRADE_CONFIRM=0
JULIA_OLD=""
while getopts ":yu:" opt; do
  case $opt in
    y)
      SKIP_CONFIRM=1
      ;;
    u)
      UPGRADE_CONFIRM=1
      JULIA_OLD=${OPTARG}
      ;;
    \?)
      echo "Invalid option: -$OPTARG" >&2
      exit 1;
      ;;
  esac
done

JULIA_LATEST=1.5

# For Linux, this script installs Julia into $JULIA_DOWNLOAD and make a
# link to $JULIA_INSTALL
# For MacOS, this script installs Julia into /Applications and make a
# link to $JULIA_INSTALL
if [[ "$(whoami)" == "root" ]]; then
  JULIA_DOWNLOAD=${JULIA_DOWNLOAD:-"/opt/julias"}
  JULIA_INSTALL=${JULIA_INSTALL:-"/usr/local/bin"}
else
  JULIA_DOWNLOAD=${JULIA_DOWNLOAD:-"$HOME/packages/julias"}
  JULIA_INSTALL=${JULIA_INSTALL:-"$HOME/.local/bin"}
fi
WGET="wget --retry-connrefused -t 3"

function header() {
  echo "JILL - Julia Installer 4 Linux (and MacOS) - Light"
  echo "Copyright (C) 2017 Abel Soares Siqueira <abel.s.siqueira@gmail.com>"
  echo "Distributed under terms of the GPLv3 license."
}

function badfolder() {
  echo "The folder '$JULIA_INSTALL' is not on your PATH, you can"
  echo "- 1) Add it to your path; or"
  echo "- 2) Run 'JULIA_INSTALL=otherfolder ./jill.sh'"
  if [[ "$SKIP_CONFIRM" == "0" ]]; then
    read -p "Do you want to add '$JULIA_INSTALL' into your PATH? (Y/N) " -n 1 -r
    echo
    if [[ ! $REPLY =~ ^[Yy] ]]; then
      echo "Aborted"
      exit 1
    fi
  fi
  echo 'export PATH="'"$JULIA_INSTALL"':$PATH"' | tee -a ~/.bashrc
  echo ""
  echo "run 'source ~/.bashrc' or restart your bash to reload the PATH"
  echo ""
}

function hi() {
  header
  if [[ ! ":$PATH:" == *":$JULIA_INSTALL:"* ]]; then
    badfolder
  fi
  mkdir -p $JULIA_INSTALL # won't create if it's aborted earlier
  echo "This script will:"
  echo ""
  # TODO: Expand to install older Julia?
  echo "  - Download latest stable Julia"
  echo "  - Create a link for julia"
  echo "  - Create a link for julia-VER"
  echo ""
  echo "Download folder: $JULIA_DOWNLOAD"
  echo "Link folder: $JULIA_INSTALL"
  echo ""
  if [ ! -d $JULIA_DOWNLOAD ]; then
    echo "Download folder will be created if required"
  fi
  if [ ! -w $JULIA_INSTALL ]; then
    echo "You don't have write permission to $JULIA_INSTALL."
    exit 1
  fi
}

function confirm() {
  read -p "Do you accept these terms? (Y/N) " -n 1 -r
  echo
  if [[ ! $REPLY =~ ^[Yy] ]]; then
    echo "Aborted"
    exit 1
  fi
}

function get_url_from_platform_arch_version() {
  platform=$1
  arch=$2
  version=$3
  # TODO: Accept ARM and FreeBSD
  [[ $arch == *"64" ]] && bit=64 || bit=32
  [[ $arch == "mac"* ]] && suffix=mac64.dmg || suffix=$platform-$arch.tar.gz
  minor=$(echo $version | cut -d. -f1-2 | cut -d- -f1)
  url=https://julialang-s3.julialang.org/bin/$platform/x$bit/$minor/julia-$version-$suffix
  echo $url
}

function install_julia_linux() {
  mkdir -p $JULIA_DOWNLOAD
  cd $JULIA_DOWNLOAD
  arch=$(uname -m)

  # Download specific version if requested
  if [ -n "${JULIA_VERSION+set}" ]; then
    version=$JULIA_VERSION
  else
    version=$JULIA_LATEST-latest
  fi
  echo "Downloading Julia version $version"
  if [ ! -f julia-$version.tar.gz ]; then
    url=$(get_url_from_platform_arch_version linux $arch $version)
    $WGET -c $url -O julia-$version.tar.gz
  else
    echo "already downloaded"
  fi
  if [ ! -d julia-$version ]; then
    mkdir -p julia-$version
    tar zxf julia-$version.tar.gz -C julia-$version --strip-components 1
  fi

  major=${version:0:3}
  rm -f $JULIA_INSTALL/julia{,-$major,-$version}
  julia=$PWD/julia-$version/bin/julia

  if [[ "$UPGRADE_CONFIRM" == "1" ]]; then
    old_major=${JULIA_OLD:0:3}
    cp -r ~/.julia/environments/v${old_major} ~/.julia/environments/v${major}
  fi

  # create symlink
  ln -s $julia $JULIA_INSTALL/julia
  ln -s $julia $JULIA_INSTALL/julia-$major
  ln -s $julia $JULIA_INSTALL/julia-$version
}

function install_julia_mac() {
  mkdir -p $JULIA_DOWNLOAD
  cd $JULIA_DOWNLOAD
  arch="mac64"

  # Download specific version if requested
  if [ -n "${JULIA_VERSION+set}" ]; then
    version=$JULIA_VERSION
  else
    version=$JULIA_LATEST-latest
  fi
  if [ ! -f julia-$version.dmg ]; then
    url=$(get_url_from_platform_arch_version mac $arch $version)
    $WGET -c $url -O julia-$version.dmg
  fi

  major=${version:0:3}

  hdiutil attach julia-$version.dmg -quiet -mount required -mountpoint julia-$version

  INSTALL_PATH=/Applications/julia-$major.app
  EXEC_PATH=$INSTALL_PATH/Contents/Resources/julia/bin/julia
  rm -rf $INSTALL_PATH
  cp -a julia-$version/Julia-$major.app /Applications/

  if [[ "$UPGRADE_CONFIRM" == "1" ]]; then
    old_major=${JULIA_OLD:0:3}
    cp -r ~/.julia/environments/v${old_major} ~/.julia/environments/v${major}
  fi

  # create symlink
  ln -sf $EXEC_PATH $JULIA_INSTALL/julia
  ln -sf $EXEC_PATH $JULIA_INSTALL/julia-$major
  ln -sf $EXEC_PATH $JULIA_INSTALL/julia-$version

  # post-installation
  umount julia-$version
}

# --------------------------------------------------------

# Main
hi
if [[ "$SKIP_CONFIRM" == "0" ]]; then
    confirm
fi
unameOut="$(uname -s)"
case "${unameOut}" in
    Linux*) install_julia_linux ;;
    Darwin*) install_julia_mac ;;
    # CYGWIN*)    machine=Cygwin;;
    # MINGW*)     machine=MinGw;;
    *)
        echo "Unsupported platform $(unameOut)" >&2
        exit 1
        ;;
esac