60 lines
1.3 KiB
Bash
60 lines
1.3 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Storage locations to check
|
||
|
|
locations=(
|
||
|
|
"/home/nathan"
|
||
|
|
"/home/nathan/Nextcloud"
|
||
|
|
)
|
||
|
|
|
||
|
|
icons=(
|
||
|
|
""
|
||
|
|
""
|
||
|
|
)
|
||
|
|
|
||
|
|
# Check for first arg
|
||
|
|
if [ -z "$1" ]; then
|
||
|
|
RED='\033[0;31m'
|
||
|
|
GREEN='\033[0;32m'
|
||
|
|
YELLOW='\033[0;33m'
|
||
|
|
BLUE='\033[0;34m'
|
||
|
|
NC='\033[0m' # No Color
|
||
|
|
|
||
|
|
# Check if storage is full
|
||
|
|
for i in "${!locations[@]}"; do
|
||
|
|
# Check if the location exists
|
||
|
|
location=${locations[$i]}
|
||
|
|
if [ -d "$location" ]; then
|
||
|
|
# Get icon for location
|
||
|
|
icon=${icons[$i]}
|
||
|
|
|
||
|
|
# Get the current usage
|
||
|
|
size=$(du -hs "$location" | awk '{print $1}')
|
||
|
|
echo -en "${YELLOW}${icon} ${NC}${size} "
|
||
|
|
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
else
|
||
|
|
RED='%{F#ff0000}'
|
||
|
|
GREEN='%{F#ff60d090}'
|
||
|
|
YELLOW='%{F#F0C674}'
|
||
|
|
BLUE='%{F#0000ff}'
|
||
|
|
NC='%{F-}'
|
||
|
|
|
||
|
|
# Check if storage is full
|
||
|
|
for i in "${!locations[@]}"; do
|
||
|
|
# Check if the location exists
|
||
|
|
location=${locations[$i]}
|
||
|
|
if [ -d "$location" ]; then
|
||
|
|
# Get icon for location
|
||
|
|
icon=${icons[$i]}
|
||
|
|
|
||
|
|
# Get the current usage (discard error output)
|
||
|
|
size=$(du -hs "$location" 2>/dev/null | awk '{print $1}')
|
||
|
|
echo -en "%{A1:/home/nathan/scripts/zellij.sh --cwd $location zsh:}${YELLOW}${icon} ${NC}${size}%{A} "
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
echo ""
|
||
|
|
fi
|