git.iko.soy/crush
on 94de6bc help sign in
crush/internal/tui/components/chat/header/header.go
167 lines 4.1 kB history raw
1package header
3import (
4 "fmt"
5 "strings"
7 tea "charm.land/bubbletea/v2"
8 "charm.land/lipgloss/v2"
9 "github.com/charmbracelet/crush/internal/config"
10 "github.com/charmbracelet/crush/internal/csync"
11 "github.com/charmbracelet/crush/internal/fsext"
12 "github.com/charmbracelet/crush/internal/lsp"
13 "github.com/charmbracelet/crush/internal/pubsub"
14 "github.com/charmbracelet/crush/internal/session"
15 "github.com/charmbracelet/crush/internal/tui/styles"
16 "github.com/charmbracelet/crush/internal/tui/util"
17 "github.com/charmbracelet/x/ansi"
18 "github.com/charmbracelet/x/powernap/pkg/lsp/protocol"
21type Header interface {
22 util.Model
23 SetSession(session session.Session) tea.Cmd
24 SetWidth(width int) tea.Cmd
25 SetDetailsOpen(open bool)
26 ShowingDetails() bool
29type header struct {
30 width int
31 session session.Session
32 lspClients *csync.Map[string, *lsp.Client]
33 detailsOpen bool
36func New(lspClients *csync.Map[string, *lsp.Client]) Header {
37 return &header{
38 lspClients: lspClients,
39 width: 0,
40 }
43func (h *header) Init() tea.Cmd {
44 return nil
47func (h *header) Update(msg tea.Msg) (util.Model, tea.Cmd) {
48 switch msg := msg.(type) {
49 case pubsub.Event[session.Session]:
50 if msg.Type == pubsub.UpdatedEvent {
51 if h.session.ID == msg.Payload.ID {
52 h.session = msg.Payload
53 }
54 }
55 }
56 return h, nil
59func (h *header) View() string {
60 if h.session.ID == "" {
61 return ""
62 }
64 const (
65 gap = " "
66 diag = "╱"
67 minDiags = 3
68 leftPadding = 1
69 rightPadding = 1
70 )
72 t := styles.CurrentTheme()
74 var b strings.Builder
76 b.WriteString(t.S().Base.Foreground(t.Secondary).Render("Charm™"))
77 b.WriteString(gap)
78 b.WriteString(styles.ApplyBoldForegroundGrad("CRUSH", t.Secondary, t.Primary))
79 b.WriteString(gap)
81 availDetailWidth := h.width - leftPadding - rightPadding - lipgloss.Width(b.String()) - minDiags
82 details := h.details(availDetailWidth)
84 remainingWidth := h.width -
85 lipgloss.Width(b.String()) -
86 lipgloss.Width(details) -
87 leftPadding -
88 rightPadding
90 if remainingWidth > 0 {
91 b.WriteString(t.S().Base.Foreground(t.Primary).Render(
92 strings.Repeat(diag, max(minDiags, remainingWidth)),
93 ))
94 b.WriteString(gap)
95 }
97 b.WriteString(details)
99 return t.S().Base.Padding(0, rightPadding, 0, leftPadding).Render(b.String())
102func (h *header) details(availWidth int) string {
103 s := styles.CurrentTheme().S()
105 var parts []string
107 errorCount := 0
108 for l := range h.lspClients.Seq() {
109 for _, diagnostics := range l.GetDiagnostics() {
110 for _, diagnostic := range diagnostics {
111 if diagnostic.Severity == protocol.SeverityError {
112 errorCount++
113 }
114 }
115 }
116 }
118 if errorCount > 0 {
119 parts = append(parts, s.Error.Render(fmt.Sprintf("%s%d", styles.ErrorIcon, errorCount)))
120 }
122 agentCfg := config.Get().Agents[config.AgentCoder]
123 model := config.Get().GetModelByType(agentCfg.Model)
124 percentage := (float64(h.session.CompletionTokens+h.session.PromptTokens) / float64(model.ContextWindow)) * 100
125 formattedPercentage := s.Muted.Render(fmt.Sprintf("%d%%", int(percentage)))
126 parts = append(parts, formattedPercentage)
128 const keystroke = "ctrl+d"
129 if h.detailsOpen {
130 parts = append(parts, s.Muted.Render(keystroke)+s.Subtle.Render(" close"))
131 } else {
132 parts = append(parts, s.Muted.Render(keystroke)+s.Subtle.Render(" open "))
133 }
135 dot := s.Subtle.Render(" • ")
136 metadata := strings.Join(parts, dot)
137 metadata = dot + metadata
139 // Truncate cwd if necessary, and insert it at the beginning.
140 const dirTrimLimit = 4
141 cwd := fsext.DirTrim(fsext.PrettyPath(config.Get().WorkingDir()), dirTrimLimit)
142 cwd = ansi.Truncate(cwd, max(0, availWidth-lipgloss.Width(metadata)), "…")
143 cwd = s.Muted.Render(cwd)
145 return cwd + metadata
148func (h *header) SetDetailsOpen(open bool) {
149 h.detailsOpen = open
152// SetSession implements Header.
153func (h *header) SetSession(session session.Session) tea.Cmd {
154 h.session = session
155 return nil
158// SetWidth implements Header.
159func (h *header) SetWidth(width int) tea.Cmd {
160 h.width = width
161 return nil
164// ShowingDetails implements Header.
165func (h *header) ShowingDetails() bool {
166 return h.detailsOpen