#!/bin/bash

#variant to use conda insted of pip, not tested

#Src: https://medium.com/swlh/how-to-install-jupyterhub-using-conda-without-runing-as-root-and-make-it-a-service-59b843fead12

set -e

if [ "$EUID" -ne 0 ]
  then echo "Please run as root"
  exit
fi

#apt-get install -y --no-install-recommends libnvinfer7=7.1.3-1+cuda11.0 \
#    libnvinfer-dev=7.1.3-1+cuda11.0 \
#    libnvinfer-plugin7=7.1.3-1+cuda11.0

#add public ssh keys
if [ ! -s "~/.ssh/authorized_keys" ]; then
  cat id_*.pub >> ~/.ssh/authorized_keys
fi

echo ""
echo "Install supermicro system tools..."
apt-get update
apt-get install -y ipmitool && modprobe ipmi_devintf

#echo "Install a recent notejs for jupyterhub from nodesource..."
#curl -fsSL https://deb.nodesource.com/setup_current.x | bash -
#apt-get install -y nodejs npm

echo "Install anaconda..."
wget -O /tmp/anaconda3.sh https://repo.anaconda.com/archive/Anaconda3-2022.05-Linux-x86_64.sh
bash /tmp/anaconda3.sh

echo "Install JupyterHub..."
apt-get install -y language-pack-de
conda install --yes jupyterhub jupyterlab notebook configurable-http-proxy jupyterlab-git jupyter-server-proxy
#jupyter labextension install jupyter-matplotlib

echo "Setup JupyterHub..."
mkdir /etc/jupyterhub/
cp jupyterhub_config.py /etc/jupyterhub/jupyterhub_config.py
cp jupyterhub.service /etc/systemd/system/jupyterhub.service
systemctl enable jupyterhub
systemctl start jupyterhub

echo "Install Tensorflow..."
conda install --yes tensorflow-gpu

echo "Install cantera..."
conda install --yes --channel cantera cantera

echo "Install common Python libs..."
conda install --yes h5py tqdm pandas ipywidgets pillow matplotlib ipympl opencv-python

#System monitoring
#echo "Install netdata..."
#wget -O /tmp/netdata-kickstart.sh https://my-netdata.io/kickstart.sh && sh /tmp/netdata-kickstart.sh
#echo "nvidia_smi: yes" >> /usr/lib/netdata/conf.d/python.d.conf
#service netdata restart
#on http://0.0.0.0:19999

echo "Install gpuview..."
python3 -m pip install gpuview
gpuview service --host 127.0.0.1 --port 9988

echo "Disable services that delays startup by wating for network connectivity..."
systemctl disable cloud-config
systemctl disable cloud-final.service
systemctl disable iscsid.service
systemctl disable open-iscsi.service


echo "show gpu state..."
nvidia-smi

echo ""
echo "Setup successfully finished"