mirror of
https://gitlab.com/mattia.basaglia/ASCII-Pony.git
synced 2024-11-21 11:47:59 +01:00
420 lines
10 KiB
Bash
Executable file
420 lines
10 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
# These global variables can be overridden by config
|
|
# Name of the pony
|
|
PONY=random
|
|
# What info to show (must have a function defined as ponyget_* to work)
|
|
INFO=(User Hostname IP Distro Kernel Uptime Load Shell Packages CPU Threads RAM Swap Disk)
|
|
# File system type to show disk usage (default/empty=all) (see df for specific values)
|
|
FSTYPE=
|
|
# Interface to get the IP address from
|
|
IP_INTERFACE="eth0"
|
|
|
|
function ponyget_Shell()
|
|
{
|
|
echo "$SHELL"
|
|
}
|
|
|
|
function ponyget_User()
|
|
{
|
|
whoami
|
|
}
|
|
|
|
function ponyget_Distro()
|
|
{
|
|
lsb_release -isr | paste "-d " - -
|
|
}
|
|
|
|
function pccolor()
|
|
{
|
|
local color="32";
|
|
if [ "$1" -gt 66 ]
|
|
then
|
|
color="31"
|
|
elif [ "$1" -gt 33 ]
|
|
then
|
|
color="33"
|
|
fi
|
|
echo -e "\x1b[$color;1m"
|
|
}
|
|
|
|
function ponyget_RAM()
|
|
{
|
|
local ramtable=$(free -h | head -n -1)
|
|
local used=$(echo "$ramtable" | tail -n 1 | sed -r "s/ +/\t/g" | cut -f 3)
|
|
local total=$(echo "$ramtable" | sed -n 2p | sed -r "s/ +/\t/g" | cut -f 2)
|
|
|
|
ramtable=$(free -m | head -n -1)
|
|
local used_M=$(echo "$ramtable" | tail -n 1 | 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"
|
|
|
|
|
|
echo -e "$(pccolor $percent)$used\x1b[0m / $total"
|
|
}
|
|
|
|
function ponyget_Swap()
|
|
{
|
|
local swaptable=$(free -h | grep Swap)
|
|
local used=$(echo "$swaptable" | sed -r "s/ +/\t/g" | cut -f 3)
|
|
local total=$(echo "$swaptable" | sed -r "s/ +/\t/g" | cut -f 2)
|
|
|
|
swaptable=$(free -m | grep Swap)
|
|
local used_M=$(echo "$swaptable" | sed -r "s/ +/\t/g" | cut -f 3)
|
|
local total_M=$(echo "$swaptable" | sed -r "s/ +/\t/g" | cut -f 2)
|
|
let percent="$used_M * 100 / $total_M"
|
|
|
|
|
|
echo -e "$(pccolor $percent)$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_Threads()
|
|
{
|
|
nproc --all
|
|
}
|
|
|
|
function ponyget_CPU_Usage()
|
|
{
|
|
printf %.1f%% "$(top -b -n1 | grep "Cpu(s):" | sed -r -e "s/^[^0-9]+//" -e "s/[^0-9]+,\s+/ + /g" -e "s/[^0-9]+$//" | cut -d ' ' -f 1-5 | bc)"
|
|
}
|
|
|
|
function ponyget_CPU_Temperature()
|
|
{
|
|
echo "$(echo "scale=1; $(cat /sys/devices/virtual/thermal/thermal_zone0/temp) / 1000" | bc) C"
|
|
}
|
|
|
|
function ponyget_Uptime()
|
|
{
|
|
uptime | grep -oE "up.*user" | \
|
|
sed -r -e "s/up\s*(.+),\s*[0-9]+\s*user/\1/" -e "s/\s+/ /g"
|
|
}
|
|
|
|
function ponyget_Load()
|
|
{
|
|
uptime | sed -r -e "s/.*load average:\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 dfoption=$FSTYPE
|
|
[ "$dfoption" ] && dfoption="-t $FSTYPE"
|
|
local diskusage=$(df -lh $dfoption -x overlay -x tmpfs -x devtmpfs --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/%// )
|
|
|
|
echo -e "$(pccolor $percent)$used\x1b[0m / $total"
|
|
}
|
|
|
|
function ponyget_Containers()
|
|
{
|
|
local active="$(lxc-ls --active | wc -l)"
|
|
local total="$(lxc-ls | wc -l)"
|
|
local color=32
|
|
if [ "$active" -lt "$total" ]
|
|
then
|
|
color=31
|
|
fi
|
|
echo -e "\x1b[$color;1m$active\x1b[0m / $total active"
|
|
}
|
|
|
|
function ponyget_IP()
|
|
{
|
|
[ -z "$1" ] && inet="inet" || inet="$1"
|
|
ip -o addr | grep -vE "lo\s+inet" | grep -F "$IP_INTERFACE" | grep -Po "$inet\s+\K[^/]+" | xargs echo
|
|
}
|
|
|
|
function ponyget_IPv6()
|
|
{
|
|
ponyget_IP inet6
|
|
}
|
|
|
|
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 * /"
|
|
echo
|
|
echo -e "\t$(bold --list), $(bold --list-ponies)"
|
|
echo -e "\t\tShows a list of possible values for $(bold --pony)."
|
|
|
|
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 list_ponies()
|
|
{
|
|
ponydir=$(get_data_file "rendered/ansi/")
|
|
echo random
|
|
find "$ponydir" -name '*.ansi' -exec basename {} .ansi \; | sort
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
# Returns the number of displayed characters, ignoring color codes
|
|
function strlen_color()
|
|
{
|
|
echo -n "$1" | sed -r 's/\x1b\[[0-9;]+m//g' | wc -c
|
|
}
|
|
|
|
# Truncates a string
|
|
function truncate_color()
|
|
{
|
|
local length="$(strlen_color "$1")"
|
|
local target="$2"
|
|
if [ "$length" -gt "$target" ]
|
|
then
|
|
let cut="$target-$length"
|
|
echo -n "$1" | head -c $cut
|
|
return
|
|
fi
|
|
echo "$1"
|
|
}
|
|
|
|
# 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
|
|
;;
|
|
--list|--list-ponies)
|
|
list_ponies
|
|
exit 0
|
|
;;
|
|
--pony|-p)
|
|
shift
|
|
PONY="$1"
|
|
;;
|
|
-p=*|--pony=*)
|
|
PONY="$(echo "$1" | sed -r "s/^(-p|--pony)=//")"
|
|
;;
|
|
--info|-i)
|
|
infostring=""
|
|
while [ "$2" ] && ! echo "$2" | grep -q -e "-"
|
|
do
|
|
infostring="$infostring $2"
|
|
shift
|
|
done
|
|
select_info "$infostring"
|
|
;;
|
|
-i=*|--info=*)
|
|
select_info "$(echo "$1" | sed -r "s/^(-i|--info)=//")"
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
# Evaluate info
|
|
info_values=()
|
|
let info_key_max_length=0
|
|
let info_val_max_length=0
|
|
function addinfo()
|
|
{
|
|
info_values+=("${2}")
|
|
local key_length=$(echo "$1" | wc -c)
|
|
local val_length=$(echo "${2}" | wc -c)
|
|
[ $key_length -gt $info_key_max_length ] && info_key_max_length=$key_length;
|
|
[ $val_length -gt $info_val_max_length ] && info_val_max_length=$val_length;
|
|
}
|
|
for info in ${INFO[*]}
|
|
do
|
|
if [ "$(type -t ponyget_${info})" = "function" ]
|
|
then
|
|
addinfo $info "$(ponyget_${info})"
|
|
else
|
|
addinfo $info "unsupported"
|
|
fi
|
|
done
|
|
|
|
if [ -f "$PONY" ]
|
|
then
|
|
ponyfile="$PONY"
|
|
elif [ "$PONY" = nopony ]
|
|
then
|
|
ponyfile=/dev/null
|
|
else
|
|
ponydir=$(get_data_file "rendered/ansi/")
|
|
ponyfile="$ponydir/$PONY.ansi"
|
|
if [ -n "$ponydir" -a '(' \! -f "$ponyfile" -o "$PONY" = random ')' ]
|
|
then
|
|
ponyfile="$(find "$ponydir" -name '*.ansi' | shuf | head -n 1)"
|
|
fi
|
|
fi
|
|
|
|
|
|
declare -A info_firstx # Number of characters before the system info output
|
|
info_firstx[applejack-nohat]=64
|
|
info_firstx[applejack]=70
|
|
info_firstx[bigmac]=42
|
|
info_firstx[celestia]=76
|
|
info_firstx[cadance]=55
|
|
info_firstx[derpy]=61
|
|
info_firstx[fluttershy]=67
|
|
info_firstx[great-and-powerful]=68
|
|
info_firstx[luna]=67
|
|
info_firstx[lyra]=61
|
|
info_firstx[pinkie-pie]=61
|
|
info_firstx[rainbow-dash]=61
|
|
info_firstx[rainbow]=61
|
|
info_firstx[rarity]=61
|
|
info_firstx[rose]=61
|
|
info_firstx[trixie]=61
|
|
info_firstx[twilight-alicorn]=61
|
|
info_firstx[twilight-unicorn]=61
|
|
info_firstx[vinyl-scratch-glasses]=69
|
|
info_firstx[vinyl-scratch-noglasses]=69
|
|
|
|
declare -A info_firsty # Number of lines before the system info output
|
|
info_firsty[luna]=18
|
|
info_firsty[rarity]=11
|
|
info_firsty[trixie]=16
|
|
info_firsty[twilight-unicorn]=10
|
|
info_firsty[twilight-alicorn]=10
|
|
|
|
if [ -f "$ponyfile" ]
|
|
then
|
|
lines=$(cat "$ponyfile" | wc -l) # cat to avoid printing file name
|
|
let info_index=0
|
|
COLUMNS=$(tput cols || echo 1024)
|
|
|
|
|
|
if [ "${info_firstx[$PONY]}" ]
|
|
then
|
|
info_x=${info_firstx[$PONY]}
|
|
else
|
|
info_x=80
|
|
fi
|
|
|
|
if [ "${info_firsty[$PONY]}" ]
|
|
then
|
|
start_line=${info_firsty[$PONY]}
|
|
else
|
|
let start_line=($lines-${#INFO[@]})/2
|
|
fi
|
|
|
|
|
|
|
|
let current_line=1
|
|
while IFS= read -r line
|
|
do
|
|
if [ $info_index -lt ${#INFO[@]} -a $current_line -ge $start_line ]
|
|
then
|
|
let msgsize="$COLUMNS - $info_x - $info_key_max_length - 4"
|
|
line="$(truncate_color "$line" $info_x)"
|
|
printf "%s \x1b[31;1m%-${info_key_max_length}s\x1b[0m: %s %s\n" \
|
|
"$line$backspaces" \
|
|
"${INFO[$info_index]}" \
|
|
"$(truncate_color "${info_values[$info_index]}" $msgsize)"
|
|
let info_index++
|
|
else
|
|
echo -n "$line"
|
|
echo -e "\x1b[0m"
|
|
fi
|
|
let current_line++
|
|
done < "$ponyfile"
|
|
else
|
|
let info_index=0
|
|
while [ $info_index -lt ${#INFO[@]} ]
|
|
do
|
|
printf "\x1b[31;1m%-${info_key_max_length}s\x1b[0m: %s\n" \
|
|
"${INFO[$info_index]}" \
|
|
"${info_values[$info_index]}"
|
|
let info_index++
|
|
done
|
|
fi
|
|
|