Add quotes around location variable.

This commit is contained in:
2025-11-10 15:08:26 -07:00
parent 68eb84af52
commit d1cd9dddf1

View File

@@ -5,16 +5,16 @@ function storage {
location="$1"
style="$2"
if [[ -z $location ]]; then
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}
df -h "$location" | tail -n 1 | awk {print $5}
elif [[ "$style" == "/" || "$style" == "use" || "$style" == "usage" ]]; then
df -h $location | tail -n 1 | awk {print $3"/"$2}
df -h "$location" | tail -n 1 | awk {print $3"/"$2}
else
df -h $location | tail -n 1 | awk {print $5" Free, "$3"/"$2" Used"}
df -h "$location" | tail -n 1 | awk {print $5" Free, "$3"/"$2" Used"}
fi
}