Compare commits
8 Commits
31a0cb1a9f
...
master
Author | SHA1 | Date | |
---|---|---|---|
21204251c9 | |||
6d79ff93cb | |||
db83cbbffc | |||
5d633dbdbb | |||
9ed55d2171 | |||
cc860cb554 | |||
ef42a7089f | |||
e46aba65be |
@@ -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
|
118
tmux-oneshot.sh
118
tmux-oneshot.sh
@@ -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
|
Reference in New Issue
Block a user