Compare commits

...

8 Commits

2 changed files with 123 additions and 46 deletions

View File

@@ -1,22 +1,28 @@
#!/bin/bash
# tmux-daemonizer.sh
# Idempotent tmux daemon manager with optional per-command workdir
# tmux-daemon.sh
# Idempotent tmux daemon manager with config file support
# Load config file if it exists
SCRIPT_DIR=$(dirname "$0")
SCRIPT_NAME=$(basename "$0" .sh)
CONFIG_FILE="${SCRIPT_DIR}/${SCRIPT_NAME}.conf"
if [ -f "$CONFIG_FILE" ]; then
echo "Loading config from $CONFIG_FILE"
source "$CONFIG_FILE"
else
echo "Config file $CONFIG_FILE not found, generating template..."
cat > "$CONFIG_FILE" << 'EOF'
# tmux-daemon.conf
# Configuration file for tmux-daemon.sh
# Session name
SESSION="example"
echo "[main] SESSION: $SESSION"
# Whether to attach to session after setup (0 or 1)
ATTACH_SESSION=0
while getopts "a" opt; do
case "$opt" in
a) ATTACH_SESSION=1 ;;
\?) echo "Usage: $0 [-a]" >&2
exit 1 ;;
esac
done
shift $((OPTIND-1))
# Define commands: "workdir:::command" OR just "command"
# Commands to run (array format)
# Format: "workdir:::command" OR just "command"
CMDS=(
"ping google.com"
"ping google.com"
@@ -26,6 +32,23 @@ CMDS=(
"ping google.com"
"ping google.com"
)
EOF
echo "Generated $CONFIG_FILE with default values. Please edit and run again."
exit 0
fi
# Validate required variables
if [ -z "$SESSION" ]; then
echo "Error: SESSION must be set in $CONFIG_FILE" >&2
exit 1
fi
if [ ${#CMDS[@]} -eq 0 ]; then
echo "Error: CMDS array must be set in $CONFIG_FILE" >&2
exit 1
fi
echo "[main] SESSION: $SESSION"
echo "[main] Commands: ${CMDS[*]}"
# Create session if missing
@@ -115,4 +138,4 @@ echo "[main] Done."
if [ $ATTACH_SESSION -eq 1 ]; then
echo "[main] Attaching to tmux session: $SESSION"
tmux attach -t $SESSION
fi
fi

View File

@@ -2,40 +2,88 @@
# tmux-oneshot.sh
# Run a single command in a tmux session
# Hardcoded command and settings (uncomment and modify as needed)
# SESSION="mytest"
# COMMAND="ping google.com"
# ATTACH_SESSION=1
# Built-in prompt patterns (command_pattern:prompt_pattern:timeout_seconds)
PROMPT_PATTERNS=(
"ssh*:Password|password:5"
"sudo*:Password|password|\\[sudo\\] password for:5"
"su*:Password|password:5"
"mysql*:Enter password:3"
"docker*:Password:2"
)
SESSION=""
ATTACH_SESSION=0
# 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
}
while getopts "t:a" opt; do
case "$opt" in
t) SESSION="$OPTARG" ;;
a) ATTACH_SESSION=1 ;;
\?) echo "Usage: $0 [-t <session_name>] [-a] <command>" >&2
exit 1 ;;
esac
done
shift $((OPTIND-1))
# Use hardcoded command if no arguments provided
if [ $# -eq 0 ]; then
if [ -n "$COMMAND" ]; then
echo "Using hardcoded command: $COMMAND"
else
echo "Error: Command is required" >&2
echo "Usage: $0 [-t <session_name>] [-a] <command>" >&2
exit 1
fi
# Load config file if it exists
SCRIPT_DIR=$(dirname "$0")
SCRIPT_NAME=$(basename "$0" .sh)
CONFIG_FILE="${SCRIPT_DIR}/${SCRIPT_NAME}.conf"
if [ -f "$CONFIG_FILE" ]; then
echo "Loading config from $CONFIG_FILE"
source "$CONFIG_FILE"
else
COMMAND="$*"
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
# Use hardcoded session name, or first word of command if not provided
# 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
# Use session name from config, or first word of first command if not set
if [ -z "$SESSION" ]; then
SESSION=$(echo "$COMMAND" | awk '{print $1}')
SESSION=$(echo "${COMMANDS[0]}" | awk '{print $1}')
fi
# Create session if missing
@@ -46,11 +94,17 @@ else
echo "Session $SESSION exists, reusing..."
fi
# Send command to session
echo "Running command: $COMMAND"
tmux send-keys -t $SESSION "$COMMAND" C-m
# Send commands individually with proper timing
echo "Running commands: ${COMMANDS[*]}"
for cmd in "${COMMANDS[@]}"; do
echo "Running: $cmd"
tmux send-keys -t $SESSION "$cmd" C-m
# Wait for command to complete or prompt for input with timeout
wait_for_prompt "$cmd"
done
if [ $ATTACH_SESSION -eq 1 ]; then
echo "Attaching to tmux session: $SESSION"
tmux attach -t $SESSION
fi
fi