store

type TaskRecord struct {
	TaskID       string          `json:"task_id"`
	TaskType     string          `json:"task_type"`
	State        string          `json:"status"` // State drives UI rendering ("PENDING_USER", "QUEUED_EXTERNALLY", "COMPLETED")
	RenderConfig json.RawMessage `json:"render_config"`

	// Parent coordinates — used to wake the parent workflow when this task finishes
	ParentWorkflowID string `json:"parent_workflow_id"`
	ParentRunID      string `json:"parent_run_id"`
	ParentNodeID     string `json:"parent_node_id"`

	RootWorkflowID string `json:"root_workflow_id"` // top-level consignment ID

	// Active subtask execution coordinates — used to resume/wake the currently active subtask step via the API.
	// WARNING: Since the store only holds a single set of coordinates, only one subtask can be active at any given time
	// (strictly sequential execution). Parallel/concurrent subtasks inside a single Task Workflow are not supported.
	TaskWorkflowID       string `json:"task_workflow_id"`
	TaskRunID            string `json:"task_run_id"`
	SubTaskNodeID        string `json:"subtask_node_id"`
	ActiveTaskTemplateID string `json:"active_task_template_id,omitempty"`

	// Data holds generic, dynamic task execution state variables.
	Data map[string]any `json:"data"`

	CreatedAt time.Time `json:"created_at"`
	UpdatedAt time.Time `json:"updated_at"`
}

type TaskStore interface {
	SaveTask(context context.Context, record TaskRecord)
	GetTask(context context.Context, taskID string) (TaskRecord, bool)
	GetTaskByWorkflowID(context context.Context, workflowID string) (TaskRecord, bool)
	GetAllTasks(context context.Context, parentWorkflowID string) []TaskRecord
}