#!/usr/bin/env bash

function storage {
	location="$1"
	style="$2"

	if [[ -z $location ]]; then
		echo "ERROR: Location must be passed. $location"
	fi

	if [[ "$style" == "%" || "$style" == "pct" || "$style" == "percent" ]]; then
		df -h $location | tail -n 1 | awk {print $5}
	elif [[ "$style" == "/" || "$style" == "use" || "$style" == "usage" ]]
		df -h $location | tail -n 1 | awk {print $3"/"$2}
	else
		df -h $location | tail -n 1 | awk {print $5" Free, "$3"/"$2" Used"}
	fi
}

export dwm_storage_style="usage"

function storage-root {
	storage / "$1"
}
function storage-root-dwm {
	storage-root "$dwm_storage_style"
}

function storage-home {
	storage /home "$1"
}
function storage-home-dwm {
	storage-root "$dwm_storage_style"
}