allow dropping into an ssh shell straight from the dashboard

Author: AlpinDale

README.md

@@ -68,19 +68,24 @@ The tool will:
 - While in the dashboard, press `c` to return to host selection to add/remove hosts
 - Press `n` to cycle through connected hosts (like tmux sessions)
 - Press `t` to toggle overview mode, showing all selected hosts at once with GPU pressure summaries
+- Press `s` to exit the dashboard and drop into an interactive SSH shell with the current host
 - All connections remain active - no need to reconnect!
 
 ### Configuration
 
 **Update Interval:**
 
-Control how often the dashboard refreshes (default: 5 seconds):
+Control how often the dashboard refreshes in seconds (default: 5). Supports decimal values for sub-second updates:
 
 ```bash
-ssh-dashboard -interval 1
+# Update every second
+ssh-dashboard -n 1
+
+# Update 10 times per second (100ms)
+ssh-dashboard -n 0.1
 
 # or with an env var
-export SSH_DASHBOARD_INTERVAL=1
+export SSH_DASHBOARD_INTERVAL=0.5
 ssh-dashboard
 ```
 
@@ -90,6 +95,7 @@ ssh-dashboard
 - `Enter` - Connect to selected host(s)
 - `n` - Switch to next host (when multiple hosts selected)
 - `t` - Toggle overview screen (shows all hosts at once)
+- `s` - Exit and SSH into current host
 - `c` - Add hosts (from dashboard, returns to host selection)
 
 ## SSH Configuration

cmd/ssh_dashboard/main.go

@@ -4,7 +4,9 @@ import (
 	"flag"
 	"fmt"
 	"os"
+	"os/exec"
 	"strconv"
+	"syscall"
 	"time"
 
 	"github.com/alpindale/ssh-dashboard/internal"
@@ -58,9 +60,30 @@ func main() {
 		os.Exit(1)
 	}
 
-	p := tea.NewProgram(internal.InitialModel(hosts, interval), tea.WithAltScreen())
-	if _, err := p.Run(); err != nil {
+	initialModel := internal.InitialModel(hosts, interval)
+	p := tea.NewProgram(initialModel, tea.WithAltScreen())
+	finalModel, err := p.Run()
+	if err != nil {
 		fmt.Fprintf(os.Stderr, "Error running program: %v\n", err)
 		os.Exit(1)
 	}
+
+	if m, ok := finalModel.(internal.Model); ok {
+		if sshHost := m.GetSSHOnExit(); sshHost != "" {
+			sshPath, err := exec.LookPath("ssh")
+			if err != nil {
+				fmt.Fprintf(os.Stderr, "Error finding ssh: %v\n", err)
+				os.Exit(1)
+			}
+
+			args := []string{"ssh", sshHost}
+			env := os.Environ()
+
+			err = syscall.Exec(sshPath, args, env)
+			if err != nil {
+				fmt.Fprintf(os.Stderr, "Error executing ssh: %v\n", err)
+				os.Exit(1)
+			}
+		}
+	}
 }

internal/ui.go

@@ -36,6 +36,7 @@ type Model struct {
 	err            error
 	width          int
 	height         int
+	sshOnExit      string
 }
 
 type TickMsg time.Time
@@ -141,6 +142,10 @@ func InitialModel(hosts []SSHHost, updateInterval time.Duration) Model {
 	}
 }
 
+func (m Model) GetSSHOnExit() string {
+	return m.sshOnExit
+}
+
 func (m *Model) updateListSelection() {
 	items := m.list.Items()
 	selectedMap := make(map[string]bool)
@@ -232,6 +237,19 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			} else if m.screen == ScreenOverview {
 				m.screen = ScreenDashboard
 			}
+		case "s":
+			if m.screen == ScreenDashboard {
+				if len(m.selectedHosts) > 0 {
+					currentHost := m.selectedHosts[m.currentHostIdx]
+					m.sshOnExit = currentHost.Name
+					for _, client := range m.clients {
+						if client != nil {
+							client.Close()
+						}
+					}
+					return m, tea.Quit
+				}
+			}
 		}
 
 	case tea.WindowSizeMsg:
@@ -625,7 +643,7 @@ func renderDashboard(hostName string, info *SystemInfo, updateInterval time.Dura
 	if multiHost {
 		navHint = " | 'n' next | 't' overview"
 	}
-	subtitle := fmt.Sprintf("Last Updated: %s | Interval: %s%s | 'c' add hosts | 'q' quit",
+	subtitle := fmt.Sprintf("Last Updated: %s | Interval: %s%s | 's' shell | 'c' add hosts | 'q' quit",
 		lastUpdate.Format("15:04:05"), formatInterval(updateInterval), navHint)
 
 	b.WriteString(titleStyle.Render(title))