1#!/usr/bin/env bash
  2# shpool-connect - Interactive shpool session selector with fuzzel/fzf
  3#
  4# Features:
  5# - Lists available hosts from SSH config (detects shpool-enabled hosts)
  6# - Lists sessions on selected host (shows connected/disconnected status)
  7# - Attaches to selected session using kitty (wayland) or current terminal
  8# - Handles already-attached sessions (force detach/attach)
  9set -euo pipefail
 10
 11# Error handler - show error and wait for user input before exiting
 12error_handler() {
 13    local exit_code=$?
 14    local line_no=$1
 15    if [[ $exit_code -ne 0 ]]; then
 16        echo ""
 17        echo "Error on line $line_no: Command exited with status $exit_code"
 18        echo "Press Enter to close..."
 19        read -r
 20    fi
 21    exit $exit_code
 22}
 23trap 'error_handler $LINENO' ERR
 24
 25# Extract shpool-enabled hosts from SSH config
 26# These are hosts with the pattern "host/*" (shpool session aliases)
 27get_shpool_hosts() {
 28    ssh -G "dummy.home/test" 2>/dev/null | head -1 >/dev/null  # validate ssh config works
 29    # Parse SSH config for "Host X/*" patterns and extract X
 30    grep -E '^Host [^*]+/\*$' ~/.ssh/config 2>/dev/null | \
 31        sed 's/^Host //' | sed 's|/\*$||' | sort -u
 32}
 33
 34# Detect environment
 35DESKTOP="${XDG_CURRENT_DESKTOP:-}"
 36GRAPHICS=1
 37
 38if [[ "$DESKTOP" != "sway" && "$DESKTOP" != "niri" ]]; then
 39    if ! command -v xset &>/dev/null; then
 40        GRAPHICS=0
 41    elif ! timeout 1s xset q &>/dev/null 2>&1; then
 42        GRAPHICS=0
 43    fi
 44fi
 45
 46# Select function based on environment
 47select_item() {
 48    local prompt="$1"
 49    if [[ GRAPHICS -eq 0 ]]; then
 50        fzf --prompt "$prompt"
 51    elif [[ "$DESKTOP" == "sway" || "$DESKTOP" == "niri" ]]; then
 52        fuzzel --dmenu --prompt "$prompt" --lines 15 --width 50
 53    else
 54        fzf --prompt "$prompt"
 55    fi
 56}
 57
 58notify() {
 59    local title="$1"
 60    local msg="$2"
 61    if [[ GRAPHICS -eq 1 ]]; then
 62        notify-send "$title" "$msg"
 63    else
 64        echo "$title: $msg"
 65    fi
 66}
 67
 68# Step 1: Select host
 69shpool_hosts=$(get_shpool_hosts)
 70
 71if [[ -z "$shpool_hosts" ]]; then
 72    notify "shpool" "No shpool-enabled hosts found in SSH config"
 73    exit 1
 74fi
 75
 76selected_host=$(echo "$shpool_hosts" | select_item "Host: ")
 77
 78if [[ -z "$selected_host" ]]; then
 79    exit 0
 80fi
 81
 82# Step 2: Get sessions on the host with status
 83# Format: "session-name (status)" where status is connected/disconnected
 84session_list=$(ssh "$selected_host" shpool list 2>/dev/null | tail -n +2 | awk '{
 85    status = $3
 86    if (status == "connected") {
 87        print $1 " (attached)"
 88    } else {
 89        print $1
 90    }
 91}')
 92
 93if [[ -z "$session_list" ]]; then
 94    notify "shpool" "No sessions on $selected_host"
 95    exit 0
 96fi
 97
 98# Add option to create new session
 99sessions="[new session]"$'\n'"$session_list"
100
101selected_session=$(echo "$sessions" | select_item "Session on $selected_host: ")
102
103# Strip status suffix if present (e.g., "session (attached)" -> "session")
104selected_session="${selected_session% (attached)}"
105
106if [[ -z "$selected_session" ]]; then
107    exit 0
108fi
109
110# Handle new session creation
111if [[ "$selected_session" == "[new session]" ]]; then
112    # Prompt for session name
113    if [[ GRAPHICS -eq 0 ]]; then
114        echo -n "Session name: "
115        read -r selected_session
116    elif [[ "$DESKTOP" == "sway" || "$DESKTOP" == "niri" ]]; then
117        selected_session=$(echo "" | fuzzel --dmenu --prompt "New session name: " --width 30)
118    else
119        echo -n "Session name: "
120        read -r selected_session
121    fi
122
123    if [[ -z "$selected_session" ]]; then
124        exit 0
125    fi
126fi
127
128# Step 3: Attach to session
129ssh_target="${selected_host}/${selected_session}"
130
131if [[ GRAPHICS -eq 1 ]] && [[ "$DESKTOP" == "sway" || "$DESKTOP" == "niri" ]]; then
132    # Use kitty for graphical wayland environments
133    if command -v kitty &>/dev/null; then
134        exec kitty --single-instance --instance-group=shpool ssh "$ssh_target"
135    else
136        notify "shpool" "kitty not found, falling back to terminal"
137        exec ssh "$ssh_target"
138    fi
139else
140    # Terminal mode - just SSH directly
141    exec ssh "$ssh_target"
142fi