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

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

help()
{
  thisfilename=$(basename -- "$0")
  echo "Add virtualhost to this server, including shell user and MySQL database."
  echo ""
  echo "usage: $thisfilename virtualhost [OPTIONS]"
  echo ""
  echo "  -h            Print this help."
  echo "  -u USERNAME   Username to use for this virtualhost. Optional, defaults to first 8 alphanumeric characters of virtualhost."
  echo "  -p PASSWORD   Password for username. Optional, random password generated if none specified."
  echo "  -j            Whether or not to jail the user. Optional, default is to not jail user."
  exit
}

# check for and set virtualhost
if [ -n "$1" ]; then
  if [ $1 == "-h" ]; then
    help
  elif vhost::validate_domain $1; then
    virtualhost="${1,,}"
    shift
  else
    echo "ERROR: Invalid virtualhost: $1"
    exit 1
  fi
else
  help
fi

while getopts "hu:p:j" opt; do
  case "${opt}" in
      h )
        help
        exit;;
      u )
        username=${OPTARG}
        ;;
      p )
        password=${OPTARG}
        ;;
      j )
        jail=true
        ;;
      \? )
        echo "Invalid option: $OPTARG"
        exit 1
        ;;
      : )
        echo "Invalid option: $OPTARG requires an argument"
        exit 1
      ;;
  esac
done

# check virtualhost
if [ ! -n "$virtualhost" ]; then
  echo "virtualhost not set"
  exit 1
fi

if [ -d /srv/www/$virtualhost ] || [ -f /etc/apache2/sites-available/$virtualhost.conf ]; then
  echo "virtualhost for $virtualhost already installed"
  exit 1
fi

# check for and set username
if [ ! -n "$username" ]; then
  username=`echo $virtualhost | sed 's|\.||'`
  username=`echo ${username:0:8}`
  if grep -q "^$username:" /etc/passwd; then
    # username already exists, try another
    username=`echo $virtualhost | sed 's|\.||'`
    username=`echo ${username:0:7}`
    if grep -q "^$username:" /etc/passwd; then
      # username already exists, try another
      username=`echo $virtualhost | sed 's|\.||'`
      username=`echo ${username:0:6}`
      if grep -q "^$username:" /etc/passwd; then
        # username already exists, try another
        username=`echo $virtualhost | sed 's|\.||'`
        username=`echo ${username:0:5}`
        if grep -q "^$username:" /etc/passwd; then
          # username already exists, try another
          username=`echo $virtualhost | sed 's|\.||'`
          username=`echo ${username:0:9}`
          if grep -q "^$username:" /etc/passwd; then
            # username already exists, try another
            username=`echo $virtualhost | sed 's|\.||'`
            username=`echo ${username:0:10}`
            if grep -q "^$username:" /etc/passwd; then
              # username already exists, try another
              username=`echo $virtualhost | sed 's|\.||'`
              username=`echo ${username:0:11}`
              if grep -q "^$username:" /etc/passwd; then
                # username already exists, try another
                username=`echo $virtualhost | sed 's|\.||'`
                username=`echo ${username:0:12}`
                if grep -q "^$username:" /etc/passwd; then
                  echo "trouble setting unique username, specify '-u USERNAME' to use an existing username"
                  exit 1
                fi
              fi
            fi
          fi
        fi
      fi
    fi
  fi
fi

if ! grep -q "^$username:" /etc/passwd; then
  # check for and set password
  if [ ! -n "$password" ]; then
    password=`/usr/bin/pwgen 12 1`
  fi
  # add user
  /usr/local/bin/vhost-user-add.sh $username -p "$password"
  # if jail option is set then jail user
  if [[ $jail = true ]]; then
    /usr/local/bin/vhost-user-jail.sh $username > /dev/null 2>&1
  fi
fi

# add virtualhost
/usr/local/bin/vhost-add.sh $virtualhost $username > /dev/null 2>&1

# add mysql database
/usr/local/bin/vhost-mysql-db-add.sh $virtualhost > /dev/null 2>&1