in pkg/rules/mcp/client_setup.go [96:184]
func handleClientRequest(method string, request *mcpRequest, message interface{}) error {
switch method {
case string(mcp.MethodToolsCall):
if msg, ok := message.(struct {
Name string `json:"name"`
Arguments map[string]interface{} `json:"arguments,omitempty"`
Meta *struct {
ProgressToken mcp.ProgressToken `json:"progressToken,omitempty"`
} `json:"_meta,omitempty"`
}); ok {
if request != nil {
request.operationName = "execute_tool"
request.methodName = msg.Name
}
}
return nil
case string(mcp.MethodPromptsGet):
if msg, ok := message.(struct {
// The name of the prompt or prompt template.
Name string `json:"name"`
// Arguments to use for templating the prompt.
Arguments map[string]string `json:"arguments,omitempty"`
}); ok {
if request != nil {
request.input["prompt_name"] = msg.Name
}
}
return nil
case string(mcp.MethodResourcesRead):
if msg, ok := message.(struct {
URI string `json:"uri"`
Arguments map[string]interface{} `json:"arguments,omitempty"`
}); ok {
if request != nil {
request.input["resources_uri"] = msg.URI
}
}
return nil
case string(mcp.MethodInitialize):
if msg, ok := message.(struct {
ProtocolVersion string `json:"protocolVersion"`
Capabilities mcp.ClientCapabilities `json:"capabilities"`
ClientInfo mcp.Implementation `json:"clientInfo"`
}); ok {
if request != nil {
request.input["client_info_name"] = msg.ClientInfo.Name
request.input["client_info_version"] = msg.ClientInfo.Version
}
}
return nil
case string(mcp.MethodResourcesList):
if msg, ok := message.(struct {
Cursor mcp.Cursor `json:"cursor,omitempty"`
}); ok {
if request != nil {
request.input["cursor"] = msg.Cursor
}
}
return nil
case string(mcp.MethodResourcesTemplatesList):
if msg, ok := message.(struct {
Cursor mcp.Cursor `json:"cursor,omitempty"`
}); ok {
if request != nil {
request.input["cursor"] = msg.Cursor
}
}
return nil
case string(mcp.MethodPromptsList):
if msg, ok := message.(struct {
Cursor mcp.Cursor `json:"cursor,omitempty"`
}); ok {
if request != nil {
request.input["cursor"] = msg.Cursor
}
}
return nil
case string(mcp.MethodToolsList):
if msg, ok := message.(struct {
Cursor mcp.Cursor `json:"cursor,omitempty"`
}); ok {
if request != nil {
request.input["cursor"] = msg.Cursor
}
}
return nil
}
return errors.New("client method not match")
}