#!/bin/bash
#
# vmail-stack
# https://git.stack-source.com/msb/vmail-stack
# MIT License Copyright (c) 2021 Matthew Saunders Brown

# load include file
source $(dirname $0)/vmail.sh

help()
{
  thisfilename=$(basename -- "$0")
  echo "Modify an existing domain."
  echo ""
  echo "usage: $thisfilename domain [OPTIONS]"
  echo ""
  echo "  -h          Print this help."
  echo "  -l LIMIT    Mailbox limit - the maximum number of mailboxes under this domain."
  echo "  -q QUOTA    Mailbox Quota Default - default quota for new mailboxes under this domain."
  echo "  -s STATUS   1 for enabled, 0 for disabled. Default is in db structure and is normally set to 1."
  exit
}

# check for and verify domain
if [ -n "$1" ]; then
  if [ $1 == "-h" ]; then
    help
  else
    domain=$1
    shift
    if vmail::validate_domain $domain; then
      domain_id=`mysql --defaults-extra-file=$MYSQL_CONNECTION_INFO_FILE -s -r -N -e "SELECT id from vm_domains WHERE vm_domains.domain='$domain';"`
      if [ -z $domain_id ] ; then
        echo "ERROR: Email domain $domain does not exist."
        exit
      fi
    else
      echo "ERROR: $domain is not a valid domain name."
      exit
    fi
  fi
else
  help
fi

# set any options that were passed
while getopts "hl:q:s:" opt; do
  case "${opt}" in
    h )
      help
      exit;;
    l )
      limit=${OPTARG}
      ;;
    q )
      quota=${OPTARG}
      ;;
    s )
      status=${OPTARG}
      ;;
    \? )
      echo "Invalid option: $OPTARG" 1>&2
      ;;
    : )
      echo "Invalid option: $OPTARG requires an argument" 1>&2
    ;;
  esac
done

# set update portion of command to empty string
dbset=""

# check for quota update
if [ -n "$quota" ]; then
  # make quota uppercase in case it is set to NULL
  quota=`echo $quota | tr [:lower:] [:upper:]`
  if [[ "$quota" =~ ^[0-9]+$ ]] || [[ "$quota" == "NULL" ]]; then
    dbset=" mbox_quota_default=\"$quota\""
  else
    echo "ERROR: quota (-q) must numeric or NULL"
    exit 1
  fi
fi

# check for limit update
if [ ! -z "$limit" ]; then
  # make limit uppercase in case it is set to NULL
  limit=`echo $limit | tr [:lower:] [:upper:]`
  if [[ "$limit" =~ ^[0-9]+$ ]] || [[ "$limit" == "NULL" ]]; then
    if [ ! -z "$dbset" ]; then
      dbset="$dbset,"
    fi
    dbset="$dbset mbox_limit=\"$limit\""
  else
    echo "ERROR: limit (-l) must numeric or NULL"
    exit 1
  fi
fi

# check for status update
if [ ! -z "$status" ]; then
  if [ "$status" == 0 ] || [ "$status" == 1 ]; then
    if [ ! -z "$dbset" ]; then
      dbset="$dbset,"
    fi
    dbset="$dbset status=\"$status\""
  else
    echo "ERROR: status (-s) must be 1 or 0"
    exit
  fi
fi

if [ -n "$dbset" ]; then

  # build query
  dbcmd="mysql --defaults-extra-file=$MYSQL_CONNECTION_INFO_FILE -e 'UPDATE vm_domains SET $dbset WHERE domain.id=\"$domain_id\";'"
#   eval $dbcmd
  echo $dbcmd

else

  echo "ERROR: No values passed for update."

fi