294 lines
8.7 KiB
Bash
294 lines
8.7 KiB
Bash
# ============================================================================
|
|
# Long-Running Command Notifications
|
|
# ============================================================================
|
|
# Sends notifications when long-running commands complete.
|
|
# Integrates with the existing timer in the adlee theme.
|
|
#
|
|
# Features:
|
|
# - Desktop notifications (notify-send/libnotify)
|
|
# - Terminal bell fallback
|
|
# - Sound notification (optional)
|
|
# - Configurable thresholds
|
|
# - Smart filtering (no notifications for editors, etc.)
|
|
# ============================================================================
|
|
|
|
# Prevent double-sourcing
|
|
[[ -n "$_DF_NOTIFY_LOADED" ]] && return 0
|
|
typeset -g _DF_NOTIFY_LOADED=1
|
|
|
|
# ============================================================================
|
|
# Configuration
|
|
# ============================================================================
|
|
|
|
# Minimum duration (seconds) before notification is sent
|
|
typeset -g DF_NOTIFY_THRESHOLD="${DF_NOTIFY_THRESHOLD:-60}"
|
|
|
|
# Enable/disable notifications
|
|
typeset -g DF_NOTIFY_ENABLED="${DF_NOTIFY_ENABLED:-true}"
|
|
|
|
# Notification methods (space-separated): desktop bell sound
|
|
typeset -g DF_NOTIFY_METHODS="${DF_NOTIFY_METHODS:-desktop bell}"
|
|
|
|
# Sound file for audio notification (optional)
|
|
typeset -g DF_NOTIFY_SOUND="${DF_NOTIFY_SOUND:-/usr/share/sounds/freedesktop/stereo/complete.oga}"
|
|
|
|
# Commands to ignore (editors, pagers, interactive tools)
|
|
typeset -g DF_NOTIFY_IGNORE_CMDS="${DF_NOTIFY_IGNORE_CMDS:-vim nvim nano vi less more man htop top btop watch ssh tmux}"
|
|
|
|
# Only notify on failure (exit code != 0)
|
|
typeset -g DF_NOTIFY_ONLY_FAILURES="${DF_NOTIFY_ONLY_FAILURES:-false}"
|
|
|
|
# ============================================================================
|
|
# Internal State
|
|
# ============================================================================
|
|
|
|
typeset -g _df_notify_cmd=""
|
|
typeset -g _df_notify_start=0
|
|
|
|
# ============================================================================
|
|
# Notification Functions
|
|
# ============================================================================
|
|
|
|
# Check if command should be ignored
|
|
_df_notify_should_ignore() {
|
|
local cmd="$1"
|
|
local first_word="${cmd%% *}"
|
|
|
|
# Check against ignore list
|
|
for ignore in ${(s: :)DF_NOTIFY_IGNORE_CMDS}; do
|
|
[[ "$first_word" == "$ignore" ]] && return 0
|
|
done
|
|
|
|
# Ignore backgrounded commands
|
|
[[ "$cmd" == *'&'* ]] && return 0
|
|
|
|
# Ignore commands run with nohup
|
|
[[ "$cmd" == nohup* ]] && return 0
|
|
|
|
return 1
|
|
}
|
|
|
|
# Send desktop notification
|
|
_df_notify_desktop() {
|
|
local title="$1"
|
|
local body="$2"
|
|
local urgency="${3:-normal}"
|
|
local icon="${4:-terminal}"
|
|
|
|
if command -v notify-send &>/dev/null; then
|
|
notify-send --urgency="$urgency" --icon="$icon" --app-name="Terminal" "$title" "$body" 2>/dev/null
|
|
return 0
|
|
fi
|
|
|
|
# macOS fallback
|
|
if command -v osascript &>/dev/null; then
|
|
osascript -e "display notification \"$body\" with title \"$title\"" 2>/dev/null
|
|
return 0
|
|
fi
|
|
|
|
return 1
|
|
}
|
|
|
|
# Send terminal bell
|
|
_df_notify_bell() {
|
|
printf '\a'
|
|
}
|
|
|
|
# Play sound notification
|
|
_df_notify_sound() {
|
|
local sound_file="$1"
|
|
|
|
if [[ -f "$sound_file" ]]; then
|
|
if command -v paplay &>/dev/null; then
|
|
paplay "$sound_file" &>/dev/null &
|
|
elif command -v aplay &>/dev/null; then
|
|
aplay -q "$sound_file" &>/dev/null &
|
|
elif command -v afplay &>/dev/null; then
|
|
afplay "$sound_file" &>/dev/null &
|
|
fi
|
|
fi
|
|
}
|
|
|
|
# Format duration for display
|
|
_df_notify_format_duration() {
|
|
local secs=$1
|
|
|
|
if (( secs >= 3600 )); then
|
|
printf "%dh %dm %ds" $((secs/3600)) $((secs%3600/60)) $((secs%60))
|
|
elif (( secs >= 60 )); then
|
|
printf "%dm %ds" $((secs/60)) $((secs%60))
|
|
else
|
|
printf "%ds" $secs
|
|
fi
|
|
}
|
|
|
|
# Main notification function
|
|
_df_notify_send() {
|
|
local cmd="$1"
|
|
local exit_code="$2"
|
|
local duration="$3"
|
|
|
|
# Skip if disabled
|
|
[[ "$DF_NOTIFY_ENABLED" != "true" ]] && return
|
|
|
|
# Skip if below threshold
|
|
(( duration < DF_NOTIFY_THRESHOLD )) && return
|
|
|
|
# Skip ignored commands
|
|
_df_notify_should_ignore "$cmd" && return
|
|
|
|
# Skip if only failures and this succeeded
|
|
[[ "$DF_NOTIFY_ONLY_FAILURES" == "true" && $exit_code -eq 0 ]] && return
|
|
|
|
# Build notification content
|
|
local title icon urgency
|
|
local duration_str=$(_df_notify_format_duration "$duration")
|
|
local cmd_short="${cmd:0:50}"
|
|
[[ ${#cmd} -gt 50 ]] && cmd_short="${cmd_short}..."
|
|
|
|
if (( exit_code == 0 )); then
|
|
title="✓ Command Complete"
|
|
icon="dialog-information"
|
|
urgency="normal"
|
|
else
|
|
title="✗ Command Failed (exit $exit_code)"
|
|
icon="dialog-error"
|
|
urgency="critical"
|
|
fi
|
|
|
|
local body="$cmd_short\nDuration: $duration_str"
|
|
|
|
# Send notifications based on configured methods
|
|
for method in ${(s: :)DF_NOTIFY_METHODS}; do
|
|
case "$method" in
|
|
desktop)
|
|
_df_notify_desktop "$title" "$body" "$urgency" "$icon"
|
|
;;
|
|
bell)
|
|
_df_notify_bell
|
|
;;
|
|
sound)
|
|
[[ -n "$DF_NOTIFY_SOUND" ]] && _df_notify_sound "$DF_NOTIFY_SOUND"
|
|
;;
|
|
esac
|
|
done
|
|
}
|
|
|
|
# ============================================================================
|
|
# Hook Functions
|
|
# ============================================================================
|
|
|
|
# Called before command execution
|
|
_df_notify_preexec() {
|
|
_df_notify_cmd="$1"
|
|
_df_notify_start=$SECONDS
|
|
}
|
|
|
|
# Called after command completion
|
|
_df_notify_precmd() {
|
|
local exit_code=$?
|
|
|
|
# Skip if no command was tracked
|
|
[[ -z "$_df_notify_cmd" ]] && return
|
|
[[ $_df_notify_start -eq 0 ]] && return
|
|
|
|
local duration=$((SECONDS - _df_notify_start))
|
|
|
|
# Send notification
|
|
_df_notify_send "$_df_notify_cmd" "$exit_code" "$duration"
|
|
|
|
# Reset state
|
|
_df_notify_cmd=""
|
|
_df_notify_start=0
|
|
}
|
|
|
|
# ============================================================================
|
|
# User Commands
|
|
# ============================================================================
|
|
|
|
# Toggle notifications
|
|
df_notify_toggle() {
|
|
if [[ "$DF_NOTIFY_ENABLED" == "true" ]]; then
|
|
DF_NOTIFY_ENABLED="false"
|
|
echo "Notifications: OFF"
|
|
else
|
|
DF_NOTIFY_ENABLED="true"
|
|
echo "Notifications: ON"
|
|
fi
|
|
}
|
|
|
|
# Set notification threshold
|
|
df_notify_threshold() {
|
|
if [[ -z "$1" ]]; then
|
|
echo "Current threshold: ${DF_NOTIFY_THRESHOLD}s"
|
|
echo "Usage: df_notify_threshold <seconds>"
|
|
else
|
|
DF_NOTIFY_THRESHOLD="$1"
|
|
echo "Threshold set to: ${DF_NOTIFY_THRESHOLD}s"
|
|
fi
|
|
}
|
|
|
|
# Test notification
|
|
df_notify_test() {
|
|
echo "Sending test notification..."
|
|
_df_notify_desktop "Test Notification" "This is a test notification from dotfiles" "normal" "terminal"
|
|
_df_notify_bell
|
|
echo "Done. Did you see/hear it?"
|
|
}
|
|
|
|
# Show notification status
|
|
df_notify_status() {
|
|
source "${DOTFILES_DIR:-$HOME/.dotfiles}/zsh/lib/bootstrap.zsh" 2>/dev/null
|
|
|
|
df_print_func_name "Notification Status"
|
|
echo ""
|
|
df_print_section "Configuration"
|
|
df_print_indent "Enabled: $DF_NOTIFY_ENABLED"
|
|
df_print_indent "Threshold: ${DF_NOTIFY_THRESHOLD}s"
|
|
df_print_indent "Methods: $DF_NOTIFY_METHODS"
|
|
df_print_indent "Only fail: $DF_NOTIFY_ONLY_FAILURES"
|
|
|
|
echo ""
|
|
df_print_section "Capabilities"
|
|
|
|
if command -v notify-send &>/dev/null; then
|
|
df_print_indent "Desktop: ✓ (notify-send)"
|
|
elif command -v osascript &>/dev/null; then
|
|
df_print_indent "Desktop: ✓ (osascript/macOS)"
|
|
else
|
|
df_print_indent "Desktop: ✗ (install libnotify)"
|
|
fi
|
|
|
|
df_print_indent "Bell: ✓ (always available)"
|
|
|
|
if [[ -n "$DF_NOTIFY_SOUND" && -f "$DF_NOTIFY_SOUND" ]]; then
|
|
df_print_indent "Sound: ✓ ($DF_NOTIFY_SOUND)"
|
|
else
|
|
df_print_indent "Sound: ✗ (no sound file configured)"
|
|
fi
|
|
|
|
echo ""
|
|
df_print_section "Ignored Commands"
|
|
df_print_indent "$DF_NOTIFY_IGNORE_CMDS"
|
|
}
|
|
|
|
# ============================================================================
|
|
# Aliases
|
|
# ============================================================================
|
|
|
|
alias notify-toggle='df_notify_toggle'
|
|
alias notify-test='df_notify_test'
|
|
alias notify-status='df_notify_status'
|
|
|
|
# ============================================================================
|
|
# Initialize Hooks
|
|
# ============================================================================
|
|
|
|
# Only set up hooks if not already done (avoid duplicates)
|
|
if [[ -z "$_DF_NOTIFY_HOOKS_SET" ]]; then
|
|
autoload -Uz add-zsh-hook
|
|
add-zsh-hook preexec _df_notify_preexec
|
|
add-zsh-hook precmd _df_notify_precmd
|
|
typeset -g _DF_NOTIFY_HOOKS_SET=1
|
|
fi
|