common

type Config struct {
	// DatadogYAML is the content of the datadog.yaml file
	DatadogYAML DatadogConfig
	// SecurityAgentYAML is the content of the security-agent.yaml file
	SecurityAgentYAML *SecurityAgentConfig
	// SystemProbeYAML is the content of the system-probe.yaml file
	SystemProbeYAML *SystemProbeConfig
	// InjectTracerYAML is the content of the inject/tracer.yaml file
	InjectTracerYAML InjectTracerConfig
	// IntegrationConfigs is the content of the integration configuration files under conf.d/
	IntegrationConfigs map[string]IntegrationConfig
}

type DatadogConfig struct {
	APIKey               string                     `yaml:"api_key"`
	Hostname             string                     `yaml:"hostname,omitempty"`
	Site                 string                     `yaml:"site,omitempty"`
	Proxy                DatadogConfigProxy         `yaml:"proxy,omitempty"`
	Env                  string                     `yaml:"env,omitempty"`
	Tags                 []string                   `yaml:"tags,omitempty"`
	LogsEnabled          bool                       `yaml:"logs_enabled,omitempty"`
	DJM                  DatadogConfigDJM           `yaml:"djm_config,omitempty"`
	ProcessConfig        DatadogConfigProcessConfig `yaml:"process_config,omitempty"`
	ExpectedTagsDuration string                     `yaml:"expected_tags_duration,omitempty"`
	RemoteUpdates        bool                       `yaml:"remote_updates,omitempty"`
	Installer            DatadogConfigInstaller     `yaml:"installer,omitempty"`
	DDURL                string                     `yaml:"dd_url,omitempty"`
}

type DatadogConfigDJM struct {
	Enabled bool `yaml:"enabled,omitempty"`
}

type DatadogConfigInstaller struct {
	Registry DatadogConfigInstallerRegistry `yaml:"registry,omitempty"`
}

type DatadogConfigInstallerRegistry struct {
	URL  string `yaml:"url,omitempty"`
	Auth string `yaml:"auth,omitempty"`
}

type DatadogConfigProcessConfig struct {
	ExpvarPort int `yaml:"expvar_port,omitempty"`
}

type DatadogConfigProxy struct {
	HTTP    string   `yaml:"http,omitempty"`
	HTTPS   string   `yaml:"https,omitempty"`
	NoProxy []string `yaml:"no_proxy,omitempty"`
}

type InjectTracerConfig struct {
	Version                        int                        `yaml:"version,omitempty"`
	ConfigSources                  string                     `yaml:"config_sources,omitempty"`
	AdditionalEnvironmentVariables []InjectTracerConfigEnvVar `yaml:"additional_environment_variables,omitempty"`
}

type InjectTracerConfigEnvVar struct {
	Key   string `yaml:"key"`
	Value string `yaml:"value"`
}

type IntegrationConfig struct {
	InitConfig any                     `yaml:"init_config"`
	Instances  []any                   `yaml:"instances,omitempty"`
	Logs       []IntegrationConfigLogs `yaml:"logs,omitempty"`
}

type IntegrationConfigInstanceSpark struct {
	SparkURL         string `yaml:"spark_url"`
	SparkClusterMode string `yaml:"spark_cluster_mode"`
	ClusterName      string `yaml:"cluster_name"`
	StreamingMetrics bool   `yaml:"streaming_metrics"`
}

type IntegrationConfigInstanceYarn struct {
	ResourceManagerURI string `yaml:"resourcemanager_uri"`
	ClusterName        string `yaml:"cluster_name"`
}

type IntegrationConfigLogs struct {
	Type    string `yaml:"type,omitempty"`
	Path    string `yaml:"path,omitempty"`
	Service string `yaml:"service,omitempty"`
	Source  string `yaml:"source,omitempty"`
	Tags    string `yaml:"tags,omitempty"`
}

type Output struct {
	// contains filtered or unexported fields
}

type Packages struct {
	// contains filtered or unexported fields
}

type RuntimeSecurityConfig struct {
	Enabled bool `yaml:"enabled,omitempty"`
}

type SecurityAgentComplianceConfig struct {
	Enabled bool `yaml:"enabled,omitempty"`
}

type SecurityAgentConfig struct {
	ComplianceConfig      SecurityAgentComplianceConfig `yaml:"compliance_config,omitempty"`
	RuntimeSecurityConfig RuntimeSecurityConfig         `yaml:"runtime_security_config,omitempty"`
}

type Setup struct {
	Out                     *Output
	Env                     *env.Env
	Ctx                     context.Context
	Span                    *telemetry.Span
	Packages                Packages
	Config                  Config
	DdAgentAdditionalGroups []string
	// contains filtered or unexported fields
}

type SystemProbeConfig struct {
	RuntimeSecurityConfig RuntimeSecurityConfig `yaml:"runtime_security_config,omitempty"`
}