Create UCP environment + DTR using docker-machine and virtualbox

ucp.sh

#!/bin/bash
# DDC set up with Docker Machine
echo Test $0 $1 $2 $3
if [ "$1" == "" ]; then
  echo "Usage setup.sh <number of UCP nodes> <number of DTR nodes> <number of worker node>"
  exit 1
fi

if [ "$UCP_ADMIN" == "" ]; then
  UCP_ADMIN="moby"
fi

if [ "$UCP_PASSWORD" == "" ]; then
  UCP_PASSWORD="orcaorca"
fi

if [ "$ENVWORKERLABEL" == "" ]; then
  ENVWORKERLABEL="Test"
fi

#echo "Using $UCP_ADMIN and $UCP_PASSWORD"
set UCP_EXISTS=$(docker-machine ls -q ucp0)
if [ "$1" == "0" ]; then
  echo "No UCP nodes specified, assuming existing UCP cluster"
    if [ -n "UCP_EXISTS" ]; then
        echo "No existing UCP nodes specified, I need them so exiting"
        exit -1
    fi
else
    # Set up UCP HA
    echo "******************** Creating UCP cluster"
    docker-machine create -d virtualbox --virtualbox-memory "4096" ucp0 && docker-machine ssh ucp0 docker swarm init --advertise-addr $(docker-machine ip ucp0)
    echo "Swarm manager created"

    echo "Creating manager nodes"
    for (( COUNT=1; COUNT \< $1; COUNT++))
    do
        echo "Creating UCP node $COUNT"
        docker-machine create -d virtualbox --virtualbox-memory "4096" ucp$COUNT && docker-machine ssh ucp$COUNT docker swarm join --token $(docker-machine ssh ucp0 docker swarm join-token -q manager) $(docker-machine ip ucp0)
    done
    echo "----------- Installing UCP ------------------"
    docker-machine ssh ucp0 docker run --rm --tty --name ucp -v /var/run/docker.sock:/var/run/docker.sock docker/ucp install --host-address $(docker-machine ip ucp0) --admin-username "${UCP_ADMIN}" --admin-password "${UCP_PASSWORD}" --swarm-port 2378 --controller-port 9443

    echo "Create vizualiser"
    docker-machine ssh ucp0 docker service create \
    --name=viz \
    --publish=8082:8080/tcp \
    --constraint=node.role==manager \
    --mount=type=bind,src=/var/run/docker.sock,dst=/var/run/docker.sock \
    manomarks/visualizer
fi

if [ "$3" == "0" ]; then
  echo "No worker nodes specified, skipping"
else
    # Worker nodes
    echo "***************** Creating and joining $3 $ENVWORKERLABEL worker nodes to Swarm cluster"
    rm swarmnodes
    #COUNT = 0
    for (( COUNT=0; COUNT < $3; COUNT++))
    do
        echo "Creating node$COUNT"
        docker-machine create -d virtualbox --engine-label environment=$ENVWORKERLABEL node$COUNT  && echo "Created node$COUNT" && echo node$COUNT >> swarmnodes && docker-machine ssh node$COUNT docker swarm join --token $(docker-machine ssh ucp0 docker swarm join-token -q worker) $(docker-machine ip ucp0)
    done
fi

if [ "$2" == "0" ]; then
  echo "No DTR nodes specified"

    echo "All done you now have a working DDC cluster. You can access UCP at https://$(docker-machine ip ucp0):9443, and the Visualiser at http://$(docker-machine ip ucp0):8082"
  exit 1
else
    echo "***************** Create DTR cluster"
    echo "----------- Creating first DTR node ----------"
    docker-machine create -d virtualbox --virtualbox-memory "3072" dtr0
    # Join Swarm
    echo "----------- DTR joining Swarm ---------"
    docker-machine ssh dtr0 docker swarm join --token $(docker-machine ssh ucp0 docker swarm join-token -q worker) $(docker-machine ip ucp0)
    echo "--------------- Installing DTR -----------"
    # Install DTR
    docker-machine ssh ucp0 docker run --rm --tty --name dtr docker/dtr install --debug --ucp-url https://$(docker-machine ip ucp0):9443 --dtr-external-url https://$(docker-machine ip dtr0):8443  --replica-http-port "8090" --replica-https-port "8443" --ucp-node dtr0 --ucp-username "${UCP_ADMIN}" --ucp-password "${UCP_PASSWORD}" --ucp-insecure-tls --replica-id AB0000000000

    echo "--------------- Installing DTR nodes -------------"
    for (( COUNT=1; COUNT \< $2; COUNT++))
    do
        echo "Create DTR node $COUNT"
        docker-machine create -d virtualbox --virtualbox-memory "3072" dtr$COUNT && docker-machine ssh dtr$COUNT docker swarm join --token $(docker-machine ssh ucp0 docker swarm join-token -q worker) $(docker-machine ip ucp0)
        #docker-machine ssh dtr$COUNT docker run --rm --tty -p 809$COUNT:80 844$COUNT:443 docker/dtr join --debug --ucp-node dtr$COUNT --ucp-insecure-tls --ucp-url https://$(docker-machine ip ucp0) --ucp-username $UCP_ADMIN --ucp-password $UCP_PASSWORD --existing-replica-id 1234567890AB --replica-http-port "809$COUNT" --replica-https-port "844$COUNT" --replica-id AB000000000$COUNT
        docker-machine ssh ucp0 docker run --rm --tty -p 809$COUNT:80 844$COUNT:443 docker/dtr join --debug --ucp-node dtr$COUNT --ucp-insecure-tls --ucp-url https://$(docker-machine ip ucp0):9443 --ucp-username "${UCP_ADMIN}" --ucp-password "${UCP_PASSWORD}" --existing-replica-id 1234567890AB --replica-http-port "809$COUNT" --replica-https-port "844$COUNT" --replica-id AB000000000$COUNT
    done

    if [ $? -ne 0 ]; then
        echo "Something went wrong"
        exit $?
    else
        echo "All done you now have a working DDC cluster. You can access UCP at https://$(docker-machine ip ucp0):9443, and DTR at https://$(docker-machine ip dtr0):8443 and the Visualiser at http://$(docker-machine ip ucp0):8082"
    fi
fi