export PIP_PACKAGE_NAME="ipywidgets"
export EXTENSION_NAME="widgetsnbextension"
#---
echo "Downloading on-start.sh..."
# save the existing on-start script into on-start.sh
aws sagemaker describe-notebook-instance-lifecycle-config --notebook-instance-lifecycle-config-name "$CONFIGURATION_NAME" | jq '.OnStart[0].Content'  | tr -d '"' | base64 --decode > on-start.sh
#---
echo "Adding extenstions install to on-start.sh..."
echo '' >> on-start.sh
echo '# install jupyter extension' >> on-start.sh
echo "export PIP_PACKAGE_NAME=\"${PIP_PACKAGE_NAME}\"" >> on-start.sh
echo "export EXTENSION_NAME=\"${EXTENSION_NAME}\"" >> on-start.sh
echo 'curl https://raw.githubusercontent.com/mariokostelac/sagemaker-setup/master/scripts/install-nb-extension/on-start.sh | bash' >> on-start.sh
#---
echo "Uploading on-start.sh..."
# update the lifecycle configuration config with updated on-start.sh script
aws sagemaker update-notebook-instance-lifecycle-config \
    --notebook-instance-lifecycle-config-name "$CONFIGURATION_NAME" \
    --on-start Content="$((cat on-start.sh)| base64)"