#!/bin/bash

###############################
# Script to add a student     #
# Must be run as root         #
# For use with Zential Server #
###############################

# Check if is running as root
if [ "$(id -u)" != "0" ]; then
  echo "This script must be run as root" 1>&2
  exit 1
fi

USERNAME=$1
NAME=$2
DEFAULT_QUOTA=256000 # 256MB

# Extract surname
LAST_NAME=$(echo $NAME | awk '{print $NF}')

# Check if arguments are not empty
if [ -z "$USERNAME" ] || [ -z "$NAME" ]; then
  echo "Usage: $0 <username> <name>"
  exit 1
fi

# Check if $USERNAME has spaces
if [[ $USERNAME =~ " " ]]; then
  echo "Username cannot have spaces"
  exit 1
fi

# Check if $USERNAME starts with a number
if [[ $USERNAME =~ ^[0-9] ]]; then
  echo "Username cannot start with a number. Appending 'u' to the beginning of the username..."
  USERNAME="u$USERNAME"
fi

NAME=$(echo $NAME | tr "[A-Z]" "[a-z]" | sed -e "s/\b\(.\)/\u\1/g")

# Check if $USERNAME is already in use
if samba-tool user show $USERNAME >/dev/null 2>&1; then
  echo "User $USERNAME already exists"
  exit 1
fi

# Create user
echo "Creating user $USERNAME with name $NAME and password $USERNAME"
samba-tool user create $USERNAME $USERNAME --userou='OU=Professores,OU=IFMS-PP' --given-name="$NAME" --surname="$LAST_NAME" --home-drive='H:' --home-directory="\\\\acadsrv.ACAD.PP.IFMS.EDU.BR\\$USERNAME" --use-username-as-cn --must-change-at-next-login

# Add user to group alunos
echo "Adding user $USERNAME to group professores"
samba-tool group addmembers professores $USERNAME

# Create home directory
echo "Creating home directory for user $USERNAME"
mkdir /home/$USERNAME
echo "Setting permissions for home directory"
chmod 700 /home/$USERNAME/ -R
chown $USERNAME:'domain users' /home/$USERNAME/ -R

# Set user quota
./pdc-setquota.sh $USERNAME $DEFAULT_QUOTA

echo "User $USERNAME ($NAME) created. Password must be changed at next login."