ci: add async update notifier

Author: AlpinDale

internal/ui/model.go

@@ -36,10 +36,13 @@ type Model struct {
 	width          int
 	height         int
 	sshOnExit      string
+	updateInfo     internal.UpdateInfo
 }
 
 type TickMsg time.Time
 
+type UpdateCheckMsg internal.UpdateInfo
+
 type SystemInfoMsg struct {
 	hostName string
 	info     *internal.SystemInfo
@@ -72,6 +75,11 @@ func (h hostItem) Description() string {
 	return ""
 }
 
+func checkForUpdates() tea.Msg {
+	updateInfo := internal.CheckForUpdates()
+	return UpdateCheckMsg(updateInfo)
+}
+
 func censorHostname(hostname string) string {
 	if hostname == "" {
 		return ""
@@ -201,7 +209,7 @@ func (m *Model) updateListSelection() {
 
 func (m Model) Init() tea.Cmd {
 	if m.screen == ScreenConnecting && len(m.selectedHosts) > 0 {
-		return tea.Batch(m.spinner.Tick, m.connectToHosts())
+		return tea.Batch(m.spinner.Tick, m.connectToHosts(), checkForUpdates)
 	}
-	return m.spinner.Tick
+	return tea.Batch(m.spinner.Tick, checkForUpdates)
 }

internal/ui/update.go

@@ -3,6 +3,7 @@ package ui
 import (
 	"time"
 
+	"github.com/alpindale/ssh-dashboard/internal"
 	tea "github.com/charmbracelet/bubbletea"
 )
 
@@ -144,6 +145,9 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			}
 		}
 
+	case UpdateCheckMsg:
+		m.updateInfo = internal.UpdateInfo(msg)
+
 	case TickMsg:
 		// update every 10 seconds
 		return m, tea.Batch(m.gatherAllSysInfo(), m.tick())

internal/ui/view.go

@@ -8,6 +8,24 @@ import (
 	"github.com/charmbracelet/lipgloss"
 )
 
+func (m Model) renderUpdateNotification() string {
+	if !m.updateInfo.Available {
+		return ""
+	}
+
+	updateStyle := lipgloss.NewStyle().
+		Foreground(lipgloss.Color("226")).
+		Bold(true)
+
+	currentVer := m.updateInfo.CurrentVersion
+	if !strings.HasPrefix(currentVer, "v") {
+		currentVer = "v" + currentVer
+	}
+
+	return updateStyle.Render(fmt.Sprintf("\n\n⬆  Update available! %s → %s",
+		currentVer, m.updateInfo.LatestVersion))
+}
+
 func (m Model) View() string {
 	switch m.screen {
 	case ScreenHostList:
@@ -30,6 +48,7 @@ func (m Model) View() string {
 		}
 		versionFooter := fmt.Sprintf("\nv%s", internal.ShortVersion())
 		listView += lipgloss.NewStyle().Foreground(lipgloss.Color("240")).Render(versionFooter)
+		listView += m.renderUpdateNotification()
 		return listView
 
 	case ScreenConnecting:
@@ -50,12 +69,14 @@ func (m Model) View() string {
 			if len(m.selectedHosts) > 1 {
 				hostIndicator = fmt.Sprintf(" [%d/%d]", m.currentHostIdx+1, len(m.selectedHosts))
 			}
-			return renderDashboard(currentHost.Name+hostIndicator, sysInfo, m.updateInterval, lastUpdate, m.width, m.height, len(m.selectedHosts) > 1)
+			dashboardView := renderDashboard(currentHost.Name+hostIndicator, sysInfo, m.updateInterval, lastUpdate, m.width, m.height, len(m.selectedHosts) > 1)
+			return dashboardView + m.renderUpdateNotification()
 		}
 		return m.renderConnectingProgress()
 
 	case ScreenOverview:
-		return m.renderOverview()
+		overviewView := m.renderOverview()
+		return overviewView + m.renderUpdateNotification()
 	}
 
 	return ""

internal/updater.go

@@ -0,0 +1,100 @@
+package internal
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strings"
+	"time"
+)
+
+const (
+	githubAPIURL       = "https://api.github.com/repos/alpindale/ssh-dashboard/releases/latest"
+	updateCheckTimeout = 3 * time.Second
+)
+
+type UpdateInfo struct {
+	Available      bool
+	LatestVersion  string
+	CurrentVersion string
+}
+
+type githubRelease struct {
+	TagName string `json:"tag_name"`
+	HTMLURL string `json:"html_url"`
+}
+
+func CheckForUpdates() UpdateInfo {
+	currentVersion := Version
+	if currentVersion == "dev" {
+		return UpdateInfo{Available: false, CurrentVersion: currentVersion}
+	}
+
+	latestVersion, err := fetchLatestVersion()
+	if err != nil {
+		return UpdateInfo{Available: false, CurrentVersion: currentVersion}
+	}
+
+	needsUpdate := compareVersions(currentVersion, latestVersion)
+
+	return UpdateInfo{
+		Available:      needsUpdate,
+		LatestVersion:  latestVersion,
+		CurrentVersion: currentVersion,
+	}
+}
+
+func fetchLatestVersion() (string, error) {
+	client := &http.Client{
+		Timeout: updateCheckTimeout,
+	}
+
+	resp, err := client.Get(githubAPIURL)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return "", fmt.Errorf("github api returned status %d", resp.StatusCode)
+	}
+
+	var release githubRelease
+	if err := json.NewDecoder(resp.Body).Decode(&release); err != nil {
+		return "", err
+	}
+
+	return release.TagName, nil
+}
+
+func compareVersions(current, latest string) bool {
+	current = strings.TrimPrefix(current, "v")
+	latest = strings.TrimPrefix(latest, "v")
+
+	currentBase := strings.Split(strings.Split(current, "-")[0], "+")[0]
+	latestBase := strings.Split(strings.Split(latest, "-")[0], "+")[0]
+
+	currentParts := strings.Split(currentBase, ".")
+	latestParts := strings.Split(latestBase, ".")
+
+	for len(currentParts) < 3 {
+		currentParts = append(currentParts, "0")
+	}
+	for len(latestParts) < 3 {
+		latestParts = append(latestParts, "0")
+	}
+
+	for i := 0; i < 3; i++ {
+		var currentNum, latestNum int
+		fmt.Sscanf(currentParts[i], "%d", &currentNum)
+		fmt.Sscanf(latestParts[i], "%d", &latestNum)
+
+		if latestNum > currentNum {
+			return true
+		} else if latestNum < currentNum {
+			return false
+		}
+	}
+
+	return false
+}