tmux-kanagawa/scripts/network_bandwidth.sh

129 lines
3 KiB
Bash
Executable file

#!/usr/bin/env bash
# INTERVAL is equal to 1s because we want to express the bandwidth in sec
readonly INTERVAL=1
# UPLOAD and DOWNLOAD index
readonly UPLOAD=0
readonly DOWNLOAD=1
# SIZE index are the multiple of the unit byte and value the internationally recommended unit symbol in sec
readonly SIZE=(
[1]='B/s'
[1024]='kB/s'
[1048576]='MB/s'
[1073741824]='GB/s'
)
# interface_get try to automaticaly get the used interface if network_name is empty
interface_get() {
name="$(tmux show-option -gqv "@dracula-network-bandwidth")"
if [[ -z $name ]]; then
case "$(uname -s)" in
Linux)
if type ip >/dev/null; then
name="$(ip -o route get 192.168.0.0 | awk '{print $5}')"
fi
;;
Darwin)
if type route >/dev/null; then
name="$(route -n get 192.168.0.0 2>/dev/null | awk '/interface: / {print $2}')"
fi
;;
esac
fi
echo "$name"
}
# interface_bytes give an interface name and return both tx/rx Bytes, separated by whitespace (upload first)
interface_bytes() {
case "$(uname -s)" in
Linux)
upload=$(cat "/sys/class/net/$1/statistics/tx_bytes")
download=$(cat "/sys/class/net/$1/statistics/rx_bytes")
echo "$upload $download"
;;
Darwin)
# column 7 is Ibytes (in bytes, rx, download) and column 10 is Obytes (out bytes, tx, upload)
netstat -nbI "$1" | tail -n1 | awk '{print $10 " " $7}'
;;
esac
}
# get_bandwidth return the number of bytes exchanged for tx and rx
get_bandwidth() {
local upload=0
local download=0
IFS=' ' read -r upload download <<< "$(interface_bytes "$1")"
# wait for interval to calculate the difference
sleep "$INTERVAL"
IFS=' ' read -r new_upload new_download <<< "$(interface_bytes "$1")"
upload=$(( $new_upload - $upload ))
download=$(( $new_download - $download ))
# set to 0 by default
echo "${upload:-0} ${download:-0}"
}
# bandwidth_to_unit convert bytes into its highest unit and add unit symbol in sec
bandwidth_to_unit() {
local size=1
for i in "${!SIZE[@]}"; do
if (($1 < i)); then
break
fi
size="$i"
done
local result="0.00"
if (($1 != 0)); then
result="$(bc <<<"scale=2; $1 / $size")"
fi
echo "$result ${SIZE[$size]}"
}
main() {
counter=0
bandwidth=()
network_name=""
show_interface="$(tmux show-option -gqv "@dracula-network-bandwidth-show-interface")"
interval_update="$(tmux show-option -gqv "@dracula-network-bandwidth-interval")"
if [[ -z $interval_update ]]; then
interval_update=0
fi
if ! command -v bc &> /dev/null
then
echo "command bc could not be found!"
exit 1
fi
while true; do
if ((counter == 0)); then
counter=60
network_name="$(interface_get)"
fi
IFS=" " read -ra bandwidth <<<"$(get_bandwidth "$network_name")"
if [[ $show_interface == "true" ]]; then echo -n "[$network_name] "; fi
echo "$(bandwidth_to_unit "${bandwidth[$DOWNLOAD]}") • ↑ $(bandwidth_to_unit "${bandwidth[$UPLOAD]}")"
((counter = counter - 1))
sleep "$interval_update"
done
}
#run main driver
main