configure-node.sh

#!/bin/bash

mkdir -p /home/ec2-user/logs

CONFIG_DONE_FILE=/home/ec2-user/container-configured
config_done() {
  touch ${CONFIG_DONE_FILE}
  echo "Couchbase Admin UI: http://localhost:8091" \
     "\nLogin credentials: Administrator / password"
     sleep infinity
}

if [ -e ${CONFIG_DONE_FILE} ]; then
  echo "Container previously configured."
  config_done
else
  echo "Configuring Couchbase Server.  Please wait (~60 sec)..."
fi

export PATH=/opt/couchbase/bin:${PATH}

wait_for_uri() {
  expected=$1
  shift
  uri=$1
  echo "Waiting for $uri to be available..."
  while true; do
    status=$(curl -s -w "%{http_code}" -o /dev/null $*)
    if [ "x$status" = "x$expected" ]; then
      break
    fi
    echo "$uri not up yet, waiting 2 seconds..."
    sleep 2
  done
  echo "$uri ready, continuing"
}

panic() {
  cat <<EOF

@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@
Error during initial configuration - aborting
@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@
EOF
  exit
}

couchbase_cli_check() {
  couchbase-cli $* || {
    echo Previous couchbase-cli command returned error code $?
    panic
  }
}

curl_check() {
  status=$(curl -sS -w "%{http_code}" -o /tmp/curl.txt $*)
  cat /tmp/curl.txt
  rm /tmp/curl.txt
  if [ "$status" -lt 200 -o "$status" -ge 300 ]; then
    echo
    echo Previous curl command returned HTTP status $status
    panic
  fi
}

wait_for_uri 200 http://localhost:8091/ui/index.html
echo "Couchbase Server up!"

my_ip=$(ifconfig | grep "inet 10" | awk '{print $2}')

if [[ ! -z $E2E_PARENT_CLUSTER ]]; then
  echo "Adding node to cluster $E2E_PARENT_CLUSTER with private IP $my_ip"
  couchbase_cli_check server-add -c $E2E_PARENT_CLUSTER -u Administrator -p password --server-add $my_ip \
    --server-add-username Administrator --server-add-password password --services data,index,query
  echo
  echo "Rebalancing cluster"
  couchbase_cli_check rebalance -c $E2E_PARENT_CLUSTER -u Administrator -p password 
  echo
else 
  echo "Set up the cluster"
  couchbase_cli_check cluster-init -c localhost --cluster-name couchbase-lite-test --cluster-username Administrator \
    --cluster-password password --services data,index,query --cluster-ramsize 8192 --cluster-index-ramsize 2048 --index-storage-setting default
  echo
fi

# Set alternate address for external access (after cluster init/join)
if [[ ! -z $E2E_PUBLIC_HOSTNAME ]]; then
  echo "Setting alternate address to $E2E_PUBLIC_HOSTNAME for external access (node: $my_ip)"
  couchbase_cli_check setting-alternate-address -c localhost -u Administrator -p password \
    --set --node $my_ip --hostname $E2E_PUBLIC_HOSTNAME
  echo
fi

echo "Verify credentials"
curl_check http://localhost:8091/settings/web -d port=8091 -d username=Administrator -d password=password -u Administrator:password
echo

echo "Create RBAC 'admin' user"
couchbase_cli_check user-manage --set \
  -c localhost -u Administrator -p password \
  --rbac-username admin --rbac-password password \
  --auth-domain local \
  --roles 'admin'
echo

echo "Couchbase Server configured"

config_done