Skip to main content

magi/
waiter.rs

1//! The waiter: something that keeps waiting on a question after the process
2//! that asked it is gone.
3//!
4//! `magi ask` blocks in the agent's own shell tool, and that tool kills a wait
5//! that runs long (see `ask::WAIT_SLICE`). An agent can also background the
6//! call and simply finish. Either way the question outlives the only process
7//! that would have read the owner's answer, and the owner's reply lands in a
8//! file nobody is polling - the phone then says "waiting for the agent" while
9//! there is no agent to wait for.
10//!
11//! This module is the guarantee that something is always on the other end. It
12//! runs inside `magi serve`, on its own cadence rather than the queue's, and
13//! for every question a `magi ask` filed it decides one of three things:
14//!
15//! - **the asker is still there** (its [`Lease`] is fresh): do nothing. A
16//!   second agent must never be started while the first is blocked;
17//! - **the asker is gone and the owner has said or answered something the agent
18//!   has not read**: resume the *same seat's* CLI session with the question,
19//!   the thread so far and the owner's words, so the agent that holds the
20//!   context carries on;
21//! - **nobody answered before `answer_timeout`**: abandon it, in the same words
22//!   the asker would have used.
23//!
24//! # Shape
25//!
26//! Same split as [`crate::queue`] and [`crate::ask`]: [`decide`] is pure - a
27//! question, a lease, a clock and a bool in, an [`Action`] out - and every
28//! filesystem or process effect is in [`Waiter::tick`]. State lives on disk (the
29//! question record and its lease), so a restarted daemon picks up exactly where
30//! the last one stopped, including an answer nobody has read yet.
31//!
32//! # What it will not do
33//!
34//! It never starts a fresh consultant. If the seat's session cannot be resumed
35//! (`agent::has_session` says no, the working directory is gone, the run is
36//! over, the agent left the roster) the question stays as it is, the owner
37//! gets a notification saying why, and nothing else runs. A new agent would
38//! have to be caught up on everything the first one knew, and would act on the
39//! owner's words without the context they were written against.
40//!
41//! It never assumes a resume worked either: a turn that fails or times out
42//! leaves the word undelivered, and a later tick tries again.
43
44use std::collections::HashMap;
45use std::path::PathBuf;
46use std::time::{Duration, Instant};
47
48use anyhow::Result;
49use jiff::Timestamp;
50
51use crate::agent::{self, Invocation, SeatState};
52use crate::ask::{Lease, Question, QuestionStatus, Questions, Waiter as Note, WaiterKind, Who};
53use crate::config::{AgentSpec, Config};
54use crate::notices::{self, Notice};
55use crate::prompt;
56use crate::run::RunState;
57
58/// How often the waiter looks at the store.
59pub const TICK: Duration = Duration::from_secs(5);
60
61/// Wall-clock limit for one resumed turn. A resumed agent may go on to do real
62/// work once it has the owner's word, but it is one conversation turn, not a
63/// whole node.
64const DELIVERY_TIMEOUT: Duration = Duration::from_secs(15 * 60);
65
66/// How long a failed or refused delivery is left alone before the same input
67/// is tried again. Long enough not to hammer a seat that is out of quota,
68/// short enough that a fixed problem does not wait a day.
69const RETRY_AFTER: Duration = Duration::from_secs(10 * 60);
70
71/// What the agent has not read yet.
72#[derive(Debug, Clone, PartialEq, Eq)]
73pub enum Word {
74    /// The owner spoke back without deciding.
75    Said(String),
76    /// The owner decided.
77    Answered(String),
78}
79
80/// What to do about one question, this tick.
81#[derive(Debug, Clone, PartialEq, Eq)]
82pub enum Action {
83    /// Nothing: not ours, someone is waiting, or nothing new to hand over.
84    Idle,
85    /// The deadline passed with no answer and nobody waiting.
86    Expire,
87    /// Resume the asking seat with this.
88    Deliver(Word),
89}
90
91/// The whole decision, with no I/O.
92///
93/// `seat_busy` is whether the asking seat's CLI is still mid-turn - the agent
94/// that exited its `magi ask` may not have exited itself, and resuming a
95/// conversation that is running would fork it.
96///
97/// Order matters: a word the owner gave before the deadline is delivered even
98/// if the deadline has passed by the time the waiter looks, and only then does
99/// an unanswered question expire.
100pub fn decide(
101    q: &Question,
102    lease: Option<&Lease>,
103    seat_busy: bool,
104    default_timeout: u64,
105    now: Timestamp,
106) -> Action {
107    decide_owned(q, lease, seat_busy, default_timeout, now, false)
108}
109
110/// [`decide`], told whether the daemon owns the answer's action.
111///
112/// `daemon_owns_action` is true when the daemon will apply the action (the
113/// task exists and is in a state `daemon::decide_action` acts on). An answer
114/// that carries a [`crate::ask::ChoiceAction`] is then the daemon's alone
115/// (`daemon::apply_choice_actions`, idempotent through `Task::actions_applied`
116/// under the task's claim): resuming the dead seat here as well would run the
117/// same seat twice when the action is a resume or a requeue. With no task
118/// there is no one to apply it, so the words are delivered as before.
119pub fn decide_owned(
120    q: &Question,
121    lease: Option<&Lease>,
122    seat_busy: bool,
123    default_timeout: u64,
124    now: Timestamp,
125    daemon_owns_action: bool,
126) -> Action {
127    if q.cwd.is_none() || lease.is_some_and(|l| l.fresh(now)) {
128        return Action::Idle;
129    }
130    match q.status {
131        QuestionStatus::Abandoned => Action::Idle,
132        QuestionStatus::Answered => match q.resolution() {
133            _ if q.chosen_action().is_some_and(|a| a.maintenance())
134                || (daemon_owns_action && q.chosen_action().is_some()) =>
135            {
136                Action::Idle
137            }
138            Some(a) if !q.answer_delivered && !seat_busy => Action::Deliver(Word::Answered(a)),
139            _ => Action::Idle,
140        },
141        QuestionStatus::Open => {
142            if let Some(said) = q.unread_from_owner() {
143                if seat_busy {
144                    return Action::Idle;
145                }
146                return Action::Deliver(Word::Said(said));
147            }
148            let secs = if q.answer_timeout > 0 {
149                q.answer_timeout
150            } else {
151                default_timeout
152            };
153            let deadline = q.last_activity().saturating_add(secs as i64);
154            if now.as_second() > deadline {
155                Action::Expire
156            } else {
157                Action::Idle
158            }
159        }
160    }
161}
162
163/// Everything needed to resume one seat.
164struct Target {
165    coordination: Option<crate::config::Coordination>,
166    fetched_base: Option<crate::run::FetchedBase>,
167    spec: AgentSpec,
168    seat: SeatState,
169    cwd: PathBuf,
170    sessions: bool,
171    allow_write: bool,
172    language: String,
173}
174
175/// The waiter's state: the store it watches and what it must remember between
176/// ticks (only which inputs it has already failed on).
177pub struct Waiter {
178    store: Questions,
179    home: PathBuf,
180    /// The config the conductor runs under, for a question the conductor asked
181    /// (it has no run whose snapshot could say).
182    conduct_cfg: Option<Config>,
183    /// `answer_timeout` for a question that did not record its own.
184    default_timeout: u64,
185    /// Inputs already tried and failed, by question id: `(key, until)`.
186    memo: HashMap<String, (String, Instant)>,
187}
188
189impl Waiter {
190    /// A waiter over `store`, with `home` as the magi home its runs live under.
191    pub fn new(store: Questions, home: PathBuf, conduct_cfg: Option<Config>) -> Self {
192        let default_timeout = conduct_cfg
193            .as_ref()
194            .map_or(Config::default().graph.answer_timeout, |c| {
195                c.graph.answer_timeout
196            });
197        Self {
198            store,
199            home,
200            conduct_cfg,
201            default_timeout,
202            memo: HashMap::new(),
203        }
204    }
205
206    /// Is this question's asking seat still mid-turn?
207    fn seat_busy(&self, q: &Question, now: Timestamp) -> bool {
208        if q.run == crate::conduct::NODE {
209            return crate::conduct::busy(&self.home);
210        }
211        RunState::load_under(&q.run, &self.home).is_ok_and(|s| {
212            s.seats_active()
213                .any(|(k, a)| *k == q.seat && a.remaining_secs(now) > 0)
214        })
215    }
216
217    /// Whether the daemon applies this question's chosen action, so the
218    /// waiter must not deliver the answer (see [`decide_owned`]).
219    fn daemon_owns_action(&self, q: &Question) -> bool {
220        q.chosen_action().is_some_and(|a| a.maintenance())
221            || crate::daemon::task_of_question(
222                &crate::queue::Queue::at(self.home.join("queue")).list(),
223                q,
224            )
225            .is_some_and(|t| crate::daemon::action_standing(t, q).daemon_owns())
226    }
227
228    /// Look at every question once and act on what needs acting on. `halt` is
229    /// asked between questions and during a delivery: true means the daemon is
230    /// parking, and whatever is in flight is dropped, undelivered and
231    /// resumable.
232    pub async fn tick(&mut self, now: Timestamp, halt: &(dyn Fn() -> bool + Sync)) {
233        for q in self.store.list() {
234            if halt() {
235                return;
236            }
237            let lease = self.store.read_lease(&q.id);
238            let owned = self.daemon_owns_action(&q);
239            match decide_owned(&q, lease.as_ref(), false, self.default_timeout, now, owned) {
240                Action::Idle => {}
241                Action::Expire => self.expire(&q),
242                Action::Deliver(word) => {
243                    // A question with a deputy is the deputy's to read
244                    // (`crate::deputy`); resuming the conductor's own seat
245                    // here would fork a conversation that is not its own.
246                    if q.deputy.is_some() {
247                        // Unless that deputy is spent: then nobody will read
248                        // the word, and the deadline still has to retire the
249                        // question.
250                        if crate::deputy::exhausted_past_deadline(
251                            &q,
252                            crate::deputy::can_start(
253                                self.conduct_cfg.as_ref(),
254                                crate::deputy::agent_of(&q),
255                            ),
256                            self.default_timeout,
257                            now,
258                        ) {
259                            self.expire(&q);
260                        }
261                        continue;
262                    }
263                    if self.seat_busy(&q, now) {
264                        continue;
265                    }
266                    let key = format!("{}:{:?}", q.thread.len(), word);
267                    if matches!(self.memo.get(&q.id), Some((k, until)) if *k == key && Instant::now() < *until)
268                    {
269                        continue;
270                    }
271                    if let Err(e) = self.deliver(&q, &word, halt).await {
272                        tracing::warn!("question {}: {e:#}", q.short());
273                    }
274                }
275            }
276        }
277    }
278
279    /// Abandon an unanswered question whose deadline passed with nobody
280    /// waiting - the words the asker's own wait would have used.
281    fn expire(&self, q: &Question) {
282        let secs = if q.answer_timeout > 0 {
283            q.answer_timeout
284        } else {
285            self.default_timeout
286        };
287        let startable =
288            crate::deputy::can_start(self.conduct_cfg.as_ref(), crate::deputy::agent_of(q));
289        let why = format!("no answer within {}s of asking", secs.max(1));
290        let done = self.store.update(&q.id, |r| {
291            // Decided again on the record as it is now: the owner may have
292            // spoken since the tick read it, and a word given in time is
293            // delivered, not abandoned.
294            let now = Timestamp::now();
295            let lease = self.store.read_lease(&r.id);
296            if decide(r, lease.as_ref(), false, self.default_timeout, now) != Action::Expire
297                && !(lease.as_ref().is_none_or(|l| !l.fresh(now))
298                    && crate::deputy::exhausted_past_deadline(
299                        r,
300                        startable,
301                        self.default_timeout,
302                        now,
303                    ))
304            {
305                return Ok(false);
306            }
307            r.abandon(&why);
308            r.waiter = None;
309            Ok(true)
310        });
311        match done {
312            Ok((_, false)) => {}
313            Ok((_, true)) => {
314                self.store.drop_lease(&q.id);
315                tracing::warn!(
316                    "question {} went unanswered for {secs}s with nobody waiting; \
317                     it stays as the record of it",
318                    q.short()
319                );
320            }
321            Err(e) => tracing::warn!("could not expire question {}: {e:#}", q.short()),
322        }
323    }
324
325    /// Admission uses the very same target resolution as actual delivery.
326    pub fn can_resume(&self, q: &Question) -> bool {
327        self.target(q).is_ok()
328    }
329
330    /// Work out how to resume the seat, or why it cannot be.
331    fn target(&self, q: &Question) -> std::result::Result<Target, String> {
332        let cwd = q
333            .cwd
334            .as_deref()
335            .map(PathBuf::from)
336            .filter(|p| p.is_dir())
337            .ok_or("the directory the agent was working in is gone")?;
338        let (cfg, seat, allow_write, fetched_base) = if q.run == crate::conduct::NODE {
339            let cfg = self
340                .conduct_cfg
341                .clone()
342                .ok_or("the conductor's configuration is not available")?;
343            let seat = crate::conduct::load_seat(&self.home)
344                .ok_or("the conductor has no recorded session")?;
345            (cfg, seat, false, None)
346        } else {
347            let state = RunState::load_under(&q.run, &self.home)
348                .map_err(|_| "the run that asked is gone".to_owned())?;
349            if !state.status.resumable() {
350                return Err(format!(
351                    "the run that asked has already {}",
352                    state.status.as_str()
353                ));
354            }
355            let seat = state
356                .seats
357                .get(&q.seat)
358                .cloned()
359                .ok_or("the run has no record of the asking seat")?;
360            let write = matches!(q.node.as_str(), "implement" | "fix");
361            (state.config, seat, write, state.fetched_base)
362        };
363        let spec = cfg
364            .agent(&seat.agent)
365            .map_err(|_| format!("agent `{}` is no longer in the roster", seat.agent))?
366            .clone();
367        if !agent::has_session(spec.kind, &seat, cfg.graph.sessions) {
368            return Err("the seat has no session to resume".to_owned());
369        }
370        Ok(Target {
371            coordination: cfg.coordination.clone(),
372            fetched_base,
373            spec,
374            seat,
375            cwd,
376            sessions: cfg.graph.sessions,
377            allow_write,
378            language: cfg.graph.language.clone(),
379        })
380    }
381
382    /// Say why the owner's word cannot reach anybody, once per input.
383    fn refuse(&mut self, q: &Question, key: String, why: &str) {
384        tracing::warn!("question {} cannot be delivered: {why}", q.short());
385        notices::raise_in(
386            &self.home,
387            Notice::warn(
388                &format!("question:{}", q.id),
389                format!(
390                    "Question {} \"{}\": what you said cannot reach the agent \
391                     ({why}). It is recorded, but nothing will read it.",
392                    q.short(),
393                    q.summary
394                ),
395            ),
396        );
397        let _ = self.store.update(&q.id, |r| {
398            r.waiter = None;
399            Ok(())
400        });
401        self.memo
402            .insert(q.id.clone(), (key, Instant::now() + RETRY_AFTER));
403    }
404
405    /// Resume the asking seat with the owner's word.
406    async fn deliver(
407        &mut self,
408        q: &Question,
409        word: &Word,
410        halt: &(dyn Fn() -> bool + Sync),
411    ) -> Result<()> {
412        let key = format!("{}:{:?}", q.thread.len(), word);
413        let mut target = match self.target(q) {
414            Ok(t) => t,
415            Err(why) => {
416                self.refuse(q, key, &why);
417                return Ok(());
418            }
419        };
420
421        let claim = self.store.root().join(format!("{}.claim", q.id));
422        if !take_claim(&claim, DELIVERY_TIMEOUT + Duration::from_secs(60)) {
423            return Ok(());
424        }
425        let _release = Release(claim);
426
427        // Re-read under the claim: the asker may have come back, or the owner
428        // may have said more, since the tick's copy was taken.
429        let q = self.store.get(&q.id)?;
430        let now = Timestamp::now();
431        let lease = self.store.read_lease(&q.id);
432        let owned = self.daemon_owns_action(&q);
433        let Action::Deliver(word) =
434            decide_owned(&q, lease.as_ref(), false, self.default_timeout, now, owned)
435        else {
436            return Ok(());
437        };
438        let snapshot = q.thread.len();
439
440        // Ownership of an action answer is settled against the daemon's task
441        // claim: hold it while the lease goes up, so the daemon either sees the
442        // fresh lease or has already applied the action (and we step aside).
443        // A claim held by a running competition is not the daemon applying an
444        // action; that case is delivered, with the re-check below.
445        let queue = crate::queue::Queue::at(self.home.join("queue"));
446        let task_id = crate::daemon::task_of_question(&queue.list(), &q).map(|t| t.id.clone());
447        // Never go on without the claim: a failed claim means an attempt (or
448        // the daemon) holds the task, and whatever it does next decides who
449        // owns this answer. Step aside; the next tick judges it afresh.
450        let _task_claim = match (&task_id, q.chosen_action()) {
451            (Some(id), Some(_)) => match queue.claim(id) {
452                Ok(c) => Some(c),
453                Err(e) => {
454                    tracing::debug!(
455                        "question {}: task claim not available ({e:#}), deferring delivery",
456                        q.short()
457                    );
458                    return Ok(());
459                }
460            },
461            _ => None,
462        };
463        // Applied, stale and pending are all stop conditions; only a busy task
464        // (or no action at all) is delivered to.
465        let task_still_ours = || {
466            if q.chosen_action().is_some_and(|a| a.maintenance()) {
467                false
468            } else {
469                match &task_id {
470                    Some(id) => queue
471                        .get(id)
472                        .map(|t| !crate::daemon::action_standing(&t, &q).daemon_owns())
473                        .unwrap_or(true),
474                    None => true,
475                }
476            }
477        };
478        if !task_still_ours() {
479            return Ok(());
480        }
481
482        self.store.beat(&q.id, WaiterKind::Daemon);
483        // Re-check after the lease is up: a daemon that decided before it
484        // existed has by now written its mark (or is blocked on our claim).
485        if !task_still_ours() {
486            return Ok(());
487        }
488        self.store.update(&q.id, |r| {
489            r.waiter = Some(Note {
490                kind: WaiterKind::Daemon,
491                since: now,
492            });
493            Ok(())
494        })?;
495        tracing::info!(
496            "question {}: the asker is gone, resuming seat {} with the owner's word",
497            q.short(),
498            q.seat
499        );
500
501        let thread: Vec<(&str, &str)> = q
502            .thread
503            .iter()
504            .map(|t| {
505                (
506                    if t.who == Who::Operator {
507                        "operator"
508                    } else {
509                        "agent"
510                    },
511                    t.body.as_str(),
512                )
513            })
514            .collect();
515        // What the owner said and the agent has not read has its own section in
516        // the prompt, so the recap stops at what it already read.
517        let shown = match word {
518            Word::Said(_) => &thread[..q.delivered_turns.min(thread.len())],
519            Word::Answered(_) => &thread[..],
520        };
521        let owner_word = match &word {
522            Word::Said(s) => prompt::OwnerWord::Said(s),
523            Word::Answered(a) => prompt::OwnerWord::Answered(a),
524        };
525        let body = prompt::question_resumed(
526            &q.id,
527            &q.summary,
528            &q.detail,
529            shown,
530            &owner_word,
531            &target.language,
532        );
533
534        let artifacts = self.store.root().join(format!("{}.delivery", q.id));
535        let stem = format!("deliver-{}", now.as_second());
536        let cache_dir = None;
537        let inv = Invocation {
538            scope: crate::agent::InvocationScope::Repository,
539            coordination: target.coordination.as_ref(),
540            fetched_base: target.fetched_base.as_ref(),
541            cwd: &target.cwd,
542            prompt: &body,
543            timeout: DELIVERY_TIMEOUT,
544            allow_write: target.allow_write,
545            unsandboxed: false,
546            sessions: target.sessions,
547            artifacts: &artifacts,
548            stem: &stem,
549            run: &q.run,
550            node: &q.node,
551            cache_dir,
552            attachments: &[],
553            writable: &[],
554        };
555
556        let store = self.store.clone();
557        let id = q.id.clone();
558        let out = {
559            let fut = agent::invoke(&target.spec, &mut target.seat, &inv);
560            tokio::pin!(fut);
561            let mut beat = tokio::time::interval(Duration::from_secs(1));
562            let mut beats = 0u32;
563            loop {
564                tokio::select! {
565                    r = &mut fut => break Some(r),
566                    _ = beat.tick() => {
567                        if halt() {
568                            break None;
569                        }
570                        beats += 1;
571                        if beats % 20 == 0 {
572                            store.beat(&id, WaiterKind::Daemon);
573                        }
574                    }
575                }
576            }
577        };
578
579        let Some(out) = out else {
580            // Parking: leave everything as it was, resumable.
581            let _ = self.store.update(&q.id, |r| {
582                r.waiter = None;
583                Ok(())
584            });
585            return Ok(());
586        };
587        let out = match out {
588            Ok(o) if o.usable() => o,
589            other => {
590                let why = match other {
591                    Ok(o) if o.timed_out => "the resumed turn timed out".to_owned(),
592                    Ok(o) if o.quota_exhausted() => "the agent is out of quota".to_owned(),
593                    Ok(o) => format!(
594                        "the resumed turn failed ({})",
595                        notices::exit_status(o.exit_code)
596                    ),
597                    Err(e) => format!("the agent could not be started: {e:#}"),
598                };
599                self.refuse(&q, key, &why);
600                return Ok(());
601            }
602        };
603
604        let text = out.text.trim().to_owned();
605        self.store.update(&q.id, |r| {
606            r.waiter = None;
607            match &word {
608                Word::Answered(_) => r.answer_delivered = true,
609                Word::Said(_) => {
610                    r.delivered_turns = r.delivered_turns.max(snapshot);
611                    // An agent that answered in prose instead of calling
612                    // `magi ask --thread` still spoke; keep it on the record so
613                    // the owner reads it. If it did use --thread, or the owner
614                    // said more meanwhile, the record already moved on.
615                    let agent_spoke = r.thread[snapshot.min(r.thread.len())..]
616                        .iter()
617                        .any(|t| t.who == Who::Agent);
618                    if !agent_spoke && r.thread.len() == snapshot && r.status.open() {
619                        let choices = r.choices.clone();
620                        r.reply(text, choices)?;
621                    }
622                }
623            }
624            Ok(())
625        })?;
626        self.memo.remove(&q.id);
627        Ok(())
628    }
629}
630
631/// Take the delivery claim, so two waiters (`magi serve` twice, or a stray
632/// second daemon) cannot resume the same seat at once. A claim older than a
633/// whole delivery plus slack was left by a process that died.
634pub(crate) fn take_claim(path: &std::path::Path, stale_after: Duration) -> bool {
635    let attempt = || {
636        std::fs::OpenOptions::new()
637            .write(true)
638            .create_new(true)
639            .open(path)
640            .is_ok()
641    };
642    if attempt() {
643        return true;
644    }
645    let stale = std::fs::metadata(path)
646        .and_then(|m| m.modified())
647        .ok()
648        .and_then(|t| t.elapsed().ok())
649        .is_some_and(|age| age > stale_after);
650    if stale {
651        let _ = std::fs::remove_file(path);
652        return attempt();
653    }
654    false
655}
656
657/// Removes the delivery claim when the delivery ends, however it ends.
658pub(crate) struct Release(pub(crate) PathBuf);
659
660impl Drop for Release {
661    fn drop(&mut self) {
662        let _ = std::fs::remove_file(&self.0);
663    }
664}
665
666/// The waiter's loop, until `stop` is asked for.
667///
668/// A stop ends the loop between ticks; a park also drops a delivery in flight.
669/// Neither writes anything to a question: the lease just ages out and the next
670/// waiter - this daemon restarted, or another - reads the same state off disk.
671pub async fn run(mut waiter: Waiter, stop: crate::daemon::Stop) {
672    let halt = {
673        let stop = stop.clone();
674        move || stop.parking()
675    };
676    while !stop.stopped() {
677        waiter.tick(Timestamp::now(), &halt).await;
678        tokio::time::sleep(TICK).await;
679    }
680}
681
682#[cfg(test)]
683mod tests {
684    use super::*;
685
686    fn ts(secs: i64) -> Timestamp {
687        Timestamp::from_second(secs).unwrap()
688    }
689
690    fn asked(at: i64) -> Question {
691        let mut q = Question::new(
692            "run".into(),
693            "implement".into(),
694            "impl-A".into(),
695            "which?".into(),
696            String::new(),
697            vec![],
698        );
699        q.cwd = Some("/tmp".into());
700        q.asked_at = ts(at);
701        q.answer_timeout = 1000;
702        q
703    }
704
705    fn lease(at: i64) -> Lease {
706        Lease {
707            kind: WaiterKind::Asker,
708            pid: 1,
709            beat_at: ts(at),
710        }
711    }
712
713    #[test]
714    fn a_fresh_lease_means_nobody_else_acts() {
715        let mut q = asked(0);
716        q.say("why?").unwrap();
717        assert_eq!(
718            decide(&q, Some(&lease(95)), false, 86_400, ts(100)),
719            Action::Idle
720        );
721    }
722
723    #[test]
724    fn a_stale_lease_delivers_what_the_owner_said() {
725        let mut q = asked(0);
726        q.say("why?").unwrap();
727        assert_eq!(
728            decide(&q, Some(&lease(0)), false, 86_400, ts(500)),
729            Action::Deliver(Word::Said("why?".into()))
730        );
731        assert_eq!(
732            decide(&q, None, true, 86_400, ts(500)),
733            Action::Idle,
734            "a seat still mid-turn is not resumed"
735        );
736    }
737
738    #[test]
739    fn an_answer_nobody_read_is_delivered_once() {
740        let mut q = asked(0);
741        q.choices = vec!["A".into()];
742        q.answer(crate::ask::Answer::Choice("A".into())).unwrap();
743        assert_eq!(
744            decide(&q, None, false, 86_400, ts(10)),
745            Action::Deliver(Word::Answered("A".into()))
746        );
747        q.answer_delivered = true;
748        assert_eq!(decide(&q, None, false, 86_400, ts(10)), Action::Idle);
749    }
750
751    #[test]
752    fn only_asked_questions_and_only_before_the_deadline() {
753        let mut q = asked(0);
754        assert_eq!(decide(&q, None, false, 86_400, ts(999)), Action::Idle);
755        assert_eq!(decide(&q, None, false, 86_400, ts(1001)), Action::Expire);
756        q.cwd = None;
757        assert_eq!(decide(&q, None, false, 86_400, ts(1001)), Action::Idle);
758    }
759
760    #[test]
761    fn the_deadline_runs_from_the_last_turn_not_from_asking() {
762        let mut q = asked(0);
763        q.thread.push(crate::ask::Turn {
764            who: Who::Agent,
765            body: "context".into(),
766            at: ts(4000),
767            note: None,
768        });
769        q.delivered_turns = 1;
770        assert_eq!(decide(&q, None, false, 86_400, ts(4500)), Action::Idle);
771        assert_eq!(decide(&q, None, false, 86_400, ts(5001)), Action::Expire);
772    }
773
774    #[test]
775    fn a_word_given_in_time_is_delivered_after_the_deadline() {
776        let mut q = asked(0);
777        q.say("wait").unwrap();
778        assert!(matches!(
779            decide(&q, None, false, 86_400, ts(5000)),
780            Action::Deliver(_)
781        ));
782    }
783
784    #[test]
785    fn an_action_answer_is_left_to_the_daemon() {
786        let mut q = asked(0);
787        q.choices = vec!["A".into()];
788        q.actions
789            .insert("A".into(), crate::ask::ChoiceAction::Requeue);
790        q.answer(crate::ask::Answer::Choice("A".into())).unwrap();
791        assert_eq!(
792            decide_owned(&q, None, false, 86_400, ts(10), true),
793            Action::Idle,
794            "the daemon applies the action; the waiter must not also resume the seat"
795        );
796        // Delivered or not, a stale lease changes nothing.
797        q.answer_delivered = true;
798        assert_eq!(
799            decide_owned(&q, None, false, 86_400, ts(10), true),
800            Action::Idle
801        );
802        // No task to apply it: the words are delivered as before.
803        q.answer_delivered = false;
804        assert_eq!(
805            decide_owned(&q, None, false, 86_400, ts(10), false),
806            Action::Deliver(Word::Answered("A".into()))
807        );
808        // No action: delivered even when a task exists.
809        let mut plain = asked(0);
810        plain.choices = vec!["A".into()];
811        plain
812            .answer(crate::ask::Answer::Choice("A".into()))
813            .unwrap();
814        assert_eq!(
815            decide_owned(&plain, None, false, 86_400, ts(10), true),
816            Action::Deliver(Word::Answered("A".into()))
817        );
818    }
819}