airflow

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

type AirflowConfig struct {
	BaseURL     string
	APIKey      string
	Timeout     time.Duration
	MaxRetries  int
	RetryDelay  time.Duration
	WebserverID string
}

type ChannelConfig struct {
	Type     string            `json:"type"`
	Settings map[string]string `json:"settings"`
}

type DAG struct {
	ID               string                 `json:"dag_id"`
	Description      string                 `json:"description"`
	IsPaused         bool                   `json:"is_paused"`
	IsActive         bool                   `json:"is_active"`
	FileToken        string                 `json:"file_token"`
	Owners           []string               `json:"owners"`
	Tags             []string               `json:"tags"`
	ScheduleInterval string                 `json:"schedule_interval"`
	DefaultArgs      map[string]interface{} `json:"default_args"`
}

type DAGRun struct {
	ID              string                 `json:"dag_run_id"`
	DAGID           string                 `json:"dag_id"`
	LogicalDate     time.Time              `json:"logical_date"`
	ExecutionDate   time.Time              `json:"execution_date"`
	StartDate       *time.Time             `json:"start_date"`
	EndDate         *time.Time             `json:"end_date"`
	State           string                 `json:"state"`
	ExternalTrigger bool                   `json:"external_trigger"`
	Conf            map[string]interface{} `json:"conf"`
}

type NotificationConfig struct {
	OnSuccess []string                 `json:"on_success"`
	OnFailure []string                 `json:"on_failure"`
	OnRetry   []string                 `json:"on_retry"`
	Channels  map[string]ChannelConfig `json:"channels"`
}

type ScanConfiguration struct {
	Target          string                 `json:"target"`
	ScanType        string                 `json:"scan_type"`
	Scanners        []string               `json:"scanners"`
	MaxConcurrency  int                    `json:"max_concurrency"`
	ConditionalFlow map[string]interface{} `json:"conditional_flow"`
	Notifications   NotificationConfig     `json:"notifications"`
}

type SecurityScanDAG struct {
	*DAG
	ScanConfig ScanConfiguration `json:"scan_config"`
}

type TaskInstance struct {
	TaskID         string     `json:"task_id"`
	DAGID          string     `json:"dag_id"`
	DAGRunID       string     `json:"dag_run_id"`
	ExecutionDate  time.Time  `json:"execution_date"`
	StartDate      *time.Time `json:"start_date"`
	EndDate        *time.Time `json:"end_date"`
	Duration       float64    `json:"duration"`
	State          string     `json:"state"`
	TryNumber      int        `json:"try_number"`
	MaxTries       int        `json:"max_tries"`
	Hostname       string     `json:"hostname"`
	Pool           string     `json:"pool"`
	Queue          string     `json:"queue"`
	PriorityWeight int        `json:"priority_weight"`
	Operator       string     `json:"operator"`
	QueuedWhen     *time.Time `json:"queued_when"`
	PID            *int       `json:"pid"`
	ExecutorConfig string     `json:"executor_config"`
}