#!/usr/bin/env bash

set -e

VERSION=${1:-"latest"}
USERNAME=${2:-"automatic"}

# If in automatic mode, determine if a user already exists, if not use vscode
if [ "${USERNAME}" = "auto" ] || [ "${USERNAME}" = "automatic" ]; then
    USERNAME=""
    POSSIBLE_USERS=("vscode" "node" "codespace" "$(awk -v val=1000 -F ":" '$3==val{print $1}' /etc/passwd)")
    for CURRENT_USER in ${POSSIBLE_USERS[@]}; do
        if id -u ${CURRENT_USER} > /dev/null 2>&1; then
            USERNAME=${CURRENT_USER}
            break
        fi
    done
    if [ "${USERNAME}" = "" ]; then
        USERNAME=vscode
    fi
elif [ "${USERNAME}" = "none" ]; then
    USERNAME=root
    USER_UID=0
    USER_GID=0
fi

# Use sudo to run as non-root user is not already running
sudoUserIf()
{
  if [ "$(id -u)" -eq 0 ] && [ "${USERNAME}" != "root" ]; then
    sudo -u ${USERNAME} "$@"
  else
    "$@"
  fi
}

# If we don't yet have Python, install it now.
if ! python --version > /dev/null ; then
  echo "You need to install Python before installing JupyterLab."
  exit 1
fi

# If we don't already have JupyterLab installed, install it now.
if ! jupyter-lab --version > /dev/null ; then
  echo "Installing JupyterLab..."
  if [ "${VERSION}" = "latest" ]; then
    pip install jupyterlab
  else
    pip install jupyterlab=="${VERSION}" --no-cache-dir
  fi
fi