mcp

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

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

type MockContent struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

type MockResource struct {
	URI         string `json:"uri"`
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	MimeType    string `json:"mimeType,omitempty"`
}

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

type MockTool struct {
	Name        string          `json:"name"`
	Description string          `json:"description"`
	InputSchema json.RawMessage `json:"inputSchema"`
}

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

type MockTransport struct {
	Tools            []MockTool
	Resources        []MockResource
	ResourceContents []MockResourceContent
	ToolCallResults  map[string]MockToolResult // tool name -> custom result

	// FailOnInvalidToolParams makes tools/call return an error for any call.
	FailOnInvalidToolParams bool

	// InitializeResult overrides the initialize response.
	InitializeResult json.RawMessage

	// SendError makes Connect return an error.
	SendError bool
	// contains filtered or unexported fields
}

type Resource struct {
	// URI is the unique identifier for the resource.
	URI string

	// Name is the human-readable name.
	Name string

	// Description describes the resource.
	Description string

	// MimeType is the content type.
	MimeType string
}

type ResourceContent struct {
	// URI is the resource URI.
	URI string

	// MimeType is the content type.
	MimeType string

	// Text is the text content (for text resources).
	Text string

	// Blob is the binary content (for binary resources, base64 encoded).
	Blob string
}

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

type ServerConfig struct {
	// Name is a unique identifier for this server.
	Name string

	// Transport specifies how to connect to the server.
	// Options: "stdio", "sse", "http"
	Transport string

	// Command is the command to run for stdio transport.
	Command string

	// Args are arguments for the command.
	Args []string

	// Env are environment variables for the command.
	Env map[string]string

	// URL is the server URL for sse/http transport.
	URL string

	// Headers are HTTP headers for sse/http transport.
	Headers map[string]string
}

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

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

type Transport interface {
	// Connect establishes the connection.
	Connect(ctx context.Context) error

	// Close closes the connection.
	Close() error

	// Send sends a request and returns the response.
	Send(ctx context.Context, method string, params any) (json.RawMessage, error)

	// OnNotification registers a handler for server notifications.
	OnNotification(handler func(method string, params json.RawMessage))
}