acp

type ACPSession interface {
	SendPrompt(ctx context.Context, text string, onUpdate func(*SessionUpdateParams)) (*SessionPromptResult, error)
	GetMessages() []Message
	GetLastMessage() *Message
	SessionID() string
	Close() error
}

type AgentCapabilities struct {
	LoadSession        bool                `json:"loadSession,omitempty"`
	PromptCapabilities *PromptCapabilities `json:"promptCapabilities,omitempty"`
	MCP                *MCPCapabilities    `json:"mcp,omitempty"`
}

type AgentResult struct {
	// StopReason indicates why the agent stopped.
	StopReason StopReason `json:"stopReason"`

	// Content is the final response content from the agent.
	Content string `json:"content,omitempty"`

	// Usage contains token usage statistics.
	Usage *TokenUsage `json:"usage,omitempty"`

	// Duration is the total execution time.
	Duration time.Duration `json:"duration"`

	// Error contains any error that occurred (when StopReason is error).
	Error error `json:"error,omitempty"`
}

type AuthMethod struct {
	ID          string `json:"id"`
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
}

type Client struct {
	// contains filtered or unexported fields
}

type ClientCapabilities struct {
	FS       *FSCapabilities `json:"fs,omitempty"`
	Terminal bool            `json:"terminal,omitempty"`
}

type ClientConfig struct {
	Command string
	Args    []string
	Env     map[string]string
	Cwd     string

	// Timeout for initialization
	InitTimeout time.Duration

	// Logger for ACP activity logging. If nil, a no-op logger is used.
	Logger *zap.Logger
}

type FSCapabilities struct {
	ReadTextFile  bool `json:"readTextFile,omitempty"`
	WriteTextFile bool `json:"writeTextFile,omitempty"`
}

type InitializeParams struct {
	ProtocolVersion    int                `json:"protocolVersion"`
	ClientCapabilities ClientCapabilities `json:"clientCapabilities,omitempty"`
}

type InitializeResult struct {
	ProtocolVersion   int               `json:"protocolVersion"`
	AgentCapabilities AgentCapabilities `json:"agentCapabilities,omitempty"`
	AuthMethods       []AuthMethod      `json:"authMethods,omitempty"`
}

type JSONRPCError struct {
	Code    int         `json:"code"`
	Message string      `json:"message"`
	Data    interface{} `json:"data,omitempty"`
}

type JSONRPCNotification struct {
	JSONRPC string          `json:"jsonrpc"`
	Method  string          `json:"method"`
	Params  json.RawMessage `json:"params,omitempty"`
}

type JSONRPCRequest struct {
	JSONRPC string      `json:"jsonrpc"`
	ID      int         `json:"id"`
	Method  string      `json:"method"`
	Params  interface{} `json:"params,omitempty"`
}

type JSONRPCResponse struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      *int            `json:"id"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   *JSONRPCError   `json:"error,omitempty"`
}

type MCPCapabilities struct {
	HTTP bool `json:"http,omitempty"`
	SSE  bool `json:"sse,omitempty"`
}

type MCPServer struct {
	Name    string            `json:"name"`
	Command string            `json:"command,omitempty"`
	Args    []string          `json:"args,omitempty"`
	Env     map[string]string `json:"env,omitempty"`
	URL     string            `json:"url,omitempty"`
}

type Message struct {
	Role       string         `json:"role"`
	Content    string         `json:"content"`
	Name       string         `json:"name,omitempty"`
	ToolCallID string         `json:"toolCallId,omitempty"`
	ToolCalls  []ToolCallInfo `json:"toolCalls,omitempty"`
}

type MessageContent struct {
	Type string `json:"type"`
	Text string `json:"text,omitempty"`
}

type MockClient struct {
	// contains filtered or unexported fields
}

type MockSession struct {

	// Test configuration
	SendPromptFunc func(ctx context.Context, text string, onUpdate func(*SessionUpdateParams)) (*SessionPromptResult, error)
	Updates        []*SessionUpdateParams // Updates to send during SendPrompt
	PromptResult   *SessionPromptResult   // Result to return from SendPrompt
	PromptError    error                  // Error to return from SendPrompt
	// contains filtered or unexported fields
}

type Process struct {
	// contains filtered or unexported fields
}

type ProcessConfig struct {
	Command string
	Args    []string
	Env     map[string]string
	Cwd     string
}

type PromptCapabilities struct {
	Image           bool `json:"image,omitempty"`
	Audio           bool `json:"audio,omitempty"`
	EmbeddedContext bool `json:"embeddedContext,omitempty"`
}

type PromptContent struct {
	Type     string          `json:"type"`
	Text     string          `json:"text,omitempty"`
	Resource *PromptResource `json:"resource,omitempty"`
}

type PromptResource struct {
	URI      string `json:"uri"`
	MimeType string `json:"mimeType,omitempty"`
	Text     string `json:"text,omitempty"`
}

type Session struct {
	// contains filtered or unexported fields
}

type SessionNewParams struct {
	Cwd        string      `json:"cwd"`
	MCPServers []MCPServer `json:"mcpServers"` // Required, can be empty array
}

type SessionNewResult struct {
	SessionID string `json:"sessionId"`
}

type SessionPromptParams struct {
	SessionID string          `json:"sessionId"`
	Prompt    []PromptContent `json:"prompt"`
}

type SessionPromptResult struct {
	StopReason string `json:"stopReason"`
}

type SessionUpdate struct {
	// Type indicates what kind of update this is.
	Type SessionUpdateType `json:"type"`

	// Content is the text content (for message chunks).
	Content string `json:"content,omitempty"`

	// ToolCall is set for tool_call updates.
	ToolCall *ToolCall `json:"toolCall,omitempty"`

	// ToolCallUpdate is set for tool_call_update updates.
	ToolCallUpdate *ToolCallUpdate `json:"toolCallUpdate,omitempty"`

	// StopReason is set when the agent stops.
	StopReason StopReason `json:"stopReason,omitempty"`
}

type SessionUpdateParams struct {
	SessionID string               `json:"sessionId"`
	Update    SessionUpdatePayload `json:"update"`
}

type SessionUpdatePayload struct {
	SessionUpdate string `json:"sessionUpdate"`

	// For agent_message_chunk - Content is *MessageContent
	// For tool_call_update - Content is []ToolCallContent (parsed via ContentArray)
	Content json.RawMessage `json:"content,omitempty"`

	// For tool_call
	ToolCallID string                 `json:"toolCallId,omitempty"`
	Name       string                 `json:"name,omitempty"`
	Title      string                 `json:"title,omitempty"`
	Kind       string                 `json:"kind,omitempty"`
	Arguments  map[string]interface{} `json:"arguments,omitempty"`

	// For tool_call_update
	Status    string      `json:"status,omitempty"`
	RawInput  interface{} `json:"rawInput,omitempty"`
	RawOutput interface{} `json:"rawOutput,omitempty"`
}

type SessionUpdateType string

type StopReason string

type TokenUsage struct {
	// PromptTokens is the number of tokens in the input/prompt.
	PromptTokens int `json:"promptTokens"`

	// CompletionTokens is the number of tokens in the output/completion.
	CompletionTokens int `json:"completionTokens"`

	// CachedTokens is the number of prompt tokens that were cache hits.
	CachedTokens int `json:"cachedTokens"`

	// TotalTokens is the sum of prompt and completion tokens.
	TotalTokens int `json:"totalTokens"`
}

type ToolCall struct {
	// ID is a unique identifier for this tool call within the session.
	// Maps to ACP's toolCallId.
	ID string `json:"id"`

	// Name is the name of the tool being invoked.
	Name string `json:"name"`

	// Title is a human-readable description of what the tool is doing.
	// Optional - if empty, clients should use the tool name.
	Title string `json:"title,omitempty"`

	// Kind is the category of tool being invoked.
	Kind ToolKind `json:"kind,omitempty"`

	// Arguments are the parameters passed to the tool.
	Arguments map[string]interface{} `json:"arguments,omitempty"`

	// Status is the current execution status.
	Status ToolCallStatus `json:"status"`
}

type ToolCallContent struct {
	Type    string          `json:"type"`
	Content *MessageContent `json:"content,omitempty"`
	// For diff type
	Path    string `json:"path,omitempty"`
	OldText string `json:"oldText,omitempty"`
	NewText string `json:"newText,omitempty"`
}

type ToolCallInfo struct {
	ID        string                 `json:"id"`
	Name      string                 `json:"name"`
	Arguments map[string]interface{} `json:"arguments,omitempty"`
}

type ToolCallStatus string

type ToolCallUpdate struct {
	// ID identifies which tool call this update is for.
	ID string `json:"id"`

	// Status is the new execution status, if changed.
	Status ToolCallStatus `json:"status,omitempty"`

	// Content is any output produced by the tool so far.
	Content string `json:"content,omitempty"`

	// Duration is how long the tool has been running.
	Duration time.Duration `json:"duration,omitempty"`

	// Error contains error information if the tool failed.
	Error error `json:"error,omitempty"`
}

type ToolKind string