pipelinex

type AIConfig struct {
	Intent      string   `yaml:"intent"`      // 核心意图描述
	Constraints []string `yaml:"constraints"` // 关键约束列表
	Template    string   `yaml:"template"`    // 模板标识
	GeneratedAt string   `yaml:"generatedAt"` // 生成时间
	Version     int      `yaml:"version"`     // 版本号
}

type Adapter interface {
	// Config 适配器配置
	Config(ctx context.Context, config map[string]any) error
}

type Bridge interface {
	// Conn 连接到环境中
	Conn(ctx context.Context, adapter Adapter) (Executor, error)
}

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

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

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

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

type DefaultMetadataStoreFactory struct{}

type Edge interface {
	// Source 返回边的源节点
	Source() Node
	// Target 返回边的目标节点
	Target() Node
	// Expression 返回边的条件表达式（pongo2模板语法）
	// 如果返回空字符串，表示无条件边，总是可以遍历
	Expression() string
	// Evaluate 评估条件表达式，返回bool表示是否通过
	Evaluate(ctx EvaluationContext) (bool, error)
	// ID 返回边的唯一标识符（格式：source->target）
	ID() string
}

type Entry struct {
	Pipeline  string    `json:"pipeline"`
	BuildID   string    `json:"buildId"`
	Node      string    `json:"node"`
	Step      string    `json:"step"`
	Timestamp time.Time `json:"timestamp"`
	Level     Level     `json:"level"`
	Message   string    `json:"message"`
	Output    string    `json:"output"` // 命令标准输出/错误
}

type EvaluationContext interface {
	Get(key string) (any, bool)
	All() map[string]any
	WithNode(node Node) EvaluationContext
	WithPipeline(pipeline Pipeline) EvaluationContext
	WithParams(params map[string]any) EvaluationContext
}

type Event string

type Executor interface {
	// Prepare 准备环境
	Prepare(ctx context.Context) error
	// Destruction 销毁环境
	Destruction(ctx context.Context) error
	// Transfer 传输需要执行的数据，并且反回执行的结果
	Transfer(ctx context.Context, in chan<- any, out <-chan any)
}

type ExecutorConfig struct {
	Type   string                 `yaml:"type"`
	Config map[string]interface{} `yaml:"config"`
}

type Graph interface {
	GraphReader
	//AddVertex 添加顶点
	AddVertex(node Node)
	//AddEdge 添加边
	AddEdge(edge Edge) error
}

type GraphReader interface {
	//Nodes
	Nodes() map[string]Node
	//Edges 返回所有的边
	Edges() []Edge
	//Traversal 遍历图结构
	Traversal(ctx context.Context, evalCtx EvaluationContext, fn TraversalFn) error
}

type HTTPMetadataConfig struct {
	URL     string            `yaml:"url"`
	Method  string            `yaml:"method"`
	Headers map[string]string `yaml:"headers"`
	Timeout string            `yaml:"timeout"`
}

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

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

type Level string

type Listener interface {
	// 处理对应的事件将事件发生的对应的流水线和对应的事件作为参数传入
	Handle(p Pipeline, event Event)
	// 获取当前注册的Event
	Events() []Event
}

type ListeningFn func(p Pipeline)

type LoggingConfig struct {
	Endpoint string            `yaml:"endpoint"`
	Headers  map[string]string `yaml:"headers"`
	Timeout  string            `yaml:"timeout"`
	Retry    int               `yaml:"retry"`
}

type Metadata map[string]any

type MetadataConfig struct {
	Type string                 `yaml:"type"`
	Data map[string]interface{} `yaml:"data"`
}

type MetadataStore interface {
	// Get 获取元数据值
	Get(ctx context.Context, key string) (string, error)
	// Set 设置元数据值
	Set(ctx context.Context, key string, value string) error
	// Delete 删除元数据
	Delete(ctx context.Context, key string) error
	// Close 关闭元数据存储连接
	Close() error
}

type MetadataStoreFactory interface {
	// Create 根据配置创建MetadataStore实例
	Create(config MetadataConfig) (MetadataStore, error)
}

type Node interface {
	//ID 获取节点唯一id
	Id() string
	//PipelineId 获取节点所属的流水线id
	PipelineId() string
	//Status 获取节点状态
	Status() string
	//Get 获取节点属性数据
	Get(key string) string
	// Set 设置节点属性数据
	Set(key string, value any)
}

type NodeConfig struct {
	Executor string                 `yaml:"executor"`
	Image    string                 `yaml:"image"`
	Steps    []Step                 `yaml:"steps"`
	Config   map[string]interface{} `yaml:"Config"`
}

type Pipeline interface {
	//ID 流水线的id
	Id() string
	//GetGraph 返回图结构
	GetGraph() Graph
	//SetGraph 设置图结构
	SetGraph(graph Graph)
	//Status 返回流水线的整体状态
	Status() string
	//SetMetadata 设置元数据
	SetMetadata(store MetadataStore)
	//Metadata 获取元数据
	Metadata() Metadata
	//Listening 流水线执行事件监听设置
	Listening(listener Listener)
	//Done流水线是否执行完成
	Done() <-chan struct{}
	//Run执行流水线
	Run(ctx context.Context) error
	//Notify 执行的步骤通知流水线
	Notify()
	//Cancel 取消流水线
	Cancel()
}

type PipelineConfig struct {
	Version   string                    `yaml:"Version"`
	Name      string                    `yaml:"Name"`
	Metadate  MetadataConfig            `yaml:"Metadate"`
	AI        AIConfig                  `yaml:"AI"`
	Param     map[string]interface{}    `yaml:"Param"`
	Executors map[string]ExecutorConfig `yaml:"Executors"`
	Logging   LoggingConfig             `yaml:"Logging"`
	Graph     string                    `yaml:"Graph"`
	Status    map[string]string         `yaml:"Status"`
	Nodes     map[string]NodeConfig     `yaml:"Nodes"`
}

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

type Pongo2TemplateEngine struct{}

type Pusher interface {
	// Push 推送单条日志
	Push(ctx context.Context, entry Entry) error

	// PushBatch 批量推送
	PushBatch(ctx context.Context, entries []Entry) error

	// Close 关闭连接，刷新缓冲
	Close() error
}

type RedisMetadataConfig struct {
	Host     string `yaml:"host"`
	Port     int    `yaml:"port"`
	DB       int    `yaml:"db"`
	Username string `yaml:"username"`
	Password string `yaml:"password"`
}

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

type Runtime interface {
	//获取流水线状态
	Get(id string) (Pipeline, error)
	//取消运行中的流水线
	Cancel(ctx context.Context, id string) error
	//执行异步流水线
	RunAsync(ctx context.Context, id string, config string, listener Listener) (Pipeline, error)
	//执行同步流水线
	RunSync(ctx context.Context, id string, config string, listener Listener) (Pipeline, error)
	//移除流水线记录
	Rm(id string)
	//runtime已经执行完成
	Done() chan struct{}
	//通知runtime
	Notify(data interface{}) error
	//反回runtime公共
	Ctx() context.Context
	//停止后台处理
	StopBackground()
	// 启动后台
	StartBackground()
	// 设置日志推送器
	SetPusher(pusher Pusher)
	// 设置模板引擎
	SetTemplateEngine(engine TemplateEngine)
}

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

type Step struct {
	Name string `yaml:"name"`
	Run  string `yaml:"run"`
}

type TemplateEngine interface {
	// EvaluateBool 评估模板表达式，返回布尔值
	EvaluateBool(expression string, ctx map[string]any) (bool, error)
	// EvaluateString 评估模板表达式，返回字符串
	EvaluateString(expression string, ctx map[string]any) (string, error)
	// Validate 验证表达式语法是否正确
	Validate(expression string) error
}

type TraversalFn func(ctx context.Context, node Node) error