pkg

type Closure struct {
	HostContext *HostContext
	ExtraFacts  map[string]interface{}
}

type ConcreteModuleInputProvider interface {
	ToCode() string
	GetVariableUsage() []string
	Validate() error
	// HasRevert indicates if this input defines a revert action.
	HasRevert() bool
	// ProvidesVariables returns a list of variable names this input might define (e.g., keys in set_fact).
	ProvidesVariables() []string
}

type FactProvider interface {
	AsFacts() map[string]interface{}
}

type Graph struct {
	RequiredInputs []string
	Tasks          [][]Task
}

type GraphExecutor interface {
	Execute(hostContexts map[string]*HostContext, orderedGraph [][]Task, cfg *config.Config) error
	Revert(executedTasks []map[string]chan Task, hostContexts map[string]*HostContext, cfg *config.Config) error
}

type GraphNode interface {
	String() string
	ToCode() string
}

type Group struct {
	Hosts map[string]*Host       `yaml:"hosts"`
	Vars  map[string]interface{} `yaml:"vars"`
}

type Host struct {
	Name    string
	Host    string `yaml:"host"`
	Vars    map[string]interface{}
	Groups  map[string]string `yaml:"groups"`
	IsLocal bool
}

type HostContext struct {
	Host    *Host
	Facts   *sync.Map
	History *sync.Map
	// contains filtered or unexported fields
}

type IgnoredTaskError struct {
	OriginalErr error
}

type Inventory struct {
	Hosts  map[string]*Host       `yaml:"all"`
	Vars   map[string]interface{} `yaml:"vars"`
	Groups map[string]*Group      `yaml:"groups"`
}

type LocalGraphExecutor struct {
	Runner TaskRunner
}

type Module interface {
	InputType() reflect.Type
	OutputType() reflect.Type
	Execute(params ConcreteModuleInputProvider, c *Closure, runAs string) (ModuleOutput, error)
	Revert(params ConcreteModuleInputProvider, c *Closure, previous ModuleOutput, runAs string) (ModuleOutput, error)
	ParameterAliases() map[string]string
}

type ModuleInput struct {
	// Actual holds the concrete *ShellInput, *CopyInput, etc.
	// Let the standard marshalers handle this field when marshaling the parent Task struct,
	// using the custom Task MarshalJSON.
	Actual ConcreteModuleInputProvider `json:"actual,omitempty" yaml:"actual,omitempty"`
}

type ModuleOutput interface {
	Changed() bool
	String() string
}

type RevertableChange[T comparable] struct {
	Before T
	After  T
}

type Task struct {
	Id           int         `yaml:"id" json:"id"`
	Name         string      `yaml:"name" json:"name"`
	Module       string      `yaml:"module" json:"module"`
	Params       ModuleInput `yaml:"params" json:"params"`
	Validate     string      `yaml:"validate" json:"validate,omitempty"`
	Before       string      `yaml:"before" json:"before,omitempty"`
	After        string      `yaml:"after" json:"after,omitempty"`
	When         string      `yaml:"when" json:"when,omitempty"`
	Register     string      `yaml:"register" json:"register,omitempty"`
	RunAs        string      `yaml:"run_as" json:"run_as,omitempty"`
	IgnoreErrors bool        `yaml:"ignore_errors,omitempty" json:"ignore_errors,omitempty"`
	FailedWhen   string      `yaml:"failed_when,omitempty" json:"failed_when,omitempty"`
	ChangedWhen  string      `yaml:"changed_when,omitempty" json:"changed_when,omitempty"`
	Loop         interface{} `yaml:"loop,omitempty" json:"loop,omitempty"`
	DelegateTo   string      `yaml:"delegate_to,omitempty" json:"delegate_to,omitempty"`
}

type TaskResult struct {
	Output   ModuleOutput
	Error    error // This can now be nil, a normal error, or an IgnoredTaskError
	Closure  *Closure
	Task     Task
	Duration time.Duration
	Status   TaskStatus
	Failed   bool
	Changed  bool
	// ExecutionSpecificOutput can store runner-specific results, e.g., SpageActivityResult for Temporal.
	ExecutionSpecificOutput interface{}
}

type TaskRunner interface {
	ExecuteTask(ctx context.Context, task Task, closure *Closure, cfg *config.Config) TaskResult
	RevertTask(ctx context.Context, task Task, closure *Closure, cfg *config.Config) TaskResult
}

type TaskStatus string