#! /bin/bash

# ? This is done to ignore the message "Make sure not to read and write
# ? the same file in the same pipeline", which is a result of ${DATABASE}
# ? being used below. (This disables the message file-wide.)
# shellcheck disable=SC2094

# shellcheck source=../scripts/helper-functions.sh
. /usr/local/bin/helper-functions.sh

DATABASE=${DATABASE:-/tmp/docker-mailserver/dovecot-quotas.cf}
USER_DATABASE=${USER_DATABASE:-/tmp/docker-mailserver/postfix-accounts.cf}

USER="${1}"
shift
QUOTA="${*}"

function usage  { echo "Usage: setquota <user@domain> [<quota>]" ; }

[[ -z ${USER} ]] && { usage ; errex "no username specified" ; }
[[ ${USER} =~ .*\@.* ]] || { usage ; errex "username must include the domain" ; }

if ! grep -qE "^${USER}\|" "${USER_DATABASE}"
then
  usage; errex "user ${USER} does not exist"
fi

# check quota
if [[ -n ${QUOTA} ]] && ! echo "${QUOTA}" | grep -qE "^([0-9]+(B|k|M|G|T)|0)\$"
then
  usage
  errex "invalid quota format. e.g. 302M (B (byte), k (kilobyte), M (megabyte), G (gigabyte) or T (terabyte))"
fi

create_lock # Protect config file with lock to avoid race conditions

touch "${DATABASE}"
if [[ -z ${QUOTA} ]]
then
  read -r -s "Enter quota (e.g. 10M): " QUOTA
  echo
  [[ -z "${QUOTA}" ]] && errex "Quota must not be empty. Use 0 for unlimited quota"
fi

# check quota
if [[ -n ${QUOTA} ]] && ! echo "${QUOTA}" | grep -qE "^([0-9]+(B|k|M|G|T)|0)\$"
then
  usage
  errex "invalid quota format. e.g. 302M (B (byte), k (kilobyte), M (megabyte), G (gigabyte) or T (terabyte))"
fi

delquota "${USER}"
echo "${USER}:${QUOTA}" >>"${DATABASE}"