config

type APIConfig struct {
	Method  string            `yaml:"method"`
	URL     string            `yaml:"url"`
	Headers map[string]string `yaml:"headers,omitempty"`
	Auth    *AuthConfig       `yaml:"auth,omitempty"`
	Body    interface{}       `yaml:"body,omitempty"`
	Timeout time.Duration     `yaml:"timeout,omitempty"`
}

type APIKeyConfig struct {
	Header string `yaml:"header,omitempty"` // Header name (e.g., "X-API-Key")
	Query  string `yaml:"query,omitempty"`  // Query param name (e.g., "api_key")
	Value  string `yaml:"value"`            // The API key value
}

type AssertConfig struct {
	Expr string `yaml:"expr"`
	Msg  string `yaml:"msg,omitempty"`
}

type AuthConfig struct {
	Bearer string           `yaml:"bearer,omitempty"`  // Bearer token value
	Basic  *BasicAuthConfig `yaml:"basic,omitempty"`   // Basic auth credentials
	APIKey *APIKeyConfig    `yaml:"api_key,omitempty"` // API key auth
}

type BasicAuthConfig struct {
	Username string `yaml:"username"`
	Password string `yaml:"password"`
}

type CleanupStep struct {
	DBStep *DBStepConfig `yaml:"-"` // populated by custom unmarshaling
}

type DBStepConfig struct {
	Database string `yaml:"-"` // resolved database name (from YAML key)

	// SQL fields (Postgres, MySQL, SQLite)
	Query  string        `yaml:"query,omitempty"`
	Params []interface{} `yaml:"params,omitempty"`

	// MongoDB fields
	Collection string                 `yaml:"collection,omitempty"`
	Operation  string                 `yaml:"operation,omitempty"`
	Filter     map[string]interface{} `yaml:"filter,omitempty"`
	Document   interface{}            `yaml:"document,omitempty"`
	Documents  []interface{}          `yaml:"documents,omitempty"`
	Update     map[string]interface{} `yaml:"update,omitempty"`
}

type DatabaseDriver string

type EnvConfig struct {
	APIBase      string            `yaml:"api_base,omitempty"`
	Databases    map[string]string `yaml:"databases,omitempty"`
	KafkaBrokers string            `yaml:"kafka_brokers,omitempty"`
	Redis        string            `yaml:"redis,omitempty"`
}

type FlowConfig struct {
	Name        string        `yaml:"name"`
	Description string        `yaml:"description,omitempty"`
	Tags        []string      `yaml:"tags,omitempty"`
	Timeout     time.Duration `yaml:"timeout,omitempty"`
	FailFast    bool          `yaml:"fail_fast,omitempty"`
	Env         EnvConfig     `yaml:"env,omitempty"`
	Setup       []SetupStep   `yaml:"setup,omitempty"`
	Cleanup     []CleanupStep `yaml:"cleanup,omitempty"`
	Steps       []Step        `yaml:"steps"`
}

type GlobalConfig struct {
	Env      EnvConfig            `yaml:"env,omitempty"`
	Profiles map[string]EnvConfig `yaml:"profiles,omitempty"`
}

type KafkaConfig struct {
	Action  string                 `yaml:"action,omitempty"` // "produce" or "consume" (default: consume)
	Topic   string                 `yaml:"topic"`
	Timeout time.Duration          `yaml:"timeout,omitempty"`
	Match   map[string]interface{} `yaml:"match,omitempty"`
	Key     string                 `yaml:"key,omitempty"`
	Message interface{}            `yaml:"message,omitempty"` // for produce
	Headers map[string]string      `yaml:"headers,omitempty"` // for produce
}

type RedisConfig struct {
	Action string        `yaml:"action"`
	Key    string        `yaml:"key"`
	Value  interface{}   `yaml:"value,omitempty"`
	TTL    time.Duration `yaml:"ttl,omitempty"`
}

type RetryConfig struct {
	Times    int           `yaml:"times"`              // max retry attempts (default 1 = no retry)
	Interval time.Duration `yaml:"interval,omitempty"` // wait between retries (default 1s)
	Backoff  string        `yaml:"backoff,omitempty"`  // "linear" or "exponential" (default: linear)
}

type ScriptConfig struct {
	Lang       string        `yaml:"lang,omitempty"`
	Run        string        `yaml:"run"`
	AssertExit *int          `yaml:"assert_exit,omitempty"`
	Timeout    time.Duration `yaml:"timeout,omitempty"`
}

type SeedConfig struct {
	Target string                 `yaml:"target,omitempty"` // database name; defaults to first DB if omitted
	Table  string                 `yaml:"table"`
	Data   map[string]interface{} `yaml:"data"`
}

type SetupStep struct {
	Seed   *SeedConfig   `yaml:"seed,omitempty"`
	DBStep *DBStepConfig `yaml:"-"` // populated by custom unmarshaling
}

type Step struct {
	Name   string            `yaml:"name"`
	When   string            `yaml:"when,omitempty"`
	Delay  time.Duration     `yaml:"delay,omitempty"`
	Retry  *RetryConfig      `yaml:"retry,omitempty"`
	API    *APIConfig        `yaml:"api,omitempty"`
	DBStep *DBStepConfig     `yaml:"-"` // populated by custom unmarshaling
	Kafka  *KafkaConfig      `yaml:"kafka,omitempty"`
	Redis  *RedisConfig      `yaml:"redis,omitempty"`
	Script *ScriptConfig     `yaml:"script,omitempty"`
	Assert []AssertConfig    `yaml:"assert,omitempty"`
	Save   map[string]string `yaml:"save,omitempty"`
}