Publish

1 files changed, 77 insertions, 0 deletions

diff --git a/node-admin/scripts/vm.sh b/node-admin/scripts/vm.sh
new file mode 100755
index 00000000000..19542a7c392
--- /dev/null
+++ b/node-admin/scripts/vm.sh
@@ -0,0 +1,77 @@
+#!/bin/bash
+# Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+set -e
+
+source "${0%/*}/common-vm.sh"
+
+DOCKER_VM_WAS_STARTED=false
+
+if ! docker-machine status "$DOCKER_VM_NAME" &> /dev/null; then
+  # Machine does not exist and we have to create and start
+  docker-machine create -d virtualbox \
+              --virtualbox-disk-size "$DOCKER_VM_DISK_SIZE_IN_MB" \
+              --virtualbox-memory "$DOCKER_VM_MEMORY_SIZE_IN_MB" \
+              --virtualbox-cpu-count "$DOCKER_VM_CPU_COUNT" \
+              --virtualbox-hostonly-cidr "$DOCKER_VM_HOST_CIDR" \
+              "$DOCKER_VM_NAME"
+
+  eval $(docker-machine env "$DOCKER_VM_NAME")
+
+  # Node admin expects different names for the certificates. Just symlink docker has 
+  # generated for us to match those in node-admin/src/main/application/services.xml.
+  (
+    cd "$DOCKER_CERT_PATH"
+    ln -s ca.pem ca_cert.pem
+    ln -s key.pem client_key.pem
+    ln -s cert.pem client_cert.pem
+  )
+  DOCKER_VM_WAS_STARTED=true
+fi
+
+
+VESPA_VM_STATUS=$(docker-machine status "$DOCKER_VM_NAME")
+if [ "$VESPA_VM_STATUS" == "Stopped" ]; then
+    docker-machine start "$DOCKER_VM_NAME"
+    DOCKER_VM_WAS_STARTED=true
+    VESPA_VM_STATUS=$(docker-machine status "$DOCKER_VM_NAME")
+fi
+
+if [ "$VESPA_VM_STATUS" != "Running" ]; then
+  echo "Unable to get Docker machine $DOCKER_VM_NAME up and running."
+  echo "You can try to manually remove the machine: docker-machine rm -y $DOCKER_VM_NAME "
+  echo "  and then rerun this script."
+  echo "Exiting."
+  exit 1
+fi
+
+if $DOCKER_VM_WAS_STARTED; then
+  # Put anything that is not persisted between VM restarts in here.
+  # Set up NAT for the $HOST_BRIDGE_INTERFACE interface so that we can connect directly from OS X.
+  docker-machine ssh "$DOCKER_VM_NAME" sudo /usr/local/sbin/iptables -t nat -A POSTROUTING -s "$HOST_BRIDGE_NETWORK" ! -o "$HOST_BRIDGE_INTERFACE" -j MASQUERADE
+  docker-machine ssh "$DOCKER_VM_NAME" sudo /usr/local/sbin/iptables -A FORWARD -o "$HOST_BRIDGE_INTERFACE" -m conntrack --ctstate RELATED,ESTABLISHED -j ACCEPT
+
+  # Install dependencies used by setup scripts
+  docker-machine ssh "$DOCKER_VM_NAME" tce-load -wi python bash
+fi
+
+# Get the environment for our VM
+eval $(docker-machine env "$DOCKER_VM_NAME")
+
+if [ $# -ge 1 ]; then
+  declare -r ARG_SCRIPT=$1
+  shift
+
+  declare -r ARG_SCRIPT_BASE=$(basename "$ARG_SCRIPT")
+  declare -r ARG_SCRIPT_DIR=$(cd $(dirname "$ARG_SCRIPT") && pwd -P)
+  declare -r ARG_SCRIPT_ABS="$ARG_SCRIPT_DIR/$ARG_SCRIPT_BASE"
+
+  if ! docker-machine ssh "$DOCKER_VM_NAME" which "$ARG_SCRIPT_ABS" &> /dev/null; then 
+    echo "Provided script file does not exist or is not executable in VM : $ARG_SCRIPT_ABS"
+    echo "Usage: $0 [SCRIPT] [SCRIPT_ARGS...]"
+    exit 1
+  fi
+
+  # Start the provided script. This works because the $HOME directory is mapped in the same location in the VM. 
+  docker-machine ssh "$DOCKER_VM_NAME" "CONTAINER_CERT_PATH=$DOCKER_CERT_PATH NETWORK_TYPE=vm $ARG_SCRIPT_ABS $*"
+fi
+