agent

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

type ArtifactFinalizer = runtime.ArtifactFinalizer

type Options struct {
	// Dry is a dry-run mode. It does not execute the actual command.
	// Dry run does not create runstore data.
	Dry bool
	// RetryTarget is the target status (runstore of execution) to retry.
	// If it's specified the agent will execute the DAG with the same
	// configuration as the specified history.
	RetryTarget *exec.DAGRunStatus
	// ParentDAGRun is the dag-run reference of the parent dag-run.
	// It is required for sub dag-runs to identify the parent dag-run.
	ParentDAGRun exec.DAGRunRef
	// ProgressDisplay indicates if the progress display should be shown.
	// This is typically enabled for CLI execution in a TTY environment.
	ProgressDisplay bool
	// ExtraEnvs are additional execution-scoped env vars injected into the DAG run context.
	ExtraEnvs []string
	// StepRetry is the name of the step to retry, if specified.
	StepRetry string
	// RetryPath identifies a persisted child DAG step retry.
	RetryPath exec.RetryPath
	// WorkerID is the identifier of the worker executing this DAG run.
	// For distributed execution, this is set to the worker's ID.
	// For local execution, this defaults to "local".
	WorkerID string
	// StatusPusher is used to push status updates to a remote coordinator.
	// When nil, status is written to local filesystem via the run-state attempt.
	StatusPusher StatusPusher
	// SubWorkflowRunnerFactory creates a runner for child workflows.
	SubWorkflowRunnerFactory SubWorkflowRunnerFactory
	// LogWriterFactory is used to create log writers for step output.
	// When nil, logs are written to local filesystem.
	LogWriterFactory exec.LogWriterFactory
	// QueuedRun indicates this execution is from a queued item.
	// When true, the agent will find the existing dag-run (created by enqueue)
	// instead of creating a new one. This is used for distributed execution
	// where the dag-run directory was already created by the scheduler.
	QueuedRun bool
	// AttemptID is the attempt ID from the coordinator.
	// When set, the agent creates an attempt with this ID instead of generating a new one.
	AttemptID string
	// PreparedAttempt is an exact attempt that was created or reopened before proc acquisition.
	// This is used for local execution so the proc heartbeat can include the final attempt ID.
	PreparedAttempt exec.DAGRunAttempt
	// RunStateStore records execution state for this DAG run.
	RunStateStore runstate.Store
	// DAGRunStore is the store for dag-run data. Nil for remote worker execution.
	DAGRunStore exec.DAGRunStore
	// QueueStore is the store for queued dag-run items. Nil when queues are unavailable.
	QueueStore exec.QueueStore
	// StateStore is the persistent state store shared across DAG runs.
	StateStore dagstate.Store
	// SecretStore resolves local registry refs and runtime profile secrets.
	SecretStore secretpkg.Store
	// SecretReferenceResolver resolves DAG-level registry refs.
	// When nil, SecretStore supplies the local resolver.
	SecretReferenceResolver secrets.ReferenceResolver
	// ProfileStore resolves named runtime profiles.
	ProfileStore profilepkg.Store
	// ProfileName selects the runtime profile for this DAG run.
	ProfileName string
	// ServiceRegistry is the registry for service discovery.
	ServiceRegistry exec.ServiceRegistry
	// RootDAGRun is the root dag-run reference for sub-DAG runs.
	RootDAGRun exec.DAGRunRef
	// PeerConfig is the configuration for peer communication.
	PeerConfig config.Peer
	// TriggerType indicates how this DAG run was initiated.
	TriggerType core.TriggerType
	// TriggerActor identifies the attributable actor that initiated the DAG run.
	TriggerActor string
	// DefaultExecMode is the server-level default execution mode.
	DefaultExecMode config.ExecutionMode
	// ScheduleTime is the RFC 3339 timestamp of when this run was scheduled.
	// Set by the scheduler for cron-triggered runs; empty for manual runs.
	ScheduleTime string
	// ArtifactDir is the per-run artifact directory when artifact storage is enabled.
	ArtifactDir string
	// DAGRunLogDir is the base log directory used for child DAG runs created by executors.
	DAGRunLogDir string
	// DAGRunArtifactDir is the base artifact directory used for child DAG runs created by executors.
	DAGRunArtifactDir string
	// ArtifactFinalizer persists artifacts before the final terminal status is written.
	ArtifactFinalizer ArtifactFinalizer
	// RemoteDAGLoader loads a DAG from a remote source when the local DAG store misses.
	// When nil, no remote fallback is attempted.
	RemoteDAGLoader RemoteDAGLoader
	// SocketServerFactory creates the local status/control transport.
	// When nil, the default Unix socket transport is used.
	SocketServerFactory SocketServerFactory
}

type ProgressReporter interface {
	// Start begins the progress display
	Start()

	// Stop stops the progress display
	Stop()

	// UpdateNode updates the progress for a specific node
	UpdateNode(node *exec.Node)

	// UpdateStatus updates the overall DAG status
	UpdateStatus(status *exec.DAGRunStatus)

	// SetDAGRunInfo sets the DAG run ID and parameters
	SetDAGRunInfo(dagRunID, params string)
}

type RemoteDAGLoader func(ctx context.Context, name string) (*core.DAG, error)

type Sender interface {
	Send(ctx context.Context, from string, to []string, subject, body string, attachments []string) error
}

type SenderFn func(ctx context.Context, from string, to []string, subject, body string, attachments []string) error

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

type SocketServer interface {
	Serve(ctx context.Context, listen chan error) error
	Shutdown(ctx context.Context) error
}

type SocketServerFactory func(addr string, handlerFunc sock.HTTPHandlerFunc) (SocketServer, error)

type StatusPusher = runtime.StatusPusher

type SubWorkflowRunnerFactory func(ctx context.Context) (runtimeexec.SubWorkflowRunner, error)