implement core logic (#3)

2026-07-14 12:13:34 +02:00 · 2025-08-27 16:56:42 +01:00
parent e9225188e8
commit 8e3f7ebc37
27 changed files with 1675 additions and 245 deletions
@@ -16,8 +16,11 @@ package ui

 import (
 	"fmt"
+	"strings"
+	"time"

 	"github.com/Adembc/lazyssh/internal/core/domain"
+	"github.com/atotto/clipboard"
 	"github.com/gdamore/tcell/v2"
 	"github.com/rivo/tview"
 )
@@ -34,7 +37,7 @@ func (t *tui) handleGlobalKeys(event *tcell.EventKey) *tcell.EventKey {

 	switch event.Rune() {
 	case 'q':
-		t.app.Stop()
+		t.handleQuit()
 		return nil
 	case '/':
 		t.handleSearchToggle()
@@ -48,8 +51,26 @@ func (t *tui) handleGlobalKeys(event *tcell.EventKey) *tcell.EventKey {
 	case 'd':
 		t.handleServerDelete()
 		return nil
-	case '?':
-		t.handleHelpShow()
+	case 'p':
+		t.handleServerPin()
+		return nil
+	case 's':
+		t.handleSortToggle()
+		return nil
+	case 'S':
+		t.handleSortReverse()
+		return nil
+	case 'c':
+		t.handleCopyCommand()
+		return nil
+	case 'g':
+		t.handlePingSelected()
+		return nil
+	case 'r':
+		t.handleRefreshBackground()
+		return nil
+	case 't':
+		t.handleTagsEdit()
 		return nil
 	}

@@ -61,8 +82,52 @@ func (t *tui) handleGlobalKeys(event *tcell.EventKey) *tcell.EventKey {
 	return event
 }

+func (t *tui) handleQuit() {
+	t.app.Stop()
+}
+
+func (t *tui) handleServerPin() {
+	if server, ok := t.serverList.GetSelectedServer(); ok {
+		pinned := server.PinnedAt.IsZero()
+		_ = t.serverService.SetPinned(server.Alias, pinned)
+		t.refreshServerList()
+	}
+}
+
+func (t *tui) handleSortToggle() {
+	t.sortMode = t.sortMode.ToggleField()
+	t.showStatusTemp("Sort: " + t.sortMode.String())
+	t.updateListTitle()
+	t.refreshServerList()
+}
+
+func (t *tui) handleSortReverse() {
+	t.sortMode = t.sortMode.Reverse()
+	t.showStatusTemp("Sort: " + t.sortMode.String())
+	t.updateListTitle()
+	t.refreshServerList()
+}
+
+func (t *tui) handleCopyCommand() {
+	if server, ok := t.serverList.GetSelectedServer(); ok {
+		cmd := BuildSSHCommand(server)
+		if err := clipboard.WriteAll(cmd); err == nil {
+			t.showStatusTemp("Copied: " + cmd)
+		} else {
+			t.showStatusTemp("Failed to copy to clipboard")
+		}
+	}
+}
+
+func (t *tui) handleTagsEdit() {
+	if server, ok := t.serverList.GetSelectedServer(); ok {
+		t.showEditTagsForm(server)
+	}
+}
+
 func (t *tui) handleSearchInput(query string) {
 	filtered, _ := t.serverService.ListServers(query)
+	sortServersForUI(filtered, t.sortMode)
 	t.serverList.UpdateServers(filtered)
 	if len(filtered) == 0 {
 		t.details.ShowEmpty()
@@ -100,12 +165,22 @@ func (t *tui) handleServerEdit() {
 }

 func (t *tui) handleServerSave(server domain.Server, original *domain.Server) {
+	var err error
 	if original != nil {
 		// Edit mode
-		_ = t.serverService.UpdateServer(*original, server)
+		err = t.serverService.UpdateServer(*original, server)
 	} else {
 		// Add mode
-		_ = t.serverService.AddServer(server)
+		err = t.serverService.AddServer(server)
+	}
+	if err != nil {
+		// Stay on form; show a small modal with the error
+		modal := tview.NewModal().
+			SetText(fmt.Sprintf("Save failed: %v", err)).
+			AddButtons([]string{"Close"}).
+			SetDoneFunc(func(buttonIndex int, buttonLabel string) { t.handleModalClose() })
+		t.app.SetRoot(modal, true)
+		return
 	}

 	t.refreshServerList()
@@ -114,8 +189,7 @@ func (t *tui) handleServerSave(server domain.Server, original *domain.Server) {

 func (t *tui) handleServerDelete() {
 	if server, ok := t.serverList.GetSelectedServer(); ok {
-		_ = t.serverService.DeleteServer(server)
-		t.refreshServerList()
+		t.showDeleteConfirmModal(server)
 	}
 }

@@ -123,14 +197,66 @@ func (t *tui) handleFormCancel() {
 	t.returnToMain()
 }

-func (t *tui) handleHelpShow() {
-	t.showHelpModal()
+func (t *tui) handlePingSelected() {
+	if server, ok := t.serverList.GetSelectedServer(); ok {
+		alias := server.Alias
+
+		t.showStatusTemp(fmt.Sprintf("Pinging %s…", alias))
+		go func() {
+			up, dur, err := t.serverService.Ping(server)
+			t.app.QueueUpdateDraw(func() {
+				if err != nil {
+					t.showStatusTempColor(fmt.Sprintf("Ping %s: DOWN (%v)", alias, err), "#FF6B6B")
+					return
+				}
+				if up {
+					t.showStatusTempColor(fmt.Sprintf("Ping %s: UP (%s)", alias, dur), "#A0FFA0")
+				} else {
+					t.showStatusTempColor(fmt.Sprintf("Ping %s: DOWN", alias), "#FF6B6B")
+				}
+			})
+		}()
+	}
 }

 func (t *tui) handleModalClose() {
 	t.returnToMain()
 }

+// handleRefreshBackground refreshes the server list in the background without leaving the current screen.
+// It preserves the current search query and selection, shows transient status, and avoids concurrent runs.
+func (t *tui) handleRefreshBackground() {
+	currentIdx := t.serverList.GetCurrentItem()
+	query := ""
+	if t.searchVisible {
+		query = t.searchBar.InputField.GetText()
+	}
+
+	t.showStatusTemp("Refreshing…")
+
+	go func(prevIdx int, q string) {
+		servers, err := t.serverService.ListServers(q)
+		if err != nil {
+			t.app.QueueUpdateDraw(func() {
+				t.showStatusTempColor(fmt.Sprintf("Refresh failed: %v", err), "#FF6B6B")
+			})
+			return
+		}
+		sortServersForUI(servers, t.sortMode)
+		t.app.QueueUpdateDraw(func() {
+			t.serverList.UpdateServers(servers)
+			// Try to restore selection if still valid
+			if prevIdx >= 0 && prevIdx < t.serverList.List.GetItemCount() {
+				t.serverList.SetCurrentItem(prevIdx)
+				if srv, ok := t.serverList.GetSelectedServer(); ok {
+					t.details.UpdateServer(srv)
+				}
+			}
+			t.showStatusTemp(fmt.Sprintf("Refreshed %d servers", len(servers)))
+		})
+	}(currentIdx, query)
+}
+
 // =============================================================================
 // UI Display Functions (show UI elements/modals)
 // =============================================================================
@@ -144,40 +270,89 @@ func (t *tui) showSearchBar() {
 }

 func (t *tui) showConnectModal(server domain.Server) {
-	msg := fmt.Sprintf("SSH to %s (%s@%s:%d)\n\nThis is a mock action.",
+	msg := fmt.Sprintf("SSH to %s (%s@%s:%d)\n\nConfirm to start an SSH session .",
 		server.Alias, server.User, server.Host, server.Port)

 	modal := tview.NewModal().
 		SetText(msg).
-		AddButtons([]string{"OK"}).
+		AddButtons([]string{"Confirm", "Cancel"}).
 		SetDoneFunc(func(buttonIndex int, buttonLabel string) {
+			if buttonIndex == 0 {
+				// Suspend the TUI while running the external ssh command.
+				t.app.Suspend(func() {
+					err := t.serverService.SSH(server.Alias)
+					if err != nil {
+						// Show a brief status after we resume
+						t.app.QueueUpdateDraw(func() {
+							if strings.Contains(strings.ToLower(err.Error()), "timeout") {
+								t.showStatusTempColor("SSH timeout, returning to list", "#FF6B6B")
+							} else {
+								t.showStatusTempColor("SSH failed: "+err.Error(), "#FF6B6B")
+							}
+						})
+					}
+				})
+				// Refresh to reflect updated last seen and ssh count
+				t.refreshServerList()
+			}
 			t.handleModalClose()
 		})

 	t.app.SetRoot(modal, true)
 }

-func (t *tui) showHelpModal() {
-	text := "Keyboard shortcuts:\n\n" +
-		"  ↑/↓          Navigate\n" +
-		"  Enter        SSH connect (mock)\n" +
-		"  a            Add server (mock)\n" +
-		"  e            Edit server (mock)\n" +
-		"  d            Delete entry (mock)\n" +
-		"  /            Focus search\n" +
-		"  q            Quit\n" +
-		"  ?            Help\n"
+func (t *tui) showDeleteConfirmModal(server domain.Server) {
+	msg := fmt.Sprintf("Delete server %s (%s@%s:%d)?\n\nThis action cannot be undone.",
+		server.Alias, server.User, server.Host, server.Port)

 	modal := tview.NewModal().
-		SetText(text).
-		AddButtons([]string{"Close"}).
+		SetText(msg).
+		AddButtons([]string{"Cancel", "Confirm"}).
 		SetDoneFunc(func(buttonIndex int, buttonLabel string) {
+			if buttonIndex == 1 {
+				_ = t.serverService.DeleteServer(server)
+				t.refreshServerList()
+			}
 			t.handleModalClose()
 		})

 	t.app.SetRoot(modal, true)
 }

+func (t *tui) showEditTagsForm(server domain.Server) {
+	form := tview.NewForm()
+	form.SetBorder(true).
+		SetTitle(fmt.Sprintf("Edit Tags: %s", server.Alias)).
+		SetTitleAlign(tview.AlignLeft)
+
+	defaultTags := strings.Join(server.Tags, ", ")
+	form.AddInputField("Tags (comma):", defaultTags, 40, nil, nil)
+
+	form.AddButton("Save", func() {
+		text := strings.TrimSpace(form.GetFormItem(0).(*tview.InputField).GetText())
+		var tags []string
+		if text != "" {
+			for _, part := range strings.Split(text, ",") {
+				if s := strings.TrimSpace(part); s != "" {
+					tags = append(tags, s)
+				}
+			}
+		}
+		newServer := server
+		newServer.Tags = tags
+		_ = t.serverService.UpdateServer(server, newServer)
+		// Refresh UI and go back
+		t.refreshServerList()
+		t.returnToMain()
+		t.showStatusTemp("Tags updated")
+	})
+	form.AddButton("Cancel", func() { t.returnToMain() })
+	form.SetCancelFunc(func() { t.returnToMain() })
+
+	t.app.SetRoot(form, true)
+	t.app.SetFocus(form)
+}
+
 // =============================================================================
 // UI State Management (hide UI elements)
 // =============================================================================
@@ -200,9 +375,35 @@ func (t *tui) refreshServerList() {
 		query = t.searchBar.InputField.GetText()
 	}
 	filtered, _ := t.serverService.ListServers(query)
+	sortServersForUI(filtered, t.sortMode)
 	t.serverList.UpdateServers(filtered)
 }

 func (t *tui) returnToMain() {
 	t.app.SetRoot(t.root, true)
 }
+
+// showStatusTemp displays a temporary message in the status bar (default green) and then restores the default text.
+func (t *tui) showStatusTemp(msg string) {
+	if t.statusBar == nil {
+		return
+	}
+	t.showStatusTempColor(msg, "#A0FFA0")
+}
+
+// showStatusTempColor displays a temporary colored message in the status bar and restores default text after 2s.
+func (t *tui) showStatusTempColor(msg string, color string) {
+	if t.statusBar == nil {
+		return
+	}
+	t.statusBar.SetText("[" + color + "]" + msg + "[-]")
+	time.AfterFunc(2*time.Second, func() {
+		if t.app != nil {
+			t.app.QueueUpdateDraw(func() {
+				if t.statusBar != nil {
+					t.statusBar.SetText(DefaultStatusText())
+				}
+			})
+		}
+	})
+}