error_injection

type ErrorType string

type ExecutionTrace struct {
	// JobID is the identifier of the job being traced.
	JobID string `json:"job_id"`
	// StartTime is when the job execution began.
	StartTime time.Time `json:"start_time"`
	// EndTime is when the job execution completed.
	EndTime time.Time `json:"end_time,omitempty"`
	// Points contains records for all injection points evaluated during execution.
	Points []PointRecord `json:"points,omitempty"`
}

type GlobalInjectionConfig struct {
	// Enabled determines whether global error injection is active.
	Enabled bool `json:"enabled"`
	// Rules contains the global injection rules to apply.
	Rules []InjectionRule `json:"rules,omitempty"`
	// ExcludedPoints lists injection points that should never have errors injected.
	ExcludedPoints []string `json:"excluded_points,omitempty"`
	// GlobalProbability is the default probability for all injection points in chaos mode.
	GlobalProbability float64 `json:"global_probability,omitempty"`
	// PointWeights overrides probability per injection point in chaos mode.
	// Keys are point IDs (e.g., "rm.provision"), values are probabilities (0.0-1.0).
	// Point-specific probability takes precedence over GlobalProbability.
	PointWeights map[string]float64 `json:"point_weights,omitempty"`
}

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

type InjectedError struct {
	// Type is the category of the injected error.
	Type ErrorType `json:"type"`
	// Code is the gRPC/HTTP status code.
	Code string `json:"code"`
	// Message is the rendered error description.
	Message string `json:"message"`
	// Details contains additional structured error information.
	Details map[string]string `json:"details,omitempty"`
}

type InjectionConfig struct {
	// JobID is the identifier of the job this config applies to.
	JobID string `json:"job_id"`
	// Enabled determines whether error injection is active for this job.
	Enabled bool `json:"enabled"`
	// Rules contains the injection rules to apply.
	Rules []InjectionRule `json:"rules,omitempty"`
	// GlobalProbability is the default probability for rules without explicit probability.
	GlobalProbability float64 `json:"global_probability,omitempty"`
}

type InjectionContextKey string

type InjectionPoint struct {
	// ID is a unique identifier in the format "component.action".
	ID string `json:"id"`
	// Component is the pipeline component name.
	Component string `json:"component"`
	// Action is the specific operation within the component.
	Action string `json:"action"`
	// Description provides a human-readable explanation of the injection point.
	Description string `json:"description"`
	// Templates contains pre-defined error templates for this injection point.
	Templates map[ErrorType]*InjectionTemplate `json:"templates,omitempty"`
}

type InjectionRule struct {
	// PointRef references the injection point by ID.
	PointRef string `json:"point_ref"`
	// ErrorType specifies which template to use from the injection point.
	ErrorType ErrorType `json:"error_type"`
	// Probability is the chance (0.0-1.0) that the error will trigger.
	Probability float64 `json:"probability"`
	// Overrides allows customization of template placeholder values.
	Overrides map[string]string `json:"overrides,omitempty"`
}

type InjectionTemplate struct {
	// Type is the category of error this template produces.
	Type ErrorType `json:"type"`
	// Code is the gRPC/HTTP status code for the error.
	Code string `json:"code"`
	// MessageTemplate is a Go template string with {{.variable}} syntax for error message.
	MessageTemplate string `json:"message_template"`
	// Placeholders contains template variables with their default values.
	Placeholders map[string]string `json:"placeholders,omitempty"`
	// DetailsTemplate contains optional structured error details templates.
	DetailsTemplate map[string]string `json:"details_template,omitempty"`
}

type Injector interface {
	// CheckPoint checks if an error should be injected at the given point.
	// Returns nil if no error should be injected, or an InjectedError if one should be.
	CheckPoint(ctx context.Context, pointID string) error

	// GetTrace retrieves the execution trace for a job.
	GetTrace(ctx context.Context, jobID string) *ExecutionTrace

	// GetGlobalConfig returns the current global configuration.
	GetGlobalConfig() *GlobalInjectionConfig

	// SetGlobalConfig updates the global configuration.
	SetGlobalConfig(config *GlobalInjectionConfig) error

	// RenderTemplate renders an error from the template for a given point.
	RenderTemplate(pointID string, errorType ErrorType, overrides map[string]string) (*InjectedError, error)
}

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

type PointRecord struct {
	// PointID is the identifier of the injection point that was evaluated.
	PointID string `json:"point_id"`
	// Timestamp is when the injection point was evaluated.
	Timestamp time.Time `json:"timestamp"`
	// Triggered indicates whether an error was actually injected.
	Triggered bool `json:"triggered"`
	// ContextSnapshot captures the relevant context at the time of evaluation.
	ContextSnapshot map[string]string `json:"context_snapshot,omitempty"`
	// Error contains the injected error if one was triggered.
	Error *InjectedError `json:"error,omitempty"`
	// TemplateUsed is the error type of the template that was applied.
	TemplateUsed ErrorType `json:"template_used,omitempty"`
	// OverridesApplied contains the placeholder overrides that were used.
	OverridesApplied map[string]string `json:"overrides_applied,omitempty"`
	// ProbabilityRoll is the random value that was rolled against the probability.
	ProbabilityRoll float64 `json:"probability_roll"`
}

type TraceStore interface {
	// AppendPoint adds a point record to the trace for the given job ID
	AppendPoint(jobID string, point PointRecord) error

	// Get retrieves the execution trace for the given job ID
	Get(jobID string) (*ExecutionTrace, error)

	// Delete removes the execution trace for the given job ID
	Delete(jobID string) error

	// List returns the most recent execution traces, limited to the specified count
	List(limit int) ([]*ExecutionTrace, error)
}