dag

type ExecutionInfo struct {
	ID     string // External ID of the job
	Stdout string // Stdout of the job
	Stderr string // Stderr of the job
	Status string // Status of the job
}

type IOSpec interface {
	NodeName() string // The node ID from the graph spec (not the internal node ID)
	ID() string       // The node's input ID from the graph spec
	CID() string
	Context() string
	Path() string
	IsRoot() bool
}

type Node[T any] interface {
	ID() int32
	Get(context.Context) (NodeRepresentation[T], error)
	AddChild(context.Context, Node[T]) error
	AddParentChildRelationship(context.Context, Node[T]) error
	AddParent(context.Context, Node[T]) error
	AddInput(context.Context, T) error
	AddOutput(context.Context, T) error
	SetMetadata(context.Context, NodeMetadata) error
	SetResults(context.Context, NodeResult) error
}

type NodeMetadata struct {
	CreatedAt time.Time
	StartedAt time.Time
	EndedAt   time.Time
	Status    string // Status of the execution
}

type NodeRepresentation[T any] struct {
	Id          int32        // Keep track of the node's ID, useful during debugging
	Name        string       // Name of the node
	QueueItemID uuid.UUID    // ID of the queue item
	Work        Work[T]      // The work to be done by the node
	Children    []Node[T]    // Children of the node
	Parents     []Node[T]    // Parents of the node
	Inputs      []T          // Input data
	Outputs     []T          // Output data (which is fed into the inputs of its children)
	Metadata    NodeMetadata // Metadata about the node
	Results     NodeResult   // Result of the node
}

type NodeResult struct {
	ID      string // External ID of the execution
	StdOut  string // Stdout of the execution
	StdErr  string // Stderr of the execution
	Skipped bool   // Whether the execution was skipped
}

type NodeSpec[T any] struct {
	OwnerID uuid.UUID
	Name    string
	Work    Work[T]
}

type NodeStore[T any] interface {
	NewNode(context.Context, NodeSpec[T]) (Node[T], error)
	GetNode(context.Context, int32) (Node[T], error)
}

type Status int64

type Work[T any] func(ctx context.Context, inputs []T, statusChan chan NodeResult) []T

type WorkRepository[T any] interface {
	Get(context.Context, int32) (Work[T], error)
	Set(context.Context, int32, Work[T]) error
}