Skip to main content

magi/
daemon.rs

1//! The unattended loop: take the next task, run the graph, record what
2//! happened, take the next one.
3//!
4//! This is what turns magi from a command a human types into something an
5//! agent can hand work to. [`crate::queue`] is the mailbox; this module is the
6//! thing that empties it. Nothing here decides *how* a task is implemented —
7//! that is [`crate::graph`] — it only decides which task runs next, and what a
8//! finished run means for the task that produced it.
9//!
10//! # One run at a time, on purpose
11//!
12//! There is no `--jobs` flag and there will not be one. A single run is
13//! already internally parallel: candidates implement concurrently and judges
14//! rank concurrently, so the machine is not idle while one task is in flight.
15//! The real constraint is not CPU but the agent CLIs' quota, and two graphs at
16//! once doubles the burn rate on exactly the resource whose exhaustion produces
17//! [`RunStatus::Stalled`]. Serialising the loop is what keeps a full backlog
18//! from converting the whole day's quota into a pile of untrustworthy verdicts.
19//!
20//! # A crash is legible, and the loop notices on its own
21//!
22//! The task is written as [`crate::queue::TaskStatus::Running`], with its run
23//! id, *before* the graph starts, and is only rewritten once the run reaches a
24//! terminal status. A daemon killed mid-run therefore leaves the task
25//! `Running` and pointing at the run that was in flight. The alternative —
26//! reverting the task to `Queued` on the way out — would hide the abandoned
27//! run and re-spend its quota on the next poll.
28//!
29//! A task left `Running` forever is not the point, though:
30//! [`crate::queue::TaskStatus::runnable`] never offers it again, so a daemon
31//! that died mid-run would otherwise strand its task for good.
32//! [`reclaim_orphaned_running`] runs on every poll and settles exactly the
33//! tasks no live process is actually driving — proven by [`Queue::claim`]
34//! succeeding rather than by a staleness guess — against whatever their last
35//! run actually became, through the same [`settle`] a live finish uses. A run
36//! that genuinely cannot be read still holds its task for a human; the run's
37//! own report explains how far it got.
38//!
39//! # Retries are bounded
40//!
41//! Every attempt at a task consumes one of [`Opts::max_attempts`], after which
42//! the task is [`crate::queue::TaskStatus::Held`] for a human. The one
43//! exception is a run that ended `Stalled`: the panel collapsed because the
44//! agent CLIs hit their quota, which is a fact about the machine and not about
45//! the task, so it must not spend an attempt. Without that exception a quota
46//! outage would quietly hold the entire backlog, and the operator would come
47//! back to a reset quota and nothing left that the loop is willing to run.
48
49use std::path::{Path, PathBuf};
50use std::sync::Arc;
51use std::sync::atomic::{AtomicBool, Ordering};
52use std::sync::{Mutex, MutexGuard};
53use std::time::Duration;
54
55use anyhow::{Context, Result, bail};
56use jiff::Timestamp;
57use serde::{Deserialize, Serialize};
58use tokio::sync::Notify;
59
60use crate::ask::{self, Questions};
61use crate::clean;
62use crate::conduct::Conductor;
63use crate::config::{Config, MergeMode};
64use crate::graph::Runner;
65use crate::land;
66use crate::queue::{Queue, Task, TaskStatus};
67use crate::run::{QuotaLoss, RunState, RunStatus};
68use crate::triage;
69
70/// On-disk format for [`Status`]. Bumped when a field's meaning changes.
71pub const SCHEMA: u32 = 1;
72
73/// How often the status file is refreshed. A reader treats a status file older
74/// than [`STALE_SECS`] as "no daemon", so the heartbeat has to be brisk enough
75/// that a busy daemon is never mistaken for a dead one.
76pub const HEARTBEAT: Duration = Duration::from_secs(5);
77
78/// How old a heartbeat may be before a reader calls the daemon dead. Six
79/// missed beats: long enough to survive a slow filesystem, short enough that
80/// a crashed daemon is not still reported as running a task.
81///
82/// The single threshold every reader shares — the web UI's `/api/health` and
83/// `magi doctor` both call [`Reading::running`] rather than each comparing
84/// against their own copy of this number, so a crashed daemon cannot look
85/// alive on one screen and dead on another.
86pub const STALE_SECS: i64 = 30;
87
88/// Default queue poll interval.
89pub const POLL: Duration = Duration::from_secs(5);
90
91/// How old a claim has to be before startup sweeps it. Longer than any run
92/// this graph plausibly takes, so a sweep cannot pull a task out from under a
93/// daemon that is merely slow.
94pub const STALE_CLAIM: Duration = Duration::from_secs(6 * 60 * 60);
95
96/// How long a task may sit [`TaskStatus::Running`] with no live daemon's
97/// heartbeat naming it before [`crate::conduct`] is shown it as stalled.
98///
99/// [`reclaim_orphaned_running`] settles most crashes immediately, on every
100/// poll, by attempting the task's own claim: a dead pid is proof enough for
101/// [`sweep_stale_claims`] to drop the lock the same tick, and the very next
102/// claim attempt succeeds. But a lock whose pid cannot be parsed at all — an
103/// empty or corrupt `.lock` file — falls back to [`STALE_CLAIM`]'s six-hour
104/// age instead, since there is nothing else to check (see
105/// [`sweep_stale_claims`]'s own doc). For as long as that lock survives, the
106/// claim keeps failing and `reclaim_orphaned_running` correctly leaves the
107/// task `running` — see
108/// `stalled_tasks_still_reaches_a_task_reclaim_could_not_claim_yet` for
109/// exactly this ordering. `stalled_tasks` is what surfaces that task to the
110/// conductor well before the mechanical six-hour sweep would, and thirty
111/// minutes is comfortably below `STALE_CLAIM` while still being generous
112/// enough that a task merely late to publish its first [`HEARTBEAT`] is
113/// never mistaken for abandoned.
114pub const STALLED_RUNNING: Duration = Duration::from_secs(30 * 60);
115
116/// What the loop is working on, for the status file.
117#[derive(Debug, Clone, Default, PartialEq, Eq, Serialize, Deserialize)]
118#[serde(default)]
119pub struct Current {
120    /// Task id being run.
121    pub task: String,
122    /// Run id the task produced.
123    pub run: String,
124}
125
126/// The daemon's liveness, published to `<home>/daemon.json`.
127///
128/// This is the only interface between the loop and the web UI, which is why it
129/// carries `updated_at` as well as `started_at`: a reader cannot tell a
130/// running daemon from a `SIGKILL`ed one by the file's existence alone, but it
131/// can compare the heartbeat against the clock.
132#[derive(Debug, Clone, Serialize, Deserialize)]
133pub struct Status {
134    /// On-disk format version.
135    pub schema: u32,
136    /// Process id, so a human can find or kill the daemon.
137    pub pid: u32,
138    /// When this process started.
139    pub started_at: Timestamp,
140    /// Last heartbeat.
141    pub updated_at: Timestamp,
142    /// True when the queue has nothing runnable.
143    pub idle: bool,
144    /// Every task and run currently in flight. More than one entry means the
145    /// loop is driving more than one run at once — see
146    /// [`crate::config::Daemon::max_concurrent_runs`]. Empty, not absent, when
147    /// nothing is running, so a reader never has to treat "no field" and "an
148    /// empty list" as two different kinds of idle.
149    pub current: Vec<Current>,
150    /// Tasks that reached a terminal status in this process.
151    pub completed: usize,
152    /// Queue polls since start, so a wedged loop shows up as a frozen count.
153    pub polls: u64,
154}
155
156impl Status {
157    /// A fresh, idle status for this process.
158    #[must_use]
159    pub fn new() -> Self {
160        let now = Timestamp::now();
161        Self {
162            schema: SCHEMA,
163            pid: std::process::id(),
164            started_at: now,
165            updated_at: now,
166            idle: true,
167            current: Vec::new(),
168            completed: 0,
169            polls: 0,
170        }
171    }
172}
173
174impl Default for Status {
175    fn default() -> Self {
176        Self::new()
177    }
178}
179
180/// How the loop should behave.
181#[derive(Debug, Clone)]
182pub struct Opts {
183    /// Repository used by tasks that name none.
184    pub repo: PathBuf,
185    /// Explicit `magi.toml`, instead of the discovered layer stack.
186    pub config: Option<PathBuf>,
187    /// Queue poll interval.
188    pub poll: Duration,
189    /// Attempts a task gets before it is held for a human.
190    pub max_attempts: usize,
191    /// Drain what is runnable now, then return, instead of waiting for more.
192    pub once: bool,
193    /// Merge mode override (`none`, `local`, `pr`); `None` keeps the config's.
194    pub merge: Option<String>,
195    /// Where the janitor's [`crate::clean::fold_orphaned_worktrees`] and
196    /// [`crate::git::worktree_prune`] look for and reclaim worktrees.
197    /// `None` resolves to [`crate::run::default_worktree_root`] - the
198    /// operator's real `~/wt/<repo>` - the same way a run with no
199    /// [`crate::config::Graph::worktree_root`] resolves its own. A caller
200    /// that does not own that directory (a test, an embedding that manages
201    /// worktrees itself) must set this, or every idle tick reclaims worktrees
202    /// out from under whoever actually does.
203    pub worktrees_root: Option<PathBuf>,
204}
205
206impl Default for Opts {
207    fn default() -> Self {
208        Self {
209            repo: PathBuf::from("."),
210            config: None,
211            poll: POLL,
212            max_attempts: 2,
213            once: false,
214            merge: None,
215            worktrees_root: None,
216        }
217    }
218}
219
220/// How many runs a plain `usize` from config may drive concurrently, floored
221/// at one. A `0` in a config file would otherwise stall the loop entirely -
222/// no runnable task could ever start - which is never what an operator who
223/// wrote `0` meant.
224fn max_concurrent(n: usize) -> usize {
225    n.max(1)
226}
227
228/// Where the status file lives.
229#[must_use]
230pub fn status_path() -> PathBuf {
231    crate::run::home().join("daemon.json")
232}
233
234/// Publish the status file for this process.
235pub fn write_status(status: &Status) -> Result<()> {
236    write_status_to(&status_path(), status)
237}
238
239/// Publish a status to an explicit path.
240///
241/// Written to a sibling `.tmp` and renamed, because the web UI reads this file
242/// on every health poll and must never see a half-written one.
243pub fn write_status_to(path: &Path, status: &Status) -> Result<()> {
244    if let Some(parent) = path.parent() {
245        std::fs::create_dir_all(parent).with_context(|| format!("create {}", parent.display()))?;
246    }
247    let body = serde_json::to_string_pretty(status).context("serialize daemon status")?;
248    let tmp = path.with_extension("json.tmp");
249    std::fs::write(&tmp, &body).with_context(|| format!("write {}", tmp.display()))?;
250    std::fs::rename(&tmp, path).with_context(|| format!("replace {}", path.display()))?;
251    Ok(())
252}
253
254/// Delete the status file. Called on the way out so a clean exit reads as
255/// "no daemon" rather than as a daemon whose heartbeat merely stopped.
256pub fn clear_status() {
257    clear_status_at(&status_path());
258}
259
260/// Delete a status file at an explicit path, so the loop's teardown and
261/// [`clear_status`] cannot drift apart: the loop is handed the path it
262/// published to, and a test can watch a temp file disappear.
263fn clear_status_at(path: &Path) {
264    let _ = std::fs::remove_file(path);
265}
266
267/// A cooperative stop, shared with whoever asked the loop to run.
268///
269/// Cloning is how the request travels: [`serve_until`] keeps one handle, the
270/// Ctrl-C listener and the web UI keep others, and every clone points at the
271/// same flag. There is no channel because there is nothing to send — the only
272/// message is "stop", it is idempotent, and a flag cannot be missed by a
273/// receiver that was not listening yet.
274///
275/// The handle also answers the question the operator's screen asks next: a
276/// stop does not take effect until the run in flight has finished, so
277/// [`Stop::finishing`] reports "asked to stop, still working" rather than
278/// leaving a caller to infer it from a heartbeat and hope.
279#[derive(Debug, Clone, Default)]
280pub struct Stop {
281    /// Set once, never cleared: a stop is not something an operator takes back
282    /// half way through, and a clearable flag would let a start racing a stop
283    /// resurrect a loop that is already unwinding.
284    stopped: Arc<AtomicBool>,
285    /// How many runs are in flight, so `finishing` can distinguish a stop
286    /// that has landed from one that is waiting on `execute`. A count, not a
287    /// flag, because more than one run can be in flight at once - see
288    /// [`crate::config::Daemon::max_concurrent_runs`] - and the last one to
289    /// finish is the one that should turn "finishing" off.
290    busy: Arc<std::sync::atomic::AtomicUsize>,
291    /// Wakes the idle wait. Without this a stop would not be seen until the
292    /// poll interval elapsed, and an operator tapping stop on a phone would
293    /// watch a button do nothing for five seconds.
294    wake: Arc<Notify>,
295    /// Handed to the run in flight, so a stop can also mean "park at the next
296    /// node boundary" instead of "finish the whole competition first".
297    pause: crate::graph::Pause,
298}
299
300impl Stop {
301    /// A stop nobody has asked for yet.
302    #[must_use]
303    pub fn new() -> Self {
304        Self::default()
305    }
306
307    /// Ask the loop to stop. Idempotent, and safe to call before the loop
308    /// starts: the flag is checked before the first poll.
309    pub fn stop(&self) {
310        self.stopped.store(true, Ordering::SeqCst);
311        // `notify_one` rather than `notify_waiters` because the loop may not be
312        // parked yet: this stores a permit, so a wait that registers a moment
313        // later returns at once instead of sleeping out the whole interval.
314        self.wake.notify_one();
315    }
316
317    /// Has a stop been asked for?
318    #[must_use]
319    pub fn stopped(&self) -> bool {
320        self.stopped.load(Ordering::SeqCst)
321    }
322
323    /// Has a stop been asked for that has not taken effect yet, because a run
324    /// is still in flight?
325    ///
326    /// This is the state a screen has to be able to show. A stop never abandons
327    /// a run — see [`serve_until`] — so between the tap and the loop's return
328    /// there is a window of tens of minutes in which "running" and "stopped"
329    /// are both misleading answers.
330    #[must_use]
331    pub fn finishing(&self) -> bool {
332        self.stopped() && self.busy_now()
333    }
334
335    /// Ask the loop to stop *and* the run in flight to park at its next node
336    /// boundary.
337    ///
338    /// The plain [`Stop::stop`] never abandons a run, which is right when the
339    /// operator only wants the queue to drain: a competition is tens of
340    /// minutes and its worktrees are paid for. But an operator who wants to
341    /// replace the binary cannot wait out a run that has an hour left, and
342    /// killing the process loses whatever the seats in flight had not written.
343    /// Parking costs at most the node in progress and leaves the run
344    /// resumable.
345    pub fn park(&self) {
346        self.pause.park();
347        self.stop();
348    }
349
350    /// Has a park been asked for?
351    #[must_use]
352    pub fn parking(&self) -> bool {
353        self.pause.parked()
354    }
355
356    /// The pause handle to give a runner.
357    #[must_use]
358    pub fn pause(&self) -> crate::graph::Pause {
359        self.pause.clone()
360    }
361
362    /// Is any run in flight right now?
363    ///
364    /// `finishing` answers "a stop is waiting on a run", which is false until
365    /// someone asks to stop. An upgrade needs the plain question, because it
366    /// is about to be the one asking.
367    #[must_use]
368    pub fn busy_now(&self) -> bool {
369        self.busy.load(Ordering::SeqCst) > 0
370    }
371
372    /// Mark one more run as in flight, for [`Stop::finishing`].
373    fn enter(&self) {
374        self.busy.fetch_add(1, Ordering::SeqCst);
375    }
376
377    /// Mark one run as finished. The last one out is what makes
378    /// [`Stop::busy_now`] false again.
379    fn exit(&self) {
380        self.busy.fetch_sub(1, Ordering::SeqCst);
381    }
382
383    /// Wait out one poll interval, returning early once a stop is asked for.
384    async fn idle(&self, poll: Duration) {
385        tokio::select! {
386            () = tokio::time::sleep(poll) => {}
387            () = self.wake.notified() => {}
388        }
389    }
390}
391
392/// The daemon's published state, read permissively.
393///
394/// This mirrors [`Status`], but is a separate declaration on purpose: every
395/// field defaults, so a status file from an older or newer magi still yields
396/// a usable reading — one this build has never heard of — instead of a parse
397/// error that hides the daemon entirely.
398#[derive(Debug, Clone, Default, Deserialize)]
399#[serde(default)]
400pub struct Reading {
401    /// Format version the daemon claims.
402    pub schema: u32,
403    /// Daemon process id, for an operator who wants to stop it.
404    pub pid: Option<u32>,
405    /// When that process started.
406    pub started_at: Option<Timestamp>,
407    /// Last heartbeat. Absent means the file is unusable, hence not running.
408    pub updated_at: Option<Timestamp>,
409    /// True when the queue had nothing runnable at the last poll.
410    pub idle: bool,
411    /// What the daemon is working on. Empty means idle; more than one entry
412    /// means more than one run is in flight at once.
413    ///
414    /// `deserialize_with` rather than the plain derive: a daemon started
415    /// before this field became a list is still out there writing the old
416    /// shape — a single `{"task":...,"run":...}` object, or its absence —
417    /// on every heartbeat until it is restarted, and a live process reading
418    /// that file during the rollout must still see it as running rather than
419    /// as absent. A bare type change here would fail the whole struct's
420    /// deserialization on a type mismatch, defeating the permissiveness this
421    /// type exists for.
422    #[serde(deserialize_with = "de_current")]
423    pub current: Vec<Current>,
424    /// Tasks this daemon process has finished.
425    pub completed: u64,
426    /// Queue polls this daemon process has made.
427    pub polls: u64,
428}
429
430/// Accept the old single-`Current`-or-absent shape as well as the current
431/// list, so a reader never has to know which build wrote the file.
432fn de_current<'de, D>(deserializer: D) -> std::result::Result<Vec<Current>, D::Error>
433where
434    D: serde::Deserializer<'de>,
435{
436    #[derive(Deserialize)]
437    #[serde(untagged)]
438    enum Shape {
439        Many(Vec<Current>),
440        One(Current),
441    }
442    Ok(
443        Option::<Shape>::deserialize(deserializer)?.map_or_else(Vec::new, |shape| match shape {
444            Shape::Many(v) => v,
445            Shape::One(c) => vec![c],
446        }),
447    )
448}
449
450impl Reading {
451    /// Seconds since the last heartbeat, or `None` when there has never been
452    /// one.
453    #[must_use]
454    pub fn age_secs(&self, now: Timestamp) -> Option<i64> {
455        self.updated_at
456            .map(|at| (now.as_second() - at.as_second()).max(0))
457    }
458
459    /// Whether the loop counts as running: a heartbeat no older than
460    /// [`STALE_SECS`]. The alternative is a reader that claims a task is in
461    /// progress hours after the daemon that owned it was killed.
462    #[must_use]
463    pub fn running(&self, now: Timestamp) -> bool {
464        self.age_secs(now).is_some_and(|secs| secs <= STALE_SECS)
465    }
466}
467
468/// Read `<home>/daemon.json` permissively, or `None` when there is nothing
469/// usable there.
470///
471/// Missing, half-written and unparseable all collapse to `None`, because the
472/// only question a reader asks is whether a daemon is alive, and a file it
473/// cannot read is not evidence that one is.
474#[must_use]
475pub fn read_status(home: &Path) -> Option<Reading> {
476    let body = std::fs::read_to_string(home.join("daemon.json")).ok()?;
477    serde_json::from_str(&body).ok()
478}
479
480/// Every run a live daemon is working on right now.
481///
482/// One definition of liveness, because deleting a task and deleting a run are
483/// both gated on it from both the CLI and the web UI - four callers that must
484/// never disagree about whether the same thing is in flight. A stale heartbeat
485/// reads as "no daemon": that is [`Reading::running`]'s judgement, and a task
486/// left at `running` or a run left at `implementing` by a killed daemon is a
487/// leftover record rather than work in progress. More than one entry once
488/// [`crate::config::Daemon::max_concurrent_runs`] is more than one - a caller
489/// after "the one thing in flight" wants [`is_working_on`] or
490/// [`is_working_on_task`], not this directly.
491#[must_use]
492pub fn current_work(home: &Path, now: Timestamp) -> Vec<Current> {
493    read_status(home)
494        .filter(|reading| reading.running(now))
495        .map(|reading| reading.current)
496        .unwrap_or_default()
497}
498
499/// Whether a live daemon is working on this run at this moment.
500#[must_use]
501pub fn is_working_on(home: &Path, run: &str, now: Timestamp) -> bool {
502    current_work(home, now).iter().any(|c| c.run == run)
503}
504
505/// Whether a live daemon is working on a run whose short id is this one.
506///
507/// For a worktree that has no run record to compare against at all -
508/// [`crate::clean::fold_orphaned_worktrees`]'s whole reason to exist - a full
509/// id is not available to hand to [`is_working_on`]. The short id is: a run's
510/// worktree bay is named after it (see [`crate::run::RunState::worktree_root`]),
511/// and it is exactly the gap between the daemon claiming a task and
512/// `RunState::new` saving the first `run.json` that this exists to protect -
513/// a run genuinely in flight but invisible to a scan of `runs/`.
514#[must_use]
515pub fn is_working_on_short(home: &Path, short: &str, now: Timestamp) -> bool {
516    current_work(home, now)
517        .iter()
518        .any(|c| crate::run::short_of(&c.run) == short)
519}
520
521/// Whether a live daemon is working on this task at this moment.
522#[must_use]
523pub fn is_working_on_task(home: &Path, task: &str, now: Timestamp) -> bool {
524    current_work(home, now).iter().any(|c| c.task == task)
525}
526
527/// Remove claim files whose owner is provably dead, or that have simply
528/// outlived `older_than`, and return the task ids swept.
529///
530/// A daemon killed with `SIGKILL` never runs [`crate::queue::Claim`]'s
531/// destructor, and the orphaned `.lock` file would make its task permanently
532/// unclaimable — the backlog would stop for good at exactly the task that was
533/// in flight when the machine went down.
534///
535/// The pid recorded in the lock is the authority whenever it can be read at
536/// all; age is only a fallback for when it cannot be.
537///
538/// - **A parseable pid wins outright.** [`crate::proc::pid_alive`] decides,
539///   full stop — dead sweeps the lock immediately, regardless of age; alive
540///   protects it, regardless of age. This is what lets a lock be reclaimed in
541///   seconds instead of waiting out [`STALE_CLAIM`]: a lock made 33 minutes
542///   before this daemon even started, next to a `queued` task, no longer has
543///   to sit for six hours before anything notices its owner is gone.
544/// - **A pid that cannot be parsed at all** — an empty or corrupt lock file —
545///   falls back to `older_than`, since there is nothing else to check.
546///
547/// Age must never override a *positive* liveness confirmation. `sweep`
548/// [`poll`]s concurrently with every attempt this daemon itself has spawned —
549/// see [`InFlightGuard`] — not only between them the way a single sequential
550/// loop once did, so a run that legitimately runs longer than `older_than`
551/// (a multi-round review, a long land wait carried across several resumed
552/// attempts) still has this very process's own live pid sitting in its own
553/// lock file on every later sweep. Deciding by age alone in that case would
554/// delete this daemon's own still-valid claim on its own in-flight task,
555/// which [`reclaim_orphaned_running`] would then read as abandoned and hand
556/// to a second attempt — two `Runner`s writing the same `run.json` and the
557/// same worktree at once. `pid_alive` answering "alive" for anything it
558/// cannot determine (a live process, a pid this build cannot check, one
559/// under another account) is exactly what keeps that path from ever
560/// firing on a guess.
561///
562/// [`STALE_CLAIM`] itself stays large: a helper program missing or its
563/// output unreadable must not be license to guess, and the risk of an
564/// unparseable lock outliving a genuinely dead owner is bounded by an order
565/// of magnitude above any plausible run rather than by a positive check.
566///
567/// Runs on every poll, not only at startup — a daemon up for days must keep
568/// noticing a lock some other, now-dead, daemon left behind just as readily
569/// as one it trips over on the way up.
570pub fn sweep_stale_claims(queue: &Queue, older_than: Duration) -> Vec<String> {
571    sweep_stale_claims_with(queue, older_than, crate::proc::pid_alive)
572}
573
574/// [`sweep_stale_claims`] with its process-query boundary supplied by the
575/// caller. This keeps the lock policy testable where process listing is
576/// unavailable, while production still uses the platform query above.
577fn sweep_stale_claims_with<F>(queue: &Queue, older_than: Duration, pid_alive: F) -> Vec<String>
578where
579    F: Fn(u32) -> bool,
580{
581    let this_process = std::process::id();
582    let mut swept: Vec<String> = std::fs::read_dir(queue.root())
583        .into_iter()
584        .flatten()
585        .flatten()
586        .map(|e| e.path())
587        .filter(|p| p.extension().is_some_and(|x| x == "lock"))
588        .filter(|p| {
589            match std::fs::read_to_string(p)
590                .ok()
591                .and_then(|body| body.trim().parse::<u32>().ok())
592            {
593                // This process wrote it and is asking the question right
594                // now, so it is definitionally still alive - settled without
595                // spawning a helper process at all.
596                Some(pid) if pid == this_process => false,
597                Some(pid) => !pid_alive(pid),
598                None => p
599                    .metadata()
600                    .and_then(|m| m.modified())
601                    .and_then(|t| t.elapsed().map_err(std::io::Error::other))
602                    .is_ok_and(|age| age >= older_than),
603            }
604        })
605        .filter(|p| std::fs::remove_file(p).is_ok())
606        .filter_map(|p| {
607            p.file_stem()
608                .and_then(|s| s.to_str())
609                .map(std::borrow::ToOwned::to_owned)
610        })
611        .collect();
612    swept.sort_unstable();
613    swept
614}
615
616/// Is `task` stalled: [`TaskStatus::Running`], past [`STALLED_RUNNING`], with
617/// no live daemon's heartbeat naming it? Deterministic — no model call, and
618/// the exact test [`stalled_tasks`] uses to decide what `crate::conduct` is
619/// shown.
620fn is_stalled(task: &Task, home: &Path, now: Timestamp) -> bool {
621    task.status == TaskStatus::Running
622        && (now.as_second() - task.updated_at.as_second()) >= STALLED_RUNNING.as_secs() as i64
623        && !is_working_on_task(home, &task.id, now)
624}
625
626/// Every task [`is_stalled`] right now — "止まったタスク" in
627/// `crate::conduct`'s vocabulary.
628fn stalled_tasks(queue: &Queue, home: &Path, now: Timestamp) -> Vec<Task> {
629    queue
630        .list()
631        .into_iter()
632        .filter(|t| is_stalled(t, home, now))
633        .collect()
634}
635
636/// Runnable tasks a dependency can still be set on — "runnable なタスク" in
637/// `crate::conduct`'s vocabulary. Deliberately `Queued` only, not
638/// `Failed`-and-so-also-runnable: a task that already attempted and lost
639/// belongs in [`finished_tasks`], where the question is a recovery, not a
640/// dependency.
641fn queued_tasks(queue: &Queue) -> Vec<Task> {
642    queue
643        .list()
644        .into_iter()
645        .filter(|t| t.status == TaskStatus::Queued)
646        .collect()
647}
648
649/// `Failed`/`Held` tasks nobody has decided a recovery for yet — "終わった
650/// タスク" in `crate::conduct`'s vocabulary.
651fn finished_tasks(queue: &Queue) -> Vec<Task> {
652    queue
653        .list()
654        .into_iter()
655        .filter(|t| matches!(t.status, TaskStatus::Failed | TaskStatus::Held))
656        .collect()
657}
658
659/// Deterministically resolve `Task::blocked_by`: a dependency task that
660/// reached `Done`, or a question that was answered, is removed — no model
661/// involved, on every poll. An answered question's content is copied onto
662/// the task ([`Task::record_answer`]) before its id is dropped, so it
663/// reaches the next `crate::conduct` prompt and the next run's instruction
664/// (see [`instruction_for`]) rather than only clearing the block.
665///
666/// A `blocked_by` id that names neither an existing task nor an existing
667/// question - `magi task rm` (or an operator by hand) deleted it while this
668/// task was still waiting - is caught before any of that: it can never
669/// become `Done` or `Answered`, so the ordinary loop below would otherwise
670/// leave the task `blocked` forever with nothing to notice. Such a task is
671/// quarantined to a machine hold instead ([`crate::queue::missing_blocker_hold_reason`]),
672/// which puts it in front of `crate::triage::run_once`'s own walk the next
673/// time it runs - see that module's doc for why the choice is "ask a human",
674/// never "assume the missing dependency was satisfied and unblock anyway".
675fn resolve_blockers(queue: &Queue, questions: &Questions) {
676    for listed in queue.list() {
677        if listed.status != TaskStatus::Blocked || listed.blocked_by.is_empty() {
678            continue;
679        }
680        let Ok(_claim) = queue.claim(&listed.id) else {
681            continue;
682        };
683        let Ok(mut task) = queue.get(&listed.id) else {
684            continue;
685        };
686        if task.status != TaskStatus::Blocked {
687            continue;
688        }
689        let missing = crate::queue::missing_blockers(queue, questions, &task.blocked_by);
690        if !missing.is_empty() {
691            task.hold_machine(Some(crate::queue::missing_blocker_hold_reason(
692                &task.blocked_by,
693                &missing,
694            )));
695            record(queue, &mut task);
696            continue;
697        }
698        let mut changed = false;
699        for id in task.blocked_by.clone() {
700            if let Ok(dep) = queue.get(&id) {
701                if dep.status == TaskStatus::Done {
702                    task.unblock(&id);
703                    changed = true;
704                }
705                continue;
706            }
707            if let Ok(q) = questions.get(&id)
708                && q.status == ask::QuestionStatus::Answered
709            {
710                let answer = match &q.answer {
711                    Some(ask::Answer::Choice(c) | ask::Answer::Text(c)) => c.clone(),
712                    None => String::new(),
713                };
714                task.record_answer(q.summary.clone(), answer);
715                task.unblock(&id);
716                changed = true;
717            }
718        }
719        if changed {
720            record(queue, &mut task);
721        }
722    }
723}
724
725/// Retire an unanswered conductor question after its task no longer refers to
726/// it. Conductor questions use the task id in `Question::run`, so run-based
727/// cleanup cannot observe a manual release or completion.
728///
729/// Restricted to `Question::node == crate::conduct::NODE`: an ordinary run's
730/// own question also carries a `run`, and a run id that happens to collide
731/// with some task's id is not this loop's business — only a conductor
732/// question actually uses the task id that way. One `Questions::list()` scan
733/// is taken up front and matched against the in-memory task set, rather than
734/// calling `Questions::open_for` (a full disk scan on its own) once per task.
735fn reconcile_task_questions(queue: &Queue, questions: &Questions) {
736    let tasks = queue.list();
737    let by_id: std::collections::BTreeMap<&str, &Task> =
738        tasks.iter().map(|t| (t.id.as_str(), t)).collect();
739    let referenced: std::collections::BTreeSet<&str> = tasks
740        .iter()
741        .flat_map(|task| task.blocked_by.iter().map(String::as_str))
742        .collect();
743
744    for mut question in questions.list() {
745        if !question.status.open() || question.node != crate::conduct::NODE {
746            continue;
747        }
748        // Keep questions a task still names, including when the reference
749        // moved to a dependent task.
750        if referenced.contains(question.id.as_str()) {
751            continue;
752        }
753        let Some(task) = by_id.get(question.run.as_str()) else {
754            continue;
755        };
756        question.abandon(format!(
757            "task {} no longer waits for this answer",
758            task.short()
759        ));
760        if let Err(e) = questions.put(&mut question) {
761            tracing::warn!(
762                "could not retire question {} for task {}: {e:#}",
763                question.short(),
764                task.short()
765            );
766        }
767    }
768}
769
770/// What a finished run tells the queue about the task it came from.
771///
772/// A struct rather than a fourth and fifth boolean argument: the two flags
773/// answer different questions about the same run, and a call site passing
774/// `(…, true, false)` is one transposition away from refunding attempts
775/// forever.
776#[derive(Debug, Clone, Copy)]
777pub struct Verdict {
778    /// Where the graph stopped.
779    pub status: RunStatus,
780    /// The run opened a pull request.
781    pub left_pr: bool,
782    /// At least one seat was lost to a rate limit.
783    pub quota_hit: bool,
784    /// The run parked at a node boundary because it was asked to.
785    pub parked: bool,
786    /// The run never produced a single candidate a judge could look at.
787    ///
788    /// Distinct from `quota_hit`: a run can lose a seat to a rate limit and
789    /// still have another candidate worth judging, in which case the loss was
790    /// not the reason nothing came of the run. This is `true` only when the
791    /// implement wave ended with nothing viable at all.
792    pub no_viable_candidates: bool,
793}
794
795/// Record a finished run against the task it came from.
796///
797/// Kept pure and separate from the loop because this mapping *is* the retry
798/// policy, and a policy that can only be exercised by spawning a graph is a
799/// policy nobody checks. The table:
800///
801/// | run status                           | task becomes        | attempt spent |
802/// |---------------------------------------|---------------------|---------------|
803/// | parked at a boundary                  | `Failed` (requeued) | **no**        |
804/// | `Merged`, `Ready`                      | `Done`               | yes          |
805/// | `Stalled`, quota hit                   | `Failed` (requeued) | **no**        |
806/// | `Failed`, quota hit, no viable cand.   | `Failed` (requeued) | **no**        |
807/// | `Stalled`, no quota                    | `Failed`, or `Held`  | yes          |
808/// | `Blocked` with a PR                    | `Held`               | yes          |
809/// | `Blocked`, `Failed` otherwise          | `Failed`, or `Held`  | yes          |
810/// | `VerifiedNoop`                          | `Held`               | yes          |
811/// | anything non-terminal                  | `Failed`, or `Held`  | yes          |
812///
813/// The `VerifiedNoop` row is independent of the `Failed`-quota row above it,
814/// deliberately: every candidate agreeing there is nothing to write is not a
815/// machine fact about a rate limit, it is an unverified claim about the
816/// *task* that a human still has to check — see [`RunStatus::VerifiedNoop`]'s
817/// own doc and [`Task::handed_off`]. `Held` rather than `Done` on purpose: the
818/// claim could be wrong (a misread instruction, a stale check), and closing
819/// the task automatically on an implementer's say-so would be the exact
820/// failure mode task 391f's own audit was raised to avoid. `attempt spent` is
821/// `yes` here for the same reason it is on the `Blocked`-with-a-PR row just
822/// above, which settles through the same [`Task::handed_off`]: `Held` is not
823/// `Failed`-and-requeued, so nothing retries this task on the same unverified
824/// claim regardless of whether the one already-spent attempt is refunded, and
825/// [`Task::release`] resets the count to zero anyway the moment a human looks
826/// at the evidence and lets it run again.
827///
828/// The `Stalled`-quota and `Failed`-quota rows are the ones worth reading
829/// twice, together. A quorum lost to rate limits is a property of the machine
830/// and not of the task, so the attempt is refunded and a reset quota picks
831/// the work up where it stopped — and that is just as true when every
832/// implement seat lost the same race and `after_implement` bails with nothing
833/// to judge, which surfaces as `Failed` rather than `Stalled` but is the same
834/// machine fact. The `no_viable_candidates` guard is what keeps that row
835/// narrow: a `Failed` run that produced a real candidate which then lost for
836/// some other reason still spends the attempt, exactly like the quorum lost
837/// to judges that answered with the wrong shape is ordinary flakiness, and
838/// refunding *that* takes the bound off the retry loop entirely: run e633
839/// stalled with `quota: []` after two judges wrote unusable JSON, was
840/// refunded, and the next attempt paid for a fresh hour-long implement wave
841/// before it could fail the same way. `max_attempts` exists precisely so
842/// that cannot repeat forever.
843///
844/// A non-terminal status means `execute` returned while the graph was still
845/// mid-flight, which is a bug rather than a verdict; it is treated as a
846/// failure so that a task cannot loop on it either.
847///
848/// `left_pr` splits the `Blocked` row, and it is the difference between a run
849/// that failed and a run that finished into a gate. See [`Task::handed_off`].
850pub fn settle(task: &mut Task, verdict: Verdict, detail: &str, max_attempts: usize) {
851    // A parked run is the operator's own doing, and its work is intact on
852    // disk. The task goes back in line with its attempt refunded so the next
853    // loop resumes the same run - which `one_task` prefers over competing
854    // again - and so that swapping the binary a few times cannot exhaust a
855    // budget meant for agents that actually misbehaved.
856    if verdict.parked {
857        task.stall(detail);
858        return;
859    }
860    match verdict.status {
861        RunStatus::Merged | RunStatus::Ready => task.succeed(),
862        RunStatus::Stalled if verdict.quota_hit => task.stall(detail),
863        RunStatus::Failed if verdict.quota_hit && verdict.no_viable_candidates => {
864            task.stall(detail)
865        }
866        RunStatus::Stalled | RunStatus::Failed => task.fail(detail, max_attempts),
867        RunStatus::Blocked if verdict.left_pr => task.handed_off(detail),
868        RunStatus::Blocked => task.fail(detail, max_attempts),
869        RunStatus::VerifiedNoop => task.handed_off(detail),
870        other => task.fail(
871            format!(
872                "the graph stopped at `{}` without reaching a terminal status: {detail}",
873                label(other)
874            ),
875            max_attempts,
876        ),
877    }
878}
879
880/// [`settle`], plus attaching the run's own [`diagnostic`] excerpt once the
881/// task ends up held.
882///
883/// The one place [`attempt`] (a live finish) and [`reclaim`] (recovering one a
884/// dead daemon never got back to) share this, so the two cannot drift into
885/// disagreeing about which held tasks get a diagnostic.
886fn settle_and_diagnose(
887    task: &mut Task,
888    verdict: Verdict,
889    detail: &str,
890    max_attempts: usize,
891    state: &RunState,
892) {
893    settle(task, verdict, detail, max_attempts);
894    if task.status == TaskStatus::Held {
895        task.diagnostic = diagnostic(state);
896    }
897}
898
899/// Reconcile a task left at [`TaskStatus::Running`] by a daemon that never
900/// got back to [`settle`] for it — a crash, a `SIGKILL`, or a run carried on
901/// by some other means entirely, like a manual `magi run` resume that
902/// finishes the graph outside the queue's bookkeeping.
903///
904/// Pure and separate from [`reclaim_orphaned_running`] for the same reason
905/// `settle` is separate from `attempt`: a task recovered this way must land
906/// exactly where a live daemon would have put it — the same policy table,
907/// not a second one that quietly drifts from it — and that is only checkable
908/// without spawning a real run.
909fn reclaim(task: &mut Task, last_run: Option<RunState>, max_attempts: usize) {
910    match last_run {
911        Some(state) => {
912            let verdict = Verdict {
913                status: state.status,
914                left_pr: state.pr.is_some(),
915                quota_hit: !state.quota.is_empty(),
916                parked: state.parked,
917                no_viable_candidates: state.viable().is_empty(),
918            };
919            let detail = format!(
920                "recovered a `running` task whose daemon never recorded the outcome: {}",
921                describe(&state)
922            );
923            settle_and_diagnose(task, verdict, &detail, max_attempts, &state);
924        }
925        None => {
926            let why = "task was `running` with no live daemon and no readable \
927                       run to recover; held for a human to check what happened";
928            task.last_error = Some(why.to_owned());
929            // The phone shows `hold_reason`, so a task held by the machine
930            // says why there too and not only in `last_error`.
931            task.hold_machine(Some(why.to_owned()));
932        }
933    }
934}
935
936/// Find every task left at `running` that no live process is actually
937/// driving, and settle each one against whatever its last run became.
938///
939/// # Why a claim is proof, not a guess
940///
941/// [`poll`] takes a task's [`Queue::claim`] *before* [`Task::start`] writes
942/// `running`, and the guard is held for the task's whole time in that status:
943/// `attempt` does not return, and the loop does not move past the scope
944/// holding the claim, until the run has settled. So a `running` task whose
945/// lock is gone cannot have a live owner — this process or any other —
946/// without needing a staleness threshold or a pid check the way
947/// [`sweep_stale_claims`] does for the narrower case of a lock left next to a
948/// task that never got as far as `running` at all. Taking the claim here is
949/// the whole test: it either fails, because something really does hold it
950/// and the task is left alone, or it succeeds, which is the proof — and it is
951/// kept for the rest of the decision so nothing else can start a competing
952/// run while this one is being written.
953///
954/// Called on every poll, not only at startup, for the reason
955/// [`sweep_stale_claims`] now is too: a daemon that has been up for days must
956/// keep noticing this, not only on the one morning it happened to restart.
957fn reclaim_orphaned_running(queue: &Queue, max_attempts: usize) -> Vec<String> {
958    let mut reclaimed = Vec::new();
959    for listed in queue.list() {
960        if listed.status != TaskStatus::Running {
961            continue;
962        }
963        let Ok(_claim) = queue.claim(&listed.id) else {
964            continue;
965        };
966        // Re-read under the claim: a release or an edit landed by a human
967        // between the listing above and the claim just taken must not be
968        // clobbered by a decision based on the stale copy.
969        let Ok(mut task) = queue.get(&listed.id) else {
970            continue;
971        };
972        if task.status != TaskStatus::Running {
973            continue;
974        }
975        let last_run = task.runs.last().and_then(|id| RunState::load(id).ok());
976        // `execute` normally abandons a run's own open questions the moment
977        // `status` lands somewhere non-resumable (see `graph::Runner::settle_questions`),
978        // but a daemon that crashed *inside* that path - mid `land`'s CI wait,
979        // say - can leave a `run.json` already at `Merged`/`Ready`/`Failed`
980        // with the question still `open`, because the process died before
981        // reaching that call. `reclaim` itself stays pure on purpose (see its
982        // own doc), so the same cleanup runs here instead, against the run
983        // this reclaim is already reading. `settle_run` costs nothing when
984        // `execute` already got there first.
985        if let Some(state) = &last_run
986            && let Err(e) = ask::Questions::open().settle_run(&state.id, state.status)
987        {
988            tracing::warn!("abandon questions for {}: {e:#}", state.id);
989        }
990        reclaim(&mut task, last_run, max_attempts);
991        record(queue, &mut task);
992        reclaimed.push(task.id.clone());
993    }
994    reclaimed
995}
996
997/// Find every run whose `run.json` is provably dead — every seat it still
998/// lists as [`crate::run::RunState::active`] has overrun its own timeout, and
999/// [`crate::run::RunState::liveness`] reads [`crate::run::Liveness::Dead`],
1000/// not merely "no daemon claims it" — and fail it, clearing the leftover
1001/// active seats so the run stops reading as `implementing` (or whichever
1002/// node) forever.
1003///
1004/// [`reclaim_orphaned_running`] settles the *task* a dead daemon left
1005/// `running`, using whatever `run.json` already says — but nothing in that
1006/// path, nor in [`reclaim`], ever writes back to the run itself (`reclaim`
1007/// stays pure on purpose, see its own doc), so a `run.json` a killed process
1008/// never got back to sits exactly where it was left: `active` full of seats
1009/// nobody will ever answer for, `status` stuck on whatever node was in
1010/// flight. `magi show` already tells an operator this in prose (`no live
1011/// daemon claims this run right now`); this is what makes that fact durable
1012/// on disk, the same way a task's own `TaskStatus::Running` does not get to
1013/// stay stuck once nothing is driving it.
1014///
1015/// Runs on every poll, not only at startup, for the reason
1016/// [`sweep_stale_claims`] and [`reclaim_orphaned_running`] already are: a
1017/// daemon up for days must keep noticing a run some other, now-dead, daemon
1018/// left behind just as readily as one it trips over on the way up.
1019///
1020/// Walks `home.join("runs")` directly and reads each `run.json` on its own,
1021/// rather than the process-global [`RunState::load`] / [`crate::run::list_ids`] —
1022/// the same reason [`crate::clean`]'s housekeeping passes take an explicit
1023/// `runs` directory instead: `home` here is a parameter precisely so a test
1024/// can point it away from the operator's real history (see [`drive`]'s own
1025/// doc), and a scan that fell through to the global home anyway would walk
1026/// whichever directory some *other* process or test pinned into that
1027/// `OnceLock` first — mutating runs this call was never handed.
1028fn reclaim_abandoned_runs(home: &Path, now: Timestamp) -> Vec<String> {
1029    reclaim_abandoned_runs_with(
1030        home,
1031        now,
1032        crate::proc::pid_status,
1033        crate::proc::process_started_at,
1034    )
1035}
1036
1037/// [`reclaim_abandoned_runs`] with its `driver_pid` liveness/identity queries
1038/// supplied by the caller — mirrors [`sweep_stale_claims_with`], which exists
1039/// for the identical reason: this sweep's real damage (wiping a run's active
1040/// seats and failing it) has to be provable against an injected answer in a
1041/// test, not just the real process table.
1042fn reclaim_abandoned_runs_with<F, G>(
1043    home: &Path,
1044    now: Timestamp,
1045    query: F,
1046    identity: G,
1047) -> Vec<String>
1048where
1049    F: Fn(u32) -> Option<bool>,
1050    G: Fn(u32) -> Option<String>,
1051{
1052    let mut abandoned = Vec::new();
1053    for entry in std::fs::read_dir(home.join("runs"))
1054        .into_iter()
1055        .flatten()
1056        .flatten()
1057    {
1058        let id = entry.file_name().to_string_lossy().into_owned();
1059        if !crate::run::is_run_id(&id) {
1060            continue;
1061        }
1062        // Unreadable is `clean::fold_due`'s problem, not this one's — see
1063        // that module's docs for why a run this cannot parse is left alone
1064        // rather than guessed at. A different schema number is not that: this
1065        // touches only `status` and `active`, never a field whose meaning a
1066        // schema bump changed, so an old record's values serve this exactly
1067        // as well as a current one's (see `clean::read_state`'s own doc for
1068        // the same reasoning applied to folding).
1069        let Ok(body) = std::fs::read_to_string(entry.path().join("run.json")) else {
1070            continue;
1071        };
1072        let Ok(mut state) = serde_json::from_str::<RunState>(&body) else {
1073            continue;
1074        };
1075        if state.status.done() || !state.active_all_overrun(now) {
1076            continue;
1077        }
1078        // Not `!is_working_on(..)` alone: that is only "no *daemon* claims
1079        // it", which is also the normal, healthy shape of a manual `magi
1080        // run` / `magi review` sharing this same home — this scan walks
1081        // every run on disk, not only ones this daemon itself started. Such
1082        // a run's active seats can legitimately sit past their own timeout
1083        // for a little while (the CLI finishing up, its result still being
1084        // collected) without the process driving it having died. `liveness`
1085        // is what actually tells the two apart, by corroborating
1086        // `driver_pid` against the process it names — see its own doc. Only
1087        // its strongest, provable answer licenses wiping this run's active
1088        // seats and failing it out from under whatever is still running it.
1089        let daemon_claims = is_working_on(home, &id, now);
1090        if state.liveness_with(daemon_claims, &query, &identity) != crate::run::Liveness::Dead {
1091            continue;
1092        }
1093        state.abandon("daemon");
1094        if let Err(e) = state.save_under(home) {
1095            tracing::warn!("could not persist abandoned run {id}: {e:#}");
1096            continue;
1097        }
1098        // The seat that asked is gone for good now, exactly like any other
1099        // door `graph::Runner::settle_questions` closes the moment `status`
1100        // lands somewhere non-resumable - see that method's own doc. Nothing
1101        // else reaches this one before the next `janitor()` startup pass
1102        // (`clean::abandon_settled_questions`), and a daemon that stays up
1103        // for days must not leave an open question badging the operator
1104        // until it happens to restart.
1105        if let Err(e) = Questions::at(home.join("questions")).settle_run(&id, state.status) {
1106            tracing::warn!("abandon questions for {id}: {e:#}");
1107        }
1108        abandoned.push(id);
1109    }
1110    abandoned
1111}
1112
1113/// Run the loop until Ctrl-C, or until the queue drains with [`Opts::once`].
1114///
1115/// A thin wrapper over [`serve_until`] with a stop nothing but Ctrl-C ever
1116/// sets, so there is one loop body rather than two that drift apart the first
1117/// time the retry policy changes on only one of them.
1118pub async fn serve(opts: Opts) -> Result<()> {
1119    serve_until(opts, Stop::new()).await
1120}
1121
1122/// [`serve`], but stopping when `stop` is set as well as on Ctrl-C.
1123///
1124/// Neither a signal nor a `stop` abandons a run in flight. Killing the graph
1125/// mid-node leaves worktrees, branches and agent sessions behind, and every
1126/// agent call already paid for is lost; finishing the run costs the operator a
1127/// wait and saves them a cleanup. A stop therefore only sets a flag: the
1128/// current `execute` runs to its terminal status, the task's outcome is
1129/// recorded, and only then does the loop return. That window is what
1130/// [`Stop::finishing`] is for. An operator who genuinely wants the run dead
1131/// still has a second Ctrl-C, which the runtime turns into a process kill —
1132/// and the task left `Running` then tells the next daemon, and the next human,
1133/// where to look.
1134///
1135/// While the queue is empty the stop is honoured within one wakeup rather than
1136/// one poll interval: the wait is a `select!` against [`Stop`]'s notify, so a
1137/// caller that taps stop does not sit through the remainder of a sleep.
1138pub async fn serve_until(opts: Opts, stop: Stop) -> Result<()> {
1139    let signal = {
1140        let stop = stop.clone();
1141        tokio::spawn(async move {
1142            if tokio::signal::ctrl_c().await.is_ok() {
1143                stop.stop();
1144                tracing::info!("shutdown requested; a run in flight will be finished first");
1145            }
1146        })
1147    };
1148
1149    let worktrees_root = opts
1150        .worktrees_root
1151        .clone()
1152        .unwrap_or_else(crate::run::default_worktree_root);
1153    let outcome = drive(
1154        &opts,
1155        &Queue::open(),
1156        &status_path(),
1157        &crate::run::home(),
1158        &worktrees_root,
1159        &stop,
1160    )
1161    .await;
1162
1163    signal.abort();
1164    outcome
1165}
1166
1167/// The loop proper: setup, poll, teardown, with the queue and the status file
1168/// supplied rather than discovered.
1169///
1170/// All three of `home`, `worktrees_root` and the queue/status paths are
1171/// parameters rather than resolved here, for the same reason:
1172/// [`crate::run::home`] is process-global and its override is a `OnceLock`,
1173/// so a unit test that pinned it would fight every other test in the binary,
1174/// and a loop that resolved its own worktree bay could only be exercised
1175/// against the operator's real `~/wt/<repo>` - publishing over a live
1176/// daemon's status file, claiming tasks out of a live backlog, and, since
1177/// [`janitor`] runs on every idle tick, reclaiming worktrees out from under
1178/// whatever the operator actually has on disk.
1179async fn drive(
1180    opts: &Opts,
1181    queue: &Queue,
1182    status_file: &Path,
1183    home: &Path,
1184    worktrees_root: &Path,
1185    stop: &Stop,
1186) -> Result<()> {
1187    // The status file is a *snapshot*, not a stream of events: a reader only
1188    // ever wants the latest values, and every tick rewrites the whole file
1189    // anyway. A shared `Mutex<Status>` therefore says exactly what is meant,
1190    // while an mpsc channel would force the loop to re-send unchanged fields on
1191    // every heartbeat — or the heartbeat to keep its own shadow copy of them —
1192    // for no gain. The lock is only ever held across a field assignment, never
1193    // across an await.
1194    let status = Arc::new(Mutex::new(Status::new()));
1195    write_status_to(status_file, &lock(&status)).context("publish the daemon status file")?;
1196    let beat = tokio::spawn(heartbeat(Arc::clone(&status), status_file.to_path_buf()));
1197
1198    // Read once at startup, not per task: how many runs this loop drives at
1199    // once is a property of the machine running it, not of whichever
1200    // repository a given task happens to name - see
1201    // `Config::daemon.max_concurrent_runs`'s doc for why that is a machine
1202    // fact in the same sense the agent roster is.
1203    let daemon_cfg = prepare(&opts.repo, opts)
1204        .map(|c| c.daemon)
1205        .unwrap_or_default();
1206    let concurrency = max_concurrent(daemon_cfg.max_concurrent_runs);
1207
1208    tracing::info!(
1209        "magi serve: queue {} (poll {}s, {} attempts per task, {} run(s) at once{})",
1210        queue.root().display(),
1211        opts.poll.as_secs(),
1212        opts.max_attempts,
1213        concurrency,
1214        if daemon_cfg.pause_for_interrupts {
1215            ", interrupts enabled"
1216        } else {
1217            ""
1218        }
1219    );
1220
1221    // `--once` drains an already-idle queue without reaching the idle wait,
1222    // but must still perform the startup cleanup.
1223    janitor(&opts.repo, opts, home, worktrees_root).await;
1224
1225    let outcome = poll(
1226        opts,
1227        queue,
1228        &status,
1229        home,
1230        worktrees_root,
1231        stop,
1232        DispatchLimits {
1233            max_concurrent: concurrency,
1234            pause_for_interrupts: daemon_cfg.pause_for_interrupts,
1235        },
1236    )
1237    .await;
1238
1239    beat.abort();
1240    clear_status_at(status_file);
1241    outcome
1242}
1243
1244/// Refresh the status file on a fixed tick.
1245///
1246/// Separate from the loop because a run takes tens of minutes: a status file
1247/// written only between tasks would look stale for the whole of every run, and
1248/// a reader would report the daemon dead exactly while it was busiest.
1249async fn heartbeat(status: Arc<Mutex<Status>>, path: PathBuf) {
1250    loop {
1251        tokio::time::sleep(HEARTBEAT).await;
1252        let snapshot = {
1253            let mut guard = lock(&status);
1254            guard.updated_at = Timestamp::now();
1255            guard.clone()
1256        };
1257        if let Err(e) = write_status_to(&path, &snapshot) {
1258            // A failed heartbeat must not take the daemon down: the loop is the
1259            // product, the status file is only the window onto it.
1260            tracing::warn!("could not refresh the daemon status file: {e:#}");
1261        }
1262    }
1263}
1264
1265/// Whether a task's last run is sitting in `land`'s merge-approval wait, and
1266/// if so, whether that wait is over.
1267#[derive(Debug, Clone, Copy, PartialEq, Eq)]
1268enum LandResume {
1269    /// The task's last run is not parked on a land approval; schedule it
1270    /// like any other candidate.
1271    NotLanding,
1272    /// Parked in `land`, waiting on a question nobody has answered yet.
1273    /// Left alone: attempting it now would only re-observe the same pull
1274    /// request and park again, spending a `gh` call on a decision that has
1275    /// not changed since the last time this was checked.
1276    StillWaiting,
1277    /// Parked in `land`, and the question is settled - answered or
1278    /// abandoned. Resuming this is the one kind of candidate that must not
1279    /// wait on a free [`Config::daemon`] concurrency slot: see [`poll`].
1280    Ready,
1281}
1282
1283/// Classify a runnable candidate by whether it is parked on a land-merge
1284/// approval. Read-only - no claim taken, nothing written - so it is cheap
1285/// enough to call on every candidate, every poll.
1286fn land_resume_state(task: &Task) -> LandResume {
1287    let Some(run_id) = task.runs.last() else {
1288        return LandResume::NotLanding;
1289    };
1290    let Ok(state) = RunState::load(run_id) else {
1291        return LandResume::NotLanding;
1292    };
1293    if state.status != RunStatus::Landing || !state.parked {
1294        return LandResume::NotLanding;
1295    }
1296    let store = ask::Questions::open();
1297    let waiting = store
1298        .list()
1299        .into_iter()
1300        .filter(|q| &q.run == run_id && q.node == land::APPROVAL_NODE)
1301        .max_by(|a, b| a.id.cmp(&b.id));
1302    let Some(mut q) = waiting else {
1303        return LandResume::Ready;
1304    };
1305    if !q.status.open() {
1306        return LandResume::Ready;
1307    }
1308    // `ask::ask_and_wait`'s own deadline is what used to retire a question
1309    // nobody ever answered; land's approval bypasses that wait entirely (see
1310    // `land::approval_gate`), so the same deadline has to be enforced here
1311    // instead, or `graph.answer_timeout` silently stops meaning anything for
1312    // a land approval and a run can sit `StillWaiting` forever with nobody
1313    // told to look at it.
1314    let timeout = Duration::from_secs(state.config.graph.answer_timeout);
1315    let elapsed = Timestamp::now().as_second() - q.asked_at.as_second();
1316    if elapsed >= 0 && elapsed as u64 >= timeout.as_secs() {
1317        q.abandon(format!(
1318            "no answer within {}s of asking",
1319            timeout.as_secs().max(1)
1320        ));
1321        // If this can't be persisted, do not treat the wait as settled on a
1322        // guess: fall through and try again next poll.
1323        if store.put(&mut q).is_ok() {
1324            return LandResume::Ready;
1325        }
1326    }
1327    LandResume::StillWaiting
1328}
1329
1330/// How often the loop rechecks for new work while something it already
1331/// started is still running, rather than sleeping out the whole
1332/// [`Opts::poll`] interval.
1333///
1334/// Short on purpose: this is what lets a land-merge approval that comes back
1335/// while another task is mid-competition be noticed and resumed within a
1336/// fraction of a second, not within the next multi-second poll.
1337const RECHECK_WHILE_BUSY: Duration = Duration::from_millis(200);
1338
1339/// How often [`poll`] rechecks the shared build cache against its cap at a
1340/// boundary between runs (see [`maybe_prune_cache_between_runs`]), instead of
1341/// waiting for the queue to run dry.
1342///
1343/// A queue that never empties means the `janitor` call at the bottom of this
1344/// loop's fully-idle branch can go unreached for as long as the backlog
1345/// lasts. Five minutes is far below a single gate's own 1200s timeout, so a
1346/// cache that started the day at its 10 GiB cap cannot grow anywhere near the
1347/// 81.8 GiB an idle-only check let it reach before this existed, and it is
1348/// well above the cost of a `dir_size` walk over a multi-gigabyte cache, so a
1349/// backlog of short tasks does not pay for that walk on every poll.
1350const CACHE_CHECK_INTERVAL_SECS: u64 = 5 * 60;
1351
1352/// Frees one attempt's concurrency slot - `Stop`'s busy count and its entry
1353/// in `Status::current` - on drop, so both are released even if the attempt
1354/// panics rather than returning.
1355///
1356/// A `Drop` impl rather than statements written after the `.await` it
1357/// guards: a panic unwinds straight past code placed "after" a call, and
1358/// `Runner::execute`'s chain reaches deep enough into agent-output parsing
1359/// that ruling a panic out there is not a bet this loop can make. Without
1360/// this, one panicking run would leave [`Stop::busy_now`] stuck `true`
1361/// forever - the idle branch in [`poll`], and with it the janitor, would
1362/// never run again - and a ghost entry in `Status::current` naming a task
1363/// nothing is still working on.
1364struct InFlightGuard<'a> {
1365    status: &'a Arc<Mutex<Status>>,
1366    stop: &'a Stop,
1367    task_id: &'a str,
1368}
1369
1370impl Drop for InFlightGuard<'_> {
1371    fn drop(&mut self) {
1372        lock(self.status).current.retain(|c| c.task != self.task_id);
1373        self.stop.exit();
1374    }
1375}
1376
1377/// State of [`poll`]'s own interrupt-scheduling sequence - see
1378/// [`crate::config::Daemon::pause_for_interrupts`]. Advanced once per tick by
1379/// [`advance_interrupt`] and consulted by [`interrupt_gate`], both pure and
1380/// both kept free of `Task`'s non-identity fields on purpose: every decision
1381/// here turns only on task ids and which ones are in flight, so the "never
1382/// more than one run at once" and "exactly one resume" invariants can be
1383/// pinned down with a plain `#[test]`, no `Runner`, no tokio, no fixture
1384/// queue - which is exactly the coverage this feature's first two attempts
1385/// were missing.
1386///
1387/// Deliberately in-memory only, not written to disk anywhere: a daemon
1388/// restart mid-sequence loses track of which run it had asked to park and
1389/// which task was meant to run first, and simply falls back to `Idle` -
1390/// see [`drive`]'s own setup. The parked run itself is not lost - it is
1391/// sitting in the queue exactly like any other resumable, interrupted task,
1392/// `RunStatus::resumable` and [`Task::interrupt`] both intact on disk - it
1393/// just resumes on the ordinary priority order rather than guaranteed to go
1394/// first. Giving that guarantee a crash-proof memory would mean a new queue
1395/// field and a recovery ordering to go with it, which is exactly the
1396/// complexity this feature's constraints rule out for the one property that
1397/// actually matters: at most one run, ever, at once.
1398#[derive(Debug, Clone, PartialEq, Eq)]
1399enum Interrupt {
1400    /// No interrupt sequence in progress. Ordinary dispatch applies.
1401    Idle,
1402    /// `interrupt_task` is runnable and exactly one other run is in flight;
1403    /// `parked` names that one task's id. Dispatch is withheld from
1404    /// everyone, including `interrupt_task` itself, until it has left
1405    /// flight - only then is `interrupt_task` let through.
1406    ///
1407    /// `parked` is a `Vec` rather than a bare id for symmetry with
1408    /// [`Interrupt::Running`] and [`Interrupt::Resuming`], but
1409    /// [`advance_interrupt`]'s own `Idle` branch only ever starts a sequence
1410    /// when exactly one run is in flight, so it is guaranteed to hold
1411    /// exactly one entry in practice - see that branch's own doc for why
1412    /// more than one is deliberately never attempted.
1413    Parking {
1414        parked: Vec<String>,
1415        interrupt_task: String,
1416    },
1417    /// `interrupt_task` is dispatched and in flight alone. Dispatch is
1418    /// withheld from everyone until it leaves flight - merged, failed, held,
1419    /// it makes no difference - at which point the sequence moves to
1420    /// [`Interrupt::Resuming`], never straight back to [`Interrupt::Idle`]:
1421    /// going straight to `Idle` would hand `parked` back to ordinary
1422    /// priority-order dispatch, where a higher-priority task filed in the
1423    /// meantime could start ahead of it.
1424    Running {
1425        parked: Vec<String>,
1426        interrupt_task: String,
1427    },
1428    /// `interrupt_task` left flight; `parked` still names the one run this
1429    /// sequence owes a resume. Dispatch is withheld from everyone except
1430    /// that task - see [`interrupt_gate`] - so the resume this feature
1431    /// promises is never raced by, or run alongside, an unrelated candidate.
1432    /// Ends the moment it is seen in flight, or - see `advance_interrupt`'s
1433    /// own doc on abandonment - the moment it is no longer runnable at all.
1434    Resuming { parked: Vec<String> },
1435}
1436
1437/// One tick of the interrupt scheduler's own state machine. Pure: `in_flight`
1438/// and `runnable` are read-only snapshots of this tick's reality, and the
1439/// only side effect the caller still owes the world is asking whichever
1440/// `Pause` handles `parked` names to actually park - see [`poll`]'s own call
1441/// site.
1442///
1443/// `runnable` only has to carry `id` and `interrupt`; the whole [`Task`] is
1444/// accepted rather than a narrower type because that is what [`poll`] already
1445/// has on hand from [`runnable`], and building a second, smaller list on
1446/// every tick just to satisfy this signature would cost more than it proves.
1447///
1448/// Abandonment: [`Interrupt::Parking`] and [`Interrupt::Resuming`] both fall
1449/// back to a task they are waiting on no longer being [`runnable`] - held,
1450/// blocked, deleted, or finished by some other means entirely, all of which
1451/// an operator can do to a task sitting in the queue with no claim on it at
1452/// all, at any moment, interrupt sequence or not. Without this check the
1453/// sequence would wait forever for a dispatch that can never come, and
1454/// `interrupt_gate` would withhold every other task in the queue right along
1455/// with it - a single `magi task hold` on the wrong id turning into a
1456/// daemon that never dispatches anything again.
1457fn advance_interrupt(state: Interrupt, in_flight: &[String], runnable: &[Task]) -> Interrupt {
1458    match state {
1459        Interrupt::Idle => {
1460            // Not just "something to interrupt": exactly one thing. More
1461            // than one run in flight only happens above the default
1462            // `max_concurrent_runs = 1`, and `parked` guarantees "exactly
1463            // one resume, never run alongside anything else" only because
1464            // it is only ever seeded with exactly one id - see
1465            // `Interrupt::Resuming`'s own doc on why releasing more than one
1466            // parked id back to ordinary dispatch cannot be made safe
1467            // against that same setting's own extra concurrency slots.
1468            // Waiting here for the herd to settle to one is the
1469            // simplification this feature's own constraints ask for rather
1470            // than a second concurrency model to reconcile with the first.
1471            if in_flight.len() != 1 {
1472                return Interrupt::Idle;
1473            }
1474            match runnable.iter().find(|t| t.interrupt) {
1475                Some(t) => Interrupt::Parking {
1476                    parked: in_flight.to_vec(),
1477                    interrupt_task: t.id.clone(),
1478                },
1479                None => Interrupt::Idle,
1480            }
1481        }
1482        Interrupt::Parking {
1483            parked,
1484            interrupt_task,
1485        } => {
1486            if in_flight.iter().any(|id| parked.contains(id)) {
1487                // Still waiting for what was in flight to actually stop.
1488                Interrupt::Parking {
1489                    parked,
1490                    interrupt_task,
1491                }
1492            } else if in_flight.contains(&interrupt_task) {
1493                Interrupt::Running {
1494                    parked,
1495                    interrupt_task,
1496                }
1497            } else if runnable.iter().any(|t| t.id == interrupt_task) {
1498                // The parked run(s) are gone, but the interrupt task has not
1499                // been dispatched yet on this tick - `interrupt_gate` is
1500                // what lets it through next.
1501                Interrupt::Parking {
1502                    parked,
1503                    interrupt_task,
1504                }
1505            } else {
1506                // The interrupt task itself is no longer runnable - see this
1507                // function's own doc on abandonment. The parked run(s) still
1508                // get their guaranteed resume; there is simply no interrupt
1509                // to run ahead of them any longer.
1510                Interrupt::Resuming { parked }
1511            }
1512        }
1513        Interrupt::Running {
1514            parked,
1515            interrupt_task,
1516        } => {
1517            if in_flight.contains(&interrupt_task) {
1518                Interrupt::Running {
1519                    parked,
1520                    interrupt_task,
1521                }
1522            } else {
1523                // The interrupt task's own run reached a terminal status,
1524                // whichever one - this is the *only* trigger that moves the
1525                // sequence on, driven straight off the same in-flight
1526                // bookkeeping `poll` already reaps every tick, not a second,
1527                // independent poll of anything.
1528                Interrupt::Resuming { parked }
1529            }
1530        }
1531        Interrupt::Resuming { parked } => {
1532            if in_flight.iter().any(|id| parked.contains(id)) {
1533                // One of the parked runs has been dispatched - the resume
1534                // this sequence owed is fulfilled. Whatever else is left in
1535                // `parked` (ordinarily nothing, at the default concurrency
1536                // of one) rejoins ordinary priority-order dispatch, same as
1537                // any other runnable task.
1538                Interrupt::Idle
1539            } else if runnable.iter().any(|t| parked.contains(&t.id)) {
1540                Interrupt::Resuming { parked }
1541            } else {
1542                // Abandonment (see this function's own doc): nothing left in
1543                // `parked` is even runnable any longer.
1544                Interrupt::Idle
1545            }
1546        }
1547    }
1548}
1549
1550/// [`Interrupt`], but with [`crate::config::Daemon::pause_for_interrupts`]
1551/// folded in: disabled, the sequence can never leave [`Interrupt::Idle`], so
1552/// a task marked [`Task::interrupt`] on a daemon that has not opted in is
1553/// indistinguishable from any other runnable task - exactly the "off does
1554/// nothing" this feature promises.
1555fn advance_interrupt_tick(
1556    enabled: bool,
1557    state: Interrupt,
1558    in_flight: &[String],
1559    runnable: &[Task],
1560) -> Interrupt {
1561    if !enabled {
1562        return Interrupt::Idle;
1563    }
1564    advance_interrupt(state, in_flight, runnable)
1565}
1566
1567/// Which of this tick's runnable candidates the interrupt sequence actually
1568/// allows to be dispatched. Pure, and separate from [`advance_interrupt`] so
1569/// each half is assertable on its own: this is the half that keeps a
1570/// competition and an interrupt from ever running at the same moment.
1571fn interrupt_gate(state: &Interrupt, in_flight: &[String], candidates: Vec<Task>) -> Vec<Task> {
1572    match state {
1573        Interrupt::Idle => candidates,
1574        Interrupt::Parking {
1575            parked,
1576            interrupt_task,
1577        } => {
1578            if in_flight.iter().any(|id| parked.contains(id)) {
1579                Vec::new()
1580            } else {
1581                candidates
1582                    .into_iter()
1583                    .filter(|t| &t.id == interrupt_task)
1584                    .collect()
1585            }
1586        }
1587        Interrupt::Running { .. } => Vec::new(),
1588        // At most one: even if `parked` names more than one id (more than
1589        // one run was in flight when the sequence began, only possible
1590        // above the default `max_concurrent_runs = 1`), only the first match
1591        // is offered. Capping this to a single candidate - not merely to
1592        // `parked`'s own ids - is what makes "exactly one resume, never two
1593        // dispatched together" true regardless of how many ordinary slots
1594        // happen to be free this tick.
1595        Interrupt::Resuming { parked } => candidates
1596            .into_iter()
1597            .find(|t| parked.contains(&t.id))
1598            .into_iter()
1599            .collect(),
1600    }
1601}
1602
1603/// The daemon-loop knobs [`poll`] needs from [`crate::config::Daemon`],
1604/// bundled into one parameter so `poll`'s own signature stays readable -
1605/// see [`drive`]'s call site for where these are actually read.
1606struct DispatchLimits {
1607    /// How many *ordinary* candidates run at once. See
1608    /// [`crate::config::Daemon::max_concurrent_runs`].
1609    max_concurrent: usize,
1610    /// See [`crate::config::Daemon::pause_for_interrupts`].
1611    pause_for_interrupts: bool,
1612}
1613
1614/// Which semaphore, if any, dispatching a candidate should draw its permit
1615/// from.
1616///
1617/// Pure and separate from [`poll`]'s loop body for the same reason
1618/// [`advance_interrupt`] is: the choice between "skip the ordinary pool
1619/// entirely", "spend the one urgent slot" and "spend an ordinary slot" is
1620/// exactly the policy this feature adds, and a policy only exercisable by
1621/// running the whole loop is a policy nobody checks.
1622#[derive(Debug, Clone, Copy, PartialEq, Eq)]
1623enum PermitKind {
1624    /// A land-merge resume (see [`LandResume::Ready`]): bypasses every slot.
1625    /// Checked first - a task can be both a land resume and marked
1626    /// [`Task::urgent`], and the resume's own "must not queue behind
1627    /// anything" guarantee takes precedence.
1628    None,
1629    /// [`Task::urgent`]: the one extra slot in `urgent_sem`, spent instead of
1630    /// (never in addition to trying) the ordinary pool. This is what lets
1631    /// an urgent task dispatch while every ordinary slot is already checked
1632    /// out, and what stops a second urgent task from opening a third run: it
1633    /// waits on this same one-slot semaphore rather than falling through to
1634    /// the ordinary one.
1635    Urgent,
1636    /// The ordinary `max_concurrent_runs` pool - unaffected by either of the
1637    /// above.
1638    Ordinary,
1639}
1640
1641/// [`PermitKind`] for one candidate. `priority` is [`LandResume::Ready`]'s
1642/// own boolean, already computed by the caller from [`land_resume_state`].
1643fn permit_kind(priority: bool, urgent: bool) -> PermitKind {
1644    if priority {
1645        PermitKind::None
1646    } else if urgent {
1647        PermitKind::Urgent
1648    } else {
1649        PermitKind::Ordinary
1650    }
1651}
1652
1653/// Poll the queue until stopped, factored out so [`drive`] owns only setup and
1654/// teardown and cannot skip the teardown on an early return.
1655///
1656/// `limits.max_concurrent` bounds how many *ordinary* candidates run at once,
1657/// see [`crate::config::Daemon::max_concurrent_runs`]. A run parked on a
1658/// land approval that has since been answered is dispatched outside that
1659/// bound the moment [`land_resume_state`] reports it [`LandResume::Ready`]:
1660/// the whole point of parking there is that it must not queue behind
1661/// whatever else the loop happens to be running, even at the default of one.
1662/// A [`Task::urgent`] candidate is exempt from the same bound the same way,
1663/// through its own one-slot `urgent_sem` (see [`permit_kind`]). Every one of
1664/// these exemptions is still subject to the interrupt gate below, urgent
1665/// included: a land-resume or urgent candidate is exactly as much "something
1666/// else running" as an ordinary one from the interrupt sequence's point of
1667/// view, and letting either slip through while a run is being parked, or
1668/// while the interrupt task itself has the floor, is precisely the second
1669/// run 75dd's own "at most one run, ever, at once" guarantee must never
1670/// allow - see [`Interrupt`]'s own doc.
1671async fn poll(
1672    opts: &Opts,
1673    queue: &Queue,
1674    status: &Arc<Mutex<Status>>,
1675    home: &Path,
1676    worktrees_root: &Path,
1677    stop: &Stop,
1678    limits: DispatchLimits,
1679) -> Result<()> {
1680    let DispatchLimits {
1681        max_concurrent,
1682        pause_for_interrupts,
1683    } = limits;
1684    // Only consulted by `once`, where a task that just failed is still
1685    // `runnable` and would otherwise be picked up again inside the same drain.
1686    // In the long-running mode a later poll retrying a failed task is the point,
1687    // and the attempt counter is what bounds it.
1688    let mut attempted: Vec<String> = Vec::new();
1689    let sem = Arc::new(tokio::sync::Semaphore::new(max_concurrent));
1690    // One extra, permanent slot for `--urgent` tasks (see `Task::urgent`),
1691    // entirely separate from `sem`: an urgent candidate must dispatch
1692    // alongside whatever `sem` already has checked out, never by waiting for
1693    // one of those ordinary slots to free up and never by growing
1694    // `max_concurrent_runs` itself. Sized at one, not unbounded - see
1695    // `permit_kind`'s own doc - so a second `--urgent` task queues behind the
1696    // first on this same slot rather than opening a third run.
1697    let urgent_sem = Arc::new(tokio::sync::Semaphore::new(1));
1698    // A quota hit is a fact about the machine, not the task that happened to
1699    // surface it, and every other *ordinary* candidate is no less likely to
1700    // hit the same wall - see the warning below. A land-merge resume is
1701    // exempt: it is a human decision finishing, not a fresh competition, and
1702    // must not sit out a quota cooldown it did not cause.
1703    let quota_cooldown_until: Arc<Mutex<Option<Timestamp>>> = Arc::new(Mutex::new(None));
1704    let mut inflight: tokio::task::JoinSet<()> = tokio::task::JoinSet::new();
1705    let mut conductor = Conductor::new();
1706    // See `maybe_prune_cache_between_runs`'s own doc: this is the cache check
1707    // a congested queue would otherwise starve of the fully-idle branch below.
1708    let mut cache_last_checked: Option<Timestamp> = None;
1709    // See `Interrupt`'s own doc: in-memory only, advanced once per tick.
1710    let mut interrupt = Interrupt::Idle;
1711    // The `Pause` handed to each dispatched candidate's own `Runner` - see
1712    // `attempt`'s new parameter - kept here so the tick that decides to park
1713    // a run for an interrupt can reach that specific run's handle and no
1714    // other's. Pruned to whatever is still in flight at the top of every
1715    // tick, so a finished attempt's handle does not linger.
1716    let mut interrupt_pauses: std::collections::HashMap<String, crate::graph::Pause> =
1717        std::collections::HashMap::new();
1718
1719    while !stop.stopped() {
1720        lock(status).polls += 1;
1721
1722        // Reap whatever finished since the last tick without blocking on
1723        // anything still running. `InFlightGuard` already released the slot
1724        // even if the spawned attempt panicked; this only surfaces that it
1725        // happened, since a panic swallowed here otherwise leaves no trace.
1726        while let Some(result) = inflight.try_join_next() {
1727            if let Err(e) = result {
1728                tracing::error!("a spawned attempt did not finish cleanly: {e}");
1729            }
1730        }
1731
1732        let swept = sweep_stale_claims(queue, STALE_CLAIM);
1733        if !swept.is_empty() {
1734            tracing::warn!(
1735                "swept {} stale claim(s) left behind by an earlier daemon: {}",
1736                swept.len(),
1737                swept.join(", ")
1738            );
1739        }
1740        // Capture stalled work before reclaiming it. A dead daemon's ordinary
1741        // lock is swept and reclaimed in this same poll, but the conductor
1742        // must still see that it was stranded rather than only its mechanical
1743        // terminal state.
1744        let now = Timestamp::now();
1745
1746        // No run this daemon spawned is mid-compile right now, whether or
1747        // not another candidate is about to start - see
1748        // `maybe_prune_cache_between_runs`'s own doc for why this cannot
1749        // wait for the queue to run dry.
1750        if !stop.busy_now() {
1751            maybe_prune_cache_between_runs(
1752                &opts.repo,
1753                opts,
1754                home,
1755                stop,
1756                &mut cache_last_checked,
1757                now,
1758            )
1759            .await;
1760        }
1761
1762        let stalled = stalled_tasks(queue, home, now);
1763        let stalled_ids: std::collections::BTreeSet<_> =
1764            stalled.iter().map(|task| task.id.clone()).collect();
1765        let reclaimed = reclaim_orphaned_running(queue, opts.max_attempts);
1766        if !reclaimed.is_empty() {
1767            tracing::warn!(
1768                "reclaimed {} task(s) left `running` by a daemon that never \
1769                 recorded the outcome: {}",
1770                reclaimed.len(),
1771                reclaimed.join(", ")
1772            );
1773        }
1774        let abandoned_runs = reclaim_abandoned_runs(home, now);
1775        if !abandoned_runs.is_empty() {
1776            tracing::warn!(
1777                "failed {} run(s) left behind by a killed process, past every \
1778                 active seat's own timeout: {}",
1779                abandoned_runs.len(),
1780                abandoned_runs.join(", ")
1781            );
1782        }
1783
1784        // `home`, not `ask::Questions::open()`'s own process-global default:
1785        // `poll` is handed its home explicitly precisely so a test can point
1786        // it elsewhere, the same reason `Queue::at` and the status file path
1787        // are parameters rather than resolved here - see `drive`'s own doc.
1788        let questions = Questions::at(home.join("questions"));
1789
1790        // Deterministic: no model, run before the conductor sees anything so
1791        // its input reflects the queue's current, already-resolved state.
1792        resolve_blockers(queue, &questions);
1793        reconcile_task_questions(queue, &questions);
1794
1795        // The conductor gets one look per cycle, right before the loop takes
1796        // its next task, and only when there is something new to look at -
1797        // see `Conductor::worth_a_look`'s own doc for why "stalled is
1798        // non-empty" is the wrong test. Checked before `prepare` so an
1799        // unchanged cycle never pays for a synchronous config load.
1800        let finished: Vec<Task> = finished_tasks(queue)
1801            .into_iter()
1802            .filter(|task| !stalled_ids.contains(&task.id))
1803            .collect();
1804        let queued = queued_tasks(queue);
1805        // An empty queue has nothing to arrange. In particular, do not let
1806        // the conductor's initial snapshot cause synchronous config I/O
1807        // between the caller's stop notification and the idle wait below.
1808        if !(queued.is_empty() && stalled.is_empty() && finished.is_empty())
1809            && conductor.worth_a_look(queue, &stalled, &finished)
1810        {
1811            match prepare(&opts.repo, opts) {
1812                Ok(cfg) => {
1813                    conductor
1814                        .maybe_run(
1815                            &cfg,
1816                            &opts.repo,
1817                            queue,
1818                            &questions,
1819                            home,
1820                            &queued,
1821                            &stalled,
1822                            &finished,
1823                            opts.max_attempts,
1824                        )
1825                        .await;
1826                }
1827                Err(e) => tracing::warn!("conductor: no config: {e:#}"),
1828            }
1829        }
1830
1831        let candidates: Vec<Task> = runnable(queue)
1832            .into_iter()
1833            .filter(|t| !opts.once || !attempted.contains(&t.id))
1834            .collect();
1835
1836        // A task id only stays a key here while its attempt is genuinely in
1837        // flight; `status.current` is the same liveness fact `InFlightGuard`
1838        // maintains for the phone's own status file, so this piggybacks on
1839        // it rather than tracking a second copy of the same thing.
1840        let in_flight: Vec<String> = lock(status)
1841            .current
1842            .iter()
1843            .map(|c| c.task.clone())
1844            .collect();
1845        interrupt_pauses.retain(|id, _| in_flight.contains(id));
1846
1847        interrupt =
1848            advance_interrupt_tick(pause_for_interrupts, interrupt, &in_flight, &candidates);
1849        if let Interrupt::Parking {
1850            parked,
1851            interrupt_task,
1852        } = &interrupt
1853        {
1854            let reason = format!(
1855                "task {} asked to run first",
1856                crate::run::short_of(interrupt_task)
1857            );
1858            for id in parked {
1859                if let Some(pause) = interrupt_pauses.get(id) {
1860                    pause.park_because(reason.clone());
1861                }
1862            }
1863        }
1864        let candidates = interrupt_gate(&interrupt, &in_flight, candidates);
1865
1866        let cooling_down =
1867            lock(&quota_cooldown_until).is_some_and(|until| Timestamp::now() < until);
1868
1869        let mut started_any = false;
1870        for candidate in candidates {
1871            if stop.stopped() {
1872                break;
1873            }
1874
1875            let resume = land_resume_state(&candidate);
1876            if resume == LandResume::StillWaiting {
1877                continue;
1878            }
1879            let priority = resume == LandResume::Ready;
1880
1881            if !priority && cooling_down {
1882                continue;
1883            }
1884            let permit = match permit_kind(priority, candidate.urgent) {
1885                PermitKind::None => None,
1886                PermitKind::Urgent => match Arc::clone(&urgent_sem).try_acquire_owned() {
1887                    Ok(p) => Some(p),
1888                    // The one urgent slot is already spoken for by another
1889                    // `--urgent` task's run. Keep looking rather than falling
1890                    // back to the ordinary pool - see `PermitKind::Urgent`'s
1891                    // own doc - a later candidate might still be an ordinary
1892                    // task with a free slot, or another priority resume.
1893                    Err(_) => continue,
1894                },
1895                PermitKind::Ordinary => match Arc::clone(&sem).try_acquire_owned() {
1896                    Ok(p) => Some(p),
1897                    // No ordinary slot free right now. A later candidate in
1898                    // this same list might still be a priority resume or an
1899                    // urgent task, so keep looking rather than stopping here.
1900                    Err(_) => continue,
1901                },
1902            };
1903
1904            // A claim we cannot take means another daemon, or a human running
1905            // `magi run`, got there first. That is not the task's fault and
1906            // must not spend one of its attempts: move to the next candidate
1907            // rather than recording a failure.
1908            let Ok(claim) = queue.claim(&candidate.id) else {
1909                tracing::info!("task {} is claimed elsewhere; skipping", candidate.short());
1910                continue;
1911            };
1912            // Re-read under the claim: the task on disk may have been held or
1913            // edited between the listing and the lock.
1914            let mut task = match queue.get(&candidate.id) {
1915                Ok(t) if t.status.runnable() => t,
1916                Ok(_) => continue,
1917                Err(e) => {
1918                    tracing::warn!("could not re-read task {}: {e:#}", candidate.short());
1919                    continue;
1920                }
1921            };
1922            let task_id = task.id.clone();
1923            attempted.push(task_id.clone());
1924            lock(status).idle = false;
1925            // A stop asked for from here on is "finishing", not "stopped": the
1926            // run gets to reach a terminal status before the loop returns.
1927            stop.enter();
1928            started_any = true;
1929
1930            // A fresh, unshared handle - never `stop.pause()` - so parking
1931            // this run for an interrupt cannot leak into any other run this
1932            // loop ever drives. See `Pause`'s own doc.
1933            let run_pause = crate::graph::Pause::new();
1934            interrupt_pauses.insert(task_id.clone(), run_pause.clone());
1935
1936            let opts = opts.clone();
1937            let queue = queue.clone();
1938            let status = Arc::clone(status);
1939            let stop = stop.clone();
1940            let quota_cooldown_until = Arc::clone(&quota_cooldown_until);
1941            inflight.spawn(async move {
1942                // Held for the whole attempt: dropping either at the end of
1943                // this task is what releases the claim and, for an ordinary
1944                // candidate, frees its concurrency slot back to the loop.
1945                let _claim = claim;
1946                let _permit = permit;
1947                // See `InFlightGuard`: this must survive a panic inside `attempt`.
1948                let _inflight = InFlightGuard {
1949                    status: &status,
1950                    stop: &stop,
1951                    task_id: &task_id,
1952                };
1953                let quota = attempt(&opts, &queue, &status, &stop, run_pause, &mut task).await;
1954                lock(&status).completed += 1;
1955                // A quota loss is a fact about the machine, not this task, and
1956                // the next ordinary candidate the loop offers is no less
1957                // likely to hit the same wall: without a cooldown here a
1958                // whole backlog can be run - and failed - in the seconds it
1959                // takes each attempt to notice the CLI is out of quota.
1960                if !quota.is_empty() {
1961                    let with_hint = quota.iter().find(|q| q.reset.is_some());
1962                    let hint = with_hint.and_then(|q| q.reset.as_deref());
1963                    let reset_at = with_hint.and_then(|q| {
1964                        parse_reset_hint(q.reset.as_deref()?, Timestamp::now(), q.at)
1965                    });
1966                    let wait = quota_wait(
1967                        reset_at,
1968                        Timestamp::now(),
1969                        QUOTA_WAIT_FALLBACK,
1970                        QUOTA_WAIT_CAP,
1971                    );
1972                    let secs = i64::try_from(wait.as_secs()).unwrap_or(i64::MAX);
1973                    let until = Timestamp::now()
1974                        .checked_add(jiff::SignedDuration::from_secs(secs))
1975                        .unwrap_or(Timestamp::MAX);
1976                    *lock(&quota_cooldown_until) = Some(until);
1977                    match hint {
1978                        Some(h) => tracing::warn!(
1979                            "quota hit; waiting {}s before taking another ordinary task \
1980                             (CLI reported reset: {h})",
1981                            wait.as_secs()
1982                        ),
1983                        None => tracing::warn!(
1984                            "quota hit; waiting {}s before taking another ordinary task \
1985                             (no reset hint reported)",
1986                            wait.as_secs()
1987                        ),
1988                    }
1989                }
1990            });
1991        }
1992
1993        if started_any {
1994            continue;
1995        }
1996
1997        if stop.busy_now() {
1998            // Something started on an earlier tick is still running. Recheck
1999            // soon rather than sleeping out the whole poll interval - a freed
2000            // slot, or a land approval answered mid-run, must not sit idle
2001            // for it.
2002            stop.idle(RECHECK_WHILE_BUSY.min(opts.poll)).await;
2003            continue;
2004        }
2005
2006        // Truly idle: nothing new to start and nothing still running.
2007        lock(status).idle = true;
2008        if opts.once {
2009            // A one-shot drain must perform the same post-work cleanup as a
2010            // daemon that reached a normal idle interval. The startup pass
2011            // cannot see runs or cache files produced by this drain.
2012            janitor(&opts.repo, opts, home, worktrees_root).await;
2013            triage_held(queue, home, opts).await;
2014            break;
2015        }
2016        stop.idle(opts.poll).await;
2017        if stop.stopped() {
2018            continue;
2019        }
2020        // Housekeeping only after a full quiet interval. Running it before
2021        // the first idle wait can block the executor while an operator's
2022        // stop request is waiting to be scheduled, defeating Stop's retained
2023        // wake permit. No run can start while this branch is active, so the
2024        // janitor still never races an in-flight compile.
2025        janitor(&opts.repo, opts, home, worktrees_root).await;
2026        triage_held(queue, home, opts).await;
2027    }
2028
2029    // Never return while a run is still in flight, whichever way the loop
2030    // above exited: a stop only sets a flag - see `serve_until` - and
2031    // returning here while `inflight` still holds spawned work would abandon
2032    // it exactly as a mid-node kill would.
2033    while let Some(result) = inflight.join_next().await {
2034        if let Err(e) = result {
2035            tracing::error!("a spawned attempt did not finish cleanly: {e}");
2036        }
2037    }
2038    Ok(())
2039}
2040
2041/// Run one claimed task to a terminal status and record the outcome.
2042///
2043/// Every transition is flushed to the queue as it happens, so the state on disk
2044/// is what actually occurred rather than what this process still intends to
2045/// write.
2046async fn attempt(
2047    opts: &Opts,
2048    queue: &Queue,
2049    status: &Arc<Mutex<Status>>,
2050    stop: &Stop,
2051    interrupt_pause: crate::graph::Pause,
2052    task: &mut Task,
2053) -> Vec<QuotaLoss> {
2054    let repo = repo_for(task, &opts.repo);
2055    tracing::info!(
2056        "task {} — {} (repo {})",
2057        task.short(),
2058        task.title,
2059        repo.display()
2060    );
2061
2062    let mut config = match prepare(&repo, opts) {
2063        Ok(c) => c,
2064        Err(e) => {
2065            // A setup failure spends an attempt even though no run was minted.
2066            // Without that, a task naming a repository that does not exist
2067            // would be retried at every poll for as long as the daemon lives.
2068            task.attempts += 1;
2069            task.fail(format!("config: {e:#}"), opts.max_attempts);
2070            record(queue, task);
2071            return Vec::new();
2072        }
2073    };
2074    apply_solo(&mut config, task);
2075
2076    // The free-space gate, checked *before* anything is minted: a task that
2077    // waits out a full disk costs nothing yet, and must not spend an attempt
2078    // or start a run the machine cannot finish. Held tasks stay in the list
2079    // for the human to see, and `magi task release` re-queues them when space
2080    // comes back - the same recovery as any other hold. A volume whose free
2081    // space cannot be measured closes the gate too: starting a run blind on a
2082    // disk that may be full is how the machine ends up with 6.7 GB free.
2083    if let Some(reason) = disk_gate(&repo, &config) {
2084        task.last_error = Some(reason.clone());
2085        task.hold_machine(Some(reason.clone()));
2086        record(queue, task);
2087        tracing::warn!("holding {} for want of disk space: {reason}", task.short());
2088        return Vec::new();
2089    }
2090
2091    // A resumable run of this task is carried on, never re-competed. The
2092    // candidates are built and paid for, and a fresh competition races a
2093    // second implementation against them.
2094    //
2095    // Two runs paid for that lesson. Run 01c2 was blocked and the loop
2096    // started 3cbf on the same task a moment later, duplicating two and a
2097    // half hours of agent work. Then b25f stalled on a judge that timed out
2098    // and one that answered with no JSON - `quota: 0`, so nothing the machine
2099    // was to blame for - and 4043 started **one second** later, buying three
2100    // fresh implementations to reach the same panel. `RunStatus::resumable`
2101    // rather than `!done()` is what catches the second case: a stall is
2102    // terminal, and its cheap recovery re-asks only the absent seats.
2103    //
2104    // A load failure is warned about rather than silently read as "not
2105    // resumable": the alternative is exactly what let a schema mismatch on
2106    // run `eba2` fall through to a full re-competition with nobody told why.
2107    // `crate::conduct` is what actually offers a better answer than
2108    // `Runner::start` here (see `Recovery::Review`), once this task's next
2109    // failure shows it up as `held`/`failed` with the run state unreadable.
2110    let unfinished = (!task.fresh_start)
2111        .then(|| unfinished_run(&task.runs, task.short()))
2112        .flatten();
2113    // `crate::conduct` chose `Review` for this task on an earlier cycle: its
2114    // branch survived, and this reopens exactly that branch as a
2115    // review-only pass rather than resuming or competing again. Consumed
2116    // (cleared) here whichever way this goes, so it never outlives this one
2117    // attempt - see `queue::Task::review_branch`.
2118    let review_branch = task.review_branch.take();
2119    let branch_exists = match &review_branch {
2120        Some(branch) => crate::git::branch_exists(&repo, branch)
2121            .await
2122            .unwrap_or(false),
2123        None => false,
2124    };
2125    let starter = choose_starter(
2126        review_branch.as_deref(),
2127        branch_exists,
2128        unfinished.as_deref(),
2129    );
2130    let started = match &starter {
2131        Starter::Review(branch) => {
2132            tracing::info!(
2133                "task {} reopens `{branch}` as a review-only pass",
2134                task.short()
2135            );
2136            Runner::review(&repo, branch, config).await
2137        }
2138        Starter::Resume(id) => {
2139            tracing::info!("resuming run {id} rather than competing again");
2140            Runner::resume(id).map(|mut r| {
2141                if let Some(instruction) =
2142                    prepare_instruction(&starter, Some(&r.state.instruction), task)
2143                {
2144                    r.state.instruction = instruction;
2145                }
2146                r
2147            })
2148        }
2149        Starter::Start => {
2150            if let Some(branch) = &review_branch {
2151                tracing::warn!(
2152                    "conductor chose review for task {} but branch `{branch}` no longer \
2153                     exists; requeuing as a fresh competition instead",
2154                    task.short()
2155                );
2156            }
2157            let instruction = prepare_instruction(&starter, None, task)
2158                .unwrap_or_else(|| task.instruction.clone());
2159            Runner::start(&repo, instruction, config).await
2160        }
2161    };
2162    let mut runner = match started {
2163        Ok(r) => r,
2164        Err(e) => {
2165            task.attempts += 1;
2166            task.fail(format!("could not start the run: {e:#}"), opts.max_attempts);
2167            record(queue, task);
2168            return Vec::new();
2169        }
2170    };
2171    // A stop that means "park" reaches the graph through this handle.
2172    runner.on_pause(stop.pause());
2173    // `poll`'s interrupt scheduler reaches this one run - and no other -
2174    // through this handle. See `Pause`'s own doc for why these are never
2175    // the same one.
2176    runner.watch_interrupt(interrupt_pause);
2177
2178    // `start` has minted the run, so the task can now point at it. Persisting
2179    // `Running` before `execute` is what makes a crash mid-run legible.
2180    let run = runner.state.id.clone();
2181    task.start(run.clone());
2182    record(queue, task);
2183    lock(status).current.push(Current {
2184        task: task.id.clone(),
2185        run,
2186    });
2187
2188    let detail = match runner.execute().await {
2189        Ok(()) => describe(&runner.state),
2190        Err(e) => format!("{e:#}"),
2191    };
2192    let verdict = Verdict {
2193        status: runner.state.status,
2194        // A run that opened a pull request handed its work over, whatever the
2195        // gate then decided about merging it.
2196        left_pr: runner.state.pr.is_some(),
2197        // Only a rate limit earns the task its attempt back.
2198        quota_hit: !runner.state.quota.is_empty(),
2199        // A run that parked was asked to stop; that is not a failure and must
2200        // not spend an attempt, or replacing the binary a few times would
2201        // exhaust a task's budget without an agent ever misbehaving.
2202        parked: runner.state.parked,
2203        // A quota loss that left nothing viable is the same machine fact as a
2204        // `Stalled` quota loss; see `settle`'s doc table.
2205        no_viable_candidates: runner.state.viable().is_empty(),
2206    };
2207    settle_and_diagnose(task, verdict, &detail, opts.max_attempts, &runner.state);
2208    record(queue, task);
2209    tracing::info!(
2210        "task {} is {} after run {} ({})",
2211        task.short(),
2212        task.status.as_str(),
2213        runner.state.short(),
2214        label(runner.state.status)
2215    );
2216    runner.state.quota
2217}
2218
2219/// Cut this attempt's candidate count to one when the task asked to run
2220/// alone.
2221///
2222/// Pure and separate from [`attempt`] so the one thing this feature changes -
2223/// which `candidates` a `solo` task's run is built with - can be asserted
2224/// without minting a run: `attempt` drives `graph::Runner`, which spawns real
2225/// agent CLIs, and no test may do that. `config` is mutated in place, taken by
2226/// value from the caller's own copy, so a repository's `magi.toml` on disk is
2227/// never touched - only the `Config` this one attempt hands to `Runner::start`.
2228fn apply_solo(config: &mut Config, task: &Task) {
2229    if task.solo {
2230        config.graph.candidates = 1;
2231    }
2232}
2233
2234/// Load the config for a task's repository, with the merge override applied.
2235fn prepare(repo: &Path, opts: &Opts) -> Result<Config> {
2236    let (mut config, _layers) = Config::discover(repo, opts.config.as_deref())?;
2237    if let Some(mode) = &opts.merge {
2238        config.merge.mode = merge_mode(mode)?;
2239    }
2240    Ok(config)
2241}
2242
2243/// Prune the shared build cache back under its cap at a safe boundary
2244/// between runs, so a queue that never empties - and so never reaches
2245/// [`poll`]'s fully-idle branch, where the ordinary [`janitor`] pass lives -
2246/// does not leave the cache to grow unchecked for as long as the backlog
2247/// lasts.
2248///
2249/// Called from [`poll`] only when `stop.busy_now()` is already `false`: the
2250/// same liveness fact the idle branch's own janitor call rests on - no run
2251/// this daemon spawned is still mid-compile - so pruning here races nothing.
2252/// The caller must not call this while a run is in flight; there is no
2253/// second `busy_now()` check inside this function, on purpose, because there
2254/// is nothing left to check that `busy_now()` has not already answered.
2255///
2256/// A stop that has already been asked for *is* checked here, for a different
2257/// reason. [`clean::prune_cache_if_over_limit`] walks the whole cache
2258/// synchronously before it decides anything, so the poll loop cannot get back
2259/// to its own `stopped()` test until that walk is over — and a loop already
2260/// on its way out must not make the operator wait out housekeeping it is
2261/// about to stop needing. This is the same call the idle branch makes when it
2262/// rechecks `stop.stopped()` after its wait before reaching [`janitor`], and
2263/// it matters more here: `busy_now()` is false throughout, so
2264/// [`Stop::finishing`] would report a stop as already landed while the walk
2265/// still held the loop. Nothing is lost by skipping — the cap is a standing
2266/// policy, and the next daemon's startup pass measures the same cache.
2267///
2268/// Rate-limited by [`CACHE_CHECK_INTERVAL_SECS`] rather than run on every
2269/// poll: a busy loop reaches this the instant one run's `InFlightGuard` drops
2270/// and the next has not yet claimed a task, which can be every few
2271/// milliseconds, and re-walking a multi-gigabyte cache that often would cost
2272/// more than the growth it is guarding against.
2273async fn maybe_prune_cache_between_runs(
2274    repo: &Path,
2275    opts: &Opts,
2276    home: &Path,
2277    stop: &Stop,
2278    last_checked: &mut Option<Timestamp>,
2279    now: Timestamp,
2280) {
2281    if stop.stopped() || !cache_check_due(*last_checked, now, CACHE_CHECK_INTERVAL_SECS) {
2282        return;
2283    }
2284    *last_checked = Some(now);
2285    let cfg = match prepare(repo, opts) {
2286        Ok(cfg) => cfg,
2287        Err(e) => {
2288            tracing::warn!("cache check: no config: {e:#}");
2289            return;
2290        }
2291    };
2292    match clean::prune_cache_if_over_limit(&cfg, home) {
2293        Ok(Some(pruned)) if pruned.files > 0 => tracing::info!(
2294            "housekeep: pruned {} file(s) ({} bytes) from the shared cache between runs",
2295            pruned.files,
2296            pruned.freed
2297        ),
2298        Ok(_) => {}
2299        Err(e) => tracing::warn!("housekeep: prune cache: {e:#}"),
2300    }
2301}
2302
2303/// Whether [`maybe_prune_cache_between_runs`] should re-measure the cache
2304/// now, given when it last did (if ever). Pure, so the cadence is asserted
2305/// directly rather than by waiting out real minutes in a test.
2306fn cache_check_due(last_checked: Option<Timestamp>, now: Timestamp, interval_secs: u64) -> bool {
2307    last_checked.is_none_or(|last| clean::due(now, last, interval_secs))
2308}
2309
2310/// The disk janitor, with its housekeeping logged rather than fatal.
2311///
2312/// Called only at the loop's idle points, for the reason the caller documents:
2313/// a prune racing a live compile would delete files mid-build. The config is
2314/// re-read on every call because the repository that just ran may not be the
2315/// daemon's own default, and the cache directory is a repository fact.
2316///
2317/// `home` and `worktrees_root` are parameters rather than [`crate::run::home`]
2318/// and [`crate::run::default_worktree_root`] read here, for the same reason
2319/// [`drive`] takes its queue and status file rather than resolving them: a
2320/// test driving the loop must not reach through to the operator's real home
2321/// or worktree bay just because the janitor runs on every idle tick.
2322/// `worktrees_root` staying unread by [`clean::fold_due`] once made this easy
2323/// to get wrong silently - a test's `home` was already isolated, but nothing
2324/// exercised the parameter next to it, so a real worktree bay stayed wired in
2325/// underneath. The moment [`clean::fold_orphaned_worktrees`] started reading
2326/// it for real, every test in this file that drives the loop at all started
2327/// sweeping the operator's actual `~/wt/<repo>` instead of a fixture's.
2328async fn janitor(repo: &Path, opts: &Opts, home: &Path, worktrees_root: &Path) {
2329    let cfg = match prepare(repo, opts) {
2330        Ok(cfg) => cfg,
2331        Err(e) => {
2332            tracing::warn!("housekeep: no config: {e:#}");
2333            return;
2334        }
2335    };
2336    // A run's own worktree lives under `config.graph.worktree_root` when the
2337    // repository sets one - the same precedence `RunState::worktree_root`
2338    // uses - and `worktrees_root` only stands in for the *default* an
2339    // unconfigured repository resolves to (see this function's own
2340    // parameter, or the test fixture wiring one to a fake path). Housekeeping
2341    // that always swept the default regardless of this override would never
2342    // see, and so never reclaim, a single worktree for a repository that
2343    // relocated them elsewhere.
2344    let worktrees_root = cfg.graph.worktree_root.as_deref().unwrap_or(worktrees_root);
2345    let out = clean::housekeep(&cfg, home, worktrees_root, repo, Timestamp::now()).await;
2346    // Reported whenever there is anything to say, not only when `folded > 0`:
2347    // the incident this exists to prevent was 90 of 93 runs skipped and 0
2348    // folded, on every single pass, for months - a report gated on `folded`
2349    // would have stayed silent through every one of them.
2350    if out.folded > 0 || out.unreadable > 0 || out.orphaned_worktrees > 0 {
2351        let mut extra = Vec::new();
2352        if out.unreadable > 0 {
2353            extra.push(format!("{} unreadable", out.unreadable));
2354        }
2355        if out.orphaned_worktrees > 0 {
2356            extra.push(format!("{} orphaned worktree(s)", out.orphaned_worktrees));
2357        }
2358        let detail = if extra.is_empty() {
2359            String::new()
2360        } else {
2361            format!(" ({})", extra.join(", "))
2362        };
2363        tracing::info!("housekeep: folded {} run(s){detail}", out.folded);
2364    }
2365    if out.cache_files > 0 {
2366        tracing::info!(
2367            "housekeep: pruned {} file(s) ({} bytes) from the shared cache",
2368            out.cache_files,
2369            out.cache_freed
2370        );
2371    }
2372    if out.questions_abandoned > 0 {
2373        tracing::info!(
2374            "housekeep: abandoned {} question(s) left open by a finished run",
2375            out.questions_abandoned
2376        );
2377    }
2378}
2379
2380/// Run [`triage::run_once`] and log whatever it did, the same "only when
2381/// there is something to say" rule [`janitor`] follows for its own report.
2382///
2383/// Called at the same idle points as [`janitor`] - once per full poll
2384/// interval, never mid-attempt - for the same reason: it is not liveness
2385/// critical, and a task's own `hold_reason` string is the one thing this
2386/// would otherwise re-check (via [`crate::disk::free_bytes`]) on every busy
2387/// tick for no benefit.
2388async fn triage_held(queue: &Queue, home: &Path, opts: &Opts) {
2389    let questions = Questions::at(home.join("questions"));
2390    let report = triage::run_once(queue, &questions, opts.config.as_deref(), Timestamp::now());
2391    if report.is_empty() {
2392        return;
2393    }
2394    if !report.quarantined.is_empty() {
2395        tracing::info!(
2396            "triage: held {} blocked task(s) whose blocked-on task or \
2397             question no longer exists: {}",
2398            report.quarantined.len(),
2399            report.quarantined.join(", ")
2400        );
2401    }
2402    if !report.resumed.is_empty() {
2403        tracing::info!(
2404            "triage: resumed {} held task(s) whose machine hold had resolved: {}",
2405            report.resumed.len(),
2406            report.resumed.join(", ")
2407        );
2408    }
2409    if !report.asked.is_empty() {
2410        tracing::info!(
2411            "triage: asked about {} held task(s): {}",
2412            report.asked.len(),
2413            report.asked.join(", ")
2414        );
2415    }
2416    if !report.answered.is_empty() {
2417        tracing::info!(
2418            "triage: applied {} operator answer(s): {}",
2419            report.answered.len(),
2420            report.answered.join(", ")
2421        );
2422    }
2423}
2424
2425/// The free-space gate: what stands between this task and a new run, if
2426/// anything. `Some(reason)` holds the task; `None` lets it start.
2427///
2428/// A zero [`Config::disk::min_free_bytes`] opens the gate unconditionally -
2429/// the operator opted out. A measurement failure is a gate, not a pass: both
2430/// sides of "cannot tell" are served by not starting.
2431fn disk_gate(repo: &Path, config: &Config) -> Option<String> {
2432    disk_gate_with(repo, config, crate::disk::free_bytes)
2433}
2434
2435/// [`disk_gate`] with its free-space measurement supplied by the caller, so a
2436/// test can assert the exact wiring `attempt` runs - config's threshold in,
2437/// task-holding reason out - without asking the real machine's disk anything.
2438fn disk_gate_with<F: Fn(&Path) -> Result<u64>>(
2439    repo: &Path,
2440    config: &Config,
2441    free_bytes: F,
2442) -> Option<String> {
2443    let min = config.disk.min_free_bytes;
2444    if min == 0 {
2445        return None;
2446    }
2447    match free_bytes(repo) {
2448        Ok(free) => crate::disk::gate(free, min),
2449        Err(e) => Some(format!(
2450            "could not measure free space on {} ({e}); the disk gate refuses \
2451             to let a run start blind",
2452            repo.display()
2453        )),
2454    }
2455}
2456
2457/// How long to wait before offering another task when a run lost a seat to a
2458/// rate limit and its [`QuotaLoss::reset`] carried no hint [`parse_reset_hint`]
2459/// could read, or carried nothing at all. Long enough that a quota outage
2460/// cannot burn through a whole backlog in the few seconds each doomed attempt
2461/// takes to fail; short enough that a quota which clears early is not left
2462/// idle for the fallback's sake.
2463const QUOTA_WAIT_FALLBACK: Duration = Duration::from_secs(5 * 60);
2464
2465/// Longest a parsed reset hint may push the wait out to. The hint comes from
2466/// the CLI's own words, not a contract, so a parsing slip that lands a day
2467/// away must not leave the loop asleep for a day.
2468const QUOTA_WAIT_CAP: Duration = Duration::from_secs(30 * 60);
2469
2470/// How long [`poll`] should wait before offering the next task, after a run
2471/// lost at least one seat to a rate limit.
2472///
2473/// Pure and separate from the loop so the policy can be exercised without a
2474/// real quota outage. `reset_at` is the time [`parse_reset_hint`] made of the
2475/// CLI's free-text hint, if it could; `fallback` is what to wait when there is
2476/// nothing to parse, or the parsed time has already passed; `cap` bounds how
2477/// far a parsed hint is trusted to push the wait out.
2478fn quota_wait(
2479    reset_at: Option<Timestamp>,
2480    now: Timestamp,
2481    fallback: Duration,
2482    cap: Duration,
2483) -> Duration {
2484    match reset_at {
2485        Some(at) if at > now => {
2486            let secs = u64::try_from(at.as_second() - now.as_second()).unwrap_or(0);
2487            Duration::from_secs(secs).min(cap)
2488        }
2489        _ => fallback,
2490    }
2491}
2492
2493/// Best-effort reading of a [`QuotaLoss::reset`] hint into a concrete time.
2494///
2495/// `reset` is deliberately free text — see [`crate::agent::Quota`], which
2496/// explains why parsing it exactly "would be a bug factory" — so this only
2497/// recognises the shapes actually observed in the wild, and returns `None`
2498/// for anything else rather than guess at a format nobody has seen.
2499///
2500/// `recorded` is when the loss was noted ([`QuotaLoss::at`]). It anchors the
2501/// relative shape (`"in 1h2m49s"`, agy's), which counts from the moment the CLI
2502/// said it, not from whenever this loop happens to read it: anchoring on `now`
2503/// would push the reset later on every read and would read an already-elapsed
2504/// reset as still in the future. (A long hint is still clamped by
2505/// [`QUOTA_WAIT_CAP`]; the anchor matters for short hints and elapsed ones.)
2506fn parse_reset_hint(text: &str, now: Timestamp, recorded: Timestamp) -> Option<Timestamp> {
2507    parse_reset_hint_zoned(text, now)
2508        .or_else(|| parse_reset_hint_dated(text))
2509        .or_else(|| parse_reset_hint_relative(text, recorded))
2510}
2511
2512/// agy's shape: `"in 1h2m49s"` - `in`, then hours/minutes/seconds, each unit
2513/// optional but at least one required, in that order. A bare number or any
2514/// unknown unit is refused.
2515fn parse_reset_hint_relative(text: &str, recorded: Timestamp) -> Option<Timestamp> {
2516    let rest = text.trim().trim_end_matches('.').strip_prefix("in ")?;
2517    let mut rest = rest.trim();
2518    if rest.is_empty() {
2519        return None;
2520    }
2521    let mut total: i64 = 0;
2522    let mut matched = false;
2523    for (unit, secs) in [('h', 3600), ('m', 60), ('s', 1)] {
2524        if let Some((digits, tail)) = rest.split_once(unit)
2525            && !digits.is_empty()
2526            && digits.bytes().all(|b| b.is_ascii_digit())
2527        {
2528            total += digits.parse::<i64>().ok()?.checked_mul(secs)?;
2529            rest = tail;
2530            matched = true;
2531        }
2532    }
2533    if !rest.is_empty() || !matched {
2534        return None;
2535    }
2536    recorded
2537        .checked_add(jiff::SignedDuration::from_secs(total))
2538        .ok()
2539}
2540
2541/// Reads a 12-hour `"H:MMam/pm"` clock reading (whitespace trimmed,
2542/// case-insensitive) into a 24-hour hour and minute. Shared by every
2543/// reset-hint shape below.
2544fn parse_12h_clock(clock: &str) -> Option<(i8, i8)> {
2545    let clock = clock.trim().to_lowercase();
2546    let (digits, pm) = clock
2547        .strip_suffix("am")
2548        .map(|d| (d, false))
2549        .or_else(|| clock.strip_suffix("pm").map(|d| (d, true)))?;
2550    let (h, m) = digits.trim().split_once(':')?;
2551    let mut hour: i8 = h.trim().parse().ok()?;
2552    let minute: i8 = m.trim().parse().ok()?;
2553    if !(1..=12).contains(&hour) || !(0..=59).contains(&minute) {
2554        return None;
2555    }
2556    if pm && hour != 12 {
2557        hour += 12;
2558    } else if !pm && hour == 12 {
2559        hour = 0;
2560    }
2561    Some((hour, minute))
2562}
2563
2564/// The Claude CLI's shape: `"H:MMam/pm (Zone)"`, naming only a clock reading
2565/// and a zone, never a date. A clock reading already past today is read as
2566/// tomorrow's: a CLI naming a same-day reset that has already gone by means
2567/// the window rolled over while nothing was watching.
2568fn parse_reset_hint_zoned(text: &str, now: Timestamp) -> Option<Timestamp> {
2569    let open = text.find('(')?;
2570    let close = text.rfind(')')?;
2571    if close <= open {
2572        return None;
2573    }
2574    let zone = text[open + 1..close].trim();
2575    let (hour, minute) = parse_12h_clock(&text[..open])?;
2576    let tz = jiff::tz::TimeZone::get(zone).ok()?;
2577    let candidate = now
2578        .to_zoned(tz)
2579        .with()
2580        .hour(hour)
2581        .minute(minute)
2582        .second(0)
2583        .millisecond(0)
2584        .microsecond(0)
2585        .nanosecond(0)
2586        .build()
2587        .ok()?;
2588    let mut at = candidate.timestamp();
2589    if at <= now {
2590        at += jiff::SignedDuration::from_hours(24);
2591    }
2592    Some(at)
2593}
2594
2595/// The Codex CLI's shape: `"Mon DDth, YYYY H:MMam/pm"` (English month
2596/// abbreviation, an ordinal day, a 4-digit year, a 12-hour clock reading),
2597/// with no zone at all — unlike [`parse_reset_hint_zoned`], so there is no
2598/// "already past today" correction to make: the year already disambiguates
2599/// it. Scanned as a five-word window so it can be pulled out of the middle
2600/// of a full sentence, e.g. Codex's actual wording: "...or try again at Sep
2601/// 19th, 2026 5:10 PM." The result is read as UTC, same as this crate reads
2602/// any other timestamp with no zone attached.
2603fn parse_reset_hint_dated(text: &str) -> Option<Timestamp> {
2604    let words: Vec<&str> = text.split_whitespace().collect();
2605    if words.len() < 5 {
2606        return None;
2607    }
2608    (0..=words.len() - 5)
2609        .find_map(|start| parse_dated_window(&words[start..start + 5], words.get(start + 5)))
2610}
2611
2612/// One five-word window: month, `"DDth,"`, `"YYYY"`, `"H:MM"`, `"am/pm"`. A
2613/// parenthesis right after the window is refused rather than ignored — it
2614/// reads as an explicit zone annotation on a shape that otherwise carries
2615/// none, and guessing UTC anyway would be exactly the silent misread this
2616/// module's parsing otherwise avoids.
2617fn parse_dated_window(window: &[&str], trailing: Option<&&str>) -> Option<Timestamp> {
2618    if trailing.is_some_and(|next| next.starts_with('(')) {
2619        return None;
2620    }
2621    let month = month_number(window[0])?;
2622    let day_token = window[1].strip_suffix(',')?.to_lowercase();
2623    let day_digits = ["st", "nd", "rd", "th"]
2624        .iter()
2625        .find_map(|suffix| day_token.strip_suffix(*suffix))?;
2626    let day: i8 = day_digits.parse().ok()?;
2627    let year_token = window[2];
2628    if year_token.len() != 4 || !year_token.bytes().all(|b| b.is_ascii_digit()) {
2629        return None;
2630    }
2631    let year: i16 = year_token.parse().ok()?;
2632    // The am/pm word carries the sentence's own trailing punctuation, e.g.
2633    // the period ending "...at Sep 19th, 2026 5:10 PM." — strip it before
2634    // reusing the same 12-hour clock reader the bracketed shape uses.
2635    let ampm = window[4].trim_matches(|c: char| !c.is_ascii_alphabetic());
2636    let (hour, minute) = parse_12h_clock(&format!("{}{}", window[3], ampm))?;
2637    let date = jiff::civil::Date::new(year, month, day).ok()?;
2638    let candidate = date
2639        .at(hour, minute, 0, 0)
2640        .to_zoned(jiff::tz::TimeZone::UTC)
2641        .ok()?;
2642    Some(candidate.timestamp())
2643}
2644
2645/// The 3-letter English month abbreviation [`parse_reset_hint_dated`] reads,
2646/// case-insensitively, into a 1-based month number.
2647fn month_number(name: &str) -> Option<i8> {
2648    const NAMES: [&str; 12] = [
2649        "jan", "feb", "mar", "apr", "may", "jun", "jul", "aug", "sep", "oct", "nov", "dec",
2650    ];
2651    let lower = name.to_lowercase();
2652    NAMES
2653        .iter()
2654        .position(|n| *n == lower.as_str())
2655        .map(|i| i as i8 + 1)
2656}
2657
2658/// Resuming a `Blocked` run that already spent every review round its own
2659/// config allowed cannot make progress: `graph::Runner`'s review loop walks
2660/// `(reviews.len()+1)..=max_rounds`, which is empty once `reviews.len()` has
2661/// reached `max_rounds`, so `execute` would settle straight back to
2662/// `Blocked` without asking anyone anything. Read-only against a state this
2663/// build never mutates — `src/graph.rs` stays untouched — but without this
2664/// check, [`unfinished_run`] would keep reporting such a run as still
2665/// "unfinished", and `crate::conduct::Recovery::Requeue` (whose whole
2666/// promise is a fresh competition when a design needs to change) would
2667/// silently resume the exhausted run instead, spending an attempt on a
2668/// cycle that cannot change anything.
2669fn exhausted_review_budget(state: &RunState) -> bool {
2670    state.status == RunStatus::Blocked && state.reviews.len() >= state.config.graph.review_rounds
2671}
2672
2673/// This task's *most recent* run, if resuming it would actually make
2674/// progress. `short` is only for the warning's own message.
2675///
2676/// Only ever `runs.last()` — never a search back through older history.
2677/// `runs` accumulates one entry per fresh `Runner::start`/`Runner::review`
2678/// mint, oldest first, and every entry before the last one was already
2679/// superseded at the moment it was minted: the daemon only ever starts a new
2680/// run when the previous one was not worth resuming (unresumable, exhausted,
2681/// or unreadable), or when `crate::conduct::Recovery::Review` deliberately
2682/// opens a fresh review-only run alongside an older, already-failed
2683/// competition. Searching further back would let an old run that merely
2684/// *looks* resumable — a `Stalled` competition an earlier `Review` pass left
2685/// behind, say — get resumed instead of the fresh competition
2686/// `crate::conduct::Recovery::Requeue` actually promised, reviving history
2687/// nothing asked to revisit.
2688///
2689/// Two runs paid for the "prefer resuming over restarting" half of this
2690/// lesson, which is why this still checks `runs.last()` rather than always
2691/// restarting. Run 01c2 was blocked and the loop started 3cbf on the same
2692/// task a moment later, duplicating two and a half hours of agent work. Then
2693/// b25f stalled on a judge that timed out and one that answered with no JSON
2694/// — `quota: 0`, so nothing the machine was to blame for — and 4043 started
2695/// **one second** later, buying three fresh implementations to reach the
2696/// same panel. `RunStatus::resumable` rather than `!done()` is what catches
2697/// the second case: a stall is terminal, and its cheap recovery re-asks only
2698/// the absent seats. [`exhausted_review_budget`] is the other half: a run
2699/// that is technically `resumable()` but provably cannot progress must not
2700/// count as "unfinished" either, or `Recovery::Requeue` becomes a silent
2701/// no-op instead of the fresh competition it promises.
2702///
2703/// A load failure is warned about rather than silently read as "not
2704/// resumable": the alternative is exactly what let a schema mismatch on run
2705/// `eba2` fall through to a full re-competition with nobody told why.
2706/// `crate::conduct` is what actually offers a better answer than
2707/// `Runner::start` here (see `Recovery::Review`), once this task's next
2708/// failure shows it up as `held`/`failed` with the run state unreadable.
2709fn unfinished_run(runs: &[String], short: &str) -> Option<String> {
2710    unfinished_run_with(runs, short, RunState::load)
2711}
2712
2713/// [`unfinished_run`] with an injected state reader. Tests provide their
2714/// fixtures directly rather than touching the process-global run home.
2715fn unfinished_run_with<F>(runs: &[String], short: &str, load: F) -> Option<String>
2716where
2717    F: FnOnce(&str) -> Result<RunState>,
2718{
2719    let id = runs.last()?;
2720    match load(id) {
2721        Ok(s) if s.status.resumable() && !exhausted_review_budget(&s) => Some(id.clone()),
2722        Ok(_) => None,
2723        Err(e) => {
2724            tracing::warn!("could not read run {id} for task {short}: {e:#}");
2725            None
2726        }
2727    }
2728}
2729
2730/// Which of the three ways [`attempt`] can mint or continue a run this task
2731/// should use.
2732#[derive(Debug, Clone, PartialEq, Eq)]
2733enum Starter {
2734    /// `crate::graph::Runner::review` against a branch `crate::conduct` chose
2735    /// and that still exists.
2736    Review(String),
2737    /// `crate::graph::Runner::resume` on an unfinished run of this task.
2738    Resume(String),
2739    /// `crate::graph::Runner::start`: a fresh competition.
2740    Start,
2741}
2742
2743/// Decide which of [`Runner::review`], [`Runner::resume`] or [`Runner::start`]
2744/// this attempt should use. Pure, and separate from [`attempt`], so the
2745/// routing itself is assertable without spawning a real graph or a git
2746/// process: `attempt`'s own `crate::git::branch_exists` call has already
2747/// happened by the time this is called.
2748///
2749/// `review_branch` wins whenever `branch_exists` confirms it; a `review_branch`
2750/// whose branch is gone falls all the way through to [`Starter::Start`], not
2751/// to [`Starter::Resume`] — `crate::conduct` chose review over resuming the
2752/// old (likely `Blocked`) run in the first place, and a branch that vanished
2753/// out from under that choice is not evidence resuming it would fare better.
2754fn choose_starter(
2755    review_branch: Option<&str>,
2756    branch_exists: bool,
2757    unfinished: Option<&str>,
2758) -> Starter {
2759    match review_branch {
2760        Some(branch) if branch_exists => Starter::Review(branch.to_owned()),
2761        Some(_) => Starter::Start,
2762        None => match unfinished {
2763            Some(id) => Starter::Resume(id.to_owned()),
2764            None => Starter::Start,
2765        },
2766    }
2767}
2768
2769/// Which repository a task runs in. A task that names none — the normal case
2770/// for one filed from a phone — runs in the daemon's own default.
2771fn repo_for(task: &Task, fallback: &Path) -> PathBuf {
2772    if task.repo.as_os_str().is_empty() || task.repo == Path::new(".") {
2773        return fallback.to_path_buf();
2774    }
2775    task.repo.clone()
2776}
2777
2778/// The header [`append_answers`] appends operator answers under. Shared with
2779/// [`strip_answers_block`] so a resumed run's instruction can be refreshed
2780/// rather than grown a new block on every resume.
2781const ANSWERS_HEADER: &str = "\n\n# Operator answers\n\n";
2782
2783/// Render the first `count` answers in the block appended to an instruction.
2784fn answers_block(task: &Task, count: usize) -> String {
2785    let mut s = ANSWERS_HEADER.to_owned();
2786    for a in &task.answers[..count] {
2787        s.push_str(&format!("- {}: {}\n", a.question, a.answer));
2788    }
2789    s
2790}
2791
2792/// Append every answer `crate::conduct` has collected for `task` onto `base`,
2793/// in the shape both [`instruction_for`] and [`resumed_instruction`] use.
2794fn append_answers(base: &str, task: &Task) -> String {
2795    if task.answers.is_empty() {
2796        return base.to_owned();
2797    }
2798    let mut s = base.to_owned();
2799    s.push_str(&answers_block(task, task.answers.len()));
2800    s
2801}
2802
2803/// Drop the prior answer block only when it is exactly the suffix this task
2804/// could have appended on an earlier resume. An `ANSWERS_HEADER` written by
2805/// the task author is ordinary instruction text, not a block to remove.
2806fn strip_answers_block<'a>(instruction: &'a str, task: &Task) -> &'a str {
2807    for count in (1..=task.answers.len()).rev() {
2808        let block = answers_block(task, count);
2809        if let Some(base) = instruction.strip_suffix(&block) {
2810            return base;
2811        }
2812    }
2813    instruction
2814}
2815
2816/// The instruction handed to `Runner::start`: the task's own text, plus any
2817/// operator answers `crate::conduct` collected for it (see
2818/// [`Task::answers`]), so a decision the operator actually made reaches the
2819/// implementers rather than only clearing the block that was waiting on it.
2820///
2821/// Appended rather than merged into [`Task::instruction`] itself, so the
2822/// task's own record stays exactly what its author wrote.
2823fn instruction_for(task: &Task) -> String {
2824    append_answers(&task.instruction, task)
2825}
2826
2827/// The instruction a resumed run should carry on with: whatever it already
2828/// had, refreshed with the task's *current* operator answers.
2829///
2830/// A resumable run's own `RunState::instruction` predates any answer
2831/// `crate::conduct` collects after the run parks, so resuming it unchanged —
2832/// the behaviour before this function existed — silently drops the very
2833/// decision the operator made to unblock it. Re-stripping any block this
2834/// function appended on an earlier resume before re-appending the current
2835/// list (rather than blindly appending again) is what keeps a task resumed
2836/// three times over three answered questions from carrying the same answer
2837/// three times.
2838fn resumed_instruction(old_instruction: &str, task: &Task) -> String {
2839    append_answers(strip_answers_block(old_instruction, task), task)
2840}
2841
2842/// What [`attempt`] should tell a [`Starter`] about `task`'s current operator
2843/// answers before handing it to `Runner` — the actual boundary between
2844/// [`choose_starter`]'s routing and the graph, factored out so it is
2845/// assertable without a real repository, git branch, or agent CLI.
2846///
2847/// `Starter::Review` deliberately answers `None`: `Runner::review` builds its
2848/// instruction from the reviewed branch's own commit log because there is no
2849/// task statement to speak of for hand-written work, and splicing operator
2850/// answers into that text would contradict the very message it sends
2851/// reviewers ("there is no task statement").
2852fn prepare_instruction(
2853    starter: &Starter,
2854    old_instruction: Option<&str>,
2855    task: &Task,
2856) -> Option<String> {
2857    match starter {
2858        Starter::Start => Some(instruction_for(task)),
2859        Starter::Resume(_) => Some(resumed_instruction(
2860            old_instruction.expect("a resumed run always has a prior instruction"),
2861            task,
2862        )),
2863        Starter::Review(_) => None,
2864    }
2865}
2866
2867/// Persist a transition. A queue write failure is logged rather than fatal: the
2868/// run already happened, and taking the daemon down would only add a lost
2869/// backlog to a full disk.
2870fn record(queue: &Queue, task: &mut Task) {
2871    if let Err(e) = queue.put(task) {
2872        tracing::error!("could not record task {}: {e:#}", task.short());
2873    }
2874}
2875
2876/// Every runnable task, in the order the loop should try them.
2877///
2878/// The head of this list is exactly what [`Queue::next_runnable`] offers; the
2879/// tail exists so that a claim somebody else holds costs the loop the next
2880/// candidate rather than a whole poll interval of idleness.
2881fn runnable(queue: &Queue) -> Vec<Task> {
2882    let mut tasks: Vec<Task> = queue
2883        .list()
2884        .into_iter()
2885        .filter(|t| t.status.runnable())
2886        .collect();
2887    tasks.sort_unstable_by(|a, b| b.priority.cmp(&a.priority).then(a.id.cmp(&b.id)));
2888    tasks
2889}
2890
2891/// Why a run ended where it did, in one line, for [`Task::last_error`].
2892///
2893/// A stalled run names the seats the quota took out: "out of quota" is not
2894/// actionable, while "judge-2, judge-3 hit a limit" tells the operator which
2895/// agent to replace or which plan to top up.
2896///
2897/// Uses [`RunStatus::display_label`] rather than [`label`]/`as_str` on
2898/// purpose: unlike `label`'s other callers (an internal log line, an
2899/// already-a-bug fallback message), this string becomes `Task::last_error`
2900/// verbatim, which the phone renders in the same alarm-styled box an
2901/// ordinary failure gets — see `web::tests` and `assets/ui/app.js`'s
2902/// `.err` styling. A bare `verified_noop` there would read exactly like the
2903/// failure this whole feature exists to tell apart from one.
2904fn describe(state: &RunState) -> String {
2905    let mut detail = if state.status == RunStatus::Stalled {
2906        let mut seats: Vec<&str> = state.quota.iter().map(|q| q.seat.as_str()).collect();
2907        seats.sort_unstable();
2908        seats.dedup();
2909        if seats.is_empty() {
2910            "the judging panel lost its quorum".to_owned()
2911        } else {
2912            format!(
2913                "the judging panel lost its quorum; quota took out {}",
2914                seats.join(", ")
2915            )
2916        }
2917    } else {
2918        format!("run ended {}", state.status.display_label())
2919    };
2920    if let Some(last) = state.events.last() {
2921        detail.push_str(&format!(" ({}: {})", last.node, last.message));
2922    }
2923    detail.push_str(&format!(" [run {}]", state.id));
2924    detail
2925}
2926
2927/// Upper bound on [`Task::diagnostic`]'s length, in bytes.
2928///
2929/// The task file lives in the backlog indefinitely; a diagnostic is an
2930/// excerpt of the run's own `artifacts/`, not a copy of them, so this has to
2931/// stay small regardless of how much a gate command or a candidate printed.
2932const DIAGNOSTIC_MAX: usize = 4_000;
2933
2934/// Tail kept from a single failing command's output inside a diagnostic.
2935/// Smaller than [`crate::graph`]'s own `OUTPUT_TAIL` on purpose: this is a
2936/// pointer for a human deciding whether to go read the full artifact by hand,
2937/// not a replacement for reading it.
2938const DIAGNOSTIC_OUTPUT_TAIL: usize = 800;
2939
2940/// Assemble a bounded diagnostic excerpt from a held task's own run, so
2941/// `magi task show` says more than the one-line reason in [`describe`].
2942///
2943/// The one-liner answers "where did the run stop"; this answers "what would a
2944/// human have found opening `artifacts/` by hand" — the point of the whole
2945/// feature is the case that one-liner actively misleads on: a run held as "no
2946/// candidate produced a change" can mean the implementer actually finished
2947/// the task (opened a PR, merged it, tagged a release) and only left a clean
2948/// local worktree behind, which reads as "nothing happened" unless someone
2949/// goes and reads what the agent actually said. `None` when the run carries
2950/// none of the three shapes this recognises — an ordinary run held for
2951/// something not diagnosable from `RunState` alone still explains itself
2952/// through `Task::last_error`.
2953fn diagnostic(state: &RunState) -> Option<String> {
2954    let mut parts: Vec<String> = Vec::new();
2955
2956    // Gate failure: which check(s), and the tail of what each printed.
2957    for o in state.gate.iter().filter(|o| !o.ok()) {
2958        parts.push(format!(
2959            "gate `{}` failed ({:?}):\n{}",
2960            o.command,
2961            o.code,
2962            crate::run::tail(&o.output_tail, DIAGNOSTIC_OUTPUT_TAIL)
2963        ));
2964    }
2965
2966    // The land loop gave up because the fixer declined while checks were
2967    // still red: the message already names them (see `land::run`).
2968    if let Some(last) = state
2969        .events
2970        .iter()
2971        .rev()
2972        .find(|e| e.node == "land" && e.message.contains("fixer produced no commit"))
2973    {
2974        parts.push(last.message.clone());
2975    }
2976
2977    // No viable candidate: every implementer's own final word, sanitized the
2978    // same way a judge would have read it, so a run that actually finished
2979    // the job does not read as an unexplained failure. A verified no-op is
2980    // called out ahead of its own summary and apart from an ordinary
2981    // failure's `why` — this is the one candidate shape whose diagnostic a
2982    // human is expected to actually judge, not just skim.
2983    if state.viable().is_empty() {
2984        for c in &state.candidates {
2985            if let Some(evidence) = &c.verified_noop {
2986                parts.push(format!(
2987                    "candidate {} (agent-verified no-op, unconfirmed by magi): {evidence}",
2988                    c.label
2989                ));
2990            } else if !c.summary.trim().is_empty() {
2991                parts.push(format!("candidate {}: {}", c.label, c.summary.trim()));
2992            } else if let Some(why) = &c.failed {
2993                parts.push(format!("candidate {}: {why}", c.label));
2994            }
2995        }
2996    }
2997
2998    if parts.is_empty() {
2999        return None;
3000    }
3001    // `run::tail` prefixes an "N earlier bytes omitted" marker whose own
3002    // length depends on N, so asking it for exactly `DIAGNOSTIC_MAX` can come
3003    // back slightly over. Leave it enough room to always land under the
3004    // limit.
3005    Some(crate::run::tail(
3006        &parts.join("\n\n"),
3007        DIAGNOSTIC_MAX.saturating_sub(100),
3008    ))
3009}
3010
3011/// Stable lower-case name for a run status, for an internal log line and the
3012/// "graph stopped without reaching a terminal status" bug message in
3013/// [`settle`] — never for [`Task::last_error`] itself; see [`describe`]'s own
3014/// doc for why that one reads [`RunStatus::display_label`] instead. One
3015/// definition of a status's name, on the type that owns it: this table used
3016/// to live here as a second copy, and a status renamed in one place would
3017/// have gone on reading correctly in the other.
3018fn label(status: RunStatus) -> &'static str {
3019    status.as_str()
3020}
3021
3022/// Parse a merge mode override.
3023fn merge_mode(mode: &str) -> Result<MergeMode> {
3024    match mode {
3025        "none" => Ok(MergeMode::None),
3026        "local" => Ok(MergeMode::Local),
3027        "pr" => Ok(MergeMode::Pr),
3028        other => bail!("unknown merge mode `{other}`; expected none, local or pr"),
3029    }
3030}
3031
3032/// Take the status lock, recovering from a poisoned one.
3033///
3034/// A panic elsewhere must not silently stop the heartbeat: the status is plain
3035/// data, and the worst a poisoned lock can hold is a stale timestamp.
3036fn lock<T>(mutex: &Mutex<T>) -> MutexGuard<'_, T> {
3037    mutex
3038        .lock()
3039        .unwrap_or_else(std::sync::PoisonError::into_inner)
3040}
3041
3042#[cfg(test)]
3043mod tests {
3044    use super::*;
3045    use crate::queue::{Source, TaskStatus};
3046    use crate::run::{Candidate, CommandOutcome};
3047    use pretty_assertions::assert_eq;
3048
3049    fn task() -> Task {
3050        Task::new(
3051            "add retries".to_owned(),
3052            "add retries".to_owned(),
3053            PathBuf::from("/repo"),
3054            Source::Human,
3055        )
3056    }
3057
3058    /// A runnable task marked to interrupt, with an id fixed for assertions
3059    /// rather than the random one [`Task::new`] mints.
3060    fn interrupt_task(id: &str) -> Task {
3061        let mut t = task();
3062        t.id = id.to_owned();
3063        t.interrupt = true;
3064        t
3065    }
3066
3067    /// An ordinary runnable task with an id fixed for assertions.
3068    fn task_with_id(id: &str) -> Task {
3069        let mut t = task();
3070        t.id = id.to_owned();
3071        t
3072    }
3073
3074    /// A runnable task marked `--urgent`, with an id fixed for assertions.
3075    fn urgent_task(id: &str) -> Task {
3076        let mut t = task();
3077        t.id = id.to_owned();
3078        t.urgent = true;
3079        t
3080    }
3081
3082    /// The land-resume exemption wins outright, whether or not the candidate
3083    /// also happens to be marked [`Task::urgent`]: a resume's own "must not
3084    /// queue behind anything" guarantee cannot be weaker just because the
3085    /// same task was also filed with `--urgent`.
3086    #[test]
3087    fn permit_kind_prefers_a_land_resume_over_the_urgent_slot() {
3088        assert_eq!(permit_kind(true, false), PermitKind::None);
3089        assert_eq!(permit_kind(true, true), PermitKind::None);
3090    }
3091
3092    /// The one property this whole feature exists for: `--urgent` draws from
3093    /// its own slot, never the ordinary `max_concurrent_runs` pool - and an
3094    /// ordinary candidate draws from the ordinary pool exactly as before,
3095    /// untouched by the urgent slot's existence.
3096    #[test]
3097    fn permit_kind_separates_urgent_from_ordinary() {
3098        assert_eq!(permit_kind(false, true), PermitKind::Urgent);
3099        assert_eq!(permit_kind(false, false), PermitKind::Ordinary);
3100    }
3101
3102    /// The exact wiring `attempt` runs before minting anything: a config's
3103    /// `min_free_bytes` in, a task-holding reason naming both numbers out.
3104    /// Free space is injected rather than asked of the real disk - the point
3105    /// of [`disk_gate_with`] existing separately from [`disk_gate`] - so this
3106    /// is deterministic on every machine this test runs on, never dependent
3107    /// on how full the CI runner's own disk happens to be.
3108    #[test]
3109    fn disk_gate_with_holds_a_task_below_the_threshold_and_names_both_numbers() {
3110        let cfg = Config::default();
3111        let repo = Path::new("/any/repo/path");
3112
3113        let reason =
3114            disk_gate_with(repo, &cfg, |_| Ok(1024)).expect("must hold below the threshold");
3115        assert!(reason.contains("1024"), "{reason}");
3116        assert!(
3117            reason.contains(&cfg.disk.min_free_bytes.to_string()),
3118            "{reason}"
3119        );
3120
3121        assert_eq!(
3122            disk_gate_with(repo, &cfg, |_| Ok(cfg.disk.min_free_bytes)),
3123            None,
3124            "exactly at the floor is open"
3125        );
3126        assert_eq!(
3127            disk_gate_with(repo, &cfg, |_| Ok(cfg.disk.min_free_bytes + 1)),
3128            None,
3129            "comfortably above the floor is open"
3130        );
3131    }
3132
3133    #[test]
3134    fn disk_gate_with_opens_unconditionally_when_the_operator_opted_out() {
3135        let mut cfg = Config::default();
3136        cfg.disk.min_free_bytes = 0;
3137        let repo = Path::new("/any/repo/path");
3138        assert_eq!(
3139            disk_gate_with(repo, &cfg, |_| Ok(0)),
3140            None,
3141            "a zero floor never measures at all"
3142        );
3143    }
3144
3145    #[test]
3146    fn disk_gate_with_closes_rather_than_starts_blind_when_it_cannot_measure() {
3147        let cfg = Config::default();
3148        let repo = Path::new("/any/repo/path");
3149        let reason = disk_gate_with(repo, &cfg, |_| Err(anyhow::anyhow!("no df on this box")))
3150            .expect("a measurement failure must close the gate, not open it");
3151        assert!(reason.contains("could not measure"), "{reason}");
3152    }
3153
3154    #[test]
3155    fn no_interrupt_task_leaves_the_sequence_idle_even_with_something_in_flight() {
3156        let ordinary = task();
3157        let next = advance_interrupt(
3158            Interrupt::Idle,
3159            std::slice::from_ref(&ordinary.id),
3160            std::slice::from_ref(&ordinary),
3161        );
3162        assert_eq!(next, Interrupt::Idle);
3163    }
3164
3165    #[test]
3166    fn an_interrupt_task_with_nothing_in_flight_never_starts_a_sequence() {
3167        // Nothing to interrupt - this is just an ordinary candidate, and the
3168        // loop's normal dispatch will pick it up like any other.
3169        let marked = interrupt_task("marked");
3170        let next = advance_interrupt(Interrupt::Idle, &[], std::slice::from_ref(&marked));
3171        assert_eq!(next, Interrupt::Idle);
3172    }
3173
3174    #[test]
3175    fn an_interrupt_task_with_something_in_flight_starts_parking_it() {
3176        let marked = interrupt_task("marked");
3177        let next = advance_interrupt(
3178            Interrupt::Idle,
3179            &["running".to_owned()],
3180            std::slice::from_ref(&marked),
3181        );
3182        assert_eq!(
3183            next,
3184            Interrupt::Parking {
3185                parked: vec!["running".to_owned()],
3186                interrupt_task: "marked".to_owned(),
3187            }
3188        );
3189    }
3190
3191    /// R1-1-2 / R2-1-2: above the default `max_concurrent_runs`, more than
3192    /// one run can be in flight when a task becomes runnable and marked.
3193    /// Parking all of them would mean `Resuming` later has more than one id
3194    /// to release back to ordinary dispatch, which cannot be made safe
3195    /// against that same setting's own extra concurrency slots letting two
3196    /// of them start together - see `advance_interrupt`'s own `Idle` branch.
3197    /// The simplification the task's own constraints ask for: do not begin
3198    /// a sequence at all until the herd settles back to exactly one.
3199    #[test]
3200    fn more_than_one_run_in_flight_never_starts_an_interrupt_sequence() {
3201        let marked = interrupt_task("marked");
3202
3203        let two = advance_interrupt(
3204            Interrupt::Idle,
3205            &["a".to_owned(), "b".to_owned()],
3206            std::slice::from_ref(&marked),
3207        );
3208        assert_eq!(two, Interrupt::Idle);
3209
3210        let none = advance_interrupt(Interrupt::Idle, &[], std::slice::from_ref(&marked));
3211        assert_eq!(none, Interrupt::Idle, "nothing to interrupt either");
3212    }
3213
3214    #[test]
3215    fn parking_holds_until_every_parked_id_has_actually_left_flight() {
3216        let state = Interrupt::Parking {
3217            parked: vec!["running".to_owned()],
3218            interrupt_task: "marked".to_owned(),
3219        };
3220        // Still in flight: no change.
3221        let still_going = advance_interrupt(state.clone(), &["running".to_owned()], &[]);
3222        assert_eq!(still_going, state);
3223
3224        // Left flight, but the interrupt task has not been dispatched yet on
3225        // this tick - stays `Parking` so `interrupt_gate` can let it through,
3226        // as long as it is still runnable.
3227        let stopped_but_not_yet_dispatched =
3228            advance_interrupt(state.clone(), &[], &[interrupt_task("marked")]);
3229        assert_eq!(stopped_but_not_yet_dispatched, state);
3230
3231        // Left flight, and the interrupt task is now in flight itself.
3232        let dispatched = advance_interrupt(state, &["marked".to_owned()], &[]);
3233        assert_eq!(
3234            dispatched,
3235            Interrupt::Running {
3236                parked: vec!["running".to_owned()],
3237                interrupt_task: "marked".to_owned(),
3238            }
3239        );
3240    }
3241
3242    #[test]
3243    fn the_sequence_moves_to_resuming_the_instant_the_interrupt_tasks_own_run_leaves_flight() {
3244        let state = Interrupt::Running {
3245            parked: vec!["running".to_owned()],
3246            interrupt_task: "marked".to_owned(),
3247        };
3248        let still_running = advance_interrupt(state.clone(), &["marked".to_owned()], &[]);
3249        assert_eq!(still_running, state);
3250
3251        // Whatever it ended as - merged, failed, held - is not this
3252        // function's concern: leaving flight is the only trigger, driven
3253        // straight off the same in-flight list `poll` already reaps. It does
3254        // not go straight to `Idle`: see `Interrupt::Running`'s own doc for
3255        // why that would let an unrelated task start ahead of, or alongside,
3256        // the guaranteed resume.
3257        let ended = advance_interrupt(state, &[], &[task_with_id("running")]);
3258        assert_eq!(
3259            ended,
3260            Interrupt::Resuming {
3261                parked: vec!["running".to_owned()]
3262            }
3263        );
3264    }
3265
3266    #[test]
3267    fn resuming_ends_the_instant_a_parked_task_is_seen_in_flight() {
3268        let state = Interrupt::Resuming {
3269            parked: vec!["running".to_owned()],
3270        };
3271        let still_waiting = advance_interrupt(state.clone(), &[], &[task_with_id("running")]);
3272        assert_eq!(still_waiting, state);
3273
3274        let dispatched = advance_interrupt(state, &["running".to_owned()], &[]);
3275        assert_eq!(dispatched, Interrupt::Idle);
3276    }
3277
3278    /// R1-2-1: an interrupt task that stops being runnable - held, blocked,
3279    /// or otherwise moved on by an operator with no claim standing in the
3280    /// way - must not wedge the sequence (and so the whole loop's dispatch,
3281    /// via `interrupt_gate`) waiting forever for a dispatch that can never
3282    /// come. The parked run still gets its resume.
3283    #[test]
3284    fn an_interrupt_task_that_stops_being_runnable_abandons_the_wait_without_losing_the_parked_run()
3285    {
3286        let state = Interrupt::Parking {
3287            parked: vec!["running".to_owned()],
3288            interrupt_task: "marked".to_owned(),
3289        };
3290        // `marked` has been held/blocked/deleted since the sequence began:
3291        // it no longer appears in `runnable` at all.
3292        let next = advance_interrupt(state, &[], &[]);
3293        assert_eq!(
3294            next,
3295            Interrupt::Resuming {
3296                parked: vec!["running".to_owned()]
3297            },
3298            "abandoning the interrupt must not abandon the resume it owes"
3299        );
3300    }
3301
3302    /// The same abandonment, one step later: `Resuming` itself must not wait
3303    /// forever for a parked task that has since become unrunnable.
3304    #[test]
3305    fn resuming_abandons_a_parked_task_that_stops_being_runnable() {
3306        let state = Interrupt::Resuming {
3307            parked: vec!["running".to_owned()],
3308        };
3309        let next = advance_interrupt(state, &[], &[]);
3310        assert_eq!(
3311            next,
3312            Interrupt::Idle,
3313            "nothing is left to wait for; the loop must not stay wedged"
3314        );
3315    }
3316
3317    #[test]
3318    fn disabled_by_config_the_sequence_can_never_leave_idle() {
3319        let marked = interrupt_task("marked");
3320        let next = advance_interrupt_tick(
3321            false,
3322            Interrupt::Idle,
3323            &["running".to_owned()],
3324            std::slice::from_ref(&marked),
3325        );
3326        assert_eq!(
3327            next,
3328            Interrupt::Idle,
3329            "an unmarked, unconfigured daemon must behave exactly as before"
3330        );
3331    }
3332
3333    #[test]
3334    fn the_gate_blocks_everyone_while_something_parked_is_still_in_flight() {
3335        let state = Interrupt::Parking {
3336            parked: vec!["running".to_owned()],
3337            interrupt_task: "marked".to_owned(),
3338        };
3339        let candidates = vec![interrupt_task("marked"), task()];
3340        let allowed = interrupt_gate(&state, &["running".to_owned()], candidates);
3341        assert!(
3342            allowed.is_empty(),
3343            "nothing may dispatch - not even the interrupt task itself - \
3344             until the parked run has actually stopped"
3345        );
3346    }
3347
3348    /// `interrupt_gate` knows nothing about [`Task::urgent`] and must not
3349    /// grow a special case for it: an urgent candidate that is not the
3350    /// interrupt task withholds exactly like an ordinary one for as long as
3351    /// the parked run it is racing has not actually left flight. A version
3352    /// of this feature once bypassed the gate for urgent candidates, on the
3353    /// theory that a genuinely separate concurrency lane could not collide
3354    /// with 75dd's own; it could, and did - the bypassed task dispatched
3355    /// alongside a run 75dd's own state machine had only *asked* to park,
3356    /// not yet confirmed gone, which is exactly the second run
3357    /// `Interrupt::Parking`'s own doc says must never happen. There is no
3358    /// tick during Parking/Running/Resuming where letting an extra
3359    /// candidate through is safe, urgent or not - the parked run may still
3360    /// be genuinely mid-call.
3361    #[test]
3362    fn urgent_gains_no_exemption_from_an_active_interrupt_sequence() {
3363        for state in [
3364            Interrupt::Parking {
3365                parked: vec!["running".to_owned()],
3366                interrupt_task: "marked".to_owned(),
3367            },
3368            Interrupt::Running {
3369                parked: vec!["running".to_owned()],
3370                interrupt_task: "marked".to_owned(),
3371            },
3372            Interrupt::Resuming {
3373                parked: vec!["running".to_owned()],
3374            },
3375        ] {
3376            let candidates = vec![
3377                interrupt_task("marked"),
3378                urgent_task("hot"),
3379                task_with_id("ordinary"),
3380            ];
3381            let allowed = interrupt_gate(&state, &["running".to_owned()], candidates);
3382            assert!(
3383                !allowed.iter().any(|t| t.id == "hot"),
3384                "an urgent candidate must wait out the same gate as anything \
3385                 else while the run it would run alongside has not actually \
3386                 left flight, for state {state:?}: {allowed:?}"
3387            );
3388        }
3389    }
3390
3391    /// The one case where marking a task `--urgent` and `interrupt` at once
3392    /// is not redundant: once the interrupt sequence's own guaranteed
3393    /// resume/dispatch actually admits the task, it is unaffected by also
3394    /// carrying `urgent` - `interrupt_gate` decides purely on identity,
3395    /// never on the urgent flag.
3396    #[test]
3397    fn a_task_marked_both_urgent_and_interrupt_is_admitted_once_the_gate_itself_says_so() {
3398        let state = Interrupt::Resuming {
3399            parked: vec!["hot".to_owned()],
3400        };
3401        let candidates = vec![urgent_task("hot"), task()];
3402        let allowed = interrupt_gate(&state, &[], candidates);
3403        assert_eq!(
3404            allowed.iter().filter(|t| t.id == "hot").count(),
3405            1,
3406            "the gate's own decision is unaffected by the urgent flag: {allowed:?}"
3407        );
3408    }
3409
3410    #[test]
3411    fn the_gate_lets_only_the_interrupt_task_through_once_parked_work_has_stopped() {
3412        let state = Interrupt::Parking {
3413            parked: vec!["running".to_owned()],
3414            interrupt_task: "marked".to_owned(),
3415        };
3416        let other = task();
3417        let candidates = vec![interrupt_task("marked"), other.clone()];
3418        let allowed = interrupt_gate(&state, &[], candidates);
3419        assert_eq!(allowed.len(), 1);
3420        assert_eq!(allowed[0].id, "marked");
3421    }
3422
3423    #[test]
3424    fn the_gate_blocks_everyone_while_the_interrupt_task_itself_is_in_flight() {
3425        let state = Interrupt::Running {
3426            parked: vec!["running".to_owned()],
3427            interrupt_task: "marked".to_owned(),
3428        };
3429        let candidates = vec![task(), task()];
3430        let allowed = interrupt_gate(&state, &["marked".to_owned()], candidates);
3431        assert!(allowed.is_empty());
3432    }
3433
3434    /// R1-1-1 / R1-1-2: even when more than one task was in flight when the
3435    /// sequence began (only reachable above the default
3436    /// `max_concurrent_runs = 1`), `Resuming` offers at most one of them -
3437    /// never both in the same tick, which is what "exactly one resume, no
3438    /// simultaneous run" actually requires structurally rather than by
3439    /// coincidence of how many ordinary slots happen to be free.
3440    #[test]
3441    fn the_gate_offers_at_most_one_candidate_while_resuming_even_with_two_parked() {
3442        let state = Interrupt::Resuming {
3443            parked: vec!["a".to_owned(), "c".to_owned()],
3444        };
3445        let candidates = vec![task_with_id("a"), task_with_id("c"), task_with_id("other")];
3446        let allowed = interrupt_gate(&state, &[], candidates);
3447        assert_eq!(
3448            allowed.len(),
3449            1,
3450            "at most one candidate may be offered while resuming: {allowed:?}"
3451        );
3452        assert_eq!(allowed[0].id, "a");
3453    }
3454
3455    #[test]
3456    fn the_gate_offers_nothing_while_resuming_if_no_parked_task_is_runnable() {
3457        let state = Interrupt::Resuming {
3458            parked: vec!["a".to_owned()],
3459        };
3460        let allowed = interrupt_gate(&state, &[], vec![task_with_id("other")]);
3461        assert!(allowed.is_empty());
3462    }
3463
3464    /// The invariant the completion criteria ask for by name: across a whole
3465    /// simulated sequence, there is never a tick where the gate would let
3466    /// through both the parked run's resume and the interrupt task, and
3467    /// exactly one candidate resumes the instant the interrupt task's run
3468    /// ends - never zero, never more than one.
3469    #[test]
3470    fn a_full_sequence_never_gates_two_runs_through_at_once_and_resumes_exactly_one() {
3471        let running = task(); // id: whatever `Task::new` minted
3472        let marked = interrupt_task("marked");
3473
3474        let mut state = Interrupt::Idle;
3475        // Tick 1: `running` is in flight, `marked` becomes runnable.
3476        let in_flight = vec![running.id.clone()];
3477        state = advance_interrupt_tick(true, state, &in_flight, std::slice::from_ref(&marked));
3478        let gated = interrupt_gate(&state, &in_flight, vec![marked.clone(), running.clone()]);
3479        assert!(gated.is_empty(), "still waiting on `running` to park");
3480
3481        // Tick 2: `running` parked and left flight; nothing dispatched yet.
3482        state = advance_interrupt_tick(true, state, &[], &[marked.clone(), running.clone()]);
3483        let gated = interrupt_gate(&state, &[], vec![marked.clone(), running.clone()]);
3484        assert_eq!(
3485            gated.iter().map(|t| t.id.as_str()).collect::<Vec<_>>(),
3486            vec!["marked"],
3487            "only the interrupt task may be offered to the dispatcher now"
3488        );
3489
3490        // Tick 3: `marked` is now in flight (dispatched from tick 2's gate).
3491        state = advance_interrupt_tick(
3492            true,
3493            state,
3494            &["marked".to_owned()],
3495            std::slice::from_ref(&running),
3496        );
3497        let gated = interrupt_gate(
3498            &state,
3499            &["marked".to_owned()],
3500            vec![marked.clone(), running.clone()],
3501        );
3502        assert!(
3503            gated.is_empty(),
3504            "the parked run must not be offered back while the interrupt \
3505             task is still running"
3506        );
3507
3508        // Tick 4: `marked`'s run reached a terminal status and left flight.
3509        // A higher-priority ordinary task `other` is also runnable now - it
3510        // must not be let through instead of, or alongside, `running`.
3511        let other = task_with_id("other");
3512        state = advance_interrupt_tick(true, state, &[], &[running.clone(), other.clone()]);
3513        assert_eq!(
3514            state,
3515            Interrupt::Resuming {
3516                parked: vec![running.id.clone()]
3517            }
3518        );
3519        let gated = interrupt_gate(&state, &[], vec![other.clone(), running.clone()]);
3520        assert_eq!(
3521            gated.iter().map(|t| t.id.as_str()).collect::<Vec<_>>(),
3522            vec![running.id.as_str()],
3523            "exactly the parked run resumes - not the unrelated task, even \
3524             though it was offered first"
3525        );
3526
3527        // Tick 5: `running` is now in flight (dispatched from tick 4's
3528        // gate). Only now does the sequence end and ordinary dispatch fully
3529        // resume.
3530        state = advance_interrupt_tick(
3531            true,
3532            state,
3533            std::slice::from_ref(&running.id),
3534            std::slice::from_ref(&other),
3535        );
3536        assert_eq!(state, Interrupt::Idle);
3537        let gated = interrupt_gate(
3538            &state,
3539            std::slice::from_ref(&running.id),
3540            vec![other.clone()],
3541        );
3542        assert_eq!(
3543            gated.iter().map(|t| t.id.as_str()).collect::<Vec<_>>(),
3544            vec![other.id.as_str()],
3545            "ordinary dispatch is unrestricted again"
3546        );
3547    }
3548
3549    #[test]
3550    fn every_run_status_settles_the_task_it_came_from() {
3551        // run status, resulting task status, attempts still standing after one
3552        let table = [
3553            (RunStatus::Merged, TaskStatus::Done, 1),
3554            (RunStatus::Ready, TaskStatus::Done, 1),
3555            (RunStatus::Stalled, TaskStatus::Failed, 0),
3556            (RunStatus::Blocked, TaskStatus::Failed, 1),
3557            (RunStatus::Failed, TaskStatus::Failed, 1),
3558            (RunStatus::VerifiedNoop, TaskStatus::Held, 1),
3559            (RunStatus::Prep, TaskStatus::Failed, 1),
3560            (RunStatus::Implementing, TaskStatus::Failed, 1),
3561            (RunStatus::Judging, TaskStatus::Failed, 1),
3562            (RunStatus::Deliberating, TaskStatus::Failed, 1),
3563            (RunStatus::Voting, TaskStatus::Failed, 1),
3564            (RunStatus::Reviewing, TaskStatus::Failed, 1),
3565            (RunStatus::Gating, TaskStatus::Failed, 1),
3566        ];
3567        for (run, want, attempts) in table {
3568            let mut t = task();
3569            t.start("20260902-000000-aaaa".to_owned());
3570            settle(
3571                &mut t,
3572                Verdict {
3573                    status: run,
3574                    left_pr: false,
3575                    parked: false,
3576                    quota_hit: matches!(run, RunStatus::Stalled),
3577                    no_viable_candidates: false,
3578                },
3579                "why",
3580                2,
3581            );
3582            assert_eq!(t.status, want, "task status after {}", label(run));
3583            assert_eq!(t.attempts, attempts, "attempts after {}", label(run));
3584        }
3585    }
3586
3587    #[test]
3588    fn a_quota_stall_costs_the_task_no_attempt_but_a_block_does() {
3589        let mut stalled = task();
3590        stalled.start("20260902-000000-aaaa".to_owned());
3591        settle(
3592            &mut stalled,
3593            Verdict {
3594                status: RunStatus::Stalled,
3595                left_pr: false,
3596                parked: false,
3597                quota_hit: true,
3598                no_viable_candidates: false,
3599            },
3600            "quota",
3601            1,
3602        );
3603        assert_eq!(stalled.attempts, 0);
3604        assert!(
3605            stalled.status.runnable(),
3606            "a machine problem must leave the task in line"
3607        );
3608
3609        let mut blocked = task();
3610        blocked.start("20260902-000000-aaaa".to_owned());
3611        settle(
3612            &mut blocked,
3613            Verdict {
3614                status: RunStatus::Blocked,
3615                left_pr: false,
3616                parked: false,
3617                quota_hit: false,
3618                no_viable_candidates: false,
3619            },
3620            "findings open",
3621            1,
3622        );
3623        assert_eq!(blocked.attempts, 1);
3624        assert_eq!(
3625            blocked.status,
3626            TaskStatus::Held,
3627            "the last attempt hands the task to a human"
3628        );
3629    }
3630
3631    #[test]
3632    fn a_run_that_opened_a_pull_request_is_never_re_competed() {
3633        // Attempts to spare: without the pull request this task would go
3634        // straight back in line and run the whole competition again.
3635        let mut delivered = task();
3636        delivered.start("20260903-080619-01c2".to_owned());
3637        settle(
3638            &mut delivered,
3639            Verdict {
3640                status: RunStatus::Blocked,
3641                left_pr: true,
3642                parked: false,
3643                quota_hit: false,
3644                no_viable_candidates: false,
3645            },
3646            "no check status",
3647            4,
3648        );
3649        assert_eq!(
3650            delivered.status,
3651            TaskStatus::Held,
3652            "a pull request waiting on CI or a person is not a retryable failure"
3653        );
3654        assert!(
3655            !delivered.status.runnable(),
3656            "the loop must not pick this task up again"
3657        );
3658        assert_eq!(
3659            delivered.last_error.as_deref(),
3660            Some("no check status"),
3661            "the operator needs to be told what the gate was waiting for"
3662        );
3663
3664        // The same status without a pull request is a plain failure, and with
3665        // attempts left it is retried.
3666        let mut empty_handed = task();
3667        empty_handed.start("20260903-080619-01c2".to_owned());
3668        settle(
3669            &mut empty_handed,
3670            Verdict {
3671                status: RunStatus::Blocked,
3672                left_pr: false,
3673                parked: false,
3674                quota_hit: false,
3675                no_viable_candidates: false,
3676            },
3677            "findings open",
3678            4,
3679        );
3680        assert_eq!(empty_handed.status, TaskStatus::Failed);
3681        assert!(empty_handed.status.runnable());
3682    }
3683
3684    #[test]
3685    fn a_verified_noop_run_hands_off_rather_than_closing_or_auto_retrying() {
3686        // Every candidate agreed, with evidence, that nothing belonged in the
3687        // worktree. That is not a confirmed success to close automatically -
3688        // a human still has to check the claim - and it is not an ordinary
3689        // failure either, so this settles exactly like a pull request nobody
3690        // merged yet: `Held`, same as `Blocked` with a PR.
3691        let mut noop = task();
3692        noop.start("20260912-131304-391f".to_owned());
3693        settle(
3694            &mut noop,
3695            Verdict {
3696                status: RunStatus::VerifiedNoop,
3697                left_pr: false,
3698                parked: false,
3699                quota_hit: false,
3700                no_viable_candidates: true,
3701            },
3702            "candidate A: already fixed by b32cfc4, on main",
3703            4,
3704        );
3705        assert_eq!(
3706            noop.status,
3707            TaskStatus::Held,
3708            "an unverified claim is a request for a human, not a failure"
3709        );
3710        assert!(
3711            !noop.status.runnable(),
3712            "the loop must not requeue this on the same unverified claim"
3713        );
3714        // `Task::release` resets attempts to zero the moment a human looks at
3715        // the evidence and lets it run again, so it does not matter here
3716        // whether the one attempt already spent stays spent - what matters is
3717        // that nothing retries this task unattended in the meantime.
3718        assert_eq!(noop.attempts, 1);
3719    }
3720
3721    #[test]
3722    fn parking_costs_the_task_no_attempt_and_leaves_it_in_line() {
3723        // Parking is the operator asking for the process back - to replace the
3724        // binary, most of all. The run's work is intact on disk, so this is
3725        // not a failed attempt, and charging for it would mean a few upgrades
3726        // could exhaust a budget meant for agents that misbehaved.
3727        let mut parked = task();
3728        parked.start("20260903-183634-2d98".to_owned());
3729        settle(
3730            &mut parked,
3731            Verdict {
3732                status: RunStatus::Implementing,
3733                left_pr: false,
3734                quota_hit: false,
3735                parked: true,
3736                no_viable_candidates: false,
3737            },
3738            "parked after `implementing`",
3739            2,
3740        );
3741        assert_eq!(parked.attempts, 0, "a park is refunded");
3742        assert!(
3743            parked.status.runnable(),
3744            "and the task stays in line so the next loop resumes its run"
3745        );
3746        assert_eq!(
3747            parked.last_error.as_deref(),
3748            Some("parked after `implementing`"),
3749            "the card says where it stopped"
3750        );
3751
3752        // Without the park flag the same non-terminal status is what it always
3753        // was: `execute` returning mid-flight, which is a bug and spends an
3754        // attempt so a task cannot loop on it forever.
3755        let mut broken = task();
3756        broken.start("20260903-183634-2d98".to_owned());
3757        settle(
3758            &mut broken,
3759            Verdict {
3760                status: RunStatus::Implementing,
3761                left_pr: false,
3762                quota_hit: false,
3763                parked: false,
3764                no_viable_candidates: false,
3765            },
3766            "returned mid-flight",
3767            2,
3768        );
3769        assert_eq!(broken.attempts, 1);
3770    }
3771
3772    #[test]
3773    fn only_a_rate_limit_buys_the_task_its_attempt_back() {
3774        // Run e633: quorum lost because two judges answered with the wrong
3775        // JSON shape, `quota: []`. Refunding that takes the bound off the
3776        // retry loop, and each retry pays for a fresh hour-long implement
3777        // wave before it can fail the same way.
3778        let mut flaky = task();
3779        flaky.start("20260903-123023-e633".to_owned());
3780        settle(
3781            &mut flaky,
3782            Verdict {
3783                status: RunStatus::Stalled,
3784                left_pr: false,
3785                parked: false,
3786                quota_hit: false,
3787                no_viable_candidates: false,
3788            },
3789            "verdict rests on 1 of 3 judges",
3790            2,
3791        );
3792        assert_eq!(
3793            flaky.attempts, 1,
3794            "flakiness spends an attempt, so `max_attempts` still bounds it"
3795        );
3796        assert!(flaky.status.runnable(), "and it is still worth retrying");
3797
3798        // The same status, lost to a rate limit, is the machine's fault.
3799        let mut limited = task();
3800        limited.start("20260903-123023-e633".to_owned());
3801        settle(
3802            &mut limited,
3803            Verdict {
3804                status: RunStatus::Stalled,
3805                left_pr: false,
3806                parked: false,
3807                quota_hit: true,
3808                no_viable_candidates: false,
3809            },
3810            "judge-2, judge-3 out of quota",
3811            2,
3812        );
3813        assert_eq!(limited.attempts, 0, "a quota window is refunded");
3814        assert!(limited.status.runnable());
3815
3816        // And the bound really binds: a task that keeps stalling on flakiness
3817        // reaches a human instead of running the roster forever.
3818        let mut worn = task();
3819        for _ in 0..2 {
3820            worn.release();
3821        }
3822        worn.start("20260903-123023-e633".to_owned());
3823        worn.attempts = 2;
3824        settle(
3825            &mut worn,
3826            Verdict {
3827                status: RunStatus::Stalled,
3828                left_pr: false,
3829                parked: false,
3830                quota_hit: false,
3831                no_viable_candidates: false,
3832            },
3833            "no quorum again",
3834            2,
3835        );
3836        assert_eq!(worn.status, TaskStatus::Held);
3837        assert!(!worn.status.runnable());
3838    }
3839
3840    #[test]
3841    fn a_quota_wipeout_that_leaves_nothing_to_judge_also_costs_no_attempt() {
3842        // The implement wave loses every seat to the same rate limit and
3843        // `after_implement` bails with nothing viable, which surfaces as
3844        // `Failed` rather than `Stalled`. That is the same machine fact the
3845        // `Stalled`-quota row already refunds, and must be refunded the same
3846        // way, or a quota outage quietly holds every task it touches instead
3847        // of leaving them in line for the reset.
3848        let mut wiped_out = task();
3849        wiped_out.start("20260907-025000-a1b2".to_owned());
3850        settle(
3851            &mut wiped_out,
3852            Verdict {
3853                status: RunStatus::Failed,
3854                left_pr: false,
3855                parked: false,
3856                quota_hit: true,
3857                no_viable_candidates: true,
3858            },
3859            "no candidate produced a change; nothing to judge",
3860            2,
3861        );
3862        assert_eq!(wiped_out.attempts, 0, "a total quota wipeout is refunded");
3863        assert!(
3864            wiped_out.status.runnable(),
3865            "a machine problem must leave the task in line"
3866        );
3867
3868        // This is the exemption that must stay narrow: a candidate that did
3869        // produce a change, and then failed for some other reason, still
3870        // spends the attempt even though a seat elsewhere hit its quota.
3871        // Otherwise every ordinary failure that happens to share a run with
3872        // an unrelated rate limit would be refunded for free.
3873        let mut partial_progress = task();
3874        partial_progress.start("20260907-025500-c3d4".to_owned());
3875        settle(
3876            &mut partial_progress,
3877            Verdict {
3878                status: RunStatus::Failed,
3879                left_pr: false,
3880                parked: false,
3881                quota_hit: true,
3882                no_viable_candidates: false,
3883            },
3884            "gate failed on the winning candidate",
3885            2,
3886        );
3887        assert_eq!(
3888            partial_progress.attempts, 1,
3889            "a candidate that actually produced a change spends the attempt \
3890             even though some other seat hit its quota"
3891        );
3892        assert!(partial_progress.status.runnable());
3893    }
3894
3895    #[test]
3896    fn reclaim_refunds_a_recovered_quota_wipeout_the_same_way_a_live_settle_does() {
3897        // `reclaim` builds its own `Verdict` from a `RunState` it loads off
3898        // disk, and that construction must reach the same conclusion as the
3899        // one `attempt` builds from a live run, or a crash at exactly the
3900        // wrong moment gives a recovered task a different policy than one a
3901        // daemon finished settling itself.
3902        let mut t = task();
3903        t.start("20260907-025000-a1b2".to_owned());
3904        let mut state = run_state(RunStatus::Failed);
3905        state.quota.push(QuotaLoss {
3906            seat: "cand-a".to_owned(),
3907            node: "implement".to_owned(),
3908            at: Timestamp::now(),
3909            reset: None,
3910        });
3911        assert!(
3912            state.viable().is_empty(),
3913            "no candidate was added, so nothing is viable"
3914        );
3915        reclaim(&mut t, Some(state), 2);
3916        assert_eq!(t.attempts, 0, "a recovered quota wipeout is refunded");
3917        assert!(t.status.runnable());
3918    }
3919
3920    #[test]
3921    fn a_held_task_is_never_offered_to_the_loop() {
3922        let dir = tempfile::tempdir().unwrap();
3923        let queue = Queue::at(dir.path().to_path_buf());
3924        for (n, priority) in [(1, 0), (2, 5), (3, 5)] {
3925            let mut t = task();
3926            t.id = format!("2026090{n}-000000-000{n}");
3927            t.priority = priority;
3928            queue.put(&mut t).unwrap();
3929        }
3930        let mut held = task();
3931        held.id = "20260909-000000-9999".to_owned();
3932        held.priority = 99;
3933        held.hold_machine(None);
3934        queue.put(&mut held).unwrap();
3935
3936        let order: Vec<String> = runnable(&queue).into_iter().map(|t| t.id).collect();
3937        assert_eq!(order.len(), 3);
3938        assert!(!order.contains(&held.id));
3939        assert_eq!(
3940            order.first().cloned(),
3941            queue.next_runnable().map(|t| t.id),
3942            "the loop's first candidate is exactly what the queue offers"
3943        );
3944        assert_eq!(
3945            order,
3946            vec![
3947                "20260902-000000-0002".to_owned(),
3948                "20260903-000000-0003".to_owned(),
3949                "20260901-000000-0001".to_owned(),
3950            ],
3951            "priority first, then oldest, so nothing starves"
3952        );
3953    }
3954
3955    #[test]
3956    fn sweep_removes_an_old_unparseable_lock_and_keeps_a_live_one() {
3957        let dir = tempfile::tempdir().unwrap();
3958        let queue = Queue::at(dir.path().to_path_buf());
3959        let mut old = task();
3960        old.id = "20260101-000000-old0".to_owned();
3961        queue.put(&mut old).unwrap();
3962        let mut fresh = task();
3963        fresh.id = "20260101-000000-new0".to_owned();
3964        queue.put(&mut fresh).unwrap();
3965
3966        // No parseable pid at all, so age is the only signal there is to
3967        // check - unlike a real `Queue::claim`, which always names a real,
3968        // and therefore alive, pid this test cannot fake as dead.
3969        std::fs::write(dir.path().join(format!("{}.lock", old.id)), "not a pid").unwrap();
3970        std::thread::sleep(Duration::from_millis(60));
3971        let live = queue.claim(&fresh.id).unwrap();
3972
3973        let swept = sweep_stale_claims(&queue, Duration::from_millis(50));
3974        assert_eq!(swept, vec![old.id.clone()]);
3975        assert!(
3976            queue.claim(&old.id).is_ok(),
3977            "an unparseable lock older than the threshold is swept"
3978        );
3979        assert!(
3980            queue.claim(&fresh.id).is_err(),
3981            "a live pid protects its lock regardless of age"
3982        );
3983        drop(live);
3984    }
3985
3986    #[test]
3987    fn an_old_lock_whose_pid_is_still_alive_is_never_swept_by_age_alone() {
3988        // The regression this guards: `sweep` now runs concurrently with
3989        // every attempt this daemon itself has spawned (see
3990        // `InFlightGuard`), not only between them the way a single
3991        // sequential loop once did. A run that legitimately outlives
3992        // `older_than` still has this very process's own live pid sitting in
3993        // its own lock file on every later sweep, and deciding by age alone
3994        // would delete that still-valid claim out from under the attempt
3995        // that holds it - which `reclaim_orphaned_running` would then read
3996        // as abandoned and hand to a second, competing attempt.
3997        let dir = tempfile::tempdir().unwrap();
3998        let queue = Queue::at(dir.path().to_path_buf());
3999        let mut t = task();
4000        t.id = "20260101-000000-live".to_owned();
4001        queue.put(&mut t).unwrap();
4002
4003        let claim = queue.claim(&t.id).unwrap();
4004        std::thread::sleep(Duration::from_millis(60));
4005
4006        let swept = sweep_stale_claims(&queue, Duration::from_millis(50));
4007        assert!(
4008            swept.is_empty(),
4009            "a lock naming a live pid must never be swept by age, no matter how old: {swept:?}"
4010        );
4011        assert!(
4012            queue.claim(&t.id).is_err(),
4013            "the lock still protects its task"
4014        );
4015        drop(claim);
4016    }
4017
4018    /// このテストプロセスにはなり得ない決定的なフィクスチャ PID。
4019    /// OS 上の状態は意図的に無関係で、各利用箇所が方針問い合わせを注入する。
4020    fn injected_dead_pid() -> u32 {
4021        std::process::id().checked_add(1).unwrap_or(1)
4022    }
4023
4024    #[test]
4025    fn a_lock_naming_a_dead_pid_is_swept_at_once_regardless_of_age() {
4026        let dir = tempfile::tempdir().unwrap();
4027        let queue = Queue::at(dir.path().to_path_buf());
4028        let mut t = task();
4029        t.id = "20260101-000000-dead".to_owned();
4030        queue.put(&mut t).unwrap();
4031        let dead_pid = injected_dead_pid();
4032
4033        // Written directly rather than through `Queue::claim`, which would
4034        // stamp this test process's own very much alive pid and defeat the
4035        // point: this is what a `.lock` left by a `SIGKILL`ed daemon looks
4036        // like moments after it died, not six hours later.
4037        std::fs::write(
4038            dir.path().join(format!("{}.lock", t.id)),
4039            dead_pid.to_string(),
4040        )
4041        .unwrap();
4042
4043        let swept = sweep_stale_claims_with(&queue, Duration::from_secs(6 * 60 * 60), |pid| {
4044            pid != dead_pid
4045        });
4046        assert_eq!(
4047            swept,
4048            vec![t.id.clone()],
4049            "a dead owner is reclaimed immediately, not after STALE_CLAIM"
4050        );
4051        assert!(queue.claim(&t.id).is_ok(), "the task is claimable again");
4052    }
4053
4054    #[test]
4055    fn sweeping_on_every_poll_catches_a_lock_that_appears_after_the_first_sweep() {
4056        let dir = tempfile::tempdir().unwrap();
4057        let queue = Queue::at(dir.path().to_path_buf());
4058        let mut t = task();
4059        t.id = "20260101-000000-late".to_owned();
4060        queue.put(&mut t).unwrap();
4061        let dead_pid = injected_dead_pid();
4062
4063        // Tick one, standing in for the sweep `poll` already runs at
4064        // startup: nothing to find yet.
4065        assert!(
4066            sweep_stale_claims(&queue, Duration::from_secs(6 * 60 * 60)).is_empty(),
4067            "nothing has claimed the task yet"
4068        );
4069
4070        // A second daemon claims the task and dies before it ever writes
4071        // `running`, well after this loop's own startup sweep already ran.
4072        std::fs::write(
4073            dir.path().join(format!("{}.lock", t.id)),
4074            dead_pid.to_string(),
4075        )
4076        .unwrap();
4077
4078        // Tick two, standing in for a poll long into this daemon's uptime:
4079        // the same function, called again, notices what only just appeared -
4080        // proving the sweep is not a one-shot startup check.
4081        let swept = sweep_stale_claims_with(&queue, Duration::from_secs(6 * 60 * 60), |pid| {
4082            pid != dead_pid
4083        });
4084        assert_eq!(swept, vec![t.id.clone()]);
4085    }
4086
4087    #[test]
4088    fn a_running_task_behind_a_dead_daemons_lock_recovers_once_swept_and_keeps_its_history() {
4089        // `reclaim_orphaned_running` looks up the task's last run, which
4090        // touches `run::home()`; the first call anywhere in this binary wins,
4091        // so this is a no-op if another test already pinned one, and either
4092        // way the run id below is never written under it.
4093        crate::run::set_home(std::env::temp_dir().join("magi-daemon-test-home"));
4094        let dir = tempfile::tempdir().unwrap();
4095        let queue = Queue::at(dir.path().to_path_buf());
4096        let mut t = task();
4097        t.id = "20260101-000000-crsh".to_owned();
4098        t.status = TaskStatus::Running;
4099        t.attempts = 1;
4100        // No `run.json` behind this id: standing in for a run this test does
4101        // not need to make readable, since the point is the lock, not the
4102        // recovery table `reclaim` already has its own tests for.
4103        t.runs.push("20260904-000000-4043".to_owned());
4104        queue.put(&mut t).unwrap();
4105        let dead_pid = injected_dead_pid();
4106
4107        // The crashed daemon's own claim, naming a pid nothing on the
4108        // machine holds anymore.
4109        std::fs::write(
4110            dir.path().join(format!("{}.lock", t.id)),
4111            dead_pid.to_string(),
4112        )
4113        .unwrap();
4114
4115        // Before the lock is swept the task looks claimed, and
4116        // `reclaim_orphaned_running` must leave it alone - this is exactly
4117        // the bug: a `running` task stranded behind a dead daemon's lock,
4118        // invisible to the claim-as-proof check because the lock outlived
4119        // the process that wrote it.
4120        assert!(reclaim_orphaned_running(&queue, 2).is_empty());
4121        assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Running);
4122
4123        let swept = sweep_stale_claims_with(&queue, Duration::from_secs(6 * 60 * 60), |pid| {
4124            pid != dead_pid
4125        });
4126        assert_eq!(swept, vec![t.id.clone()]);
4127
4128        let reclaimed = reclaim_orphaned_running(&queue, 2);
4129        assert_eq!(reclaimed, vec![t.id.clone()]);
4130        let after = queue.get(&t.id).unwrap();
4131        assert_eq!(
4132            after.status,
4133            TaskStatus::Held,
4134            "no run.json to recover from, so a human is asked"
4135        );
4136        assert_eq!(
4137            after.runs,
4138            vec!["20260904-000000-4043".to_owned()],
4139            "the crashed run's id is kept as evidence, not discarded"
4140        );
4141    }
4142
4143    #[test]
4144    fn a_lock_is_kept_when_the_process_query_is_unavailable() {
4145        let dir = tempfile::tempdir().unwrap();
4146        let queue = Queue::at(dir.path().to_path_buf());
4147        let mut t = task();
4148        t.id = "20260101-000000-unknown".to_owned();
4149        queue.put(&mut t).unwrap();
4150        let dead_pid = injected_dead_pid();
4151        std::fs::write(
4152            dir.path().join(format!("{}.lock", t.id)),
4153            dead_pid.to_string(),
4154        )
4155        .unwrap();
4156
4157        let swept = sweep_stale_claims_with(&queue, Duration::ZERO, |_| true);
4158        assert!(swept.is_empty(), "an unknown pid must keep its lock");
4159        assert!(queue.claim(&t.id).is_err(), "the lock remains protective");
4160    }
4161
4162    fn run_state(status: RunStatus) -> RunState {
4163        let mut state = RunState::new(
4164            PathBuf::from("/repo"),
4165            "main".to_owned(),
4166            "abc1234def".to_owned(),
4167            "add retries".to_owned(),
4168            Config::default(),
4169        );
4170        state.status = status;
4171        state
4172    }
4173
4174    fn candidate(label: char, summary: &str, empty: bool, failed: Option<&str>) -> Candidate {
4175        Candidate {
4176            index: 0,
4177            label,
4178            agent: "claude".to_owned(),
4179            branch: format!("magi/x/{label}"),
4180            worktree: PathBuf::from("/repo"),
4181            summary: summary.to_owned(),
4182            stat: String::new(),
4183            files: 0,
4184            commits: usize::from(!empty),
4185            empty,
4186            failed: failed.map(str::to_owned),
4187            verified_noop: None,
4188            duration_ms: 0,
4189            folded: false,
4190        }
4191    }
4192
4193    #[test]
4194    fn diagnostic_names_the_failing_gate_checks_and_their_output() {
4195        let mut state = run_state(RunStatus::Blocked);
4196        state.gate = vec![
4197            CommandOutcome {
4198                command: "cargo make check".to_owned(),
4199                code: Some(0),
4200                output_tail: "ok".to_owned(),
4201                duration_ms: 0,
4202                resource_blocked: false,
4203            },
4204            CommandOutcome {
4205                command: "cargo test".to_owned(),
4206                code: Some(101),
4207                output_tail: "thread 'x' panicked: assertion failed".to_owned(),
4208                duration_ms: 0,
4209                resource_blocked: false,
4210            },
4211        ];
4212        let d = diagnostic(&state).expect("a failing gate must produce a diagnostic");
4213        assert!(d.contains("cargo test"), "{d}");
4214        assert!(
4215            !d.contains("cargo make check"),
4216            "a passing check is not a diagnostic: {d}"
4217        );
4218        assert!(d.contains("assertion failed"), "{d}");
4219    }
4220
4221    #[test]
4222    fn diagnostic_names_the_checks_the_fixer_gave_up_in_front_of() {
4223        let mut state = run_state(RunStatus::Blocked);
4224        state.event(
4225            "land",
4226            "stopped: the fixer produced no commit while 2 check(s) were failing \
4227             (build, lint); stopping instead of looping on an unchanged tree",
4228        );
4229        let d = diagnostic(&state).expect("a stalled land loop must produce a diagnostic");
4230        assert!(d.contains("build"), "{d}");
4231        assert!(d.contains("lint"), "{d}");
4232        assert!(d.contains("fixer produced no commit"), "{d}");
4233    }
4234
4235    #[test]
4236    fn describe_never_leaves_a_verified_noop_reading_as_a_bare_status_code() {
4237        // `describe`'s output becomes `Task::last_error` verbatim, and the
4238        // phone renders that in the same alarm-styled box an ordinary
4239        // failure gets. A bare `verified_noop` there would read exactly like
4240        // the failure this status exists to be told apart from.
4241        let state = run_state(RunStatus::VerifiedNoop);
4242        let d = describe(&state);
4243        assert!(
4244            d.contains("agent-verified no-op"),
4245            "expected the display label, not the wire spelling: {d}"
4246        );
4247        assert!(!d.contains("verified_noop"), "{d}");
4248    }
4249
4250    #[test]
4251    fn diagnostic_carries_a_candidates_own_final_word_when_none_was_viable() {
4252        // The whole point of the feature: a run held as "no candidate produced
4253        // a change" can mean the implementer actually finished the task and
4254        // only left a clean local tree behind - see AGENTS.md on this exact
4255        // failure mode. The diagnostic has to carry what the agent actually
4256        // said, not just the fact that nothing was there to judge.
4257        let mut state = run_state(RunStatus::Failed);
4258        state.candidates = vec![candidate(
4259            'A',
4260            "opened pull request #42, merged it, tagged v1.2.3 and published the release",
4261            true,
4262            None,
4263        )];
4264        let d = diagnostic(&state).expect("an empty candidate with a summary must be surfaced");
4265        assert!(d.contains("candidate A"), "{d}");
4266        assert!(d.contains("tagged v1.2.3"), "{d}");
4267    }
4268
4269    #[test]
4270    fn diagnostic_falls_back_to_a_candidates_failure_reason_when_it_has_no_summary() {
4271        let mut state = run_state(RunStatus::Failed);
4272        state.candidates = vec![candidate('A', "", true, Some("agent timed out"))];
4273        let d = diagnostic(&state).expect("a candidate's own failure reason must be surfaced");
4274        assert!(d.contains("candidate A"), "{d}");
4275        assert!(d.contains("agent timed out"), "{d}");
4276    }
4277
4278    #[test]
4279    fn diagnostic_is_none_when_nothing_recognisable_explains_the_hold() {
4280        // A viable candidate existed, the gate never ran, and nothing land
4281        // said matches - `Task::last_error` is left to explain this one alone.
4282        let mut state = run_state(RunStatus::Failed);
4283        state.candidates = vec![candidate('A', "did the work", false, None)];
4284        assert!(diagnostic(&state).is_none());
4285    }
4286
4287    #[test]
4288    fn diagnostic_is_bounded_however_much_a_run_printed() {
4289        let mut state = run_state(RunStatus::Blocked);
4290        state.gate = vec![
4291            CommandOutcome {
4292                command: "cargo test".to_owned(),
4293                code: Some(101),
4294                output_tail: "x".repeat(50_000),
4295                duration_ms: 0,
4296                resource_blocked: false,
4297            },
4298            CommandOutcome {
4299                command: "cargo clippy".to_owned(),
4300                code: Some(1),
4301                output_tail: "y".repeat(50_000),
4302                duration_ms: 0,
4303                resource_blocked: false,
4304            },
4305        ];
4306        state.candidates = vec![
4307            candidate('A', &"z".repeat(50_000), true, None),
4308            candidate('B', &"w".repeat(50_000), true, None),
4309        ];
4310        let d = diagnostic(&state).expect("plenty here to diagnose");
4311        assert!(
4312            d.len() <= DIAGNOSTIC_MAX,
4313            "diagnostic grew to {} bytes, unbounded",
4314            d.len()
4315        );
4316    }
4317
4318    #[test]
4319    fn settle_and_diagnose_attaches_a_diagnostic_only_once_the_task_is_held() {
4320        let mut state = run_state(RunStatus::Blocked);
4321        state.gate = vec![CommandOutcome {
4322            command: "cargo test".to_owned(),
4323            code: Some(101),
4324            output_tail: "assertion failed".to_owned(),
4325            duration_ms: 0,
4326            resource_blocked: false,
4327        }];
4328        let verdict = Verdict {
4329            status: RunStatus::Blocked,
4330            left_pr: false,
4331            quota_hit: false,
4332            parked: false,
4333            no_viable_candidates: false,
4334        };
4335
4336        // Attempt one of two still has a retry coming: no diagnostic yet, the
4337        // task is going to run again and this run's evidence would go stale.
4338        let mut t = task();
4339        t.start("run-1".to_owned());
4340        settle_and_diagnose(&mut t, verdict, "gate failed", 2, &state);
4341        assert_eq!(t.status, TaskStatus::Failed);
4342        assert!(t.diagnostic.is_none());
4343
4344        // Attempt two exhausts the budget: now it is held, and the
4345        // diagnostic is what `magi task show` has to say more than one line.
4346        t.start("run-2".to_owned());
4347        settle_and_diagnose(&mut t, verdict, "gate failed", 2, &state);
4348        assert_eq!(t.status, TaskStatus::Held);
4349        let d = t.diagnostic.expect("a held task must carry its diagnostic");
4350        assert!(d.contains("cargo test"), "{d}");
4351    }
4352
4353    fn approval_question(run: &str) -> ask::Question {
4354        ask::Question::new(
4355            run.to_owned(),
4356            land::APPROVAL_NODE.to_owned(),
4357            "land".to_owned(),
4358            "merge?".to_owned(),
4359            String::new(),
4360            vec!["merge".to_owned(), "hold".to_owned()],
4361        )
4362    }
4363
4364    #[test]
4365    fn land_resume_state_leaves_a_fresh_open_question_waiting() {
4366        crate::run::set_home(std::env::temp_dir().join("magi-daemon-test-home"));
4367        let mut state = run_state(RunStatus::Landing);
4368        state.id = "20260101-000000-fre1".to_owned();
4369        state.parked = true;
4370        state.save().unwrap();
4371        ask::Questions::open()
4372            .put(&mut approval_question(&state.id))
4373            .unwrap();
4374
4375        let mut t = task();
4376        t.runs.push(state.id.clone());
4377        assert_eq!(
4378            land_resume_state(&t),
4379            LandResume::StillWaiting,
4380            "nobody has answered and the timeout has not passed"
4381        );
4382    }
4383
4384    #[test]
4385    fn land_resume_state_abandons_a_question_that_outlived_answer_timeout() {
4386        // `ask::ask_and_wait`'s own deadline used to retire a question
4387        // nobody answered; land's approval bypasses that wait (see
4388        // `land::approval_gate`), so this is now the only place
4389        // `graph.answer_timeout` is enforced for a land approval at all.
4390        crate::run::set_home(std::env::temp_dir().join("magi-daemon-test-home"));
4391        let mut state = run_state(RunStatus::Landing);
4392        state.id = "20260101-000000-exp1".to_owned();
4393        state.parked = true;
4394        state.config.graph.answer_timeout = 60;
4395        state.save().unwrap();
4396
4397        let store = ask::Questions::open();
4398        let mut q = approval_question(&state.id);
4399        q.asked_at = Timestamp::now() - jiff::SignedDuration::from_secs(120);
4400        store.put(&mut q).unwrap();
4401
4402        let mut t = task();
4403        t.runs.push(state.id.clone());
4404        assert_eq!(
4405            land_resume_state(&t),
4406            LandResume::Ready,
4407            "an expired question must not be waited on forever"
4408        );
4409
4410        let after = store.get(&q.id).unwrap();
4411        assert!(
4412            !after.status.open(),
4413            "the question is abandoned, not silently ignored"
4414        );
4415        assert!(
4416            after.resolution().is_none(),
4417            "an abandoned question is not read as a decision"
4418        );
4419    }
4420
4421    #[test]
4422    fn reclaim_settles_a_running_task_against_its_last_run() {
4423        let mut t = task();
4424        t.start("20260904-000000-4043".to_owned());
4425        reclaim(&mut t, Some(run_state(RunStatus::Ready)), 2);
4426        assert_eq!(
4427            t.status,
4428            TaskStatus::Done,
4429            "a run that actually finished must not stay `running` forever"
4430        );
4431    }
4432
4433    #[test]
4434    fn reclaim_reuses_the_same_retry_policy_as_a_live_settle() {
4435        // A blocked run with attempts left goes back to `Failed`, exactly as
4436        // it would from `attempt` itself - `reclaim` must not invent a second
4437        // policy for a task a daemon merely stopped without reporting.
4438        let mut t = task();
4439        t.start("20260904-000000-4043".to_owned());
4440        reclaim(&mut t, Some(run_state(RunStatus::Blocked)), 2);
4441        assert_eq!(t.status, TaskStatus::Failed);
4442        assert!(t.status.runnable());
4443    }
4444
4445    #[test]
4446    fn reclaim_holds_a_running_task_whose_run_cannot_be_found() {
4447        let mut t = task();
4448        t.start("20260904-000000-4043".to_owned());
4449        reclaim(&mut t, None, 2);
4450        assert_eq!(t.status, TaskStatus::Held);
4451        assert!(
4452            t.last_error
4453                .as_deref()
4454                .is_some_and(|e| e.contains("running")),
4455            "the operator needs to know why this task was held"
4456        );
4457    }
4458
4459    #[test]
4460    fn orphaned_running_tasks_are_reclaimed_but_live_ones_are_left_alone() {
4461        let dir = tempfile::tempdir().unwrap();
4462        let queue = Queue::at(dir.path().to_path_buf());
4463
4464        // No run recorded, so this never has to touch `RunState::load`.
4465        let mut orphaned = task();
4466        orphaned.id = "20260904-000000-orph".to_owned();
4467        orphaned.status = TaskStatus::Running;
4468        orphaned.attempts = 1;
4469        queue.put(&mut orphaned).unwrap();
4470
4471        let mut alive = task();
4472        alive.id = "20260904-000000-live".to_owned();
4473        alive.status = TaskStatus::Running;
4474        alive.attempts = 1;
4475        queue.put(&mut alive).unwrap();
4476        let _held_by_a_live_daemon = queue.claim(&alive.id).unwrap();
4477
4478        let mut queued = task();
4479        queued.id = "20260904-000000-wait".to_owned();
4480        queue.put(&mut queued).unwrap();
4481
4482        let reclaimed = reclaim_orphaned_running(&queue, 2);
4483        assert_eq!(reclaimed, vec![orphaned.id.clone()]);
4484
4485        assert_eq!(
4486            queue.get(&orphaned.id).unwrap().status,
4487            TaskStatus::Held,
4488            "nothing was driving it and there was no run to recover"
4489        );
4490        assert_eq!(
4491            queue.get(&alive.id).unwrap().status,
4492            TaskStatus::Running,
4493            "a live claim must protect the task it belongs to"
4494        );
4495        assert_eq!(queue.get(&queued.id).unwrap().status, TaskStatus::Queued);
4496    }
4497
4498    /// Read a run.json back from an explicit `home`, the same way
4499    /// `reclaim_abandoned_runs` itself does - never through the
4500    /// process-global `RunState::load`, which this test's own `home` (an
4501    /// isolated tempdir, never pinned into the shared `OnceLock`) does not
4502    /// use at all.
4503    fn read_run_under(home: &Path, id: &str) -> RunState {
4504        let body = std::fs::read_to_string(home.join("runs").join(id).join("run.json")).unwrap();
4505        serde_json::from_str(&body).unwrap()
4506    }
4507
4508    #[test]
4509    fn reclaim_abandoned_runs_fails_a_run_whose_active_seats_are_all_provably_dead() {
4510        let dir = tempfile::tempdir().unwrap();
4511        let home = dir.path().to_path_buf();
4512        let now = Timestamp::now();
4513        let overrun_seat = || crate::run::ActiveSeat {
4514            node: "implement".to_owned(),
4515            started_at: now - jiff::SignedDuration::new(21_000, 0),
4516            timeout_secs: 3_600,
4517            attempt: 0,
4518            task: None,
4519            command: None,
4520            index: None,
4521            total: None,
4522        };
4523
4524        let mut dead = run_state(RunStatus::Implementing);
4525        dead.id = "20260101-000000-dead".to_owned();
4526        dead.active.insert("impl-A".to_owned(), overrun_seat());
4527        // A `driver_pid` the injected query below confirms gone outright —
4528        // `liveness` reads this as `Dead`, not merely "no daemon claims it".
4529        dead.driver_pid = Some(4242);
4530        dead.save_under(&home).unwrap();
4531
4532        // Same shape, but a live daemon's heartbeat names it: must be left
4533        // exactly alone, however far past its own timeout the seat sits.
4534        let mut alive = run_state(RunStatus::Implementing);
4535        alive.id = "20260101-000000-aliv".to_owned();
4536        alive.active.insert("impl-A".to_owned(), overrun_seat());
4537        alive.save_under(&home).unwrap();
4538        let mut status = Status::new();
4539        status.current = vec![Current {
4540            task: "20260101-000000-task".to_owned(),
4541            run: alive.id.clone(),
4542        }];
4543        write_status_to(&home.join("daemon.json"), &status).unwrap();
4544
4545        // The abandoned seat left an open question behind: nobody is left to
4546        // read an answer once the run is failed, and this must not wait for
4547        // some later daemon startup's own sweep to notice that.
4548        let questions = Questions::at(home.join("questions"));
4549        let mut q = ask::Question::new(
4550            dead.id.clone(),
4551            "implement".to_owned(),
4552            "impl-A".to_owned(),
4553            "Which storage backend?".to_owned(),
4554            String::new(),
4555            vec!["SQLite".to_owned(), "Redis".to_owned()],
4556        );
4557        questions.put(&mut q).unwrap();
4558
4559        let abandoned = reclaim_abandoned_runs_with(
4560            &home,
4561            now,
4562            |pid| if pid == 4242 { Some(false) } else { None },
4563            |_| panic!("a query answering Dead outright needs no identity corroboration"),
4564        );
4565        assert_eq!(abandoned, vec![dead.id.clone()]);
4566
4567        let reloaded = read_run_under(&home, &dead.id);
4568        assert_eq!(reloaded.status, RunStatus::Failed);
4569        assert!(reloaded.active.is_empty());
4570        assert!(
4571            !questions.get(&q.id).unwrap().status.open(),
4572            "the failed run's own open question must be settled in the same pass"
4573        );
4574
4575        let still_alive = read_run_under(&home, &alive.id);
4576        assert_eq!(
4577            still_alive.status,
4578            RunStatus::Implementing,
4579            "a live daemon's claim protects it"
4580        );
4581        assert!(!still_alive.active.is_empty());
4582    }
4583
4584    /// The exact shape a review round flagged as broken: `magi serve` running
4585    /// in this same `home` scans *every* run on disk, including a manual
4586    /// `magi review` / `magi run` this daemon never started and that
4587    /// therefore claims no heartbeat of its own. Before this scan asked
4588    /// `liveness` rather than just `is_working_on`, a manual run whose active
4589    /// seat merely ran a little past its own timeout — the CLI finishing up,
4590    /// its result still being collected — got wiped and failed by a daemon
4591    /// that had nothing to do with it, out from under a process that was
4592    /// still very much running.
4593    #[test]
4594    fn reclaim_abandoned_runs_leaves_a_live_manual_run_alone_even_though_no_daemon_claims_it() {
4595        let dir = tempfile::tempdir().unwrap();
4596        let home = dir.path().to_path_buf();
4597        let now = Timestamp::now();
4598
4599        let mut manual = run_state(RunStatus::Reviewing);
4600        manual.id = "20260101-000000-manl".to_owned();
4601        manual.active.insert(
4602            "review-1".to_owned(),
4603            crate::run::ActiveSeat {
4604                node: "review".to_owned(),
4605                started_at: now - jiff::SignedDuration::new(21_000, 0),
4606                timeout_secs: 3_600,
4607                attempt: 0,
4608                task: None,
4609                command: None,
4610                index: None,
4611                total: None,
4612            },
4613        );
4614        // Not claimed by any daemon (no `daemon.json` at all in this `home`),
4615        // but a real, still-running process: `liveness` must corroborate this
4616        // as `Live`, not read the missing daemon claim as death.
4617        manual.driver_pid = Some(4242);
4618        manual.driver_started_at = Some("2026-09-22T10:00:00Z".to_owned());
4619        manual.save_under(&home).unwrap();
4620
4621        let abandoned = reclaim_abandoned_runs_with(
4622            &home,
4623            now,
4624            |pid| if pid == 4242 { Some(true) } else { None },
4625            |pid| {
4626                if pid == 4242 {
4627                    Some("2026-09-22T10:00:00Z".to_owned())
4628                } else {
4629                    None
4630                }
4631            },
4632        );
4633        assert!(
4634            abandoned.is_empty(),
4635            "a manual run a real process is still driving must never be reclaimed: {abandoned:?}"
4636        );
4637
4638        let reloaded = read_run_under(&home, &manual.id);
4639        assert_eq!(reloaded.status, RunStatus::Reviewing);
4640        assert!(!reloaded.active.is_empty());
4641    }
4642
4643    #[test]
4644    fn an_already_claimed_task_is_skipped_rather_than_failed() {
4645        let dir = tempfile::tempdir().unwrap();
4646        let queue = Queue::at(dir.path().to_path_buf());
4647        let mut only = task();
4648        queue.put(&mut only).unwrap();
4649
4650        let _elsewhere = queue.claim(&only.id).unwrap();
4651        let candidates = runnable(&queue);
4652        assert_eq!(candidates.len(), 1, "the task is still runnable");
4653        assert!(
4654            queue.claim(&candidates[0].id).is_err(),
4655            "the loop cannot take a claim somebody else holds"
4656        );
4657
4658        let after = queue.get(&only.id).unwrap();
4659        assert_eq!(after.status, TaskStatus::Queued);
4660        assert_eq!(
4661            after.attempts, 0,
4662            "losing the race is not an attempt at the task"
4663        );
4664        assert_eq!(after.last_error, None);
4665    }
4666
4667    #[test]
4668    fn the_status_file_round_trips_and_its_heartbeat_advances() {
4669        let dir = tempfile::tempdir().unwrap();
4670        let path = dir.path().join("daemon.json");
4671
4672        let mut status = Status::new();
4673        status.idle = false;
4674        status.completed = 7;
4675        status.current = vec![Current {
4676            task: "20260902-000000-t111".to_owned(),
4677            run: "20260902-000001-r111".to_owned(),
4678        }];
4679        write_status_to(&path, &status).unwrap();
4680        let first: Status = serde_json::from_str(&std::fs::read_to_string(&path).unwrap()).unwrap();
4681        assert_eq!(first.schema, SCHEMA);
4682        assert_eq!(first.pid, std::process::id());
4683        assert!(!first.idle);
4684        assert_eq!(first.completed, 7);
4685        assert_eq!(first.current, status.current);
4686        assert!(
4687            !path.with_extension("json.tmp").exists(),
4688            "the temp file is renamed, not left behind"
4689        );
4690
4691        std::thread::sleep(Duration::from_millis(5));
4692        status.updated_at = Timestamp::now();
4693        status.polls = 3;
4694        write_status_to(&path, &status).unwrap();
4695        let second: Status =
4696            serde_json::from_str(&std::fs::read_to_string(&path).unwrap()).unwrap();
4697        assert!(
4698            second.updated_at > first.updated_at,
4699            "a reader can only detect staleness if the heartbeat moves"
4700        );
4701        assert_eq!(
4702            second.started_at, first.started_at,
4703            "the start time is not a heartbeat"
4704        );
4705        assert_eq!(second.polls, 3);
4706    }
4707
4708    #[test]
4709    fn reading_counts_as_running_only_while_its_heartbeat_is_fresh() {
4710        let dir = tempfile::tempdir().unwrap();
4711
4712        assert!(read_status(dir.path()).is_none(), "no file, no daemon");
4713
4714        let mut status = Status::new();
4715        status.updated_at = Timestamp::now() - jiff::SignedDuration::from_secs(60);
4716        write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
4717        let stale = read_status(dir.path()).unwrap();
4718        assert!(
4719            !stale.running(Timestamp::now()),
4720            "a minute without a heartbeat is a dead daemon, not a busy one"
4721        );
4722        assert!(stale.age_secs(Timestamp::now()).is_some_and(|s| s >= 55));
4723
4724        status.updated_at = Timestamp::now();
4725        write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
4726        let fresh = read_status(dir.path()).unwrap();
4727        assert!(fresh.running(Timestamp::now()));
4728    }
4729
4730    #[test]
4731    fn only_a_live_daemon_on_this_very_run_counts_as_working_on_it() {
4732        let dir = tempfile::tempdir().unwrap();
4733        let now = Timestamp::now();
4734        let mine = "20260903-080619-01c2";
4735
4736        assert!(
4737            !is_working_on(dir.path(), mine, now),
4738            "no status file means nobody is working on anything"
4739        );
4740
4741        let mut status = Status::new();
4742        status.current = vec![Current {
4743            task: "20260903-080340-0167".to_owned(),
4744            run: mine.to_owned(),
4745        }];
4746        status.updated_at = now;
4747        write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
4748        assert!(is_working_on(dir.path(), mine, now));
4749        assert!(
4750            !is_working_on(dir.path(), "20260903-105039-3cbf", now),
4751            "a daemon busy with one run is not working on another"
4752        );
4753
4754        // A killed daemon stops writing heartbeats but leaves the file behind
4755        // naming the run it died in. That run must not be undeletable forever.
4756        status.updated_at = now - jiff::SignedDuration::from_secs(600);
4757        write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
4758        assert!(
4759            !is_working_on(dir.path(), mine, now),
4760            "a stale heartbeat is a dead daemon, so its run is a leftover"
4761        );
4762    }
4763
4764    #[test]
4765    fn is_working_on_short_matches_by_the_worktree_bays_own_name() {
4766        let dir = tempfile::tempdir().unwrap();
4767        let now = Timestamp::now();
4768
4769        assert!(
4770            !is_working_on_short(dir.path(), "01c2", now),
4771            "no status file means nobody is working on anything"
4772        );
4773
4774        let mut status = Status::new();
4775        status.current = vec![Current {
4776            task: "20260903-080340-0167".to_owned(),
4777            run: "20260903-080619-01c2".to_owned(),
4778        }];
4779        status.updated_at = now;
4780        write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
4781        assert!(
4782            is_working_on_short(dir.path(), "01c2", now),
4783            "the run's short id is the last block of its full id"
4784        );
4785        assert!(
4786            !is_working_on_short(dir.path(), "3cbf", now),
4787            "a daemon busy with one worktree bay is not working on another"
4788        );
4789    }
4790
4791    #[test]
4792    fn a_newer_status_file_still_yields_a_reading() {
4793        let dir = tempfile::tempdir().unwrap();
4794        // A field this build has never heard of must not turn the reading into
4795        // nothing at all; that is the whole reason the reader is permissive.
4796        std::fs::write(
4797            dir.path().join("daemon.json"),
4798            serde_json::json!({
4799                "schema": 2,
4800                "updated_at": Timestamp::now().to_string(),
4801                "idle": true,
4802                "surprise": { "nested": [1, 2, 3] },
4803            })
4804            .to_string(),
4805        )
4806        .unwrap();
4807
4808        let reading = read_status(dir.path()).expect("a forward-compatible read");
4809        assert!(reading.running(Timestamp::now()));
4810        assert!(reading.idle);
4811        assert!(reading.current.is_empty());
4812    }
4813
4814    #[test]
4815    fn an_older_daemons_single_object_current_still_reads_as_a_one_item_list() {
4816        // A daemon started before `current` became a list keeps writing this
4817        // shape on every heartbeat until it is restarted. A rolling upgrade
4818        // - a newer `magi web` or `magi doctor` reading an older `magi
4819        // serve`'s heartbeat - must still see the run it is on, not "no
4820        // daemon" from a type mismatch failing the whole struct.
4821        let dir = tempfile::tempdir().unwrap();
4822        std::fs::write(
4823            dir.path().join("daemon.json"),
4824            serde_json::json!({
4825                "schema": 1,
4826                "pid": 4242,
4827                "updated_at": Timestamp::now().to_string(),
4828                "idle": false,
4829                "current": {"task": "20260902-140501-aaaa", "run": "20260902-140502-bbbb"},
4830                "completed": 3,
4831                "polls": 9,
4832            })
4833            .to_string(),
4834        )
4835        .unwrap();
4836
4837        let reading = read_status(dir.path()).expect("an older shape must still parse");
4838        assert!(reading.running(Timestamp::now()));
4839        assert_eq!(
4840            reading.current,
4841            vec![Current {
4842                task: "20260902-140501-aaaa".to_owned(),
4843                run: "20260902-140502-bbbb".to_owned(),
4844            }]
4845        );
4846    }
4847
4848    #[test]
4849    fn an_absent_or_null_current_reads_as_idle_not_a_parse_failure() {
4850        let dir = tempfile::tempdir().unwrap();
4851        std::fs::write(
4852            dir.path().join("daemon.json"),
4853            serde_json::json!({
4854                "schema": 1,
4855                "updated_at": Timestamp::now().to_string(),
4856                "idle": true,
4857                "current": null,
4858            })
4859            .to_string(),
4860        )
4861        .unwrap();
4862        let with_null = read_status(dir.path()).expect("null must still parse");
4863        assert!(with_null.current.is_empty());
4864
4865        std::fs::write(
4866            dir.path().join("daemon.json"),
4867            serde_json::json!({
4868                "schema": 1,
4869                "updated_at": Timestamp::now().to_string(),
4870                "idle": true,
4871            })
4872            .to_string(),
4873        )
4874        .unwrap();
4875        let absent = read_status(dir.path()).expect("a missing field must still parse");
4876        assert!(absent.current.is_empty());
4877    }
4878
4879    #[test]
4880    fn a_task_without_a_repository_runs_in_the_daemons_default() {
4881        let fallback = Path::new("/default");
4882        let mut blank = task();
4883        blank.repo = PathBuf::new();
4884        assert_eq!(repo_for(&blank, fallback), PathBuf::from("/default"));
4885        let mut dot = task();
4886        dot.repo = PathBuf::from(".");
4887        assert_eq!(repo_for(&dot, fallback), PathBuf::from("/default"));
4888        assert_eq!(
4889            repo_for(&task(), fallback),
4890            PathBuf::from("/repo"),
4891            "a task that names a repository keeps it"
4892        );
4893    }
4894
4895    #[test]
4896    fn a_solo_task_runs_with_one_candidate_and_a_plain_task_keeps_the_configs() {
4897        // Three seats said out loud. What `solo` promises is one candidate
4898        // *whatever the config asks for*, so the contrast has to be a number
4899        // this test owns - it used to be `Config::default()`'s, which became
4900        // 1 when one implementation became the default and left the two
4901        // halves of this test asserting the same thing.
4902        let mut solo_cfg = Config::default();
4903        solo_cfg.graph.candidates = 3;
4904        let mut solo_task = task();
4905        solo_task.solo = true;
4906        apply_solo(&mut solo_cfg, &solo_task);
4907        assert_eq!(solo_cfg.graph.candidates, 1);
4908
4909        let mut plain_cfg = Config::default();
4910        plain_cfg.graph.candidates = 3;
4911        let plain_task = task();
4912        assert!(!plain_task.solo);
4913        apply_solo(&mut plain_cfg, &plain_task);
4914        assert_eq!(
4915            plain_cfg.graph.candidates, 3,
4916            "a task that did not ask to run alone keeps the config's candidates"
4917        );
4918    }
4919
4920    #[test]
4921    fn merge_overrides_are_parsed_or_refused() {
4922        assert_eq!(merge_mode("none").unwrap(), MergeMode::None);
4923        assert_eq!(merge_mode("local").unwrap(), MergeMode::Local);
4924        assert_eq!(merge_mode("pr").unwrap(), MergeMode::Pr);
4925        assert!(merge_mode("squash").is_err());
4926    }
4927
4928    #[test]
4929    fn quota_wait_uses_a_future_reset_time_capped_and_falls_back_otherwise() {
4930        let now = Timestamp::now();
4931        let fallback = Duration::from_secs(300);
4932        let cap = Duration::from_secs(1800);
4933
4934        // No reset hint at all: the fallback.
4935        assert_eq!(quota_wait(None, now, fallback, cap), fallback);
4936
4937        // A reset ten minutes out, well inside the cap: waited for exactly.
4938        let soon = now + jiff::SignedDuration::from_secs(600);
4939        assert_eq!(
4940            quota_wait(Some(soon), now, fallback, cap),
4941            Duration::from_secs(600)
4942        );
4943
4944        // A reset already in the past is not trusted: the fallback, not a
4945        // zero or negative wait that would spin the loop right back around.
4946        let past = now - jiff::SignedDuration::from_secs(60);
4947        assert_eq!(quota_wait(Some(past), now, fallback, cap), fallback);
4948
4949        // A reset further out than the cap is trusted for direction but not
4950        // for magnitude: a parsing slip must not sleep the loop for a day.
4951        let far = now + jiff::SignedDuration::from_secs(3 * 3600);
4952        assert_eq!(quota_wait(Some(far), now, fallback, cap), cap);
4953    }
4954
4955    #[test]
4956    fn parse_reset_hint_reads_the_claude_cli_shape_and_rolls_a_past_clock_to_tomorrow() {
4957        let now = "2026-09-07T02:50:00Z".parse::<Timestamp>().unwrap();
4958
4959        let at = parse_reset_hint("4:50am (UTC)", now, now).expect("a recognised shape parses");
4960        assert_eq!(at.to_string(), "2026-09-07T04:50:00Z");
4961
4962        // Same clock reading, but it has already gone by today: read as
4963        // tomorrow's, since the CLI would not still be reporting a limit past
4964        // its own stated reset.
4965        let already_past =
4966            parse_reset_hint("1:00am (UTC)", now, now).expect("a recognised shape parses");
4967        assert_eq!(already_past.to_string(), "2026-09-08T01:00:00Z");
4968
4969        assert!(
4970            parse_reset_hint("session limit reached", now, now).is_none(),
4971            "free text with no recognised shape is not guessed at"
4972        );
4973        assert!(
4974            parse_reset_hint("4:50am (Nowhere/Fake)", now, now).is_none(),
4975            "an unresolvable zone name is not guessed at either"
4976        );
4977    }
4978
4979    #[test]
4980    fn parse_reset_hint_reads_the_codex_cli_shape_with_no_year_rollover_needed() {
4981        let now = "2026-09-07T02:50:00Z".parse::<Timestamp>().unwrap();
4982
4983        let at = parse_reset_hint(
4984            "You've hit your usage limit. Visit \
4985             https://chatgpt.com/codex/settings/usage to purchase more \
4986             credits or try again at Sep 19th, 2026 5:10 PM.",
4987            now,
4988            now,
4989        )
4990        .expect("the codex reset wording is a recognised shape");
4991        assert_eq!(at.to_string(), "2026-09-19T17:10:00Z");
4992
4993        // The month is explicit, so a date already earlier in the same
4994        // sentence-implied year than `now` is trusted as written rather than
4995        // rolled forward a year the way the bracketed shape rolls a
4996        // same-day clock reading to tomorrow.
4997        let earlier = parse_reset_hint("try again at Jan 2nd, 2026 1:00 AM.", now, now)
4998            .expect("an explicit year needs no rollover");
4999        assert_eq!(earlier.to_string(), "2026-01-02T01:00:00Z");
5000
5001        assert!(
5002            parse_reset_hint("try again at Sep 19th, 26 5:10 PM.", now, now).is_none(),
5003            "a two-digit year is not the documented shape and is not guessed at"
5004        );
5005        assert!(
5006            parse_reset_hint("try again at Sept 19th, 2026 5:10 PM.", now, now).is_none(),
5007            "a four-letter month name is not the documented three-letter abbreviation"
5008        );
5009        assert!(
5010            parse_reset_hint("try again at Sep 19th, 2026 5:10 PM (UTC).", now, now).is_none(),
5011            "an explicit zone on the dated shape is a format nobody has \
5012             documented, and is refused rather than guessed at as UTC"
5013        );
5014    }
5015
5016    #[test]
5017    fn parse_reset_hint_reads_agys_relative_shape_from_when_the_loss_was_recorded() {
5018        let now = "2026-09-24T12:00:00Z".parse::<Timestamp>().unwrap();
5019        let recorded = "2026-09-24T08:00:00Z".parse::<Timestamp>().unwrap();
5020
5021        let at = parse_reset_hint("in 1h2m49s", now, recorded).expect("agy's shape parses");
5022        assert_eq!(at.as_second() - recorded.as_second(), 3769);
5023
5024        let partial = parse_reset_hint("in 45m", now, recorded).expect("units are optional");
5025        assert_eq!(partial.as_second() - recorded.as_second(), 45 * 60);
5026
5027        for bad in ["in ", "in 45", "in 3x", "in m", "in 1h junk", "1h2m"] {
5028            assert!(
5029                parse_reset_hint(bad, now, recorded).is_none(),
5030                "{bad:?} must not be guessed at"
5031            );
5032        }
5033    }
5034
5035    /// A loop whose queue lives in a temp tree and whose poll interval is far
5036    /// longer than the test's patience, so anything that waits out a poll
5037    /// instead of noticing the stop fails rather than merely being slow.
5038    fn idle_loop(dir: &Path) -> (Opts, Queue, PathBuf, PathBuf, PathBuf) {
5039        let config = dir.join("magi.toml");
5040        std::fs::write(
5041            &config,
5042            "[disk]\nmin_free_bytes = 0\nauto_fold = false\ncache_limit_bytes = 0\n",
5043        )
5044        .unwrap();
5045        let opts = Opts {
5046            poll: Duration::from_secs(30),
5047            config: Some(config),
5048            // The explicit fixture config keeps startup cleanup from reading
5049            // machine configuration. This fictional repository likewise
5050            // keeps any best-effort git cleanup away from this checkout.
5051            repo: dir.join("repo"),
5052            ..Opts::default()
5053        };
5054        // The status file goes in a directory that does not exist yet, so its
5055        // creation is itself evidence the loop published one. `worktrees`
5056        // must be just as fictional: the janitor reclaims worktrees under it
5057        // for real, and a test that let it fall through to
5058        // `crate::run::default_worktree_root()` would have it reclaim
5059        // worktrees out of the operator's real `~/wt/<repo>`, not a fixture -
5060        // which is exactly what happened before this function took the
5061        // parameter at all.
5062        let home = dir.join("home");
5063        let worktrees = dir.join("wt");
5064        (
5065            opts,
5066            Queue::at(dir.join("queue")),
5067            home.join("daemon.json"),
5068            home,
5069            worktrees,
5070        )
5071    }
5072
5073    #[test]
5074    fn a_stop_is_idempotent_and_once_set_stays_set() {
5075        let stop = Stop::new();
5076        assert!(!stop.stopped());
5077
5078        stop.stop();
5079        assert!(stop.stopped());
5080        stop.stop();
5081        assert!(stop.stopped(), "a second stop is not a toggle");
5082
5083        let shared = stop.clone();
5084        assert!(
5085            shared.stopped(),
5086            "a clone is the same stop; that is how the loop and its caller share one"
5087        );
5088    }
5089
5090    #[test]
5091    fn only_a_stop_with_a_run_in_flight_reads_as_finishing() {
5092        let stop = Stop::new();
5093        stop.enter();
5094        assert!(
5095            !stop.finishing(),
5096            "a busy loop nobody has asked to stop is just running"
5097        );
5098
5099        stop.stop();
5100        assert!(
5101            stop.finishing(),
5102            "a stop asked for mid-run has not landed until the run is settled"
5103        );
5104
5105        stop.exit();
5106        assert!(
5107            !stop.finishing(),
5108            "once the run is settled the stop has landed and there is nothing to finish"
5109        );
5110    }
5111
5112    #[test]
5113    fn finishing_stays_true_until_the_last_of_several_runs_exits() {
5114        let stop = Stop::new();
5115        stop.enter();
5116        stop.enter();
5117        stop.stop();
5118        assert!(stop.finishing(), "two runs still in flight");
5119
5120        stop.exit();
5121        assert!(
5122            stop.finishing(),
5123            "one run finished, but a sibling is still working"
5124        );
5125
5126        stop.exit();
5127        assert!(
5128            !stop.finishing(),
5129            "the last run out is what actually lands the stop"
5130        );
5131    }
5132
5133    #[tokio::test]
5134    async fn a_loop_already_asked_to_stop_returns_without_waiting_out_a_poll() {
5135        let dir = tempfile::tempdir().unwrap();
5136        let (opts, queue, status_file, home, worktrees) = idle_loop(dir.path());
5137        let stop = Stop::new();
5138        stop.stop();
5139
5140        let began = std::time::Instant::now();
5141        tokio::time::timeout(
5142            Duration::from_secs(2),
5143            drive(&opts, &queue, &status_file, &home, &worktrees, &stop),
5144        )
5145        .await
5146        .expect("a stopped loop must return, not sit out its poll interval")
5147        .expect("the loop's own setup and teardown must not fail");
5148        assert!(
5149            began.elapsed() < opts.poll,
5150            "returned only after {:?}, which is a poll interval, not a stop",
5151            began.elapsed()
5152        );
5153    }
5154
5155    #[tokio::test]
5156    async fn a_stop_while_idle_wakes_the_wait_instead_of_sleeping_it_out() {
5157        let dir = tempfile::tempdir().unwrap();
5158        let (opts, queue, status_file, home, worktrees) = idle_loop(dir.path());
5159        let stop = Stop::new();
5160
5161        // Asked for after the loop is already parked on its empty queue, which
5162        // is the case an operator tapping stop on a phone actually hits.
5163        let asker = {
5164            let stop = stop.clone();
5165            tokio::spawn(async move {
5166                tokio::time::sleep(Duration::from_millis(20)).await;
5167                stop.stop();
5168            })
5169        };
5170
5171        let began = std::time::Instant::now();
5172        tokio::time::timeout(
5173            Duration::from_secs(2),
5174            drive(&opts, &queue, &status_file, &home, &worktrees, &stop),
5175        )
5176        .await
5177        .expect("a stop asked for while idle must wake the wait")
5178        .expect("the loop's own setup and teardown must not fail");
5179        asker.await.unwrap();
5180        assert!(
5181            began.elapsed() < opts.poll,
5182            "returned only after {:?}, so the stop waited on the sleep",
5183            began.elapsed()
5184        );
5185    }
5186
5187    #[tokio::test]
5188    async fn a_stopped_loop_leaves_no_status_file_claiming_it_is_running() {
5189        let dir = tempfile::tempdir().unwrap();
5190        let (opts, queue, status_file, home, worktrees) = idle_loop(dir.path());
5191        let stop = Stop::new();
5192        stop.stop();
5193
5194        tokio::time::timeout(
5195            Duration::from_secs(2),
5196            drive(&opts, &queue, &status_file, &home, &worktrees, &stop),
5197        )
5198        .await
5199        .expect("a stopped loop must return")
5200        .expect("the loop's own setup and teardown must not fail");
5201
5202        assert!(
5203            home.is_dir(),
5204            "the loop did publish a status file, so its removal is the teardown and not an absence"
5205        );
5206        assert!(
5207            !status_file.exists(),
5208            "a stopped loop clears its status file"
5209        );
5210        assert!(
5211            read_status(&home).is_none(),
5212            "a reader must see no daemon at all, not a heartbeat that merely stopped"
5213        );
5214    }
5215
5216    #[tokio::test]
5217    async fn once_runs_startup_housekeeping_before_an_empty_queue_exits() {
5218        let dir = tempfile::tempdir().unwrap();
5219        let (mut opts, queue, status_file, home, worktrees) = idle_loop(dir.path());
5220        opts.once = true;
5221
5222        let mut settled = RunState::new(
5223            dir.path().join("repo"),
5224            "main".to_owned(),
5225            "abc1234".to_owned(),
5226            "fixture".to_owned(),
5227            Config::default(),
5228        );
5229        settled.status = RunStatus::Ready;
5230        let run_dir = home.join("runs").join(&settled.id);
5231        std::fs::create_dir_all(&run_dir).unwrap();
5232        std::fs::write(
5233            run_dir.join("run.json"),
5234            serde_json::to_string_pretty(&settled).unwrap(),
5235        )
5236        .unwrap();
5237        let questions = Questions::at(home.join("questions"));
5238        let mut question = ask::Question::new(
5239            settled.id.clone(),
5240            "review".to_owned(),
5241            "reviewer-1".to_owned(),
5242            "Continue?".to_owned(),
5243            String::new(),
5244            Vec::new(),
5245        );
5246        questions.put(&mut question).unwrap();
5247
5248        drive(&opts, &queue, &status_file, &home, &worktrees, &Stop::new())
5249            .await
5250            .unwrap();
5251
5252        assert_eq!(
5253            questions.get(&question.id).unwrap().status,
5254            ask::QuestionStatus::Abandoned,
5255            "an empty --once drain still performs startup question cleanup"
5256        );
5257    }
5258
5259    #[test]
5260    fn cache_check_due_fires_immediately_then_waits_out_its_own_interval() {
5261        let t0 = "2026-09-15T00:00:00Z".parse::<Timestamp>().unwrap();
5262
5263        assert!(
5264            cache_check_due(None, t0, CACHE_CHECK_INTERVAL_SECS),
5265            "never checked before: due at once"
5266        );
5267
5268        let one_sec_later = t0 + jiff::SignedDuration::from_secs(1);
5269        assert!(
5270            !cache_check_due(Some(t0), one_sec_later, CACHE_CHECK_INTERVAL_SECS),
5271            "well inside the interval: not due yet"
5272        );
5273
5274        let at_the_edge = t0 + jiff::SignedDuration::from_secs(CACHE_CHECK_INTERVAL_SECS as i64);
5275        assert!(
5276            !cache_check_due(Some(t0), at_the_edge, CACHE_CHECK_INTERVAL_SECS),
5277            "exactly at the edge: not yet due, same convention as `clean::due`"
5278        );
5279
5280        let past_it = t0 + jiff::SignedDuration::from_secs(CACHE_CHECK_INTERVAL_SECS as i64 + 1);
5281        assert!(
5282            cache_check_due(Some(t0), past_it, CACHE_CHECK_INTERVAL_SECS),
5283            "past the interval: due again"
5284        );
5285    }
5286
5287    /// A `magi.toml` whose `[verify] gate` names `cache_dir` as its shared
5288    /// `CARGO_TARGET_DIR`, capped at `limit_bytes`, plus a repository path
5289    /// that is never created - the fixtures [`maybe_prune_cache_between_runs`]
5290    /// and the congestion test below both need, and must not drift apart.
5291    fn cache_check_opts(dir: &Path, cache_dir: &Path, limit_bytes: u64) -> Opts {
5292        let config = dir.join("magi.toml");
5293        // A literal (single-quoted) TOML string, not a basic one: the cache
5294        // path is a Windows path full of backslashes, and a basic string
5295        // would have TOML try to interpret `\U` (from `\Users\...`) as a
5296        // Unicode escape and fail to parse - the same trap `magi.toml`'s own
5297        // `{{ vars.cache }}` rendering documents.
5298        std::fs::write(
5299            &config,
5300            format!(
5301                "[disk]\nmin_free_bytes = 0\nauto_fold = false\ncache_limit_bytes = {limit_bytes}\n\n\
5302                 [verify]\ngate = ['CARGO_TARGET_DIR={} cargo make check']\n",
5303                cache_dir.display()
5304            ),
5305        )
5306        .unwrap();
5307        Opts {
5308            config: Some(config),
5309            repo: dir.join("repo"),
5310            ..Opts::default()
5311        }
5312    }
5313
5314    #[tokio::test]
5315    async fn maybe_prune_cache_between_runs_reprunes_only_once_its_own_interval_elapses() {
5316        let dir = tempfile::tempdir().unwrap();
5317        let home = dir.path().join("home");
5318        let cache_dir = dir.path().join("cache");
5319        std::fs::create_dir_all(&cache_dir).unwrap();
5320        std::fs::write(cache_dir.join("a"), vec![0u8; 10]).unwrap();
5321        let opts = cache_check_opts(dir.path(), &cache_dir, 1);
5322
5323        // Nobody has asked this daemon to stop, which is the ordinary case;
5324        // the skip that a stop buys is asserted by its own test below.
5325        let running = Stop::new();
5326        let mut last_checked = None;
5327        let t0 = "2026-09-15T00:00:00Z".parse::<Timestamp>().unwrap();
5328        maybe_prune_cache_between_runs(&opts.repo, &opts, &home, &running, &mut last_checked, t0)
5329            .await;
5330        assert_eq!(
5331            crate::disk::dir_size(&cache_dir),
5332            0,
5333            "over the cap on the first check ever: pruned at once, no idle queue required"
5334        );
5335        assert_eq!(last_checked, Some(t0));
5336
5337        // A fresh oversized file lands, but the next check is not due yet.
5338        std::fs::write(cache_dir.join("b"), vec![0u8; 10]).unwrap();
5339        let too_soon = t0 + jiff::SignedDuration::from_secs(1);
5340        maybe_prune_cache_between_runs(
5341            &opts.repo,
5342            &opts,
5343            &home,
5344            &running,
5345            &mut last_checked,
5346            too_soon,
5347        )
5348        .await;
5349        assert_eq!(
5350            crate::disk::dir_size(&cache_dir),
5351            10,
5352            "too soon since the last check: left alone rather than rescanned every call"
5353        );
5354        assert_eq!(
5355            last_checked,
5356            Some(t0),
5357            "an idle check does not reset the clock"
5358        );
5359
5360        // Once the interval elapses, the same oversized cache is caught again.
5361        let due_again = t0 + jiff::SignedDuration::from_secs(CACHE_CHECK_INTERVAL_SECS as i64 + 1);
5362        maybe_prune_cache_between_runs(
5363            &opts.repo,
5364            &opts,
5365            &home,
5366            &running,
5367            &mut last_checked,
5368            due_again,
5369        )
5370        .await;
5371        assert_eq!(
5372            crate::disk::dir_size(&cache_dir),
5373            0,
5374            "due again: pruned back under the cap"
5375        );
5376    }
5377
5378    /// A stop must not queue behind housekeeping. The prune below is a
5379    /// synchronous walk of the whole cache with no await point in it, so a
5380    /// loop that entered it could not get back to its own `stopped()` test
5381    /// until the walk finished - and because no run is in flight at this
5382    /// boundary, `Stop::finishing` would meanwhile tell the operator's screen
5383    /// the stop had already landed. The idle branch has always made this same
5384    /// check before reaching `janitor`; the between-runs path makes it too.
5385    #[tokio::test]
5386    async fn a_stop_already_asked_for_skips_the_between_runs_cache_walk() {
5387        let dir = tempfile::tempdir().unwrap();
5388        let home = dir.path().join("home");
5389        let cache_dir = dir.path().join("cache");
5390        std::fs::create_dir_all(&cache_dir).unwrap();
5391        std::fs::write(cache_dir.join("a"), vec![0u8; 10]).unwrap();
5392        let opts = cache_check_opts(dir.path(), &cache_dir, 1);
5393
5394        let stop = Stop::new();
5395        stop.stop();
5396        assert!(
5397            !stop.finishing(),
5398            "no run is in flight at a between-runs boundary, so nothing else \
5399             would tell the operator this stop had not taken effect yet"
5400        );
5401
5402        let mut last_checked = None;
5403        let t0 = "2026-09-15T00:00:00Z".parse::<Timestamp>().unwrap();
5404        maybe_prune_cache_between_runs(&opts.repo, &opts, &home, &stop, &mut last_checked, t0)
5405            .await;
5406        assert_eq!(
5407            crate::disk::dir_size(&cache_dir),
5408            10,
5409            "over its cap, and due for the first check ever, but a stop outranks \
5410             it: the cap is a standing policy the next start measures again"
5411        );
5412        assert_eq!(
5413            last_checked, None,
5414            "a check that never happened must not claim the interval"
5415        );
5416    }
5417
5418    /// The regression this whole change exists for: gate timeouts on runs
5419    /// 52da/2f7f/5991/0915 traced back to the shared cache sitting at 81.8
5420    /// GiB against a 10 GiB cap, because the operator's queue never had a
5421    /// quiet moment for `poll`'s fully-idle branch to reach the ordinary
5422    /// `janitor` pass.
5423    ///
5424    /// Reproduced here with a task whose repository is never created:
5425    /// `Runner::start` fails at `git::toplevel` in a few milliseconds,
5426    /// spawning no agent CLI, so the task keeps failing and re-queuing
5427    /// (`Task::fail` with attempts still under the budget leaves it
5428    /// `Failed`, which `TaskStatus::runnable` still offers) for as long as
5429    /// the loop keeps polling - exactly the "queue with no idle moment"
5430    /// this task describes, produced without a real competition.
5431    #[tokio::test]
5432    async fn cache_prune_reaches_a_queue_that_never_goes_idle() {
5433        let dir = tempfile::tempdir().unwrap();
5434        let cache_dir = dir.path().join("cache");
5435        std::fs::create_dir_all(&cache_dir).unwrap();
5436        std::fs::write(cache_dir.join("stale"), vec![0u8; 4096]).unwrap();
5437
5438        let mut opts = cache_check_opts(dir.path(), &cache_dir, 1);
5439        opts.poll = Duration::from_millis(20);
5440        opts.max_attempts = 1_000;
5441
5442        let queue = Queue::at(dir.path().join("queue"));
5443        let mut t = Task::new(
5444            "x".to_owned(),
5445            "x".to_owned(),
5446            opts.repo.clone(),
5447            Source::Human,
5448        );
5449        queue.put(&mut t).unwrap();
5450
5451        let home = dir.path().join("home");
5452        let worktrees = dir.path().join("wt");
5453        let status_file = home.join("daemon.json");
5454        let stop = Stop::new();
5455        let stopper = {
5456            let stop = stop.clone();
5457            tokio::spawn(async move {
5458                tokio::time::sleep(Duration::from_millis(400)).await;
5459                stop.stop();
5460            })
5461        };
5462
5463        tokio::time::timeout(
5464            Duration::from_secs(10),
5465            drive(&opts, &queue, &status_file, &home, &worktrees, &stop),
5466        )
5467        .await
5468        .expect("the loop must not hang on a queue that keeps producing failing work")
5469        .expect("the loop's own setup and teardown must not fail");
5470        stopper.await.unwrap();
5471
5472        let after = queue.get(&t.id).unwrap();
5473        assert!(
5474            after.attempts >= 2,
5475            "the harness must actually have retried more than once, or this is not \
5476             exercising a busy queue at all (got {} attempt(s))",
5477            after.attempts
5478        );
5479        assert!(
5480            after.status.runnable(),
5481            "still under its attempt budget: the queue never reached a natural idle \
5482             on its own, only the external stop ended the test"
5483        );
5484
5485        assert_eq!(
5486            crate::disk::dir_size(&cache_dir),
5487            0,
5488            "an oversized cache must not be left to grow unboundedly just because the \
5489             queue kept the loop busy the whole time"
5490        );
5491    }
5492
5493    #[test]
5494    fn task_question_reconciliation_keeps_references_and_retires_manual_releases() {
5495        let dir = tempfile::tempdir().unwrap();
5496        let queue = Queue::at(dir.path().join("queue"));
5497        let questions = Questions::at(dir.path().join("questions"));
5498        let mut task = task();
5499        queue.put(&mut task).unwrap();
5500
5501        let mut task_question = ask::Question::new(
5502            task.id.clone(),
5503            crate::conduct::NODE.to_owned(),
5504            "conduct".to_owned(),
5505            "Which backend?".to_owned(),
5506            String::new(),
5507            Vec::new(),
5508        );
5509        questions.put(&mut task_question).unwrap();
5510        task.block(vec![task_question.id.clone()], None);
5511        queue.put(&mut task).unwrap();
5512
5513        let mut run_question = ask::Question::new(
5514            "20260101-000000-run1".to_owned(),
5515            "review".to_owned(),
5516            "reviewer-1".to_owned(),
5517            "Run question".to_owned(),
5518            String::new(),
5519            Vec::new(),
5520        );
5521        questions.put(&mut run_question).unwrap();
5522
5523        // A question from another node whose `run` happens to equal this
5524        // task's id — the same field, filled in for an unrelated reason. Only
5525        // `crate::conduct::NODE` questions use `run` as a task id; this one
5526        // must never be touched by this reconciliation, even after release.
5527        let mut coincidental = ask::Question::new(
5528            task.id.clone(),
5529            "review".to_owned(),
5530            "reviewer-1".to_owned(),
5531            "Unrelated review question".to_owned(),
5532            String::new(),
5533            Vec::new(),
5534        );
5535        questions.put(&mut coincidental).unwrap();
5536
5537        reconcile_task_questions(&queue, &questions);
5538        assert!(questions.get(&task_question.id).unwrap().status.open());
5539        assert!(questions.get(&run_question.id).unwrap().status.open());
5540        assert!(questions.get(&coincidental.id).unwrap().status.open());
5541
5542        task.release();
5543        queue.put(&mut task).unwrap();
5544        reconcile_task_questions(&queue, &questions);
5545        assert_eq!(
5546            questions.get(&task_question.id).unwrap().status,
5547            ask::QuestionStatus::Abandoned
5548        );
5549        assert!(
5550            questions.get(&run_question.id).unwrap().status.open(),
5551            "run questions remain the run janitor's responsibility"
5552        );
5553        assert!(
5554            questions.get(&coincidental.id).unwrap().status.open(),
5555            "a non-conductor question must not be abandoned just because its \
5556             run id coincides with a task id"
5557        );
5558    }
5559
5560    #[test]
5561    fn a_freshly_started_running_task_is_never_stalled() {
5562        let dir = tempfile::tempdir().unwrap();
5563        let mut t = task();
5564        t.start("run-1".to_owned());
5565        // `updated_at` is `Timestamp::now()`, left alone: no live daemon
5566        // named in `dir`, but nowhere near `STALLED_RUNNING` yet.
5567        assert!(!is_stalled(&t, dir.path(), Timestamp::now()));
5568    }
5569
5570    #[test]
5571    fn a_long_running_task_with_no_live_daemon_is_stalled() {
5572        let dir = tempfile::tempdir().unwrap();
5573        let mut t = task();
5574        t.start("run-1".to_owned());
5575        t.updated_at = Timestamp::now()
5576            - jiff::SignedDuration::from_secs(STALLED_RUNNING.as_secs() as i64 + 60);
5577        assert!(is_stalled(&t, dir.path(), Timestamp::now()));
5578        assert_eq!(
5579            stalled_tasks(
5580                &Queue::at(dir.path().join("q")),
5581                dir.path(),
5582                Timestamp::now()
5583            )
5584            .len(),
5585            0,
5586            "the task was never written to this queue"
5587        );
5588    }
5589
5590    #[test]
5591    fn a_long_running_task_a_live_daemon_still_names_is_not_stalled() {
5592        let dir = tempfile::tempdir().unwrap();
5593        let mut t = task();
5594        t.id = "20260903-080340-0167".to_owned();
5595        t.start("20260903-080619-01c2".to_owned());
5596        t.updated_at = Timestamp::now()
5597            - jiff::SignedDuration::from_secs(STALLED_RUNNING.as_secs() as i64 + 60);
5598
5599        let mut status = Status::new();
5600        status.current = vec![Current {
5601            task: t.id.clone(),
5602            run: "20260903-080619-01c2".to_owned(),
5603        }];
5604        write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
5605
5606        assert!(
5607            !is_stalled(&t, dir.path(), Timestamp::now()),
5608            "a live daemon's own heartbeat rules out stalled, however long the task has run"
5609        );
5610    }
5611
5612    /// Rewrite a task's `updated_at` on disk directly, bypassing
5613    /// `Queue::put`'s own `Timestamp::now()` stamping - the only way to make
5614    /// a fixture look like it has genuinely been `running` for a while.
5615    fn backdate_task(queue: &Queue, id: &str, seconds_ago: i64) {
5616        let path = queue.path_of(id);
5617        let body = std::fs::read_to_string(&path).unwrap();
5618        let mut v: serde_json::Value = serde_json::from_str(&body).unwrap();
5619        let old = Timestamp::now() - jiff::SignedDuration::from_secs(seconds_ago);
5620        v["updated_at"] = serde_json::Value::String(old.to_string());
5621        std::fs::write(&path, serde_json::to_string_pretty(&v).unwrap()).unwrap();
5622    }
5623
5624    #[test]
5625    fn stalled_tasks_still_reaches_a_task_reclaim_could_not_claim_yet() {
5626        // The realistic `poll()` ordering, not `is_stalled` in isolation:
5627        // `reclaim_orphaned_running` runs first, on every poll, and settles
5628        // any `running` task whose claim it can actually take. For most
5629        // crashes that is immediate - a dead pid is proof enough for
5630        // `sweep_stale_claims` to drop the lock the same tick, and the very
5631        // next claim attempt succeeds. But a lock whose pid cannot be parsed
5632        // at all falls back to `STALE_CLAIM`'s six-hour age instead (see
5633        // `sweep_stale_claims`'s own doc), so the lock - and the claim
5634        // failure behind it - can legitimately outlive many polls. This is
5635        // exactly the gap `stalled_tasks` exists to surface well before that
5636        // six-hour sweep would: reclaim leaves the task `running`, and it
5637        // must still reach the conductor as stalled.
5638        let dir = tempfile::tempdir().unwrap();
5639        let queue = Queue::at(dir.path().join("queue"));
5640        let home = dir.path().join("home");
5641
5642        let mut t = task();
5643        t.id = "20260101-000001-lock".to_owned();
5644        t.start("run-1".to_owned());
5645        queue.put(&mut t).unwrap();
5646        backdate_task(&queue, &t.id, STALLED_RUNNING.as_secs() as i64 + 60);
5647        std::fs::write(
5648            dir.path().join("queue").join(format!("{}.lock", t.id)),
5649            "not a pid",
5650        )
5651        .unwrap();
5652
5653        let now = Timestamp::now();
5654        assert!(
5655            reclaim_orphaned_running(&queue, 2).is_empty(),
5656            "the unparseable lock is still well within STALE_CLAIM, so the claim fails \
5657             and reclaim must leave the task alone"
5658        );
5659        assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Running);
5660
5661        let stalled = stalled_tasks(&queue, &home, now);
5662        assert_eq!(
5663            stalled.len(),
5664            1,
5665            "reclaim's inability to claim it yet must not hide it from the conductor"
5666        );
5667        assert_eq!(stalled[0].id, t.id);
5668    }
5669
5670    #[test]
5671    fn ordinary_dead_daemon_task_is_shown_stalled_before_reclaim_and_can_be_requeued() {
5672        let dir = tempfile::tempdir().unwrap();
5673        crate::run::set_home(dir.path().join("run-home"));
5674        let queue = Queue::at(dir.path().join("queue"));
5675        let home = dir.path().join("home");
5676        let questions = Questions::at(dir.path().join("questions"));
5677
5678        let mut t = task();
5679        t.id = "20260101-000003-dead".to_owned();
5680        t.start("missing-run".to_owned());
5681        queue.put(&mut t).unwrap();
5682        backdate_task(&queue, &t.id, STALLED_RUNNING.as_secs() as i64 + 60);
5683
5684        // This is the real poll ordering: retain the deterministic stalled
5685        // input before a claim proves the owner is gone and reclaims it.
5686        let stalled = stalled_tasks(&queue, &home, Timestamp::now());
5687        assert_eq!(
5688            stalled.iter().map(|task| &task.id).collect::<Vec<_>>(),
5689            [&t.id]
5690        );
5691        assert_eq!(reclaim_orphaned_running(&queue, 2), [t.id.clone()]);
5692        assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Held);
5693
5694        // Reclaim drops its guard before conductor decisions are applied, so
5695        // the decision for the captured stalled input has a real write path.
5696        crate::conduct::apply(
5697            &queue,
5698            &questions,
5699            &crate::conduct::Verdict {
5700                decisions: vec![crate::conduct::Decision {
5701                    id: t.id.clone(),
5702                    recovery: Some(crate::conduct::Recovery::Requeue),
5703                    ..crate::conduct::Decision::default()
5704                }],
5705            },
5706        )
5707        .unwrap();
5708        assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Queued);
5709    }
5710
5711    #[test]
5712    fn stalled_tasks_reports_exactly_the_tasks_is_stalled_agrees_on() {
5713        let dir = tempfile::tempdir().unwrap();
5714        let queue = Queue::at(dir.path().join("queue"));
5715        let home = dir.path().join("home");
5716
5717        let mut fresh = task();
5718        fresh.id = "20260101-000001-aaaa".to_owned();
5719        fresh.start("run-1".to_owned());
5720        queue.put(&mut fresh).unwrap();
5721
5722        let mut old = task();
5723        old.id = "20260101-000002-bbbb".to_owned();
5724        old.start("run-2".to_owned());
5725        queue.put(&mut old).unwrap();
5726        backdate_task(&queue, &old.id, STALLED_RUNNING.as_secs() as i64 + 60);
5727
5728        let stalled = stalled_tasks(&queue, &home, Timestamp::now());
5729        assert_eq!(stalled.len(), 1);
5730        assert_eq!(stalled[0].id, old.id);
5731    }
5732
5733    #[test]
5734    fn queued_and_finished_task_views_partition_by_status() {
5735        let dir = tempfile::tempdir().unwrap();
5736        let queue = Queue::at(dir.path().join("queue"));
5737
5738        let mut queued = task();
5739        queued.id = "20260101-000001-aaaa".to_owned();
5740        queue.put(&mut queued).unwrap();
5741
5742        let mut failed = task();
5743        failed.id = "20260101-000002-bbbb".to_owned();
5744        failed.start("run-1".to_owned());
5745        failed.fail("gate red", 5);
5746        queue.put(&mut failed).unwrap();
5747
5748        let mut held = task();
5749        held.id = "20260101-000003-cccc".to_owned();
5750        held.hold_machine(None);
5751        queue.put(&mut held).unwrap();
5752
5753        let mut running = task();
5754        running.id = "20260101-000004-dddd".to_owned();
5755        running.start("run-2".to_owned());
5756        queue.put(&mut running).unwrap();
5757
5758        let queued_ids: Vec<String> = queued_tasks(&queue).into_iter().map(|t| t.id).collect();
5759        assert_eq!(queued_ids, [queued.id.clone()]);
5760
5761        let mut finished_ids: Vec<String> =
5762            finished_tasks(&queue).into_iter().map(|t| t.id).collect();
5763        finished_ids.sort_unstable();
5764        let mut want = vec![failed.id.clone(), held.id.clone()];
5765        want.sort_unstable();
5766        assert_eq!(finished_ids, want);
5767    }
5768
5769    #[test]
5770    fn resolve_blockers_clears_a_done_dependency_and_keeps_an_unresolved_one() {
5771        let dir = tempfile::tempdir().unwrap();
5772        let queue = Queue::at(dir.path().join("queue"));
5773        let questions = ask::Questions::at(dir.path().join("questions"));
5774
5775        let mut dep = task();
5776        dep.id = "20260101-000001-dep0".to_owned();
5777        dep.succeed();
5778        queue.put(&mut dep).unwrap();
5779
5780        let mut still_going = task();
5781        still_going.id = "20260101-000002-dep1".to_owned();
5782        queue.put(&mut still_going).unwrap();
5783
5784        let mut blocked = task();
5785        blocked.id = "20260101-000003-main".to_owned();
5786        blocked.block(
5787            vec![dep.id.clone(), still_going.id.clone()],
5788            Some("waits on both".to_owned()),
5789        );
5790        queue.put(&mut blocked).unwrap();
5791
5792        resolve_blockers(&queue, &questions);
5793
5794        let after = queue.get(&blocked.id).unwrap();
5795        assert_eq!(
5796            after.status,
5797            TaskStatus::Blocked,
5798            "one dependency is still outstanding"
5799        );
5800        assert_eq!(after.blocked_by, [still_going.id.clone()]);
5801    }
5802
5803    #[test]
5804    fn resolve_blockers_carries_an_answers_content_onto_the_task_and_unblocks_it() {
5805        let dir = tempfile::tempdir().unwrap();
5806        let queue = Queue::at(dir.path().join("queue"));
5807        let questions = ask::Questions::at(dir.path().join("questions"));
5808
5809        let mut q = crate::ask::Question::new(
5810            "20260101-000001-main".to_owned(),
5811            crate::conduct::NODE.to_owned(),
5812            "conduct".to_owned(),
5813            "Which backend?".to_owned(),
5814            String::new(),
5815            Vec::new(),
5816        );
5817        questions.put(&mut q).unwrap();
5818        q.answer(crate::ask::Answer::Text("SQLite".to_owned()))
5819            .unwrap();
5820        questions.put(&mut q).unwrap();
5821
5822        let mut blocked = task();
5823        blocked.id = "20260101-000001-main".to_owned();
5824        blocked.block(vec![q.id.clone()], Some("which backend?".to_owned()));
5825        queue.put(&mut blocked).unwrap();
5826
5827        resolve_blockers(&queue, &questions);
5828
5829        let after = queue.get(&blocked.id).unwrap();
5830        assert_eq!(
5831            after.status,
5832            TaskStatus::Queued,
5833            "the only blocker resolved"
5834        );
5835        assert_eq!(after.answers.len(), 1);
5836        assert_eq!(after.answers[0].question, "Which backend?");
5837        assert_eq!(after.answers[0].answer, "SQLite");
5838
5839        // And the run this task starts next is told about it.
5840        let instruction = instruction_for(&after);
5841        assert!(instruction.contains("Which backend?"));
5842        assert!(instruction.contains("SQLite"));
5843    }
5844
5845    #[test]
5846    fn resolve_blockers_restores_a_held_task_to_held_instead_of_queuing_it() {
5847        // Reproduces the reported bug (task 3958): a task held out of
5848        // attempts, blocked on a `crate::conduct` follow-up question, must
5849        // come back `held` once that question is answered - never `queued`,
5850        // whatever the answer said - or it silently re-enters the
5851        // competition queue with its attempts already exhausted.
5852        let dir = tempfile::tempdir().unwrap();
5853        let queue = Queue::at(dir.path().join("queue"));
5854        let questions = ask::Questions::at(dir.path().join("questions"));
5855
5856        let mut q = crate::ask::Question::new(
5857            "20260101-000001-main".to_owned(),
5858            crate::conduct::NODE.to_owned(),
5859            "conduct".to_owned(),
5860            "How should this be handled?".to_owned(),
5861            String::new(),
5862            Vec::new(),
5863        );
5864        questions.put(&mut q).unwrap();
5865        q.answer(crate::ask::Answer::Text(
5866            "leave it held, a human will look at it later".to_owned(),
5867        ))
5868        .unwrap();
5869        questions.put(&mut q).unwrap();
5870
5871        let mut held = task();
5872        held.id = "20260101-000001-main".to_owned();
5873        held.hold_machine(Some("out of attempts".to_owned()));
5874        held.block(vec![q.id.clone()], Some("what now?".to_owned()));
5875        queue.put(&mut held).unwrap();
5876
5877        resolve_blockers(&queue, &questions);
5878
5879        let after = queue.get(&held.id).unwrap();
5880        assert_eq!(after.status, TaskStatus::Held);
5881        assert_eq!(after.hold_reason.as_deref(), Some("out of attempts"));
5882        assert_eq!(
5883            after.answers[0].answer,
5884            "leave it held, a human will look at it later"
5885        );
5886    }
5887
5888    #[test]
5889    fn resolve_blockers_holds_a_task_whose_dependency_was_deleted() {
5890        // Reproduces the reported bug: a task blocked on a task id that was
5891        // removed (`magi task rm`, or deleted by hand) can never see that id
5892        // reach `Done`, so the ordinary per-id loop has nothing to notice and
5893        // would otherwise leave the task `blocked` forever with no way for an
5894        // operator to find out why.
5895        let dir = tempfile::tempdir().unwrap();
5896        let queue = Queue::at(dir.path().join("queue"));
5897        let questions = ask::Questions::at(dir.path().join("questions"));
5898
5899        let mut still_going = task();
5900        still_going.id = "20260101-000002-dep1".to_owned();
5901        queue.put(&mut still_going).unwrap();
5902
5903        let mut blocked = task();
5904        blocked.id = "20260101-000003-main".to_owned();
5905        blocked.block(
5906            vec!["20260101-000001-gone".to_owned(), still_going.id.clone()],
5907            Some("waits on both".to_owned()),
5908        );
5909        queue.put(&mut blocked).unwrap();
5910
5911        resolve_blockers(&queue, &questions);
5912
5913        let after = queue.get(&blocked.id).unwrap();
5914        assert_eq!(
5915            after.status,
5916            TaskStatus::Held,
5917            "a missing dependency must not leave the task blocked forever"
5918        );
5919        assert_eq!(after.hold_source, Some(crate::queue::HoldSource::Machine));
5920        assert!(after.blocked_by.is_empty());
5921        let reason = after.hold_reason.as_deref().unwrap_or_default();
5922        assert!(
5923            reason.contains("20260101-000001-gone"),
5924            "the missing id must be named so an operator can tell what happened: {reason}"
5925        );
5926        assert!(
5927            reason.contains(&still_going.id),
5928            "the still-valid dependency must not silently vanish from the record: {reason}"
5929        );
5930    }
5931
5932    #[test]
5933    fn instruction_for_is_unchanged_without_any_answers() {
5934        let t = task();
5935        assert_eq!(instruction_for(&t), t.instruction);
5936    }
5937
5938    #[test]
5939    fn resumed_instruction_is_unchanged_without_any_answers() {
5940        let t = task();
5941        assert_eq!(resumed_instruction(&t.instruction, &t), t.instruction);
5942    }
5943
5944    #[test]
5945    fn resumed_instruction_carries_a_new_answer_onto_the_old_run() {
5946        let mut t = task();
5947        t.record_answer("Which backend?".to_owned(), "SQLite".to_owned());
5948        // The run's own instruction on disk predates the answer: it is the
5949        // plain original text `Runner::start` saved before the operator was
5950        // ever asked anything.
5951        let old = t.instruction.clone();
5952
5953        let refreshed = resumed_instruction(&old, &t);
5954        assert!(refreshed.starts_with(&old), "the original text is kept");
5955        assert!(refreshed.contains("Which backend?"));
5956        assert!(refreshed.contains("SQLite"));
5957    }
5958
5959    #[test]
5960    fn resumed_instruction_keeps_an_original_answers_heading() {
5961        let mut t = task();
5962        t.instruction = "Context\n\n# Operator answers\n\nThis is part of the task.".to_owned();
5963        t.record_answer("Which backend?".to_owned(), "SQLite".to_owned());
5964
5965        let refreshed = resumed_instruction(&t.instruction, &t);
5966
5967        assert!(
5968            refreshed.starts_with(&t.instruction),
5969            "an answers heading in the original instruction is not the appended block"
5970        );
5971        assert_eq!(refreshed.matches(ANSWERS_HEADER).count(), 2);
5972        assert!(refreshed.contains("Which backend?"));
5973        assert!(refreshed.contains("SQLite"));
5974
5975        let repeated = resumed_instruction(&refreshed, &t);
5976        assert_eq!(
5977            repeated, refreshed,
5978            "only the final appended block is refreshed"
5979        );
5980    }
5981
5982    #[test]
5983    fn resumed_instruction_does_not_duplicate_across_repeated_resumes() {
5984        let mut t = task();
5985        t.record_answer("Which backend?".to_owned(), "SQLite".to_owned());
5986
5987        // A first resume appends the block; a second resume of the same run,
5988        // with no new answer in between, must reproduce exactly the same
5989        // text rather than appending the block a second time.
5990        let once = resumed_instruction(&t.instruction, &t);
5991        let twice = resumed_instruction(&once, &t);
5992        assert_eq!(once, twice);
5993        assert_eq!(once.matches("Which backend?").count(), 1);
5994
5995        // A later answer replaces the block wholesale rather than growing it.
5996        t.record_answer("Which cache?".to_owned(), "Redis".to_owned());
5997        let refreshed = resumed_instruction(&once, &t);
5998        assert_eq!(refreshed.matches(ANSWERS_HEADER).count(), 1);
5999        assert!(refreshed.contains("Which backend?"));
6000        assert!(refreshed.contains("Which cache?"));
6001    }
6002
6003    #[test]
6004    fn prepare_instruction_covers_all_three_starters() {
6005        let mut t = task();
6006        t.record_answer("Which backend?".to_owned(), "SQLite".to_owned());
6007
6008        // Start: a fresh run gets the task text plus every answer so far —
6009        // exactly `instruction_for`.
6010        assert_eq!(
6011            prepare_instruction(&Starter::Start, None, &t),
6012            Some(instruction_for(&t))
6013        );
6014
6015        // Resume: the run's prior instruction is refreshed with the answer,
6016        // not discarded and not left stale.
6017        let old = t.instruction.clone();
6018        assert_eq!(
6019            prepare_instruction(&Starter::Resume("some-run".to_owned()), Some(&old), &t),
6020            Some(resumed_instruction(&old, &t))
6021        );
6022
6023        // Review: a review-only pass builds its own instruction from the
6024        // branch's history in `crate::graph`, with no task statement at all -
6025        // this boundary must leave it alone.
6026        assert_eq!(
6027            prepare_instruction(&Starter::Review("magi/eba2/A".to_owned()), Some(&old), &t),
6028            None
6029        );
6030    }
6031
6032    #[test]
6033    fn choose_starter_prefers_review_over_resume_when_the_branch_survived() {
6034        assert_eq!(
6035            choose_starter(Some("magi/eba2/A"), true, Some("some-run")),
6036            Starter::Review("magi/eba2/A".to_owned())
6037        );
6038    }
6039
6040    #[test]
6041    fn choose_starter_falls_back_to_start_when_the_review_branch_is_gone() {
6042        assert_eq!(
6043            choose_starter(Some("magi/eba2/A"), false, Some("some-run")),
6044            Starter::Start,
6045            "a vanished review branch must not fall back to resuming the old run either"
6046        );
6047    }
6048
6049    #[test]
6050    fn choose_starter_resumes_or_starts_when_there_is_no_review_choice_at_all() {
6051        assert_eq!(
6052            choose_starter(None, false, Some("some-run")),
6053            Starter::Resume("some-run".to_owned())
6054        );
6055        assert_eq!(choose_starter(None, false, None), Starter::Start);
6056    }
6057
6058    #[test]
6059    fn an_explicit_release_forces_a_fresh_competition_even_with_a_resumable_run() {
6060        let mut released = task();
6061        released.start("stalled-run".to_owned());
6062        released.requeue();
6063        let unfinished = (!released.fresh_start)
6064            .then(|| Some("stalled-run".to_owned()))
6065            .flatten();
6066        assert_eq!(
6067            choose_starter(None, false, unfinished.as_deref()),
6068            Starter::Start,
6069            "release keeps run history but must not resume it"
6070        );
6071        assert_eq!(released.runs, ["stalled-run"]);
6072    }
6073
6074    #[test]
6075    fn an_ordinary_release_keeps_a_resumable_run_available() {
6076        let mut released = task();
6077        released.start("stalled-run".to_owned());
6078        released.release();
6079        let unfinished = (!released.fresh_start)
6080            .then(|| Some("stalled-run".to_owned()))
6081            .flatten();
6082        assert_eq!(
6083            choose_starter(None, false, unfinished.as_deref()),
6084            Starter::Resume("stalled-run".to_owned()),
6085            "manual release must preserve the normal resume path"
6086        );
6087    }
6088
6089    #[test]
6090    fn a_blocked_run_that_spent_every_review_round_has_exhausted_its_budget() {
6091        let mut state = run_state(RunStatus::Blocked);
6092        state.config.graph.review_rounds = 3;
6093        state.reviews = vec![review_round(1), review_round(2), review_round(3)];
6094        assert!(exhausted_review_budget(&state));
6095
6096        // One round still unused: resuming can still ask a reviewer something.
6097        state.reviews.pop();
6098        assert!(!exhausted_review_budget(&state));
6099
6100        // Exhausted rounds on a non-`Blocked` status (a stall, say) do not
6101        // count: only a `Blocked` run re-enters the review loop on resume.
6102        let mut stalled = run_state(RunStatus::Stalled);
6103        stalled.config.graph.review_rounds = 1;
6104        stalled.reviews = vec![review_round(1)];
6105        assert!(!exhausted_review_budget(&stalled));
6106    }
6107
6108    fn review_round(round: usize) -> crate::run::ReviewRound {
6109        crate::run::ReviewRound {
6110            round,
6111            head: "deadbeef".to_owned(),
6112            verified_head: None,
6113            verified_at: None,
6114            reviews: Vec::new(),
6115            e2e: Vec::new(),
6116            verify_retried: false,
6117            e2e_deferred: false,
6118            e2e_defer_reason: None,
6119            fix: None,
6120            blocking: 0,
6121            answered: 1,
6122            expected: 1,
6123            clean: false,
6124            progressed: true,
6125            vote_split: false,
6126            reconsideration: Vec::new(),
6127            verdict: None,
6128        }
6129    }
6130
6131    #[test]
6132    fn unfinished_run_skips_a_round_exhausted_blocked_run_so_requeue_means_a_fresh_competition() {
6133        // Mirrors the failure this exists to close: a task's last run ended
6134        // `Blocked` with the review budget spent, `crate::conduct` chose
6135        // `Recovery::Requeue` (`Task::release`, which keeps `runs` as
6136        // evidence), and without this check `attempt` would go on treating
6137        // that exhausted run as "unfinished" and resume it - `graph::Runner`'s
6138        // review loop iterates zero times over an already-spent budget, so
6139        // the resumed run settles right back to `Blocked` having asked nobody
6140        // anything, and `Requeue`'s promised fresh competition never happens.
6141        let mut exhausted = RunState::new(
6142            PathBuf::from("/repo"),
6143            "main".to_owned(),
6144            "abc1234def".to_owned(),
6145            "add retries".to_owned(),
6146            Config::default(),
6147        );
6148        exhausted.status = RunStatus::Blocked;
6149        exhausted.config.graph.review_rounds = 1;
6150        exhausted.reviews = vec![review_round(1)];
6151
6152        assert_eq!(
6153            unfinished_run_with(&[exhausted.id.clone()], "t", |_| Ok(exhausted.clone())),
6154            None,
6155            "an exhausted `Blocked` run must not be offered as resumable"
6156        );
6157
6158        // A `Blocked` run with rounds still unused is genuinely worth
6159        // resuming, and must still be found.
6160        let mut has_budget_left = RunState::new(
6161            PathBuf::from("/repo"),
6162            "main".to_owned(),
6163            "abc1234def".to_owned(),
6164            "add retries".to_owned(),
6165            Config::default(),
6166        );
6167        has_budget_left.status = RunStatus::Blocked;
6168        has_budget_left.config.graph.review_rounds = 3;
6169        has_budget_left.reviews = vec![review_round(1)];
6170
6171        assert_eq!(
6172            unfinished_run_with(&[has_budget_left.id.clone()], "t", |_| {
6173                Ok(has_budget_left.clone())
6174            }),
6175            Some(has_budget_left.id.clone())
6176        );
6177    }
6178
6179    #[test]
6180    fn unfinished_run_never_falls_back_to_an_older_resumable_run() {
6181        // A task whose history holds an *older* run that still looks
6182        // resumable (say, a competition `Runner::review` was started
6183        // alongside after that older run went `Stalled`) and a *newest* run
6184        // that is `Blocked` with its review budget spent. `Recovery::Requeue`
6185        // on this task must mean a fresh competition — falling back to the
6186        // stale, superseded `Stalled` run instead would resurrect history
6187        // nothing asked to revisit and silently defeat the requeue.
6188        let mut older_stalled = RunState::new(
6189            PathBuf::from("/repo"),
6190            "main".to_owned(),
6191            "abc1234def".to_owned(),
6192            "add retries".to_owned(),
6193            Config::default(),
6194        );
6195        older_stalled.status = RunStatus::Stalled;
6196
6197        let mut newest_exhausted = RunState::new(
6198            PathBuf::from("/repo"),
6199            "main".to_owned(),
6200            "abc1234def".to_owned(),
6201            "add retries".to_owned(),
6202            Config::default(),
6203        );
6204        newest_exhausted.status = RunStatus::Blocked;
6205        newest_exhausted.config.graph.review_rounds = 1;
6206        newest_exhausted.reviews = vec![review_round(1)];
6207
6208        assert_eq!(
6209            unfinished_run_with(
6210                &[older_stalled.id.clone(), newest_exhausted.id.clone()],
6211                "t",
6212                |_| Ok(newest_exhausted.clone())
6213            ),
6214            None,
6215            "the newest run is exhausted, so nothing here is worth resuming - \
6216             least of all the older, already-superseded run"
6217        );
6218    }
6219
6220    #[test]
6221    fn unfinished_run_warns_and_skips_a_run_it_cannot_read() {
6222        assert_eq!(
6223            unfinished_run_with(&["20260101-000000-gone".to_owned()], "t", |_| {
6224                Err(anyhow::anyhow!("fixture is absent"))
6225            }),
6226            None
6227        );
6228    }
6229}