message.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945
  1. package chat
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "maps"
  6. "slices"
  7. "strings"
  8. "time"
  9. "github.com/charmbracelet/lipgloss/v2"
  10. "github.com/charmbracelet/lipgloss/v2/compat"
  11. "github.com/charmbracelet/x/ansi"
  12. "github.com/muesli/reflow/truncate"
  13. "github.com/sst/opencode-sdk-go"
  14. "github.com/sst/opencode/internal/app"
  15. "github.com/sst/opencode/internal/components/diff"
  16. "github.com/sst/opencode/internal/styles"
  17. "github.com/sst/opencode/internal/theme"
  18. "github.com/sst/opencode/internal/util"
  19. "golang.org/x/text/cases"
  20. "golang.org/x/text/language"
  21. )
  22. type blockRenderer struct {
  23. textColor compat.AdaptiveColor
  24. backgroundColor compat.AdaptiveColor
  25. border bool
  26. borderColor *compat.AdaptiveColor
  27. borderLeft bool
  28. borderRight bool
  29. paddingTop int
  30. paddingBottom int
  31. paddingLeft int
  32. paddingRight int
  33. marginTop int
  34. marginBottom int
  35. }
  36. type renderingOption func(*blockRenderer)
  37. func WithTextColor(color compat.AdaptiveColor) renderingOption {
  38. return func(c *blockRenderer) {
  39. c.textColor = color
  40. }
  41. }
  42. func WithBackgroundColor(color compat.AdaptiveColor) renderingOption {
  43. return func(c *blockRenderer) {
  44. c.backgroundColor = color
  45. }
  46. }
  47. func WithNoBorder() renderingOption {
  48. return func(c *blockRenderer) {
  49. c.border = false
  50. }
  51. }
  52. func WithBorderColor(color compat.AdaptiveColor) renderingOption {
  53. return func(c *blockRenderer) {
  54. c.borderColor = &color
  55. }
  56. }
  57. func WithBorderLeft() renderingOption {
  58. return func(c *blockRenderer) {
  59. c.borderLeft = true
  60. c.borderRight = false
  61. }
  62. }
  63. func WithBorderRight() renderingOption {
  64. return func(c *blockRenderer) {
  65. c.borderLeft = false
  66. c.borderRight = true
  67. }
  68. }
  69. func WithBorderBoth(value bool) renderingOption {
  70. return func(c *blockRenderer) {
  71. if value {
  72. c.borderLeft = true
  73. c.borderRight = true
  74. }
  75. }
  76. }
  77. func WithMarginTop(padding int) renderingOption {
  78. return func(c *blockRenderer) {
  79. c.marginTop = padding
  80. }
  81. }
  82. func WithMarginBottom(padding int) renderingOption {
  83. return func(c *blockRenderer) {
  84. c.marginBottom = padding
  85. }
  86. }
  87. func WithPadding(padding int) renderingOption {
  88. return func(c *blockRenderer) {
  89. c.paddingTop = padding
  90. c.paddingBottom = padding
  91. c.paddingLeft = padding
  92. c.paddingRight = padding
  93. }
  94. }
  95. func WithPaddingLeft(padding int) renderingOption {
  96. return func(c *blockRenderer) {
  97. c.paddingLeft = padding
  98. }
  99. }
  100. func WithPaddingRight(padding int) renderingOption {
  101. return func(c *blockRenderer) {
  102. c.paddingRight = padding
  103. }
  104. }
  105. func WithPaddingTop(padding int) renderingOption {
  106. return func(c *blockRenderer) {
  107. c.paddingTop = padding
  108. }
  109. }
  110. func WithPaddingBottom(padding int) renderingOption {
  111. return func(c *blockRenderer) {
  112. c.paddingBottom = padding
  113. }
  114. }
  115. func renderContentBlock(
  116. app *app.App,
  117. content string,
  118. width int,
  119. options ...renderingOption,
  120. ) string {
  121. t := theme.CurrentTheme()
  122. renderer := &blockRenderer{
  123. textColor: t.TextMuted(),
  124. backgroundColor: t.BackgroundPanel(),
  125. border: true,
  126. borderLeft: true,
  127. borderRight: false,
  128. paddingTop: 1,
  129. paddingBottom: 1,
  130. paddingLeft: 2,
  131. paddingRight: 2,
  132. }
  133. for _, option := range options {
  134. option(renderer)
  135. }
  136. borderColor := t.BackgroundPanel()
  137. if renderer.borderColor != nil {
  138. borderColor = *renderer.borderColor
  139. }
  140. style := styles.NewStyle().
  141. Foreground(renderer.textColor).
  142. Background(renderer.backgroundColor).
  143. PaddingTop(renderer.paddingTop).
  144. PaddingBottom(renderer.paddingBottom).
  145. PaddingLeft(renderer.paddingLeft).
  146. PaddingRight(renderer.paddingRight).
  147. AlignHorizontal(lipgloss.Left)
  148. if renderer.border {
  149. style = style.
  150. BorderStyle(lipgloss.ThickBorder()).
  151. BorderLeft(true).
  152. BorderRight(true).
  153. BorderLeftForeground(t.BackgroundPanel()).
  154. BorderLeftBackground(t.Background()).
  155. BorderRightForeground(t.BackgroundPanel()).
  156. BorderRightBackground(t.Background())
  157. if renderer.borderLeft {
  158. style = style.BorderLeftForeground(borderColor)
  159. }
  160. if renderer.borderRight {
  161. style = style.BorderRightForeground(borderColor)
  162. }
  163. }
  164. content = style.Render(content)
  165. if renderer.marginTop > 0 {
  166. for range renderer.marginTop {
  167. content = "\n" + content
  168. }
  169. }
  170. if renderer.marginBottom > 0 {
  171. for range renderer.marginBottom {
  172. content = content + "\n"
  173. }
  174. }
  175. return content
  176. }
  177. func renderText(
  178. app *app.App,
  179. message opencode.MessageUnion,
  180. text string,
  181. author string,
  182. showToolDetails bool,
  183. width int,
  184. extra string,
  185. isThinking bool,
  186. fileParts []opencode.FilePart,
  187. agentParts []opencode.AgentPart,
  188. toolCalls ...opencode.ToolPart,
  189. ) string {
  190. t := theme.CurrentTheme()
  191. var ts time.Time
  192. backgroundColor := t.BackgroundPanel()
  193. var content string
  194. switch casted := message.(type) {
  195. case opencode.AssistantMessage:
  196. bg := t.Background()
  197. if isThinking {
  198. bg = t.BackgroundPanel()
  199. }
  200. ts = time.UnixMilli(int64(casted.Time.Created))
  201. content = util.ToMarkdown(text, width, bg)
  202. if isThinking {
  203. content = styles.NewStyle().Background(bg).Foreground(t.TextMuted()).Render("Thinking") + "\n\n" + content
  204. }
  205. case opencode.UserMessage:
  206. ts = time.UnixMilli(int64(casted.Time.Created))
  207. base := styles.NewStyle().Foreground(t.Text()).Background(backgroundColor)
  208. var result strings.Builder
  209. lastEnd := int64(0)
  210. // Apply highlighting to filenames and base style to rest of text BEFORE wrapping
  211. textLen := int64(len(text))
  212. // Collect all parts to highlight (both file and agent parts)
  213. type highlightPart struct {
  214. start int64
  215. end int64
  216. color compat.AdaptiveColor
  217. }
  218. var highlights []highlightPart
  219. // Add file parts with secondary color
  220. for _, filePart := range fileParts {
  221. highlights = append(highlights, highlightPart{
  222. start: filePart.Source.Text.Start,
  223. end: filePart.Source.Text.End,
  224. color: t.Secondary(),
  225. })
  226. }
  227. // Add agent parts with secondary color (same as file parts)
  228. for _, agentPart := range agentParts {
  229. highlights = append(highlights, highlightPart{
  230. start: agentPart.Source.Start,
  231. end: agentPart.Source.End,
  232. color: t.Secondary(),
  233. })
  234. }
  235. // Sort highlights by start position
  236. slices.SortFunc(highlights, func(a, b highlightPart) int {
  237. if a.start < b.start {
  238. return -1
  239. }
  240. if a.start > b.start {
  241. return 1
  242. }
  243. return 0
  244. })
  245. // Merge overlapping highlights to prevent duplication
  246. merged := make([]highlightPart, 0)
  247. for _, part := range highlights {
  248. if len(merged) == 0 {
  249. merged = append(merged, part)
  250. continue
  251. }
  252. last := &merged[len(merged)-1]
  253. // If current part overlaps with the last one, merge them
  254. if part.start <= last.end {
  255. if part.end > last.end {
  256. last.end = part.end
  257. }
  258. } else {
  259. merged = append(merged, part)
  260. }
  261. }
  262. for _, part := range merged {
  263. highlight := base.Foreground(part.color)
  264. start, end := part.start, part.end
  265. if end > textLen {
  266. end = textLen
  267. }
  268. if start > textLen {
  269. start = textLen
  270. }
  271. if start > lastEnd {
  272. result.WriteString(base.Render(text[lastEnd:start]))
  273. }
  274. if start < end {
  275. result.WriteString(highlight.Render(text[start:end]))
  276. }
  277. lastEnd = end
  278. }
  279. if lastEnd < textLen {
  280. result.WriteString(base.Render(text[lastEnd:]))
  281. }
  282. // wrap styled text
  283. styledText := result.String()
  284. wrappedText := ansi.WordwrapWc(styledText, width-6, " -")
  285. content = base.Width(width - 6).Render(wrappedText)
  286. }
  287. timestamp := ts.
  288. Local().
  289. Format("02 Jan 2006 03:04 PM")
  290. if time.Now().Format("02 Jan 2006") == timestamp[:11] {
  291. timestamp = timestamp[12:]
  292. }
  293. // Check if this is an assistant message with mode (agent) information
  294. var modelAndAgentSuffix string
  295. if assistantMsg, ok := message.(opencode.AssistantMessage); ok && assistantMsg.Mode != "" {
  296. // Find the agent index by name to get the correct color
  297. var agentIndex int
  298. for i, agent := range app.Agents {
  299. if agent.Name == assistantMsg.Mode {
  300. agentIndex = i
  301. break
  302. }
  303. }
  304. // Get agent color based on the original agent index (same as status bar)
  305. agentColor := util.GetAgentColor(agentIndex)
  306. // Style the agent name with the same color as status bar
  307. agentName := strings.Title(assistantMsg.Mode)
  308. styledAgentName := styles.NewStyle().Foreground(agentColor).Bold(true).Render(agentName)
  309. modelAndAgentSuffix = fmt.Sprintf(" | %s | %s", assistantMsg.ModelID, styledAgentName)
  310. }
  311. var info string
  312. if modelAndAgentSuffix != "" {
  313. // For assistant messages: "timestamp | modelID | agentName"
  314. info = fmt.Sprintf("%s%s", timestamp, modelAndAgentSuffix)
  315. } else {
  316. // For user messages: "author (timestamp)"
  317. info = fmt.Sprintf("%s (%s)", author, timestamp)
  318. }
  319. info = styles.NewStyle().Foreground(t.TextMuted()).Render(info)
  320. if !showToolDetails && toolCalls != nil && len(toolCalls) > 0 {
  321. content = content + "\n\n"
  322. for _, toolCall := range toolCalls {
  323. title := renderToolTitle(toolCall, width-2)
  324. style := styles.NewStyle()
  325. if toolCall.State.Status == opencode.ToolPartStateStatusError {
  326. style = style.Foreground(t.Error())
  327. }
  328. title = style.Render(title)
  329. title = "∟ " + title + "\n"
  330. content = content + title
  331. }
  332. }
  333. sections := []string{content, info}
  334. if extra != "" {
  335. sections = append(sections, "\n"+extra)
  336. }
  337. content = strings.Join(sections, "\n")
  338. switch message.(type) {
  339. case opencode.UserMessage:
  340. return renderContentBlock(
  341. app,
  342. content,
  343. width,
  344. WithTextColor(t.Text()),
  345. WithBorderColor(t.Secondary()),
  346. )
  347. case opencode.AssistantMessage:
  348. if isThinking {
  349. return renderContentBlock(
  350. app,
  351. content,
  352. width,
  353. WithTextColor(t.Text()),
  354. WithBackgroundColor(t.BackgroundPanel()),
  355. WithBorderColor(t.BackgroundPanel()),
  356. )
  357. }
  358. return renderContentBlock(
  359. app,
  360. content,
  361. width,
  362. WithNoBorder(),
  363. WithBackgroundColor(t.Background()),
  364. )
  365. }
  366. return ""
  367. }
  368. func renderToolDetails(
  369. app *app.App,
  370. toolCall opencode.ToolPart,
  371. permission opencode.Permission,
  372. width int,
  373. ) string {
  374. measure := util.Measure("chat.renderToolDetails")
  375. defer measure("tool", toolCall.Tool)
  376. ignoredTools := []string{"todoread"}
  377. if slices.Contains(ignoredTools, toolCall.Tool) {
  378. return ""
  379. }
  380. if toolCall.State.Status == opencode.ToolPartStateStatusPending {
  381. title := renderToolTitle(toolCall, width)
  382. return renderContentBlock(app, title, width)
  383. }
  384. var result *string
  385. if toolCall.State.Output != "" {
  386. result = &toolCall.State.Output
  387. }
  388. toolInputMap := make(map[string]any)
  389. if toolCall.State.Input != nil {
  390. value := toolCall.State.Input
  391. if m, ok := value.(map[string]any); ok {
  392. toolInputMap = m
  393. keys := make([]string, 0, len(toolInputMap))
  394. for key := range toolInputMap {
  395. keys = append(keys, key)
  396. }
  397. slices.Sort(keys)
  398. }
  399. }
  400. body := ""
  401. t := theme.CurrentTheme()
  402. backgroundColor := t.BackgroundPanel()
  403. borderColor := t.BackgroundPanel()
  404. defaultStyle := styles.NewStyle().Background(backgroundColor).Width(width - 6).Render
  405. permissionContent := ""
  406. if permission.ID != "" {
  407. borderColor = t.Warning()
  408. base := styles.NewStyle().Background(backgroundColor)
  409. text := base.Foreground(t.Text()).Bold(true).Render
  410. muted := base.Foreground(t.TextMuted()).Render
  411. permissionContent = "Permission required to run this tool:\n\n"
  412. permissionContent += text(
  413. "enter ",
  414. ) + muted(
  415. "accept ",
  416. ) + text(
  417. "a",
  418. ) + muted(
  419. " accept always ",
  420. ) + text(
  421. "esc",
  422. ) + muted(
  423. " reject",
  424. )
  425. }
  426. if permission.Metadata != nil {
  427. metadata, ok := toolCall.State.Metadata.(map[string]any)
  428. if metadata == nil || !ok {
  429. metadata = map[string]any{}
  430. }
  431. maps.Copy(metadata, permission.Metadata)
  432. toolCall.State.Metadata = metadata
  433. }
  434. if toolCall.State.Metadata != nil {
  435. metadata := toolCall.State.Metadata.(map[string]any)
  436. switch toolCall.Tool {
  437. case "read":
  438. var preview any
  439. if metadata != nil {
  440. preview = metadata["preview"]
  441. }
  442. if preview != nil && toolInputMap["filePath"] != nil {
  443. filename := toolInputMap["filePath"].(string)
  444. body = preview.(string)
  445. body = util.RenderFile(filename, body, width, util.WithTruncate(6))
  446. }
  447. case "edit":
  448. if filename, ok := toolInputMap["filePath"].(string); ok {
  449. var diffField any
  450. if metadata != nil {
  451. diffField = metadata["diff"]
  452. }
  453. if diffField != nil {
  454. patch := diffField.(string)
  455. var formattedDiff string
  456. if width < 120 {
  457. formattedDiff, _ = diff.FormatUnifiedDiff(
  458. filename,
  459. patch,
  460. diff.WithWidth(width-2),
  461. )
  462. } else {
  463. formattedDiff, _ = diff.FormatDiff(
  464. filename,
  465. patch,
  466. diff.WithWidth(width-2),
  467. )
  468. }
  469. body = strings.TrimSpace(formattedDiff)
  470. style := styles.NewStyle().
  471. Background(backgroundColor).
  472. Foreground(t.TextMuted()).
  473. Padding(1, 2).
  474. Width(width - 4)
  475. if diagnostics := renderDiagnostics(metadata, filename, backgroundColor, width-6); diagnostics != "" {
  476. diagnostics = style.Render(diagnostics)
  477. body += "\n" + diagnostics
  478. }
  479. title := renderToolTitle(toolCall, width)
  480. title = style.Render(title)
  481. content := title + "\n" + body
  482. if permissionContent != "" {
  483. permissionContent = styles.NewStyle().
  484. Background(backgroundColor).
  485. Padding(1, 2).
  486. Render(permissionContent)
  487. content += "\n" + permissionContent
  488. }
  489. content = renderContentBlock(
  490. app,
  491. content,
  492. width,
  493. WithPadding(0),
  494. WithBorderColor(borderColor),
  495. WithBorderBoth(permission.ID != ""),
  496. )
  497. return content
  498. }
  499. }
  500. case "write":
  501. if filename, ok := toolInputMap["filePath"].(string); ok {
  502. if content, ok := toolInputMap["content"].(string); ok {
  503. body = util.RenderFile(filename, content, width)
  504. if diagnostics := renderDiagnostics(metadata, filename, backgroundColor, width-4); diagnostics != "" {
  505. body += "\n\n" + diagnostics
  506. }
  507. }
  508. }
  509. case "bash":
  510. command := toolInputMap["command"].(string)
  511. body = fmt.Sprintf("```console\n$ %s\n", command)
  512. stdout := metadata["stdout"]
  513. if stdout != nil {
  514. body += ansi.Strip(fmt.Sprintf("%s", stdout))
  515. }
  516. stderr := metadata["stderr"]
  517. if stderr != nil {
  518. body += ansi.Strip(fmt.Sprintf("%s", stderr))
  519. }
  520. body += "```"
  521. body = util.ToMarkdown(body, width, backgroundColor)
  522. case "webfetch":
  523. if format, ok := toolInputMap["format"].(string); ok && result != nil {
  524. body = *result
  525. body = util.TruncateHeight(body, 10)
  526. if format == "html" || format == "markdown" {
  527. body = util.ToMarkdown(body, width, backgroundColor)
  528. }
  529. }
  530. case "todowrite":
  531. todos := metadata["todos"]
  532. if todos != nil {
  533. for _, item := range todos.([]any) {
  534. todo := item.(map[string]any)
  535. content := todo["content"].(string)
  536. switch todo["status"] {
  537. case "completed":
  538. body += fmt.Sprintf("- [x] %s\n", content)
  539. case "cancelled":
  540. // strike through cancelled todo
  541. body += fmt.Sprintf("- [ ] ~~%s~~\n", content)
  542. case "in_progress":
  543. // highlight in progress todo
  544. body += fmt.Sprintf("- [ ] `%s`\n", content)
  545. default:
  546. body += fmt.Sprintf("- [ ] %s\n", content)
  547. }
  548. }
  549. body = util.ToMarkdown(body, width, backgroundColor)
  550. }
  551. case "task":
  552. summary := metadata["summary"]
  553. if summary != nil {
  554. toolcalls := summary.([]any)
  555. steps := []string{}
  556. for _, item := range toolcalls {
  557. data, _ := json.Marshal(item)
  558. var toolCall opencode.ToolPart
  559. _ = json.Unmarshal(data, &toolCall)
  560. step := renderToolTitle(toolCall, width-2)
  561. step = "∟ " + step
  562. steps = append(steps, step)
  563. }
  564. body = strings.Join(steps, "\n")
  565. }
  566. body = defaultStyle(body)
  567. default:
  568. if result == nil {
  569. empty := ""
  570. result = &empty
  571. }
  572. body = *result
  573. body = util.TruncateHeight(body, 10)
  574. body = defaultStyle(body)
  575. }
  576. }
  577. error := ""
  578. if toolCall.State.Status == opencode.ToolPartStateStatusError {
  579. error = toolCall.State.Error
  580. }
  581. if error != "" {
  582. body = styles.NewStyle().
  583. Width(width - 6).
  584. Foreground(t.Error()).
  585. Background(backgroundColor).
  586. Render(error)
  587. }
  588. if body == "" && error == "" && result != nil {
  589. body = *result
  590. body = util.TruncateHeight(body, 10)
  591. body = defaultStyle(body)
  592. }
  593. if body == "" {
  594. body = defaultStyle("")
  595. }
  596. title := renderToolTitle(toolCall, width)
  597. content := title + "\n\n" + body
  598. if permissionContent != "" {
  599. content += "\n\n\n" + permissionContent
  600. }
  601. return renderContentBlock(
  602. app,
  603. content,
  604. width,
  605. WithBorderColor(borderColor),
  606. WithBorderBoth(permission.ID != ""),
  607. )
  608. }
  609. func renderToolName(name string) string {
  610. switch name {
  611. case "webfetch":
  612. return "Fetch"
  613. case "invalid":
  614. return "Invalid"
  615. default:
  616. normalizedName := name
  617. if after, ok := strings.CutPrefix(name, "opencode_"); ok {
  618. normalizedName = after
  619. }
  620. return cases.Title(language.Und).String(normalizedName)
  621. }
  622. }
  623. func getTodoPhase(metadata map[string]any) string {
  624. todos, ok := metadata["todos"].([]any)
  625. if !ok || len(todos) == 0 {
  626. return "Plan"
  627. }
  628. counts := map[string]int{"pending": 0, "completed": 0}
  629. for _, item := range todos {
  630. if todo, ok := item.(map[string]any); ok {
  631. if status, ok := todo["status"].(string); ok {
  632. counts[status]++
  633. }
  634. }
  635. }
  636. total := len(todos)
  637. switch {
  638. case counts["pending"] == total:
  639. return "Creating plan"
  640. case counts["completed"] == total:
  641. return "Completing plan"
  642. default:
  643. return "Updating plan"
  644. }
  645. }
  646. func getTodoTitle(toolCall opencode.ToolPart) string {
  647. if toolCall.State.Status == opencode.ToolPartStateStatusCompleted {
  648. if metadata, ok := toolCall.State.Metadata.(map[string]any); ok {
  649. return getTodoPhase(metadata)
  650. }
  651. }
  652. return "Plan"
  653. }
  654. func renderToolTitle(
  655. toolCall opencode.ToolPart,
  656. width int,
  657. ) string {
  658. if toolCall.State.Status == opencode.ToolPartStateStatusPending {
  659. title := renderToolAction(toolCall.Tool)
  660. return styles.NewStyle().Width(width - 6).Render(title)
  661. }
  662. toolArgs := ""
  663. toolArgsMap := make(map[string]any)
  664. if toolCall.State.Input != nil {
  665. value := toolCall.State.Input
  666. if m, ok := value.(map[string]any); ok {
  667. toolArgsMap = m
  668. keys := make([]string, 0, len(toolArgsMap))
  669. for key := range toolArgsMap {
  670. keys = append(keys, key)
  671. }
  672. slices.Sort(keys)
  673. firstKey := ""
  674. if len(keys) > 0 {
  675. firstKey = keys[0]
  676. }
  677. toolArgs = renderArgs(&toolArgsMap, firstKey)
  678. }
  679. }
  680. title := renderToolName(toolCall.Tool)
  681. switch toolCall.Tool {
  682. case "read":
  683. toolArgs = renderArgs(&toolArgsMap, "filePath")
  684. title = fmt.Sprintf("%s %s", title, toolArgs)
  685. case "edit", "write":
  686. if filename, ok := toolArgsMap["filePath"].(string); ok {
  687. title = fmt.Sprintf("%s %s", title, util.Relative(filename))
  688. }
  689. case "bash":
  690. if description, ok := toolArgsMap["description"].(string); ok {
  691. title = fmt.Sprintf("%s %s", title, description)
  692. }
  693. case "task":
  694. description := toolArgsMap["description"]
  695. subagent := toolArgsMap["subagent_type"]
  696. if description != nil && subagent != nil {
  697. title = fmt.Sprintf("%s[%s] %s", title, subagent, description)
  698. } else if description != nil {
  699. title = fmt.Sprintf("%s %s", title, description)
  700. }
  701. case "webfetch":
  702. toolArgs = renderArgs(&toolArgsMap, "url")
  703. title = fmt.Sprintf("%s %s", title, toolArgs)
  704. case "todowrite":
  705. title = getTodoTitle(toolCall)
  706. case "todoread":
  707. return "Plan"
  708. case "invalid":
  709. if actualTool, ok := toolArgsMap["tool"].(string); ok {
  710. title = renderToolName(actualTool)
  711. }
  712. default:
  713. toolName := renderToolName(toolCall.Tool)
  714. title = fmt.Sprintf("%s %s", toolName, toolArgs)
  715. }
  716. title = truncate.StringWithTail(title, uint(width-6), "...")
  717. if toolCall.State.Error != "" {
  718. t := theme.CurrentTheme()
  719. title = styles.NewStyle().Foreground(t.Error()).Render(title)
  720. }
  721. return title
  722. }
  723. func renderToolAction(name string) string {
  724. switch name {
  725. case "task":
  726. return "Delegating..."
  727. case "bash":
  728. return "Writing command..."
  729. case "edit":
  730. return "Preparing edit..."
  731. case "webfetch":
  732. return "Fetching from the web..."
  733. case "glob":
  734. return "Finding files..."
  735. case "grep":
  736. return "Searching content..."
  737. case "list":
  738. return "Listing directory..."
  739. case "read":
  740. return "Reading file..."
  741. case "write":
  742. return "Preparing write..."
  743. case "todowrite", "todoread":
  744. return "Planning..."
  745. case "patch":
  746. return "Preparing patch..."
  747. }
  748. return "Working..."
  749. }
  750. func renderArgs(args *map[string]any, titleKey string) string {
  751. if args == nil || len(*args) == 0 {
  752. return ""
  753. }
  754. keys := make([]string, 0, len(*args))
  755. for key := range *args {
  756. keys = append(keys, key)
  757. }
  758. slices.Sort(keys)
  759. title := ""
  760. parts := []string{}
  761. for _, key := range keys {
  762. value := (*args)[key]
  763. if value == nil {
  764. continue
  765. }
  766. if key == "filePath" || key == "path" {
  767. value = util.Relative(value.(string))
  768. }
  769. if key == titleKey {
  770. title = fmt.Sprintf("%s", value)
  771. continue
  772. }
  773. parts = append(parts, fmt.Sprintf("%s=%v", key, value))
  774. }
  775. if len(parts) == 0 {
  776. return title
  777. }
  778. return fmt.Sprintf("%s (%s)", title, strings.Join(parts, ", "))
  779. }
  780. // Diagnostic represents an LSP diagnostic
  781. type Diagnostic struct {
  782. Range struct {
  783. Start struct {
  784. Line int `json:"line"`
  785. Character int `json:"character"`
  786. } `json:"start"`
  787. } `json:"range"`
  788. Severity int `json:"severity"`
  789. Message string `json:"message"`
  790. }
  791. // renderDiagnostics formats LSP diagnostics for display in the TUI
  792. func renderDiagnostics(
  793. metadata map[string]any,
  794. filePath string,
  795. backgroundColor compat.AdaptiveColor,
  796. width int,
  797. ) string {
  798. if diagnosticsData, ok := metadata["diagnostics"].(map[string]any); ok {
  799. if fileDiagnostics, ok := diagnosticsData[filePath].([]any); ok {
  800. var errorDiagnostics []string
  801. for _, diagInterface := range fileDiagnostics {
  802. diagMap, ok := diagInterface.(map[string]any)
  803. if !ok {
  804. continue
  805. }
  806. // Parse the diagnostic
  807. var diag Diagnostic
  808. diagBytes, err := json.Marshal(diagMap)
  809. if err != nil {
  810. continue
  811. }
  812. if err := json.Unmarshal(diagBytes, &diag); err != nil {
  813. continue
  814. }
  815. // Only show error diagnostics (severity === 1)
  816. if diag.Severity != 1 {
  817. continue
  818. }
  819. line := diag.Range.Start.Line + 1 // 1-based
  820. column := diag.Range.Start.Character + 1 // 1-based
  821. errorDiagnostics = append(
  822. errorDiagnostics,
  823. fmt.Sprintf("Error [%d:%d] %s", line, column, diag.Message),
  824. )
  825. }
  826. if len(errorDiagnostics) == 0 {
  827. return ""
  828. }
  829. t := theme.CurrentTheme()
  830. var result strings.Builder
  831. for _, diagnostic := range errorDiagnostics {
  832. if result.Len() > 0 {
  833. result.WriteString("\n\n")
  834. }
  835. diagnostic = ansi.WordwrapWc(diagnostic, width, " -")
  836. result.WriteString(
  837. styles.NewStyle().
  838. Background(backgroundColor).
  839. Foreground(t.Error()).
  840. Render(diagnostic),
  841. )
  842. }
  843. return result.String()
  844. }
  845. }
  846. return ""
  847. // diagnosticsData should be a map[string][]Diagnostic
  848. // strDiagnosticsData := diagnosticsData.Raw()
  849. // diagnosticsMap := gjson.Parse(strDiagnosticsData).Value().(map[string]any)
  850. // fileDiagnostics, ok := diagnosticsMap[filePath]
  851. // if !ok {
  852. // return ""
  853. // }
  854. // diagnosticsList, ok := fileDiagnostics.([]any)
  855. // if !ok {
  856. // return ""
  857. // }
  858. }