81 lines
2.2 KiB
Bash
81 lines
2.2 KiB
Bash
#!/bin/bash
|
|
# tmux-oneshot.sh
|
|
# Run a single command in a tmux session
|
|
|
|
# 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-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
|
|
|
|
# Use session name from config, or first word of first command if not set
|
|
if [ -z "$SESSION" ]; then
|
|
SESSION=$(echo "${COMMANDS[0]}" | awk '{print $1}')
|
|
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 command(s) to session using temporary scripts (like tmux-daemon.sh)
|
|
echo "Running commands: ${COMMANDS[*]}"
|
|
for cmd in "${COMMANDS[@]}"; do
|
|
echo "Running: $cmd"
|
|
# Create temporary script to run command (preserves special characters)
|
|
tmpfile=$(mktemp)
|
|
cat >"$tmpfile" <<EOF
|
|
#!/bin/bash
|
|
$cmd
|
|
EOF
|
|
chmod +x "$tmpfile"
|
|
tmux send-keys -t $SESSION "$tmpfile" C-m
|
|
sleep 1 # Brief pause between commands
|
|
done
|
|
|
|
if [ $ATTACH_SESSION -eq 1 ]; then
|
|
echo "Attaching to tmux session: $SESSION"
|
|
tmux attach -t $SESSION
|
|
fi |