feat: health check

Author: whoisYoges

README.md

@@ -66,6 +66,7 @@ make run
 - e: edit selected host
 - y: duplicate selected host
 - d: delete selected host (with confirmation)
+- h: run health check for all saved hosts (green/yellow/red indicators)
 - enter: connect to selected host
 - x: export backup
 - i: import backup

internal/tui/app.go

@@ -48,6 +48,9 @@ type hostVerifyDoneMsg struct {
 type hostTrustDoneMsg struct{ err error }
 type saveKeyPassDoneMsg struct{ err error }
 type dashboardTrustDoneMsg struct{ err error }
+type healthCheckDoneMsg struct {
+	statuses map[string]hostHealthStatus
+}
 
 type formUserConfig struct {
 	Username            string
@@ -75,20 +78,22 @@ type model struct {
 	err      string
 
 	// Dashboard
-	store         *host.Store
-	filtered      []host.Host
-	hostCursor    int
-	search        textinput.Model
-	searchFocused bool
-	confirmDelete bool
-	connErr       string
-	exportErr     string
-	exportDir     string
-	importErr     string
-	importPath    string
-	importReturn  phase
-	userCursor    int
-	selectedHost  host.Host
+	store          *host.Store
+	filtered       []host.Host
+	hostCursor     int
+	search         textinput.Model
+	searchFocused  bool
+	confirmDelete  bool
+	connErr        string
+	exportErr      string
+	exportDir      string
+	importErr      string
+	importPath     string
+	importReturn   phase
+	userCursor     int
+	selectedHost   host.Host
+	healthChecking bool
+	healthStatuses map[string]hostHealthStatus
 
 	// Host form
 	formInputs             []textinput.Model
@@ -192,6 +197,8 @@ func (m model) initDashboard() (model, error) {
 	m.hostCursor = 0
 	m.confirmDelete = false
 	m.connErr = ""
+	m.healthChecking = false
+	m.healthStatuses = make(map[string]hostHealthStatus)
 	m.filtered = store.Filter("")
 	m.phase = phaseDashboard
 	return m, nil
@@ -290,6 +297,10 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			return m, nil
 		}
 		return m.connectSSHWithResolved(m.connectHost, m.connectUser, m.connectResolved, m.pendingKeyPassphrase, m.pendingKeyPassSave)
+	case healthCheckDoneMsg:
+		m.healthChecking = false
+		m.healthStatuses = msg.statuses
+		return m, nil
 	}
 
 	switch m.phase {

internal/tui/dashboard.go

@@ -77,6 +77,12 @@ func (m model) updateDashboardNormal(msg tea.Msg) (tea.Model, tea.Cmd) {
 		return m, textinput.Blink
 	case "i":
 		return m.startImportFlow()
+	case "h":
+		if m.healthChecking {
+			return m, nil
+		}
+		m.healthChecking = true
+		return m, healthCheckAllCmd(m.store.Hosts, m.encKey)
 	case "j", "down":
 		if m.hostCursor < len(m.filtered)-1 {
 			m.hostCursor++
@@ -356,6 +362,9 @@ func (m model) viewDashboard() string {
 	}
 
 	view := title + "\n" + searchLine + "\n\n" + panels
+	if m.healthChecking {
+		view += "\n" + hintStyle.Render(" Running health check across all saved hosts...")
+	}
 
 	if m.connErr != "" {
 		errBanner := errorStyle.Render(" ✗ " + m.connErr)
@@ -377,7 +386,7 @@ func (m model) renderHostList(maxW, maxH int) string {
 	}
 
 	// Column widths — adapt to available space.
-	available := maxW - 3
+	available := maxW - 6
 	if available < 10 {
 		available = 10
 	}
@@ -402,6 +411,7 @@ func (m model) renderHostList(maxW, maxH int) string {
 
 	for i := offset; i < end; i++ {
 		h := m.filtered[i]
+		indicator := m.healthIndicator(m.hostHealth(h.ID))
 
 		cursor := "  "
 		style := lipgloss.NewStyle().Foreground(text)
@@ -420,7 +430,7 @@ func (m model) renderHostList(maxW, maxH int) string {
 		}
 		hostColStr := lipgloss.NewStyle().Width(colHost).MaxWidth(colHost).Render(hostStr)
 
-		line := fmt.Sprintf("%s%s %s", cursor, aliasStr, hostColStr)
+		line := fmt.Sprintf("%s%s %s %s", cursor, indicator, aliasStr, hostColStr)
 		b.WriteString(style.Render(line))
 		if i < end-1 {
 			b.WriteByte('\n')
@@ -469,6 +479,7 @@ func (m model) renderDetails() string {
 	}
 
 	lines := []string{
+		render("Health", healthLabel(m.hostHealth(h.ID))),
 		render("Alias", h.Alias),
 		render("Host", h.Hostname),
 		render("Users", strings.Join(users, ", ")),
@@ -502,9 +513,9 @@ func (m model) renderCommands(maxW int) string {
 	return "  " + pad(cmd("/", "Search"), col) + cmd("l", "Lock Session") + "\n" +
 		"  " + pad(cmd("a", "Add"), col) + cmd("c", "Change Master Key") + "\n" +
 		"  " + pad(cmd("e", "Edit"), col) + cmd("y", "Duplicate") + "\n" +
-		"  " + pad(cmd("d", "Delete"), col) + cmd("⏎", "Connect") + "\n" +
+		"  " + pad(cmd("d", "Delete"), col) + cmd("h", "Health Check") + "\n" +
 		"  " + pad(cmd("x", "Export Backup"), col) + cmd("i", "Import Backup") + "\n" +
-		"  " + cmd("q", "Quit")
+		"  " + pad(cmd("⏎", "Connect"), col) + cmd("q", "Quit")
 }
 
 func (m model) updateUserSelect(msg tea.Msg) (tea.Model, tea.Cmd) {

internal/tui/healthcheck.go

@@ -0,0 +1,157 @@
+package tui
+
+import (
+	"context"
+	"errors"
+	"os/exec"
+	"strings"
+	"time"
+
+	tea "github.com/charmbracelet/bubbletea"
+
+	"github.com/managedssh/managedssh/internal/host"
+	"github.com/managedssh/managedssh/internal/sshclient"
+	"github.com/managedssh/managedssh/internal/vault"
+)
+
+type hostHealthStatus int
+
+const (
+	healthUnknown hostHealthStatus = iota
+	healthRed
+	healthYellow
+	healthGreen
+)
+
+func healthCheckAllCmd(hosts []host.Host, encKey []byte) tea.Cmd {
+	copiedHosts := make([]host.Host, len(hosts))
+	copy(copiedHosts, hosts)
+	return func() tea.Msg {
+		statuses := make(map[string]hostHealthStatus, len(copiedHosts))
+		for _, h := range copiedHosts {
+			statuses[h.ID] = checkHostHealth(h, encKey)
+		}
+		return healthCheckDoneMsg{statuses: statuses}
+	}
+}
+
+func checkHostHealth(h host.Host, encKey []byte) hostHealthStatus {
+	users := h.AccountNames()
+	if len(users) == 0 {
+		return healthRed
+	}
+
+	for _, username := range users {
+		_, resolved, ok := h.ResolveAccount(username)
+		if !ok {
+			continue
+		}
+
+		var password []byte
+		if resolved.AuthType == "password" && len(resolved.Password) > 0 {
+			dec, err := vault.Decrypt(encKey, resolved.Password)
+			if err != nil {
+				continue
+			}
+			password = dec
+		}
+
+		var keyData []byte
+		if resolved.AuthType == "key" && len(resolved.EncKey) > 0 {
+			dec, err := vault.Decrypt(encKey, resolved.EncKey)
+			if err != nil {
+				zeroBytes(password)
+				continue
+			}
+			keyData = dec
+		}
+
+		var keyPassphrase []byte
+		if resolved.AuthType == "key" && len(resolved.EncKeyPass) > 0 {
+			dec, err := vault.Decrypt(encKey, resolved.EncKeyPass)
+			if err != nil {
+				zeroBytes(password)
+				zeroBytes(keyData)
+				continue
+			}
+			keyPassphrase = dec
+		}
+
+		err := sshclient.Verify(sshclient.VerifyConfig{
+			Host:          h.Hostname,
+			Port:          h.Port,
+			User:          username,
+			Password:      password,
+			KeyPath:       resolved.KeyPath,
+			KeyData:       keyData,
+			KeyPassphrase: keyPassphrase,
+		})
+		zeroBytes(password)
+		zeroBytes(keyData)
+		zeroBytes(keyPassphrase)
+		if err == nil {
+			return healthGreen
+		}
+
+		var unknown *sshclient.UnknownHostError
+		if errors.As(err, &unknown) {
+			return healthGreen
+		}
+	}
+
+	if hostRespondsToPing(h.Hostname) {
+		return healthYellow
+	}
+	return healthRed
+}
+
+func hostRespondsToPing(hostname string) bool {
+	hostname = strings.TrimSpace(hostname)
+	if hostname == "" {
+		return false
+	}
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	defer cancel()
+
+	cmd := exec.CommandContext(ctx, "ping", "-c", "1", hostname)
+	if err := cmd.Run(); err == nil {
+		return true
+	}
+	return false
+}
+
+func (m model) healthIndicator(status hostHealthStatus) string {
+	switch status {
+	case healthGreen:
+		return lipHealthGreen.Render("●")
+	case healthYellow:
+		return lipHealthYellow.Render("●")
+	case healthRed:
+		return lipHealthRed.Render("●")
+	default:
+		if m.healthChecking {
+			return lipHealthPending.Render("○")
+		}
+		return lipHealthUnknown.Render("○")
+	}
+}
+
+func healthLabel(status hostHealthStatus) string {
+	switch status {
+	case healthGreen:
+		return "Green (reachable)"
+	case healthYellow:
+		return "Yellow (ping OK, SSH uncertain)"
+	case healthRed:
+		return "Red (unreachable/invalid)"
+	default:
+		return "Unknown"
+	}
+}
+
+func (m model) hostHealth(hostID string) hostHealthStatus {
+	if status, ok := m.healthStatuses[hostID]; ok {
+		return status
+	}
+	return healthUnknown
+}

internal/tui/styles.go

@@ -78,4 +78,10 @@ var (
 
 	cmdDescStyle = lipgloss.NewStyle().
 			Foreground(text)
+
+	lipHealthGreen   = lipgloss.NewStyle().Foreground(lipgloss.Color("#22C55E"))
+	lipHealthYellow  = lipgloss.NewStyle().Foreground(lipgloss.Color("#F59E0B"))
+	lipHealthRed     = lipgloss.NewStyle().Foreground(lipgloss.Color("#EF4444"))
+	lipHealthPending = lipgloss.NewStyle().Foreground(lipgloss.Color("#60A5FA"))
+	lipHealthUnknown = lipgloss.NewStyle().Foreground(subtle)
 )