#!/bin/bash
set -euo pipefail

MONGODB_PORT=""
MONGODB_KEY_FILE_PATH=""
#MONGODB_KEY_FILE_PASSWORD=""
MONGODB_CA_FILE_PATH=""
MONGODB_CRL_FILE_PATH=""
REPLICA_SET_NAME=""
MONGODB_FQDN=""
MONGODB_IP=""

while [[ $# -gt 1 ]]; do
  arg="$1"
  case $arg in
      --mongodb-port)
          MONGODB_PORT="$2"
          shift
          ;;
      --mongodb-key-file-path)
          MONGODB_KEY_FILE_PATH="$2"
          shift
          ;;
      --mongodb-key-file-password)
          # TODO(Krish) move this to a mapped file later
          MONGODB_KEY_FILE_PASSWORD="$2"
          shift
          ;;
      --mongodb-ca-file-path)
          MONGODB_CA_FILE_PATH="$2"
          shift
          ;;
      --mongodb-crl-file-path)
          MONGODB_CRL_FILE_PATH="$2"
          shift
          ;;
      --replica-set-name)
          REPLICA_SET_NAME="$2"
          shift
          ;;
      --mongodb-fqdn)
          MONGODB_FQDN="$2"
          shift
          ;;
      --mongodb-ip)
          MONGODB_IP="$2"
          shift
          ;;
      --storage-engine-cache-size)
          STORAGE_ENGINE_CACHE_SIZE="$2"
          shift
          ;;
      *)
          echo "Unknown option: $1"
          exit 1
          ;;
  esac
  shift
done

# sanity checks
if [[ -z "${REPLICA_SET_NAME:?REPLICA_SET_NAME not specified. Exiting!}" || \
    -z "${MONGODB_PORT:?MONGODB_PORT not specified. Exiting!}" || \
    -z "${MONGODB_FQDN:?MONGODB_FQDN not specified. Exiting!}" || \
    -z "${MONGODB_IP:?MONGODB_IP not specified. Exiting!}" || \
    -z "${MONGODB_KEY_FILE_PATH:?MONGODB_KEY_FILE_PATH not specified. Exiting!}" || \
    -z "${MONGODB_CA_FILE_PATH:?MONGODB_CA_FILE_PATH not specified. Exiting!}" || \
    -z "${MONGODB_CRL_FILE_PATH:?MONGODB_CRL_FILE_PATH not specified. Exiting!}" || \
    -z "${STORAGE_ENGINE_CACHE_SIZE:=''}" ]] ; then
    #-z "${MONGODB_KEY_FILE_PASSWORD:?MongoDB Key File Password not specified. Exiting!}" || \
  exit 1
else
  echo REPLICA_SET_NAME="$REPLICA_SET_NAME"
  echo MONGODB_PORT="$MONGODB_PORT"
  echo MONGODB_FQDN="$MONGODB_FQDN"
  echo MONGODB_IP="$MONGODB_IP"
  echo MONGODB_KEY_FILE_PATH="$MONGODB_KEY_FILE_PATH"
  echo MONGODB_CA_FILE_PATH="$MONGODB_CA_FILE_PATH"
  echo MONGODB_CRL_FILE_PATH="$MONGODB_CRL_FILE_PATH"
  echo STORAGE_ENGINE_CACHE_SIZE="$STORAGE_ENGINE_CACHE_SIZE"
fi

MONGODB_CONF_FILE_PATH=/etc/mongod.conf
HOSTS_FILE_PATH=/etc/hosts

# configure the mongod.conf file
sed -i "s|MONGODB_PORT|${MONGODB_PORT}|g" ${MONGODB_CONF_FILE_PATH}
sed -i "s|MONGODB_KEY_FILE_PATH|${MONGODB_KEY_FILE_PATH}|g" ${MONGODB_CONF_FILE_PATH}
#sed -i "s|MONGODB_KEY_FILE_PASSWORD|${MONGODB_KEY_FILE_PASSWORD}|g" ${MONGODB_CONF_FILE_PATH}
sed -i "s|MONGODB_CA_FILE_PATH|${MONGODB_CA_FILE_PATH}|g" ${MONGODB_CONF_FILE_PATH}
sed -i "s|MONGODB_CRL_FILE_PATH|${MONGODB_CRL_FILE_PATH}|g" ${MONGODB_CONF_FILE_PATH}
sed -i "s|REPLICA_SET_NAME|${REPLICA_SET_NAME}|g" ${MONGODB_CONF_FILE_PATH}
if [ ! -z "$STORAGE_ENGINE_CACHE_SIZE" ]; then
  if [[ "$STORAGE_ENGINE_CACHE_SIZE" =~ ^[0-9]+(G|M|T)B$ ]]; then
    sed -i.bk "s|STORAGE_ENGINE_CACHE_SIZE|${STORAGE_ENGINE_CACHE_SIZE}|g" ${MONGODB_CONF_FILE_PATH}
  else
    echo "Invalid Value for storage engine cache size $STORAGE_ENGINE_CACHE_SIZE"
    exit 1
  fi
else
  sed -i.bk "/cache_size=/d" ${MONGODB_CONF_FILE_PATH}
fi

# add the hostname and ip to hosts file
echo "${MONGODB_IP} ${MONGODB_FQDN}" >> $HOSTS_FILE_PATH

# create the directory if it does not exist, where MongoDB can store the data
# and config files; this assumes that the data directory is mounted at
# /data/db/main and the config directory is mounted at /data/configdb
mkdir -p /data/db/main /data/configdb/main

# start mongod
echo "INFO: starting mongod..."

# TODO Uncomment the first exec command and use it instead of the second one
# after https://github.com/docker-library/mongo/issues/172 is resolved. Check
# for other bugs too.
#exec /entrypoint.sh mongod --config ${MONGODB_CONF_FILE_PATH}
exec /usr/bin/mongod --config ${MONGODB_CONF_FILE_PATH}