129 lines
3 KiB
Bash
Executable file
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
|