#!/bin/sh
# Requires: rofi, wireplumber (wpctl)
# Audio menu: volume, mute, and which sink the sound comes out of. This is
# the rofi replacement for the gum-based audio-menu/audio-output pair, so it
# no longer needs a terminal window to run in.
set -u

STEP=${ROFI_AUDIO_STEP:-5}
NOTIFY_ID=7303

notify() {
    if command -v dunstify >/dev/null 2>&1; then
        dunstify -r "$NOTIFY_ID" "Audio" "$1"
    else
        notify-send "Audio" "$1"
    fi
}

# wpctl has no machine-readable listing, so carve the Sinks block out of
# `status`. Output: "<*|space>\t<id>\t<name>".
sinks() {
    wpctl status 2>/dev/null | awk '
        /^Audio/                                { audio = 1; next }
        /^[A-Z]/                                { audio = 0 }
        audio && /Sinks:/ && !/endpoints:/      { insec = 1; next }
        audio && /Devices:|Sources:|Streams:|endpoints:|Filters:/ { insec = 0; next }
        insec {
            line = $0
            prefix = line; sub(/[0-9].*$/, "", prefix)
            def = (prefix ~ /\*/) ? "*" : " "
            sub(/^[^0-9]*/, "", line)
            if (line !~ /^[0-9]+\./) next
            id = line;   sub(/\..*$/, "", id)
            name = line; sub(/^[0-9]+\.[[:space:]]*/, "", name)
            sub(/[[:space:]]*\[vol:.*$/, "", name)
            sub(/[[:space:]]+$/, "", name)
            print def "\t" id "\t" name
        }
    '
}

# @DEFAULT_AUDIO_SINK@ always resolves to whatever is current, so the volume
# actions never need to know which sink that is.
volume() { wpctl get-volume @DEFAULT_AUDIO_SINK@ 2>/dev/null; }

vol_percent() { volume | awk '{ printf "%d", $2 * 100 }'; }

# The icon tracks the level the same way the status bar blocks do, so a
# notification reads the same whether it came from here or from a media key.
vol_icon() {
    case "$(volume)" in
        *MUTED*) printf '󰝟' ;;
        *) v=$(vol_percent)
           if   [ "$v" -eq 0 ];  then printf '󰕿'
           elif [ "$v" -lt 34 ]; then printf '󰖀'
           elif [ "$v" -lt 67 ]; then printf '󰕾'
           else                       printf '󰕾'
           fi ;;
    esac
}

report() { notify "$(vol_icon)  $(vol_percent)%"; }

SINKS=$(sinks)
[ -n "$SINKS" ] || { notify "No sinks found — is pipewire running?"; exit 1; }

CURRENT=$(printf '%s\n' "$SINKS" | awk -F'\t' '$1 == "*" { print $3; exit }')
MIC=$(wpctl get-volume @DEFAULT_AUDIO_SOURCE@ 2>/dev/null | grep -q MUTED && echo "muted" || echo "live")

ACTIONS=$(printf '󰝝  volume up (+%s%%)\n󰝞  volume down (-%s%%)\n󰝟  toggle mute\n󰍬  toggle mic (currently %s)\n󰓃  automatic output\n' "$STEP" "$STEP" "$MIC")

# Sinks are listed by name with their wpctl id hidden behind two spaces, the
# same split the actions use, so one case statement handles both.
SINKLIST=$(printf '%s\n' "$SINKS" | awk -F'\t' '{
    icon = ($1 == "*") ? "󰄬" : "󰓃"
    printf "%s  %s\t%s\n", icon, $3, $2
}')

CHOICE=$(printf '%s\n%s\n' "$ACTIONS" "$(printf '%s\n' "$SINKLIST" | cut -f1)" |
    rofi -dmenu -i -no-custom -l 10 -p "audio" \
        -mesg "$(vol_icon)  $(vol_percent)% — ${CURRENT:-no default sink}") || exit 0
[ -n "$CHOICE" ] || exit 0

case "$CHOICE" in
    *"volume up"*)
        # The cap keeps a stray keystroke from pushing the sink into the
        # clipping range above 100%.
        wpctl set-volume -l 1.0 @DEFAULT_AUDIO_SINK@ "$STEP%+"
        report ;;
    *"volume down"*)
        wpctl set-volume @DEFAULT_AUDIO_SINK@ "$STEP%-"
        report ;;
    *"toggle mute"*)
        wpctl set-mute @DEFAULT_AUDIO_SINK@ toggle
        report ;;
    *"toggle mic"*)
        wpctl set-mute @DEFAULT_AUDIO_SOURCE@ toggle
        wpctl get-volume @DEFAULT_AUDIO_SOURCE@ | grep -q MUTED &&
            notify "󰍭  microphone muted" || notify "󰍬  microphone live" ;;
    *"automatic output"*)
        wpctl clear-default
        notify "󰓃  output: automatic (wireplumber picks)" ;;
    *)
        ID=$(printf '%s\n' "$SINKLIST" | awk -F'\t' -v c="$CHOICE" '$1 == c { print $2; exit }')
        [ -n "$ID" ] || exit 0
        wpctl set-default "$ID"
        notify "󰓃  output: ${CHOICE##*  }" ;;
esac