common

type AuthContext struct {
	Authenticated         bool   `json:"authenticated"`
	PrincipalID           string `json:"principal_id,omitempty"`
	PrincipalType         string `json:"principal_type,omitempty"` // e.g. "api_key"
	KeyID                 string `json:"key_id,omitempty"`
	Gateway               string `json:"gateway,omitempty"`
	AuthHeader            string `json:"auth_header,omitempty"`
	CredentialFingerprint string `json:"credential_fingerprint,omitempty"`
	InternalSessionID     string `json:"internal_session_id,omitempty"`
	TransportSessionID    string `json:"transport_session_id,omitempty"`
}

type BaseMcpEvent struct {

	// Indicates the event status - is related to http status codes:
	// 20x - ok
	// 40x - expected error - client might be able to resolve it by rephrasing the query
	// 50x - unexpected error - proxy ran into unexpected error.
	Status int `json:"status"`

	// Timestamp is the exact time when the log entry was created.
	Timestamp time.Time `json:"timestamp"`

	// Transport identifies the proxy type: "stdio", "http", "websocket".
	Transport string `json:"transport"`

	// RequestID uniquely identifies a single request/response pair.
	RequestID string `json:"request_id"`

	// SessionID groups multiple requests within the same proxy session.
	SessionID string `json:"session_id,omitempty"`

	// ServerID uniquely identifies the MCP server instance handling this request.
	ServerID string `json:"server_id,omitempty"`

	// Direction indicates the communication flow perspective:
	// "request" (client→server),
	// "response" (server→client), or
	// "system" (proxy lifecycle events).
	// This field remains stable regardless of success/failure status.
	Direction McpEventDirection `json:"direction"`

	// MessageType categorizes the content/outcome: "request", "response", "error", or "system".
	// Unlike Direction, this changes to "error" for failed responses, enabling filtering.
	// by operational status (e.g., "all errors" vs "all responses regardless of success").
	// This orthogonal design supports both flow analysis (Direction) and status monitoring (MessageType).
	MessageType McpMessageType `json:"message_type"`

	// Success indicates whether the operation completed successfully.
	Success bool `json:"success"`

	// Error contains error details if Success is false.
	Error string `json:"error,omitempty"`

	// ProcessingErrors indicate errors during processing of this event.
	ProcessingErrors map[string]error `json:"-"`

	// Metadata holds additional context-specific key-value pairs.
	Metadata map[string]string `json:"metadata,omitempty"`

	// Modified indicates that the event has been modified at least once since being received.
	Modified bool `json:"modified"`
}

type EventAnnotation struct {
	Type      string                   `json:"type,omitempty"`
	Processor string                   `json:"processor,omitempty"`
	Action    string                   `json:"action,omitempty"`
	Category  string                   `json:"category,omitempty"`
	Severity  string                   `json:"severity,omitempty"`
	Message   string                   `json:"message,omitempty"`
	Findings  []EventAnnotationFinding `json:"findings,omitempty"`
	Details   map[string]any           `json:"details,omitempty"`
}

type EventAnnotationFinding struct {
	Rule string `json:"rule,omitempty"`
	Path string `json:"path,omitempty"`
}

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

type LogEntry struct {
	BaseMcpEvent
	Routing     RoutingContext    `json:"routing"`
	ToolCall    *ToolCallLog      `json:"tool_call,omitempty"`
	Annotations []EventAnnotation `json:"annotations,omitempty"`
}

type LogLevel int

type LoggerOptions struct {
	Level         string
	Output        string
	FilePath      string
	ConsoleWriter io.Writer
}

type McpEventDirection string

type McpMessageType string

type McpTransportType string

type MetaContext struct {
	BaseMcpEvent
	Annotations []EventAnnotation `json:"-"`
}

type PromptDefinition struct {
	Prompt string `yaml:"prompt"`
}

type RoutingContext struct {
	// Transport describes the used transport for this connection (http or stdio)
	Transport McpTransportType `json:"transport,omitempty"`

	// Gateway is the logical grouping of MCP servers
	Gateway string `json:"gateway,omitempty"`

	// ServerName identifies the specific MCP server
	ServerName string `json:"server_name,omitempty"`

	// Endpoint is the HTTP path or identifier for this proxy
	Endpoint string `json:"endpoint,omitempty"`

	// DownstreamURL is the target MCP server URL being proxied to
	DownstreamURL string `json:"downstream_url,omitempty"`

	// DownstreamCommand is the target MCP server command being proxied to
	DownstreamCommand string `json:"downstream_cmd,omitempty"`

	// Args is the target MCP server command args being used
	Args []string `json:"args,omitempty"`
}

type ToolCallLog struct {
	// Name is the tool name being called
	Name string `json:"name"`

	// OriginalName is the tool name before any namespace transformations
	OriginalName string `json:"original_name,omitempty"`

	// Arguments contains the tool call arguments as raw JSON
	Arguments json.RawMessage `json:"arguments,omitempty"`

	// Result contains the tool call result as raw JSON (for responses)
	Result json.RawMessage `json:"result,omitempty"`

	// IsError indicates if the tool call resulted in an error
	IsError bool `json:"is_error,omitempty"`
}