#!/bin/bash

# simple screen information script
# similar to archey and screenfetch without annoying ASCII graphics

# this script is provided with NO GUARANTEE and NO SUPPORT
# if it breaks or does not do what you want, FIX IT YOURSELF

VERSION="3.4"   # updated for changes to free -m

# read wm names from a file
#WMLIST='/usr/bin/wmlist'
#wms=($(cat $WMLIST)) # read wmlist from file

# or use wm array -- add any that need to be recognized
wms=( 2bwm 2wm 9wm aewm afterstep ahwm alopex amiwm antiwm awesome blackbox bspwm catwm clfswm ctwm cwm dminiwm dragonflywm dwm echinus \
	euclid-wm evilpoison evilwm fluxbox flwm fvwm-crystal goomwwm hcwm herbstluftwm i3 icewm jwm karmen larswm lwm matwm2 mcwm monsterwm \
	musca notion nwm olwm openbox oroborus pekwm ratpoison sapphire sawfish sscrotwm sithwm smallwm snapwm spectrwm stumpwm subtle tfwm tinywm tritium twm \
	uwm vtwm w9wm weewm wind windowlab wm2 wmaker wmfs wmii wmx xfwm4 xmonad xoat yeahwm )

# define colors for color-echo
red="\e[31m"
grn="\e[32m"
ylw="\e[33m"
cyn="\e[36m"
blu="\e[34m"
prp="\e[35m"
rst="\e[0m"

color-echo() {  # print with colors
echo -e " $ylw$1: $rst$2"
}

#print-kernel() {
#color-echo 'Kernel' "$(uname -smr)"
#}

print-uptime() {
up=$(</proc/uptime)
up=${up//.*}                # string before first . is seconds
days=$((${up}/86400))       # seconds divided by 86400 is days
hours=$((${up}/3600%24))    # seconds divided by 3600 mod 24 is hours
mins=$((${up}/60%60))       # seconds divided by 60 mod 60 is mins
color-echo "Uptime" $days'd '$hours'h '$mins'm'
}

print-shell() {
color-echo 'Shell' $SHELL
}

print-cpu() {
arm=$(grep ARM /proc/cpuinfo)    # ARM procinfo uses different format
if [[ "$arm" != "" ]]; then
	cpu=$(grep -m1 -i 'Processor' /proc/cpuinfo)
else
	cpu=$(grep -m1 -i 'model name' /proc/cpuinfo)
fi
color-echo 'CPU' "${cpu#*: }" # everything after colon is processor name
}

print-gpu() {
gpu=$(lspci | grep VGA | awk -F ': ' '{print $2}' | sed 's/(rev ..)//g')
color-echo 'GPU' "$gpu"
}

print-packages() {
packages=$(pacman -Qs | grep local | wc -l)
color-echo 'Packages' "$packages"
}

print-disk() {
# field 2 on line 2 is total, field 3 on line 2 is used
disk=$(df -h /home/ | awk 'NR==2 {total=$2; used=$3; print used" / "total}')
color-echo 'Disk' "$disk"
}

print-mem() {
# field 2 on line 2 is total, field 3 on line 2 is used (in new format)
# field 2 on line 2 is total, field 3 on line 3 is used (in old format)

if [[ $(free -h) =~ "buffers" ]]; then # using old format
	mem=$(free -h | awk 'NR==2 {total=$2} NR==3 {used=$3; print used" / "total}')
else # using new format
	mem=$(free -h | awk 'NR==2 {total=$2} NR==2 {used=$3; print used" / "total}')
fi
color-echo 'Memory' "$mem"
}

print-wm() {
for wm in ${wms[@]}; do          # pgrep through wmname array
	pid=$(pgrep -x -u $USER $wm) # if found, this wmname has running process
	if [[ "$pid" ]]; then
		color-echo 'WM' $wm
		break
	fi
done
}

print-de() {
if [[ $(pgrep -x -u $USER lxsession) ]]; then         # if lxsession is running, assume LXDE
	color-echo 'DE' 'LXDE'
elif [ $(pgrep -x -u $USER xfce4-session) ]; then   # if xfce4-session is running, assume Xfce
	color-echo 'DE' 'Xfce'
fi
}

print-distro() {
[[ -e /etc/os-release ]] && source /etc/os-release
if [[ -n "$PRETTY_NAME" ]]; then
	color-echo 'OS' "$PRETTY_NAME"
else
	color-echo 'OS' "not found"
fi
}

print-colors() {
NAMES=('█ black' '█ red' '█ green' '█ yellow' '█ blue' '█ magenta' '█ cyan' '█ white')
for f in {0..7}; do
	echo -en "\033[m\033[$(($f+30))m  ${NAMES[$f]}" # normal colors
done
echo
for f in {0..7}; do
	echo -en "\033[m\033[1;$(($f+30))m  ${NAMES[$f]}" # bold colors
done
echo -e "$rst\n"
}

if [[ $1 = '-v' ]]; then # print version information and exit
	echo $(basename "$0") / version: $VERSION / wm count: ${#wms[*]}
	exit
fi

#echo -e "\n  $prp$USER@$HOSTNAME$rst\n"
echo
echo -en "\033[35;1m          ,.  ,.      |\033[0m" && print-distro
echo -en "\033[35;1m           ||  ||     |\033[0m" && print-uptime
echo -en "\033[35;1m          ,''--''.    |\033[0m" && print-shell
echo -en "\033[35;1m         : (.)(.) :   |\033[0m" && print-wm
echo -en "\033[35;1m        ,'        \`.  |\033[0m" && print-packages
#echo -en "\033[35;1m        :          :  |\033[0m" && print-kernel
echo -en "\033[35;1m        :          :  |\033[0m" && print-cpu
echo -en "\033[35;1m  -boo- \`._m____m_,'  |\033[0m" && print-gpu
echo
print-colors
#
#${redf} ▄██████▄${reset}
#${redf}▄${whitef}█▀█${redf}██${whitef}█▀█${redf}██▄${reset}
#${redf}█${whitef}▄▄█${redf}██${whitef}▄▄█${redf}███${reset}
#${redf}████████████${reset}
#${redf}██▀██▀▀██▀██${reset}
#${redf}▀ ▀ ▀ ▀${reset}