engine

type AsyncJob struct {
	TaskID      int64          `json:"task_id"`
	Node        string         `json:"node"`
	StepAdapter string         `json:"step_adapter"`
	Input       map[string]any `json:"input"`
	Hash        string         `json:"hash"`
}

type AsyncJobQueue interface {

	// 发布任务
	Publish(ctx context.Context, job AsyncJob) error

	// Worker消费
	Consume(ctx context.Context, group string, consumer string) (*AsyncJob, string, error)

	// ack
	Ack(ctx context.Context, id string) error
}

type AwaitTraceInfo struct {
	AwaitType        domain.AwaitType          `json:"await_type"`
	Source           domain.AwaitSource        `json:"source"`
	Status           domain.AwaitBindingStatus `json:"status"`
	Provider         *string                   `json:"provider,omitempty"`
	ProviderTaskID   *string                   `json:"provider_task_id,omitempty"`
	ResultPayload    map[string]any            `json:"result_payload,omitempty"`
	ErrorMessage     string                    `json:"error_message,omitempty"`
	PollAttempts     int                       `json:"poll_attempts,omitempty"`
	WaitingStartedAt *time.Time                `json:"waiting_started_at,omitempty"`
	CompletedAt      *time.Time                `json:"completed_at,omitempty"`
}

type CheckpointOutputRebuilder func(
	nodeDef nodes.Node,
	runtime *domain.NodeRuntime,
) (map[string]any, error)

type ClosureValidationIssue struct {
	Level     ClosureValidationLevel `json:"level"`
	NodeName  string                 `json:"node_name"`
	EdgeKey   string                 `json:"edge_key,omitempty"`
	FieldName string                 `json:"field_name,omitempty"`
	Message   string                 `json:"message"`
}

type ClosureValidationLevel string

type ClosureValidationMode string

type ClosureValidationResult struct {
	Valid  bool                     `json:"valid"`
	Issues []ClosureValidationIssue `json:"issues,omitempty"`
}

type DirtyPlan struct {
	DirtyNodes   map[string]string
	ReuseNodes   map[string]bool
	MapItemReuse map[string]map[int]bool
	PatchedNodes map[string]bool
}

type EdgeRuntimeState string

type Engine struct {
	WorkflowVersionRepo repository.WorkflowVersionRepository
	WorkflowRepo        repository.WorkflowRepository
	// contains filtered or unexported fields
}

type EngineOption func(*engineConfig)

type ExecutionReason string

type MapItemFrame struct {
	Index  int            `json:"index"`
	Output map[string]any `json:"output"`
	Reused bool           `json:"reused"`
}

type NodePlan struct {
	Name         string
	Label        string
	NodeType     definition.NodeType
	Action       PlanAction
	Reason       ExecutionReason
	ReuseKind    domain.ReuseKind
	MapItemReuse map[int]bool
	Patches      []domain.RuntimePatch

	// 调试 / 观测信息
	ParentTaskID *int64
	ParentNode   *string
}

type NodeReplayResult struct {
	TaskID         int64          `json:"task_id"`
	NodeName       string         `json:"node_name"`
	NodeState      string         `json:"node_state"`
	NodeType       string         `json:"node_type"`
	Tool           string         `json:"tool,omitempty"`
	ResolvedInput  map[string]any `json:"resolved_input,omitempty"`
	OriginalOutput map[string]any `json:"original_output,omitempty"`
	OriginalError  string         `json:"original_error,omitempty"`
	ReplayOutput   map[string]any `json:"replay_output,omitempty"`
	ReplayError    string         `json:"replay_error,omitempty"`
	Executed       bool           `json:"executed"`
}

type NodeReplayService interface {
	ReplayTaskNode(ctx context.Context, taskID int64, nodeName string, execute bool) (*NodeReplayResult, error)
}

type NodeTraceFrame struct {
	Name             string           `json:"name"`
	Index            int              `json:"index"`
	BizIndex         int              `json:"biz_index"`
	State            domain.NodeState `json:"state"`
	StartedAt        *time.Time       `json:"started_at,omitempty"`
	FinishedAt       *time.Time       `json:"finished_at,omitempty"`
	DurationMs       *int64           `json:"duration_ms,omitempty"`
	ResolvedInput    map[string]any   `json:"resolved_input"`
	Output           map[string]any   `json:"output"`
	OutputPersisted  bool             `json:"output_persisted"`
	ActivatedEdges   map[string]bool  `json:"activated_edges"`
	ReuseKind        domain.ReuseKind `json:"reuse_kind,omitempty"`
	ReusedFromTaskID *int64           `json:"reused_from_task_id,omitempty"`
	Error            string           `json:"error,omitempty"`
	// Await 节点附加信息
	AwaitInfo *AwaitTraceInfo `json:"await_info,omitempty"`
	// Map/Loop 节点逐 item 结果
	MapItems []MapItemFrame `json:"map_items,omitempty"`
}

type ParentFanoutNodeKind string

type PlanAction string

type ReplayTrace struct {
	TaskID            int64             `json:"task_id"`
	WorkflowVersionID int64             `json:"workflow_version_id"`
	Status            domain.TaskStatus `json:"status"`
	StartedAt         time.Time         `json:"started_at"`
	FinishedAt        *time.Time        `json:"finished_at,omitempty"`
	Input             map[string]any    `json:"input"`
	Nodes             []NodeTraceFrame  `json:"nodes"`
}

type RetryTrigger string

type RunCancellationResult struct {
	TaskID                  int64
	Reason                  string
	AlreadyCanceled         bool
	CanceledTaskIDs         []int64
	CanceledNodeIDs         []int64
	CanceledAwaitBindingIDs []int64
}

type RunCancellationService interface {
	CancelForSupersededRevision(ctx context.Context, taskID int64) (*RunCancellationResult, error)
}

type RunPlan struct {
	TaskID       int64
	Mode         RunPlanMode
	ResumeFrom   string
	ParentTaskID *int64
	Nodes        map[string]*NodePlan
	TopoOrder    []string
}

type RunPlanMode string

type RunRedoService interface {
	RedoRun(
		ctx context.Context,
		sourceTaskID int64,
		resumeSpec *domain.ResumeSpec,
		overrideInput map[string]any,
		editAction string,
		editLabel string,
		note string,
	) (*domain.Task, error)
}

type RunResult struct {
	Status RunStatus
	Err    error

	// 可选
	SuspendReason string
	SuspendNode   string
}

type RunStatus string

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

type TaskRetryService interface {
	PrepareTaskRetry(ctx context.Context, taskID int64, trigger RetryTrigger, resumeFrom string, patches []domain.RuntimePatch) error
	ClearMapChildCheckpointResult(ctx context.Context, parentTaskID int64, nodeName string, childTaskID int64) error
}