5bd7ab4d2a<!-- Provide a brief summary of your changes --> Route `interceptors/list` and `interceptor/invoke` through the receiving middleware chain so that middleware installed (audit, etc.) fires for interceptor protocol methods the same way it does for standard MCP methods. ## Motivation and Context <!-- Why is this change needed? What problem does it solve? --> Before this change, custom JSON-RPC methods were dispatched via a separate code path that bypassed receiving middleware entirely. This meant any middleware an operator installed would silently not apply to the interceptor protocol endpoints. The fix adds `AddReceivingCustomMethod[P, R, T]` as a package-level generic function to the vendored go-sdk/mcp package. Custom methods registered through it are stored in a per-server customMethods map and merged into `receivingMethodInfos()`, so `defaultReceivingMethodHandler` dispatches them exactly like built-in methods. External types satisfy the sealed `Params`/`Result` interfaces by embedding the `ParamsBase`/`ResultBase` types exported from the mcp package. ## How Has This Been Tested? <!-- Have you tested this in a real application? Which scenarios were tested? --> Existing test suites and new ones for verifying middleware paths. ## Breaking Changes <!-- Will users need to update their code or configurations? --> no. ## Types of changes <!-- What types of changes does your code introduce? Put an `x` in all the boxes that apply: --> - [x] Bug fix (non-breaking change which fixes an issue) - [ ] New feature (non-breaking change which adds functionality) - [ ] Breaking change (fix or feature that would cause existing functionality to change) - [ ] Documentation update ## Checklist <!-- Go over all the following points, and put an `x` in all the boxes that apply. --> - [x] I have read the [MCP Documentation](https://modelcontextprotocol.io) - [x] My code follows the repository's style guidelines - [x] New and existing tests pass locally - [x] I have added appropriate error handling - [x] I have added or updated documentation as needed ## Additional context <!-- Add any other context, implementation notes, or design decisions --> The changes to `vendor/go-sdk/mcp/server.go` are intentionally minimal and carry a `NOTE` comment acknowledging they are additions to the upstream SDK needed for this reference implementation. --------- Signed-off-by: Ukjae Jeong <jeongukjae@gmail.com>
106 lines
3.4 KiB
Go
106 lines
3.4 KiB
Go
// Copyright 2025 The MCP Interceptors Authors. All rights reserved.
|
|
// Use of this source code is governed by an Apache-2.0
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package interceptors
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
)
|
|
|
|
// JSON-RPC method names for the interceptor protocol.
|
|
const (
|
|
MethodList = "interceptors/list"
|
|
MethodInvoke = "interceptor/invoke"
|
|
)
|
|
|
|
// Event name constants for standard MCP methods.
|
|
const (
|
|
// Server Features
|
|
EventToolsList = "tools/list"
|
|
EventToolsCall = "tools/call"
|
|
EventPromptsList = "prompts/list"
|
|
EventPromptsGet = "prompts/get"
|
|
EventResourcesList = "resources/list"
|
|
EventResourcesRead = "resources/read"
|
|
EventResourcesSubscribe = "resources/subscribe"
|
|
)
|
|
|
|
// --- interceptors/list ---
|
|
|
|
// ListParams are the optional parameters for the interceptors/list method.
|
|
type ListParams struct {
|
|
mcp.ParamsBase
|
|
Event string `json:"event,omitempty"` // Filter by event name; empty = all
|
|
}
|
|
|
|
// InterceptorInfo describes a registered interceptor in wire format.
|
|
type InterceptorInfo struct {
|
|
Name string `json:"name"`
|
|
Version string `json:"version,omitempty"`
|
|
Description string `json:"description,omitempty"`
|
|
Type InterceptorType `json:"type"`
|
|
Hook Hook `json:"hook"`
|
|
PriorityHint Priority `json:"priorityHint,omitempty"`
|
|
Compat *Compat `json:"compat,omitempty"`
|
|
ConfigSchema json.RawMessage `json:"configSchema,omitempty"`
|
|
Mode Mode `json:"mode"`
|
|
FailOpen bool `json:"failOpen,omitempty"`
|
|
}
|
|
|
|
// InfoFromInterceptor builds an InterceptorInfo from an Interceptor.
|
|
func InfoFromInterceptor(i Interceptor) InterceptorInfo {
|
|
m := i.GetMetadata()
|
|
return InterceptorInfo{
|
|
Name: m.Name,
|
|
Version: m.Version,
|
|
Description: m.Description,
|
|
Type: i.GetType(),
|
|
Hook: m.Hook,
|
|
PriorityHint: m.PriorityHint,
|
|
Compat: m.Compat,
|
|
ConfigSchema: m.ConfigSchema,
|
|
Mode: m.Mode,
|
|
FailOpen: m.FailOpen,
|
|
}
|
|
}
|
|
|
|
// ListResult is the result of the interceptors/list method.
|
|
type ListResult struct {
|
|
mcp.ResultBase
|
|
Interceptors []InterceptorInfo `json:"interceptors"`
|
|
}
|
|
|
|
// --- interceptor/invoke ---
|
|
|
|
// InvokeParams are the parameters for the interceptor/invoke method.
|
|
// The caller specifies which interceptor to invoke by name.
|
|
type InvokeParams struct {
|
|
mcp.ParamsBase
|
|
Name string `json:"name"`
|
|
Event string `json:"event"`
|
|
Phase InterceptionPhase `json:"phase"`
|
|
Payload json.RawMessage `json:"payload"`
|
|
TimeoutMs int64 `json:"timeoutMs,omitempty"`
|
|
Config map[string]any `json:"config,omitempty"`
|
|
Context *InvocationContext `json:"context,omitempty"`
|
|
}
|
|
|
|
// InvokeResult is the result of the interceptor/invoke method.
|
|
// Fields are type-specific: validators populate Validation, mutators populate Mutation + Payload.
|
|
type InvokeResult struct {
|
|
mcp.ResultBase
|
|
Interceptor string `json:"interceptor"`
|
|
Type InterceptorType `json:"type"`
|
|
Phase InterceptionPhase `json:"phase"`
|
|
DurationMs int64 `json:"durationMs"`
|
|
|
|
// Validation result (type == "validation")
|
|
Validation *ValidationResult `json:"validation,omitempty"`
|
|
|
|
// Mutation result (type == "mutation")
|
|
Mutation *MutationResult `json:"mutation,omitempty"`
|
|
Payload json.RawMessage `json:"payload,omitempty"` // Mutated payload (only for mutators)
|
|
}
|