#!/bin/bash
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 JupyterHub..."
apt-get install -y language-pack-de python3-pip
python3 -m pip install jupyterhub
npm install -g configurable-http-proxy
python3 -m pip install jupyterlab notebook jupyterlab-git
jupyter labextension install @jupyterlab/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..."
python3 -m pip install tensorflow-gpu

echo "Install cantera..."
apt-add-repository -y ppa:cantera-team/cantera
apt-get install -y cantera-python3

echo "Install common Python libs..."
python3 -m pip install h5py tqdm pandas ipywidgets pillow matplotlib ipympl opencv-python
#python3 -m pip install scipy scikit-learn numba

#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"