1package chat
2
3import (
4 "encoding/json"
5 "strings"
6
7 "github.com/charmbracelet/crush/internal/message"
8 "github.com/charmbracelet/crush/internal/stringext"
9 "github.com/charmbracelet/crush/internal/ui/styles"
10)
11
12// GenericToolMessageItem is a message item that represents an unknown tool call.
13type GenericToolMessageItem struct {
14 *baseToolMessageItem
15}
16
17var _ ToolMessageItem = (*GenericToolMessageItem)(nil)
18
19// NewGenericToolMessageItem creates a new [GenericToolMessageItem].
20func NewGenericToolMessageItem(
21 sty *styles.Styles,
22 toolCall message.ToolCall,
23 result *message.ToolResult,
24 canceled bool,
25) ToolMessageItem {
26 return newBaseToolMessageItem(sty, toolCall, result, &GenericToolRenderContext{}, canceled)
27}
28
29// GenericToolRenderContext renders unknown/generic tool messages.
30type GenericToolRenderContext struct{}
31
32// RenderTool implements the [ToolRenderer] interface.
33func (g *GenericToolRenderContext) RenderTool(sty *styles.Styles, width int, opts *ToolRenderOpts) string {
34 cappedWidth := cappedMessageWidth(width)
35 name := genericPrettyName(opts.ToolCall.Name)
36
37 if opts.IsPending() {
38 return pendingTool(sty, name, opts.Anim)
39 }
40
41 var params map[string]any
42 if err := json.Unmarshal([]byte(opts.ToolCall.Input), ¶ms); err != nil {
43 return toolErrorContent(sty, &message.ToolResult{Content: "Invalid parameters"}, cappedWidth)
44 }
45
46 var toolParams []string
47 if len(params) > 0 {
48 parsed, _ := json.Marshal(params)
49 toolParams = append(toolParams, string(parsed))
50 }
51
52 header := toolHeader(sty, opts.Status, name, cappedWidth, opts.Compact, toolParams...)
53 if opts.Compact {
54 return header
55 }
56
57 if earlyState, ok := toolEarlyStateContent(sty, opts, cappedWidth); ok {
58 return joinToolParts(header, earlyState)
59 }
60
61 if !opts.HasResult() || opts.Result.Content == "" {
62 return header
63 }
64
65 bodyWidth := cappedWidth - toolBodyLeftPaddingTotal
66
67 // Handle image data.
68 if opts.Result.Data != "" && strings.HasPrefix(opts.Result.MIMEType, "image/") {
69 body := sty.Tool.Body.Render(toolOutputImageContent(sty, opts.Result.Data, opts.Result.MIMEType))
70 return joinToolParts(header, body)
71 }
72
73 // Try to parse result as JSON for pretty display.
74 var result json.RawMessage
75 var body string
76 if err := json.Unmarshal([]byte(opts.Result.Content), &result); err == nil {
77 prettyResult, err := json.MarshalIndent(result, "", " ")
78 if err == nil {
79 body = sty.Tool.Body.Render(toolOutputCodeContent(sty, "result.json", string(prettyResult), 0, bodyWidth, opts.ExpandedContent))
80 } else {
81 body = sty.Tool.Body.Render(toolOutputPlainContent(sty, opts.Result.Content, bodyWidth, opts.ExpandedContent))
82 }
83 } else if looksLikeMarkdown(opts.Result.Content) {
84 body = sty.Tool.Body.Render(toolOutputCodeContent(sty, "result.md", opts.Result.Content, 0, bodyWidth, opts.ExpandedContent))
85 } else {
86 body = sty.Tool.Body.Render(toolOutputPlainContent(sty, opts.Result.Content, bodyWidth, opts.ExpandedContent))
87 }
88
89 return joinToolParts(header, body)
90}
91
92// genericPrettyName converts a snake_case or kebab-case tool name to a
93// human-readable title case name.
94func genericPrettyName(name string) string {
95 name = strings.ReplaceAll(name, "_", " ")
96 name = strings.ReplaceAll(name, "-", " ")
97 return stringext.Capitalize(name)
98}