#! /usr/bin/env sh

#############################################################################
#
# Usage:
#
#############################################################################

#### Colors definition
BLACK='\033[30;40m'
RED='\033[0;31m'
REDB='\033[1;31m'
GREEN='\033[1;32m'
YELLOW='\033[1;33m'
BLUE='\033[34;40m'
MAGENTA='\033[0;35m'
CYAN='\033[36;40m'
WHITE='\033[0;37m'
WHITEB='\033[1;37m'
RESET='\033[0m'


printf '%b' "${MAGENTA}++++++++++++++++++++: ${WHITE}Disk Usage${RESET} ${MAGENTA}:+++++++++++++++++++++${RESET}"
# Root partition
# Replace by rootfs in few kernel version
DISK_USAGE=$(df | grep -m1 "/$" | awk '{print $5}')
printf '%b' "\n${MAGENTA}+ ${WHITEB}/${RESET}\t[ ${DISK_USAGE} ] "

# Other partition
for PART in /boot /home /opt /tmp /usr /var /var/lib/vz; do
  # "/...$" : $ to grep only the mount point and not sub-directories (/var: OK; /mnt/temp: nOK)
  if (df | grep "${PART}$" > /dev/null); then
    DISK_USAGE=$(df | grep "${PART}$" | awk '{print $5}')
    printf '%b' "\n${MAGENTA}+ ${WHITEB}${PART}${RESET}\t[ ${DISK_USAGE} ] "
  fi
done

### IF ZFS
if [ $(command -v zpool) ]; then
  printf '%b' "\n${MAGENTA}+++++++++++++++++++ ${WHITEB}ZFS${RESET} ${WHITE}Partitions${RESET} ${MAGENTA}:++++++++++++++++++${RESET}"
  NUMBER_PART=$(df -T | grep zfs | wc -l)
  LINE=1

  while [ "${LINE}" -le "${NUMBER_PART}" ]; do
    PART_NAME=$(df -T | grep zfs | sed -n "${LINE},${LINE}p" | awk '{print $1}')
    DISK_USAGE=$(df -T | grep zfs | sed -n "${LINE},${LINE}p" | awk '{print $6}')
    printf '%b' "\n${MAGENTA}+ ${WHITEB}${PART_NAME}${RESET} [ ${DISK_USAGE} ] "
    LINE=$((LINE+1))
    # Bash ONLY:  (( LINE++ ))
  done
fi
### FI ZFS

printf '%b' "${RESET}\n"