mirror of
https://gitlab.com/mattia.basaglia/ASCII-Pony.git
synced 2024-11-22 04:07:59 +01:00
319 lines
6.5 KiB
Bash
Executable file
319 lines
6.5 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
PONY=random
|
|
INFO=( User Hostname Distro Kernel Uptime Shell Packages RAM Disk )
|
|
|
|
function ponyget_Shell()
|
|
{
|
|
grep $USER /etc/passwd | cut -f7 -d:
|
|
}
|
|
|
|
function ponyget_User()
|
|
{
|
|
whoami
|
|
}
|
|
|
|
function ponyget_Distro()
|
|
{
|
|
lsb_release -isr | paste "-d " - -
|
|
}
|
|
|
|
|
|
function ponyget_RAM()
|
|
{
|
|
local ramtable=$(free -h)
|
|
local used=$(echo "$ramtable" | sed -n 3p | sed -r "s/ +/\t/g" | cut -f 3)
|
|
local total=$(echo "$ramtable" | sed -n 2p | sed -r "s/ +/\t/g" | cut -f 2)
|
|
|
|
local ramtable=$(free -m)
|
|
local used_M=$(echo "$ramtable" | sed -n 3p | sed -r "s/ +/\t/g" | cut -f 3)
|
|
local total_M=$(echo "$ramtable" | sed -n 2p | sed -r "s/ +/\t/g" | cut -f 2)
|
|
let percent="$used_M * 100 / $total_M"
|
|
|
|
local color="32";
|
|
if [ "$percent" -gt 66 ]
|
|
then
|
|
color="31"
|
|
elif [ "$percent" -gt 33 ]
|
|
then
|
|
color="33"
|
|
fi
|
|
|
|
echo -e "\x1b[$color;1m$used\x1b[0m / $total"
|
|
}
|
|
|
|
function ponyget_Kernel()
|
|
{
|
|
uname -r -m
|
|
}
|
|
|
|
function ponyget_Hostname()
|
|
{
|
|
hostname
|
|
}
|
|
|
|
function ponyget_CPU()
|
|
{
|
|
cat /proc/cpuinfo | grep "model name" | head -n 1 | sed -r "s/model name\s: //"
|
|
}
|
|
|
|
function ponyget_Uptime()
|
|
{
|
|
uptime | grep -oE "up\s+[^,]+" | sed -r "s/up\s+//"
|
|
}
|
|
|
|
function ponyget_Packages()
|
|
{
|
|
if which dpkg &>/dev/null
|
|
then
|
|
dpkg --get-selections | grep -v deinstall | wc -l
|
|
elif which rpm &>/dev/null
|
|
then
|
|
rpm -qa | wc -l
|
|
elif which pacman &>/dev/null
|
|
then
|
|
pacman -Q | wc -l
|
|
fi
|
|
}
|
|
|
|
function ponyget_Disk()
|
|
{
|
|
local diskusage=$(df -lh --total | tail -n 1 | sed -r "s/ +/\t/g" )
|
|
local used=$(echo "$diskusage" | cut -f 3)
|
|
local total=$(echo "$diskusage" | cut -f 2)
|
|
local percent=$(echo "$diskusage" | cut -f 5 | sed s/%// )
|
|
|
|
local color="32";
|
|
if [ "$percent" -gt 66 ]
|
|
then
|
|
color="31"
|
|
elif [ "$percent" -gt 33 ]
|
|
then
|
|
color="33"
|
|
fi
|
|
|
|
echo -e "\x1b[$color;1m$used\x1b[0m / $total"
|
|
}
|
|
|
|
function bold()
|
|
{
|
|
echo -en "\x1b[1m${*}\x1b[22m"
|
|
}
|
|
|
|
function underline()
|
|
{
|
|
echo -en "\x1b[4m${*}\x1b[24m"
|
|
}
|
|
|
|
function title()
|
|
{
|
|
echo
|
|
bold ${*}
|
|
echo
|
|
}
|
|
|
|
function help()
|
|
{
|
|
title NAME
|
|
echo -e "\t$(bold $0) - show a pony and some system information"
|
|
|
|
title SYNOPSIS
|
|
echo -e "\t$(bold $0) [$(bold --pony) $(underline pony)|$(bold -p=)$(underline pony)] [$(bold --info) $(underline id)|$(bold -i=)$(underline id)...]"
|
|
echo -e "\t$(bold $0) $(bold help)|$(bold --help)|$(bold -h)"
|
|
|
|
title OPTIONS
|
|
echo -e "\t$(bold --pony) $(underline pony), $(bold -p=)$(underline pony)"
|
|
echo -e "\t\tSelect a pony (default: $PONY)."
|
|
echo
|
|
echo -e "\t$(bold --info) $(underline id), $(bold -i=)$(underline id)"
|
|
echo -e "\t\tShow the given info (default: ${INFO[@]})."
|
|
echo -e "\t\tThis option supports multiples IDs separated by commas, spaces or colons."
|
|
echo -e "\t\tAvailable IDs:"
|
|
declare -F | grep ponyget_ | sed "s/declare -f ponyget_/\t\t * /"
|
|
|
|
title CONFIGURATION
|
|
echo -e "\tYou can override $(bold PONY) and $(bold INFO) in the config files."
|
|
echo -e "\tConfiguration files:"
|
|
echo -e "\t * $(underline PREFIX)$(bold /share/ascii-pony/systempony.conf) (system)"
|
|
echo -e "\t * $(bold ~/.systempony) (user)"
|
|
|
|
echo
|
|
}
|
|
|
|
function select_info()
|
|
{
|
|
INFO=($(echo "${*}" | column -t -s:,))
|
|
}
|
|
|
|
SELFDIR=$(dirname $(readlink -se "${BASH_SOURCE[0]}"))
|
|
function get_data_file()
|
|
{
|
|
if [ -e "$SELFDIR/$1" ]
|
|
then
|
|
# Not installed
|
|
echo "$SELFDIR/$1"
|
|
elif [ -e "$SELFDIR/../share/ascii-pony/$1" ]
|
|
then
|
|
# Installed with PREFIX=$SELFDIR/..
|
|
echo "$SELFDIR/../share/ascii-pony/$1"
|
|
fi
|
|
}
|
|
|
|
# Read global config
|
|
globalconfig=$(get_data_file systempony.conf)
|
|
if [ -r "$globalconfig" ]
|
|
then
|
|
source "$globalconfig"
|
|
fi
|
|
# Read user config
|
|
if [ -r ~/.systempony ]
|
|
then
|
|
source ~/.systempony
|
|
fi
|
|
|
|
# Read parameters
|
|
while [ "$1" ]
|
|
do
|
|
case "$1" in
|
|
--help|-h|help)
|
|
help
|
|
exit 0
|
|
;;
|
|
--pony)
|
|
shift
|
|
PONY=$1
|
|
;;
|
|
-p=*)
|
|
PONY=$(echo "$1" | sed "s/-p=//")
|
|
;;
|
|
--info)
|
|
infostring=""
|
|
while [ "$2" ] && ! echo "$2" | grep -q -e "-"
|
|
do
|
|
infostring="$infostring $2"
|
|
shift
|
|
done
|
|
select_info $infostring
|
|
;;
|
|
-i=*)
|
|
select_info "$(echo "$1" | sed "s/-i=//")"
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
# Evaluate info
|
|
infoval=()
|
|
let maxkeyl=0
|
|
let maxvall=0
|
|
function addinfo()
|
|
{
|
|
infoval+=("${2}")
|
|
local keyl=$(echo "$1" | wc -c)
|
|
local vall=$(echo "${2}" | wc -c)
|
|
[ $keyl -gt $maxkeyl ] && maxkeyl=$keyl;
|
|
[ $vall -gt $maxvall ] && maxvall=$vall;
|
|
}
|
|
for info in ${INFO[*]}
|
|
do
|
|
if [ "$(type -t ponyget_${info})" = "function" ]
|
|
then
|
|
addinfo $info "$(ponyget_${info})"
|
|
else
|
|
addinfo $info "unsupported"
|
|
fi
|
|
done
|
|
|
|
ponydir=$(get_data_file "rendered/ansi/")
|
|
ponyfile="$ponydir/$PONY.colored.txt"
|
|
if [ -n "$ponydir" -a '(' -z "$PONY" -o "$PONY" = random ')' ]
|
|
then
|
|
ponyfile="$(find rendered/ansi/ -name '*.colored.txt' | shuf | head -n 1)"
|
|
fi
|
|
|
|
# WARNING most of the variable names below this point are pretty bad and should be refactored
|
|
|
|
declare -A ponyline # Number of characters before the system info output
|
|
ponyline[applejack-nohat]=64
|
|
ponyline[applejack]=70
|
|
ponyline[bigmac]=42
|
|
ponyline[celestia]=76
|
|
ponyline[cadance]=55
|
|
ponyline[derpy]=32
|
|
ponyline[fluttershy]=67
|
|
ponyline[great-and-powerful]=68
|
|
ponyline[luna]=67
|
|
ponyline[lyra]=61
|
|
ponyline[pinkie-pie]=61
|
|
ponyline[rainbow-dash]=61
|
|
ponyline[rainbow]=61
|
|
ponyline[rarity]=61
|
|
ponyline[rose]=61
|
|
ponyline[trixie]=61
|
|
ponyline[twilight-alicorn]=61
|
|
ponyline[twilight-unicorn]=61
|
|
ponyline[vinyl-scratch-glasses]=69
|
|
ponyline[vinyl-scratch-noglasses]=69
|
|
|
|
declare -A ponystart # Number of lines before the system info output
|
|
ponystart[luna]=18
|
|
ponystart[rarity]=11
|
|
ponystart[trixie]=16
|
|
|
|
if [ -f "$ponyfile" ]
|
|
then
|
|
lines=$(cat "$ponyfile" | wc -l) # cat to avoid printing file name
|
|
let info_index=0
|
|
COLUMNS=$(tput cols)
|
|
|
|
|
|
if [ "${ponyline[$PONY]}" ]
|
|
then
|
|
linesize=${ponyline[$PONY]}
|
|
else
|
|
linesize=80
|
|
fi
|
|
|
|
if [ "${ponystart[$PONY]}" ]
|
|
then
|
|
start_line=${ponystart[$PONY]}
|
|
else
|
|
let start_line=($lines-${#INFO[@]})/2
|
|
fi
|
|
|
|
|
|
|
|
let l=1
|
|
while IFS= read -r line
|
|
do
|
|
if [ $info_index -lt ${#INFO[@]} -a $l -ge $start_line ]
|
|
then
|
|
let msgsize="$COLUMNS - $linesize - $maxkeyl - 3"
|
|
thislinesize=$(echo "$line" | sed -r 's/\x1b\[[0-9;]+m//g' | wc -c)
|
|
if [ $thislinesize -gt $linesize ]
|
|
then
|
|
let linediff="$thislinesize - $linesize"
|
|
line="$(echo "$line" | head -c -$linediff)";
|
|
fi
|
|
printf "%s \x1b[31;1m%-${maxkeyl}s\x1b[0m: %s\n" \
|
|
"$line" \
|
|
"${INFO[$info_index]}" \
|
|
"$(echo -n "${infoval[$info_index]}" | head -c $msgsize)"
|
|
let info_index++
|
|
else
|
|
echo -n "$line"
|
|
echo -e "\x1b[0m"
|
|
fi
|
|
let l++
|
|
done < "$ponyfile"
|
|
else
|
|
let info_index=0
|
|
while [ $info_index -lt ${#INFO[@]} ]
|
|
do
|
|
printf "\x1b[31;1m%-${maxkeyl}s\x1b[0m: %s\n" \
|
|
"${INFO[$info_index]}" \
|
|
"${infoval[$info_index]}"
|
|
let info_index++
|
|
done
|
|
fi
|
|
|