mcpserver

type DataOutput struct {
	Data any `json:"data"`
}

type Doer interface {
	Do(req *http.Request) (*http.Response, error)
}

type FaultError struct {
	Code       string `json:"code"       jsonschema:"AWS error code, e.g. ThrottlingException"`
	StatusCode int    `json:"statusCode" jsonschema:"HTTP status code, e.g. 429"`
}

type FaultRule struct {
	Error       *FaultError `json:"error,omitempty"       jsonschema:"injected error (default 503 ServiceUnavailable)"`
	Probability *float64    `json:"probability,omitempty" jsonschema:"firing probability 0..1 (default 1)"`
	Service     string      `json:"service,omitempty"     jsonschema:"service name to match, e.g. dynamodb"`
	Region      string      `json:"region,omitempty"      jsonschema:"region to match"`
	Operation   string      `json:"operation,omitempty"   jsonschema:"operation name to match, e.g. PutItem"`
}

type FaultsClearInput struct{}

type FaultsListInput struct{}

type FaultsOutput struct {
	Rules []FaultRule `json:"rules"`
}

type FaultsSetInput struct {
	Rules  []FaultRule `json:"rules"            jsonschema:"fault rules; evaluated in order, first match wins"`
	Append bool        `json:"append,omitempty" jsonschema:"append to existing rules instead of replacing them"`
}

type HealthInput struct{}

type HealthOutput struct {
	Status        string   `json:"status"`
	Version       string   `json:"version"`
	Services      []string `json:"services"`
	Disabled      []string `json:"disabled,omitempty"`
	UptimeSeconds int64    `json:"uptime_seconds"`
	Goroutines    int      `json:"goroutines"`
}

type IAMDenialsInput struct {
	Service   string `json:"service,omitempty"   jsonschema:"only this service"`
	Principal string `json:"principal,omitempty" jsonschema:"only this principal ARN"`
	Since     string `json:"since,omitempty"     jsonschema:"RFC3339 time or a duration such as 5m"`
	Limit     int    `json:"limit,omitempty"     jsonschema:"max events, 1-500 (default 50)"`
}

type IAMDenialsOutput struct {
	Events  []map[string]any `json:"events"`
	Dropped int64            `json:"dropped"`
}

type InitHooksInput struct{}

type InitHooksRunInput struct {
	Stage  string `json:"stage"            jsonschema:"BOOT, START, READY or SHUTDOWN"`
	Script string `json:"script,omitempty" jsonschema:"script file name from init_hooks_list (default: whole stage)"`
}

type Options struct {
	Doer    Doer
	BaseURL string
	Version string
	// ReadOnly omits every tool that can change state.
	ReadOnly bool
}

type ResetInput struct {
	Service string `json:"service,omitempty" jsonschema:"only reset this service (default: every service)"`
}

type SnapshotDeleteInput struct {
	Name    string `json:"name"              jsonschema:"snapshot name"`
	Version string `json:"version,omitempty" jsonschema:"version to delete (default: all versions)"`
}

type SnapshotListInput struct{}

type SnapshotLoadInput struct {
	Name     string   `json:"name"               jsonschema:"snapshot name"`
	Version  string   `json:"version,omitempty"  jsonschema:"version number or 'latest' (default latest)"`
	Strategy string   `json:"strategy,omitempty" jsonschema:"overwrite (default) or service-merge"`
	Services []string `json:"services,omitempty" jsonschema:"only load these services (default: all in the snapshot)"`
	Merge    bool     `json:"merge,omitempty"    jsonschema:"merge into existing state instead of replacing it"`
}

type SnapshotSaveInput struct {
	Name     string   `json:"name"               jsonschema:"snapshot name (a new version is created if it exists)"`
	Message  string   `json:"message,omitempty"  jsonschema:"optional version message"`
	Services []string `json:"services,omitempty" jsonschema:"only save these services (default: all)"`
}

type StatusOutput struct {
	Status  string `json:"status"`
	Message string `json:"message,omitempty"`
}

type Trace struct {
	Timestamp    time.Time `json:"timestamp"`
	ID           string    `json:"id"`
	Method       string    `json:"method"`
	Path         string    `json:"path"`
	Query        string    `json:"query,omitempty"`
	Service      string    `json:"service,omitempty"`
	Operation    string    `json:"operation,omitempty"`
	ErrorCode    string    `json:"error_code,omitempty"`
	ErrorMessage string    `json:"error_message,omitempty"`
	Status       int       `json:"status"`
	DurationMS   int64     `json:"duration_ms"`
}

type TracesInput struct {
	Service   string `json:"service,omitempty"   jsonschema:"only this service, e.g. s3"`
	Operation string `json:"operation,omitempty" jsonschema:"only this operation name, e.g. PutObject (case-insensitive)"`
	Status    string `json:"status,omitempty"    jsonschema:"status code like 500, class like 4xx, or 'failed' (>=400)"`
	Limit     int    `json:"limit,omitempty"     jsonschema:"max traces to return, 1-100 (default 20)"`
}

type TracesOutput struct {
	Traces []Trace `json:"traces"`
}