#!/usr/pkg/bin/bash
# Output a sorted list of currently logged-in users, [email protected]
# Completely and utterly free, do whatever you want with this.

#defaults
c0="\e[0m"      # color reset
c1="\e[32m"     # title and total color
c2="\e[36m"     # color of [letter]
c3="\e[33m"     # color of user count per letter
c4="\e[35m"     # color of known users (from ~/.whoiknow)

# Process arguments
while getopts "hc" opt; do
       case $opt in
               h)
                       printf "Usage: whoa [-ch]\n"
                       printf "   -c removes all color, -h shows this message. \n"
                       printf "   Place a text file named .whoiknow in your root, one login name per line, to highlight known users in the output.\n"
                       exit
                       ;;
               c)
                       # color off
                       c0=""
                       c1=""
                       c2=""
                       c3=""
                       c4="*"
                       ;;
               \?)
                       printf "Usage: whoa [-ch]\n"
                       printf "   -c removes all color, -h shows this message. \n"
                       printf "   Place a text file named .whoiknow in your root, one login name per line, to highlight known users in the output.\n"
                       exit;
               ;;
       esac
done

# check for .whoiknow text file, read known usernames into array
declare knownUsers
if [ -f ~/.whoiknow ]; then
       mapfile -t knownUsers < ~/.whoiknow
fi

# Get and sort user list
USERS="$(/usr/bin/who -q | sed -e '$d')"
array=(${USERS// / })
readarray -t sorted < <(for a in "${array[@]}"; do echo "$a"; done | sort -u)

# store in associative array
declare -A alphaUsers
for i in "${!sorted[@]}"
do
       if [[ " ${knownUsers[@]} " =~ " ${sorted[i]} " ]]; then
               alphaUsers["${sorted[i]:0:1}"]+="${c4}"${sorted[i]}"${c0}, "
       else
               alphaUsers["${sorted[i]:0:1}"]+=${sorted[i]}", "
       fi
done

printf "\n     ${c1}Alphabetical User List (logged in)${c0}\n"
for x in {a..z}
do
       if [[ ! -z ${alphaUsers[$x]} ]]; then
               countTemp="${alphaUsers[$x]//[^ ]}"
               total="$(($total+${#countTemp}))"
               countTemp=$(printf "%03d" ${#countTemp})

               printf "\n     [${c2}${x^}${c0}] [${c3}$countTemp${c0}] "
               printf "${alphaUsers[$x]::-2}" #strip the last comma/space off
       fi
done

printf "\n\n     ${c1}$total users total${c0}"
printf "\n\n"