sshko.sh

#!/bin/bash

logFile=~/.sshLog.txt
[[ -e "$logFile" ]] || : > "$logFile"

# -------- Formatting helpers --------
fmt_row() {
    local left="$1" note="$2" left_w="$3" note_w="$4"
    note="${note//$'\n'/ }"
    [[ ${#left} -gt $left_w ]] && left="${left:0:$((left_w-1))}…"
    [[ ${#note} -gt $note_w ]] && note="${note:0:$((note_w-1))}…"
    printf "%-${left_w}s | %-${note_w}s" "$left" "$note"
}

# -------- File helpers --------
# Build arrays from file:
#   menu_pairs[]: <idx> "<formatted row>"
#   specs[]: left field (ssh spec)
#   notes[]: note
#   lines[]: trimmed full line as stored
build_arrays() {
    menu_pairs=(); specs=(); notes=(); lines=()
    local key=0
    while IFS= read -r raw || [[ -n "$raw" ]]; do
        [[ -z "$raw" || "$raw" =~ ^[[:space:]]*# ]] && continue
        # Normalize line (trim)
        local line="$raw"
        line="${line#"${line%%[![:space:]]*}"}"
        line="${line%"${line##*[![:space:]]}"}"

        local left="${line%%,*}"
        local note="${line#*,}"
        [[ "$note" == "$line" ]] && note=""

        # Trim fields
        left="${left#"${left%%[![:space:]]*}"}"; left="${left%"${left##*[![:space:]]}"}"
        note="${note#"${note%%[![:space:]]*}"}"; note="${note%"${note##*[![:space:]]}"}"

        local label; label=$(fmt_row "$left" "$note" "$LEFT_W" "$NOTE_W")
        menu_pairs+=("$key" "$label")
        specs+=("$left")
        notes+=("$note")
        lines+=("$line")
        ((key++))
    done < "$logFile"
}

# Delete entry by index safely (rewrite file)
delete_index() {
    local del="$1"
    local tmp
    tmp=$(mktemp)
    local i=0
    while IFS= read -r raw || [[ -n "$raw" ]]; do
        # keep comments/blank lines as-is in output
        if [[ -z "$raw" || "$raw" =~ ^[[:space:]]*# ]]; then
            echo "$raw" >> "$tmp"
            continue
        fi
        # Align with lines[] by trimming same way
        local line="$raw"
        line="${line#"${line%%[![:space:]]*}"}"
        line="${line%"${line##*[![:space:]]}"}"

        # Only count non-comment, non-empty lines toward index
        if [[ $i -ne $del ]]; then
            echo "$line" >> "$tmp"
        fi
        ((i++))
    done < "$logFile"
    mv "$tmp" "$logFile"
}

# -------- Menu / actions --------
show_menu() {
    while true; do
        if ! [ -s "$logFile" ]; then
            echo "ssh log file is empty"
            exit 1
        fi

        HEIGHT=22
        WIDTH=90
        CHOICE_HEIGHT=18
        BACKTITLE="This is made, because you are super lazy!!!!"
        TITLE="SSHko"
        MENU="Choose one of the following ssh connections:"

        LEFT_W=48
        NOTE_W=$(( WIDTH - 10 - LEFT_W ))

        build_arrays
        if [[ ${#menu_pairs[@]} -eq 0 ]]; then
            echo "ssh log file is empty"
            exit 1
        fi

        header=$(fmt_row "SSH" "NOTE" "$LEFT_W" "$NOTE_W")
        CHOICE=$(dialog --clear --title "$TITLE" --backtitle "$BACKTITLE" \
            --ok-label "Connect" --cancel-label "Exit" \
            --extra-button --extra-label "Delete" \
            --menu "$MENU\n$header" \
            $HEIGHT $WIDTH $CHOICE_HEIGHT "${menu_pairs[@]}" \
            2>&1 >/dev/tty)
        status=$?
        clear

        case $status in
            0)  # Connect
                [[ -z "$CHOICE" ]] && continue
                spec="${specs[$CHOICE]}"
                IFS=' ' read -r -a SSHARGS <<< "$spec"
                exec ssh "${SSHARGS[@]}"
                ;;
            3)  # Delete (extra button)
                [[ -z "$CHOICE" ]] && continue
                full_line="${lines[$CHOICE]}"
                dialog --yesno "Delete this entry?\n\n${full_line}" 8 70
                confirm=$?
                clear
                if [[ $confirm -eq 0 ]]; then
                    delete_index "$CHOICE"
                    # Loop to refresh menu after deletion
                fi
                ;;
            1|255) # Cancel / ESC
                exit 0
                ;;
        esac
    done
}

# -------- Main behavior --------
if [[ $# -eq 0 ]] || [[ $# -gt 1 && "$1" == "--menu" ]]; then
    show_menu
    exit
fi

# Called with args: treat ALL args as ssh spec; add if missing, then connect
sshSpec="$*"
# Exists already?
if awk -v s="$sshSpec" -F',' '
  { gsub(/^[ \t]+|[ \t]+$/,"",$1); if ($1==s){found=1; exit} }
  END { exit found?0:1 }
' "$logFile"; then
    IFS=' ' read -r -a SSHARGS <<< "$sshSpec"
    exec ssh "${SSHARGS[@]}"
else
    read -r -p "Enter a note for [${sshSpec}] (optional): " note
    echo "${sshSpec}, ${note}" >> "$logFile"
    IFS=' ' read -r -a SSHARGS <<< "$sshSpec"
    exec ssh "${SSHARGS[@]}"
fi