From 06434a49350a669263b7bc45e5210b5f3e699ced Mon Sep 17 00:00:00 2001
From: spinach <19keegandeppe@gmail.com>
Date: Wed, 8 Mar 2023 12:39:51 -0500
Subject: [PATCH] init

---
 device_battery.sh | 186 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 186 insertions(+)
 create mode 100755 device_battery.sh

diff --git a/device_battery.sh b/device_battery.sh
new file mode 100755
index 0000000..708e246
--- /dev/null
+++ b/device_battery.sh
@@ -0,0 +1,186 @@
+#!/bin/bash
+
+# parses upower -e and grabs device/battery level
+
+usage() {
+    cat <<EOF
+usage $0: [-i][-d][-b][-c][-h]
+
+Options:
+  -b, --battery-icon    uses battery level NF icon instead of percentage
+  -c, --color           enables polybar color codes
+  -d, --dev-icon        uses device NF icon instead of name 
+  -i, --icon-test       prints all icons used to test rendering
+  -h, --help            shows this message
+EOF
+}
+
+LOW_BATTERY_THRESHOLD=15
+LOW_BATTERY_COLOR='%{F#F00}' # polybar red
+CHARGING_COLOR='%{F#0F0}' # polybar green
+MOUSE_ICON=$(echo -e '\Uf098b') # bt mouse
+HEADSET_ICON=$(echo -e '\Uf0970')
+
+battery_icon() {
+    # grabs corresponding battery level icon
+    if [ "$2" == "charging" ] ; then
+        # mouse is charging, battery icon represents
+        if [ $1 -gt 95 ] ; then
+            echo -e '\Uf0085'
+        elif [ $1 -gt 85 ] ; then
+            echo -e '\Uf008b'
+        elif [ $1 -gt 75 ] ; then
+            echo -e '\Uf008a'
+        elif [ $1 -gt 65 ] ; then
+            echo -e '\Uf089e'
+        elif [ $1 -gt 55 ] ; then
+            echo -e '\Uf0089'
+        elif [ $1 -gt 45 ] ; then
+            echo -e '\Uf089d'
+        elif [ $1 -gt 35 ] ; then
+            echo -e '\Uf0088'
+        elif [ $1 -gt 25 ] ; then
+            echo -e '\Uf0087'
+        elif [ $1 -gt 15 ] ; then
+            echo -e '\Uf0086'
+        elif [ $1 -gt 5 ] ; then
+            echo -e '\Uf089c'
+        else
+            echo -e '\Uf089f'
+        fi
+    else
+        # regular battery icon
+        if [ $1 -gt 95 ] ; then
+            echo -e '\Uf0079'
+        elif [ $1 -gt 85 ] ; then
+            echo -e '\Uf0081'
+        elif [ $1 -gt 75 ] ; then
+            echo -e '\Uf0081'
+        elif [ $1 -gt 65 ] ; then
+            echo -e '\Uf0080'
+        elif [ $1 -gt 55 ] ; then
+            echo -e '\Uf007f'
+        elif [ $1 -gt 45 ] ; then
+            echo -e '\Uf007e'
+        elif [ $1 -gt 35 ] ; then
+            echo -e '\Uf007d'
+        elif [ $1 -gt 25 ] ; then
+            echo -e '\Uf007c'
+        elif [ $1 -gt 15 ] ; then
+            echo -e '\Uf007b'
+        elif [ $1 -gt 5 ] ; then
+            echo -e '\Uf007a'
+        else
+            echo -e '\Uf008e'
+        fi
+    fi
+}
+
+icon_test() {
+    echo "Icon test for $1. (0-100)"
+    for ((i=-4; i<=100; i+=10)); do
+        battery_icon $i "$1"
+    done
+}
+
+identifier() {
+    # returns device name or icon based on options; dev in $1
+    
+    if [ -n "$DEV_ICON" ] ; then
+        if [ -n "$(upower -i "$1" | grep 'mouse')" ] ; then
+            printf '%s ' "$MOUSE_ICON"
+        fi
+
+        if [ -n "$(upower -i "$1" | grep 'headset')" ] ; then
+            printf '%s ' "$HEADSET_ICON"
+        fi
+    else
+        printf '%s' "$(upower -i "$1" | grep 'model:' | awk '{print $2}')"
+    fi
+}
+
+battery() {
+    # gets device battery level; dev in $1
+
+    level=$(upower -i "$1" | grep 'percentage:' | awk '{print $2}' | tr -d '%')
+    state=$(upower -i "$1" | grep 'state:' | awk '{print $2}')
+
+    if [ "$state" == "unknown" ] ; then
+        # override to always show if device asleep
+        echo -e '\Uf04b2 ' && exit 0
+    fi
+
+    color "$state" $level
+
+    if [ -n "$BATTERY_ICON" ] ; then
+        battery_icon $level "$state"
+    else
+        printf '%d%%' $level
+    fi
+}    
+
+color() {
+    # sets battery color based on passed params
+    if [ -n "$COLOR" ] ; then
+        if [ "$1" == "charging" ] ; then
+            echo "$CHARGING_COLOR"
+        elif [ $2 -le $LOW_BATTERY_THRESHOLD ] ; then
+            echo "$LOW_BATTERY_COLOR"
+        fi
+    fi
+}
+
+# parsing long args
+for arg in "$@"; do
+    shift
+    case "$arg" in
+        '--help')           set -- "$@" '-h'    ;;
+        '--color')          set -- "$@" '-c'    ;;
+        '--icon-test')      set -- "$@" '-i'    ;;
+        '--dev-icon')       set -- "$@" '-d'    ;;
+        '--battery-icon')   set -- "$@" '-b'    ;;
+        *)                  set -- "$@" "$arg"  ;;
+    esac
+done
+
+# parsing args
+while getopts "hcidb" opt ; do
+    case "$opt" in
+        'h')
+            usage
+            exit 0
+            ;;
+        'c')
+            COLOR=true
+            ;;
+        'i')
+            echo "Headset: $HEADSET_ICON"
+            echo "Mouse: $MOUSE_ICON"
+            icon_test "default"
+            icon_test "charging"
+            exit 0
+            ;;
+        'd')
+            DEV_ICON=true
+            ;;
+        'b')
+            BATTERY_ICON=true
+            ;;
+        '?')
+            usage
+            exit 1
+            ;;
+    esac
+done
+
+# go through devs
+devs=$(upower -e | grep -v "DisplayDevice") # don't use the monitor
+
+for dev in $devs; do
+    # getting device info
+    dev_id=$(identifier "$dev")
+    dev_battery=$(battery "$dev")
+
+    printf ' %s %s' "$dev_id" "$dev_battery"
+done
+echo