workflow

type Activity interface {

	// Name returns the name of the Activity
	Name() string

	// Execute the Activity with the given parameters.
	Execute(ctx Context, parameters map[string]any) (any, error)
}

type ActivityExecutionEvent struct {
	ExecutionID  string
	WorkflowName string
	PathID       string
	StepName     string
	ActivityName string
	Parameters   map[string]any
	Result       any
	StartTime    time.Time
	EndTime      time.Time
	Duration     time.Duration
	Error        error
}

type ActivityExecutor interface {
	// ExecuteActivity runs an activity with automatic logging and checkpointing
	// In the new path-local state system, activities work directly with path state
	ExecuteActivity(ctx context.Context, stepName, pathID string, activity Activity, params map[string]interface{}, pathState *PathLocalState) (result interface{}, err error)
}

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

type ActivityLogEntry struct {
	ID          string                 `json:"id"`
	ExecutionID string                 `json:"execution_id"`
	Activity    string                 `json:"activity"`
	StepName    string                 `json:"step_name"`
	PathID      string                 `json:"path_id"`
	Parameters  map[string]interface{} `json:"parameters"`
	Result      interface{}            `json:"result,omitempty"`
	Error       string                 `json:"error,omitempty"`
	StartTime   time.Time              `json:"start_time"`
	Duration    float64                `json:"duration"`
}

type ActivityLogger interface {
	// LogActivity logs a completed activity
	LogActivity(ctx context.Context, entry *ActivityLogEntry) error

	// GetActivityHistory retrieves activity log for an execution
	GetActivityHistory(ctx context.Context, executionID string) ([]*ActivityLogEntry, error)
}

type ActivityRegistry map[string]Activity

type BaseExecutionCallbacks struct{}

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

type CatchConfig struct {
	ErrorEquals []string `json:"error_equals"`
	Next        string   `json:"next"`
	Store       string   `json:"store,omitempty"`
}

type Checkpoint struct {
	ID           string                 `json:"id"`
	ExecutionID  string                 `json:"execution_id"`
	WorkflowName string                 `json:"workflow_name"`
	Status       string                 `json:"status"`
	Inputs       map[string]interface{} `json:"inputs"`
	Outputs      map[string]interface{} `json:"outputs"`
	Variables    map[string]interface{} `json:"variables"`
	PathStates   map[string]*PathState  `json:"path_states"`
	JoinStates   map[string]*JoinState  `json:"join_states"`
	PathCounter  int                    `json:"path_counter"`
	Error        string                 `json:"error,omitempty"`
	StartTime    time.Time              `json:"start_time,omitzero"`
	EndTime      time.Time              `json:"end_time,omitzero"`
	CheckpointAt time.Time              `json:"checkpoint_at"`
}

type Checkpointer interface {
	// SaveCheckpoint saves the current execution state
	SaveCheckpoint(ctx context.Context, checkpoint *Checkpoint) error

	// LoadCheckpoint loads the latest checkpoint for an execution
	LoadCheckpoint(ctx context.Context, executionID string) (*Checkpoint, error)

	// DeleteCheckpoint removes checkpoint data for an execution
	DeleteCheckpoint(ctx context.Context, executionID string) error
}

type ChildWorkflowExecutor interface {
	// ExecuteSync runs a child workflow synchronously and waits for completion
	ExecuteSync(ctx context.Context, spec *ChildWorkflowSpec) (*ChildWorkflowResult, error)

	// ExecuteAsync starts a child workflow asynchronously and returns immediately
	ExecuteAsync(ctx context.Context, spec *ChildWorkflowSpec) (*ChildWorkflowHandle, error)

	// GetResult retrieves the result of an asynchronous execution
	GetResult(ctx context.Context, handle *ChildWorkflowHandle) (*ChildWorkflowResult, error)
}

type ChildWorkflowExecutorOptions struct {
	WorkflowRegistry WorkflowRegistry
	Activities       []Activity
	Logger           *slog.Logger
	ActivityLogger   ActivityLogger
	Checkpointer     Checkpointer
}

type ChildWorkflowHandle struct {
	ExecutionID  string `json:"execution_id"`
	WorkflowName string `json:"workflow_name"`
}

type ChildWorkflowResult struct {
	Outputs     map[string]interface{} `json:"outputs"`
	Status      ExecutionStatus        `json:"status"`
	ExecutionID string                 `json:"execution_id"`
	Duration    time.Duration          `json:"duration"`
	Error       error                  `json:"error,omitempty"`
}

type ChildWorkflowSpec struct {
	WorkflowName string                 `json:"workflow_name"`
	Inputs       map[string]interface{} `json:"inputs,omitempty"`
	Timeout      time.Duration          `json:"timeout,omitempty"`
	ParentID     string                 `json:"parent_id,omitempty"` // for tracing
	Sync         bool                   `json:"sync"`                // synchronous vs asynchronous
}

type Context interface {

	// workflow.Context embeds the context.Context interface.
	context.Context

	// workflow.Context embeds the VariableContainer interface.
	VariableContainer

	// ListInputs returns a slice containing all input names.
	ListInputs() []string

	// GetInput returns the value of an input variable.
	GetInput(key string) (value any, exists bool)

	// GetLogger returns the logger.
	GetLogger() *slog.Logger

	// GetCompiler returns the script compiler.
	GetCompiler() script.Compiler

	// GetPathID returns the current execution path ID.
	GetPathID() string

	// GetStepName returns the current step name.
	GetStepName() string
}

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

type Each struct {
	Items any    `json:"items"`
	As    string `json:"as,omitempty"`
}

type Edge struct {
	Step      string `json:"step"`
	Condition string `json:"condition,omitempty"`
	Path      string `json:"path,omitempty"`
}

type EdgeMatchingStrategy string

type ErrorOutput struct {
	Error   string      `json:"Error"`
	Cause   string      `json:"Cause"`
	Details interface{} `json:"Details,omitempty"`
}

type ExecuteActivityFunc func(ctx Context, parameters map[string]any) (any, error)

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

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

type ExecutionCallbacks interface {
	// Workflow-level callbacks
	BeforeWorkflowExecution(ctx context.Context, event *WorkflowExecutionEvent)
	AfterWorkflowExecution(ctx context.Context, event *WorkflowExecutionEvent)

	// Path-level callbacks
	BeforePathExecution(ctx context.Context, event *PathExecutionEvent)
	AfterPathExecution(ctx context.Context, event *PathExecutionEvent)

	// Activity-level callbacks
	BeforeActivityExecution(ctx context.Context, event *ActivityExecutionEvent)
	AfterActivityExecution(ctx context.Context, event *ActivityExecutionEvent)
}

type ExecutionContextOptions struct {
	PathLocalState *PathLocalState
	Logger         *slog.Logger
	Compiler       script.Compiler
	PathID         string
	StepName       string
}

type ExecutionOptions struct {
	Workflow           *Workflow
	Inputs             map[string]any
	ActivityLogger     ActivityLogger
	Checkpointer       Checkpointer
	Logger             *slog.Logger
	Formatter          WorkflowFormatter
	ExecutionID        string
	Activities         []Activity
	ScriptCompiler     script.Compiler
	ExecutionCallbacks ExecutionCallbacks
}

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

type ExecutionStatus string

type ExecutionSummary struct {
	ExecutionID  string        `json:"execution_id"`
	WorkflowName string        `json:"workflow_name"`
	Status       string        `json:"status"`
	StartTime    time.Time     `json:"start_time"`
	EndTime      time.Time     `json:"end_time,omitempty"`
	Duration     time.Duration `json:"duration"`
	Error        string        `json:"error,omitempty"`
}

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

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

type Input struct {
	Name        string      `json:"name" yaml:"name"`
	Type        string      `json:"type" yaml:"type"`
	Description string      `json:"description,omitempty" yaml:"description,omitempty"`
	Default     interface{} `json:"default,omitempty" yaml:"default,omitempty"`
}

type JitterStrategy string

type JoinConfig struct {
	// Paths specifies which named paths to wait for. If empty, waits for all active paths.
	Paths []string `json:"paths,omitempty"`

	// Count specifies the number of paths to wait for. If 0, waits for all specified paths.
	Count int `json:"count,omitempty"`

	// PathMappings specifies where to store path data. Supports two syntaxes:
	// 1. Store entire path state: "pathID": "destination"
	//    Example: "pathA": "results.pathA" stores all pathA variables under results.pathA
	// 2. Extract specific variables: "pathID.variable": "destination"
	//    Example: "pathA.result": "extracted.value" stores only pathA.result under extracted.value
	// Supports nested field extraction using dot notation for both variable names and destinations.
	PathMappings map[string]string `json:"path_mappings,omitempty"`
}

type JoinRequest struct {
	StepName    string         `json:"step_name"`
	Config      *JoinConfig    `json:"config"`
	Variables   map[string]any `json:"variables"`
	StepOutputs map[string]any `json:"step_outputs"`
}

type JoinState struct {
	StepName      string      `json:"step_name"`
	WaitingPathID string      `json:"waiting_path_id"` // The single path that's waiting
	Config        *JoinConfig `json:"config"`
	CreatedAt     time.Time   `json:"created_at"`
}

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

type NullActivityLogger struct{}

type NullCheckpointer struct{}

type Options struct {
	Name        string         `json:"name" yaml:"name"`
	Steps       []*Step        `json:"steps" yaml:"steps"`
	Description string         `json:"description,omitempty" yaml:"description,omitempty"`
	Path        string         `json:"path,omitempty" yaml:"path,omitempty"`
	Inputs      []*Input       `json:"inputs,omitempty" yaml:"inputs,omitempty"`
	Outputs     []*Output      `json:"outputs,omitempty" yaml:"outputs,omitempty"`
	State       map[string]any `json:"state,omitempty" yaml:"state,omitempty"`
}

type Output struct {
	Name        string `json:"name" yaml:"name"`
	Variable    string `json:"variable" yaml:"variable"`
	Path        string `json:"path,omitempty" yaml:"path,omitempty"`
	Description string `json:"description,omitempty" yaml:"description,omitempty"`
}

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

type PatchOptions struct {
	Variable string
	Value    any
	Delete   bool
}

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

type PathExecutionEvent struct {
	ExecutionID  string
	WorkflowName string
	PathID       string
	Status       ExecutionStatus
	StartTime    time.Time
	EndTime      time.Time
	Duration     time.Duration
	CurrentStep  string
	StepOutputs  map[string]any
	Error        error
}

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

type PathOptions struct {
	Workflow           *Workflow
	ActivityRegistry   map[string]Activity
	Logger             *slog.Logger
	Formatter          WorkflowFormatter
	Inputs             map[string]any // Initial inputs (readonly)
	Variables          map[string]any // Initial variables (will be copied)
	ActivityExecutor   ActivityExecutor
	UpdatesChannel     chan<- PathSnapshot
	ScriptCompiler     script.Compiler
	ExecutionCallbacks ExecutionCallbacks
}

type PathSnapshot struct {
	PathID      string
	Status      ExecutionStatus
	StepName    string
	StepOutput  any
	NewPaths    []PathSpec
	Error       error
	Timestamp   time.Time
	StartTime   time.Time
	EndTime     time.Time
	JoinRequest *JoinRequest // New field for join requests
}

type PathSpec struct {
	Step      *Step
	Variables map[string]any
	Name      string // Optional name for the path from Edge.Path
}

type PathState struct {
	ID           string          `json:"id"`
	Status       ExecutionStatus `json:"status"`
	CurrentStep  string          `json:"current_step"`
	StartTime    time.Time       `json:"start_time,omitzero"`
	EndTime      time.Time       `json:"end_time,omitzero"`
	ErrorMessage string          `json:"error_message,omitempty"`
	StepOutputs  map[string]any  `json:"step_outputs"`
	Variables    map[string]any  `json:"variables"`
}

type RetryConfig struct {
	ErrorEquals    []string       `json:"error_equals,omitempty"`
	MaxRetries     int            `json:"max_retries,omitempty"`
	BaseDelay      time.Duration  `json:"base_delay,omitempty"`
	MaxDelay       time.Duration  `json:"max_delay,omitempty"`
	BackoffRate    float64        `json:"backoff_rate,omitempty"`
	JitterStrategy JitterStrategy `json:"jitter_strategy,omitempty"`
	Timeout        time.Duration  `json:"timeout,omitempty"`
}

type Step struct {
	Name                 string               `json:"name"`
	Description          string               `json:"description,omitempty"`
	Store                string               `json:"store,omitempty"`
	Activity             string               `json:"activity,omitempty"`
	Parameters           map[string]any       `json:"parameters,omitempty"`
	Each                 *Each                `json:"each,omitempty"`
	Join                 *JoinConfig          `json:"join,omitempty"`
	Next                 []*Edge              `json:"next,omitempty"`
	EdgeMatchingStrategy EdgeMatchingStrategy `json:"edge_matching_strategy,omitempty"`
	Retry                []*RetryConfig       `json:"retry,omitempty"`
	Catch                []*CatchConfig       `json:"catch,omitempty"`
}

type TypedActivity[TParams, TResult any] interface {

	// Name returns the name of the Activity
	Name() string

	// Execute the Activity with the given parameters.
	Execute(ctx Context, parameters TParams) (TResult, error)
}

type TypedActivityAdapter[TParams, TResult any] struct {
	// contains filtered or unexported fields
}

type TypedActivityFunction[TParams, TResult any] struct {
	// contains filtered or unexported fields
}

type VariableContainer interface {

	// SetVariable sets the value of a state variable.
	SetVariable(key string, value any)

	// DeleteVariable deletes a state variable.
	DeleteVariable(key string)

	// ListVariables returns a slice containing all variable names.
	ListVariables() []string

	// GetVariable returns the value of a state variable.
	GetVariable(key string) (value any, exists bool)
}

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

type WorkflowError struct {
	Type    string      `json:"type"`
	Cause   string      `json:"cause"`
	Details interface{} `json:"details,omitempty"`
	Wrapped error       `json:"-"` // Original error being wrapped
}

type WorkflowExecutionEvent struct {
	ExecutionID  string
	WorkflowName string
	Status       ExecutionStatus
	StartTime    time.Time
	EndTime      time.Time
	Duration     time.Duration
	Inputs       map[string]any
	Outputs      map[string]any
	PathCount    int
	Error        error
}

type WorkflowFormatter interface {
	PrintStepStart(stepName string, activityName string)
	PrintStepOutput(stepName string, content any)
	PrintStepError(stepName string, err error)
}

type WorkflowRegistry interface {
	// Register adds a workflow to the registry
	Register(workflow *Workflow) error

	// Get retrieves a workflow by name
	Get(name string) (*Workflow, bool)

	// List returns all registered workflow names
	List() []string
}