scheduler

type BatchWaker interface {
	WakeMany(ctx context.Context, targets []WakeTarget) []error
}

type CycleResult struct {
	PendingRuns          int
	ActiveRuns           int
	SessionsScanned      int
	RecoveredLeases      int
	WakeAttempts         int
	WakeSucceeded        int
	WakeFailed           int
	NoMatchRuns          int
	RecentlyNotified     int
	UnclaimableRuns      int
	Paused               bool
	StarvedRuns          int
	SpawnRequested       int
	NeedsAttention       int
	SelectedRunIDs       []string
	NoMatchRunIDs        []string
	RecoveredRunIDs      []string
	StarvedRunIDs        []string
	SpawnRequestedRunIDs []string
	NeedsAttentionRunIDs []string
}

type EscalationActor interface {
	EmitRunStarved(ctx context.Context, work *RunSnapshot, age time.Duration) error
	RequestWorkerSpawn(ctx context.Context, work *RunSnapshot) error
	MarkRunNeedsAttention(ctx context.Context, runID string, diagnostic string) (taskpkg.Run, error)
}

type Option func(*Scheduler)

type PauseStore interface {
	GetSchedulerPause(ctx context.Context) (taskpkg.SchedulerPauseState, error)
}

type RebuildResult struct {
	PendingRuns     int
	ActiveRuns      int
	SessionsScanned int
	ClearedWakeKeys int
	RebuiltAt       time.Time
}

type RunSnapshot struct {
	Task taskpkg.Task
	Run  taskpkg.Run
}

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

type SessionSnapshot struct {
	ID           string
	AgentName    string
	WorkspaceID  string
	Channel      string
	Type         string
	State        string
	Prompting    bool
	Capabilities []string
	CreatedAt    time.Time
}

type SessionSource interface {
	Sessions(ctx context.Context) ([]SessionSnapshot, error)
}

type StarvationStore interface {
	LoadRunStarvation(ctx context.Context, runID string) (taskpkg.RunStarvation, bool, error)
	ListRunStarvation(ctx context.Context) ([]taskpkg.RunStarvation, error)
	UpsertRunStarvation(ctx context.Context, mutation taskpkg.RunStarvationMutation) (taskpkg.RunStarvation, error)
	ClearRunStarvation(ctx context.Context, runID string) error
}

type StarvationThresholds struct {
	FanOutAfter         int
	SpawnAfter          int
	EventAfter          int
	NeedsAttentionAfter int
	MinQueuedAge        time.Duration
}

type Stats struct {
	Cycles            int
	Rebuilds          int
	RecoveredLeases   int
	RecoveryErrors    int
	WakeAttempts      int
	WakeSucceeded     int
	WakeFailed        int
	NoMatchRuns       int
	RecentlyNotified  int
	UnclaimableRuns   int
	StarvedRuns       int
	SpawnRequested    int
	NeedsAttention    int
	LastCycleAt       time.Time
	LastRebuildAt     time.Time
	LastRecoveryError string
	LastWakeError     string
}

type TaskSource interface {
	PendingRuns(ctx context.Context) ([]RunSnapshot, error)
	ActiveRuns(ctx context.Context) ([]taskpkg.Run, error)
	GetRunStatus(ctx context.Context, runID string) (taskpkg.RunStatus, bool, error)
	RecoverExpiredRunLeases(
		ctx context.Context,
		recovery taskpkg.ExpiredLeaseRecovery,
		actor taskpkg.ActorContext,
	) ([]taskpkg.ExpiredLeaseRecoveryResult, error)
}

type WakeTarget struct {
	Work    RunSnapshot
	Session SessionSnapshot
	Reason  string
}

type Waker interface {
	Wake(ctx context.Context, target *WakeTarget) error
}