pipeline

type CheckpointData struct {
	StepIndex   int                    `json:"step_index"`
	StepResults map[string]*StepResult `json:"step_results"`
	Event       types.DataEvent        `json:"event"`
	HeartbeatAt time.Time              `json:"heartbeat_at"`
}

type Clock interface {
	Now() time.Time
	After(d time.Duration) <-chan time.Time
}

type Definition struct {
	Name        string
	Description string
	Enabled     bool
	Resumable   bool
	Trigger     Trigger
	Steps       []Step
	ParentName  string
}

type DefinitionReader interface {
	ListPublishedDefinitions(ctx context.Context) ([]DefinitionRecord, error)
}

type DefinitionRecord struct {
	Name        string
	Description string
	YAML        string
	UpdatedAt   time.Time
}

type EditorDefinition struct {
	Name        string         `json:"name" yaml:"name"`
	Description string         `json:"description" yaml:"description"`
	Enabled     bool           `json:"enabled" yaml:"enabled"`
	Resumable   bool           `json:"resumable" yaml:"resumable"`
	Triggers    []TriggerEntry `json:"triggers" yaml:"triggers"`
	Steps       []Step         `json:"steps" yaml:"steps"`
}

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

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

type RealClock struct{}

type RenderContext struct {
	Event types.DataEvent
	Steps map[string]map[string]any
	Input map[string]any
	// contains filtered or unexported fields
}

type RunStore interface {
	CreateRun(ctx context.Context, pipelineName, eventID, eventType, triggerSource string) (*gen.PipelineRun, error)
	UpdateRunStatus(ctx context.Context, runID int64, status int, errMsg string) error
	CreateStepRun(ctx context.Context, runID int64, stepName, capability, operation string, params map[string]any, attempt int) (*gen.PipelineStepRun, error)
	UpdateStepRun(ctx context.Context, stepRunID int64, status int, result map[string]any, errMsg string, attempt int) error
	SaveCheckpoint(ctx context.Context, runID int64, data any) error
	GetIncompleteRuns(ctx context.Context) ([]*gen.PipelineRun, error)
	GetCheckpoint(ctx context.Context, runID int64, target any) error
	GetRun(ctx context.Context, runID int64) (*gen.PipelineRun, error)
	UpdateRunHeartbeat(ctx context.Context, runID int64) error
	HasConsumed(ctx context.Context, consumerName, eventID string) (bool, error)
	RecordConsumption(ctx context.Context, consumerName, eventID string) error
	RecordResourceLink(ctx context.Context, link *gen.ResourceLink) error
}

type Step struct {
	Name       string
	Capability hub.CapabilityType
	Operation  string
	Params     map[string]any
	Retry      *backoff.Config
}

type StepCallback interface {
	OnRunStart(ctx context.Context, runID int64, pipelineName string,
		trigger string, totalSteps int, stepNames []string)
	OnStepStart(ctx context.Context, runID int64, pipelineName string,
		stepIndex int, stepName string, input map[string]any)
	OnStepDone(ctx context.Context, runID int64, pipelineName string,
		stepIndex int, stepName string, output map[string]any, elapsedMs int64)
	OnStepError(ctx context.Context, runID int64, pipelineName string,
		stepIndex int, stepName string, err error, elapsedMs int64)
	OnRunComplete(ctx context.Context, runID int64, pipelineName string,
		elapsedMs int64, failed bool, errMsg string)
}

type StepProgressEvent struct {
	RunID        int64          `json:"run_id"`
	PipelineName string         `json:"pipeline_name"`
	StepIndex    int            `json:"step_index"`
	StepName     string         `json:"step_name"`
	Status       string         `json:"status"`
	Input        map[string]any `json:"input,omitempty"`
	Output       map[string]any `json:"output,omitempty"`
	ElapsedMs    int64          `json:"elapsed_ms,omitempty"`
	Error        string         `json:"error,omitempty"`
	TotalSteps   int            `json:"total_steps,omitempty"`
}

type StepResult struct {
	Name        string         `json:"name"`
	Capability  string         `json:"capability"`
	Operation   string         `json:"operation"`
	Output      map[string]any `json:"output"`
	CompletedAt time.Time      `json:"completed_at"`
}

type Trigger struct {
	Event       string
	Cron        string
	CronTimeout time.Duration
	Webhook     *WebhookConfig
}

type TriggerEntry struct {
	Enabled     bool           `json:"enabled" yaml:"enabled"`
	Type        string         `json:"type" yaml:"type"` // "event", "cron", "webhook"
	Event       string         `json:"event,omitempty" yaml:"event,omitempty"`
	Cron        string         `json:"cron,omitempty" yaml:"cron,omitempty"`
	CronTimeout string         `json:"cron_timeout,omitempty" yaml:"cron_timeout,omitempty"`
	Webhook     *WebhookConfig `json:"webhook,omitempty" yaml:"webhook,omitempty"`
}

type WebhookAuthConfig struct {
	Token       string
	HMACSecret  string
	HMACHeader  string
	TokenHeader string
}

type WebhookConfig struct {
	Path      string
	Method    string
	Auth      WebhookAuthConfig
	Payload   config.WebhookPayloadMode
	EventType string
}