attempts

type Attempt struct {
	Attempt      int           `json:"attempt"`
	Model        string        `json:"model"`
	Status       string        `json:"status"`
	Passed       bool          `json:"passed"`
	Steps        int           `json:"steps"`
	InputTokens  int           `json:"input_tokens"`
	OutputTokens int           `json:"output_tokens"`
	CostUSD      float64       `json:"cost_usd"`
	WallTimeMS   int64         `json:"wall_time_ms"`
	Worktree     string        `json:"worktree"`
	Branch       string        `json:"branch"`
	FilesChanged int           `json:"files_changed"`
	Trajectory   string        `json:"trajectory,omitempty"`
	Checks       []CheckResult `json:"checks"`
	Error        string        `json:"error,omitempty"`
	// Removed is true when Keep cleanup deleted the worktree and branch.
	Removed bool `json:"removed,omitempty"`
	// contains filtered or unexported fields
}

type Check struct {
	Command        string `json:"command"`
	TimeoutSeconds int    `json:"timeout_seconds"`
}

type CheckResult struct {
	Command    string `json:"command"`
	ExitCode   int    `json:"exit_code"`
	DurationMS int64  `json:"duration_ms"`
	TimedOut   bool   `json:"timed_out"`
	Skipped    bool   `json:"skipped"`
	Output     string `json:"output"`
}

type CheckRunner interface {
	Command(ctx context.Context, root, command string) *exec.Cmd
}

type Link struct {
	ID          string // attempts group id, shared by every attempt
	Index       int    // 1-based attempt number
	Count       int    // attempts requested
	TraceParent string // W3C traceparent for the child's root span, or ""
}

type Options struct {
	// Models holds one model spec per attempt; its length is N.
	Models []string
	Checks []Check
	// BaseName names the worktrees: attempt i uses <BaseName>-a<i>.
	BaseName string
	// Output is the -o path. Attempt i's trajectory goes to <stem>.a<i><ext>
	// and the report to <stem>.attempts<ext>. Empty means trajectories go to
	// a temporary directory and no report file is written.
	Output    string
	CostLimit float64       // invocation-wide USD budget; 0 = none
	TimeLimit time.Duration // invocation-wide wall-clock budget; 0 = none
	// ExtraArgs are forwarded to every child run (the task among them).
	ExtraArgs []string
	Stderr    io.Writer
	// Stdout receives the winner's stdout (its submission) at the end.
	Stdout io.Writer
	// Redact masks secrets in check output; nil leaves it unchanged.
	Redact func(string) string
	// RunAgent runs one child tinyagent with args. Its stdout is buffered
	// per attempt; its stderr is passed through.
	RunAgent func(ctx context.Context, args []string, stdout, stderr io.Writer) error
	// Now is the clock; nil means time.Now.
	Now func() time.Time
	// RunAll runs every attempt instead of stopping at the first pass. The
	// lowest-numbered passing attempt still wins.
	RunAll bool
	// Keep is KeepAll (or empty), KeepFailed or KeepNone. With a winner,
	// KeepFailed removes passing non-winners and KeepNone removes every
	// non-winner through Remove. Nothing is removed without a winner.
	Keep string
	// Remove deletes one attempt's worktree and branch. Nil disables
	// cleanup.
	Remove func(a Attempt) error
	// Events, when non-nil, receives JSONL events (--json): each child
	// run's stdout events tagged with "attempt", plus attempt.started,
	// attempt.check, attempt.result and one final attempts.result. The
	// winner's stdout is then not replayed to Stdout.
	Events io.Writer
	// AttemptsID, when set, links the children to this group: each child
	// gets it (and its attempt number) in its environment, and it is
	// recorded in the report and events.
	AttemptsID string
	// TraceParent, when set with AttemptsID, is passed to each child as
	// TRACEPARENT so every attempt's spans share the parent's trace.
	TraceParent string
	// CheckEnv, when non-nil, prepares where one attempt's checks run
	// (the run's sandbox, container or command_env) for the attempt's
	// working directory, and returns a release function. An error fails
	// every check of that attempt (fail closed); nil runs checks on the
	// host with tinyagent's environment.
	CheckEnv func(ctx context.Context, root string) (CheckRunner, func(), error)
}

type Report struct {
	SchemaVersion int       `json:"schema_version"`
	AttemptsID    string    `json:"attempts_id,omitempty"`
	Requested     int       `json:"attempts_requested"`
	Winner        *int      `json:"winner"`
	WinnerBranch  string    `json:"winner_branch,omitempty"`
	StopReason    string    `json:"stop_reason"`
	StopOnPass    bool      `json:"stop_on_pass"`
	Keep          string    `json:"keep"`
	Attempts      []Attempt `json:"attempts"`
	TotalCostUSD  float64   `json:"total_cost_usd"`
	TotalWallMS   int64     `json:"total_wall_time_ms"`
}