1package chat
 2
 3import (
 4	"encoding/json"
 5
 6	"github.com/charmbracelet/crush/internal/agent/tools"
 7	"github.com/charmbracelet/crush/internal/fsext"
 8	"github.com/charmbracelet/crush/internal/message"
 9	"github.com/charmbracelet/crush/internal/ui/styles"
10)
11
12// ReferencesToolMessageItem is a message item that represents a references tool call.
13type ReferencesToolMessageItem struct {
14	*baseToolMessageItem
15}
16
17var _ ToolMessageItem = (*ReferencesToolMessageItem)(nil)
18
19// NewReferencesToolMessageItem creates a new [ReferencesToolMessageItem].
20func NewReferencesToolMessageItem(
21	sty *styles.Styles,
22	toolCall message.ToolCall,
23	result *message.ToolResult,
24	canceled bool,
25) ToolMessageItem {
26	return newBaseToolMessageItem(sty, toolCall, result, &ReferencesToolRenderContext{}, canceled)
27}
28
29// ReferencesToolRenderContext renders references tool messages.
30type ReferencesToolRenderContext struct{}
31
32// RenderTool implements the [ToolRenderer] interface.
33func (r *ReferencesToolRenderContext) RenderTool(sty *styles.Styles, width int, opts *ToolRenderOpts) string {
34	cappedWidth := cappedMessageWidth(width)
35	if opts.IsPending() {
36		return pendingTool(sty, "Find References", opts.Anim)
37	}
38
39	var params tools.ReferencesParams
40	_ = json.Unmarshal([]byte(opts.ToolCall.Input), &params)
41
42	toolParams := []string{params.Symbol}
43	if params.Path != "" {
44		toolParams = append(toolParams, "path", fsext.PrettyPath(params.Path))
45	}
46
47	header := toolHeader(sty, opts.Status, "Find References", cappedWidth, opts.Compact, toolParams...)
48	if opts.Compact {
49		return header
50	}
51
52	if earlyState, ok := toolEarlyStateContent(sty, opts, cappedWidth); ok {
53		return joinToolParts(header, earlyState)
54	}
55
56	if opts.HasEmptyResult() {
57		return header
58	}
59
60	bodyWidth := cappedWidth - toolBodyLeftPaddingTotal
61	body := sty.Tool.Body.Render(toolOutputPlainContent(sty, opts.Result.Content, bodyWidth, opts.ExpandedContent))
62	return joinToolParts(header, body)
63}