@cyperx/clawforge 1.2.0

package/tui/model.go

@@ -0,0 +1,166 @@
+package main
+
+import (
+	"time"
+
+	tea "charm.land/bubbletea/v2"
+)
+
+// RefreshTickMsg triggers periodic data reload.
+type RefreshTickMsg time.Time
+
+const refreshInterval = 2 * time.Second
+
+// Model is the top-level Bubble Tea model for the ClawForge TUI dashboard.
+type Model struct {
+	agents      []Agent
+	selected    int
+	filter      string
+	filterMode  bool
+	viewMode    string // all | running | finished
+	showHelp    bool
+	animating   bool
+	width       int
+	height      int
+	steerMode   bool
+	showPreview bool
+	steerInput  string
+	confirmStop bool
+	frame       int
+	noAnim      bool
+}
+
+// filteredAgents returns the agents matching the current filter.
+func (m Model) filteredAgents() []Agent {
+	agents := m.agents
+	switch m.viewMode {
+	case "running":
+		var tmp []Agent
+		for _, a := range agents {
+			if a.Status == "running" || a.Status == "spawned" {
+				tmp = append(tmp, a)
+			}
+		}
+		agents = tmp
+	case "finished":
+		var tmp []Agent
+		for _, a := range agents {
+			if a.Status == "done" || a.Status == "archived" || a.Status == "failed" || a.Status == "cancelled" || a.Status == "timeout" {
+				tmp = append(tmp, a)
+			}
+		}
+		agents = tmp
+	}
+	return FilterAgents(agents, m.filter)
+}
+
+// NewModel creates a new Model. If noAnim is true, the startup animation is skipped.
+func NewModel(noAnim bool) Model {
+	return Model{
+		animating: !noAnim,
+		noAnim:    noAnim,
+		viewMode:  "all",
+	}
+}
+
+// Init starts the animation tick (or loads agents directly) and kicks off refresh.
+func (m Model) Init() tea.Cmd {
+	if m.animating {
+		return animationTick()
+	}
+	// No animation: load agents immediately and start refresh cycle.
+	m.agents = LoadAgents()
+	return refreshTick()
+}
+
+// refreshTick returns a command that sends a RefreshTickMsg after refreshInterval.
+func refreshTick() tea.Cmd {
+	return tea.Tick(refreshInterval, func(t time.Time) tea.Msg {
+		return RefreshTickMsg(t)
+	})
+}
+
+// Update handles all incoming messages and dispatches to the appropriate handler.
+func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case tea.WindowSizeMsg:
+		m.width = msg.Width
+		m.height = msg.Height
+		return m, nil
+
+	case AnimationTickMsg:
+		if m.animating {
+			m.frame++
+			if m.frame >= len(forgeFrames) {
+				m.animating = false
+				m.agents = LoadAgents()
+				return m, refreshTick()
+			}
+			return m, animationTick()
+		}
+		return m, nil
+
+	case AnimationDoneMsg:
+		// Animation was skipped via keypress — load agents and start refresh.
+		m.agents = LoadAgents()
+		return m, refreshTick()
+
+	case RefreshTickMsg:
+		if !m.animating {
+			m.agents = LoadAgents()
+			// Clamp selection after refresh.
+			filtered := m.filteredAgents()
+			if m.selected >= len(filtered) && len(filtered) > 0 {
+				m.selected = len(filtered) - 1
+			}
+		}
+		return m, refreshTick()
+
+	case steerDoneMsg:
+		// Steer command finished — refresh data.
+		m.agents = LoadAgents()
+		return m, nil
+
+	case stopDoneMsg:
+		// Stop command finished — refresh data.
+		m.agents = LoadAgents()
+		return m, nil
+
+	case attachDoneMsg:
+		// Returned from tmux attach — refresh data.
+		m.agents = LoadAgents()
+		return m, nil
+
+	case nudgeDoneMsg:
+		// Nudge command finished — refresh data.
+		m.agents = LoadAgents()
+		return m, nil
+
+	case tea.KeyPressMsg:
+		if m.animating {
+			// Skip animation on any key press.
+			m.animating = false
+			m.frame = 0
+			return m, func() tea.Msg { return AnimationDoneMsg{} }
+		}
+		var cmd tea.Cmd
+		m, cmd = handleKeyPress(m, msg)
+		return m, cmd
+	}
+
+	return m, nil
+}
+
+// View renders either the animation or the dashboard based on current state.
+func (m Model) View() tea.View {
+	var content string
+	if m.animating {
+		content = renderAnimation(m.frame, m.width, m.height)
+	} else {
+		content = renderDashboard(m)
+	}
+
+	v := tea.NewView(content)
+	v.AltScreen = true
+	return v
+}

package/tui/steer.go

@@ -0,0 +1,69 @@
+package main
+
+import (
+	"fmt"
+	"os/exec"
+
+	tea "charm.land/bubbletea/v2"
+	"charm.land/lipgloss/v2"
+)
+
+// steerDoneMsg is sent when the steer command finishes.
+type steerDoneMsg struct{ err error }
+
+// handleSteerKey processes key input while in steer mode.
+func handleSteerKey(m Model, key string) (Model, tea.Cmd) {
+	switch key {
+	case "esc":
+		m.steerMode = false
+		m.steerInput = ""
+		return m, nil
+	case "enter":
+		if m.steerInput != "" {
+			agents := m.filteredAgents()
+			if len(agents) > 0 {
+				agent := agents[m.selected]
+				msg := m.steerInput
+				m.steerMode = false
+				m.steerInput = ""
+				cmd := exec.Command("clawforge", "steer", agent.ID, msg)
+				return m, tea.ExecProcess(cmd, func(err error) tea.Msg {
+					return steerDoneMsg{err}
+				})
+			}
+		}
+		m.steerMode = false
+		m.steerInput = ""
+		return m, nil
+	case "backspace":
+		if len(m.steerInput) > 0 {
+			m.steerInput = m.steerInput[:len(m.steerInput)-1]
+		}
+		return m, nil
+	default:
+		// Only append printable single characters.
+		if len(key) == 1 {
+			m.steerInput += key
+		} else if key == "space" {
+			m.steerInput += " "
+		}
+		return m, nil
+	}
+}
+
+// RenderSteerInput renders the steer prompt line at the bottom of the screen.
+func RenderSteerInput(m Model) string {
+	agents := m.filteredAgents()
+	id := "?"
+	if len(agents) > 0 && m.selected < len(agents) {
+		id = agents[m.selected].ID
+	}
+
+	prompt := fmt.Sprintf("⚒️  Steer agent #%s: %s", id, m.steerInput)
+	cursor := "█"
+
+	return lipgloss.NewStyle().
+		Foreground(lipgloss.Color("#FF8C00")).
+		Bold(true).
+		Render(prompt + cursor)
+}

package/tui/styles.go

@@ -0,0 +1,69 @@
+package main
+
+import (
+	"charm.land/lipgloss/v2"
+)
+
+// Forge theme colors.
+var (
+	colorPrimary   = lipgloss.Color("#FF8C00") // dark orange / amber
+	colorSecondary = lipgloss.Color("#FFA500") // orange
+	colorAccent    = lipgloss.Color("#FF6600") // red-orange
+	colorMuted     = lipgloss.Color("#CC5500") // brown-orange
+)
+
+// Header style: bold + amber.
+var headerStyle = lipgloss.NewStyle().
+	Bold(true).
+	Foreground(colorPrimary)
+
+// Table header style: bold + secondary.
+var tableHeaderStyle = lipgloss.NewStyle().
+	Bold(true).
+	Foreground(colorSecondary)
+
+// Separator style: muted.
+var separatorStyle = lipgloss.NewStyle().
+	Foreground(colorMuted)
+
+// Selected row: reverse + amber foreground.
+var selectedRowStyle = lipgloss.NewStyle().
+	Reverse(true).
+	Foreground(colorPrimary)
+
+// Status bar: dim + muted.
+var statusBarStyle = lipgloss.NewStyle().
+	Faint(true).
+	Foreground(colorMuted)
+
+// Borders: rounded, amber.
+var borderStyle = lipgloss.NewStyle().
+	Border(lipgloss.RoundedBorder()).
+	BorderForeground(colorPrimary)
+
+// Mode badge style.
+var modeBadgeStyle = lipgloss.NewStyle().
+	Bold(true).
+	Padding(0, 1).
+	Foreground(lipgloss.Color("#000000")).
+	Background(colorSecondary)
+
+// Status indicator styles.
+var (
+	statusRunningStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("#00FF00"))
+	statusIdleStyle    = lipgloss.NewStyle().Foreground(lipgloss.Color("#FFFF00"))
+	statusFailedStyle  = lipgloss.NewStyle().Foreground(lipgloss.Color("#FF0000"))
+	statusDoneStyle    = lipgloss.NewStyle().Foreground(lipgloss.Color("#808080"))
+)
+
+// Help overlay style.
+var helpOverlayStyle = lipgloss.NewStyle().
+	Border(lipgloss.RoundedBorder()).
+	BorderForeground(colorPrimary).
+	Padding(1, 2).
+	Foreground(colorSecondary)
+
+// Animation frame style: centered, amber.
+var animFrameStyle = lipgloss.NewStyle().
+	Foreground(colorAccent).
+	Bold(true)