@@ -0,0 +1,98 @@
1+package chat
2+
3+import (
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.
13+type GenericToolMessageItem struct {
14+ *baseToolMessageItem
15+}
16+
17+var _ ToolMessageItem = (*GenericToolMessageItem)(nil)
18+
19+// NewGenericToolMessageItem creates a new [GenericToolMessageItem].
20+func 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.
30+type GenericToolRenderContext struct{}
31+
32+// RenderTool implements the [ToolRenderer] interface.
33+func (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.
94+func genericPrettyName(name string) string {
95+ name = strings.ReplaceAll(name, "_", " ")
96+ name = strings.ReplaceAll(name, "-", " ")
97+ return stringext.Capitalize(name)
98+}