mcp

type Config struct {
	ConfigPath string `json:"config_path,omitempty"`
	LocalOnly  bool   `json:"local_only"`
	Source     string `json:"source"`
	MaxResults int    `json:"max_results"`
}

type ConfigOption func(*Config)

type InitializeResult struct {
	ProtocolVersion string     `json:"protocolVersion"`
	ServerInfo      ServerInfo `json:"serverInfo"`
	Capabilities    ServerCaps `json:"capabilities"`
}

type ListConversationsInput struct {
	AccountIDs   []string `json:"account_ids,omitempty"`
	Limit        int      `json:"limit,omitempty"`
	Offset       int      `json:"offset,omitempty"`
	UnreadOnly   bool     `json:"unread_only,omitempty"`
	IncludeMuted bool     `json:"include_muted,omitempty"`
	MutedOnly    bool     `json:"muted_only,omitempty"`
	PinnedOnly   bool     `json:"pinned_only,omitempty"`
	ArchivedOnly bool     `json:"archived_only,omitempty"`
	Type         string   `json:"type,omitempty"`
}

type ListConversationsOutput struct {
	Conversations []api.Chat `json:"conversations"`
	Total         int        `json:"total"`
	HasMore       bool       `json:"has_more,omitempty"`
}

type ReadMessagesInput struct {
	ChatID string `json:"chat_id"`
	Limit  int    `json:"limit,omitempty"`
}

type ReadMessagesOutput struct {
	ChatID     string        `json:"chat_id"`
	ThreadName string        `json:"thread_name,omitempty"`
	Messages   []api.Message `json:"messages"`
	HasMore    bool          `json:"has_more,omitempty"`
	NextCursor string        `json:"next_cursor,omitempty"`
}

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

type Response struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      json.RawMessage `json:"id,omitempty"`
	Result  any             `json:"result,omitempty"`
	Error   *ResponseError  `json:"error,omitempty"`
}

type ResponseError struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
}

type SearchMessagesInput struct {
	Query      string   `json:"query"`
	ChatIDs    []string `json:"chat_ids,omitempty"`
	AccountIDs []string `json:"account_ids,omitempty"`
	Sender     string   `json:"sender,omitempty"`
	ChatType   string   `json:"chat_type,omitempty"`
	Limit      int      `json:"limit,omitempty"`
}

type SearchMessagesOutput struct {
	Query    string              `json:"query"`
	Messages []api.Message       `json:"messages"`
	Chats    map[string]api.Chat `json:"chats,omitempty"`
	HasMore  bool                `json:"has_more,omitempty"`
	Cursor   string              `json:"cursor,omitempty"`
}

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

type ServerCaps struct {
	Tools ToolsCapability `json:"tools"`
}

type ServerInfo struct {
	Name    string `json:"name"`
	Version string `json:"version"`
}

type ServerOption func(*Server)

type Tool struct {
	Name        string         `json:"name"`
	Description string         `json:"description,omitempty"`
	InputSchema map[string]any `json:"inputSchema,omitempty"`
}

type ToolCallParams struct {
	Name      string          `json:"name"`
	Arguments json.RawMessage `json:"arguments,omitempty"`
}

type ToolCallResult struct {
	Content []ToolContent `json:"content,omitempty"`
	IsError bool          `json:"isError,omitempty"`
}

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

type ToolRuntime interface {
	ListConversations(context.Context, ListConversationsInput) (ListConversationsOutput, error)
	ReadMessages(context.Context, ReadMessagesInput) (ReadMessagesOutput, error)
	SearchMessages(context.Context, SearchMessagesInput) (SearchMessagesOutput, error)
}

type ToolsCapability struct {
	ListChanged bool `json:"listChanged"`
}

type ToolsListResult struct {
	Tools []Tool `json:"tools"`
}