jupyter.bash

#!/bin/bash

USERNAME=$1
OS_TYPE=$(uname)

if [ $# -eq 0 ]; then
    echo "Please pass your USERNAME as the argument"
    exit
fi

CMD1="jupyter-notebook --no-browser --port=7777 &"
CMD2="ssh -N -f -L localhost:7777:localhost:7777 $USERNAME@anvil.sv.vt.edu"

# ssh to Anvil, and check if a jupyter notebook is 
# already running under your name

var=$(ssh $USERNAME@128.173.49.135 ps -ef | grep jupyter | grep $USERNAME | awk '{print $1, $11}')

# If jupyter is not running, start a new notebook and background it
if [ -z "$var" ]; then
    echo Starting a jupyter notebook
    ssh -f $USERNAME@128.173.49.135 $CMD1
fi

# Open a new terminal, and create a ssh tunnel
# Kill any localhost process runnig on port 7777

pkill -f localhost
xterm -e $CMD2

# Open web browser on localhost and port 7777
# Check OS type first

if [ $OS_TYPE == 'Darwin' ]; then
    open http://localhost:7777
elif [ $OS_TYPE == 'Linux' ]; then 
    xdg-open http://localhost:7777
else
    echo Open your local browser, and type \"localhost:7777\"
fi