config

type AIConfig struct {
	Provider       string `yaml:"provider" json:"provider"`               // default "openai"
	BaseURL        string `yaml:"base_url" json:"base_url"`               // default "https://api.openai.com/v1"
	APIKey         string `yaml:"api_key" json:"api_key"`                 // supports keyring:xxx reference
	AllowPlaintext bool   `yaml:"allow_plaintext" json:"allow_plaintext"` // allow plaintext API key
	Model          string `yaml:"model" json:"model"`                     // default "gpt-4o"
	MaxTokens      int    `yaml:"max_tokens" json:"max_tokens"`           // default 8192
}

type File struct {
	SSHProxies map[string]SSHProxy `yaml:"ssh_proxies" json:"ssh_proxies"`
	Profiles   map[string]Profile  `yaml:"profiles" json:"profiles"`
	MCP        MCPConfig           `yaml:"mcp" json:"mcp"`
	Web        WebConfig           `yaml:"web" json:"web"`
	Stats      stats.StatsConfig   `yaml:"stats" json:"stats"`
	AI         AIConfig            `yaml:"ai" json:"ai"`
}

type MCPConfig struct {
	Transport        string        `yaml:"transport" json:"transport"` // stdio | streamable_http
	HTTP             MCPHTTPConfig `yaml:"http" json:"http"`
	MaxRows          int           `yaml:"max_rows" json:"max_rows"`                     // default MCP query row cap
	MaxRowsLimit     int           `yaml:"max_rows_limit" json:"max_rows_limit"`         // hard MCP ceiling for max_rows
	MaxResponseBytes int           `yaml:"max_response_bytes" json:"max_response_bytes"` // MCP query JSON size cap
}

type MCPHTTPConfig struct {
	Addr                string `yaml:"addr" json:"addr"`
	AuthToken           string `yaml:"auth_token" json:"auth_token"`                       // supports keyring:xxx reference
	AllowPlaintextToken bool   `yaml:"allow_plaintext_token" json:"allow_plaintext_token"` // allow plaintext token
}

type Options struct {
	// ConfigPath: if non-empty, only this file is read (error if not found).
	ConfigPath string

	// CLI
	CLIProfile    string
	CLIProfileSet bool
	CLIFormat     string
	CLIFormatSet  bool

	// CLI AI
	CLIAIModel      string
	CLIAIModelSet   bool
	CLIAIBaseURL    string
	CLIAIBaseURLSet bool
	CLIAIAPIKey     string
	CLIAIAPIKeySet  bool

	// ENV (injected by caller for testability)
	EnvProfile   string
	EnvFormat    string
	EnvAIModel   string
	EnvAIBaseURL string
	EnvAIAPIKey  string

	// HomeDir is used for default path resolution (auto-detected if empty).
	HomeDir string

	// WorkDir is used for default paths (falls back to process cwd if empty).
	WorkDir string
}

type Profile struct {
	Description string `yaml:"description" json:"description"` // description to distinguish databases
	Format      string `yaml:"format" json:"format"`

	// DB connection
	DB       string `yaml:"db" json:"db"`   // mysql | pg
	DSN      string `yaml:"dsn" json:"dsn"` // raw DSN (takes precedence)
	Host     string `yaml:"host" json:"host"`
	Port     int    `yaml:"port" json:"port"`
	User     string `yaml:"user" json:"user"`
	Password string `yaml:"password" json:"password"` // supports keyring:xxx reference
	Database string `yaml:"database" json:"database"`

	// Security options
	AllowPlaintext   bool `yaml:"allow_plaintext" json:"allow_plaintext"`       // allow plaintext password
	UnsafeAllowWrite bool `yaml:"unsafe_allow_write" json:"unsafe_allow_write"` // permit write-capable entrypoints; CLI also requires its runtime flag

	// Timeout settings (seconds)
	QueryTimeout  int `yaml:"query_timeout" json:"query_timeout"`   // query timeout, default 30s
	SchemaTimeout int `yaml:"schema_timeout" json:"schema_timeout"` // schema export timeout, default 60s

	// MaxRows caps rows returned by query (0 = unset; use CLI/API 0 for unlimited)
	MaxRows int `yaml:"max_rows" json:"max_rows"`

	// SSH proxy reference (refers to a name defined in ssh_proxies)
	SSHProxy string `yaml:"ssh_proxy" json:"ssh_proxy"`

	// Local port for the proxy (used by xsql proxy command)
	LocalPort int `yaml:"local_port" json:"local_port"`

	// Resolved SSH config (populated by Resolve, not read from YAML)
	SSHConfig *SSHProxy `yaml:"-" json:"-"`
}

type ProfileInfo struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	DB          string `json:"db"`
	Mode        string `json:"mode"` // "read-only" or "read-write"
}

type Resolved struct {
	ConfigPath  string
	ProfileName string
	Format      string
	Profile     Profile            // full profile for query use
	AllProfiles map[string]Profile // all configured profiles
	AI          AIConfig
}

type SSHProxy struct {
	Host           string `yaml:"host" json:"host"`
	Port           int    `yaml:"port" json:"port"`
	User           string `yaml:"user" json:"user"`
	IdentityFile   string `yaml:"identity_file" json:"identity_file"`
	Passphrase     string `yaml:"passphrase" json:"passphrase"` // supports keyring:xxx reference
	KnownHostsFile string `yaml:"known_hosts_file" json:"known_hosts_file"`
	SkipHostKey    bool   `yaml:"skip_host_key" json:"skip_host_key"` // strongly discouraged
}

type WebConfig struct {
	HTTP WebHTTPConfig `yaml:"http" json:"http"`
}

type WebHTTPConfig struct {
	Addr                string `yaml:"addr" json:"addr"`
	AuthToken           string `yaml:"auth_token" json:"auth_token"`                       // supports keyring:xxx reference
	AllowPlaintextToken bool   `yaml:"allow_plaintext_token" json:"allow_plaintext_token"` // allow plaintext token
}