1package chat
 2
 3import (
 4	"encoding/json"
 5	"strings"
 6
 7	"github.com/charmbracelet/crush/internal/message"
 8	"github.com/charmbracelet/crush/internal/ui/styles"
 9)
10
11// GenericToolMessageItem is a message item that represents an unknown tool call.
12type GenericToolMessageItem struct {
13	*baseToolMessageItem
14}
15
16var _ ToolMessageItem = (*GenericToolMessageItem)(nil)
17
18// NewGenericToolMessageItem creates a new [GenericToolMessageItem].
19func NewGenericToolMessageItem(
20	sty *styles.Styles,
21	toolCall message.ToolCall,
22	result *message.ToolResult,
23	canceled bool,
24) ToolMessageItem {
25	return newBaseToolMessageItem(sty, toolCall, result, &GenericToolRenderContext{}, canceled)
26}
27
28// GenericToolRenderContext renders unknown/generic tool messages.
29type GenericToolRenderContext struct{}
30
31// RenderTool implements the [ToolRenderer] interface.
32func (g *GenericToolRenderContext) RenderTool(sty *styles.Styles, width int, opts *ToolRenderOpts) string {
33	cappedWidth := cappedMessageWidth(width)
34	name := humanizedToolName(opts.ToolCall.Name)
35
36	if opts.IsPending() {
37		return pendingTool(sty, name, opts.Anim, opts.Compact)
38	}
39
40	var params map[string]any
41	if err := json.Unmarshal([]byte(opts.ToolCall.Input), &params); err != nil {
42		return toolErrorContent(sty, &message.ToolResult{Content: "Invalid parameters"}, cappedWidth)
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, cappedWidth, opts.Compact, toolParams...)
52	if opts.Compact {
53		return header
54	}
55
56	if earlyState, ok := toolEarlyStateContent(sty, opts, cappedWidth); ok {
57		return joinToolParts(header, earlyState)
58	}
59
60	if !opts.HasResult() || opts.Result.Content == "" {
61		return header
62	}
63
64	bodyWidth := cappedWidth - toolBodyLeftPaddingTotal
65
66	if opts.Result.Data != "" && strings.HasPrefix(opts.Result.MIMEType, "image/") {
67		body := sty.Tool.Body.Render(toolOutputImageContent(sty, opts.Result.Data, opts.Result.MIMEType))
68		return joinToolParts(header, body)
69	}
70
71	body := renderToolResultTextContent(sty, opts.Result.Content, toolResultContentWidths{Body: bodyWidth, Diff: cappedWidth}, opts.ExpandedContent)
72	return joinToolParts(header, body)
73}