core

type DefaultUnit struct {
	Name        string
	Description string
	// contains filtered or unexported fields
}

type ExecutionState interface {
	Save(state PipelineState) error
	Load() (PipelineState, error)
}

type IJob[T any] interface {
	ID() string        // 任务唯一标识
	Status() JobStatus // 任务的整体状态
	GetPolicy() IJobPolicy
	Execute(ctx context.Context, input any, current *Unit) (T, error) //
}

type IJobPolicy interface {
	RetryCount() int           // 重试次数
	MaxRetries() int           // 最大重试次数
	RetryDelay() time.Duration // 每次重试之间的延迟
}

type IStage[T any] interface {
	Name() string        // 阶段名称
	Status() StageStatus // 阶段状态
	DependsOn() []string // 依赖的阶段NAME（空表示第一个阶段）
	// contains filtered or unexported methods
}

type Job struct {
	Name    string // 任务名称
	Results any    // 存储阶段产物，key 为阶段名称

	Output       any                                                          // 输出可以是任意类型的数据
	Input        any                                                          // 输入也可以是任意类型的数据
	Policy       IJobPolicy                                                   //
	Dependencies []*IJob[any]                                                 // 该任务依赖的其他任务, 用于构建依赖关系
	Executor     func(ctx context.Context, i any, current *Unit) (any, error) // 执行器
	// contains filtered or unexported fields
}

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

type JobStatus string

type JobUnit struct {
	UnitId string    // Unit id
	Job    IJob[any] // 绑定的任务
}

type Line struct {
	From string
	To   string
}

type Pipeline struct {
	ID string

	Status     string // "pending", "running", "completed", "failed", "terminal", "retry", "resume","paused"
	StartTime  time.Time
	EndTime    time.Time
	StateStore ExecutionState

	Results map[string]any
	// contains filtered or unexported fields
}

type PipelineState struct {
	CurrentStageIndex int
	// 如果Stage内部有多个Unit并行或串行执行的上下文，也需要记录当前Unit的进度
	CurrentUnitIndex int
	Status           string
	// 可以存储上一阶段的输出数据，用于从中间点恢复
	LastOutput any
}

type Stage struct {
	Name      string `json:"name"`       // 阶段名称
	DependsOn string `json:"depends_on"` // 依赖的阶段NAME（空表示第一个阶段）

	Status StageStatus `json:"status"` // 阶段状态
	// contains filtered or unexported fields
}

type StageCallback func(stageName string, status StageStatus, self *Pipeline) bool

type StageStatus string

type StoryBoard struct {
	Stages []*Stage //这里不是真正的Stage ， 需要套一层UI-Data ，StageVo -》 Stage
	Lines  []*Line
}

type Unit interface {
	ID() string                                                     // Unit的ID，每次实例化可能不一样
	Run(ctx context.Context, input any, current *Unit) (any, error) // 执行单元的主要逻辑，会产出结果输出
}

type UnitOutput struct {
	ID     string `json:"id"`
	Output any    `json:"output"`
}

type UnitRegistry interface {
	Name() string         // Unit的名称
	Order() int           // 执行顺序
	TemplateName() string // 对应的模板名称
	SetInput(any) error   // 设置输入数据
	GetOutput() any       // 获取输出数据
}

type UnitTemplate interface {
	Name() string
	Description() string
	// Instantiate 由Storyboard在构建Stage时调用，用于根据模板创建对应的Unit实例（单元）
	Instantiate() Unit
}