setup_local.sh

#!/bin/bash

# -----------------------------
# BE CAREFUL EDITING THIS FILE:
# this script will be sourced, which means you can set env variables just once. 
# Further modifications won't have effect
# -----------------------------

jnr_path="$HOME/.RemoteJupyter"
version="1.0.0"

usage () {
    echo "Usage"
    echo "Standard usage:"
    echo "    source setup_local.sh"
    echo "Specify alias different from jnr: "
    echo "    source setup_local.sh -a <alias>"
    echo 
}


# parse arguments
while getopts ":ha:" opt; do
  	case $opt in
    	a) jnr_alias="$OPTARG"
    	;;
        h) usage 
        return 0
        ;;
    	\?) echo "Invalid option -$OPTARG" >&2
    	return 1
    	;;
  	esac
done

# set default alias if not provided
if [[ -z "$jnr_alias" ]] ; then
    jnr_alias="jnr"
fi

# check if there is already a setup file, which means we just need to update
if [[ ! -f "$jnr_path/setup.txt" ]] ; then
    # check if given alias already exists. TODO: iprove this block
    if command -v "$jnr_alias" &> /dev/null ; then
        echo "command $jnr_alias is already in use"
        echo "Please specify another alias for RemoteJupyter by running"
        echo "    source setup_local.sh -a <alias>"
        return 1
    fi
fi

# identify the type of shell
case $SHELL in
*/zsh) 
    rc_file="$HOME/.zshrc"
    ;;
*/bash)
    rc_file="$HOME/.bashrc"
    ;;
*)
    echo "Fatal: Unable to recognize shell type"
    return 1
esac

# create .RemoteJupyter folder if it doesn't exist
if [[ ! -d "$jnr_path" ]] ; then
	mkdir "$jnr_path"
fi

# create a setup file
setup_date=`date`
echo "$jnr_alias~$setup_date --- version=$version" >> "$jnr_path/setup.txt"

# copy jnr.sh into jnr_path
cp jnr.sh "$jnr_path"

# allow jnr.sh to be executed
chmod u+x "$jnr_path/jnr.sh"

# make a copy of the original rc file
cp "$rc_file" "$rc_file.pre_jnr"

# read the rc file and remove the old region where the aliases are defined
found=false
copy=true
while IFS='' read -r line ; do
    if [[ "$line" == *RemoteJupyter ]] ; then
        if [[ "$copy" == true ]] ; then
            copy=false
            found=true
        else
            copy=true
            echo "# >> RemoteJupyter" >> "$rc_file.tmp"
            echo "alias $jnr_alias='$jnr_path/jnr.sh'" >> "$rc_file.tmp"
        fi
    fi

    if [[ "$copy" == true ]] ; then
        echo "$line" >> "$rc_file.tmp"
    fi
    
done < "$rc_file"
mv "$rc_file.tmp" "$rc_file"

if [[ "$found" == false ]] ; then
    # create alias in rc file
    echo >> "$rc_file"
    echo "# >> RemoteJupyter" >> "$rc_file"
    echo "alias $jnr_alias='$jnr_path/jnr.sh'" >> "$rc_file"
    echo "# << RemoteJupyter" >> "$rc_file"
    echo >> "$rc_file"
fi

# Tell the user how to proceed
echo
echo "RemoteJupyter successfully setup with alias $jnr_alias"
echo "Restart the terminal or run"
echo "    source $rc_file"
echo "to start using."

if [[ "$found" == false ]] ; then
    echo "To obtain instruction on how to use, just type"
    echo "   $jnr_alias"
    echo
    echo "Run"
    echo "    source setup_remote.sh"
    echo "to setup the aliases also in the remote machine"
fi
echo
echo

return 0