Add tmux scripts

2025-11-21 00:39:45 +01:00
commit 3002777e4e
2 changed files with 270 additions and 0 deletions
--- a/tmux-oneshot.sh
+++ b/tmux-oneshot.sh
@@ -0,0 +1,117 @@
+#!/bin/bash
+# tmux-oneshot.sh
+# Run a single command in a tmux session
+
+# Built-in prompt patterns (command_pattern:prompt_pattern:timeout_seconds)
+PROMPT_PATTERNS=(
+    "ssh*:Password|password:10"
+    "sudo*:Password|password|\\[sudo\\] password for:10"
+    "su*:Password|password:10"
+)
+
+# Generic function to wait for prompts based on command patterns
+wait_for_prompt() {
+    local cmd="$1"
+    
+    # Find matching pattern for this command
+    for pattern in "${PROMPT_PATTERNS[@]}"; do
+        IFS=':' read -r cmd_pattern prompt_pattern timeout <<< "$pattern"
+        if [[ "$cmd" == $cmd_pattern ]]; then
+            echo "Waiting for prompt: $prompt_pattern (timeout: ${timeout}s)"
+            timeout=$((timeout * 10))  # Convert to 0.1s intervals
+            while [ $timeout -gt 0 ] && ! tmux capture-pane -t $SESSION -p | tail -3 | grep -E "($prompt_pattern)" >/dev/null 2>&1; do
+                sleep 0.1
+                timeout=$((timeout - 1))
+            done
+            if [ $timeout -eq 0 ]; then
+                echo "Timeout waiting for prompt: $prompt_pattern"
+            fi
+            return
+        fi
+    done
+    
+    # No pattern matched, brief pause
+    sleep 0.5
+}
+
+# Load config file if it exists
+SCRIPT_DIR=$(dirname "$0")
+SCRIPT_NAME=$(basename "$0" .sh)
+CONFIG_FILE="${SCRIPT_DIR}/${SCRIPT_NAME}.sh.conf"
+if [ ! -f "$CONFIG_FILE" ]; then
+    CONFIG_FILE="${SCRIPT_DIR}/${SCRIPT_NAME}.conf"
+fi
+if [ -f "$CONFIG_FILE" ]; then
+    echo "Loading config from $CONFIG_FILE"
+    source "$CONFIG_FILE"
+else
+    CONFIG_FILE="${SCRIPT_DIR}/${SCRIPT_NAME}.sh.conf"
+    echo "Config file $CONFIG_FILE not found, generating template..."
+    cat > "$CONFIG_FILE" << 'EOF'
+# tmux-oneshot.conf
+# Configuration file for tmux-oneshot.sh
+
+# Session name (leave empty to use first word of command)
+SESSION=""
+
+# For multiple commands, use an array:
+# COMMANDS=("ssh root@server" "sudo su" "password123")
+COMMANDS=()
+
+# Whether to attach to session after running command (0 or 1)
+ATTACH_SESSION=0
+EOF
+    echo "Generated $CONFIG_FILE with default values. Please edit and run again."
+    exit 0
+fi
+
+# Validate required variables
+if [ -z "$SESSION" ] && [ ${#COMMANDS[@]} -eq 0 ]; then
+    echo "Error: Either SESSION or COMMANDS must be set in $CONFIG_FILE" >&2
+    exit 1
+fi
+
+# Use command line arguments if provided, otherwise use config
+if [ $# -gt 0 ]; then
+    COMMANDS=("$@")
+    echo "Using command line arguments: $*"
+else
+    if [ ${#COMMANDS[@]} -eq 0 ]; then
+        echo "Error: No commands provided and no commands set in config" >&2
+        exit 1
+    fi
+    echo "Using config commands: ${COMMANDS[*]}"
+fi
+
+# Create session if missing
+if ! tmux has-session -t $SESSION 2>/dev/null; then
+    echo "Creating tmux session: $SESSION"
+    tmux new-session -d -s $SESSION
+else
+    echo "Session $SESSION exists, reusing..."
+fi
+
+# Send commands individually with proper timing
+echo "Running commands: ${COMMANDS[*]}"
+for i in "${!COMMANDS[@]}"; do
+    cmd="${COMMANDS[$i]}"
+    echo "Running: $cmd"
+    tmpfile=$(mktemp)
+    {
+        printf '#!/bin/bash\n'
+        printf '%s\n' "$cmd"
+    } >"$tmpfile"
+    chmod +x "$tmpfile"
+    tmux send-keys -t $SESSION -l "$tmpfile"
+    tmux send-keys -t $SESSION Enter
+    # Only wait if there is a subsequent command
+    if [ $i -lt $((${#COMMANDS[@]} - 1)) ]; then
+        # Wait for command to complete or prompt for input with timeout
+        wait_for_prompt "$cmd"
+    fi
+done
+
+if [ $ATTACH_SESSION -eq 1 ]; then
+    echo "Attaching to tmux session: $SESSION"
+    tmux attach -t $SESSION
+fi