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 name := genericPrettyName(opts.ToolCall.Name)
35
36 if opts.IsPending() {
37 return pendingTool(sty, name, opts.Anim)
38 }
39
40 var params map[string]any
41 if err := json.Unmarshal([]byte(opts.ToolCall.Input), ¶ms); err != nil {
42 return toolErrorContent(sty, &message.ToolResult{Content: "Invalid parameters"}, width)
43 }
44
45 var toolParams []string
46 if len(params) > 0 {
47 parsed, _ := json.Marshal(params)
48 toolParams = append(toolParams, string(parsed))
49 }
50
51 header := toolHeader(sty, opts.Status, name, width, opts.Compact, toolParams...)
52 if opts.Compact {
53 return header
54 }
55
56 if earlyState, ok := toolEarlyStateContent(sty, opts, width); ok {
57 return joinToolParts(header, earlyState)
58 }
59
60 if !opts.HasResult() || opts.Result.Content == "" {
61 return header
62 }
63
64 bodyWidth := width - toolBodyLeftPaddingTotal
65
66 // Handle image data.
67 if opts.Result.Data != "" && strings.HasPrefix(opts.Result.MIMEType, "image/") {
68 body := sty.Tool.Body.Render(toolOutputImageContent(sty, opts.Result.Data, opts.Result.MIMEType))
69 return joinToolParts(header, body)
70 }
71
72 // Try to parse result as JSON for pretty display.
73 var result json.RawMessage
74 var body string
75 if err := json.Unmarshal([]byte(opts.Result.Content), &result); err == nil {
76 prettyResult, err := json.MarshalIndent(result, "", " ")
77 if err == nil {
78 body = sty.Tool.Body.Render(toolOutputCodeContent(sty, "result.json", string(prettyResult), 0, bodyWidth, opts.ExpandedContent))
79 } else {
80 body = sty.Tool.Body.Render(toolOutputPlainContent(sty, opts.Result.Content, bodyWidth, opts.ExpandedContent))
81 }
82 } else if looksLikeMarkdown(opts.Result.Content) {
83 body = sty.Tool.Body.Render(toolOutputCodeContent(sty, "result.md", opts.Result.Content, 0, bodyWidth, opts.ExpandedContent))
84 } else {
85 body = sty.Tool.Body.Render(toolOutputPlainContent(sty, opts.Result.Content, bodyWidth, opts.ExpandedContent))
86 }
87
88 return joinToolParts(header, body)
89}
90
91// genericPrettyName converts a snake_case or kebab-case tool name to a
92// human-readable title case name.
93func genericPrettyName(name string) string {
94 name = strings.ReplaceAll(name, "_", " ")
95 name = strings.ReplaceAll(name, "-", " ")
96 return stringext.Capitalize(name)
97}