149 lines
4.2 KiB
Go
149 lines
4.2 KiB
Go
// Package ui provides the bubbletea terminal UI for wallarm:
|
|
// - Deploy choice: local deploy or remote assistance
|
|
// - Dashboard: lists existing nodes with actions
|
|
//
|
|
// Deployment logic runs in plain terminal (main.go), not inside the TUI.
|
|
package ui
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
|
|
"git.sechpoint.app/customer-engineering/wallarm/internal/state"
|
|
)
|
|
|
|
var (
|
|
titleStyle = lipgloss.NewStyle().Bold(true).Foreground(lipgloss.Color("63")).MarginBottom(1)
|
|
goodStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("42"))
|
|
warnStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("227"))
|
|
badStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("196"))
|
|
dimStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("240"))
|
|
activeStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("63")).Bold(true)
|
|
)
|
|
|
|
type Model struct {
|
|
state stateView
|
|
width int
|
|
height int
|
|
}
|
|
|
|
type stateView int
|
|
|
|
const (
|
|
viewRoute stateView = iota
|
|
viewDeployChoice // local deploy or remote assist?
|
|
viewDashboard
|
|
)
|
|
|
|
// Run starts the bubbletea TUI. Returns nil if user chose local deploy (exit and run form).
|
|
func Run() error {
|
|
m := Model{state: viewRoute}
|
|
p := tea.NewProgram(m)
|
|
if _, err := p.Run(); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m Model) Init() tea.Cmd { return nil }
|
|
|
|
func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case tea.KeyMsg:
|
|
switch msg.String() {
|
|
case "q", "ctrl+c":
|
|
return m, tea.Quit
|
|
case "1":
|
|
if m.state == viewDeployChoice {
|
|
return m, tea.Quit // exit TUI, main.go runs deploy form
|
|
}
|
|
case "2":
|
|
if m.state == viewDeployChoice {
|
|
return m, tea.Quit // exit TUI, main.go runs tunnel flow
|
|
}
|
|
}
|
|
case tea.WindowSizeMsg:
|
|
m.width = msg.Width
|
|
m.height = msg.Height
|
|
}
|
|
|
|
switch m.state {
|
|
case viewRoute:
|
|
if state.HasDeployment() {
|
|
m.state = viewDashboard
|
|
} else {
|
|
m.state = viewDeployChoice
|
|
}
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func (m Model) View() string {
|
|
switch m.state {
|
|
case viewRoute:
|
|
return ""
|
|
case viewDeployChoice:
|
|
return deployChoiceView(m)
|
|
case viewDashboard:
|
|
return dashboardView(m)
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// ─── Deploy Choice ───────────────────────────────────────────────────
|
|
|
|
func deployChoiceView(m Model) string {
|
|
s := titleStyle.Render("🛡️ Wallarm Setup") + "\n\n"
|
|
s += "How would you like to proceed?\n\n"
|
|
s += activeStyle.Render(" [1] Deploy locally") + "\n"
|
|
s += " Configure and deploy the Wallarm node on this server now.\n\n"
|
|
s += activeStyle.Render(" [2] Remote assistance") + "\n"
|
|
s += " Open a secure tunnel so a remote admin can deploy for you.\n\n"
|
|
s += dimStyle.Render("Press 1 or 2, q to quit")
|
|
return s
|
|
}
|
|
|
|
// ─── Dashboard ───────────────────────────────────────────────────────
|
|
|
|
func dashboardView(m Model) string {
|
|
s, err := state.Load()
|
|
if err != nil || s == nil {
|
|
return badStyle.Render("Error loading state.") + "\nPress q to quit"
|
|
}
|
|
|
|
out := titleStyle.Render("📊 Wallarm Dashboard") + "\n"
|
|
out += fmt.Sprintf("Type: %s | Cloud: %s (%s)\n", s.DeploymentType, s.CloudRegion, s.APIHost)
|
|
out += dimStyle.Render(strings.Repeat("─", 50)) + "\n\n"
|
|
|
|
if len(s.Nodes) == 0 {
|
|
out += dimStyle.Render("No nodes deployed yet.") + "\n\n"
|
|
} else {
|
|
out += activeStyle.Render("Nodes:") + "\n"
|
|
for _, n := range s.Nodes {
|
|
marker := "●"
|
|
style := goodStyle
|
|
if n.Status != "running" {
|
|
marker = "○"
|
|
style = warnStyle
|
|
}
|
|
out += style.Render(fmt.Sprintf(" %s %s — %s", marker, n.Name, n.Status))
|
|
if n.Address != "" {
|
|
out += dimStyle.Render(fmt.Sprintf(" (%s)", n.Address))
|
|
}
|
|
out += "\n"
|
|
}
|
|
out += "\n"
|
|
}
|
|
|
|
out += "Actions:\n"
|
|
out += fmt.Sprintf(" %s\n", activeStyle.Render("[a] Add node"))
|
|
out += fmt.Sprintf(" %s\n", activeStyle.Render("[c] Configure"))
|
|
out += fmt.Sprintf(" %s\n", activeStyle.Render("[r] Remove node"))
|
|
out += fmt.Sprintf(" %s\n", activeStyle.Render("[t] Start tunnel"))
|
|
out += fmt.Sprintf(" %s\n", dimStyle.Render("[q] Quit"))
|
|
|
|
return out
|
|
}
|