Dotfiles update 2025-12-25 10:23
This commit is contained in:
Aaron D. Lee
@@ -1,237 +1,96 @@
|
||||
# ============================================================================
|
||||
# SSH Session Manager with Tmux Integration
|
||||
# ============================================================================
|
||||
# Manage SSH connections with automatic tmux session handling
|
||||
# ============================================================================
|
||||
|
||||
# Source shared colors (with fallback)
|
||||
source "${0:A:h}/../lib/colors.zsh" 2>/dev/null || \
|
||||
source "$HOME/.dotfiles/zsh/lib/colors.zsh" 2>/dev/null || {
|
||||
typeset -g DF_GREEN=$'\033[0;32m' DF_BLUE=$'\033[0;34m'
|
||||
typeset -g DF_YELLOW=$'\033[1;33m' DF_CYAN=$'\033[0;36m'
|
||||
typeset -g DF_RED=$'\033[0;31m' DF_NC=$'\033[0m'
|
||||
}
|
||||
|
||||
# ============================================================================
|
||||
# Configuration
|
||||
# ============================================================================
|
||||
source "${0:A:h}/../lib/utils.zsh" 2>/dev/null || \
|
||||
source "$HOME/.dotfiles/zsh/lib/utils.zsh" 2>/dev/null
|
||||
|
||||
typeset -g SSH_PROFILES_FILE="${SSH_PROFILES_FILE:-$HOME/.dotfiles/.ssh-profiles}"
|
||||
typeset -g SSH_AUTO_TMUX="${SSH_AUTO_TMUX:-true}"
|
||||
typeset -g SSH_TMUX_SESSION_PREFIX="${SSH_TMUX_SESSION_PREFIX:-ssh}"
|
||||
typeset -g SSH_SYNC_DOTFILES="${SSH_SYNC_DOTFILES:-ask}"
|
||||
typeset -g SSH_TMUX_PREFIX="${SSH_TMUX_PREFIX:-ssh}"
|
||||
|
||||
# ============================================================================
|
||||
# Helper Functions
|
||||
# ============================================================================
|
||||
|
||||
_ssh_print_step() { echo -e "${DF_BLUE}==>${DF_NC} $1"; }
|
||||
_ssh_print_success() { echo -e "${DF_GREEN}✓${DF_NC} $1"; }
|
||||
_ssh_print_error() { echo -e "${DF_RED}✗${DF_NC} $1"; }
|
||||
_ssh_print_info() { echo -e "${DF_CYAN}ℹ${DF_NC} $1"; }
|
||||
|
||||
_ssh_init_profiles() {
|
||||
if [[ ! -f "$SSH_PROFILES_FILE" ]]; then
|
||||
mkdir -p "$(dirname "$SSH_PROFILES_FILE")"
|
||||
cat > "$SSH_PROFILES_FILE" << 'EOF'
|
||||
# SSH Connection Profiles
|
||||
# Format: name|user@host|port|key_file|options|description
|
||||
EOF
|
||||
_ssh_print_success "Created SSH profiles file: $SSH_PROFILES_FILE"
|
||||
fi
|
||||
_ssh_init() {
|
||||
df_ensure_file "$SSH_PROFILES_FILE" "# SSH Profiles: name|user@host|port|key|options|description"
|
||||
}
|
||||
|
||||
_ssh_parse_profile() {
|
||||
local name="$1"
|
||||
local line=$(grep "^${name}|" "$SSH_PROFILES_FILE" 2>/dev/null | head -1)
|
||||
_ssh_parse() {
|
||||
local line=$(grep "^${1}|" "$SSH_PROFILES_FILE" 2>/dev/null | head -1)
|
||||
[[ -z "$line" ]] && return 1
|
||||
IFS='|' read -r profile_name connection port key_file ssh_opts description <<< "$line"
|
||||
echo "$connection|$port|$key_file|$ssh_opts|$description"
|
||||
echo "$line" | cut -d'|' -f2-
|
||||
}
|
||||
|
||||
# ============================================================================
|
||||
# SSH Profile Management
|
||||
# ============================================================================
|
||||
|
||||
ssh-save() {
|
||||
local name="$1" connection="$2" port="${3:-22}" key_file="${4:-}" options="${5:-}" description="${6:-}"
|
||||
|
||||
_ssh_init_profiles
|
||||
|
||||
[[ -z "$name" || -z "$connection" ]] && {
|
||||
echo "Usage: ssh-save <name> <user@host> [port] [key_file] [options] [description]"
|
||||
return 1
|
||||
}
|
||||
|
||||
if grep -q "^${name}|" "$SSH_PROFILES_FILE" 2>/dev/null; then
|
||||
echo -e "${DF_YELLOW}⚠${DF_NC} Profile '$name' already exists"
|
||||
read -q "REPLY?Overwrite? [y/N]: "; echo
|
||||
[[ ! "$REPLY" =~ ^[Yy]$ ]] && return 1
|
||||
local name="$1" conn="$2" port="${3:-22}" key="${4:-}" opts="${5:-}" desc="${6:-}"
|
||||
[[ -z "$name" || -z "$conn" ]] && { echo "Usage: ssh-save <n> <user@host> [port] [key] [opts] [desc]"; return 1; }
|
||||
_ssh_init
|
||||
grep -q "^${name}|" "$SSH_PROFILES_FILE" 2>/dev/null && {
|
||||
df_confirm "Overwrite '$name'?" || return 1
|
||||
grep -v "^${name}|" "$SSH_PROFILES_FILE" > "${SSH_PROFILES_FILE}.tmp"
|
||||
mv "${SSH_PROFILES_FILE}.tmp" "$SSH_PROFILES_FILE"
|
||||
fi
|
||||
|
||||
echo "${name}|${connection}|${port}|${key_file}|${options}|${description}" >> "$SSH_PROFILES_FILE"
|
||||
|
||||
_ssh_print_success "Saved SSH profile: $name"
|
||||
echo " Connection: $connection"
|
||||
[[ "$port" != "22" ]] && echo " Port: $port"
|
||||
[[ -n "$key_file" ]] && echo " Key: $key_file"
|
||||
}
|
||||
echo "${name}|${conn}|${port}|${key}|${opts}|${desc}" >> "$SSH_PROFILES_FILE"
|
||||
df_print_success "Saved: $name → $conn"
|
||||
}
|
||||
|
||||
ssh-list() {
|
||||
_ssh_init_profiles
|
||||
|
||||
df_print_func_name "SSH Connection Profiles"
|
||||
|
||||
local has_profiles=false
|
||||
while IFS='|' read -r name connection port key options description; do
|
||||
[[ "$name" =~ ^# ]] && continue
|
||||
[[ -z "$name" ]] && continue
|
||||
has_profiles=true
|
||||
|
||||
echo -e "${DF_GREEN}●${DF_NC} ${DF_CYAN}$name${DF_NC}"
|
||||
echo " Connection: $connection"
|
||||
[[ "$port" != "22" && -n "$port" ]] && echo " Port: $port"
|
||||
[[ -n "$key" ]] && echo " Key: $key"
|
||||
[[ -n "$description" ]] && echo " Description: $description"
|
||||
echo
|
||||
_ssh_init
|
||||
df_print_func_name "SSH Profiles"
|
||||
local found=false
|
||||
while IFS='|' read -r name conn port key opts desc; do
|
||||
[[ "$name" =~ ^# || -z "$name" ]] && continue
|
||||
found=true
|
||||
df_print_indent "● $name → $conn"
|
||||
[[ "$port" != "22" && -n "$port" ]] && df_print_indent " Port: $port"
|
||||
[[ -n "$desc" ]] && df_print_indent " $desc"
|
||||
done < "$SSH_PROFILES_FILE"
|
||||
|
||||
[[ "$has_profiles" != true ]] && {
|
||||
_ssh_print_info "No profiles saved yet"
|
||||
echo "Create a profile with: ssh-save myserver user@example.com"
|
||||
}
|
||||
[[ "$found" != true ]] && df_print_info "No profiles. Use: ssh-save name user@host"
|
||||
}
|
||||
|
||||
ssh-delete() {
|
||||
local name="$1"
|
||||
[[ -z "$name" ]] && { echo "Usage: ssh-delete <name>"; return 1; }
|
||||
|
||||
_ssh_init_profiles
|
||||
|
||||
if ! grep -q "^${name}|" "$SSH_PROFILES_FILE" 2>/dev/null; then
|
||||
_ssh_print_error "Profile '$name' not found"
|
||||
return 1
|
||||
fi
|
||||
|
||||
grep -v "^${name}|" "$SSH_PROFILES_FILE" > "${SSH_PROFILES_FILE}.tmp"
|
||||
[[ -z "$1" ]] && { echo "Usage: ssh-delete <n>"; return 1; }
|
||||
grep -q "^${1}|" "$SSH_PROFILES_FILE" 2>/dev/null || { df_print_error "Not found: $1"; return 1; }
|
||||
grep -v "^${1}|" "$SSH_PROFILES_FILE" > "${SSH_PROFILES_FILE}.tmp"
|
||||
mv "${SSH_PROFILES_FILE}.tmp" "$SSH_PROFILES_FILE"
|
||||
_ssh_print_success "Deleted profile: $name"
|
||||
df_print_success "Deleted: $1"
|
||||
}
|
||||
|
||||
ssh-connect() {
|
||||
local name="$1"
|
||||
local session_name="${2:-${SSH_TMUX_SESSION_PREFIX}-${name}}"
|
||||
local name="$1" session="${2:-${SSH_TMUX_PREFIX}-${1}}"
|
||||
[[ -z "$name" ]] && { ssh-list; return 1; }
|
||||
_ssh_init
|
||||
local data=$(_ssh_parse "$name")
|
||||
[[ -z "$data" ]] && { df_print_error "Not found: $name"; return 1; }
|
||||
|
||||
[[ -z "$name" ]] && { echo "Usage: ssh-connect <profile_name>"; ssh-list; return 1; }
|
||||
IFS='|' read -r conn port key opts desc <<< "$data"
|
||||
df_print_step "Connecting: $name"
|
||||
[[ -n "$desc" ]] && df_print_indent "$desc"
|
||||
|
||||
_ssh_init_profiles
|
||||
|
||||
local profile_data=$(_ssh_parse_profile "$name")
|
||||
[[ -z "$profile_data" ]] && { _ssh_print_error "Profile '$name' not found"; return 1; }
|
||||
|
||||
IFS='|' read -r connection port key_file ssh_opts description <<< "$profile_data"
|
||||
|
||||
_ssh_print_step "Connecting to: $name"
|
||||
[[ -n "$description" ]] && echo " $description"
|
||||
|
||||
local ssh_cmd="ssh"
|
||||
[[ -n "$port" && "$port" != "22" ]] && ssh_cmd="$ssh_cmd -p $port"
|
||||
[[ -n "$key_file" ]] && ssh_cmd="$ssh_cmd -i $key_file"
|
||||
[[ -n "$ssh_opts" ]] && ssh_cmd="$ssh_cmd $ssh_opts"
|
||||
ssh_cmd="$ssh_cmd $connection"
|
||||
local cmd="ssh"
|
||||
[[ -n "$port" && "$port" != "22" ]] && cmd="$cmd -p $port"
|
||||
[[ -n "$key" ]] && cmd="$cmd -i $key"
|
||||
[[ -n "$opts" ]] && cmd="$cmd $opts"
|
||||
cmd="$cmd $conn"
|
||||
|
||||
if [[ "$SSH_AUTO_TMUX" == "true" ]]; then
|
||||
_ssh_print_info "Attaching to tmux session: $session_name"
|
||||
local tmux_cmd="tmux attach-session -t $session_name 2>/dev/null || tmux new-session -s $session_name"
|
||||
eval "$ssh_cmd -t '$tmux_cmd'"
|
||||
df_print_info "Tmux session: $session"
|
||||
eval "$cmd -t 'tmux attach -t $session 2>/dev/null || tmux new -s $session'"
|
||||
else
|
||||
eval "$ssh_cmd"
|
||||
eval "$cmd"
|
||||
fi
|
||||
}
|
||||
|
||||
sshf() {
|
||||
if ! command -v fzf &>/dev/null; then
|
||||
_ssh_print_error "fzf not installed"
|
||||
return 1
|
||||
fi
|
||||
|
||||
_ssh_init_profiles
|
||||
|
||||
df_require_cmd fzf || return 1
|
||||
_ssh_init
|
||||
local profiles=()
|
||||
while IFS='|' read -r name connection port key options description; do
|
||||
[[ "$name" =~ ^# ]] && continue
|
||||
[[ -z "$name" ]] && continue
|
||||
local display="$name → $connection"
|
||||
[[ -n "$description" ]] && display="$display ($description)"
|
||||
profiles+=("$name|$display")
|
||||
while IFS='|' read -r name conn port key opts desc; do
|
||||
[[ "$name" =~ ^# || -z "$name" ]] && continue
|
||||
profiles+=("$name|$name → $conn")
|
||||
done < "$SSH_PROFILES_FILE"
|
||||
|
||||
[[ ${#profiles[@]} -eq 0 ]] && { _ssh_print_info "No profiles saved"; return 1; }
|
||||
|
||||
local selection=$(printf '%s\n' "${profiles[@]}" | \
|
||||
fzf --height=50% --layout=reverse --border=rounded --prompt='SSH > ' \
|
||||
--delimiter='|' --with-nth=2)
|
||||
|
||||
[[ -n "$selection" ]] && ssh-connect "${selection%%|*}"
|
||||
[[ ${#profiles[@]} -eq 0 ]] && { df_print_info "No profiles"; return 1; }
|
||||
local sel=$(printf '%s\n' "${profiles[@]}" | fzf $(df_fzf_opts) --delimiter='|' --with-nth=2 --prompt='SSH > ')
|
||||
[[ -n "$sel" ]] && ssh-connect "${sel%%|*}"
|
||||
}
|
||||
|
||||
ssh-reconnect() {
|
||||
local name="${1:-last}"
|
||||
|
||||
if [[ "$name" == "last" ]]; then
|
||||
local last_profile=$(grep "ssh-connect" "$HISTFILE" 2>/dev/null | tail -1 | awk '{print $2}')
|
||||
[[ -z "$last_profile" ]] && { _ssh_print_error "No previous connection found"; return 1; }
|
||||
name="$last_profile"
|
||||
fi
|
||||
|
||||
_ssh_print_info "Reconnecting to: $name"
|
||||
ssh-connect "$name"
|
||||
}
|
||||
|
||||
ssh-sync-dotfiles() {
|
||||
local name="$1"
|
||||
[[ -z "$name" ]] && { echo "Usage: ssh-sync-dotfiles <profile_name>"; return 1; }
|
||||
|
||||
local profile_data=$(_ssh_parse_profile "$name")
|
||||
[[ -z "$profile_data" ]] && { _ssh_print_error "Profile '$name' not found"; return 1; }
|
||||
|
||||
IFS='|' read -r connection port key_file ssh_opts description <<< "$profile_data"
|
||||
|
||||
local dotfiles_dir="${DOTFILES_DIR:-$HOME/.dotfiles}"
|
||||
[[ ! -d "$dotfiles_dir" ]] && { _ssh_print_error "Dotfiles directory not found"; return 1; }
|
||||
|
||||
df_print_func_name "Sync Dotfiles to: $connection"
|
||||
|
||||
local rsync_cmd="rsync -avz --exclude='.git' --exclude='*.local'"
|
||||
[[ -n "$port" && "$port" != "22" ]] && rsync_cmd="$rsync_cmd -e 'ssh -p $port'"
|
||||
[[ -n "$key_file" ]] && rsync_cmd="$rsync_cmd -e 'ssh -i $key_file'"
|
||||
rsync_cmd="$rsync_cmd $dotfiles_dir/ $connection:.dotfiles/"
|
||||
|
||||
_ssh_print_info "Running: $rsync_cmd"
|
||||
|
||||
if eval "$rsync_cmd"; then
|
||||
_ssh_print_success "Dotfiles synced successfully"
|
||||
else
|
||||
_ssh_print_error "Failed to sync dotfiles"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
# ============================================================================
|
||||
# Aliases
|
||||
# ============================================================================
|
||||
|
||||
alias sshl='ssh-list'
|
||||
alias sshs='ssh-save'
|
||||
alias sshc='ssh-connect'
|
||||
alias sshd='ssh-delete'
|
||||
alias sshr='ssh-reconnect'
|
||||
alias sshsync='ssh-sync-dotfiles'
|
||||
|
||||
# ============================================================================
|
||||
# Initialization
|
||||
# ============================================================================
|
||||
|
||||
_ssh_init_profiles
|
||||
alias sshl='ssh-list' sshs='ssh-save' sshc='ssh-connect' sshd='ssh-delete'
|
||||
_ssh_init
|
||||
|
||||
Reference in New Issue
Block a user