#!/bin/bash
# This script finds the top CPU users on the machine and fires in separate
# account move tickets to the queue.  Written by Erik S, ext 834 (e@inmotion.net)

#set some variables
SESSIONID=$(date +%s)
OURSERVER=$(hostname -s)
RECIPIENT="sadmin@imhadmin.net"

if [ -z "$1" ]; then
	echo "$(echo "$0" | cut -d/ -f4) - Written by Erik S. (x834/e@inmotion.net)"
	echo
	echo "This script will generate move tickets for specified number of users"
	echo "sorted by highest CP count first.  If a move ticket has already been"
	echo "generated, the user will be skipped and a higher number can be passed."
	echo "If the --force flag is passed, move tickets will be force generated."
	echo
	echo "Usage:"
	echo "       $0 <# of accts> [--force]"
	echo
	exit 0
fi

        if [ "$2" == "--force" ]; then
                FORCE=TRUE
        fi

function generate_ticket {
        MVUSER_HOME=$(file /home*/"$MVUSER" | awk -F: '{print $1}')
	echo "$OURSERVER" > "$MVUSER_HOME/tmp/.move"
        chown "$MVUSER.$MVUSER" "$MVUSER_HOME/tmp/.move"
        echo -e "  Username       Diskspace         Homedir  Pointed Local?      Dedicated IP?           Has SSL?          local MX?    Suspended? \n $line" | /opt/sharedrads/strmailer -u root -s "Account move: $MVUSER @ $SERVER" -e $RECIPIENT
        echo "[*] generated move ticket for $MVUSER"
}

#INPUT/ERROR HANDLING
NUMBER=$1


        if expr "$NUMBER" + 1 &>/dev/null; then
                echo "$NUMBER is a number" >> /dev/null
        else
		echo "ERROR: You did not provide a number of accounts to schedule for move."
		exit 0
	fi

	if [ "$NUMBER" -lt 3 ]; then
		echo "ERROR: Number of accounts to schedule for move must be greather than 3."
		exit 0
	fi

#DUMP DATA FROM LISTACCOUNTS
sa -m | sort -nk4 | grep -v root | grep -v mysql | grep -v mail | grep -v cpanel | grep -v nobody | tail -"$NUMBER" > "/tmp/.$SESSIONID"


# User define Function (UDF)
processLine(){
  line="$*" # get all args
  MVUSER=$(echo "$line" | awk '{print $1}')
  SERVER=$(hostname -s)

	if [ "$FORCE" == "TRUE" ]; then
		generate_ticket
		return
	fi

	if [ -f "/var/cpanel/suspended/$MVUSER" ]; then #check for suspension
		echo "[*] $MVUSER account is suspended, skipping"
		return
	fi

	if [ -f "$MVUSER_HOME/tmp/.move" ]; then  #if user has already been scheduled to move once, lets check if it was our machine

		MOVEBOX=$(cat "$MVUSER_HOME/tmp/.move")

		if [ "$MOVEBOX" == "$OURSERVER" ]; then
			echo "[*] already generated ticket for $MVUSER, skipping"
			return
		fi

	else
		generate_ticket
fi

}

#Specify the file we outputted to
FILE="/tmp/.$SESSIONID"

#make sure file exist and readable
   if [ ! -f "$FILE" ]; then
  	echo "$FILE : does not exist, check to see if listacct tool is working, escalate to T3 if the problem persists."
  	exit 1
   elif [ ! -r "$FILE" ]; then
  	echo "$FILE: can not read, escalate to T3 if this problem persists"
  	exit 2
   fi

# Set loop separator to end of line
BAKIFS=$IFS
IFS=$(echo -en "\n\b")
exec 3<&0
exec 0<"$FILE"
while read -r line
do
	processLine "$line"
done
exec 0<&3

# restore $IFS which was used to determine what the field separators are
IFS=$BAKIFS

rm -f "/tmp/.$SESSIONID"
exit 0