#!/bin/bash

### Time and date ###
echo -e "$(date)"
echo -e


### Hostname ###
echo -e "$(/usr/bin/env figlet -w $(tput cols) "$(hostname)" | /usr/bin/env lolcat -f -F 0.05 -S $(echo $((0x$(hostname | md5sum | cut -c 1-10))) | cut -c 2-5))"

echo -e


### System info ###
echo -e "$(cat /etc/*release | grep PRETTY_NAME | cut -d = -f 2- | sed 's/\"//g') | Kernel: $(uname -r)
Uptime: $(uptime -p | sed 's/up //')"


### Services ###
# set column width
COLUMNS=1
# colors
green="\e[1;32m"
red="\e[1;31m"
undim="\e[0m"

services=($(cat /.motd.services))
# sort services
#IFS=$'\n' services=($(sort <<<"${services[*]}"))
#unset IFS

service_status=()
# get status of all services
for service in "${services[@]}"; do
    service_status+=($(systemctl is-active "$service"))
done

out=""
for i in "${!services[@]}"; do
    # color green if service is active, else red
    if [[ "${service_status[$i]}" == "active" ]]; then
        out+="${services[$i]}:,${green}${service_status[$i]}${undim},"
    else
        out+="${services[$i]}:,${red}${service_status[$i]}${undim},"
    fi
    # insert \n every $COLUMNS column
    if [ $((($i+1) % $COLUMNS)) -eq 0 ]; then
        out+="\n"
    fi
done
out+="\n"

printf "\nServices:\n"
printf "$out" | column -ts $',' | sed -e 's/^/  /'


### Docker ###
containers=($(docker ps --format "{{.Image}}/{{.Names}}"))
container_statuses=($(docker ps --format "{{.State}}"))
out=""
for i in ${!containers[@]}; do
	if [[ "${container_statuses[$i]}" == "running" ]]; then
		out+="${containers[$i]}:,${green}${container_statuses[$i]}${undim},"
	else
		out+="${containers[$i]}:,${red}${container_statuses[$i]}${undim},"
	fi
	out+="\n"
done
printf "\nDocker:\n"
printf "$out" | column -ts $',' | sed -e 's/^/ /'

rootless_users=($(cat /.motd.users))
for user in ${rootless_users[@]}; do
	containers=($(sudo -n DOCKER_HOST=unix:///run/user/$(id -u $user)/docker.sock docker ps --format "{{.Image}}/{{.Names}}"))
	container_statuses=($(sudo -n DOCKER_HOST=unix:///run/user/$(id -u $user)/docker.sock docker ps --format "{{.State}}"))
	out=""
	for i in ${!containers[@]}; do
		if [[ "${container_statuses[$i]}" == "running" ]]; then
			out+="${containers[$i]}:,${green}${container_statuses[$i]}${undim},"
		else
			out+="${containers[$i]}:,${red}${container_statuses[$i]}${undim},"
		fi
		out+="\n"
	done
	printf "\nDocker > $user:\n"
	printf "$out" | column -ts $',' | sed -e 's/^/ /'
done

### Disk space ###
# config
max_usage=80
bar_width=50
# colors
white="\e[39m"
green="\e[1;32m"
red="\e[1;31m"
magenta="\e[1;35m"
lightgreen="\e[1;92m"
lightred="\e[1;91m"
yellow="\e[1;33m"
lightyellow="\e[1;93m"
dim="\e[2m"
undim="\e[0m"

# disk usage: ignore zfs, squashfs & tmpfs
DISKS=$(df -H -x squashfs -x tmpfs -x devtmpfs -x zfs --output=target,pcent,size,used | tail -n+2)" 0 0 0 -"

if test $(command -v zfs); then
	DISKS+="\n"
	#DISKS+=$(zpool list -o name,capacity,size | tail -n+2)
	while read name available used usedsnap usedds usedchild compressratio; do
		ttlsize=$((($available+$used)))
		pcent=$(echo "($used/$ttlsize)*100" | bc -l | numfmt --format="%3.0f")
		gfxsize=$(numfmt --to=iec --format="%9f" $ttlsize)
		gfxused=$(numfmt --to=iec --format="%9f" $used)
		usedsnaprecursive=$(zfs get usedbysnapshots -r "$name" -H -p -t filesystem -o value | tail -n+2 | awk '{s+=$1} END {printf "%.0f", s}')
		usedsnappercent=$(echo "(($usedsnap + $usedsnaprecursive)/$ttlsize)*100" | bc -l | numfmt --format="%3.0f")
		useddspercent=$(echo "($usedds/$ttlsize)*100" | bc -l | numfmt --format="%3.0f")
		usedchildpercent=$(echo "(($usedchild - $usedsnaprecursive)/$ttlsize)*100" | bc -l | numfmt --format="%3.0f")
		DISKS+="$name $pcent% $gfxsize $gfxused $usedsnappercent $useddspercent $usedchildpercent x$compressratio\n"
	done <<< $(zfs list -H -p -d 0 -o name,available,used,usedsnap,usedds,usedchild,compressratio)
fi
printf "\nDisk usage:\n"

while read line; do
	if [ -z "$line" ]; then
		continue
	fi
    # get disk usage
    dds=$(echo "$line" | awk '{print $6}' | sed 's/%//')
    dds_width=$((($dds*$bar_width)/100))
    child=$(echo "$line" | awk '{print $7}' | sed 's/%//')
    child_width=$((($child*$bar_width)/100))
    snap=$(echo "$line" | awk '{print $5}' | sed 's/%//')
    snap_width=$((($snap*$bar_width)/100))
    usage=$(echo "$line" | awk '{print $2}' | sed 's/%//')
    used_width=$((($usage*$bar_width)/100))

	#echo "$line"
    #echo "$dds,$dds_width|$child,$child_width|$snap,$snap_width|$usage,$used_width"

	i=0
    # [
    bar="["

	# DDS
    if [ "${usage}" -ge "${max_usage}" ]; then
        bar+=$red
    else
        bar+=$green
    fi
    for ((; i<$dds_width; i++)); do
        bar+="="
    done

    # CHILDREN
    if [ "${usage}" -ge "${max_usage}" ]; then
        bar+=$lightred
    else
        bar+=$lightgreen
    fi
    for ((; i<(($dds_width + $child_width)); i++)); do
        bar+="="
    done

	# SNAP
    bar+=$lightyellow
    for ((; i<(($dds_width + $child_width + $snap_width)); i++)); do
        bar+="="
    done

	# TOTAL USAGE
    if [ "${usage}" -ge "${max_usage}" ]; then
        bar+=$red
    else
        bar+=$green
    fi
    for ((; i<$used_width; i++)); do
        bar+="="
    done

    # FREE SPACE
    bar+="${white}${dim}"
    for ((; i<$bar_width; i++)); do
        bar+="="
    done

    # ]
    bar+="${undim}]"

    # print usage line & bar
    echo "${line}" | awk '{ printf("%-19scompratio %+5s  %+4s/%+4s  [%+3s]\n", $1, $8, $4, $3, $2); }' | sed -e 's/^/  /'
    echo -e "${bar}" | sed -e 's/^/  /'
done <<< $(echo -e "$DISKS")


### Welcome message ###
LAST_LOGIN=$(last -w --time-format iso | grep $(whoami) | head -2 | tail -1)
echo -e
echo -e "Welcome to $(hostname), $(whoami)!
Last login: $(echo "$LAST_LOGIN" | awk '{for(i=4;i<=NF;i++) print $i}' | paste -sd ' ') from $(echo "$LAST_LOGIN" | awk '{print $3}')"

### ZFS available space warning ###
if test $(command -v zfs); then
	POOLS=$(zfs list -o name,used,available -p | tail -n+2)
	while read line; do
		name=$(echo "$line" | awk '{print $1}')
		used=$(echo "$line" | awk '{print $2}')
		avail=$(echo "$line" | awk '{print $3}')

		percentage=$(echo "scale=2;$avail/($used+$avail)*100" | bc)

		if (( $(echo "$percentage < 20" | bc -l) )); then
			echo -e "${red}Warning! ${white}$name ${undim}only has ${percentage}% capacity left"
		fi
	done <<< $(echo -e "$POOLS")
fi