#!/bin/bash
# Checks the low battery warning in bin/sway-status: the real script, with a pretend battery
# and a pretend notify-send. Run by test.sh, or:
#   bash tools/test-battery-bar.sh
# shellcheck disable=SC2016  # the inner shells expand their own variables
HERE=$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")/.." && pwd)
T=$(mktemp -d) || { echo "  FAIL  couldn't make a temp folder (is /tmp full?)"; exit 1; }
fails=0
check() { # DESCRIPTION CONDITION-COMMAND...
    local what=$1
    shift
    if "$@" >/dev/null 2>&1; then printf '  ok    %s\n' "$what"; else printf '  FAIL  %s\n' "$what"; fails=$((fails + 1)); fi
}
not() { ! "$@"; }
S=$HERE/bin/sway-status
mkdir -p "$T/bin" "$T/power/BAT0" "$T/home" "$T/run"
printf '#!/bin/sh\necho "$*" >>"%s/notified"\n' "$T" >"$T/bin/notify-send"
chmod +x "$T/bin/notify-send"
: >"$T/notified"
# (this computer's own memory and logins aren't to set anything off)
printf '#!/bin/sh\n' >"$T/bin/who"
chmod +x "$T/bin/who"
printf 'MemTotal: 10000 kB\nMemAvailable: 5000 kB\n' >"$T/meminfo"
export WREN_MEMINFO=$T/meminfo
battery() { echo "$1" >"$T/power/BAT0/capacity"; echo "$2" >"$T/power/BAT0/status"; }
run() { env HOME="$T/home" XDG_RUNTIME_DIR="$T/run" WREN_POWER_DIR="$T/power" PATH="$T/bin:$PATH" sh "$S" "$@"; }
bar_bat() { # the battery's block in one line the bar is sent, as JSON
    python3 -I -c '
import json, sys
line = sys.stdin.read().strip().rstrip(",")
for b in json.loads(line):
    if b.get("name") == "bat":
        print(json.dumps(b, sort_keys=True))'
}
# lines N: the first N lines the bar is sent, for a battery as it is now (after its 2 of header)
lines() {
    local p i=0 f=$T/out
    : >"$f"
    env HOME="$T/home" XDG_RUNTIME_DIR="$T/run" WREN_POWER_DIR="$T/power" PATH="$T/bin:$PATH" sh "$S" >"$f" 2>/dev/null </dev/null &
    while [ "$i" -lt 150 ] && [ "$(wc -l <"$f")" -lt $(($1 + 2)) ]; do /bin/sleep 0.1; i=$((i + 1)); done
    p=$(cat "$T/run/sway-status.pid" 2>/dev/null)
    [ -z "$p" ] || kill "$p" 2>/dev/null
    wait 2>/dev/null
    sed -n "3,$(($1 + 2))p" "$f"
}
one_line() { lines 1; }

battery 80 Discharging
out=$(one_line | bar_bat)
check "80%: an ordinary block" bash -c 'grep -q "bat 80%" <<<"$1" && ! grep -q "background" <<<"$1"' _ "$out"
battery 16 Discharging
out=$(one_line | bar_bat)
check "16%: still ordinary" not grep -q background <<<"$out"
battery 15 Discharging
out=$(one_line | bar_bat)
check "15%: yellow, black text, readable on either theme" bash -c 'grep -q "\"background\": \"#ffd000\"" <<<"$1" && grep -q "\"color\": \"#000000\"" <<<"$1"' _ "$out"
check "...it says how much" grep -q 'bat 15%' <<<"$out"
battery 6 Discharging
check "6%: still yellow" grep -q '#ffd000' <<<"$(one_line | bar_bat)"
battery 5 Discharging
out=$(one_line | bar_bat)
check "5%: red, white text" bash -c 'grep -q "\"background\": \"#d00000\"" <<<"$1" && grep -q "\"color\": \"#ffffff\"" <<<"$1"' _ "$out"
two=$(lines 2)
n=0
while IFS= read -r l; do if grep -q '#d00000' <<<"$(printf '%s\n' "$l" | bar_bat)"; then n=$((n + 1)); fi; done <<<"$two"
check "...solid: red at every redraw, never flashing (a flash moved the rest of the bar)" test "$n" = 2
check "...short: bat 5%, dies @ 3%" test "$(grep -c 'bat 5%, dies @ 3%' <<<"$two")" = 2
battery 15 Discharging
check "15%: yellow, solid too" test "$(lines 2 | grep -c '#ffd000')" = 2
battery 5 Discharging
battery 5 Charging
out=$(one_line | bar_bat)
check "5% and charging: back to normal" not grep -q background <<<"$out"
check "...shown as charging" grep -q 'bat 5%+' <<<"$out"
battery 5 Full
check "plugged in and not going down (Full): normal" not grep -q background <<<"$(one_line | bar_bat)"
battery 5 'Not charging'
check "...Not charging: normal" not grep -q background <<<"$(one_line | bar_bat)"
rm -f "$T/power/BAT0/capacity" "$T/power/BAT0/status"
check "no battery: no block" test -z "$(one_line | bar_bat)"

# The text console's bar
battery 15 Discharging
check "the console bar: yellow and bold at 15%" grep -q 'bold,fg=yellow\]bat 15%' <<<"$(run --tmux)"
battery 5 Discharging
check "...red and bold at 5%, solid (no blinking)" grep -q 'bold,fg=red\]bat 5%' <<<"$(run --tmux)"
battery 80 Discharging
check "...plain otherwise" not grep -q 'fg=[a-z]*\]bat 80%' <<<"$(run --tmux)"

# One notification on reaching each level, in a running bar (SIGUSR1: draw again now)
battery 80 Discharging
: >"$T/notified"
env HOME="$T/home" XDG_RUNTIME_DIR="$T/run" WREN_POWER_DIR="$T/power" PATH="$T/bin:$PATH" sh "$S" >/dev/null 2>&1 </dev/null &
i=0
while [ "$i" -lt 50 ] && [ ! -s "$T/run/sway-status.pid" ]; do /bin/sleep 0.1; i=$((i + 1)); done
barpid=$(cat "$T/run/sway-status.pid")
redraw() { # (a redraw takes a moment: wait for the next one to end)
    kill -USR1 "$barpid" 2>/dev/null
    /bin/sleep 1.5
}
count() { grep -c "$1" "$T/notified"; }
/bin/sleep 1.5
check "80%: no notification" test "$(wc -l <"$T/notified")" = 0
battery 15 Discharging; redraw
check "reaching 15%: a notification" test "$(count 'Battery low')" = 1
battery 14 Discharging; redraw
battery 13 Discharging; redraw
check "...only one, as it goes on down" test "$(count 'Battery low')" = 1
battery 5 Discharging; redraw
check "reaching 5%: another, a critical one" bash -c 'grep -q "critical.*Battery very low" "$1"' _ "$T/notified"
battery 4 Discharging; redraw
check "...only one" test "$(count 'Battery very low')" = 1
battery 60 Charging; redraw
battery 15 Discharging; redraw
check "charged, then down to 15% again: it warns again" test "$(count 'Battery low')" = 2
kill "$barpid" 2>/dev/null
wait 2>/dev/null

# A clean shutdown at 3%, after a last chance to plug in
printf '#!/bin/sh\necho "$*" >>"%s/systemctl"\n' "$T" >"$T/bin/systemctl"
chmod +x "$T/bin/systemctl"
battery 5 Discharging
check "5%: it says it will shut down at 3%" grep -q 'bat 5%, dies @ 3%' <<<"$(run --tmux)"
mkdir -p "$T/home/.config/wren"; : >"$T/home/.config/wren/no-battery-shutdown"
o=$(run --tmux)
check "...or, with the shutdown switched off, to plug in" bash -c 'grep -q "bat 5%, plug in" <<<"$1" && ! grep -q "dies" <<<"$1"' _ "$o"
rm -f "$T/home/.config/wren/no-battery-shutdown"
battery 80 Discharging
export WREN_BATTERY_SHUTDOWN_WAIT=3
: >"$T/notified"; rm -f "$T/systemctl"
env HOME="$T/home" XDG_RUNTIME_DIR="$T/run" WREN_POWER_DIR="$T/power" PATH="$T/bin:$PATH" sh "$S" >"$T/out" 2>&1 </dev/null &
i=0
while [ "$i" -lt 50 ] && [ ! -s "$T/run/sway-status.pid" ]; do /bin/sleep 0.1; i=$((i + 1)); done
barpid=$(cat "$T/run/sway-status.pid")
/bin/sleep 1
battery 4 Discharging; redraw
check "4%: no shutdown" test ! -e "$T/systemctl"
battery 3 Discharging; redraw
check "3%: a warning first, not a shutdown" bash -c 'grep -q "Shutting down soon" "$1" && [ ! -e "$2" ]' _ "$T/notified" "$T/systemctl"
check "...the bar says the same as before: bat 3%, dies @ 3%" grep -q 'bat 3%, dies @ 3%' "$T/out"
battery 3 Charging; redraw; /bin/sleep 3; redraw
check "plugged in within the minute: called off" test ! -e "$T/systemctl"
battery 2 Discharging; redraw; /bin/sleep 3.5; redraw
check "left on the battery: a clean power off" grep -qx poweroff "$T/systemctl"
kill "$barpid" 2>/dev/null
wait 2>/dev/null
rm -f "$T/systemctl"
mkdir -p "$T/home/.config/wren"; touch "$T/home/.config/wren/no-battery-shutdown"
env HOME="$T/home" XDG_RUNTIME_DIR="$T/run" WREN_POWER_DIR="$T/power" PATH="$T/bin:$PATH" sh "$S" >/dev/null 2>&1 </dev/null &
/bin/sleep 5
kill "$(cat "$T/run/sway-status.pid")" 2>/dev/null
wait 2>/dev/null
check "with no-battery-shutdown in your settings: it stays on" test ! -e "$T/systemctl"
rm -f "$T/home/.config/wren/no-battery-shutdown"
unset WREN_BATTERY_SHUTDOWN_WAIT

# Memory: the same two levels, shown only then, with the battery plain
mem() { printf 'MemTotal: 10000 kB\nMemAvailable: %s kB\n' "$1" >"$T/meminfo"; }
bar_ram() { python3 -I -c '
import json, sys
for b in json.loads(sys.stdin.read().strip().rstrip(",")):
    if b.get("name") == "ram":
        print(json.dumps(b, sort_keys=True))'; }
battery 80 Discharging
mem 5000
check "memory half free: nothing about it in the bar" test -z "$(one_line | bar_ram)"
mem 1500
check "15% left: still nothing" test -z "$(one_line | bar_ram)"
mem 1400
out=$(one_line | bar_ram)
check "14% left: yellow, saying how much" bash -c 'grep -q "#ffd000" <<<"$1" && grep -q "mem 14% left" <<<"$1"' _ "$out"
mem 400
check "4% left: red" grep -q '#d00000' <<<"$(one_line | bar_ram)"
two=$(lines 2)
check "...solid red, never flashing" test "$(grep -c '#d00000' <<<"$two")" = 2
check "...the battery stays plain meanwhile" not grep -q '"bat"[^}]*background' <<<"$two"
battery 15 Discharging
check "...a low battery is yellow beside it" test "$(lines 2 | grep -c '#ffd000')" = 2
battery 80 Discharging
check "the console bar has it too, without blinking" grep -q 'bold,fg=red\]mem 4% left' <<<"$(run --tmux)"
mem 5000

# The alerts: nothing while there's nothing to say, each one only while it's so
bar_alerts() { one_line | python3 -I -c '
import json, sys
print(" ".join(b["name"] for b in json.loads(sys.stdin.read().strip().rstrip(",")) if b.get("urgent")))'; }
mkdir -p "$T/home/.local/bin" "$T/modules"
export WREN_REBOOT_FILE=$T/reboot-required WREN_MODULES=$T/modules
uname -r >"$T/kernel"
mkdir "$T/modules/$(cat "$T/kernel")"
printf '#!/bin/sh\n[ ! -e "%s/ssh" ] || echo "user pts/0 2026-11-07 10:00 (203.0.113.9)"\necho "user tty1 2026-11-07 09:00"\n' "$T" >"$T/bin/who"
printf '#!/bin/sh\ncat "%s/$1" 2>/dev/null || echo off\n' "$T" >"$T/home/.local/bin/radios"
printf '#!/bin/sh\necho "F S U A %%s /"; echo "/dev/x 1 1 1 $(cat "%s/use") /"\n' "$T" | sed 's/%%s/Capacity/' >"$T/bin/df"
printf '#!/bin/sh\n[ -e "%s/vault-open" ]\n' "$T" >"$T/bin/mountpoint"
printf '#!/bin/sh\n' >"$T/home/.local/bin/vault"
chmod +x "$T/bin/who" "$T/home/.local/bin/radios" "$T/bin/df" "$T/bin/mountpoint" "$T/home/.local/bin/vault"
echo 50% >"$T/use"
check "nothing on: no alerts" test -z "$(bar_alerts)"
touch "$T/ssh"
check "a login from another computer: ssh" grep -qw ssh <<<"$(bar_alerts)"
rm "$T/ssh"
echo on >"$T/mic"
check "the microphone on: mic" grep -qw mic <<<"$(bar_alerts)"
# the divider before an alert is a block of its own, so the alert's color stops at its text
alert_plain_divider() {
    one_line | python3 -I -c '
import json, sys
bl = json.loads(sys.stdin.read().strip().rstrip(","))
i = [k for k, b in enumerate(bl) if b.get("urgent")]
assert i and all("|" not in bl[k]["full_text"] and bl[k - 1]["full_text"] == " | " and not bl[k - 1].get("urgent") for k in i), bl'
}
check "...its colored block holds its text only, not the divider before it" alert_plain_divider
rm "$T/mic"
echo on >"$T/camera"
check "the webcam on: cam" grep -qw cam <<<"$(bar_alerts)"
rm "$T/camera"
touch "$T/reboot-required"
check "a package asks for a restart: reboot" grep -qw reboot <<<"$(bar_alerts)"
rm "$T/reboot-required"
mkdir "$T/modules/99.0-newer"
check "a newer kernel than the running one: reboot" grep -qw reboot <<<"$(bar_alerts)"
rmdir "$T/modules/99.0-newer"
echo 95% >"$T/use"
check "the disk 95% full: disk" grep -qw disk <<<"$(bar_alerts)"
echo 50% >"$T/use"
touch "$T/vault-open"
check "the vault open: vault" grep -qw vault <<<"$(bar_alerts)"
rm "$T/vault-open"
check "...and gone again: no alerts" test -z "$(bar_alerts)"
touch "$T/ssh"; echo on >"$T/mic"
check "the console bar has them too" bash -c 'grep -q "ssh login" <<<"$1" && grep -q "mic ON" <<<"$1"' _ "$(run --tmux)"
rm "$T/ssh"; echo off >"$T/mic"

# A network's name is whatever its owner typed: it must not break the bar, or the warnings go too
printf '#!/bin/sh\nprintf "evil\\033[31m\\001\\177\\"quote\\\\\\t#[x]:802-11-wireless\\n"\n' >"$T/bin/nmcli"
chmod +x "$T/bin/nmcli"
touch "$T/ssh"
check "control characters in a Wi-Fi name: the bar is still good JSON, with its warnings" grep -qw ssh <<<"$(bar_alerts)"
wifi_shown=$(one_line | python3 -I -c '
import json, sys
print(next(b["full_text"] for b in json.loads(sys.stdin.read().strip().rstrip(",")) if b.get("name") == "wifi"))' 2>/dev/null)
check "...the name shows without them, its quote kept" test "$wifi_shown" = ' | evil[31m"quote#[x]'
rm "$T/ssh" "$T/bin/nmcli"
rm -rf "$T"
[ "$fails" -eq 0 ]