config

type APIConfig struct {
	// Port where the agent API should listen on.
	Port string `json:"port"`

	// Listener used by the agent to serve its API. It takes
	// precedence over the Port field. It is closed by the agent.
	Listener net.Listener

	// Interface name that defines the IP a check should use to reach the agent API.
	IName string `json:"iname" toml:"iname"`

	// Host a check should use to reach the agent. Overrides the IName config param.
	Host string `json:"host" toml:"host"`
}

type AgentConfig struct {
	LogLevel       string `toml:"log_level"`
	LogFile        string `toml:"log_file"`
	Timeout        int    `toml:"timeout"` // Timeout to start running a check.
	ConcurrentJobs int    `toml:"concurrent_jobs"`
	// MaxMsgsInterval defines the maximun time, in seconds, the agent can
	// running without reading any message from the queue.
	MaxNoMsgsInterval      int `toml:"max_no_msgs_interval"`
	MaxProcessMessageTimes int `toml:"max_message_processed_times"`
}

type Auth struct {
	Server string `toml:"server"`
	User   string `toml:"user"`
	Pass   string `toml:"pass"`
}

type CheckConfig struct {
	AbortTimeout int               `toml:"abort_timeout"` // Time to wait for a check container to stop gracefully.
	LogLevel     string            `toml:"log_level"`     // Log level for the check default logger.
	Vars         map[string]string `toml:"vars"`          // Environment variables to inject to checks.
}

type ClusterConfig struct {
	Name   string `toml:"name"`
	Server string `toml:"server"`
}

type Config struct {
	Agent     AgentConfig   `toml:"agent"`
	Stream    StreamConfig  `toml:"stream"`
	SQSReader SQSReader     `toml:"sqs_reader"`
	SQSWriter SQSWriter     `toml:"sqs_writer"`
	S3Writer  S3Writer      `toml:"s3_writer"`
	API       APIConfig     `toml:"api"`
	Check     CheckConfig   `toml:"check"`
	Runtime   RuntimeConfig `toml:"runtime"`
	DataDog   DatadogConfig `toml:"datadog"`
}

type ContextConfig struct {
	Name      string `toml:"name"`
	Cluster   string `toml:"cluster"`
	Namespace string `toml:"namespace"`
	User      string `toml:"user"`
}

type CredentialsConfig struct {
	Name  string `toml:"name"`
	Token string `toml:"token"`
}

type DatadogConfig struct {
	Enabled bool   `toml:"metrics_enabled"`
	Statsd  string `toml:"dogstatsd"`
}

type DockerConfig struct {
	Registry RegistryConfig `toml:"registry"`
}

type KubernetesConfig struct {
	Cluster     ClusterConfig     `toml:"cluster"`
	Context     ContextConfig     `toml:"context"`
	Credentials CredentialsConfig `toml:"credentials"`
}

type PullPolicy int

type RegistryConfig struct {
	Auths               []Auth     `toml:"auths"`
	Server              string     `toml:"server"`
	User                string     `toml:"user"`
	Pass                string     `toml:"pass"`
	BackoffInterval     int        `toml:"backoff_interval"`
	BackoffMaxRetries   int        `toml:"backoff_max_retries"`
	BackoffJitterFactor float64    `toml:"backoff_jitter_factor"`
	PullPolicy          PullPolicy `toml:"pull_policy"`
}

type RuntimeConfig struct {
	Docker     DockerConfig     `toml:"docker"`
	Kubernetes KubernetesConfig `toml:"kubernetes"`
}

type S3Writer struct {
	BucketReports string `toml:"bucket_reports"`
	BucketLogs    string `toml:"bucket_logs"`
	Region        string `toml:"region"`
	LinkBase      string `toml:"link_base"`
	S3Link        bool   `toml:"s3_link"`
	Endpoint      string `toml:"endpoint"`
	PathStyle     bool   `toml:"path_style"`
}

type SQSReader struct {
	Endpoint          string `toml:"endpoint"`
	ARN               string `toml:"arn"`
	VisibilityTimeout int    `toml:"visibility_timeout"`
	PollingInterval   int    `toml:"polling_interval"`
	ProcessQuantum    int    `toml:"process_quantum"`
}

type SQSWriter struct {
	Endpoint string `toml:"endpoint"`
	ARN      string `toml:"arn"`
}

type StreamConfig struct {
	Endpoint      string `toml:"endpoint"` // Can be empty if no cancelation mechanism is required
	QueryEndpoint string `toml:"query_endpoint"`
	Timeout       int    `toml:"timeout"`
	Retries       int    `toml:"retries"`
	RetryInterval int    `toml:"retry_interval"`
}