wfruntime

type AsyncResultStore interface {
	CommitAsyncResult(ctx context.Context, task *AsyncTask, run *WorkflowRun, expectedUpdatedAt time.Time, snapshots []*RunSnapshot, events []RunEvent) error
}

type AsyncTask struct {
	DispatchID     string                  `json:"dispatch_id"`
	RunID          string                  `json:"run_id"`
	NodeID         string                  `json:"node_id"`
	ExternalTaskID string                  `json:"external_task_id"`
	Task           executor.ExecuteTask    `json:"task"`
	Result         *executor.ExecuteResult `json:"result,omitempty"`
	ResultHash     string                  `json:"-"`
	ResultApplied  bool                    `json:"-"`
	// ExpireAt is the persisted callback deadline; zero means no expiration.
	ExpireAt   time.Time      `json:"expire_at,omitempty,omitzero"`
	State      AsyncTaskState `json:"state"`
	ClaimToken string         `json:"-"`
	ClaimUntil time.Time      `json:"-"`
	CreatedAt  time.Time      `json:"created_at"`
	UpdatedAt  time.Time      `json:"updated_at"`
}

type AsyncTaskState string

type AsyncTaskStore interface {
	SaveAsyncTask(ctx context.Context, task *AsyncTask) error
	SubmitAsyncResult(ctx context.Context, dispatchID string, result executor.ExecuteResult) (bool, error)
	// ClaimCompletedAsyncTasks also atomically completes expired waiting tasks
	// with a non-retryable timeout result before claiming recovery work.
	ClaimCompletedAsyncTasks(ctx context.Context, now time.Time, limit int, lease time.Duration) ([]*AsyncTask, error)
	RenewAsyncTask(ctx context.Context, dispatchID, claimToken string, claimUntil time.Time) (bool, error)
	AcknowledgeAsyncTask(ctx context.Context, dispatchID, claimToken string) error
	ReleaseAsyncTask(ctx context.Context, dispatchID, claimToken string) error
	CancelAsyncTasks(ctx context.Context, runID string) error
}

type EventType string

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

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

type NodeRun struct {
	NodeID      string    `json:"node_id"`
	DispatchID  string    `json:"dispatch_id,omitempty"`
	Status      Status    `json:"status"`
	Attempt     int       `json:"attempt"`
	MaxAttempts int       `json:"max_attempts,omitempty"`
	StartedAt   time.Time `json:"started_at,omitempty"`
	FinishedAt  time.Time `json:"finished_at,omitempty"`
	// Input keeps the latest materialized task input so the UI can explain what
	// a node actually received during execution instead of only showing config.
	Input    any            `json:"input,omitempty"`
	Result   any            `json:"result,omitempty"`
	Error    string         `json:"error,omitempty"`
	Metadata map[string]any `json:"metadata,omitempty"`
}

type RunContext struct {
	Variables   map[string]any `json:"variables,omitempty"`
	NodeResults map[string]any `json:"node_results,omitempty"`
}

type RunEvent struct {
	RunID      string         `json:"run_id"`
	Type       EventType      `json:"type"`
	NodeID     string         `json:"node_id,omitempty"`
	Status     Status         `json:"status,omitempty"`
	Time       time.Time      `json:"time"`
	Message    string         `json:"message,omitempty"`
	Payload    map[string]any `json:"payload,omitempty"`
	WorkflowID string         `json:"workflow_id,omitempty"`
}

type RunSnapshot struct {
	RunID       string              `json:"run_id"`
	Status      Status              `json:"status"`
	NodeRuns    map[string]*NodeRun `json:"node_runs"`
	Context     RunContext          `json:"context"`
	At          time.Time           `json:"at"`
	Description string              `json:"description,omitempty"`
}

type Status string

type Store interface {
	SaveRun(ctx context.Context, run *WorkflowRun) error
	LoadRun(ctx context.Context, runID string) (*WorkflowRun, error)
	ListRuns(ctx context.Context) ([]*WorkflowRun, error)
	SaveSnapshot(ctx context.Context, snapshot *RunSnapshot) error
	Snapshots(ctx context.Context, runID string) ([]*RunSnapshot, error)
	AppendEvent(ctx context.Context, event RunEvent) error
	Events(ctx context.Context, runID string) ([]RunEvent, error)
}

type WorkflowRun struct {
	ID                 string              `json:"id"`
	WorkflowID         string              `json:"workflow_id"`
	WorkflowVersionID  string              `json:"workflow_version_id"`
	PlanID             string              `json:"plan_id"`
	RequestFingerprint string              `json:"request_fingerprint,omitempty"`
	CredentialScope    string              `json:"credential_scope,omitempty"`
	Status             Status              `json:"status"`
	CurrentNodes       []string            `json:"current_nodes,omitempty"`
	NodeRuns           map[string]*NodeRun `json:"node_runs,omitempty"`
	Context            RunContext          `json:"context"`
	CreatedAt          time.Time           `json:"created_at"`
	UpdatedAt          time.Time           `json:"updated_at"`
	StartedAt          time.Time           `json:"started_at,omitempty"`
	FinishedAt         time.Time           `json:"finished_at,omitempty"`
	// contains filtered or unexported fields
}