server

type Accumulator struct {
	Count int32
	Sum   float64
	Min   float64
	Max   float64
}

type AuthenticatedUser struct {
	UserID   int
	Username string
	IsAdmin  bool
}

type Job struct {
	JobID        int32
	WorkerID     int32
	WorkerName   string
	ProviderID   int32
	ProviderName string
	Region       string
	Flavor       string
	// HasGPU is the recruited flavor's has_gpu signal (looked up from
	// the flavor catalog at job-creation time). The provider's Create
	// uses it to swap to the operator's `gpu_image` config when set.
	// false for CPU-only flavors and when the catalog row is missing.
	HasGPU bool
	// Per-recruiter cloud image overrides (NULL when the deploy came
	// from CreateWorker without a recruiter hint, or when the
	// recruiter row left both columns NULL). Format is
	// provider-specific. Precedence at provider Create:
	// gpu_image (if HasGPU) > image > scitq.yaml default.
	Image    *string
	GPUImage *string
	// Per-recruiter swapfile sizing override (see Recruiter.swap_proportion).
	// NULL = provider.Create falls back to cfg.Scitq.SwapProportion.
	// Carried through so the provider's cloud-init can bake the right
	// `-swap` value into the worker's install command.
	SwapProportion *float32
	// Per-recruiter extra block storage (see Recruiter.extra_storage_gb /
	// extra_storage_type). NULL / 0 = no extra volume. OVH only for now;
	// the Azure provider returns an error when ExtraStorageGB > 0.
	ExtraStorageGB   *int32
	ExtraStorageType *string
	Action           rune // "C", "D", "R"
	Retry            int
	Timeout          time.Duration
	FromRecruiter    bool
}

type ObjectiveDefinition struct {
	Formula   string `json:"formula"`
	Direction string `json:"direction"` // "maximize" or "minimize"
}

type ParamSpec struct {
	Name     string `json:"name"`
	Type     string `json:"type"`
	Required bool   `json:"required"`
	// Default and Choices accept whatever JSON type the schema carries —
	// boolean, integer, float, string, etc. The Python side
	// (ParamSpec.schema in python/src/scitq2/param.py) emits native
	// types, not stringified defaults, so a YAML `default: true` round-
	// trips as the JSON literal `true`. Trying to unmarshal that into
	// `string` aborted the whole UploadTemplate with "json: cannot
	// unmarshal bool into Go struct field". Neither field is consumed
	// downstream (UploadTemplate only validates Name/Type and then
	// stores the schema verbatim), so the loose `interface{}` typing
	// has no downside.
	Default interface{}   `json:"default"`
	Help    string        `json:"help"` // Or `json:"description"` if your field is named so
	Choices []interface{} `json:"choices,omitempty"`
}

type QualityDefinition struct {
	Variables  map[string]string     `json:"variables"`            // name -> regex pattern (one capture group)
	Formula    string                `json:"formula,omitempty"`    // single-objective formula (backward compat)
	Objectives []ObjectiveDefinition `json:"objectives,omitempty"` // multi-objective: list of {formula, direction}
}

type QualityResult struct {
	Vars   map[string]float64 `json:"vars"`
	Score  float64            `json:"score"`            // primary score (first objective, or single formula)
	Scores []float64          `json:"scores,omitempty"` // all objective scores (multi-objective only)
}

type StepAgg struct {
	// Task status counters
	Waiting      int32
	Pending      int32
	Accepted     int32
	Running      int32
	Uploading    int32
	Succeeded    int32
	Failed       int32
	ReallyFailed int32          // tasks that have exhausted all retries and are failed
	Retrying     int32          // tasks currently being retried
	RetryingSet  map[int32]bool // set of task IDs currently retrying
	Total        int32

	// Accumulators for durations
	Download   Accumulator
	Upload     Accumulator
	SuccessRun Accumulator
	FailRun    Accumulator

	// RunningTasks maps taskID to start time
	RunningTasks map[int32]time.Time

	// Time bounds (epoch seconds)
	StartTime *int64
	EndTime   *int64
}

type StepAggSnapshot struct {
	Waiting      int32
	Pending      int32
	Accepted     int32
	Running      int32
	Uploading    int32
	Succeeded    int32
	Failed       int32
	ReallyFailed int32
	Retrying     int32
}

type StepCounterSnapshot struct {
	StepId            int32 `json:"stepId"`
	TotalTasks        int32 `json:"totalTasks"`
	WaitingTasks      int32 `json:"waitingTasks"`
	PendingTasks      int32 `json:"pendingTasks"`
	AcceptedTasks     int32 `json:"acceptedTasks"`
	RunningTasks      int32 `json:"runningTasks"`
	UploadingTasks    int32 `json:"uploadingTasks"`
	SuccessfulTasks   int32 `json:"successfulTasks"`
	FailedTasks       int32 `json:"failedTasks"`       // hidden retried parents
	ReallyFailedTasks int32 `json:"reallyFailedTasks"` // visible F only
	RetryingTasks     int32 `json:"retryingTasks"`
	// Accumulators — sent as flat min/max/avg for the UI's duration
	// column. UI already renders these; format matches today's
	// StepStats proto response.
	SuccessRun stepRunStats `json:"successRun"`
	FailedRun  stepRunStats `json:"failedRun"`
	RunningRun stepRunStats `json:"runningRun"`
	Download   stepRunStats `json:"download"`
	Upload     stepRunStats `json:"upload"`
	// RunningTaskStarts is the raw set of running tasks (task_id +
	// run_started_at as fractional epoch seconds). The client rebuilds
	// its local runningByStep map from this list on every snapshot so
	// its 1 s timer computes min/avg/max from an up-to-date set — R
	// transitions after mount are no longer stale.
	RunningTaskStarts []stepRunningTask `json:"runningTaskStarts"`
	StartTime         *int64            `json:"startTime,omitempty"`
	EndTime           *int64            `json:"endTime,omitempty"`
}

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

type TaskUpdateBroadcast struct {
	TaskId   int32
	Status   string
	WorkerId int32
}

type TransitionExtras struct {
	WorkerID     *int32
	FailureClass *string
}

type TransitionRequest struct {
	TaskID int32
	// New status letter: 'P'|'W'|'A'|'C'|'D'|'O'|'R'|'U'|'V'|'S'|'F'.
	NewStatus string
	// Optional guard: only transition if the CURRENT status equals
	// this value. Empty = unconditional. Mirrors the "WHERE status='W'"
	// inline guards several existing sites already use.
	ExpectStatus string
	// Extras applied in the same UPDATE. Nil pointers = leave alone.
	// The struct is intentionally typed rather than a map so a new
	// column touched by a transition forces an API extension rather
	// than sneaking in via a stringly-typed side channel.
	Extras TransitionExtras
}

type TransitionResult struct {
	// OldStatus is the row's status before the UPDATE. Empty when the
	// row didn't exist or the guard (ExpectStatus) didn't match.
	OldStatus string
	// WorkflowID / StepID scraped from the row via the RETURNING
	// clause. Used internally to bill the transition against the
	// right aggregator bucket.
	WorkflowID int32
	StepID     int32
	// Applied is true iff the UPDATE affected a row (guard matched,
	// row existed, row wasn't hidden).
	Applied bool
}