config

type APMConfigurationDefault struct {
	TraceDebug                    *bool   `yaml:"DD_TRACE_DEBUG,omitempty"`
	IntegrationsEnabled           *bool   `yaml:"DD_INTEGRATIONS_ENABLED,omitempty"`
	DataJobsCommandPattern        string  `yaml:"DD_DATA_JOBS_COMMAND_PATTERN,omitempty"`
	DataJobsSparkAppNameAsService *bool   `yaml:"DD_SPARK_APP_NAME_AS_SERVICE,omitempty"`
	RuntimeMetricsEnabled         *bool   `yaml:"DD_RUNTIME_METRICS_ENABLED,omitempty"`
	LogsInjection                 *bool   `yaml:"DD_LOGS_INJECTION,omitempty"`
	APMTracingEnabled             *bool   `yaml:"DD_APM_TRACING_ENABLED,omitempty"`
	ProfilingEnabled              *string `yaml:"DD_PROFILING_ENABLED,omitempty"`
	DataStreamsEnabled            *bool   `yaml:"DD_DATA_STREAMS_ENABLED,omitempty"`
	AppsecEnabled                 *bool   `yaml:"DD_APPSEC_ENABLED,omitempty"`
	IastEnabled                   *bool   `yaml:"DD_IAST_ENABLED,omitempty"`
	DataJobsEnabled               *bool   `yaml:"DD_DATA_JOBS_ENABLED,omitempty"`
	AppsecScaEnabled              *bool   `yaml:"DD_APPSEC_SCA_ENABLED,omitempty"`
}

type ApplicationMonitoringConfig struct {
	Default APMConfigurationDefault `yaml:"apm_configuration_default,omitempty"`
}

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
	// IntegrationConfigs is the content of the integration configuration files under conf.d/
	IntegrationConfigs map[string]IntegrationConfig
	// ApplicationMonitoringYAML is the content of the application_monitoring.yaml configuration file
	ApplicationMonitoringYAML *ApplicationMonitoringConfig
}

type DatadogConfig struct {
	APIKey               string                     `yaml:"api_key,omitempty"`
	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"`
	ExtraTags            []string                   `yaml:"extra_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"`
	LogsConfig           LogsConfig                 `yaml:"logs_config,omitempty"`
	GPUCheck             GPUCheckConfig             `yaml:"gpu,omitempty"`
	SBOM                 SBOMConfig                 `yaml:"sbom,omitempty"`
	InfrastructureMode   string                     `yaml:"infrastructure_mode,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 DelayedAgentRestartConfig struct {
	Scheduled bool
	Delay     time.Duration
	LogFile   string
}

type GPUCheckConfig struct {
	Enabled     *bool  `yaml:"enabled,omitempty"`
	NvmlLibPath string `yaml:"nvml_lib_path,omitempty"`
}

type GPUMonitoringConfig struct {
	Enabled *bool `yaml:"enabled,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"`
	AutoMultiLineDetection *bool               `yaml:"auto_multi_line_detection,omitempty"`
	LogProcessingRules     []LogProcessingRule `yaml:"log_processing_rules,omitempty"`
}

type LogProcessingRule struct {
	Type    string `yaml:"type" json:"type"`
	Name    string `yaml:"name" json:"name"`
	Pattern string `yaml:"pattern" json:"pattern"`
}

type LogsConfig struct {
	ProcessingRules []LogProcessingRule `yaml:"processing_rules"`
}

type RuntimeSecurityConfig struct {
	Enabled *bool      `yaml:"enabled,omitempty"`
	SBOM    SBOMConfig `yaml:"sbom,omitempty"`
}

type SBOMConfig struct {
	Enabled        *bool                    `yaml:"enabled,omitempty"`
	ContainerImage SBOMContainerImageConfig `yaml:"container_image,omitempty"`
	Host           SBOMHostConfig           `yaml:"host,omitempty"`
}

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

type SBOMHostConfig 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 SystemProbeConfig struct {
	RuntimeSecurityConfig RuntimeSecurityConfig `yaml:"runtime_security_config,omitempty"`
	GPUMonitoringConfig   GPUMonitoringConfig   `yaml:"gpu_monitoring,omitempty"`
}