mcp

type AuthSpec struct {
	GoogleOAuth *GoogleOAuthAuth `json:"google_oauth,omitempty"`
}

type DigestOptions struct {
	// Store is the CCR backing for retrieve_raw. When nil, digest
	// still runs but retrieve_raw returns "no raw payload" — matches
	// digest-design.md OQ1 default when --session-db is off.
	Store digest.Store

	// Threshold is the byte size below which responses bypass the
	// router entirely. Zero → DefaultAgenticWrapThreshold.
	Threshold int

	// NeverServers names MCP servers (by mcp.json key) that opt out
	// of digesting. Operator escape hatch for debug-sensitive or
	// known-tiny servers.
	NeverServers map[string]bool

	// LLMFallback, when non-nil, invokes a small-tier subagent to
	// digest MCP responses the structural JSON pruner can't reduce
	// (prose, malformed JSON, or JSON that's structurally minimal
	// and mostly-values). Callers pass a closure that owns a
	// reference to an Agent + a resolved small-model; the wrapper
	// invokes it with the raw payload and gets back a compressed
	// digest plus subagent usage numbers.
	//
	// Returned SubagentModel + token counts populate
	// digest.Savings.Subagent* on the resulting Result so /stats,
	// per-tool footer, and OTel span attributes have real cost
	// figures for the agentic path.
	//
	// A nil LLMFallback preserves the #128-shipped structural-only
	// behavior (bounded passthrough for anything structural can't
	// reduce). No opt-in from operators required to keep that.
	LLMFallback func(ctx context.Context, raw []byte) (LLMFallbackResult, error)

	// OnResult, when non-nil, fires after every successful Process
	// call with the (fully-decorated) Result. Callers use this to
	// aggregate per-call Savings into session-level counters — the
	// usage.Tracker sink in cmd/core-agent wires this to a cumulative
	// digest-savings block rendered by /context.
	//
	// Firing is best-effort: skipped on Process errors / marshal
	// failures where Result is undefined. Runs synchronously on the
	// wrapper's Run goroutine, so callers should keep the callback
	// fast (increment counters, don't do I/O).
	OnResult func(*digest.Result)
}

type ElicitorFn func(ctx context.Context, serverName string, req *mcp.ElicitRequest) (*mcp.ElicitResult, error)

type GoogleOAuthAuth struct {
	// Scopes is the OAuth 2.0 scopes requested on the access token.
	// No default — each server documents its own scope requirements,
	// and an implicit broad default (e.g. cloud-platform) would grant
	// more privilege than necessary. Explicit is safer.
	//
	// For the GKE MCP server, typical values:
	//   https://www.googleapis.com/auth/container.read-only
	//   https://www.googleapis.com/auth/container
	Scopes []string `json:"scopes"`
}

type LLMFallbackResult struct {
	Text                 string
	SubagentModel        string
	SubagentInputTokens  int
	SubagentOutputTokens int
}

type Server struct {
	Name      string
	Status    string
	Tools     []string   // tool names exposed; populated lazily by Toolset
	ToolInfos []ToolInfo // name + description pairs, parallel to Tools
	Err       error      // non-nil when Status == StatusError
	// contains filtered or unexported fields
}

type ServerSpec struct {
	Transport    string            `json:"transport"`               // "stdio" | "http"
	Command      string            `json:"command,omitempty"`       // stdio
	Args         []string          `json:"args,omitempty"`          // stdio
	Env          map[string]string `json:"env,omitempty"`           // stdio
	URL          string            `json:"url,omitempty"`           // http
	Headers      map[string]string `json:"headers,omitempty"`       // http
	Auth         *AuthSpec         `json:"auth,omitempty"`          // http
	AgenticNever bool              `json:"agentic_never,omitempty"` // skip digest wrap for this server
}

type Servers struct {
	Version              int                   `json:"version"`
	Servers              map[string]ServerSpec `json:"servers"`
	AgenticWrap          *bool                 `json:"agentic_wrap,omitempty"`           // default true; ptr for explicit off
	AgenticWrapThreshold int                   `json:"agentic_wrap_threshold,omitempty"` // default 8000
	AgenticWrapLLM       *bool                 `json:"agentic_wrap_llm,omitempty"`       // default false; ptr for explicit on
	AgenticWrapModel     string                `json:"agentic_wrap_model,omitempty"`     // MCP-specific small-model override; empty falls through to --agentic-small-model resolution
}

type ToolInfo struct {
	Name        string
	Description string
}