Merge pull request #11 from sql-machine-learning/split_provision_and_start.sh

Split provision and start.sh

Author: wangkuiyi

Vagrantfile

@@ -3,20 +3,22 @@
 
 Vagrant.configure("2") do |config|
   config.vm.box = "ubuntu/bionic64"
-  config.vm.provision "shell", path: "provision.sh"
+  config.vm.provision "shell", path: "provision.bash"
 
   # Don't forward 22.  Even if we do so, the exposed port only binds
   # to 127.0.0.1, but not 0.0.0.0.  Other ports binds to all IPs.
-  config.vm.network "forwarded_port", guest: 3306, host: 3306
-  config.vm.network "forwarded_port", guest: 50051, host: 50051
-  config.vm.network "forwarded_port", guest: 8888, host: 8888
-
-  # public_network eases debugging -- I can SSH from iMac to the VM
-  # without the need of forwarding port to the host workstation.
-  config.vm.network "public_network", ip: "192.168.0.17"
+  config.vm.network "forwarded_port", guest: 3306, host: 3306,
+                    auto_correct: true
+  config.vm.network "forwarded_port", guest: 50051, host: 50051,
+                    auto_correct: true
+  config.vm.network "forwarded_port", guest: 8888, host: 8888,
+                    auto_correct: true
 
   config.vm.provider "virtualbox" do |v|
     v.memory = 16384
     v.cpus = 8
   end
+
+  # Bind the host directory ./ into the VM.
+  config.vm.synced_folder "./", "/home/vagrant/desktop"
 end

provision.bash

@@ -0,0 +1,55 @@
+#!/bin/bash
+
+set -e  # Exit script if any error
+
+echo "Installing Docker ..."
+# c.f. https://dockr.ly/3cExcay
+if which docker > /dev/null; then
+    echo "Docker had been installed. Skip."
+else
+    curl -fsSL https://get.docker.com | sh -
+    usermod -aG docker vagrant
+fi
+echo "Done."
+
+echo "Docker pull SQLFlow images ..."
+# c.f. https://github.com/sql-machine-learning/sqlflow/blob/develop/.travis.yml
+docker pull --quiet sqlflow/sqlflow:latest
+echo "Done."
+
+# The shared folder is specified in Vagrantfile.
+VAGRANT_SHARED_FOLDER=/home/vagrant/desktop
+
+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
+fi
+echo "Done."
+
+echo "Configure minikube ..."
+mkdir -p /home/vagrant/.kube /home/vagrant/.minikube
+touch /home/vagrant/.kube/config
+chown -R vagrant /home/vagrant/.bashrc
+echo "Done."

provision.sh

@@ -0,0 +1,24 @@
+#!/bin/bash
+# Copyright 2020 The SQLFlow Authors. All rights reserved.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+set -e
+
+# NOTE: According to https://stackoverflow.com/a/16619261/724872,
+# source is very necessary here.
+source $(dirname $0)/sqlflow/scripts/travis/export_k8s_vars.sh
+
+$(dirname $0)/sqlflow/scripts/travis/start_minikube.sh
+sudo chown -R vagrant: $HOME/.minikube/
+
+$(dirname $0)/sqlflow/scripts/travis/start_argo.sh