From 0441ed825e6b0544b5d0d0cb40dc4d77fc97bfed Mon Sep 17 00:00:00 2001
From: dcarrillo <daniel.carrillo@gmail.com>
Date: Sun, 22 Mar 2020 19:32:43 +0100
Subject: [PATCH] [polybar] Add custom network usage module

---
 .config/polybar/bar.ini               |  2 +-
 .config/polybar/modules.ini           | 10 +++--
 .config/polybar/scripts/network_usage | 60 +++++++++++++++++++++++++++
 3 files changed, 68 insertions(+), 4 deletions(-)
 create mode 100755 .config/polybar/scripts/network_usage

diff --git a/.config/polybar/bar.ini b/.config/polybar/bar.ini
index ecbb554..0e1461d 100644
--- a/.config/polybar/bar.ini
+++ b/.config/polybar/bar.ini
@@ -67,7 +67,7 @@ font-4 = NotoSans-Regular:size=18:weight=bold;-7
 
 modules-left = spotify
 modules-center = custom_date
-modules-right = updates cpu_bar memory_bar vpn wireless wired alsa_bar
+modules-right = updates cpu_bar memory_bar vpn network_usage alsa_bar
 
 cursor-click = pointer
 cursor-scroll = ns-resize
diff --git a/.config/polybar/modules.ini b/.config/polybar/modules.ini
index 86e90f0..8165d57 100644
--- a/.config/polybar/modules.ini
+++ b/.config/polybar/modules.ini
@@ -80,14 +80,19 @@ label-connected = "%{A1:/usr/bin/bash ~/.config/polybar/scripts/nmcli_manager:}%
 # keep speed meter for other devices
 label-disconnected = "%{A1:/usr/bin/bash ~/.config/polybar/scripts/nmcli_manager:}%downspeed:8%  %upspeed:8% %{A-}"
 
-[module/vpn]
+[module/network_usage]
+type = custom/script
+exec = ~/.config/polybar/scripts/network_usage
+click-left = /usr/bin/bash ~/.config/polybar/scripts/nmcli_manager &
+double-click-left = /usr/bin/bash ~/.config/polybar/scripts/nmcli_manager &
+tail = true
 
+[module/vpn]
 type = custom/script
 exec = ~/.config/polybar/scripts/check_vpn
 interval = 5
 
 [module/spotify]
-
 type = custom/script
 exec = ~/.config/polybar/scripts/spotify.py
 interval = 5
@@ -107,7 +112,6 @@ time-alt = %H:%M:%S
 label = %{F#f5a70a}%{T4}%time%%{F-}%{T-} | %date%
 
 [module/custom_date]
-
 type = custom/script
 exec = ~/.config/polybar/scripts/custom_date
 interval = 1
diff --git a/.config/polybar/scripts/network_usage b/.config/polybar/scripts/network_usage
new file mode 100755
index 0000000..3c74579
--- /dev/null
+++ b/.config/polybar/scripts/network_usage
@@ -0,0 +1,60 @@
+#!/usr/bin/env bash
+
+COUNTER_FILE=/dev/shm/polybar_network_usage
+REFRESH=2
+
+get_icon_by_device()
+{
+    device=$1
+
+    device_type=$(nmcli -t device status | grep -E -m 1 "^$device" | cut -f 2 -d ":")
+
+    case "$device_type" in
+        ethernet)
+            icon=""
+        ;;
+        wifi)
+            ssid=$(nmcli -t device status | grep -E -m 1 "^$device" | cut -f 4 -d ":")
+            icon=" $ssid"
+        ;;
+        *)
+            icon=""
+        ;;
+    esac
+
+    echo $icon
+}
+
+while true; do
+    default_device=$(ip route list | grep -Fm 1 default | cut -d " " -f 5)
+    icon=$(get_icon_by_device "$default_device")
+    now=$(date +%s)
+    counter_age=$(stat --format %Z $COUNTER_FILE 2>/dev/null)
+
+    if [ -f $COUNTER_FILE ] && [ $((now-counter_age)) -lt $((REFRESH+1)) ]; then
+        last_value_in=$(cut -f 1 -d " " $COUNTER_FILE)
+        last_value_out=$(cut -f 2 -d " " $COUNTER_FILE)
+    fi
+
+    current_values=$(awk -v dev="${default_device}:" '{
+        if ($1 == dev) print $2,$10
+    }' < /proc/net/dev)
+
+    current_bytes_in=$(echo "$current_values" | cut -f 1 -d " ")
+    current_bytes_out=$(echo "$current_values" | cut -f 2 -d " ")
+
+    if [ -n "$last_value_in" ]; then
+        bits_in=$((((current_bytes_in-last_value_in) / REFRESH) * 8))
+        bits_out=$((((current_bytes_out-last_value_out) / REFRESH) * 8))
+
+        output=$(numfmt --to iec --format "%8.2f" -z "$bits_in" "$bits_out" 2>/dev/null)
+        echo "$icon $output"
+    else
+        echo "$icon --  --"
+    fi
+
+    echo "$current_values">$COUNTER_FILE
+    unset last_value
+
+    sleep $REFRESH
+done