diff options
Diffstat (limited to 'assistant.sh')
| -rwxr-xr-x | assistant.sh | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/assistant.sh b/assistant.sh new file mode 100755 index 0000000..e288afd --- /dev/null +++ b/assistant.sh @@ -0,0 +1,94 @@ +#!/bin/bash +# desktop-assistant: start, stop, restart or check the voice loop. +# Copyright (C) 2026 Danilo M. <danix@danix.xyz> +# +# This program is free software; you can redistribute it and/or modify it +# under the terms of the GNU General Public License version 2 as published +# by the Free Software Foundation. +# +# This program is distributed in the hope that it will be useful, but +# WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General +# Public License for more details. +# +# You should have received a copy of the GNU General Public License along +# with this program; if not, write to the Free Software Foundation, Inc., +# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. +# +# assistant.sh start|stop|restart|status +# +# stop: SIGTERM any running assistant.py (whisper-server and amixer die +# with it through PDEATHSIG), SIGKILL after 10 s; same for the quickshell +# overlay (OVERLAY, default the quickshell repo next to this one). +# start: run both in their own session, detached from the terminal, +# logging where the Hyprland autostart does; does nothing for one already +# running. restart: stop, then start. status: processes, children, stray +# recorders, socket, pause flags and the log's tail; exit 1 if either is +# down. Run it from the login session: it needs its +# audio, bus, Wayland display and XDG_RUNTIME_DIR. + +set -eu + +dir=$(cd "$(dirname "$0")" && pwd) +log="${XDG_RUNTIME_DIR:?not in a login session}/desktop-assistant.log" +overlay=$(realpath -m "${OVERLAY:-$dir/../quickshell/assistant}") +pat='^\S*python3 (-u )?\S*assistant\.py' +qpat="^qs -p $overlay\$" + +kill_wait() { + pkill -u "$USER" -f "$1" || return 0 + for _ in $(seq 50); do + pgrep -u "$USER" -f "$1" >/dev/null || return 0 + sleep 0.2 + done + pkill -KILL -u "$USER" -f "$1" && sleep 0.5 || true +} + +running() { pgrep -u "$USER" -f "$1" >/dev/null; } + +stop() { + kill_wait "$pat" + kill_wait "$qpat" +} + +start() { + cd "$dir" + running "$pat" || setsid -f python3 -u "$dir/assistant.py" >>"$log" 2>&1 </dev/null + running "$qpat" || setsid -f qs -p "$overlay" >/dev/null 2>&1 </dev/null + sleep 0.5 + status +} + +show() { + local pid + pid=$(pgrep -u "$USER" -of "$2") || { echo "$1: not running"; return 1; } + echo "$1: running" + ps -o pid=,etime=,rss=,args= -p "$pid" | + awk '{ printf " pid %s, up %s, %d MB\n %s", $1, $2, $3 / 1024, $4 + for (i = 5; i <= NF; i++) printf " %s", $i; print "" }' + [ "$1" = assistant ] || return 0 + pgrep -a -P "$pid" | sed 's/^/ child /' +} + +status() { + local rc=0 f + show assistant "$pat" || rc=1 + show overlay "$qpat" || rc=1 + # Children get PDEATHSIG; one left over means the assistant died under it. + pgrep -u "$USER" -af "^arecord .*desktop-assistant\.raw" | sed 's/^/stray: /' || true + [ -S "$XDG_RUNTIME_DIR/desktop-assistant.sock" ] && echo "socket: present" || echo "socket: missing" + for f in presentation gaming; do + [ "$(cat "$XDG_RUNTIME_DIR/status.$f" 2>/dev/null)" = 1 ] && echo "paused: $f mode on" + done + echo "log: $log" + tail -n 5 "$log" 2>/dev/null | sed 's/^/ /' + return $rc +} + +case "${1:-}" in + start) start ;; + stop) stop ;; + restart) stop; start ;; + status) status ;; + *) echo "usage: $0 start|stop|restart|status" >&2; exit 2 ;; +esac |
