testnet.sh

#!/usr/bin/env bash

while test $# -gt 0; do
  case "$1" in
    -h|--help)
      echo "testnet - run HERB testnet"
      echo " "
      echo "testnet [options]"
      echo " "
      echo "options:"
      echo "-h, --help                    show brief help"
      echo "-ct, --ct_treshhold=ct        specify a ciphertext shares treshhold"
      echo "-t, --decryption_threshold=t  specify a decryption threshold"
      echo "-n, --maximum_clients=n       specify maximum clients count"
      echo "-c, --node_count=c            specify node count"
      echo "-p, --bots_per_node=p         specify bots per node count"
      exit 0
      ;;
    -ct|--ct_treshhold)
      shift
      if test $# -gt 0; then
        export t1=$1
      else
        echo "no cipertext shares treshhold specified"
        exit 1
      fi
      shift
      ;;
    -t|--decryption_threshold)
      shift
      if test $# -gt 0; then
        export t2=$1
      else
        echo "no decryption treshhold specified"
        exit 1
      fi
      shift
      ;;
    -p|--bots_per_node)
      shift
      if test $# -gt 0; then
        export bots_per_node=$1
      else
        echo "no bots_per_node specified"
        exit 1
      fi
      shift
      ;;
    -n|--maximum_clients)
      shift
      if test $# -gt 0; then
        export n=$1
      else
        echo "no maximum_clients specified"
        exit 1
      fi
      shift
      ;;
    -c|--node_count)
      shift
      if test $# -gt 0; then
        export node_count=$1
      else
        echo "no node_count specified"
        exit 1
      fi
      shift
      ;;
    --no_rebuild)
      NOREBUILD=true
      shift
      ;;
    *)
      break
      ;;
  esac
done

if [[ -n $bots_per_node ]] && [[ -n $node_count ]]
then
      n=$(( $bots_per_node*$node_count ))
fi

if [[ -z $n ]]
then
      n=12
fi

if [[ -z $bots_per_node ]]
then
      bots_per_node=1
      node_count=$n
fi


if [[ -z $node_count ]]
then
      node_count=$(($n/$bots_per_node))
fi

if [[ -z $t2 ]]
then
      t2=$(((n*2)/3))
fi

if [[ -z $t1 ]]
then
      t1=$t2
fi

echo "params: $t1 $t2 $n"
echo "node_count: $node_count"
echo "bots_per_node: $bots_per_node"

sleep 3

n=$((n+1))

cur_path=$( cd "$(dirname "${BASH_SOURCE[0]}")" ; pwd -P )

cd $cur_path
rm -rf ./vendor

rm -rf ./node0_config
mkdir ./node0_config

gopath=$(whereis go | grep -oP '(?<=go: )(\S*)(?= .*)' -m 1)
PATH=$gopath:$gopath/bin:$PATH

echo $GOBIN

if [[ $NOREBUILD ]]
then
  echo "no rebuild"
else
  make prepare
  GO111MODULE=off

  docker build -t herb_testnet .
fi

HERBPATH=/go/src/github.com/dgamingfoundation/HERB

echo "run node0"

node0_full_id=$(docker run -d herb_testnet /bin/bash -c "$HERBPATH/scripts/init_chain_full.sh $t1 $t2 $n;
 sed -i 's/timeout_commit = "5s"/timeout_commit = "1s"/' /root/.hd/config/config.toml;
 hd start")
node0_id=${node0_full_id:0:12}

echo "node0: $node0_id"
echo

ccc=$((n-2))
while  ! docker exec $node0_id /bin/bash -c "[[ -f $HERBPATH/bots/client$ccc.exp ]]" ; do
sleep 2
echo "waiting ..."
done

sleep 10

docker cp $node0_id:/root/.hd ./node0_config/.hd
docker cp $node0_id:/root/.hcli ./node0_config/.hcli
docker cp $node0_id:$HERBPATH/bots ./node0_config/bots

chmod -R 0777 ./node0_config

node0_addr=$(cat ./node0_config/.hd/config/genesis.json | jq '.app_state.genutil.gentxs[0].value.memo')

echo node0_addr
echo $node0_addr

if [[ -z $node0_addr ]] || [[ $node0_addr == "null" ]] || [[ $node0_addr == null ]]
then
  echo "ERROR"
  exit 1
fi

sed -i "s/seeds = \"\"/seeds = $node0_addr/" ./node0_config/.hd/config/config.toml

nodeArray=($node0_id)

for ((i=1;i<=$node_count;i++));
do
    nodeN_full_id=$(docker create -t herb_testnet /bin/bash -c "$HERBPATH/scripts/init_chain.sh $i && hd start")
    nodeN_id=${nodeN_full_id:0:12}

    nodeArray+=($nodeN_id)

    docker cp ./node0_config/.hd/config/config.toml $nodeN_id:/root/
    docker cp ./node0_config/.hd/config/genesis.json $nodeN_id:/root/
    docker cp ./node0_config/.hcli/keys $nodeN_id:/root/
    docker cp ./node0_config/bots $nodeN_id:/root/

    docker start $nodeN_id

    echo "node_num: $i, node_id: $nodeN_id"

done

sleep 5

echo "${nodeArray[@]}" > nodeArray.txt

chmod 0777 ./nodeArray.txt

echo "${nodeArray[@]}"
echo "all nodes started"
echo "run run_clients"

for ((i=0;i<=$node_count;i++));
do
  nodeN_id=${nodeArray[$i]}
  j=$(($i*$bots_per_node))
  docker exec -t -d $nodeN_id /bin/bash -c "$HERBPATH/scripts/run_clients.sh $j $bots_per_node"

  echo "node_num: $i, node_id: $nodeN_id"
done