mcp

type AnalyticsConfig struct {
	// Enabled controls whether analytics are active. Default: true
	Enabled bool
	// AppVersion is the CLI version for tracking.
	AppVersion string
}

type ColumnInfo struct {
	Name             string `json:"name"`
	Type             string `json:"type"`
	IsPrimary        bool   `json:"is_primary"`
	IsForeignKey     bool   `json:"is_foreign_key"`
	ReferencedTable  string `json:"referenced_table,omitempty"`
	ReferencedColumn string `json:"referenced_column,omitempty"`
}

type ColumnsInput struct {
	// Connection is the name of a saved connection or environment profile.
	Connection string `json:"connection" jsonschema:"Connection name (optional if only one exists)"`
	// Schema containing the table
	Schema string `json:"schema,omitempty" jsonschema:"Schema name (uses default if omitted)"`
	// Table name to describe
	Table string `json:"table" jsonschema:"Table name to describe"`
}

type ColumnsOutput struct {
	Columns   []ColumnInfo `json:"columns"`
	Table     string       `json:"table"`
	Schema    string       `json:"schema"`
	Error     string       `json:"error,omitempty"`
	RequestID string       `json:"request_id,omitempty"` // Unique ID for request tracing
}

type ConfirmInput struct {
	// Token is the confirmation token from a previous query response
	Token string `json:"token" jsonschema:"Confirmation token from a previous whodb_query response"`
}

type ConfirmOutput struct {
	Columns     []string `json:"columns"`
	ColumnTypes []string `json:"column_types,omitempty"`
	Rows        [][]any  `json:"rows"`
	Error       string   `json:"error,omitempty"`
	Message     string   `json:"message,omitempty"`
	RequestID   string   `json:"request_id,omitempty"` // Unique ID for request tracing
}

type ConnectionInfo struct {
	Name     string `json:"name"`
	Type     string `json:"type"`
	Host     string `json:"host,omitempty"`
	Port     int    `json:"port,omitempty"`
	Database string `json:"database,omitempty"`
	Schema   string `json:"schema,omitempty"`
	Source   string `json:"source"` // "env" or "saved"
}

type ConnectionsInput struct{}

type ConnectionsOutput struct {
	Connections []ConnectionInfo `json:"connections"`
	Error       string           `json:"error,omitempty"`
	RequestID   string           `json:"request_id,omitempty"` // Unique ID for request tracing
}

type HTTPOptions struct {
	// Host to bind to (default: "localhost").
	Host string
	// Port to listen on (default: 3000).
	Port int
}

type PendingConfirmation struct {
	Token      string
	Query      string
	Connection string
	ExpiresAt  time.Time
}

type PendingInfo struct {
	Token      string `json:"token"`
	Query      string `json:"query"`
	Connection string `json:"connection"`
	ExpiresAt  string `json:"expires_at"` // ISO 8601
}

type PendingInput struct{}

type PendingOutput struct {
	Pending   []PendingInfo `json:"pending"`
	Error     string        `json:"error,omitempty"`
	RequestID string        `json:"request_id,omitempty"`
}

type QueryInput struct {
	// Connection is the name of a saved connection or environment profile.
	Connection string `json:"connection" jsonschema:"Connection name (optional if only one exists)"`
	// Query is the SQL query to execute
	Query string `json:"query" jsonschema:"SQL query to execute"`
	// Parameters for parameterized queries (optional).
	// Use placeholders in the query ($1, $2 for Postgres; ? for MySQL/SQLite).
	// Example: query="SELECT * FROM users WHERE id = $1", parameters=[42]
	Parameters []any `json:"parameters,omitempty" jsonschema:"Parameterized query values ($1/$2 for Postgres or ? for MySQL/SQLite)"`
}

type QueryOutput struct {
	Columns              []string `json:"columns"`
	ColumnTypes          []string `json:"column_types,omitempty"`
	Rows                 [][]any  `json:"rows"`
	Error                string   `json:"error,omitempty"`
	Warning              string   `json:"warning,omitempty"`
	ConfirmationRequired bool     `json:"confirmation_required,omitempty"`
	ConfirmationToken    string   `json:"confirmation_token,omitempty"`
	ConfirmationQuery    string   `json:"confirmation_query,omitempty"`
	ConfirmationExpiry   string   `json:"confirmation_expiry,omitempty"` // ISO 8601 timestamp when the token expires
	RequestID            string   `json:"request_id,omitempty"`          // Unique ID for request tracing
}

type SchemaDetail struct {
	Name   string      `json:"name"`
	Tables []TableInfo `json:"tables,omitempty"`
}

type SchemasInput struct {
	// Connection is the name of a saved connection or environment profile.
	Connection string `json:"connection" jsonschema:"Connection name (optional if only one exists)"`
	// IncludeTables returns the tables within each schema in a single call.
	// Reduces round-trips when you need both schemas and tables.
	IncludeTables bool `json:"include_tables,omitempty" jsonschema:"Set true to also return tables within each schema in a single call"`
}

type SchemasOutput struct {
	Schemas   []string       `json:"schemas"`
	Details   []SchemaDetail `json:"details,omitempty"` // Populated when include_tables=true
	Error     string         `json:"error,omitempty"`
	RequestID string         `json:"request_id,omitempty"` // Unique ID for request tracing
}

type SecurityLevel string

type SecurityOptions struct {
	ReadOnly            bool
	ConfirmWrites       bool
	SecurityLevel       SecurityLevel
	QueryTimeout        time.Duration
	MaxRows             int
	AllowMultiStatement bool
	AllowDrop           bool
	DefaultConnection   string   // Injected connection when not specified
	AllowedConnections  []string // If set, only these connections are accessible
}

type ServerOptions struct {
	// Logger for server messages (defaults to stderr).
	Logger *slog.Logger
	// Instructions provides guidance to LLMs on how to use this server.
	Instructions string
	// ReadOnly prevents INSERT, UPDATE, DELETE, DROP, CREATE, ALTER, TRUNCATE operations.
	// Default: true
	ReadOnly bool
	// ConfirmWrites enables human-in-the-loop confirmation for write operations.
	// When enabled, write operations return a confirmation token that must be approved.
	// Default: false
	ConfirmWrites bool
	// SecurityLevel controls the strictness of query validation.
	// Options: "strict", "standard", "minimal". Default: "standard"
	SecurityLevel SecurityLevel
	// QueryTimeout is the maximum time a query can run before being cancelled.
	// Default: 30 seconds
	QueryTimeout time.Duration
	// MaxRows limits the number of rows returned by queries.
	// Default: 0 (unlimited). Set via --max-rows to enable truncation.
	MaxRows int
	// AllowMultiStatement permits multiple SQL statements in one query (separated by semicolons).
	// WARNING: Enabling this increases SQL injection risk.
	// Default: false
	AllowMultiStatement bool
	// AllowDrop permits DROP/TRUNCATE operations even in allow-write mode.
	// Without this, DROP is blocked unless --confirm-writes is used
	// Default: false
	AllowDrop bool
	// EnabledTools specifies which tools to enable. If empty, all tools are enabled.
	// Valid values: "query", "schemas", "tables", "columns", "connections", "confirm"
	EnabledTools []string
	// DisabledTools specifies which tools to disable. Takes precedence over EnabledTools.
	// Valid values: "query", "schemas", "tables", "columns", "connections", "confirm"
	DisabledTools []string
	// DefaultConnection is the connection to use when none is specified.
	// This simplifies AI interaction when working with a single database.
	DefaultConnection string
	// AllowedConnections restricts which connections can be used.
	// When set, only these connections are visible and accessible.
	// If DefaultConnection is not set, the first allowed connection becomes the default.
	AllowedConnections []string
}

type StatementType string

type TableInfo struct {
	Name       string            `json:"name"`
	Attributes map[string]string `json:"attributes,omitempty"`
	Columns    []ColumnInfo      `json:"columns,omitempty"` // Populated when include_columns=true
}

type TablesInput struct {
	// Connection is the name of a saved connection or environment profile.
	Connection string `json:"connection" jsonschema:"Connection name (optional if only one exists)"`
	// Schema to list tables from (uses default if not specified)
	Schema string `json:"schema,omitempty" jsonschema:"Schema name (uses default if omitted)"`
	// IncludeColumns returns column details for each table in a single call.
	// Reduces round-trips when you need both tables and their columns.
	IncludeColumns bool `json:"include_columns,omitempty" jsonschema:"Set true to also return column details for each table in a single call"`
}

type TablesOutput struct {
	Tables    []TableInfo `json:"tables"`
	Schema    string      `json:"schema"`
	Error     string      `json:"error,omitempty"`
	RequestID string      `json:"request_id,omitempty"` // Unique ID for request tracing
}

type ToolEnablement struct {
	EnabledTools  []string
	DisabledTools []string
}

type TransportType string