func handleClientRequest()

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")
}