config

type Config struct {
	Server      ServerConfig       `mapstructure:"server"`
	Logging     LoggingConfig      `mapstructure:"logging"`
	Sources     []SourceConfig     `mapstructure:"sources"`
	CustomTools []CustomToolConfig `mapstructure:"custom_tools"`
	Security    SecurityConfig     `mapstructure:"security"`
}

type CustomToolConfig struct {
	Name        string          `mapstructure:"name"`        // Tool name (e.g., "get_daily_sales")
	Description string          `mapstructure:"description"` // Tool description for AI
	Source      string          `mapstructure:"source"`      // Which source to run against
	Query       string          `mapstructure:"query"`       // SQL query template
	Parameters  []ToolParameter `mapstructure:"parameters"`  // Optional parameters
}

type LoggingConfig struct {
	Level  string `mapstructure:"level"`
	Format string `mapstructure:"format"`
}

type PIIMaskPattern struct {
	Name        string `mapstructure:"name"`        // Pattern name (e.g., "credit_card")
	Pattern     string `mapstructure:"pattern"`     // Regex pattern
	Replacement string `mapstructure:"replacement"` // Replacement string (default: "***")
	Enabled     bool   `mapstructure:"enabled"`     // Whether this pattern is active
}

type SecurityConfig struct {
	MaxRowLimit      int              `mapstructure:"max_row_limit"`      // Maximum rows to return (default: 1000)
	EnablePIIMasking bool             `mapstructure:"enable_pii_masking"` // Enable PII data masking
	PIIPatterns      []PIIMaskPattern `mapstructure:"pii_patterns"`       // PII patterns to mask
	AllowedKeywords  []string         `mapstructure:"allowed_keywords"`   // SQL keywords to allow beyond SELECT/WITH
	BlockedKeywords  []string         `mapstructure:"blocked_keywords"`   // Additional SQL keywords to block
}

type ServerConfig struct {
	Name      string `mapstructure:"name"`
	Version   string `mapstructure:"version"`
	Transport string `mapstructure:"transport"`
	Port      int    `mapstructure:"port"`
}

type SourceConfig struct {
	Name     string `mapstructure:"name"`
	Type     string `mapstructure:"type"` // mssql, firebird, postgres
	DSN      string `mapstructure:"dsn"`  // Data Source Name (Connection String)
	ReadOnly bool   `mapstructure:"readonly"`
}

type ToolParameter struct {
	Name        string `mapstructure:"name"`
	Description string `mapstructure:"description"`
	Required    bool   `mapstructure:"required"`
	Default     string `mapstructure:"default"`
}