provision.bash

#!/bin/bash

set -e  # Exit script if any error

# The shared folder is specified in Vagrantfile.
VAGRANT_SHARED_FOLDER=/home/vagrant/desktop

source $VAGRANT_SHARED_FOLDER/sqlflow/docker/dev/find_fastest_resources.sh

echo "Setting apt-get mirror..."
rm -rf /var/lib/apt/lists/* /etc/apt/sources.list
find_fastest_apt_source >/etc/apt/sources.list
apt-get update

echo "Installing Docker ..."
# c.f. https://dockr.ly/3cExcay
if which docker > /dev/null; then
    echo "Docker had been installed. Skip."
else
    best_install_url=$(find_fastest_docker_url)
    docker_ce_mirror=$(find_fastest_docker_ce_mirror)
    echo "Using ${best_install_url}..."
    curl -sSL "${best_install_url}" | DOWNLOAD_URL=$docker_ce_mirror bash -
    best_docker_mirror=$(find_fastest_docker_registry)
    if [[ -n "${best_docker_mirror}" ]]; then
        mkdir -p /etc/docker
        cat <<-EOF >/etc/docker/daemon.json
	{ 
	  "graph": "/mnt/docker-data",
	  "storage-driver": "overlay",
	  "registry-mirrors":[ "${best_docker_mirror}" ]
	}
	EOF
    fi
    usermod -aG docker vagrant
fi
echo "Done."

echo "Install axel ..."
if which axel > /dev/null; then
    echo "axel installed. Skip."
else
    $VAGRANT_SHARED_FOLDER/sqlflow/scripts/travis/install_axel.sh
fi

echo "Export Kubernetes environment variables ..."
# NOTE: According to https://stackoverflow.com/a/16619261/724872,
# source is very necessary here.
source $VAGRANT_SHARED_FOLDER/sqlflow/scripts/travis/export_k8s_vars.sh

echo "Installing kubectl ..."
if which kubectl > /dev/null; then
    echo "kubectl installed. Skip."
else
    $VAGRANT_SHARED_FOLDER/sqlflow/scripts/travis/install_kubectl.sh
fi
echo "Done."

echo "Installing minikube ..."
if which minikube > /dev/null; then
    echo "minikube installed. Skip."
else
    $VAGRANT_SHARED_FOLDER/sqlflow/scripts/travis/install_minikube.sh
    minikube config set WantUpdateNotification false
fi
echo "Done."

echo "Copy files ..."
# In non-develop mode, we want the user see the start.bash
# immediately after she/he logs on the vm
cp "$VAGRANT_SHARED_FOLDER/start.bash" "/root/"

read -r -d '\t' files <<EOM
sqlflow/scripts/travis/export_k8s_vars.sh
sqlflow/docker/dev/find_fastest_resources.sh
sqlflow/scripts/travis/start_argo.sh
sqlflow/doc/run/k8s/install-sqlflow.yaml
\t
EOM

mkdir -p "/root/scripts"
for file in ${files[@]}; do
    cp "$VAGRANT_SHARED_FOLDER/$file" "/root/scripts/$(basename $file)"
done
echo "Done."

echo "Change root password ..."
echo "root:sqlflow" | chpasswd
sed -i -e 's/^PasswordAuthentication no/PasswordAuthentication yes/g' \
    -e 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/g' \
    /etc/ssh/sshd_config
service ssh restart
echo "Done."


echo "Disable cloudimg grub settings ..."
sed -i -e 's/^GRUB_CMDLINE_LINUX_DEFAULT="quiet splash"/GRUB_CMDLINE_LINUX_DEFAULT="quiet"/g' \
    -e 's/^#GRUB_TERMINAL=console$/GRUB_TERMINAL=console/g' /etc/default/grub
rm -f /etc/default/grub.d/50-cloudimg-settings.cfg
update-grub
echo "Done."