#! /bin/bash

# The set -e option instructs bash to immediately exit if any command has a non-zero exit status
set -e

function printErr()
    {
        echo "usage: ./startup.sh <tag> <number_of_nodes_in_cluster>"
        echo "No argument $1 supplied"
    }

if [ -z "$1" ]
  then
    printErr "<tag>"
    exit 1
fi

if [ -z "$2" ]
  then
    printErr "<number_of_nodes_in_cluster>"
    exit 1
fi

TAG=$1
NODES=$2
AWS=`which aws`
FAB=`which fab`
DEPLOYDIR=`pwd`
# It seems BIGCHAINDIR was never used, but I wasn't sure
# so I just commented-out the following two lines. -Troy
#BIGCHAINDIR=`dirname $DEPLOYDIR`
#export BIGCHAINDIR

# check if awscli is installed
if [ ! -f "$AWS" ]
    then
        echo "awscli is not installed!!!"
        exit 1
fi

# checck if python-fabric is installed
if [ ! -f "$FAB" ]
    then
        echo "python-fabric is not installed!!!"
        exit 1
fi

# checking pem-file and changing access rights
if [ ! -f "pem/bigchaindb.pem" ]
    then
        echo "File pem/bigchaindb.pem (AWS private key) is missing"
        exit 1
fi
# 0400 for pem/bigchaindb.pem
chmod 0400 pem/bigchaindb.pem

# Start the specified number of nodes
# and tag them with the specified tag
python run_and_tag.py --tag $TAG --nodes $NODES

# Wait until all those instances are running
python wait_until_all_running.py --tag $TAG

# in case of elastic ips...
python get_elastic_ips.py --tag $TAG

# generate hostlist.py and add_keys.sh
python3 create_hostlist.py --tag $TAG > hostlist.py
# make add_keys executable and execute
chmod +x add2known_hosts.sh
./add2known_hosts.sh

# resetting the rethinkdb initfile and adding the nodes to join...
cp conf/bigchaindb.conf.template conf/bigchaindb.conf
cat add2dbconf >> conf/bigchaindb.conf

# rollout base packages (dependencies) needed before
# storage backend (rethinkdb) and bigchaindb can be rolled out
fab install_base_software

# rollout storage backend (rethinkdb)
fab install_rethinkdb

# rollout bigchaindb
fab install_bigchaindb

# generate genesisblock
HORST=`tail -1 conf/bigchaindb.conf|cut -d: -f1|cut -d= -f2`
fab -H $HORST -f fab_prepare_chain.py init_bigchaindb
# initiate sharding
fab start_bigchaindb_nodes

# now cleanup!
rm add2known_hosts.sh add2dbconf

# DONE!