1use 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::notices::{self, Link, Notice};
67use crate::queue::{Queue, Task, TaskStatus};
68use crate::run::{Liveness, QuotaLoss, RunState, RunStatus};
69use crate::triage;
70
71pub const SCHEMA: u32 = 1;
73
74pub const HEARTBEAT: Duration = Duration::from_secs(5);
78
79pub const STALE_SECS: i64 = 30;
88
89pub const POLL: Duration = Duration::from_secs(5);
91
92pub const STALE_CLAIM: Duration = Duration::from_secs(6 * 60 * 60);
96
97pub const STALLED_RUNNING: Duration = Duration::from_secs(30 * 60);
116
117#[derive(Debug, Clone, Default, PartialEq, Eq, Serialize, Deserialize)]
119#[serde(default)]
120pub struct Current {
121 pub task: String,
123 pub run: String,
125}
126
127#[derive(Debug, Clone, Serialize, Deserialize)]
134pub struct Status {
135 pub schema: u32,
137 pub pid: u32,
139 pub started_at: Timestamp,
141 pub updated_at: Timestamp,
143 pub idle: bool,
145 pub current: Vec<Current>,
151 pub completed: usize,
153 pub polls: u64,
155 #[serde(default)]
157 pub question_recovery: Option<crate::question_delivery::Recovery>,
158}
159
160impl Status {
161 #[must_use]
163 pub fn new() -> Self {
164 let now = Timestamp::now();
165 Self {
166 schema: SCHEMA,
167 pid: std::process::id(),
168 started_at: now,
169 updated_at: now,
170 idle: true,
171 current: Vec::new(),
172 completed: 0,
173 polls: 0,
174 question_recovery: None,
175 }
176 }
177}
178
179impl Default for Status {
180 fn default() -> Self {
181 Self::new()
182 }
183}
184
185#[derive(Debug, Clone)]
187pub struct Opts {
188 pub repo: PathBuf,
190 pub config: Option<PathBuf>,
192 pub poll: Duration,
194 pub max_attempts: usize,
196 pub once: bool,
198 pub merge: Option<String>,
200 pub worktrees_root: Option<PathBuf>,
209}
210
211impl Default for Opts {
212 fn default() -> Self {
213 Self {
214 repo: PathBuf::from("."),
215 config: None,
216 poll: POLL,
217 max_attempts: 2,
218 once: false,
219 merge: None,
220 worktrees_root: None,
221 }
222 }
223}
224
225fn max_concurrent(n: usize) -> usize {
230 n.max(1)
231}
232
233#[must_use]
235pub fn status_path() -> PathBuf {
236 crate::run::home().join("daemon.json")
237}
238
239pub fn write_status(status: &Status) -> Result<()> {
241 write_status_to(&status_path(), status)
242}
243
244pub fn write_status_to(path: &Path, status: &Status) -> Result<()> {
249 if let Some(parent) = path.parent() {
250 std::fs::create_dir_all(parent).with_context(|| format!("create {}", parent.display()))?;
251 }
252 let body = serde_json::to_string_pretty(status).context("serialize daemon status")?;
253 let tmp = path.with_extension("json.tmp");
254 std::fs::write(&tmp, &body).with_context(|| format!("write {}", tmp.display()))?;
255 std::fs::rename(&tmp, path).with_context(|| format!("replace {}", path.display()))?;
256 Ok(())
257}
258
259pub fn clear_status() {
262 clear_status_at(&status_path());
263}
264
265fn clear_status_at(path: &Path) {
269 let _ = std::fs::remove_file(path);
270}
271
272#[derive(Debug, Clone, Default)]
285pub struct Stop {
286 stopped: Arc<AtomicBool>,
290 busy: Arc<std::sync::atomic::AtomicUsize>,
296 wake: Arc<Notify>,
300 pause: crate::graph::Pause,
303}
304
305impl Stop {
306 #[must_use]
308 pub fn new() -> Self {
309 Self::default()
310 }
311
312 pub fn stop(&self) {
315 self.stopped.store(true, Ordering::SeqCst);
316 self.wake.notify_one();
320 }
321
322 #[must_use]
324 pub fn stopped(&self) -> bool {
325 self.stopped.load(Ordering::SeqCst)
326 }
327
328 #[must_use]
336 pub fn finishing(&self) -> bool {
337 self.stopped() && self.busy_now()
338 }
339
340 pub fn park(&self, cause: crate::run::ParkCause) {
351 self.pause.park_for(cause);
352 self.stop();
353 }
354
355 #[must_use]
357 pub fn parking(&self) -> bool {
358 self.pause.parked()
359 }
360
361 #[must_use]
363 pub fn pause(&self) -> crate::graph::Pause {
364 self.pause.clone()
365 }
366
367 #[must_use]
373 pub fn busy_now(&self) -> bool {
374 self.busy.load(Ordering::SeqCst) > 0
375 }
376
377 fn enter(&self) {
379 self.busy.fetch_add(1, Ordering::SeqCst);
380 }
381
382 fn exit(&self) {
385 self.busy.fetch_sub(1, Ordering::SeqCst);
386 }
387
388 async fn idle(&self, poll: Duration) {
390 tokio::select! {
391 () = tokio::time::sleep(poll) => {}
392 () = self.wake.notified() => {}
393 }
394 }
395}
396
397#[derive(Debug, Clone, Default, Deserialize)]
404#[serde(default)]
405pub struct Reading {
406 pub schema: u32,
408 pub pid: Option<u32>,
410 pub started_at: Option<Timestamp>,
412 pub updated_at: Option<Timestamp>,
414 pub idle: bool,
416 #[serde(deserialize_with = "de_current")]
428 pub current: Vec<Current>,
429 pub completed: u64,
431 pub polls: u64,
433 #[serde(default)]
435 pub question_recovery: Option<crate::question_delivery::Recovery>,
436}
437
438fn de_current<'de, D>(deserializer: D) -> std::result::Result<Vec<Current>, D::Error>
441where
442 D: serde::Deserializer<'de>,
443{
444 #[derive(Deserialize)]
445 #[serde(untagged)]
446 enum Shape {
447 Many(Vec<Current>),
448 One(Current),
449 }
450 Ok(
451 Option::<Shape>::deserialize(deserializer)?.map_or_else(Vec::new, |shape| match shape {
452 Shape::Many(v) => v,
453 Shape::One(c) => vec![c],
454 }),
455 )
456}
457
458impl Reading {
459 #[must_use]
462 pub fn age_secs(&self, now: Timestamp) -> Option<i64> {
463 self.updated_at
464 .map(|at| (now.as_second() - at.as_second()).max(0))
465 }
466
467 #[must_use]
471 pub fn running(&self, now: Timestamp) -> bool {
472 self.age_secs(now).is_some_and(|secs| secs <= STALE_SECS)
473 }
474}
475
476#[must_use]
483pub fn read_status(home: &Path) -> Option<Reading> {
484 let body = std::fs::read_to_string(home.join("daemon.json")).ok()?;
485 serde_json::from_str(&body).ok()
486}
487
488#[must_use]
500pub fn current_work(home: &Path, now: Timestamp) -> Vec<Current> {
501 read_status(home)
502 .filter(|reading| reading.running(now))
503 .map(|reading| reading.current)
504 .unwrap_or_default()
505}
506
507#[must_use]
509pub fn is_working_on(home: &Path, run: &str, now: Timestamp) -> bool {
510 current_work(home, now).iter().any(|c| c.run == run)
511}
512
513#[must_use]
523pub fn is_working_on_short(home: &Path, short: &str, now: Timestamp) -> bool {
524 current_work(home, now)
525 .iter()
526 .any(|c| crate::run::short_of(&c.run) == short)
527}
528
529#[must_use]
531pub fn is_working_on_task(home: &Path, task: &str, now: Timestamp) -> bool {
532 current_work(home, now).iter().any(|c| c.task == task)
533}
534
535pub fn sweep_stale_claims(queue: &Queue, older_than: Duration) -> Vec<String> {
579 sweep_stale_claims_with(queue, older_than, crate::proc::pid_alive)
580}
581
582fn sweep_stale_claims_with<F>(queue: &Queue, older_than: Duration, pid_alive: F) -> Vec<String>
586where
587 F: Fn(u32) -> bool,
588{
589 let this_process = std::process::id();
590 let mut swept: Vec<String> = std::fs::read_dir(queue.root())
591 .into_iter()
592 .flatten()
593 .flatten()
594 .map(|e| e.path())
595 .filter(|p| p.extension().is_some_and(|x| x == "lock"))
596 .filter(|p| {
597 match std::fs::read_to_string(p)
598 .ok()
599 .and_then(|body| body.trim().parse::<u32>().ok())
600 {
601 Some(pid) if pid == this_process => false,
605 Some(pid) => !pid_alive(pid),
606 None => p
607 .metadata()
608 .and_then(|m| m.modified())
609 .and_then(|t| t.elapsed().map_err(std::io::Error::other))
610 .is_ok_and(|age| age >= older_than),
611 }
612 })
613 .filter(|p| std::fs::remove_file(p).is_ok())
614 .filter_map(|p| {
615 p.file_stem()
616 .and_then(|s| s.to_str())
617 .map(std::borrow::ToOwned::to_owned)
618 })
619 .collect();
620 swept.sort_unstable();
621 swept
622}
623
624fn is_stalled(task: &Task, home: &Path, now: Timestamp) -> bool {
629 task.status == TaskStatus::Running
630 && (now.as_second() - task.updated_at.as_second()) >= STALLED_RUNNING.as_secs() as i64
631 && !is_working_on_task(home, &task.id, now)
632}
633
634fn stalled_tasks(queue: &Queue, home: &Path, now: Timestamp) -> Vec<Task> {
637 queue
638 .list()
639 .into_iter()
640 .filter(|t| is_stalled(t, home, now))
641 .collect()
642}
643
644fn queued_tasks(queue: &Queue) -> Vec<Task> {
650 queue
651 .list()
652 .into_iter()
653 .filter(|t| t.status == TaskStatus::Queued)
654 .collect()
655}
656
657fn finished_tasks(queue: &Queue) -> Vec<Task> {
660 queue
661 .list()
662 .into_iter()
663 .filter(|t| matches!(t.status, TaskStatus::Failed | TaskStatus::Held))
664 .collect()
665}
666
667fn resolve_blockers(queue: &Queue, questions: &Questions) {
684 for listed in queue.list() {
685 if listed.status != TaskStatus::Blocked || listed.blocked_by.is_empty() {
686 continue;
687 }
688 let Ok(_claim) = queue.claim(&listed.id) else {
689 continue;
690 };
691 let Ok(mut task) = queue.get(&listed.id) else {
692 continue;
693 };
694 if task.status != TaskStatus::Blocked {
695 continue;
696 }
697 let deleted = queue.apply_deleted_blockers(&mut task, questions);
701 if !deleted.is_empty() {
702 record(queue, &mut task);
703 for id in &deleted {
704 queue.note_dependency_deleted(&task, id);
705 }
706 if task.status != TaskStatus::Blocked {
707 continue;
708 }
709 }
710 let missing = crate::queue::missing_blockers(queue, questions, &task.blocked_by);
711 if !missing.is_empty() {
712 let language = language_of(&task, Path::new("."));
713 let mut cause = missing.clone();
714 cause.sort();
715 cause.dedup();
716 task.notice_cause(&format!("missing-blockers:{}", cause.join(",")), None, None);
717 task.hold_machine(Some(crate::queue::missing_blocker_hold_reason_in(
718 &task.blocked_by,
719 &missing,
720 &language,
721 )));
722 record(queue, &mut task);
723 continue;
724 }
725 if let Some(q) = task.blocked_by.iter().find_map(|id| {
728 let crate::queue::Dependency::Question(full) =
729 crate::queue::dependency(queue, questions, id)
730 else {
731 return None;
732 };
733 questions.get(&full).ok().filter(|q| {
734 q.node == crate::conduct::NODE && q.status == ask::QuestionStatus::Abandoned
735 })
736 }) {
737 let language = language_of(&task, Path::new("."));
738 task.notice_cause(&format!("unanswered-question:{}", q.id), None, None);
739 task.hold_machine(Some(unanswered_question_hold_reason(&q, &language)));
740 record(queue, &mut task);
741 continue;
742 }
743 let mut changed = false;
744 for id in task.blocked_by.clone() {
745 let resolved = crate::queue::dependency(queue, questions, &id);
746 if let crate::queue::Dependency::Task(full) = &resolved {
747 if queue
748 .get(full)
749 .is_ok_and(|dep| dep.status == TaskStatus::Done)
750 {
751 task.unblock(&id);
752 changed = true;
753 }
754 continue;
755 }
756 if let crate::queue::Dependency::Question(full) = resolved
757 && let Ok(q) = questions.get(&full)
758 && q.status == ask::QuestionStatus::Answered
759 {
760 if q.node == crate::conduct::NODE
764 && q.chosen_action().is_some()
765 && !task.action_applied(&q.id)
766 {
767 continue;
768 }
769 let answer = match &q.answer {
770 Some(ask::Answer::Choice(c) | ask::Answer::Text(c)) => c.clone(),
771 None => String::new(),
772 };
773 task.record_answer(q.summary.clone(), answer);
774 task.unblock(&id);
775 changed = true;
776 }
777 }
778 if changed {
779 record(queue, &mut task);
780 }
781 }
782}
783
784fn unanswered_question_hold_reason(q: &ask::Question, language: &str) -> String {
786 if crate::lang::is_japanese(language) {
787 format!(
788 "質問 {} 「{}」 に期限内の回答がなく、取り下げられました - `magi task triage` を参照",
789 q.short(),
790 q.summary
791 )
792 } else {
793 format!(
794 "question {} \"{}\" went unanswered and was abandoned - see `magi task triage`",
795 q.short(),
796 q.summary
797 )
798 }
799}
800
801#[derive(Debug, Clone, Copy, PartialEq, Eq)]
804pub(crate) enum ActionStanding {
805 NoAction,
807 Pending,
809 Applied,
812 Stale,
814 Busy,
816}
817
818impl ActionStanding {
819 pub(crate) fn daemon_owns(self) -> bool {
821 matches!(self, Self::Pending | Self::Applied | Self::Stale)
822 }
823}
824
825pub(crate) fn action_standing(task: &Task, q: &ask::Question) -> ActionStanding {
829 if q.chosen_action().is_none() {
830 ActionStanding::NoAction
831 } else if task.action_applied(&q.id) {
832 ActionStanding::Applied
833 } else if q.node != crate::conduct::NODE && task.runs.last() != Some(&q.run) {
834 ActionStanding::Stale
835 } else if matches!(
836 task.status,
837 TaskStatus::Running | TaskStatus::Blocked | TaskStatus::Done
838 ) {
839 ActionStanding::Busy
840 } else {
841 ActionStanding::Pending
842 }
843}
844
845#[derive(Debug, Clone, PartialEq, Eq)]
847enum ActionDecision {
848 Skip,
850 Resume(String),
852 Requeue,
854 Hold,
856 Done,
858 Stale,
861 Refuse(String),
864}
865
866fn decide_action<F>(task: &Task, q: &ask::Question, p: &Phrases, load: F) -> ActionDecision
874where
875 F: FnOnce(&str) -> Result<RunState>,
876{
877 let Some(action) = q.chosen_action() else {
878 return ActionDecision::Skip;
879 };
880 if task.action_applied(&q.id)
881 || matches!(
882 task.status,
883 TaskStatus::Running | TaskStatus::Blocked | TaskStatus::Done
884 )
885 {
886 return ActionDecision::Skip;
887 }
888 if q.node != crate::conduct::NODE && task.runs.last() != Some(&q.run) {
891 return ActionDecision::Stale;
892 }
893 let repair = q.node == land::APPROVAL_NODE && q.resolution().as_deref() == Some(land::REPAIR);
894 let stopping = q.node == land::APPROVAL_NODE
895 && q.resolution()
896 .as_deref()
897 .is_some_and(crate::merge_decision::stops);
898 let decision_at = if repair || stopping {
902 q.answered_at.unwrap_or(q.asked_at)
903 } else {
904 q.asked_at
905 };
906 if task.operator_decided_at.is_some_and(|at| at > decision_at) {
907 return if repair || stopping {
908 ActionDecision::Refuse(format!(
909 "{} answer {} was superseded by a later operator decision; the task stays held. Inspect `magi show {}`; if this action is still wanted, explicitly release the existing owner with `magi task release {}`",
910 if repair { "repair" } else { "stop" },
911 q.id,
912 q.run,
913 task.id
914 ))
915 } else {
916 ActionDecision::Stale
917 };
918 }
919 match action {
920 a if a.maintenance() => ActionDecision::Skip,
921 ask::ChoiceAction::ResumeRun { .. }
922 | ask::ChoiceAction::ReleaseTask { .. }
923 | ask::ChoiceAction::RerunChecks { .. }
924 | ask::ChoiceAction::AbandonRun { .. } => unreachable!(),
925 ask::ChoiceAction::Requeue => ActionDecision::Requeue,
926 ask::ChoiceAction::Hold => ActionDecision::Hold,
927 ask::ChoiceAction::Done => ActionDecision::Done,
928 ask::ChoiceAction::Resume { run } => {
929 if task.runs.last() != Some(run) {
930 return ActionDecision::Refuse((p.resume_not_latest)(
931 q.short(),
932 ask::short_id(run),
933 ));
934 }
935 match load(run) {
936 Ok(s) if stopping => {
937 if s.merge_decision.as_ref().is_some_and(|d| d.closed)
938 || !s
939 .land_approval
940 .as_ref()
941 .is_some_and(|binding| binding.question == q.id)
942 {
943 ActionDecision::Stale
944 } else if s.liveness(false) == Liveness::Live {
945 ActionDecision::Skip
946 } else if s.status == RunStatus::Landing
947 && !s.released()
948 && s.retired_to.is_none()
949 {
950 ActionDecision::Resume(run.clone())
951 } else {
952 ActionDecision::Refuse("Owner stop decision cannot resume its original landing execution; inspect the existing run. No replacement or merge was started.".into())
953 }
954 }
955 Ok(s)
956 if q.node == land::APPROVAL_NODE
957 && q.resolution().as_deref() == Some(land::REPAIR) =>
958 {
959 if s.merge_repairs.iter().any(|r| r.question == q.id)
960 || s.land_approval.as_ref().is_some_and(|a| a.question != q.id)
961 {
962 ActionDecision::Stale
965 } else if s.liveness(false) == Liveness::Live {
966 ActionDecision::Skip
967 } else if s.status != RunStatus::Landing
968 || s.released()
969 || s.reviews.is_empty()
970 || !implementers_match(task, &s)
971 || !s
972 .land_approval
973 .as_ref()
974 .is_some_and(|a| a.question == q.id && !a.head.is_empty())
975 {
976 ActionDecision::Refuse(format!(
977 "repair question {} no longer names a supported current approval on run {}; inspect `magi show {}` and restore the existing owner's approval before retrying; do not file another task",
978 q.id, run, run
979 ))
980 } else {
981 ActionDecision::Resume(run.clone())
982 }
983 }
984 Ok(s) if s.liveness(false) == Liveness::Live => {
985 ActionDecision::Refuse("run already has a live driver".into())
986 }
987 Ok(s) if crate::handover::can_resume(&s) => ActionDecision::Resume(run.clone()),
988 Ok(_) => ActionDecision::Refuse((p.resume_cannot_progress)(
989 q.short(),
990 ask::short_id(run),
991 )),
992 Err(e) => ActionDecision::Refuse((p.resume_unreadable)(
993 q.short(),
994 ask::short_id(run),
995 &format!("{e:#}"),
996 )),
997 }
998 }
999 }
1000}
1001
1002struct Phrases {
1016 graph_stopped: fn(&str, &str) -> String,
1018 quorum_lost: &'static str,
1019 quota_took_out: &'static str,
1021 run_ended: &'static str,
1023 waiting_for_answer: &'static str,
1025 recovered_running: &'static str,
1026 no_run_to_recover: &'static str,
1027 could_not_start: &'static str,
1028 resume_not_latest: fn(&str, &str) -> String,
1029 resume_cannot_progress: fn(&str, &str) -> String,
1030 resume_unreadable: fn(&str, &str, &str) -> String,
1031 handover_refused: fn(&crate::handover::Refused) -> String,
1034 handover_hint: &'static str,
1036 reviewers_never_answered: fn(usize, usize) -> String,
1039}
1040
1041const PHRASES_EN: Phrases = Phrases {
1042 graph_stopped: |status, detail| {
1043 format!("the graph stopped at `{status}` without reaching a terminal status: {detail}")
1044 },
1045 quorum_lost: "the judging panel lost its quorum",
1046 quota_took_out: "; quota took out ",
1047 run_ended: "run ended ",
1048 waiting_for_answer: " - waiting for operator answer to question ",
1049 recovered_running: "recovered a `running` task whose daemon never recorded the outcome: ",
1050 no_run_to_recover: "task was `running` with no live daemon and no readable \
1051 run to recover; held for a human to check what happened",
1052 could_not_start: "could not start the run: ",
1053 resume_not_latest: |q, run| {
1054 format!("question {q} asked to resume run {run}, which is not this task's latest run")
1055 },
1056 resume_cannot_progress: |q, run| {
1057 format!("question {q} asked to resume run {run}, which cannot make progress")
1058 },
1059 resume_unreadable: |q, run, e| {
1060 format!("question {q} asked to resume run {run}, which could not be read: {e}")
1061 },
1062 handover_refused: |r| r.to_string(),
1063 handover_hint: " (clean up the other worktree, then release the task from the queue)",
1064 reviewers_never_answered: |missing, rounds| {
1065 format!(
1066 "{missing} reviewer seat(s) never answered after {rounds} rounds; \
1067 refusing to call it clean"
1068 )
1069 },
1070};
1071
1072const PHRASES_JA: Phrases = Phrases {
1073 graph_stopped: |status, detail| {
1074 format!("グラフが終端状態に達しないまま `{status}` で停止しました: {detail}")
1075 },
1076 quorum_lost: "審査パネルが定足数を失いました",
1077 quota_took_out: "。クォータで脱落: ",
1078 run_ended: "run 終了: ",
1079 waiting_for_answer: " - オペレーターの回答待ち: 質問 ",
1080 recovered_running: "daemon が結果を記録しないまま `running` だったタスクを回収しました: ",
1081 no_run_to_recover: "タスクは `running` でしたが、生きた daemon も回収できる run も見つかりません。\
1082 何が起きたか人が確認するため保留にしました",
1083 could_not_start: "run を開始できませんでした: ",
1084 resume_not_latest: |q, run| {
1085 format!(
1086 "質問 {q} は run {run} の再開を求めましたが、これはタスクの最新の run ではありません"
1087 )
1088 },
1089 resume_cannot_progress: |q, run| {
1090 format!("質問 {q} は run {run} の再開を求めましたが、これは進行できません")
1091 },
1092 resume_unreadable: |q, run, e| {
1093 format!("質問 {q} は run {run} の再開を求めましたが、読み込めませんでした: {e}")
1094 },
1095 handover_refused: |r| {
1096 use crate::handover::Refused;
1097 match r {
1098 Refused::Foreign { branch, path, why } => format!(
1099 "ブランチ `{branch}` は {path} にチェックアウトされており、magi は自動では\
1100 削除しません。不要なら `git worktree remove` でその worktree を削除して\
1101 から、やり直してください(詳細: {why})"
1102 ),
1103 Refused::Unsafe { branch, path, why } => format!(
1104 "ブランチ `{branch}` は {path} にチェックアウトされています。そこの作業を\
1105 コミットか破棄したうえで `git worktree remove` で worktree を削除する\
1106 か、run を破棄してよいと伝えてから、やり直してください(詳細: {why})"
1107 ),
1108 Refused::ReleaseFailed { branch, path, run } => format!(
1109 "ブランチ `{branch}` は {path} で run {run} がチェックアウトしており、その\
1110 worktree の解放に失敗したか、変更が見つかりました(未コミットの変更が\
1111 ある worktree は git が削除を拒否します)。worktree はそのまま残しました"
1112 ),
1113 }
1114 },
1115 handover_hint: "(他の worktree を片付けてから、タスクをキューから解放してください)",
1116 reviewers_never_answered: |missing, rounds| {
1117 format!(
1118 "{missing} 席のレビュアーが {rounds} ラウンドの間に一度も回答しなかったため、\
1119 クリーンとは認めません"
1120 )
1121 },
1122};
1123
1124fn phrases(language: &str) -> &'static Phrases {
1125 if crate::lang::is_japanese(language) {
1126 &PHRASES_JA
1127 } else {
1128 &PHRASES_EN
1129 }
1130}
1131
1132fn language_of(task: &Task, fallback: &Path) -> String {
1136 crate::lang::of_repo(&repo_for(task, fallback))
1137}
1138
1139pub(crate) fn task_of_question<'a>(tasks: &'a [Task], q: &ask::Question) -> Option<&'a Task> {
1143 if q.node == crate::conduct::NODE {
1144 return tasks.iter().find(|t| t.id == q.run);
1145 }
1146 tasks.iter().find(|t| t.runs.contains(&q.run))
1147}
1148
1149fn repair_choice_offers(questions: &Questions) {
1152 for q in questions.list().into_iter().filter(|q| {
1153 q.status.open()
1154 && q.node == crate::conduct::NODE
1155 && q.choices
1156 .iter()
1157 .any(|c| crate::choice_effect::describe(q, c).is_none())
1158 }) {
1159 let _ = questions.update(&q.id, |current| {
1160 if current.status.open() {
1161 crate::choice_effect::executable_conductor(current);
1162 if let Some(deputy) = &mut current.deputy {
1163 deputy.brief = crate::deputy::brief(
1164 ¤t.run,
1165 ¤t.detail,
1166 ¤t.choices,
1167 ¤t.actions,
1168 );
1169 }
1170 }
1171 Ok(())
1172 });
1173 }
1174}
1175
1176pub async fn apply_choice_actions(queue: &Queue, questions: &Questions, home: &Path) {
1179 apply_maintenance_choices(queue, questions, home).await;
1180 apply_task_choice_actions(queue, questions, home);
1181}
1182
1183async fn apply_maintenance_choices(queue: &Queue, questions: &Questions, home: &Path) {
1186 for q in questions.list() {
1187 let Some(action) = q.chosen_action().filter(|a| a.maintenance()) else {
1188 continue;
1189 };
1190 if asker_may_still_read(questions, &q, home) || !ask::valid_asset_name(&q.id) {
1193 continue;
1194 }
1195 let dir = home.join("choice-actions");
1196 if std::fs::create_dir_all(&dir).is_err() {
1197 continue;
1198 }
1199 let Ok(lock) = std::fs::OpenOptions::new()
1200 .create(true)
1201 .truncate(false)
1202 .write(true)
1203 .open(dir.join(format!("{}.lock", q.id)))
1204 else {
1205 continue;
1206 };
1207 if crate::pr_head::try_lock_file(&lock).is_err() {
1208 continue;
1209 }
1210 let fresh = !home
1211 .join("choice-actions")
1212 .join(format!("{}.json", q.id))
1213 .exists();
1214 let result = async {
1215 let receipt = dir.join(format!("{}.json", q.id));
1216 if receipt.exists() {
1217 let prior: serde_json::Value = serde_json::from_slice(&std::fs::read(&receipt)?)?;
1218 anyhow::ensure!(
1219 prior.get("status").and_then(|s| s.as_str()) == Some("applied"),
1220 "previous maintenance attempt did not confirm completion; inspect its receipt"
1221 );
1222 return Ok::<_, anyhow::Error>(());
1223 }
1224 let record = |status: &str| -> anyhow::Result<()> {
1227 let tmp = receipt.with_extension("tmp");
1228 std::fs::write(
1229 &tmp,
1230 serde_json::to_vec(&serde_json::json!({
1231 "action": action, "status": status,
1232 }))?,
1233 )?;
1234 std::fs::rename(tmp, &receipt)?;
1235 Ok(())
1236 };
1237 let applied = async {
1238 match action {
1239 ask::ChoiceAction::ResumeRun { run } => {
1240 crate::resume::validate(home, run)?;
1241 record("started")?;
1242 if !crate::resume::pin_queued(queue, run, Some(q.asked_at), Some(&q.id))? {
1243 crate::resume::validate_unqueued(home, run)?;
1244 crate::resume::start(run.clone());
1245 }
1246 }
1247 ask::ChoiceAction::ReleaseTask {
1248 task,
1249 held_at,
1250 reason,
1251 } => {
1252 anyhow::ensure!(
1253 queue.resolve_id(task)? == *task,
1254 "task action requires an exact task ID"
1255 );
1256 let _claim = queue.claim(task)?;
1257 let mut task = queue.get(task)?;
1258 if task.action_applied(&q.id) {
1259 return Ok(());
1260 }
1261 if q.node == crate::conduct::NODE
1266 && q.run == task.id
1267 && task.status == TaskStatus::Blocked
1268 && task.blocked_from == Some(TaskStatus::Held)
1269 && task.blocked_by.as_slice() == [q.id.as_str()]
1270 {
1271 task.unblock(&q.id);
1272 }
1273 anyhow::ensure!(
1274 task.status == TaskStatus::Held
1275 && task.held_at == Some(*held_at)
1276 && task.hold_reason.as_deref() == reason.as_deref()
1277 && task.operator_decided_at.is_none_or(|at| at <= q.asked_at),
1278 "task no longer has the hold this question names"
1279 );
1280 task.release_operator("choice", Some(&q.id));
1281 task.mark_action_applied(&q.id);
1282 queue.put(&mut task)?;
1283 }
1284 ask::ChoiceAction::AbandonRun { run } => {
1285 record("started")?;
1286 crate::driver::abandon_proven_dead(home, run)?;
1287 }
1288 ask::ChoiceAction::RerunChecks {
1289 repo,
1290 pr,
1291 head,
1292 runs,
1293 } => {
1294 rerun_checks_once(home, &q.id, repo, pr, head, runs, || record("started"))
1295 .await?;
1296 }
1297 _ => unreachable!(),
1298 }
1299 Ok::<_, anyhow::Error>(())
1300 }
1301 .await;
1302 match applied {
1303 Ok(()) => record("applied"),
1304 Err(e) => {
1305 record(&format!("refused or failed: {e:#}"))?;
1306 Err(e)
1307 }
1308 }
1309 }
1310 .await;
1311 if home
1316 .join("choice-actions")
1317 .join(format!("{}.json", q.id))
1318 .exists()
1319 && let Some(listed) = task_of_question(&queue.list(), &q)
1320 && !listed.action_applied(&q.id)
1321 && let Ok(_claim) = queue.claim(&listed.id)
1322 && let Ok(mut origin) = queue.get(&listed.id)
1323 {
1324 if !origin.operator_held()
1325 && origin.operator_decided_at.is_none_or(|at| at <= q.asked_at)
1326 && !matches!(origin.status, TaskStatus::Running | TaskStatus::Done)
1327 {
1328 if result.is_err() || matches!(action, ask::ChoiceAction::ResumeRun { .. }) {
1329 origin.hold_machine(Some(format!(
1330 "Maintenance choice {} settled; inspect its target before releasing this task",
1331 q.id
1332 )));
1333 } else {
1334 origin.unblock(&q.id);
1335 }
1336 }
1337 origin.mark_action_applied(&q.id);
1338 let _ = queue.put(&mut origin);
1339 }
1340 if let Err(e) = result
1341 && fresh
1342 {
1343 notices::raise_in(
1344 home,
1345 Notice::warn(
1346 &format!("action:{}", q.id),
1347 format!("Choice action {} was not completed: {e:#}", q.id),
1348 ),
1349 );
1350 }
1351 }
1352}
1353
1354pub(crate) async fn rerun_checks_once(
1361 home: &Path,
1362 request: &str,
1363 repo: &Path,
1364 pr: &str,
1365 head: &str,
1366 runs: &[String],
1367 started: impl FnOnce() -> anyhow::Result<()>,
1368) -> anyhow::Result<()> {
1369 use crate::release_watch::{GhForge, ReleaseForge};
1370 anyhow::ensure!(
1371 !runs.is_empty()
1372 && runs.len() <= 20
1373 && runs
1374 .iter()
1375 .all(|r| !r.is_empty() && r.bytes().all(|b| b.is_ascii_digit()))
1376 && !pr.starts_with('-'),
1377 "invalid rerun targets"
1378 );
1379 let snap = GhForge.snapshot(repo, pr).await?;
1380 anyhow::ensure!(
1381 snap.head == head && snap.state == crate::land::PrLifecycle::Open,
1382 "pull request head or lifecycle changed"
1383 );
1384 anyhow::ensure!(
1385 runs.iter().all(|r| snap
1386 .checks
1387 .iter()
1388 .any(|c| c.run.as_ref() == Some(r) && crate::release_watch::is_failed(c.verdict))),
1389 "named workflow is no longer failed on this pull request"
1390 );
1391 started()?;
1392 let dir = home.join("choice-actions");
1393 std::fs::create_dir_all(&dir)?;
1394 let rerun_lock = std::fs::OpenOptions::new()
1395 .create(true)
1396 .truncate(false)
1397 .write(true)
1398 .open(dir.join("reruns.lock"))?;
1399 crate::pr_head::try_lock_file(&rerun_lock)?;
1400 let ledger = dir.join("reruns.json");
1401 let mut recorded: std::collections::BTreeSet<String> = if ledger.exists() {
1402 serde_json::from_slice(&std::fs::read(&ledger)?)?
1403 } else {
1404 Default::default()
1405 };
1406 for run in runs {
1407 let key = serde_json::to_string(&(request, repo, pr, head, run))?;
1408 if !recorded.insert(key) {
1409 continue;
1410 }
1411 let tmp = ledger.with_extension("tmp");
1412 std::fs::write(&tmp, serde_json::to_vec(&recorded)?)?;
1413 std::fs::rename(tmp, &ledger)?;
1414 GhForge.rerun(repo, run).await?;
1415 }
1416 Ok(())
1417}
1418
1419pub(crate) fn apply_task_choice_actions(queue: &Queue, questions: &Questions, home: &Path) {
1428 let tasks = queue.list();
1429 for q in questions.list() {
1430 if q.resolution().as_deref() == Some(crate::merge_decision::DISMISS) {
1432 continue;
1433 }
1434 if q.chosen_action().is_none_or(|a| a.maintenance()) {
1435 continue;
1436 }
1437 let Some(listed) = task_of_question(&tasks, &q) else {
1438 if q.node == land::APPROVAL_NODE
1439 && q.resolution().as_deref().is_some_and(|choice| {
1440 choice == land::REPAIR || crate::merge_decision::stops(choice)
1441 })
1442 {
1443 notices::raise_in(
1444 home,
1445 Notice::warn(
1446 &format!("repair-owner:{}", q.id),
1447 format!(
1448 "Approval action is waiting: no task owns run {}. Inspect `magi show {}` and restore its original queue owner; do not file a competing task.",
1449 q.run, q.run
1450 ),
1451 ),
1452 );
1453 }
1454 continue;
1455 };
1456 if listed.action_applied(&q.id) {
1457 continue;
1458 }
1459 let Ok(_claim) = queue.claim(&listed.id) else {
1460 continue;
1461 };
1462 let Ok(mut task) = queue.get(&listed.id) else {
1463 continue;
1464 };
1465 if q.node == crate::conduct::NODE && task.status == TaskStatus::Blocked {
1468 let blockers: Vec<_> = task
1469 .blocked_by
1470 .iter()
1471 .filter(|id| {
1472 crate::queue::dependency(queue, questions, id)
1473 == crate::queue::Dependency::Question(q.id.clone())
1474 })
1475 .cloned()
1476 .collect();
1477 if !blockers.is_empty() {
1478 for id in blockers {
1479 task.unblock(&id);
1480 }
1481 let answer = match &q.answer {
1482 Some(ask::Answer::Choice(c) | ask::Answer::Text(c)) => c.clone(),
1483 None => String::new(),
1484 };
1485 task.record_answer(q.summary.clone(), answer);
1486 }
1487 }
1488 let language = language_of(&task, Path::new("."));
1489 let decision = decide_action(&task, &q, phrases(&language), |id| {
1490 RunState::load_under(id, home)
1491 });
1492 let ran = matches!(
1493 decision,
1494 ActionDecision::Resume(_)
1495 | ActionDecision::Requeue
1496 | ActionDecision::Hold
1497 | ActionDecision::Done
1498 );
1499 if ran && asker_may_still_read(questions, &q, home) {
1500 continue;
1509 }
1510 let done = matches!(decision, ActionDecision::Done);
1511 match decision {
1512 ActionDecision::Skip => continue,
1513 ActionDecision::Resume(run) => {
1514 task.release_operator("choice", Some(&q.id));
1515 task.resume_override = Some(crate::queue::OperatorResume {
1516 question_id: q.id.clone(),
1517 at: Timestamp::now(),
1518 conductor_rehold: None,
1519 forced: true,
1520 pinned_run: Some(run),
1521 });
1522 }
1523 ActionDecision::Stale => {}
1524 ActionDecision::Requeue => {
1525 task.release_operator("choice-requeue", Some(&q.id));
1526 task.requeue();
1527 }
1528 ActionDecision::Hold => task.hold_manual(Some(q.summary.clone())),
1529 ActionDecision::Done => task.succeed(),
1530 ActionDecision::Refuse(why) => {
1531 if task.operator_held() {
1532 task.hold_reason = Some(match task.hold_reason.take() {
1535 Some(reason) => format!("{reason}\n\n{why}"),
1536 None => why,
1537 });
1538 } else {
1539 task.notice_cause(&format!("resume-refused:{}", q.id), None, None);
1540 task.hold_machine(Some(why));
1541 }
1542 notices::raise(
1543 Notice::warn(
1544 &format!("action:{}", q.id),
1545 "An answer asked the daemon to resume a run that cannot be resumed; the task stays held.",
1546 )
1547 .link(Link::Task {
1548 id: task.id.clone(),
1549 }),
1550 );
1551 }
1552 }
1553 task.mark_action_applied(&q.id);
1554 record(queue, &mut task);
1557 if done && queue.get(&task.id).is_ok_and(|t| t.action_applied(&q.id)) {
1558 supersede_prior_runs(&task, home);
1559 }
1560 let _ = questions.update(&q.id, |r| {
1563 r.answer_delivered = true;
1564 Ok(())
1565 });
1566 }
1567}
1568
1569fn asker_may_still_read(questions: &Questions, q: &ask::Question, home: &Path) -> bool {
1573 let now = Timestamp::now();
1574 if questions.read_lease(&q.id).is_some_and(|l| l.fresh(now)) {
1575 return true;
1576 }
1577 q.node != crate::conduct::NODE
1578 && RunState::load_under(&q.run, home).is_ok_and(|s| {
1579 s.seats_active()
1580 .any(|(k, a)| *k == q.seat && a.remaining_secs(now) > 0)
1581 })
1582}
1583
1584fn reconcile_task_questions(queue: &Queue, questions: &Questions) {
1595 let tasks = queue.list();
1596 let by_id: std::collections::BTreeMap<&str, &Task> =
1597 tasks.iter().map(|t| (t.id.as_str(), t)).collect();
1598 let referenced: Vec<&str> = tasks
1599 .iter()
1600 .flat_map(|task| task.blocked_by.iter().map(String::as_str))
1601 .collect();
1602
1603 for mut question in questions.list() {
1604 if !question.status.open() || question.node != crate::conduct::NODE {
1605 continue;
1606 }
1607 if referenced
1612 .iter()
1613 .any(|id| question.id.starts_with(id) || question.id.ends_with(id))
1614 {
1615 continue;
1616 }
1617 let Some(task) = by_id.get(question.run.as_str()) else {
1618 continue;
1619 };
1620 question.abandon(format!(
1621 "task {} no longer waits for this answer",
1622 task.short()
1623 ));
1624 if let Err(e) = questions.put(&mut question) {
1625 tracing::warn!(
1626 "could not retire question {} for task {}: {e:#}",
1627 question.short(),
1628 task.short()
1629 );
1630 }
1631 }
1632}
1633
1634#[derive(Debug, Clone, Copy)]
1641pub struct Verdict {
1642 pub status: RunStatus,
1644 pub left_pr: bool,
1646 pub quota_hit: bool,
1648 pub parked: bool,
1650 pub no_viable_candidates: bool,
1657}
1658
1659pub fn settle(task: &mut Task, verdict: Verdict, detail: &str, max_attempts: usize) {
1716 settle_in(task, verdict, detail, max_attempts, &PHRASES_EN)
1717}
1718
1719fn settle_in(task: &mut Task, verdict: Verdict, detail: &str, max_attempts: usize, p: &Phrases) {
1721 if verdict.parked {
1727 task.park(detail);
1728 return;
1729 }
1730 match verdict.status {
1731 RunStatus::Merged | RunStatus::Ready
1732 if task.pr_recovery.as_ref().is_some_and(|s| {
1733 s.disposition != Some(crate::pr_recovery::Disposition::Review)
1734 }) =>
1735 {
1736 task.hold_manual(Some("Public delivery is not confirmed. Inspect the selected PR and run, then use recover-pr --disposition land with the intended source and validation configuration.".into()));
1737 }
1738 RunStatus::Merged | RunStatus::Ready => task.succeed(),
1739 RunStatus::AlreadyInBase => task.already_landed(detail),
1740 RunStatus::Stalled if verdict.quota_hit => task.stall(detail),
1741 RunStatus::Failed if verdict.quota_hit && verdict.no_viable_candidates => {
1742 task.stall(detail)
1743 }
1744 RunStatus::Stalled | RunStatus::Failed => task.fail(detail, max_attempts),
1745 RunStatus::Blocked if verdict.left_pr => task.handed_off(detail),
1746 RunStatus::Blocked => task.fail(detail, max_attempts),
1747 RunStatus::VerifiedNoop => task.handed_off(detail),
1748 other => task.fail((p.graph_stopped)(label(other), detail), max_attempts),
1749 }
1750}
1751
1752pub fn supersede_prior_runs(task: &Task, home: &Path) {
1801 let now = Timestamp::now();
1802 let last_run_succeeded = task
1809 .runs
1810 .last()
1811 .and_then(|id| RunState::load_under(id, home).ok())
1812 .is_some_and(|s| matches!(s.status, RunStatus::Merged | RunStatus::Ready));
1813 for id in task.superseded_attempts(last_run_succeeded) {
1814 let mut state = match RunState::load_under(id, home) {
1815 Ok(s) => s,
1816 Err(e) => {
1817 tracing::warn!("could not load run {id} to mark it superseded: {e:#}");
1818 continue;
1819 }
1820 };
1821 if !matches!(state.status, RunStatus::Blocked | RunStatus::Stalled) {
1822 continue;
1823 }
1824 let daemon_claims = is_working_on(home, id, now);
1825 if state.liveness(daemon_claims) == Liveness::Live {
1826 continue;
1827 }
1828 state.status = RunStatus::Superseded;
1829 if let Err(e) = state.save_under(home) {
1830 tracing::warn!("could not mark run {id} superseded: {e:#}");
1831 }
1832 }
1833}
1834
1835fn resweep_superseded_attempts(queue: &Queue, home: &Path) {
1857 for task in queue.list() {
1858 if task.status != TaskStatus::Done || task.runs.len() < 2 {
1859 continue;
1860 }
1861 supersede_prior_runs(&task, home);
1862 }
1863}
1864
1865fn settle_and_diagnose(
1872 task: &mut Task,
1873 verdict: Verdict,
1874 detail: &str,
1875 max_attempts: usize,
1876 state: &RunState,
1877) {
1878 let p = phrases(&state.config.graph.language);
1879 if let Some(d) = &state.merge_decision {
1882 use crate::merge_decision as md;
1883 if task.runs.last().is_some_and(|id| id != &state.id) {
1884 return;
1885 }
1886 if (d.choice == md::WITHDRAW && d.closed) || d.choice == md::DISMISS {
1887 task.succeed();
1888 } else if let Some(agent) = md::selected_agent(&d.choice).filter(|_| d.closed) {
1889 if state.liveness(false) != Liveness::Dead {
1890 task.hold_manual(Some("Previous driver is still live; replacement held until it exits. Release the task after checking the driver.".into()));
1891 return;
1892 }
1893 let replacement = md::Replacement {
1894 question: d.question.clone(),
1895 run: state.id.clone(),
1896 pr: d.target.pr.clone(),
1897 agent: agent.to_owned(),
1898 context: format!(
1899 "Previous PR: {}. Previous run: {}. Owner requested a new implementation.\nReview evidence:\n{}",
1900 d.target.pr,
1901 state.id,
1902 serde_json::to_string(&state.reviews).unwrap_or_default()
1903 ),
1904 };
1905 task.requeue();
1906 task.pr_recovery = None;
1907 task.review_of = None;
1908 task.replacement = Some(replacement);
1909 } else {
1910 task.hold_manual(Some(if d.choice.eq_ignore_ascii_case(land::HOLD) {
1911 format!("Decision held by owner; no further repair or merge work. {}", state.merge.as_ref().map_or("", |m| m.detail.as_str()))
1912 } else { format!("PR closure was not confirmed. Task held; inspect the existing run before starting replacement work. {}", state.merge.as_ref().map_or("", |m| m.detail.as_str())) }));
1913 }
1914 task.mark_action_applied(&d.question);
1915 if let Some(pass) = state.passes.last() {
1916 task.record_pass(&state.id, pass.index, task.attempts, None);
1917 }
1918 return;
1919 }
1920 if let Some(replacement) = &task.replacement
1921 && !state.parked
1922 && state.resource_block.is_none()
1923 && (state.viable().is_empty()
1924 || (state.status == RunStatus::Blocked && state.tally.is_none()))
1925 {
1926 task.hold_manual(Some(format!("Selected implementer `{}` was unavailable, failed or hit quota before producing a viable implementation. No other implementer was dispatched. Inspect the run, restore availability, then explicitly release this task.", replacement.agent)));
1927 return;
1928 }
1929 let attempts_before = task.attempts;
1930 if let Some(block) = state.resource_block.as_ref().filter(|_| !verdict.parked) {
1936 task.notice_cause(
1937 &format!("resource:{}:{}", block.kind, block.node),
1938 None,
1939 None,
1940 );
1941 task.last_error = Some(detail.to_owned());
1942 task.diagnostic = None;
1943 task.attempts = task.attempts.saturating_sub(1);
1944 if block.kind == "disk pressure" {
1945 task.hold_machine(Some(format!(
1946 "{}{}",
1947 crate::disk::RUNNING_PRESSURE,
1948 block.detail
1949 )));
1950 } else {
1951 task.hold_machine(Some(format!(
1952 "{}{} stayed busy past the resource-wait limit before `{}` could start; \
1953 no agent was asked and no attempt was spent",
1954 crate::cache::RESOURCE_HOLD,
1955 block.kind,
1956 block.node
1957 )));
1958 }
1959 if let Some(pass) = state.passes.last() {
1960 task.record_pass(&state.id, pass.index, attempts_before, None);
1961 }
1962 return;
1963 }
1964 let unresolved = task.hold_notice_cause.clone();
1965 if !verdict.parked
1966 && task
1967 .pr_recovery
1968 .as_ref()
1969 .is_some_and(|s| s.disposition != Some(crate::pr_recovery::Disposition::Review))
1970 && matches!(
1971 verdict.status,
1972 RunStatus::Ready | RunStatus::Merged | RunStatus::AlreadyInBase
1973 )
1974 {
1975 if state.pr_recovery == task.pr_recovery && state.public_delivery_complete() {
1976 task.succeed();
1977 } else {
1978 task.hold_manual(Some("Selected PR delivery is incomplete: local review or publication is not a confirmed public merge. Inspect the run and PR; use recover-pr --disposition land with --source-head <reviewed-full-SHA> and the required verification configuration. No automatic new attempt will start.".into()));
1979 }
1980 } else if !verdict.parked
1981 && task.pr_recovery.is_some()
1982 && matches!(
1983 verdict.status,
1984 RunStatus::Blocked | RunStatus::Failed | RunStatus::Gating
1985 )
1986 && !verdict.quota_hit
1987 {
1988 task.hold_unrequested_retry(Some(format!("Selected recovery stopped: {detail}. Inspect the retained source and run; correct the cause before explicitly releasing or selecting again. No automatic replacement will start.")));
1989 } else {
1990 settle_in(task, verdict, detail, max_attempts, p);
1991 }
1992 if !verdict.parked
1993 && !verdict.quota_hit
1994 && matches!(
1995 verdict.status,
1996 RunStatus::Blocked | RunStatus::Failed | RunStatus::Stalled
1997 )
1998 {
1999 task.recovery_failed(state);
2000 }
2001 if state.prep_failure.is_some() {
2002 task.hold_notice_cause = unresolved;
2005 task.startup_notice_cause("startup-candidates");
2006 }
2007 if let Some(pass) = state.passes.last() {
2010 let parked = match &pass.end {
2011 Some(crate::run::PassEnd::Parked { cause }) => Some(cause.clone()),
2012 _ => None,
2013 };
2014 task.record_pass(&state.id, pass.index, attempts_before, parked);
2015 }
2016 if task.status == TaskStatus::Held {
2017 if state.prep_failure.is_none() {
2018 task.notice_cause(
2019 label(verdict.status),
2020 Some(&state.id),
2021 state.passes.last().map(|p| p.index),
2022 );
2023 }
2024 task.diagnostic = diagnostic(state);
2025 note_open_question(task, &state.id, p);
2026 }
2027}
2028
2029fn note_open_question(task: &mut Task, run: &str, p: &Phrases) {
2044 let Some(home) = crate::run::try_home() else {
2045 return;
2046 };
2047 let open = Questions::at(home.join("questions")).open_for(run);
2048 let Some(q) = open.first() else {
2049 return;
2050 };
2051 let mut decisions: Vec<_> = open.iter().map(|q| q.id.as_str()).collect();
2052 decisions.sort_unstable();
2053 task.notice_cause(
2054 &format!("questions:{}", decisions.join(",")),
2055 Some(run),
2056 None,
2057 );
2058 let base = task.hold_reason.clone().unwrap_or_default();
2059 task.hold_reason = Some(format!("{base}{}{}", p.waiting_for_answer, q.short()));
2060}
2061
2062fn reclaim(task: &mut Task, last_run: Option<RunState>, max_attempts: usize, language: &str) {
2073 match last_run {
2074 Some(state) => {
2075 let verdict = Verdict {
2076 status: state.status,
2077 left_pr: state.pr.is_some(),
2078 quota_hit: !state.quota.is_empty(),
2079 parked: state.parked,
2080 no_viable_candidates: state.viable().is_empty(),
2081 };
2082 let detail = format!(
2083 "{}{}",
2084 phrases(&state.config.graph.language).recovered_running,
2085 describe(&state)
2086 );
2087 settle_and_diagnose(task, verdict, &detail, max_attempts, &state);
2088 }
2089 None => {
2090 let why = phrases(language).no_run_to_recover;
2093 task.last_error = Some(why.to_owned());
2094 task.notice_cause("missing-run", None, None);
2097 task.hold_machine(Some(why.to_owned()));
2098 }
2099 }
2100}
2101
2102fn reclaim_orphaned_running(queue: &Queue, max_attempts: usize) -> Vec<String> {
2124 let mut reclaimed = Vec::new();
2125 for listed in queue.list() {
2126 if listed.status != TaskStatus::Running {
2127 continue;
2128 }
2129 let Ok(_claim) = queue.claim(&listed.id) else {
2130 continue;
2131 };
2132 let Ok(mut task) = queue.get(&listed.id) else {
2136 continue;
2137 };
2138 if task.status != TaskStatus::Running {
2139 continue;
2140 }
2141 let last_run = task.runs.last().and_then(|id| RunState::load(id).ok());
2142 if let Some(state) = &last_run
2152 && let Err(e) = ask::Questions::open().settle_run(&state.id, state.status)
2153 {
2154 tracing::warn!("abandon questions for {}: {e:#}", state.id);
2155 }
2156 let language = if last_run.is_none() {
2157 language_of(&task, Path::new("."))
2158 } else {
2159 String::new()
2160 };
2161 reclaim(&mut task, last_run, max_attempts, &language);
2162 if task.status == TaskStatus::Done {
2163 supersede_prior_runs(&task, &crate::run::home());
2164 }
2165 record(queue, &mut task);
2166 reclaimed.push(task.id.clone());
2167 }
2168 reclaimed
2169}
2170
2171fn reclaim_abandoned_runs(home: &Path, now: Timestamp) -> Vec<String> {
2203 reclaim_abandoned_runs_with(
2204 home,
2205 now,
2206 crate::proc::pid_status,
2207 crate::proc::process_started_at,
2208 )
2209}
2210
2211fn reclaim_abandoned_runs_with<F, G>(
2217 home: &Path,
2218 now: Timestamp,
2219 query: F,
2220 identity: G,
2221) -> Vec<String>
2222where
2223 F: Fn(u32) -> Option<bool>,
2224 G: Fn(u32) -> Option<String>,
2225{
2226 let mut abandoned = Vec::new();
2227 for entry in std::fs::read_dir(home.join("runs"))
2228 .into_iter()
2229 .flatten()
2230 .flatten()
2231 {
2232 let id = entry.file_name().to_string_lossy().into_owned();
2233 if !crate::run::is_run_id(&id) {
2234 continue;
2235 }
2236 let Ok(_owner) = crate::pr_head::owner_lock(home, &id) else {
2239 continue;
2240 };
2241 let Ok(body) = std::fs::read_to_string(entry.path().join("run.json")) else {
2249 continue;
2250 };
2251 let Ok(mut state) = serde_json::from_str::<RunState>(&body) else {
2252 continue;
2253 };
2254 if state.driver_exited || state.status.done() || !state.active_all_overrun(now) {
2255 continue;
2256 }
2257 let daemon_claims = is_working_on(home, &id, now);
2269 if state.liveness_under_with(home, daemon_claims, &query, &identity)
2270 != crate::run::Liveness::Dead
2271 {
2272 continue;
2273 }
2274 state.abandon("daemon");
2275 if let Err(e) = state.save_under(home) {
2276 tracing::warn!("could not persist abandoned run {id}: {e:#}");
2277 continue;
2278 }
2279 if let Some(notice) = notices::run_ended(&state) {
2282 notices::raise_in(home, notice);
2283 }
2284 if let Err(e) = Questions::at(home.join("questions")).settle_run(&id, state.status) {
2292 tracing::warn!("abandon questions for {id}: {e:#}");
2293 }
2294 abandoned.push(id);
2295 }
2296 abandoned
2297}
2298
2299pub async fn serve(opts: Opts) -> Result<()> {
2305 serve_until(opts, Stop::new()).await
2306}
2307
2308pub async fn serve_until(opts: Opts, stop: Stop) -> Result<()> {
2325 let signal = {
2326 let stop = stop.clone();
2327 tokio::spawn(async move {
2328 if tokio::signal::ctrl_c().await.is_ok() {
2329 stop.stop();
2330 tracing::info!("shutdown requested; a run in flight will be finished first");
2331 }
2332 })
2333 };
2334
2335 let worktrees_root = opts
2336 .worktrees_root
2337 .clone()
2338 .unwrap_or_else(crate::run::default_worktree_root);
2339 let outcome = drive(
2340 &opts,
2341 &Queue::open(),
2342 &status_path(),
2343 &crate::run::home(),
2344 &worktrees_root,
2345 &stop,
2346 )
2347 .await;
2348
2349 signal.abort();
2350 outcome
2351}
2352
2353async fn drive(
2366 opts: &Opts,
2367 queue: &Queue,
2368 status_file: &Path,
2369 home: &Path,
2370 worktrees_root: &Path,
2371 stop: &Stop,
2372) -> Result<()> {
2373 let recovery_cfg = prepare(&opts.repo, opts).ok();
2381 let daemon_cfg = recovery_cfg
2382 .as_ref()
2383 .map(|c| c.daemon.clone())
2384 .unwrap_or_default();
2385 let reconciled = crate::driver::reconcile(home);
2386 if !reconciled.is_empty() {
2387 tracing::info!(runs = ?reconciled, "reconciled stopped run drivers at startup");
2388 }
2389 let mut initial_status = Status::new();
2390 initial_status.question_recovery = Some(crate::question_delivery::Recovery::new(
2391 recovery_cfg.as_ref(),
2392 daemon_cfg.max_deputies,
2393 ));
2394 let status = Arc::new(Mutex::new(initial_status));
2395 write_status_to(status_file, &lock(&status)).context("publish the daemon status file")?;
2396 let beat = tokio::spawn(heartbeat(Arc::clone(&status), status_file.to_path_buf()));
2397
2398 let concurrency = max_concurrent(daemon_cfg.max_concurrent_runs);
2404
2405 let waiter = tokio::spawn(crate::waiter::run(
2410 crate::waiter::Waiter::new(
2411 crate::ask::Questions::at(home.join("questions")),
2412 home.to_path_buf(),
2413 recovery_cfg.clone(),
2414 ),
2415 stop.clone(),
2416 ));
2417
2418 let deputies = tokio::spawn(crate::deputy::run(
2422 crate::deputy::Deputies::new(
2423 crate::ask::Questions::at(home.join("questions")),
2424 home.to_path_buf(),
2425 recovery_cfg.clone(),
2426 opts.repo.clone(),
2427 daemon_cfg.max_deputies,
2428 {
2429 let stop = stop.clone();
2430 Arc::new(move || stop.parking())
2431 },
2432 ),
2433 stop.clone(),
2434 ));
2435
2436 let fetcher = tokio::spawn(fetch_loop(opts.repo.clone(), opts.clone(), stop.clone()));
2439
2440 let release_watch = tokio::spawn(crate::release_watch::run(
2443 crate::release_watch::Watcher::new(
2444 Box::new(crate::release_watch::GhForge),
2445 home.to_path_buf(),
2446 ),
2447 {
2448 let (repo, opts) = (opts.repo.clone(), opts.clone());
2449 move || release_watch_settings(&repo, &opts)
2450 },
2451 stop.clone(),
2452 ));
2453
2454 tracing::info!(
2455 "magi serve: queue {} (poll {}s, {} attempts per task, {} run(s) at once{})",
2456 queue.root().display(),
2457 opts.poll.as_secs(),
2458 opts.max_attempts,
2459 concurrency,
2460 if daemon_cfg.pause_for_interrupts {
2461 ", interrupts enabled"
2462 } else {
2463 ""
2464 }
2465 );
2466
2467 janitor(&opts.repo, opts, home, worktrees_root).await;
2470 resweep_superseded_attempts(queue, home);
2471
2472 let outcome = poll(
2473 opts,
2474 queue,
2475 &status,
2476 home,
2477 worktrees_root,
2478 stop,
2479 DispatchLimits {
2480 max_concurrent: concurrency,
2481 pause_for_interrupts: daemon_cfg.pause_for_interrupts,
2482 },
2483 )
2484 .await;
2485
2486 beat.abort();
2487 waiter.abort();
2488 deputies.abort();
2489 fetcher.abort();
2490 release_watch.abort();
2491 clear_status_at(status_file);
2492 outcome
2493}
2494
2495fn release_watch_settings(repo: &Path, opts: &Opts) -> (Vec<PathBuf>, u64) {
2498 let Ok(cfg) = prepare(repo, opts) else {
2499 return (Vec::new(), 0);
2500 };
2501 let mut paths: Vec<PathBuf> = crate::repos::scan(&cfg.repos.roots)
2502 .into_iter()
2503 .map(|r| r.path)
2504 .collect();
2505 let mut known: Vec<PathBuf> = Queue::open().list().into_iter().map(|t| t.repo).collect();
2508 let runs = crate::run::home().join("runs");
2509 for id in crate::run::list_ids_in(&runs) {
2510 let repo = std::fs::read_to_string(runs.join(&id).join("run.json"))
2511 .ok()
2512 .and_then(|s| serde_json::from_str::<serde_json::Value>(&s).ok())
2513 .and_then(|v| v.get("repo")?.as_str().map(PathBuf::from));
2514 known.extend(repo);
2515 }
2516 for k in known {
2517 let k = k.canonicalize().unwrap_or(k);
2518 if k.join(".git").exists() && !paths.contains(&k) {
2519 paths.push(k);
2520 }
2521 }
2522 let own = repo.canonicalize().unwrap_or_else(|_| repo.to_path_buf());
2523 if !paths.contains(&own) {
2524 paths.push(own);
2525 }
2526 (paths, cfg.daemon.release_stall_minutes)
2527}
2528
2529const FETCH_TIMEOUT: Duration = Duration::from_secs(30);
2531
2532async fn fetch_loop(repo: PathBuf, opts: Opts, stop: Stop) {
2536 while !stop.stopped() {
2537 let (interval, roots) = match prepare(&repo, &opts) {
2538 Ok(c) => (c.repos.fetch_interval, c.repos.roots),
2539 Err(_) => (0, Vec::new()),
2540 };
2541 if interval > 0 && !roots.is_empty() {
2542 let r = crate::clean::fetch_origins(&roots, FETCH_TIMEOUT, || stop.stopped()).await;
2543 tracing::debug!("fetch origins: {r:?}");
2544 }
2545 let wait = if interval > 0 { interval } else { 60 };
2547 let mut slept = 0;
2548 while slept < wait && !stop.stopped() {
2549 tokio::time::sleep(Duration::from_secs(1)).await;
2550 slept += 1;
2551 }
2552 }
2553}
2554
2555async fn heartbeat(status: Arc<Mutex<Status>>, path: PathBuf) {
2561 loop {
2562 tokio::time::sleep(HEARTBEAT).await;
2563 let snapshot = {
2564 let mut guard = lock(&status);
2565 guard.updated_at = Timestamp::now();
2566 guard.clone()
2567 };
2568 if let Err(e) = write_status_to(&path, &snapshot) {
2569 tracing::warn!("could not refresh the daemon status file: {e:#}");
2572 }
2573 }
2574}
2575
2576#[derive(Debug, Clone, Copy, PartialEq, Eq)]
2579enum LandResume {
2580 NotLanding,
2583 StillWaiting,
2588 Ready,
2592}
2593
2594fn land_resume_state(task: &Task) -> LandResume {
2598 let Some(run_id) = task.runs.last() else {
2599 return LandResume::NotLanding;
2600 };
2601 let Ok(state) = RunState::load(run_id) else {
2602 return LandResume::NotLanding;
2603 };
2604 if !state.parked {
2605 return LandResume::NotLanding;
2606 }
2607 if state.status == RunStatus::Landing
2608 && state.land_approval.as_ref().is_some_and(|binding| {
2609 ask::Questions::open()
2610 .get(&binding.question)
2611 .is_ok_and(|q| {
2612 q.run == state.id
2613 && q.resolution()
2614 .as_deref()
2615 .is_some_and(crate::merge_decision::stops)
2616 })
2617 })
2618 {
2619 return LandResume::Ready;
2620 }
2621 if !implementers_match(task, &state) {
2622 return LandResume::NotLanding;
2623 }
2624 let node = if state.status == RunStatus::Landing {
2628 land::APPROVAL_NODE
2629 } else if state.github_text.as_ref().is_some_and(|g| !g.resolved) {
2630 crate::github_text::ASK_NODE
2631 } else {
2632 return LandResume::NotLanding;
2633 };
2634 let store = ask::Questions::open();
2635 let waiting = store
2636 .list()
2637 .into_iter()
2638 .filter(|q| &q.run == run_id && q.node == node)
2639 .max_by(|a, b| a.id.cmp(&b.id));
2640 let Some(mut q) = waiting else {
2641 return LandResume::Ready;
2642 };
2643 if !q.status.open() {
2644 if q.resolution().as_deref() == Some(land::REPAIR) {
2647 return if state.liveness(false) == Liveness::Live {
2648 LandResume::StillWaiting
2649 } else {
2650 LandResume::NotLanding
2651 };
2652 }
2653 return LandResume::Ready;
2654 }
2655 let timeout = Duration::from_secs(state.config.graph.answer_timeout);
2662 let elapsed = Timestamp::now().as_second() - q.asked_at.as_second();
2663 if elapsed >= 0 && elapsed as u64 >= timeout.as_secs() {
2664 q.abandon(format!(
2665 "no answer within {}s of asking",
2666 timeout.as_secs().max(1)
2667 ));
2668 if store.put(&mut q).is_ok() {
2671 return LandResume::Ready;
2672 }
2673 }
2674 LandResume::StillWaiting
2675}
2676
2677const RECHECK_WHILE_BUSY: Duration = Duration::from_millis(200);
2685
2686const CACHE_CHECK_INTERVAL_SECS: u64 = 5 * 60;
2698
2699struct InFlightGuard<'a> {
2712 status: &'a Arc<Mutex<Status>>,
2713 stop: &'a Stop,
2714 task_id: &'a str,
2715}
2716
2717impl Drop for InFlightGuard<'_> {
2718 fn drop(&mut self) {
2719 lock(self.status).current.retain(|c| c.task != self.task_id);
2720 self.stop.exit();
2721 }
2722}
2723
2724#[derive(Debug, Clone, PartialEq, Eq)]
2746enum Interrupt {
2747 Idle,
2749 Parking {
2761 parked: Vec<String>,
2762 interrupt_task: String,
2763 },
2764 Running {
2772 parked: Vec<String>,
2773 interrupt_task: String,
2774 },
2775 Resuming { parked: Vec<String> },
2782}
2783
2784fn advance_interrupt(state: Interrupt, in_flight: &[String], runnable: &[Task]) -> Interrupt {
2805 match state {
2806 Interrupt::Idle => {
2807 if in_flight.len() != 1 {
2819 return Interrupt::Idle;
2820 }
2821 match runnable.iter().find(|t| t.interrupt) {
2822 Some(t) => Interrupt::Parking {
2823 parked: in_flight.to_vec(),
2824 interrupt_task: t.id.clone(),
2825 },
2826 None => Interrupt::Idle,
2827 }
2828 }
2829 Interrupt::Parking {
2830 parked,
2831 interrupt_task,
2832 } => {
2833 if in_flight.iter().any(|id| parked.contains(id)) {
2834 Interrupt::Parking {
2836 parked,
2837 interrupt_task,
2838 }
2839 } else if in_flight.contains(&interrupt_task) {
2840 Interrupt::Running {
2841 parked,
2842 interrupt_task,
2843 }
2844 } else if runnable.iter().any(|t| t.id == interrupt_task) {
2845 Interrupt::Parking {
2849 parked,
2850 interrupt_task,
2851 }
2852 } else {
2853 Interrupt::Resuming { parked }
2858 }
2859 }
2860 Interrupt::Running {
2861 parked,
2862 interrupt_task,
2863 } => {
2864 if in_flight.contains(&interrupt_task) {
2865 Interrupt::Running {
2866 parked,
2867 interrupt_task,
2868 }
2869 } else {
2870 Interrupt::Resuming { parked }
2876 }
2877 }
2878 Interrupt::Resuming { parked } => {
2879 if in_flight.iter().any(|id| parked.contains(id)) {
2880 Interrupt::Idle
2886 } else if runnable.iter().any(|t| parked.contains(&t.id)) {
2887 Interrupt::Resuming { parked }
2888 } else {
2889 Interrupt::Idle
2892 }
2893 }
2894 }
2895}
2896
2897fn advance_interrupt_tick(
2903 enabled: bool,
2904 state: Interrupt,
2905 in_flight: &[String],
2906 runnable: &[Task],
2907) -> Interrupt {
2908 if !enabled {
2909 return Interrupt::Idle;
2910 }
2911 advance_interrupt(state, in_flight, runnable)
2912}
2913
2914fn interrupt_gate(state: &Interrupt, in_flight: &[String], candidates: Vec<Task>) -> Vec<Task> {
2919 match state {
2920 Interrupt::Idle => candidates,
2921 Interrupt::Parking {
2922 parked,
2923 interrupt_task,
2924 } => {
2925 if in_flight.iter().any(|id| parked.contains(id)) {
2926 Vec::new()
2927 } else {
2928 candidates
2929 .into_iter()
2930 .filter(|t| &t.id == interrupt_task)
2931 .collect()
2932 }
2933 }
2934 Interrupt::Running { .. } => Vec::new(),
2935 Interrupt::Resuming { parked } => candidates
2943 .into_iter()
2944 .find(|t| parked.contains(&t.id))
2945 .into_iter()
2946 .collect(),
2947 }
2948}
2949
2950struct DispatchLimits {
2954 max_concurrent: usize,
2957 pause_for_interrupts: bool,
2959}
2960
2961#[derive(Debug, Clone, Copy, PartialEq, Eq)]
2970enum PermitKind {
2971 None,
2976 Urgent,
2983 Ordinary,
2986}
2987
2988fn permit_kind(priority: bool, urgent: bool) -> PermitKind {
2991 if priority {
2992 PermitKind::None
2993 } else if urgent {
2994 PermitKind::Urgent
2995 } else {
2996 PermitKind::Ordinary
2997 }
2998}
2999
3000struct DispatchClaim {
3001 task: Task,
3002 claim: crate::queue::Claim,
3003 permit: Option<tokio::sync::OwnedSemaphorePermit>,
3004}
3005
3006fn claim_dispatch(
3009 queue: &Queue,
3010 candidate: &Task,
3011 cooling_down: bool,
3012 ordinary: &Arc<tokio::sync::Semaphore>,
3013 urgent: &Arc<tokio::sync::Semaphore>,
3014) -> Option<DispatchClaim> {
3015 let Ok(claim) = queue.claim(&candidate.id) else {
3016 tracing::info!("task {} is claimed elsewhere; skipping", candidate.short());
3017 return None;
3018 };
3019 let task = match queue.get(&candidate.id) {
3020 Ok(t) if t.status.runnable() => t,
3021 Ok(_) => return None,
3022 Err(e) => {
3023 tracing::warn!("could not re-read task {}: {e:#}", candidate.short());
3024 return None;
3025 }
3026 };
3027 let resume = land_resume_state(&task);
3028 if resume == LandResume::StillWaiting {
3029 return None;
3030 }
3031 let priority = resume == LandResume::Ready;
3032 if !priority && cooling_down {
3033 return None;
3034 }
3035 let permit = match permit_kind(priority, task.urgent) {
3038 PermitKind::None => None,
3039 PermitKind::Urgent => Some(Arc::clone(urgent).try_acquire_owned().ok()?),
3040 PermitKind::Ordinary => Some(Arc::clone(ordinary).try_acquire_owned().ok()?),
3041 };
3042 Some(DispatchClaim {
3043 task,
3044 claim,
3045 permit,
3046 })
3047}
3048
3049async fn poll(
3068 opts: &Opts,
3069 queue: &Queue,
3070 status: &Arc<Mutex<Status>>,
3071 home: &Path,
3072 worktrees_root: &Path,
3073 stop: &Stop,
3074 limits: DispatchLimits,
3075) -> Result<()> {
3076 let DispatchLimits {
3077 max_concurrent,
3078 pause_for_interrupts,
3079 } = limits;
3080 let mut attempted: Vec<String> = Vec::new();
3085 let sem = Arc::new(tokio::sync::Semaphore::new(max_concurrent));
3086 let urgent_sem = Arc::new(tokio::sync::Semaphore::new(1));
3094 let quota_cooldown_until: Arc<Mutex<Option<Timestamp>>> = Arc::new(Mutex::new(None));
3100 let mut inflight: tokio::task::JoinSet<()> = tokio::task::JoinSet::new();
3101 let mut conductor = Conductor::new();
3102 let mut cache_last_checked: Option<Timestamp> = None;
3105 let mut disk_last_checked: Option<Timestamp> = None;
3106 let mut interrupt = Interrupt::Idle;
3108 let mut interrupt_pauses: std::collections::HashMap<String, crate::graph::Pause> =
3114 std::collections::HashMap::new();
3115
3116 while !stop.stopped() {
3117 lock(status).polls += 1;
3118
3119 while let Some(result) = inflight.try_join_next() {
3124 if let Err(e) = result {
3125 tracing::error!("a spawned attempt did not finish cleanly: {e}");
3126 notices::raise(Notice::error(
3127 "loop:attempt",
3128 "A queued attempt ended abnormally; check the task it was running.",
3129 ));
3130 }
3131 }
3132
3133 let swept = sweep_stale_claims(queue, STALE_CLAIM);
3134 if !swept.is_empty() {
3135 tracing::warn!(
3136 "swept {} stale claim(s) left behind by an earlier daemon: {}",
3137 swept.len(),
3138 swept.join(", ")
3139 );
3140 }
3141 let now = Timestamp::now();
3146
3147 if !stop.busy_now() {
3152 maybe_prune_cache_between_runs(
3153 &opts.repo,
3154 opts,
3155 home,
3156 stop,
3157 &mut cache_last_checked,
3158 now,
3159 )
3160 .await;
3161 }
3162
3163 if cache_check_due(disk_last_checked, now, 30) {
3164 disk_last_checked = Some(now);
3165 if let Ok(cfg) = prepare(&opts.repo, opts) {
3166 crate::retention::sweep(home, &cfg.disk, now).await;
3167 if let Err(e) = clean::prune_cache_if_over_limit(&cfg, home) {
3170 tracing::warn!("disk pressure cache cleanup failed: {e:#}");
3171 }
3172 for current in lock(status).current.clone() {
3173 if let Ok(state) = RunState::load_under(¤t.run, home)
3174 && let Some(reason) = disk_gate(&state.repo, &state.config)
3175 {
3176 if let Some(pause) = interrupt_pauses.get(¤t.task) {
3177 pause.park_because(reason);
3178 }
3179 notices::raise_in(
3180 home,
3181 Notice::warn(
3182 "disk:pressure",
3183 "Disk pressure persists after cleanup; active runs were asked to checkpoint and park. New starts and resumes wait for free space.",
3184 ),
3185 );
3186 }
3187 }
3188 }
3189 }
3190
3191 let stalled = stalled_tasks(queue, home, now);
3192 let stalled_ids: std::collections::BTreeSet<_> =
3193 stalled.iter().map(|task| task.id.clone()).collect();
3194 let reclaimed = reclaim_orphaned_running(queue, opts.max_attempts);
3195 if !reclaimed.is_empty() {
3196 tracing::warn!(
3197 "reclaimed {} task(s) left `running` by a daemon that never \
3198 recorded the outcome: {}",
3199 reclaimed.len(),
3200 reclaimed.join(", ")
3201 );
3202 }
3203 let abandoned_runs = reclaim_abandoned_runs(home, now);
3204 if !abandoned_runs.is_empty() {
3205 tracing::warn!(
3206 "failed {} run(s) left behind by a killed process, past every \
3207 active seat's own timeout: {}",
3208 abandoned_runs.len(),
3209 abandoned_runs.join(", ")
3210 );
3211 }
3212
3213 let questions = Questions::at(home.join("questions"));
3218
3219 repair_choice_offers(&questions);
3222 resolve_blockers(queue, &questions);
3223 apply_choice_actions(queue, &questions, home).await;
3224 triage::apply_answers(queue, &questions);
3225 reconcile_task_questions(queue, &questions);
3226 {
3230 let talks = crate::talk::Talks::at(home.join("talks"));
3231 crate::chat_notify::sweep(queue, &talks, home);
3232 crate::chat_watch::sweep(home, &talks);
3233 crate::chat_notify::start_turns(&talks, stop.parking(), &mut inflight);
3234 }
3235
3236 let finished: Vec<Task> = finished_tasks(queue)
3242 .into_iter()
3243 .filter(|task| !stalled_ids.contains(&task.id))
3244 .filter(|task| !awaiting_resume_with(task, |id| RunState::load_under(id, home)))
3248 .collect();
3249 let queued = queued_tasks(queue);
3250 if !(queued.is_empty() && stalled.is_empty() && finished.is_empty())
3254 && conductor.worth_a_look(queue, &stalled, &finished)
3255 {
3256 match prepare(&opts.repo, opts) {
3257 Ok(cfg) => {
3258 conductor
3259 .maybe_run(
3260 &cfg,
3261 &opts.repo,
3262 queue,
3263 &questions,
3264 home,
3265 &queued,
3266 &stalled,
3267 &finished,
3268 opts.max_attempts,
3269 )
3270 .await;
3271 }
3272 Err(e) => {
3273 tracing::warn!("conductor: no config: {e:#}");
3274 notices::raise(Notice::warn(
3275 "loop:no-config",
3276 "The loop could not read this repository's config, so held tasks are not being triaged.",
3277 ));
3278 }
3279 }
3280 }
3281
3282 let candidates: Vec<Task> = runnable(queue)
3283 .into_iter()
3284 .filter(|t| !opts.once || !attempted.contains(&t.id))
3285 .collect();
3286
3287 let in_flight: Vec<String> = lock(status)
3292 .current
3293 .iter()
3294 .map(|c| c.task.clone())
3295 .collect();
3296 interrupt_pauses.retain(|id, _| in_flight.contains(id));
3297
3298 interrupt =
3299 advance_interrupt_tick(pause_for_interrupts, interrupt, &in_flight, &candidates);
3300 if let Interrupt::Parking {
3301 parked,
3302 interrupt_task,
3303 } = &interrupt
3304 {
3305 let reason = format!(
3306 "task {} asked to run first",
3307 crate::run::short_of(interrupt_task)
3308 );
3309 for id in parked {
3310 if let Some(pause) = interrupt_pauses.get(id) {
3311 pause.park_because(reason.clone());
3312 }
3313 }
3314 }
3315 let candidates = interrupt_gate(&interrupt, &in_flight, candidates);
3316
3317 let cooling_down =
3318 lock("a_cooldown_until).is_some_and(|until| Timestamp::now() < until);
3319
3320 let mut started_any = false;
3321 for candidate in candidates {
3322 if stop.stopped() {
3323 break;
3324 }
3325
3326 let Some(DispatchClaim {
3327 mut task,
3328 claim,
3329 permit,
3330 }) = claim_dispatch(queue, &candidate, cooling_down, &sem, &urgent_sem)
3331 else {
3332 continue;
3333 };
3334
3335 let task_id = task.id.clone();
3336 attempted.push(task_id.clone());
3337 lock(status).idle = false;
3338 stop.enter();
3341 started_any = true;
3342
3343 let run_pause = crate::graph::Pause::new();
3347 interrupt_pauses.insert(task_id.clone(), run_pause.clone());
3348
3349 let opts = opts.clone();
3350 let queue = queue.clone();
3351 let status = Arc::clone(status);
3352 let stop = stop.clone();
3353 let quota_cooldown_until = Arc::clone("a_cooldown_until);
3354 inflight.spawn(async move {
3355 let _claim = claim;
3359 let _permit = permit;
3360 let _inflight = InFlightGuard {
3362 status: &status,
3363 stop: &stop,
3364 task_id: &task_id,
3365 };
3366 let quota = attempt(&opts, &queue, &status, &stop, run_pause, &mut task).await;
3367 lock(&status).completed += 1;
3368 let now = Timestamp::now();
3374 if let Some(until) = cooldown_until("a, now) {
3375 let wait = until.as_second() - now.as_second();
3376 *lock("a_cooldown_until) = Some(until);
3377 let hint = quota
3378 .iter()
3379 .find(|q| q.reset.is_some())
3380 .and_then(|q| q.reset.as_deref());
3381 match hint {
3382 Some(h) => tracing::warn!(
3383 "quota hit; waiting {wait}s before taking another ordinary task \
3384 (CLI reported reset: {h})"
3385 ),
3386 None => tracing::warn!(
3387 "quota hit; waiting {wait}s before taking another ordinary task \
3388 (no reset hint reported)"
3389 ),
3390 }
3391 }
3392 });
3393 }
3394
3395 if started_any {
3396 continue;
3397 }
3398
3399 if stop.busy_now() {
3400 stop.idle(RECHECK_WHILE_BUSY.min(opts.poll)).await;
3405 continue;
3406 }
3407
3408 lock(status).idle = true;
3410 if opts.once {
3411 janitor(&opts.repo, opts, home, worktrees_root).await;
3415 resweep_superseded_attempts(queue, home);
3416 triage_held(queue, home, opts).await;
3417 break;
3418 }
3419 stop.idle(opts.poll).await;
3420 if stop.stopped() {
3421 continue;
3422 }
3423 janitor(&opts.repo, opts, home, worktrees_root).await;
3429 resweep_superseded_attempts(queue, home);
3430 triage_held(queue, home, opts).await;
3431 }
3432
3433 while let Some(result) = inflight.join_next().await {
3438 if let Err(e) = result {
3439 tracing::error!("a spawned attempt did not finish cleanly: {e}");
3440 notices::raise(Notice::error(
3441 "loop:attempt",
3442 "A queued attempt ended abnormally; check the task it was running.",
3443 ));
3444 }
3445 }
3446 Ok(())
3447}
3448
3449async fn public_resume_owner(
3453 home: &Path,
3454 runs: &[String],
3455 repo: &Path,
3456) -> Result<Option<RunState>> {
3457 let mut owner = None;
3458 for id in runs.iter().collect::<std::collections::BTreeSet<_>>() {
3459 let _lock = crate::pr_head::owner_lock(home, id)?;
3460 let state = RunState::load_under(id, home)?;
3461 if state.released()
3462 || state.retired_to.is_some()
3463 || state.pr_taken_over_by.is_some()
3464 || !crate::handover::landing_checkpoint(&state)
3465 {
3466 continue;
3467 }
3468 anyhow::ensure!(
3469 crate::git::toplevel(&state.repo).await?.canonicalize()?
3470 == crate::git::toplevel(repo).await?.canonicalize()?,
3471 "public owner repository differs; inspect run lineage"
3472 );
3473 anyhow::ensure!(
3474 owner.is_none(),
3475 "multiple retained public owners; inspect run lineage"
3476 );
3477 anyhow::ensure!(
3478 crate::handover::can_resume(&state),
3479 "public owner cannot resume"
3480 );
3481 let claimed = is_working_on(home, id, Timestamp::now());
3482 anyhow::ensure!(
3483 state.liveness_under(home, claimed) == crate::run::Liveness::Dead,
3484 "public owner's driver is live or unknown; wait before retrying"
3485 );
3486 owner = Some(state);
3487 }
3488 Ok(owner)
3489}
3490
3491async fn attempt(
3497 opts: &Opts,
3498 queue: &Queue,
3499 status: &Arc<Mutex<Status>>,
3500 stop: &Stop,
3501 interrupt_pause: crate::graph::Pause,
3502 task: &mut Task,
3503) -> Vec<QuotaLoss> {
3504 let repo = repo_for(task, &opts.repo);
3505 tracing::info!(
3506 "task {} — {} (repo {})",
3507 task.short(),
3508 task.title,
3509 repo.display()
3510 );
3511
3512 let public_owner = match public_resume_owner(&crate::run::home(), &task.runs, &repo).await {
3513 Ok(owner) => owner,
3514 Err(e) => {
3515 task.hold_machine(Some(format!(
3516 "Cannot verify existing public owner: {e:#}; no attempt started."
3517 )));
3518 record(queue, task);
3519 return Vec::new();
3520 }
3521 };
3522 let resume_runs = public_owner
3523 .as_ref()
3524 .map(|s| vec![s.id.clone()])
3525 .unwrap_or_else(|| task.runs.clone());
3526
3527 if let Some(previous) = resume_runs.last().and_then(|id| RunState::load(id).ok())
3531 && previous
3532 .merge_decision
3533 .as_ref()
3534 .is_some_and(|decision| !decision.closed)
3535 && previous.status != RunStatus::Landing
3536 && !crate::merge_decision::resumable_hold(&previous)
3537 {
3538 task.hold_unrequested_retry(Some(format!("The existing owner stop decision remains held. Inspect run {} and its PR; request an explicit fresh review/decision before further work. No replacement or merge was started.", previous.id)));
3539 record(queue, task);
3540 return Vec::new();
3541 }
3542 let stop_owner = resume_runs
3543 .last()
3544 .and_then(|id| RunState::load(id).ok())
3545 .filter(|state| {
3546 crate::merge_decision::resumable_hold(state)
3547 || state.status == RunStatus::Landing
3548 && state.land_approval.as_ref().is_some_and(|binding| {
3549 Questions::open().get(&binding.question).is_ok_and(|q| {
3550 q.run == state.id
3551 && q.resolution()
3552 .as_deref()
3553 .is_some_and(crate::merge_decision::stops)
3554 })
3555 })
3556 });
3557 if let Some(owner) = stop_owner {
3558 let mut runner = match Runner::resume(&owner.id) {
3559 Ok(runner) => runner,
3560 Err(e) => {
3561 task.hold_manual(Some(format!("Cannot stop the existing landing owner: {e:#}. No replacement or merge was started.")));
3562 record(queue, task);
3563 return Vec::new();
3564 }
3565 };
3566 runner.on_pause(stop.pause());
3567 runner.watch_interrupt(interrupt_pause);
3568 let quota_before = runner.state.quota.clone();
3569 task.start(owner.id.clone());
3570 record(queue, task);
3571 lock(status).current.push(Current {
3572 task: task.id.clone(),
3573 run: owner.id,
3574 });
3575 runner.state.next_pass_task_entry = true;
3576 let result = runner.execute().await;
3577 return finish_attempt(
3578 opts.max_attempts,
3579 queue,
3580 task,
3581 &runner.state,
3582 "a_before,
3583 result,
3584 );
3585 }
3586
3587 if let Some(owner) = &public_owner
3588 && (task.fresh_start || !may_resume_run(owner, task))
3589 {
3590 task.hold_machine(Some(format!("Existing PR belongs to run {}; resume its authorized instructions or explicitly resolve its owner decision before starting a successor. No attempt started.", owner.id)));
3591 record(queue, task);
3592 return Vec::new();
3593 }
3594
3595 let unfinished = (!task.fresh_start)
3596 .then(|| unfinished_run(&resume_runs, task.short()))
3597 .flatten()
3598 .filter(|id| RunState::load(id).is_ok_and(|run| may_resume_run(&run, task)));
3599 let previous = resume_runs.last().and_then(|id| RunState::load(id).ok());
3600 if !task.fresh_start
3601 && task.pr_recovery.is_none()
3602 && task.review_branch.is_none()
3603 && task.resume_override.is_none()
3604 && previous
3605 .as_ref()
3606 .is_some_and(|run| run.retired_to.is_some())
3607 && unfinished.is_none()
3608 {
3609 task.hold_unrequested_retry(Some("Recorded recovery authority is unavailable, no longer resumable or no longer authorized for these instructions. Inspect the retained lineage and select a recovery source explicitly; no replacement competition was started.".into()));
3610 record(queue, task);
3611 return Vec::new();
3612 }
3613
3614 let saved_recovery = previous.as_ref().filter(|run| {
3615 task.replacement.is_none()
3616 && task.pr_recovery.as_ref().is_some_and(|selected| {
3617 selected.check_task(&task.id, &task.instruction).is_ok()
3618 && selected_starter(selected, Some(run), unfinished.as_deref())
3619 == Starter::Resume(run.id.clone())
3620 })
3621 });
3622
3623 let explicit = match &task.overrides {
3627 Some(o) if !o.inherit_daemon_options => o.config.as_deref(),
3628 Some(_) => opts.config.as_deref(),
3629 None => opts.config.as_deref(),
3630 };
3631 let (mut config, layers) = if let Some(run) = saved_recovery {
3632 let mut saved = run.clone();
3635 if let Err(e) = saved.configure_recovery_resume() {
3636 task.hold_manual(Some(format!("[recovery] saved run {} cannot resume: {e:#}. Inspect its saved configuration and evidence before requesting a new selection.", run.id)));
3637 record(queue, task);
3638 return Vec::new();
3639 }
3640 (saved.config, Vec::new())
3641 } else {
3642 let (mut config, layers) = match Config::discover_fetched(&repo, explicit).await {
3643 Ok(loaded) => loaded,
3644 Err(e) => {
3645 let reason = format!("[config] {e:#}");
3646 record_config_failure(task, &e);
3647 task.last_error = Some(reason.clone());
3648 task.hold_machine(Some(reason));
3649 record(queue, task);
3650 return Vec::new();
3651 }
3652 };
3653 if let Err(e) = apply_options(&mut config, opts, task) {
3654 task.hold_manual(Some(format!(
3655 "[config] {e:#}; correct the merge override before releasing the task"
3656 )));
3657 record(queue, task);
3658 return Vec::new();
3659 }
3660 if let Some(selected) = &task.pr_recovery
3661 && let Err(e) = selected.configure(&mut config)
3662 {
3663 task.hold_manual(Some(format!("[recovery] {e:#}. No run started; select the PR again with an explicit disposition and corrected configuration.")));
3664 record(queue, task);
3665 return Vec::new();
3666 }
3667 (config, layers)
3668 };
3669 let config_evidence = crate::pr_recovery::ConfigEvidence {
3670 layers,
3671 overrides: task.overrides.clone(),
3672 daemon_merge: task
3673 .overrides
3674 .as_ref()
3675 .is_none_or(|o| o.inherit_daemon_options)
3676 .then(|| opts.merge.clone())
3677 .flatten(),
3678 };
3679 apply_task_implementers(&mut config, task);
3680 if let Some(replacement) = &task.replacement {
3681 let safe = RunState::load(&replacement.run).is_ok_and(|old| {
3682 old.liveness(false) == Liveness::Dead
3683 && old.merge_decision.as_ref().is_some_and(|d| {
3684 d.closed
3685 && d.question == replacement.question
3686 && d.replacement_run.as_ref().is_none_or(|run| {
3687 task.runs.last() == Some(run)
3688 && RunState::load(run).is_ok_and(|current| {
3689 current.replaces.as_ref().is_some_and(|pin| {
3690 pin.question == replacement.question
3691 && pin.agent == replacement.agent
3692 })
3693 })
3694 })
3695 })
3696 });
3697 if !safe {
3698 task.hold_manual(Some("Replacement ownership cannot be confirmed, or a replacement run was already minted before a crash. Inspect the task and its linked runs; no second replacement was started.".into()));
3699 record(queue, task);
3700 return Vec::new();
3701 }
3702 match config.agent(&replacement.agent) {
3703 Ok(agent) if crate::agent::installed(agent) => {}
3704 _ => {
3705 task.hold_manual(Some(format!("Selected implementer `{}` is unavailable or no longer configured. No substitute was dispatched. Restore it and explicitly release the task.", replacement.agent)));
3706 record(queue, task);
3707 return Vec::new();
3708 }
3709 }
3710 config.graph.implementers = 1;
3711 config.graph.land_approval = true;
3712 config.merge.mode = MergeMode::Pr;
3713 config.graph.selected_implementer = Some(replacement.agent.clone());
3714 }
3715 let p = phrases(&config.graph.language);
3716 let start_failed = p.could_not_start;
3717 if let Ok(base) =
3722 crate::git::merge_base_branch(&repo, &config.merge.remote, config.merge.base.as_deref())
3723 .await
3724 && let Ok(head) =
3725 crate::git::rev_parse(&repo, &format!("{}/{}", config.merge.remote, base)).await
3726 {
3727 let mut inputs = crate::queue::recovery_inputs(&task.sources);
3728 if let Some(branch) = task.review_of.as_ref().or(task.review_branch.as_ref()) {
3729 let source = crate::git::rev_parse(&repo, branch)
3730 .await
3731 .unwrap_or_default();
3732 inputs.push_str(&format!(";review:{branch}:{source}"));
3733 }
3734 if let Some(reason) = task.recovery_exhausted(
3735 &head,
3736 &inputs,
3737 task.attempt_limit
3738 .map_or(opts.max_attempts, std::num::NonZeroUsize::get),
3739 ) {
3740 task.hold_unrequested_retry(Some(reason));
3741 record(queue, task);
3742 return Vec::new();
3743 }
3744 task.recovery_admission = Some(crate::queue::RecoveryAdmission { base: head, inputs });
3745 }
3746
3747 if let Some(reason) = disk_gate(&repo, &config) {
3755 task.notice_cause("disk-space", None, None);
3756 task.last_error = Some(reason.clone());
3757 task.hold_machine(Some(reason.clone()));
3758 record(queue, task);
3759 tracing::warn!("holding {} for want of disk space: {reason}", task.short());
3760 notices::raise(
3763 Notice::warn(
3764 &format!("disk:{}", repo.display()),
3765 "A task was held for want of free disk space; free some, then release it from the queue.",
3766 )
3767 .link(Link::Task {
3768 id: task.id.clone(),
3769 }),
3770 );
3771 return Vec::new();
3772 }
3773
3774 let review_branch = take_review_branch(task, previous.as_ref());
3799 let branch_exists = match &review_branch {
3800 Some(branch) => crate::git::branch_exists(&repo, branch)
3801 .await
3802 .unwrap_or(false),
3803 None => false,
3804 };
3805 let proposed = match (&review_branch, &unfinished, &task.review_of) {
3809 (None, None, Some(branch)) => Starter::Review(branch.clone()),
3810 _ => choose_starter(
3811 review_branch.as_deref(),
3812 branch_exists,
3813 unfinished.as_deref(),
3814 ),
3815 };
3816 let retry_selected = task.replacement.is_some()
3817 && task.resumption.is_some()
3818 && !task.fresh_start
3819 && previous.as_ref().is_some_and(|run| {
3820 run.replaces.is_some()
3821 && ((run.status == RunStatus::Failed && run.viable().is_empty())
3822 || (run.status == RunStatus::Blocked && run.tally.is_none()))
3823 });
3824 let stop_pin = task
3825 .resume_override
3826 .as_ref()
3827 .filter(|pin| {
3828 Questions::open().get(&pin.question_id).is_ok_and(|q| {
3829 q.node == land::APPROVAL_NODE
3830 && q.resolution()
3831 .as_deref()
3832 .is_some_and(crate::merge_decision::stops)
3833 })
3834 })
3835 .and_then(|pin| pin.pinned_run.clone());
3836 let starter = if let Some(owner) = &public_owner {
3837 Starter::Resume(owner.id.clone())
3838 } else if let Some(run) = stop_pin {
3839 Starter::Resume(run)
3842 } else if retry_selected {
3843 Starter::Resume(previous.as_ref().unwrap().id.clone())
3844 } else if let Some(selected) = &task.pr_recovery {
3845 if selected.check_task(&task.id, &task.instruction).is_ok() {
3846 selected_starter(selected, previous.as_ref(), unfinished.as_deref())
3847 } else {
3848 Starter::Review(selected.branch.clone())
3849 }
3850 } else {
3851 starter_with_implementers(task, previous.as_ref(), proposed)
3852 };
3853 if let Some(pin) = &task.resume_override
3856 && let Some(run) = &pin.pinned_run
3857 && starter != Starter::Resume(run.clone())
3858 {
3859 let run = run.clone();
3860 task.notice_cause("resume-owner", Some(&run), None);
3861 task.hold_machine(Some(format!(
3862 "pinned run cannot resume; inspect `magi show {run}` and restore the existing run before `magi task release {}`; no replacement task was started",
3863 task.id
3864 )));
3865 record(queue, task);
3866 return Vec::new();
3867 }
3868 if task.replacement.is_some()
3869 && previous.as_ref().is_some_and(|run| run.replaces.is_some())
3870 && starter != Starter::Resume(previous.as_ref().unwrap().id.clone())
3871 {
3872 task.hold_manual(Some("The selected replacement already has an attempt. It cannot be resumed safely; inspect its saved pin and restore the existing run. No additional replacement was started.".into()));
3873 record(queue, task);
3874 return Vec::new();
3875 }
3876 if let Some(reason) = retained_attempt_hold(
3877 task,
3878 &starter,
3879 config.disk.retained_attempt_limit,
3880 RunState::load,
3881 ) {
3882 if let Starter::Review(branch) = &starter {
3885 task.review_branch = Some(branch.clone());
3886 }
3887 task.hold_unrequested_retry(Some(reason));
3888 record(queue, task);
3889 return Vec::new();
3890 }
3891 let attachments = match task_attachments(queue, task) {
3896 Ok(a) => a,
3897 Err(e) => {
3898 task.attempts += 1;
3899 task.fail_startup_target(
3900 "startup-attachments",
3901 e.name.clone(),
3902 format!("{start_failed}{e:#}"),
3903 opts.max_attempts,
3904 );
3905 record(queue, task);
3906 return Vec::new();
3907 }
3908 };
3909 let startup_pause = stop.pause();
3910 let startup_interrupt = interrupt_pause.clone();
3911 let wait = crate::repo_lock::WaitPolicy::new(
3912 Duration::from_secs(config.graph.timeout_resource_wait),
3913 move || startup_pause.parked() || startup_interrupt.parked(),
3914 );
3915 let started = wait.scope(Box::pin(async { match &starter {
3918 Starter::Review(_) if task.pr_recovery.is_some() => {
3919 let selected = task.pr_recovery.clone().expect("selected recovery");
3920 match selected.check_task(&task.id, &task.instruction) {
3921 Ok(()) => Runner::recover_pr(&repo, &task.id, selected, config)
3922 .await
3923 .map(|mut r| {
3924 r.state.attachments = attachments.clone();
3925 r
3926 }),
3927 Err(e) => Err(e),
3928 }
3929 }
3930 Starter::Review(branch) => {
3931 tracing::info!(
3932 "task {} reopens `{branch}` as a review-only pass",
3933 task.short()
3934 );
3935 let takeover = crate::handover::Takeover {
3938 earlier: task.earlier_attempts().to_vec(),
3939 home: crate::run::home(),
3940 choice: take_divergence_answer(branch, &config.merge.remote, task),
3941 task_instruction: Some(instruction_for(task)),
3942 };
3943 Runner::review_taking_over(
3944 &repo,
3945 branch,
3946 config,
3947 Some(takeover),
3948 crate::run::Origin::queue(&task.id),
3949 )
3950 .await
3951 }
3952 Starter::Resume(id) => {
3953 tracing::info!("resuming run {id} rather than competing again");
3954 Runner::resume(id).and_then(|mut r| {
3955 if retry_selected {
3956 for candidate in &mut r.state.candidates {
3959 candidate.failed = None;
3960 candidate.empty = false;
3961 candidate.commits = 0;
3962 }
3963 r.state.status = RunStatus::Implementing;
3964 r.state.event("implement", "owner released the held selected implementer; retrying the same attempt without fallback");
3965 }
3966 if r.state.prep_complete == Some(false) && r.state.candidates.is_empty() {
3967 r.state.sources = task.sources.clone();
3968 }
3969 if r.state.prep_failure.is_some() {
3970 let approval = task
3971 .resumption
3972 .as_ref()
3973 .context("preparation retry requires an explicit task release")?;
3974 r.retry_preparation(approval.at)?;
3975 }
3976 if let Some(instruction) =
3977 prepare_instruction(&starter, Some(&r.state.instruction), task)
3978 {
3979 r.state.instruction = instruction;
3980 }
3981 r.state.attachments = attachments.clone();
3982 if task.replacement.is_some() {
3983 r.state.config.merge.mode = MergeMode::Pr;
3984 r.state.config.graph.land_approval = true;
3985 }
3986 if task.replacement.is_none() && task.pr_recovery.is_none() && let Some(mode) = task
3990 .overrides
3991 .as_ref()
3992 .and_then(|o| o.merge.as_deref())
3993 .and_then(|m| merge_mode(m).ok())
3994 {
3995 r.state.config.merge.mode = mode;
3996 }
3997 Ok(r)
3998 })
3999 }
4000 Starter::Start => {
4001 if let Some(branch) = &review_branch {
4002 tracing::warn!(
4003 "conductor chose review for task {} but branch `{branch}` no longer \
4004 exists; requeuing as a fresh competition instead",
4005 task.short()
4006 );
4007 }
4008 let instruction = prepare_instruction(&starter, None, task)
4009 .unwrap_or_else(|| task.instruction.clone());
4010 Runner::start_naming(
4011 &repo,
4012 instruction,
4013 &task.title,
4014 config,
4015 crate::run::Origin::queue(&task.id),
4016 )
4017 .await
4018 .map(|mut r| {
4019 r.state.attachments = attachments.clone();
4020 r.state.sources = task.sources.clone();
4021 r
4022 })
4023 }
4024 }})).await;
4025 let mut runner = match started {
4026 Ok(r) => r,
4027 Err(e) if e.is::<crate::repo_lock::ResourceBlocked>() => {
4028 task.last_error = Some(e.to_string());
4029 if stop.parking() || interrupt_pause.parked() {
4030 let attempts = task.attempts;
4031 task.park(e.to_string());
4032 task.attempts = attempts;
4034 } else {
4035 task.hold_machine(Some(format!("{}{e}", crate::cache::RESOURCE_HOLD)));
4036 }
4037 record(queue, task);
4038 return Vec::new();
4039 }
4040 Err(e) if task.pr_recovery.is_some() => {
4044 task.startup_notice_cause("startup-public-head");
4045 task.attempts += 1;
4046 if let Some(partial) = e.downcast_ref::<crate::pr_recovery::StartupFailed>() {
4047 task.link_run(&partial.run);
4048 task.record_pass(&partial.run, 0, task.attempts, None);
4049 }
4050 task.last_error = Some(format!("{start_failed}{e:#}"));
4051 task.hold_unrequested_retry(Some(format!(
4052 "Selected public-head recovery could not start: {e:#}. Correct the cause and use `magi task recover-pr` with the approved public head; automatic recovery is stopped."
4053 )));
4054 record(queue, task);
4055 return Vec::new();
4056 }
4057 Err(e) if e.downcast_ref::<crate::handover::Refused>().is_some() => {
4058 let detail = match e.downcast_ref::<crate::handover::Refused>() {
4059 Some(r) => (p.handover_refused)(r),
4060 None => format!("{e:#}"),
4061 };
4062 task.startup_notice_cause("startup-handover");
4063 let reason = format!("{start_failed}{detail}");
4064 task.last_error = Some(reason.clone());
4065 let branch = match &starter {
4068 Starter::Review(branch) => Some(branch.clone()),
4069 _ => None,
4070 };
4071 task.hold_for_handover(branch, format!("{reason}{}", p.handover_hint));
4075 record(queue, task);
4076 tracing::warn!(
4077 "holding {} for a branch it cannot take over: {e:#}",
4078 task.short()
4079 );
4080 return Vec::new();
4081 }
4082 Err(e) if e.downcast_ref::<crate::reconcile::Diverged>().is_some() => {
4087 let d = e
4088 .downcast_ref::<crate::reconcile::Diverged>()
4089 .expect("checked by the guard");
4090 let mut q = ask::Question::new(
4091 task.id.clone(),
4092 crate::reconcile::NODE.to_owned(),
4093 crate::reconcile::SEAT.to_owned(),
4094 d.summary(),
4095 d.detail(),
4096 d.choices(),
4097 );
4098 match Questions::open().put(&mut q) {
4099 Ok(()) => {
4100 task.last_error = Some(format!("{e:#}"));
4101 task.review_branch = Some(d.branch.clone());
4104 task.review_instruction = Some(task.instruction.clone());
4105 task.block(vec![q.id.clone()], Some(d.summary()));
4106 }
4107 Err(put) => {
4108 tracing::warn!("could not file the divergence question: {put:#}");
4109 task.attempts += 1;
4110 task.fail_startup(
4111 "startup-question-store",
4112 format!("{start_failed}{e:#}"),
4113 opts.max_attempts,
4114 );
4115 }
4116 }
4117 record(queue, task);
4118 return Vec::new();
4119 }
4120 Err(e) => {
4121 if e.downcast_ref::<crate::reconcile::Stale>().is_some()
4127 && let Starter::Review(branch) = &starter
4128 {
4129 task.review_branch = Some(branch.clone());
4130 task.review_instruction = Some(task.instruction.clone());
4131 task.last_error = Some(format!("{start_failed}{e:#}"));
4132 task.status = crate::queue::TaskStatus::Failed;
4133 record(queue, task);
4134 return Vec::new();
4135 }
4136 task.attempts += 1;
4137 task.last_error = Some(format!("{start_failed}{e:#}"));
4138 task.startup_notice_cause("startup-source");
4139 task.hold_unrequested_retry(Some(format!(
4140 "Startup failed: {e:#}. Correct the startup source or configuration, then explicitly release the task; automatic recovery is stopped."
4141 )));
4142 record(queue, task);
4143 return Vec::new();
4144 }
4145 };
4146 if task.pr_recovery.is_some() && !matches!(starter, Starter::Resume(_)) {
4147 runner.state.recovery_config = Some(config_evidence);
4148 }
4149 if let Some(replacement) = &task.replacement {
4150 if matches!(starter, Starter::Start) {
4151 runner.state.instruction.push_str(&format!(
4152 "\n\n# Replacement review context\n{}",
4153 replacement.context
4154 ));
4155 }
4156 runner.state.replaces = Some(replacement.clone());
4157 runner.state.config.graph.selected_implementer = Some(replacement.agent.clone());
4158 if let Ok(mut old) = RunState::load(&replacement.run)
4159 && let Some(d) = &mut old.merge_decision
4160 {
4161 d.replacement_run = Some(runner.state.id.clone());
4162 old.event(
4163 "replacement",
4164 format!(
4165 "replacement run {} owns the next attempt; old PR remains closed",
4166 runner.state.id
4167 ),
4168 );
4169 if let Err(e) = old.save() {
4170 task.hold_manual(Some(format!(
4171 "Could not link replacement history: {e}; no agents dispatched"
4172 )));
4173 record(queue, task);
4174 return Vec::new();
4175 }
4176 }
4177 }
4178 runner.state.task_instruction = Some(task.instruction.clone());
4179 runner.state.followup_generation = Some(task.followup.as_ref().map_or(0, |f| f.generation));
4182 if runner.state.origin_chat.is_none() {
4183 runner.state.origin_chat = task.chat_talk().map(str::to_owned);
4184 }
4185 runner.on_pause(stop.pause());
4187 runner.watch_interrupt(interrupt_pause);
4191
4192 let run = runner.state.id.clone();
4195 task.start(run.clone());
4196 record(queue, task);
4197 lock(status).current.push(Current {
4198 task: task.id.clone(),
4199 run,
4200 });
4201
4202 let quota_before = runner.state.quota.clone();
4205 runner.state.next_pass_task_entry = true;
4206 let result = runner.execute().await;
4207 finish_attempt(
4208 opts.max_attempts,
4209 queue,
4210 task,
4211 &runner.state,
4212 "a_before,
4213 result,
4214 )
4215}
4216
4217pub fn finish_attempt(
4222 max_attempts: usize,
4223 queue: &Queue,
4224 task: &mut Task,
4225 state: &RunState,
4226 quota_before: &[QuotaLoss],
4227 result: Result<()>,
4228) -> Vec<QuotaLoss> {
4229 let detail = match result {
4230 Ok(()) => describe(state),
4231 Err(e) => format!("{e:#}"),
4232 };
4233 let fresh = losses_this_attempt(quota_before, &state.quota);
4234 let verdict = Verdict {
4235 status: state.status,
4236 left_pr: state.pr.is_some(),
4239 quota_hit: !fresh.is_empty(),
4245 parked: state.parked,
4249 no_viable_candidates: state.viable().is_empty(),
4252 };
4253 if let Some(replacement) = &state.replaces
4254 && let Some(pr) = &state.pr
4255 && let Ok(mut old) = RunState::load(&replacement.run)
4256 && let Some(d) = &mut old.merge_decision
4257 {
4258 d.replacement_pr = Some(pr.url.clone());
4259 let _ = old.save();
4260 }
4261 settle_and_diagnose(task, verdict, &detail, max_attempts, state);
4262 if state.status == RunStatus::Blocked && state.prep_failure.is_some() {
4263 task.hold_unrequested_retry(Some(format!(
4264 "Candidate setup failed: {detail}. Correct the explicit sources with `magi task sources {}` (README, \"Use\": implementation imports), or select a complete implementation with `magi task recover-pr`, then explicitly release the task.",
4265 task.short()
4266 )));
4267 }
4268 if task.status == TaskStatus::Done {
4269 supersede_prior_runs(task, &crate::run::home());
4270 }
4271 record(queue, task);
4272 tracing::info!(
4273 "task {} is {} after run {} ({})",
4274 task.short(),
4275 task.status.as_str(),
4276 state.short(),
4277 label(state.status)
4278 );
4279 fresh
4280}
4281
4282pub fn hold_if_runnable(queue: &Queue, task: &mut Task) {
4286 let waits_on_owner = withheld_text_wait(task).is_some();
4291 if task.status.runnable()
4292 && !waits_on_owner
4293 && !(task.fresh_start && task.replacement.is_some())
4294 {
4295 let why = task.last_error.clone().map_or_else(
4296 || "the run did not finish".to_owned(),
4297 |e| format!("the run did not finish: {e}"),
4298 );
4299 task.hold_unrequested_retry(Some(format!(
4300 "{why}. It was started by hand, so it is not retried \
4301 automatically; `magi task release` retries it."
4302 )));
4303 record(queue, task);
4304 }
4305}
4306
4307#[must_use]
4311pub fn loop_would_resume(run: &str) -> bool {
4312 unfinished_run(&[run.to_owned()], crate::run::short_of(run)).is_some()
4313}
4314
4315#[must_use]
4325pub fn foreign_loop(
4326 reading: Option<&Reading>,
4327 now: Timestamp,
4328 own_pid: u32,
4329) -> Option<Option<u32>> {
4330 let reading = reading.filter(|r| r.running(now))?;
4331 match reading.pid {
4332 Some(pid) if pid == own_pid => None,
4333 pid => Some(pid),
4334 }
4335}
4336
4337pub async fn run_claimed(opts: &Opts, queue: &Queue, task: &mut Task) {
4352 let status = Arc::new(Mutex::new(Status::new()));
4353 let stop = Stop::new();
4354 Box::pin(attempt(
4357 opts,
4358 queue,
4359 &status,
4360 &stop,
4361 crate::graph::Pause::new(),
4362 task,
4363 ))
4364 .await;
4365 let mut resumes = 0;
4374 while resumes < MAX_WITHHELD_RESUMES {
4375 let Some(timeout) = withheld_text_wait(task) else {
4376 break;
4377 };
4378 eprintln!(
4379 "waiting for the owner's answer on a withheld pull request text \
4380 (up to {timeout}s; Ctrl-C leaves the run parked for a later `magi serve`)"
4381 );
4382 let limit = std::time::Instant::now() + Duration::from_secs(timeout + 60);
4385 let ready = loop {
4386 match land_resume_state(task) {
4387 LandResume::StillWaiting if std::time::Instant::now() < limit => {
4388 tokio::time::sleep(WITHHELD_POLL).await;
4389 }
4390 LandResume::Ready => break true,
4391 _ => break false,
4392 }
4393 };
4394 if !ready {
4395 break;
4396 }
4397 resumes += 1;
4398 Box::pin(attempt(
4399 opts,
4400 queue,
4401 &status,
4402 &stop,
4403 crate::graph::Pause::new(),
4404 task,
4405 ))
4406 .await;
4407 }
4408 hold_if_runnable(queue, task);
4409}
4410
4411const MAX_WITHHELD_RESUMES: u32 = 3;
4414
4415const WITHHELD_POLL: Duration = Duration::from_secs(15);
4417
4418fn withheld_text_wait(task: &Task) -> Option<u64> {
4421 let id = task.runs.last()?;
4422 let s = RunState::load(id).ok()?;
4423 (s.parked && s.github_text.as_ref().is_some_and(|g| !g.resolved))
4424 .then_some(s.config.graph.answer_timeout)
4425}
4426
4427fn losses_this_attempt(before: &[QuotaLoss], after: &[QuotaLoss]) -> Vec<QuotaLoss> {
4437 after
4438 .iter()
4439 .filter(|q| !before.contains(q))
4440 .cloned()
4441 .collect()
4442}
4443
4444fn cooldown_until(quota: &[QuotaLoss], now: Timestamp) -> Option<Timestamp> {
4447 if quota.is_empty() {
4448 return None;
4449 }
4450 let with_hint = quota.iter().find(|q| q.reset.is_some());
4451 let reset_at = with_hint.and_then(|q| parse_reset_hint(q.reset.as_deref()?, now, q.at));
4452 let wait = quota_wait(reset_at, now, QUOTA_WAIT_FALLBACK, QUOTA_WAIT_CAP);
4453 let secs = i64::try_from(wait.as_secs()).unwrap_or(i64::MAX);
4454 Some(
4455 now.checked_add(jiff::SignedDuration::from_secs(secs))
4456 .unwrap_or(Timestamp::MAX),
4457 )
4458}
4459
4460fn implementers_match(task: &Task, state: &RunState) -> bool {
4462 task.replacement.as_ref().map_or_else(
4463 || {
4464 task.implementers()
4465 .is_none_or(|n| n == state.config.graph.implementers)
4466 },
4467 |replacement| {
4468 state.config.graph.selected_implementer.as_deref() == Some(&replacement.agent)
4469 && state.config.graph.implementers == 1
4470 },
4471 )
4472}
4473
4474fn starter_with_implementers(
4476 task: &Task,
4477 previous: Option<&RunState>,
4478 proposed: Starter,
4479) -> Starter {
4480 if previous.is_some_and(|s| !implementers_match(task, s)) {
4481 task.review_of
4482 .as_ref()
4483 .map_or(Starter::Start, |branch| Starter::Review(branch.clone()))
4484 } else {
4485 proposed
4486 }
4487}
4488
4489fn apply_task_implementers(config: &mut Config, task: &Task) {
4491 if let Some(n) = task.implementers() {
4492 config.graph.implementers = n;
4493 }
4494}
4495
4496#[cfg(test)]
4501fn prepare_for(repo: &Path, opts: &Opts, task: &Task) -> Result<Config> {
4502 let explicit = match &task.overrides {
4503 Some(o) if !o.inherit_daemon_options => o.config.as_deref(),
4504 _ => opts.config.as_deref(),
4505 };
4506 let (mut config, _) = Config::discover(repo, explicit)?;
4507 apply_options(&mut config, opts, task)?;
4508 Ok(config)
4509}
4510
4511fn apply_options(config: &mut Config, opts: &Opts, task: &Task) -> Result<()> {
4512 if task
4513 .overrides
4514 .as_ref()
4515 .is_none_or(|o| o.inherit_daemon_options)
4516 && let Some(mode) = opts.merge.as_deref()
4517 {
4518 config.merge.mode = merge_mode(mode)?;
4519 }
4520 if let Some(o) = &task.overrides {
4521 o.apply(config);
4522 }
4523 Ok(())
4524}
4525
4526fn prepare(repo: &Path, opts: &Opts) -> Result<Config> {
4528 let (mut config, _layers) = Config::discover(repo, opts.config.as_deref())?;
4529 if let Some(mode) = &opts.merge {
4530 config.merge.mode = merge_mode(mode)?;
4531 }
4532 Ok(config)
4533}
4534
4535async fn maybe_prune_cache_between_runs(
4566 repo: &Path,
4567 opts: &Opts,
4568 home: &Path,
4569 stop: &Stop,
4570 last_checked: &mut Option<Timestamp>,
4571 now: Timestamp,
4572) {
4573 if stop.stopped() || !cache_check_due(*last_checked, now, CACHE_CHECK_INTERVAL_SECS) {
4574 return;
4575 }
4576 *last_checked = Some(now);
4577 let cfg = match prepare(repo, opts) {
4578 Ok(cfg) => cfg,
4579 Err(e) => {
4580 tracing::warn!("cache check: no config: {e:#}");
4581 return;
4582 }
4583 };
4584 match clean::prune_cache_if_over_limit(&cfg, home) {
4585 Ok(Some(pruned)) if pruned.files > 0 => tracing::info!(
4586 "housekeep: pruned {} file(s) ({} bytes) from the shared cache between runs",
4587 pruned.files,
4588 pruned.freed
4589 ),
4590 Ok(_) => {}
4591 Err(e) => {
4592 tracing::warn!("housekeep: prune cache: {e:#}");
4593 notices::raise_in(
4594 home,
4595 Notice::warn(
4596 "housekeep:cache",
4597 "Pruning the shared build cache failed; disk usage may keep growing.",
4598 ),
4599 );
4600 }
4601 }
4602}
4603
4604fn cache_check_due(last_checked: Option<Timestamp>, now: Timestamp, interval_secs: u64) -> bool {
4608 last_checked.is_none_or(|last| clean::due(now, last, interval_secs))
4609}
4610
4611async fn janitor(repo: &Path, opts: &Opts, home: &Path, worktrees_root: &Path) {
4630 let cfg = match prepare(repo, opts) {
4631 Ok(cfg) => cfg,
4632 Err(e) => {
4633 tracing::warn!("housekeep: no config: {e:#}");
4634 return;
4635 }
4636 };
4637 let worktrees_root = cfg.graph.worktree_root.as_deref().unwrap_or(worktrees_root);
4646 let out = clean::housekeep(&cfg, home, worktrees_root, repo, Timestamp::now()).await;
4647 if out.folded > 0 || out.unreadable > 0 || out.orphaned_worktrees > 0 {
4652 let mut extra = Vec::new();
4653 if out.unreadable > 0 {
4654 extra.push(format!("{} unreadable", out.unreadable));
4655 }
4656 if out.orphaned_worktrees > 0 {
4657 extra.push(format!("{} orphaned worktree(s)", out.orphaned_worktrees));
4658 }
4659 let detail = if extra.is_empty() {
4660 String::new()
4661 } else {
4662 format!(" ({})", extra.join(", "))
4663 };
4664 tracing::info!("housekeep: folded {} run(s){detail}", out.folded);
4665 }
4666 if out.external_merges_recorded > 0 {
4667 tracing::info!(
4668 "housekeep: recorded {} run(s) as merged externally",
4669 out.external_merges_recorded
4670 );
4671 }
4672 if out.stale_pr_states_repaired > 0 {
4673 tracing::info!(
4674 "housekeep: rewrote {} run record(s) whose pull request had already settled",
4675 out.stale_pr_states_repaired
4676 );
4677 }
4678 if out.cache_files > 0 {
4679 tracing::info!(
4680 "housekeep: pruned {} file(s) ({} bytes) from the shared cache",
4681 out.cache_files,
4682 out.cache_freed
4683 );
4684 }
4685 if out.questions_abandoned > 0 {
4686 tracing::info!(
4687 "housekeep: abandoned {} question(s) left open by a finished run",
4688 out.questions_abandoned
4689 );
4690 }
4691}
4692
4693async fn triage_held(queue: &Queue, home: &Path, opts: &Opts) {
4702 let questions = Questions::at(home.join("questions"));
4703 let report = triage::run_once(queue, &questions, opts.config.as_deref(), Timestamp::now());
4704 if report.is_empty() {
4705 return;
4706 }
4707 if !report.quarantined.is_empty() {
4708 tracing::info!(
4709 "triage: held {} blocked task(s) whose blocked-on task or \
4710 question no longer exists: {}",
4711 report.quarantined.len(),
4712 report.quarantined.join(", ")
4713 );
4714 }
4715 if !report.resumed.is_empty() {
4716 tracing::info!(
4717 "triage: resumed {} held task(s) whose machine hold had resolved: {}",
4718 report.resumed.len(),
4719 report.resumed.join(", ")
4720 );
4721 }
4722 if !report.asked.is_empty() {
4723 tracing::info!(
4724 "triage: asked about {} held task(s): {}",
4725 report.asked.len(),
4726 report.asked.join(", ")
4727 );
4728 }
4729 if !report.answered.is_empty() {
4730 tracing::info!(
4731 "triage: applied {} operator answer(s): {}",
4732 report.answered.len(),
4733 report.answered.join(", ")
4734 );
4735 }
4736}
4737
4738pub(crate) fn disk_gate(repo: &Path, config: &Config) -> Option<String> {
4745 disk_gate_with(repo, config, crate::disk::free_bytes)
4746}
4747
4748fn disk_gate_with<F: Fn(&Path) -> Result<u64>>(
4752 repo: &Path,
4753 config: &Config,
4754 free_bytes: F,
4755) -> Option<String> {
4756 let min = config.disk.min_free_bytes;
4757 if min == 0 {
4758 return None;
4759 }
4760 let mut paths = vec![repo.to_owned()];
4761 if let Some(cache) = config.cache_dir() {
4762 paths.push(cache);
4763 }
4764 if let Some(root) = &config.graph.worktree_root {
4765 paths.push(root.clone());
4766 }
4767 for path in paths {
4768 let probe = if path == repo {
4769 path.as_path()
4770 } else {
4771 path.ancestors().find(|p| p.exists()).unwrap_or(&path)
4772 };
4773 let reason = match free_bytes(probe) {
4774 Ok(free) => crate::disk::gate_in(free, min, &config.graph.language),
4775 Err(e) => Some(crate::disk::unmeasured_in(
4776 probe,
4777 &e,
4778 &config.graph.language,
4779 )),
4780 };
4781 if reason.is_some() {
4782 return reason;
4783 }
4784 }
4785 None
4786}
4787
4788const QUOTA_WAIT_FALLBACK: Duration = Duration::from_secs(5 * 60);
4795
4796const QUOTA_WAIT_CAP: Duration = Duration::from_secs(30 * 60);
4800
4801fn quota_wait(
4810 reset_at: Option<Timestamp>,
4811 now: Timestamp,
4812 fallback: Duration,
4813 cap: Duration,
4814) -> Duration {
4815 match reset_at {
4816 Some(at) if at > now => {
4817 let secs = u64::try_from(at.as_second() - now.as_second()).unwrap_or(0);
4818 Duration::from_secs(secs).min(cap)
4819 }
4820 _ => fallback,
4821 }
4822}
4823
4824fn parse_reset_hint(text: &str, now: Timestamp, recorded: Timestamp) -> Option<Timestamp> {
4838 parse_reset_hint_zoned(text, now)
4839 .or_else(|| parse_reset_hint_dated(text))
4840 .or_else(|| parse_reset_hint_relative(text, recorded))
4841}
4842
4843fn parse_reset_hint_relative(text: &str, recorded: Timestamp) -> Option<Timestamp> {
4847 let rest = text.trim().trim_end_matches('.').strip_prefix("in ")?;
4848 let mut rest = rest.trim();
4849 if rest.is_empty() {
4850 return None;
4851 }
4852 let mut total: i64 = 0;
4853 let mut matched = false;
4854 for (unit, secs) in [('h', 3600), ('m', 60), ('s', 1)] {
4855 if let Some((digits, tail)) = rest.split_once(unit)
4856 && !digits.is_empty()
4857 && digits.bytes().all(|b| b.is_ascii_digit())
4858 {
4859 total += digits.parse::<i64>().ok()?.checked_mul(secs)?;
4860 rest = tail;
4861 matched = true;
4862 }
4863 }
4864 if !rest.is_empty() || !matched {
4865 return None;
4866 }
4867 recorded
4868 .checked_add(jiff::SignedDuration::from_secs(total))
4869 .ok()
4870}
4871
4872fn parse_12h_clock(clock: &str) -> Option<(i8, i8)> {
4876 let clock = clock.trim().to_lowercase();
4877 let (digits, pm) = clock
4878 .strip_suffix("am")
4879 .map(|d| (d, false))
4880 .or_else(|| clock.strip_suffix("pm").map(|d| (d, true)))?;
4881 let (h, m) = digits.trim().split_once(':')?;
4882 let mut hour: i8 = h.trim().parse().ok()?;
4883 let minute: i8 = m.trim().parse().ok()?;
4884 if !(1..=12).contains(&hour) || !(0..=59).contains(&minute) {
4885 return None;
4886 }
4887 if pm && hour != 12 {
4888 hour += 12;
4889 } else if !pm && hour == 12 {
4890 hour = 0;
4891 }
4892 Some((hour, minute))
4893}
4894
4895fn parse_reset_hint_zoned(text: &str, now: Timestamp) -> Option<Timestamp> {
4900 let open = text.find('(')?;
4901 let close = text.rfind(')')?;
4902 if close <= open {
4903 return None;
4904 }
4905 let zone = text[open + 1..close].trim();
4906 let (hour, minute) = parse_12h_clock(&text[..open])?;
4907 let tz = jiff::tz::TimeZone::get(zone).ok()?;
4908 let candidate = now
4909 .to_zoned(tz)
4910 .with()
4911 .hour(hour)
4912 .minute(minute)
4913 .second(0)
4914 .millisecond(0)
4915 .microsecond(0)
4916 .nanosecond(0)
4917 .build()
4918 .ok()?;
4919 let mut at = candidate.timestamp();
4920 if at <= now {
4921 at += jiff::SignedDuration::from_hours(24);
4922 }
4923 Some(at)
4924}
4925
4926fn parse_reset_hint_dated(text: &str) -> Option<Timestamp> {
4935 let words: Vec<&str> = text.split_whitespace().collect();
4936 if words.len() < 5 {
4937 return None;
4938 }
4939 (0..=words.len() - 5)
4940 .find_map(|start| parse_dated_window(&words[start..start + 5], words.get(start + 5)))
4941}
4942
4943fn parse_dated_window(window: &[&str], trailing: Option<&&str>) -> Option<Timestamp> {
4949 if trailing.is_some_and(|next| next.starts_with('(')) {
4950 return None;
4951 }
4952 let month = month_number(window[0])?;
4953 let day_token = window[1].strip_suffix(',')?.to_lowercase();
4954 let day_digits = ["st", "nd", "rd", "th"]
4955 .iter()
4956 .find_map(|suffix| day_token.strip_suffix(*suffix))?;
4957 let day: i8 = day_digits.parse().ok()?;
4958 let year_token = window[2];
4959 if year_token.len() != 4 || !year_token.bytes().all(|b| b.is_ascii_digit()) {
4960 return None;
4961 }
4962 let year: i16 = year_token.parse().ok()?;
4963 let ampm = window[4].trim_matches(|c: char| !c.is_ascii_alphabetic());
4967 let (hour, minute) = parse_12h_clock(&format!("{}{}", window[3], ampm))?;
4968 let date = jiff::civil::Date::new(year, month, day).ok()?;
4969 let candidate = date
4970 .at(hour, minute, 0, 0)
4971 .to_zoned(jiff::tz::TimeZone::UTC)
4972 .ok()?;
4973 Some(candidate.timestamp())
4974}
4975
4976fn month_number(name: &str) -> Option<i8> {
4979 const NAMES: [&str; 12] = [
4980 "jan", "feb", "mar", "apr", "may", "jun", "jul", "aug", "sep", "oct", "nov", "dec",
4981 ];
4982 let lower = name.to_lowercase();
4983 NAMES
4984 .iter()
4985 .position(|n| *n == lower.as_str())
4986 .map(|i| i as i8 + 1)
4987}
4988
4989#[cfg(test)]
4991fn exhausted_review_budget(state: &RunState) -> bool {
4992 !crate::handover::can_resume(state)
4995 && state.status == RunStatus::Blocked
4996 && !state.released()
4997 && state.retired_to.is_none()
4998}
4999
5000fn unfinished_run(runs: &[String], short: &str) -> Option<String> {
5037 unfinished_run_with(runs, short, RunState::load)
5038}
5039
5040pub(crate) fn unfinished_run_with<F>(runs: &[String], short: &str, mut load: F) -> Option<String>
5043where
5044 F: FnMut(&str) -> Result<RunState>,
5045{
5046 let mut id = runs.last()?.clone();
5047 let mut visited = std::collections::BTreeSet::new();
5048 let mut repo = None;
5049 loop {
5050 if !runs.contains(&id) || !visited.insert(id.clone()) {
5051 tracing::warn!("invalid or cyclic recovery authority for task {short}");
5052 return None;
5053 }
5054 let state = match load(&id) {
5055 Ok(state) if state.id == id => state,
5056 Ok(_) => return None,
5057 Err(e) => {
5058 tracing::warn!("could not read run {id} for task {short}: {e:#}");
5059 return None;
5060 }
5061 };
5062 if repo.as_ref().is_some_and(|repo| *repo != state.repo) {
5063 return None;
5064 }
5065 repo = Some(state.repo.clone());
5066 if let Some(authority) = state.retired_to {
5067 id = authority;
5068 continue;
5069 }
5070 return (crate::handover::can_resume(&state)
5071 && state.liveness(false) != crate::run::Liveness::Live)
5072 .then_some(id);
5073 }
5074}
5075
5076fn awaiting_resume_with<F>(task: &Task, load: F) -> bool
5084where
5085 F: FnOnce(&str) -> Result<RunState>,
5086{
5087 if !matches!(task.status, TaskStatus::Parked | TaskStatus::Failed)
5089 || task.fresh_start
5090 || task.review_branch.is_some()
5091 {
5092 return false;
5093 }
5094 let Some(id) = task.runs.last() else {
5095 return false;
5096 };
5097 load(id).is_ok_and(|s| s.parked && crate::handover::can_resume(&s))
5098}
5099
5100#[derive(Debug, Clone, PartialEq, Eq)]
5103enum Starter {
5104 Review(String),
5107 Resume(String),
5109 Start,
5111}
5112
5113fn retained_attempt_hold(
5117 task: &Task,
5118 starter: &Starter,
5119 limit: usize,
5120 mut load: impl FnMut(&str) -> Result<RunState>,
5121) -> Option<String> {
5122 if limit == 0 || matches!(starter, Starter::Resume(_)) {
5123 return None;
5124 }
5125 let mut retained = 0;
5126 for id in task.runs.iter().collect::<std::collections::BTreeSet<_>>() {
5127 let state = match load(id) {
5128 Ok(state) if state.id == *id => state,
5129 _ => {
5130 return Some(format!(
5131 "[recovery-backlog] Cannot verify historical run {id}. Restore its record and inspect the recovery lineage before starting another attempt. Existing history was retained."
5132 ));
5133 }
5134 };
5135 if state.retired_to.is_some() || state.released() {
5136 continue;
5137 }
5138 retained += 1;
5141 }
5142 (retained >= limit).then(|| format!(
5143 "[recovery-backlog] {retained} unretired attempts reach the limit of {limit}. Resume an existing recovery owner or inspect the lineage and retire proved-obsolete attempts before creating another run. Releases, refunds and changed upstream inputs do not reset this limit. No source or historical report was removed."
5144 ))
5145}
5146
5147fn selected_starter(
5149 selected: &crate::pr_recovery::Selection,
5150 previous: Option<&RunState>,
5151 unfinished: Option<&str>,
5152) -> Starter {
5153 if let Some(id) = unfinished
5154 && previous.is_some_and(|r| {
5155 r.id == id && r.pr_recovery.as_ref() == Some(selected) && r.reviewed_commits.is_some()
5156 })
5157 {
5158 Starter::Resume(id.to_owned())
5159 } else {
5160 Starter::Review(selected.branch.clone())
5161 }
5162}
5163
5164fn take_divergence_answer(
5168 branch: &str,
5169 remote: &str,
5170 task: &mut Task,
5171) -> Option<crate::reconcile::Choice> {
5172 let summary = crate::reconcile::summary_for(branch, remote);
5173 let (idx, choice) = task.answers.iter().enumerate().rev().find_map(|(i, a)| {
5174 (a.question == summary)
5175 .then(|| crate::reconcile::Choice::from_answer(&a.answer))
5176 .flatten()
5177 .map(|c| (i, c))
5178 })?;
5179 task.answers.remove(idx);
5180 Some(choice)
5181}
5182
5183fn choose_starter(
5195 review_branch: Option<&str>,
5196 branch_exists: bool,
5197 unfinished: Option<&str>,
5198) -> Starter {
5199 match review_branch {
5200 Some(branch) if branch_exists => Starter::Review(branch.to_owned()),
5201 Some(_) => Starter::Start,
5202 None => match unfinished {
5203 Some(id) => Starter::Resume(id.to_owned()),
5204 None => Starter::Start,
5205 },
5206 }
5207}
5208
5209fn take_review_branch(task: &mut Task, previous: Option<&RunState>) -> Option<String> {
5211 let branch = task.review_branch.take();
5212 let instruction = task.review_instruction.take();
5213 branch.filter(|branch| match &instruction {
5214 Some(authored) => *authored == task.instruction,
5215 None => previous.map_or_else(
5218 || task.review_of.as_ref() == Some(branch),
5219 |run| compatible_run_instruction(run, task),
5220 ),
5221 })
5222}
5223
5224fn may_resume_run(run: &RunState, task: &Task) -> bool {
5227 task.resume_override
5228 .as_ref()
5229 .is_some_and(|r| r.pinned_run.as_ref() == Some(&run.id))
5230 || compatible_run_instruction(run, task)
5231}
5232
5233pub(crate) fn compatible_run_instruction(run: &RunState, task: &Task) -> bool {
5235 if let Some(authored) = &run.task_instruction {
5236 return compatible_instruction(authored, task);
5237 }
5238 if compatible_instruction(&run.instruction, task) {
5239 return true;
5240 }
5241 if run.reviewed_commits.is_some() {
5244 if let Some((authored, _)) = run.instruction.split_once("\n\n# Work already on branch `") {
5245 return strip_answers_block(authored, task).trim() == task.instruction.trim();
5246 }
5247 if let Some(branch) = &task.review_of {
5248 let canonical = format!(
5249 "Review the branch `{branch}` (review-only: no implementation, no judging)."
5250 );
5251 let generated = format!(
5252 "Review the work already on branch `{branch}`. There is no task statement: what the change claims to do is whatever its commits say.\n\n"
5253 );
5254 return task.instruction == canonical && run.instruction.starts_with(&generated);
5255 }
5256 }
5257 false
5258}
5259
5260pub(crate) fn compatible_instruction(old: &str, task: &Task) -> bool {
5262 strip_answers_block(old, task) == task.instruction
5263}
5264
5265fn repo_for(task: &Task, fallback: &Path) -> PathBuf {
5268 if task.repo.as_os_str().is_empty() || task.repo == Path::new(".") {
5269 return fallback.to_path_buf();
5270 }
5271 task.repo.clone()
5272}
5273
5274const ANSWERS_HEADER: &str = "\n\n# Operator answers\n\n";
5278
5279fn answers_block(task: &Task, count: usize) -> String {
5281 let mut s = ANSWERS_HEADER.to_owned();
5282 for a in &task.answers[..count] {
5283 s.push_str(&format!("- {}: {}\n", a.question, a.answer));
5284 }
5285 s
5286}
5287
5288fn resumption_block(r: &crate::queue::Resumption) -> String {
5291 format!(
5292 "\n\n# Explicit operator resumption\n\nGeneration {} at {} approved this instruction:\n{}\n\
5293 Earlier answers (first {}) are audit history; their waiting premises are superseded by this approval.\n",
5294 r.generation, r.at, r.instruction, r.answers_through,
5295 )
5296}
5297
5298fn append_answers(base: &str, task: &Task) -> String {
5300 let mut s = base.to_owned();
5301 if !task.answers.is_empty() {
5302 s.push_str(&answers_block(task, task.answers.len()));
5303 }
5304 if let Some(r) = &task.resumption {
5305 s.push_str(&resumption_block(r));
5306 }
5307 s
5308}
5309
5310fn strip_answers_block<'a>(instruction: &'a str, task: &Task) -> &'a str {
5314 let mut instruction = instruction;
5315 for approval in task
5316 .resumption
5317 .iter()
5318 .chain(task.resumption_history.iter().rev())
5319 {
5320 if let Some(base) = instruction.strip_suffix(&resumption_block(approval)) {
5321 instruction = base;
5322 break;
5323 }
5324 }
5325 for count in (1..=task.answers.len()).rev() {
5326 let block = answers_block(task, count);
5327 if let Some(base) = instruction.strip_suffix(&block) {
5328 return base;
5329 }
5330 }
5331 instruction
5332}
5333
5334fn instruction_for(task: &Task) -> String {
5342 append_answers(&task.instruction, task)
5343}
5344
5345fn resumed_instruction(old_instruction: &str, task: &Task) -> String {
5357 append_answers(strip_answers_block(old_instruction, task), task)
5358}
5359
5360fn record_config_failure(task: &mut Task, error: &anyhow::Error) {
5362 let failure = error.downcast_ref::<crate::config::ConfigFailure>();
5363 task.notice_cause(failure.map_or("config-load", |f| f.stage), None, None);
5364 if let Some(cause) = task.hold_notice_cause.as_mut() {
5365 cause.target = failure.map(|f| f.target.clone());
5366 }
5367}
5368
5369#[derive(Debug, thiserror::Error)]
5370#[error("attachment `{name}` is recorded on the task but {path} is missing")]
5371struct MissingAttachment {
5372 name: String,
5373 path: PathBuf,
5374}
5375
5376fn task_attachments(
5378 queue: &Queue,
5379 task: &Task,
5380) -> std::result::Result<Vec<PathBuf>, MissingAttachment> {
5381 let paths = queue.attachment_paths(task);
5382 for (name, path) in task.attachments.iter().zip(&paths) {
5383 if !path.is_file() {
5384 return Err(MissingAttachment {
5385 name: name.clone(),
5386 path: path.clone(),
5387 });
5388 }
5389 }
5390 Ok(paths)
5391}
5392
5393fn prepare_instruction(
5404 starter: &Starter,
5405 old_instruction: Option<&str>,
5406 task: &Task,
5407) -> Option<String> {
5408 match starter {
5409 Starter::Start => Some(instruction_for(task)),
5410 Starter::Resume(_) => Some(resumed_instruction(
5411 old_instruction.expect("a resumed run always has a prior instruction"),
5412 task,
5413 )),
5414 Starter::Review(_) => None,
5415 }
5416}
5417
5418fn record(queue: &Queue, task: &mut Task) {
5422 if let Err(e) = queue.put(task) {
5423 tracing::error!("could not record task {}: {e:#}", task.short());
5424 let notice = Notice::error(
5425 "loop:record",
5426 "The loop could not save a task's state; check the disk.",
5427 )
5428 .link(Link::Task {
5429 id: task.id.clone(),
5430 });
5431 match queue.root().parent().filter(|p| !p.as_os_str().is_empty()) {
5433 Some(home) => notices::raise_in(home, notice),
5434 None => notices::raise(notice),
5435 }
5436 }
5437}
5438
5439fn runnable(queue: &Queue) -> Vec<Task> {
5445 let mut tasks: Vec<Task> = queue
5446 .list()
5447 .into_iter()
5448 .filter(|t| t.status.runnable())
5449 .collect();
5450 tasks.sort_unstable_by(|a, b| b.priority.cmp(&a.priority).then(a.id.cmp(&b.id)));
5451 tasks
5452}
5453
5454fn describe(state: &RunState) -> String {
5468 let p = phrases(&state.config.graph.language);
5469 let mut detail = if state.status == RunStatus::Stalled {
5470 let mut seats: Vec<&str> = state.quota.iter().map(|q| q.seat.as_str()).collect();
5471 seats.sort_unstable();
5472 seats.dedup();
5473 if seats.is_empty() {
5474 p.quorum_lost.to_owned()
5475 } else {
5476 format!("{}{}{}", p.quorum_lost, p.quota_took_out, seats.join(", "))
5477 }
5478 } else {
5479 format!("{}{}", p.run_ended, state.status.display_label())
5480 };
5481 if let Some(last) = state.events.last() {
5482 let unanswered = state
5486 .reviews
5487 .last()
5488 .filter(|r| {
5489 state.status == RunStatus::Blocked
5490 && last.node == "review"
5491 && r.incomplete()
5492 && r.blocking == 0
5493 && r.round == state.config.graph.review_rounds
5494 && r.e2e.iter().all(crate::run::CommandOutcome::ok)
5495 })
5496 .map(|r| (r.expected - r.answered, r.round));
5497 match unanswered {
5498 Some((missing, rounds)) if crate::lang::is_japanese(&state.config.graph.language) => {
5499 detail.push_str(&format!(
5500 " ({}: {})",
5501 last.node,
5502 (p.reviewers_never_answered)(missing, rounds)
5503 ));
5504 }
5505 _ => detail.push_str(&format!(" ({}: {})", last.node, last.message)),
5506 }
5507 }
5508 detail.push_str(&format!(" [run {}]", state.id));
5509 detail
5510}
5511
5512const DIAGNOSTIC_MAX: usize = 4_000;
5518
5519const DIAGNOSTIC_OUTPUT_TAIL: usize = 800;
5524
5525fn diagnostic(state: &RunState) -> Option<String> {
5539 let mut parts: Vec<String> = Vec::new();
5540
5541 for o in state.gate.iter().filter(|o| !o.ok()) {
5543 parts.push(format!(
5544 "gate `{}` failed ({}):\n{}",
5545 o.command,
5546 notices::exit_status(o.code),
5547 crate::run::tail(&o.output_tail, DIAGNOSTIC_OUTPUT_TAIL)
5548 ));
5549 }
5550
5551 if let Some(last) = state
5554 .events
5555 .iter()
5556 .rev()
5557 .find(|e| e.node == "land" && e.message.contains("fixer produced no commit"))
5558 {
5559 parts.push(last.message.clone());
5560 }
5561
5562 if state.viable().is_empty() {
5569 for c in &state.candidates {
5570 if let Some(evidence) = &c.verified_noop {
5571 parts.push(format!(
5572 "candidate {} (agent-verified no-op, unconfirmed by magi): {evidence}",
5573 c.label
5574 ));
5575 } else if !c.summary.trim().is_empty() {
5576 parts.push(format!("candidate {}: {}", c.label, c.summary.trim()));
5577 } else if let Some(why) = &c.failed {
5578 parts.push(format!("candidate {}: {why}", c.label));
5579 }
5580 }
5581 }
5582
5583 if parts.is_empty() {
5584 return None;
5585 }
5586 Some(crate::run::tail(
5591 &parts.join("\n\n"),
5592 DIAGNOSTIC_MAX.saturating_sub(100),
5593 ))
5594}
5595
5596fn label(status: RunStatus) -> &'static str {
5604 status.as_str()
5605}
5606
5607pub(crate) fn merge_mode(mode: &str) -> Result<MergeMode> {
5609 match mode {
5610 "none" => Ok(MergeMode::None),
5611 "local" => Ok(MergeMode::Local),
5612 "pr" => Ok(MergeMode::Pr),
5613 other => bail!("unknown merge mode `{other}`; expected none, local or pr"),
5614 }
5615}
5616
5617fn lock<T>(mutex: &Mutex<T>) -> MutexGuard<'_, T> {
5622 mutex
5623 .lock()
5624 .unwrap_or_else(std::sync::PoisonError::into_inner)
5625}
5626
5627#[cfg(test)]
5628mod tests {
5629 use super::*;
5630 use crate::queue::{Source, TaskStatus};
5631 use crate::run::{Candidate, CommandOutcome};
5632 use pretty_assertions::assert_eq;
5633
5634 fn task() -> Task {
5635 Task::new(
5636 "add retries".to_owned(),
5637 "add retries".to_owned(),
5638 PathBuf::from("/repo"),
5639 Source::Human,
5640 )
5641 }
5642
5643 #[test]
5644 fn selected_public_head_overrides_historical_resume_and_pin() {
5645 let mut t = task();
5646 let mut old = run_state(RunStatus::Blocked);
5647 let selection = crate::pr_recovery::Selection {
5648 url: "url".into(),
5649 branch: format!("magi/task-{}", t.id),
5650 head: "a".repeat(40),
5651 source_head: None,
5652 source_run: None,
5653 source_proof: None,
5654 disposition: Some(crate::pr_recovery::Disposition::Review),
5655 instruction: t.instruction.clone(),
5656 approved_at: Timestamp::now(),
5657 };
5658 t.resume_override = Some(crate::queue::OperatorResume {
5659 pinned_run: Some(old.id.clone()),
5660 question_id: "choice".into(),
5661 at: Timestamp::now(),
5662 conductor_rehold: None,
5663 forced: false,
5664 });
5665 assert_eq!(
5666 selected_starter(&selection, Some(&old), Some(&old.id)),
5667 Starter::Review(selection.branch.clone())
5668 );
5669 old.pr_recovery = Some(selection.clone());
5670 assert_eq!(
5671 selected_starter(&selection, Some(&old), Some(&old.id)),
5672 Starter::Review(selection.branch.clone())
5673 );
5674 old.reviewed_commits = Some(vec!["implementation".into()]);
5675 assert_eq!(
5676 selected_starter(&selection, Some(&old), Some(&old.id)),
5677 Starter::Resume(old.id.clone())
5678 );
5679 let mut changed = selection.clone();
5680 changed.head = "b".repeat(40);
5681 assert_eq!(
5682 selected_starter(&changed, Some(&old), Some(&old.id)),
5683 Starter::Review(changed.branch)
5684 );
5685 }
5686
5687 #[test]
5688 fn candidate_transplant_failure_is_one_actionable_hold() {
5689 let dir = tempfile::tempdir().unwrap();
5690 let q = Queue::at(dir.path().join("queue"));
5691 let mut t = task();
5692 let mut state = run_state(RunStatus::Blocked);
5693 state.event("prep", "cannot apply referenced commit: conflict");
5694 state.prep_failure = Some("conflict".into());
5695 t.start(state.id.clone());
5696 finish_attempt(3, &q, &mut t, &state, &[], Err(anyhow::anyhow!("conflict")));
5697 assert_eq!(t.attempts, 1);
5698 assert_eq!(t.status, TaskStatus::Held);
5699 assert!(t.operator_held());
5700 assert!(!t.status.runnable());
5701 assert!(t.hold_reason.as_deref().unwrap().contains("recover-pr"));
5702 let first = t.chat_notice_key().unwrap();
5703 t.chat_notices.push(first.clone());
5704 t.release_operator("test", None);
5705 state.id = "another-unprepared-run".into();
5706 t.start(state.id.clone());
5707 finish_attempt(
5708 3,
5709 &q,
5710 &mut t,
5711 &state,
5712 &[],
5713 Err(anyhow::anyhow!("still conflicts")),
5714 );
5715 assert_eq!(
5716 t.chat_notice_key().as_ref(),
5717 Some(&first),
5718 "no work happened"
5719 );
5720 t.release_operator("test", None);
5721 state.prep_failure = None;
5722 state.status = RunStatus::Failed;
5723 state.begin_pass();
5724 t.start(state.id.clone());
5725 finish_attempt(
5726 1,
5727 &q,
5728 &mut t,
5729 &state,
5730 &[],
5731 Err(anyhow::anyhow!("new gate failure")),
5732 );
5733 let failure = t.chat_notice_key().unwrap();
5734 assert_ne!(failure, first, "approved recovery reached a real failure");
5735 t.release_operator("test", None);
5736 state.begin_pass();
5737 t.start(state.id.clone());
5738 finish_attempt(
5739 1,
5740 &q,
5741 &mut t,
5742 &state,
5743 &[],
5744 Err(anyhow::anyhow!("new failure on resume")),
5745 );
5746 assert_ne!(
5747 t.chat_notice_key().as_ref(),
5748 Some(&failure),
5749 "a new settled pass"
5750 );
5751 }
5752
5753 #[test]
5754 fn revised_resumption_invalidates_old_pins_and_preserves_unchanged_pins() {
5755 for source in ["cli", "web"] {
5756 let mut t = task();
5757 let mut run = run_state(RunStatus::Blocked);
5758 run.instruction = "Run A original criteria".into();
5759 run.task_instruction = Some(run.instruction.clone());
5760 t.start(run.id.clone());
5761 t.release_operator("choice", Some("resume-question"));
5762 t.resume_override = Some(crate::queue::OperatorResume {
5763 question_id: "resume-question".into(),
5764 at: Timestamp::now(),
5765 conductor_rehold: None,
5766 forced: true,
5767 pinned_run: Some(run.id.clone()),
5768 });
5769 t.hold_machine(Some("not enough free space to start a run".into()));
5770 let mut unchanged = t.clone();
5771 unchanged
5772 .edit("title changed".into(), unchanged.instruction.clone())
5773 .unwrap();
5774 unchanged.release_operator(source, None);
5775 assert_eq!(
5776 unchanged
5777 .resume_override
5778 .as_ref()
5779 .unwrap()
5780 .pinned_run
5781 .as_deref(),
5782 Some(run.id.as_str())
5783 );
5784 assert!(!compatible_run_instruction(&run, &unchanged));
5785 assert!(may_resume_run(&run, &unchanged));
5786
5787 t.edit(
5788 "approved recovery".into(),
5789 "Reuse measured PR head B; fix Clippy; review".into(),
5790 )
5791 .unwrap();
5792 assert!(t.resume_override.is_none());
5793 t.release_operator(source, None);
5794 let reloaded: Task = serde_json::from_str(&serde_json::to_string(&t).unwrap()).unwrap();
5795 assert!(reloaded.resume_override.is_none());
5796 assert!(!may_resume_run(&run, &reloaded));
5797 let unfinished =
5798 unfinished_run_with(&reloaded.runs, reloaded.short(), |_| Ok(run.clone()))
5799 .filter(|_| may_resume_run(&run, &reloaded));
5800 assert_eq!(
5801 choose_starter(None, false, unfinished.as_deref()),
5802 Starter::Start
5803 );
5804 assert!(instruction_for(&reloaded).starts_with("Reuse measured PR head B"));
5805 assert_eq!(
5806 reloaded.resumption_history[0].question_id.as_deref(),
5807 Some("resume-question")
5808 );
5809
5810 unchanged.instruction = "Reuse measured PR head B".into();
5813 unchanged.hold_machine(Some("disk hold".into()));
5814 unchanged.release_operator(source, None);
5815 assert!(unchanged.resume_override.is_none());
5816 assert!(!may_resume_run(&run, &unchanged));
5817 }
5818 }
5819
5820 #[test]
5821 fn resumption_preserves_generated_review_runs_and_rejects_revised_targets() {
5822 let mut t = task();
5823 let mut run = run_state(RunStatus::Blocked);
5824 t.review_of = Some("approved-head".into());
5825 t.instruction =
5826 "Review the branch `approved-head` (review-only: no implementation, no judging)."
5827 .into();
5828 run.instruction = "Review the work already on branch `approved-head`. There is no task statement: what the change claims to do is whatever its commits say.\n\nCommit log".into();
5829 run.reviewed_commits = Some(vec!["Commit log".into()]);
5830 t.start(run.id.clone());
5831 t.hold_machine(Some("interrupted review".into()));
5832 t.release_operator("cli", None);
5833 assert!(
5834 compatible_run_instruction(&run, &t),
5835 "legacy explicit review remains resumable"
5836 );
5837 let unfinished = unfinished_run_with(&t.runs, t.short(), |_| Ok(run.clone()))
5838 .filter(|_| compatible_run_instruction(&run, &t));
5839 assert_eq!(
5840 choose_starter(None, false, unfinished.as_deref()),
5841 Starter::Resume(run.id.clone())
5842 );
5843 run.task_instruction = Some(t.instruction.clone());
5844 assert!(compatible_run_instruction(&run, &t));
5845 let reloaded: RunState =
5846 serde_json::from_str(&serde_json::to_string(&run).unwrap()).unwrap();
5847 assert!(compatible_run_instruction(&reloaded, &t));
5848 t.edit(
5849 "new approved head".into(),
5850 "Reuse measured PR head B".into(),
5851 )
5852 .unwrap();
5853 assert!(!compatible_run_instruction(&run, &t));
5854
5855 t.review_of = None;
5857 t.instruction = "Fix Clippy on approved head".into();
5858 run.task_instruction = None;
5859 run.instruction = format!(
5860 "{}\n\n# Work already on branch `approved-head` (commit subjects)\n\nCommit log",
5861 t.instruction
5862 );
5863 assert!(compatible_run_instruction(&run, &t));
5864 t.instruction.push('\n');
5865 assert!(
5866 compatible_run_instruction(&run, &t),
5867 "legacy generated review trimmed the original criteria"
5868 );
5869 t.hold_for_handover(Some("approved-head".into()), "refused".into());
5870 let mut unchanged = t.clone();
5871 assert_eq!(
5872 take_review_branch(&mut unchanged, Some(&run)).as_deref(),
5873 Some("approved-head")
5874 );
5875 t.edit("new plan".into(), "Reuse measured PR head B".into())
5876 .unwrap();
5877 assert!(take_review_branch(&mut t, Some(&run)).is_none());
5878 assert!(!compatible_run_instruction(&run, &t));
5879 t.review_branch = Some("approved-head".into());
5880 t.review_instruction = None;
5881 assert!(
5882 take_review_branch(&mut t, Some(&run)).is_none(),
5883 "legacy retained candidate cannot replace a revised approved head"
5884 );
5885 }
5886
5887 #[test]
5888 fn resumption_actions_survive_automatic_holds_of_hand_started_runs() {
5889 crate::run::pin_test_home();
5890 let dir = tempfile::tempdir().unwrap();
5891 let queue = Queue::at(dir.path().join("queue"));
5892 for resume in [true, false] {
5893 let state = run_state(RunStatus::Blocked);
5894 let mut t = task();
5895 t.start(state.id.clone());
5896 let action = if resume {
5897 ask::ChoiceAction::Resume {
5898 run: state.id.clone(),
5899 }
5900 } else {
5901 ask::ChoiceAction::Requeue
5902 };
5903 let mut q = action_question(&state.id, action);
5904 q.status = ask::QuestionStatus::Open;
5905 q.answer = None;
5906 q.answered_at = None;
5907 t.fail("run stopped with a question", 3);
5908 hold_if_runnable(&queue, &mut t);
5909 assert!(t.operator_decided_at.is_none());
5910 assert!(t.operator_held());
5911 q.answer(ask::Answer::Choice(q.choices[0].clone())).unwrap();
5912 let expected = if resume {
5913 ActionDecision::Resume(state.id.clone())
5914 } else {
5915 ActionDecision::Requeue
5916 };
5917 assert_eq!(
5918 decide_action(&t, &q, phrases("en"), |_| Ok(state.clone())),
5919 expected
5920 );
5921 t.hold_manual(Some("later explicit hold".into()));
5922 assert_eq!(
5923 decide_action(&t, &q, phrases("en"), |_| Ok(state.clone())),
5924 ActionDecision::Stale
5925 );
5926 }
5927 }
5928
5929 #[test]
5930 fn explicit_resumption_keeps_revised_instruction_and_audit_answers() {
5931 let mut t = task();
5932 t.answers.push(crate::queue::AnsweredQuestion {
5933 question: "Inspect the remote?".into(),
5934 answer: "Inspect then wait".into(),
5935 });
5936 let old = instruction_for(&t);
5937 assert!(compatible_instruction(&old, &t));
5938 t.instruction = "Reuse approved PR head and review".into();
5939 t.release_operator("cli", None);
5940 assert!(!compatible_instruction(&old, &t));
5941 let approved = instruction_for(&t);
5942 assert!(approved.contains("waiting premises are superseded"));
5943 assert!(compatible_instruction(&approved, &t));
5944 assert_eq!(resumed_instruction(&approved, &t), approved);
5945 t.release_operator("web", None);
5946 assert!(compatible_instruction(&approved, &t));
5947 assert_eq!(t.resumption_history.len(), 1);
5948 t.start("run".into());
5949 assert!(instruction_for(&t).contains("Explicit operator resumption"));
5950 }
5951
5952 #[test]
5956 fn a_resource_blocked_run_holds_its_task_and_spends_no_attempt() {
5957 let mut state = RunState::new(
5958 PathBuf::from("/repo"),
5959 "main".to_owned(),
5960 "abc1234".to_owned(),
5961 "task".to_owned(),
5962 Config::default(),
5963 );
5964 state.status = RunStatus::Blocked;
5965 state.resource_block = Some(crate::run::ResourceBlock {
5966 node: "graft".to_owned(),
5967 kind: "shared build cache".to_owned(),
5968 detail: "timed out".to_owned(),
5969 at: jiff::Timestamp::now(),
5970 });
5971 let mut t = task();
5972 t.start("20260902-000000-aaaa".to_owned());
5973 assert_eq!(t.attempts, 1);
5974 let verdict = Verdict {
5975 status: RunStatus::Blocked,
5976 left_pr: false,
5977 quota_hit: false,
5978 parked: false,
5979 no_viable_candidates: false,
5980 };
5981 settle_and_diagnose(&mut t, verdict, "blocked", 2, &state);
5982 assert_eq!(t.status, TaskStatus::Held);
5983 assert_eq!(t.attempts, 0, "the attempt is refunded");
5984 assert!(t.hold_reason.as_deref().unwrap().starts_with("[resource] "));
5985 assert!(
5986 !t.operator_held(),
5987 "a machine hold, released without a human"
5988 );
5989
5990 let mut selected = task();
5991 selected.start("20260902-000000-cccc".into());
5992 selected.replacement = Some(crate::merge_decision::Replacement {
5993 question: "fixture".into(),
5994 run: "earlier".into(),
5995 pr: "https://example.invalid/pr/1".into(),
5996 agent: "selected".into(),
5997 context: "fixture".into(),
5998 });
5999 state.resource_block.as_mut().unwrap().kind = "repository coordination".into();
6000 settle_and_diagnose(&mut selected, verdict, "blocked", 2, &state);
6001 assert_eq!(selected.attempts, 0);
6002 assert!(
6003 !selected.operator_held(),
6004 "a selected agent also retains a machine resource hold"
6005 );
6006 assert!(
6007 selected
6008 .hold_reason
6009 .as_deref()
6010 .unwrap()
6011 .starts_with("[resource] ")
6012 );
6013
6014 let mut p = task();
6016 p.start("20260902-000000-bbbb".to_owned());
6017 settle_and_diagnose(
6018 &mut p,
6019 Verdict {
6020 parked: true,
6021 ..verdict
6022 },
6023 "parked",
6024 2,
6025 &state,
6026 );
6027 assert_eq!(p.status, TaskStatus::Parked);
6028 }
6029
6030 #[test]
6031 fn record_failure_notice_links_the_task() {
6032 let dir = tempfile::tempdir().unwrap();
6033 std::fs::write(dir.path().join("queue"), b"not a directory").unwrap();
6036 let queue = Queue::at(dir.path().join("queue"));
6037 let mut t = interrupt_task("rec-fail-1");
6038 record(&queue, &mut t);
6039
6040 let found = crate::notices::Notices::at(dir.path().join("notifications")).list();
6041 assert_eq!(found.len(), 1);
6042 assert_eq!(found[0].key, "loop:record");
6043 assert_eq!(
6044 found[0].link,
6045 Some(Link::Task {
6046 id: "rec-fail-1".to_owned()
6047 })
6048 );
6049 }
6050
6051 fn interrupt_task(id: &str) -> Task {
6054 let mut t = task();
6055 t.id = id.to_owned();
6056 t.interrupt = true;
6057 t
6058 }
6059
6060 fn task_with_id(id: &str) -> Task {
6062 let mut t = task();
6063 t.id = id.to_owned();
6064 t
6065 }
6066
6067 fn urgent_task(id: &str) -> Task {
6069 let mut t = task();
6070 t.id = id.to_owned();
6071 t.urgent = true;
6072 t
6073 }
6074
6075 #[test]
6076 fn urgent_dispatch_has_one_extra_slot_and_never_uses_a_stale_flag() {
6077 let dir = tempfile::tempdir().unwrap();
6078 let queue = Queue::at(dir.path().join("queue"));
6079 let ordinary = Arc::new(tokio::sync::Semaphore::new(1));
6080 let urgent = Arc::new(tokio::sync::Semaphore::new(1));
6081 let occupied = Arc::clone(&ordinary).try_acquire_owned().unwrap();
6082 let mut first = task();
6083 queue.put(&mut first).unwrap();
6084 let stale_plain = queue.get(&first.id).unwrap();
6085 first.set_urgent(true).unwrap();
6086 queue.put(&mut first).unwrap();
6087 let dispatched = claim_dispatch(&queue, &stale_plain, false, &ordinary, &urgent).unwrap();
6088 let mut second = urgent_task("second");
6089 queue.put(&mut second).unwrap();
6090 assert!(claim_dispatch(&queue, &second, false, &ordinary, &urgent).is_none());
6091 drop(queue.claim(&second.id).unwrap());
6092 assert_eq!(queue.get(&second.id).unwrap().attempts, 0);
6093 drop(dispatched);
6094 let stale_urgent = queue.get(&first.id).unwrap();
6095 first.set_urgent(false).unwrap();
6096 queue.put(&mut first).unwrap();
6097 assert!(claim_dispatch(&queue, &stale_urgent, false, &ordinary, &urgent).is_none());
6098 assert_eq!(urgent.available_permits(), 1);
6099 drop(queue.claim(&first.id).unwrap());
6100 drop(occupied);
6101 }
6102
6103 #[test]
6104 fn urgent_dispatch_uses_saved_lane_and_releases_unsuccessful_claims() {
6105 let dir = tempfile::tempdir().unwrap();
6106 let queue = Queue::at(dir.path().join("queue"));
6107 let ordinary = Arc::new(tokio::sync::Semaphore::new(1));
6108 let urgent = Arc::new(tokio::sync::Semaphore::new(1));
6109 let mut t = task();
6110 queue.put(&mut t).unwrap();
6111 for enabled in [true, false] {
6112 let stale = queue.get(&t.id).unwrap();
6113 t.set_urgent(enabled).unwrap();
6114 queue.put(&mut t).unwrap();
6115 assert_ne!(stale.urgent, enabled);
6116 let DispatchClaim {
6117 task: fresh,
6118 claim,
6119 permit,
6120 } = claim_dispatch(&queue, &stale, false, &ordinary, &urgent).unwrap();
6121 assert_eq!(fresh.urgent, enabled);
6122 assert_eq!(ordinary.available_permits(), usize::from(enabled));
6123 assert_eq!(urgent.available_permits(), usize::from(!enabled));
6124 assert!(queue.claim(&t.id).is_err());
6125 assert_eq!(fresh.attempts, stale.attempts);
6126 drop((claim, permit));
6127 assert_eq!(ordinary.available_permits(), 1);
6128 assert_eq!(urgent.available_permits(), 1);
6129 }
6130 t.set_urgent(true).unwrap();
6131 queue.put(&mut t).unwrap();
6132 let occupied = Arc::clone(&urgent).try_acquire_owned().unwrap();
6133 assert!(claim_dispatch(&queue, &t, false, &ordinary, &urgent).is_none());
6134 assert_eq!(ordinary.available_permits(), 1, "no ordinary fallback");
6135 drop(queue.claim(&t.id).unwrap());
6136 assert_eq!(queue.get(&t.id).unwrap().attempts, t.attempts);
6137 drop(occupied);
6138 assert!(claim_dispatch(&queue, &t, true, &ordinary, &urgent).is_none());
6139 drop(queue.claim(&t.id).unwrap());
6140 let held = queue.claim(&t.id).unwrap();
6141 assert!(claim_dispatch(&queue, &t, false, &ordinary, &urgent).is_none());
6142 drop(held);
6143 t.status = TaskStatus::Held;
6144 queue.put(&mut t).unwrap();
6145 assert!(claim_dispatch(&queue, &t, false, &ordinary, &urgent).is_none());
6146 drop(queue.claim(&t.id).unwrap());
6147 std::fs::remove_file(queue.path_of(&t.id)).unwrap();
6148 assert!(claim_dispatch(&queue, &t, false, &ordinary, &urgent).is_none());
6149 drop(queue.claim(&t.id).unwrap());
6150 assert_eq!(urgent.available_permits(), 1);
6151 }
6152
6153 #[test]
6158 fn permit_kind_prefers_a_land_resume_over_the_urgent_slot() {
6159 assert_eq!(permit_kind(true, false), PermitKind::None);
6160 assert_eq!(permit_kind(true, true), PermitKind::None);
6161 }
6162
6163 #[test]
6168 fn permit_kind_separates_urgent_from_ordinary() {
6169 assert_eq!(permit_kind(false, true), PermitKind::Urgent);
6170 assert_eq!(permit_kind(false, false), PermitKind::Ordinary);
6171 }
6172
6173 #[test]
6180 fn disk_gate_with_holds_a_task_below_the_threshold_and_names_both_numbers() {
6181 let cfg = Config::default();
6182 let repo = Path::new("/any/repo/path");
6183
6184 let reason =
6185 disk_gate_with(repo, &cfg, |_| Ok(1024)).expect("must hold below the threshold");
6186 assert!(reason.contains("1024"), "{reason}");
6187 assert!(
6188 reason.contains(&cfg.disk.min_free_bytes.to_string()),
6189 "{reason}"
6190 );
6191
6192 assert_eq!(
6193 disk_gate_with(repo, &cfg, |_| Ok(cfg.disk.min_free_bytes)),
6194 None,
6195 "exactly at the floor is open"
6196 );
6197 assert_eq!(
6198 disk_gate_with(repo, &cfg, |_| Ok(cfg.disk.min_free_bytes + 1)),
6199 None,
6200 "comfortably above the floor is open"
6201 );
6202 }
6203
6204 #[test]
6205 fn disk_gate_with_opens_unconditionally_when_the_operator_opted_out() {
6206 let mut cfg = Config::default();
6207 cfg.disk.min_free_bytes = 0;
6208 let repo = Path::new("/any/repo/path");
6209 assert_eq!(
6210 disk_gate_with(repo, &cfg, |_| Ok(0)),
6211 None,
6212 "a zero floor never measures at all"
6213 );
6214 }
6215
6216 #[test]
6217 fn disk_gate_with_closes_rather_than_starts_blind_when_it_cannot_measure() {
6218 let cfg = Config::default();
6219 let repo = Path::new("/any/repo/path");
6220 let reason = disk_gate_with(repo, &cfg, |_| Err(anyhow::anyhow!("no df on this box")))
6221 .expect("a measurement failure must close the gate, not open it");
6222 assert!(reason.contains("could not measure"), "{reason}");
6223 }
6224
6225 #[test]
6226 fn no_interrupt_task_leaves_the_sequence_idle_even_with_something_in_flight() {
6227 let ordinary = task();
6228 let next = advance_interrupt(
6229 Interrupt::Idle,
6230 std::slice::from_ref(&ordinary.id),
6231 std::slice::from_ref(&ordinary),
6232 );
6233 assert_eq!(next, Interrupt::Idle);
6234 }
6235
6236 #[test]
6237 fn an_interrupt_task_with_nothing_in_flight_never_starts_a_sequence() {
6238 let marked = interrupt_task("marked");
6241 let next = advance_interrupt(Interrupt::Idle, &[], std::slice::from_ref(&marked));
6242 assert_eq!(next, Interrupt::Idle);
6243 }
6244
6245 #[test]
6246 fn an_interrupt_task_with_something_in_flight_starts_parking_it() {
6247 let marked = interrupt_task("marked");
6248 let next = advance_interrupt(
6249 Interrupt::Idle,
6250 &["running".to_owned()],
6251 std::slice::from_ref(&marked),
6252 );
6253 assert_eq!(
6254 next,
6255 Interrupt::Parking {
6256 parked: vec!["running".to_owned()],
6257 interrupt_task: "marked".to_owned(),
6258 }
6259 );
6260 }
6261
6262 #[test]
6271 fn more_than_one_run_in_flight_never_starts_an_interrupt_sequence() {
6272 let marked = interrupt_task("marked");
6273
6274 let two = advance_interrupt(
6275 Interrupt::Idle,
6276 &["a".to_owned(), "b".to_owned()],
6277 std::slice::from_ref(&marked),
6278 );
6279 assert_eq!(two, Interrupt::Idle);
6280
6281 let none = advance_interrupt(Interrupt::Idle, &[], std::slice::from_ref(&marked));
6282 assert_eq!(none, Interrupt::Idle, "nothing to interrupt either");
6283 }
6284
6285 #[test]
6286 fn parking_holds_until_every_parked_id_has_actually_left_flight() {
6287 let state = Interrupt::Parking {
6288 parked: vec!["running".to_owned()],
6289 interrupt_task: "marked".to_owned(),
6290 };
6291 let still_going = advance_interrupt(state.clone(), &["running".to_owned()], &[]);
6293 assert_eq!(still_going, state);
6294
6295 let stopped_but_not_yet_dispatched =
6299 advance_interrupt(state.clone(), &[], &[interrupt_task("marked")]);
6300 assert_eq!(stopped_but_not_yet_dispatched, state);
6301
6302 let dispatched = advance_interrupt(state, &["marked".to_owned()], &[]);
6304 assert_eq!(
6305 dispatched,
6306 Interrupt::Running {
6307 parked: vec!["running".to_owned()],
6308 interrupt_task: "marked".to_owned(),
6309 }
6310 );
6311 }
6312
6313 #[test]
6314 fn the_sequence_moves_to_resuming_the_instant_the_interrupt_tasks_own_run_leaves_flight() {
6315 let state = Interrupt::Running {
6316 parked: vec!["running".to_owned()],
6317 interrupt_task: "marked".to_owned(),
6318 };
6319 let still_running = advance_interrupt(state.clone(), &["marked".to_owned()], &[]);
6320 assert_eq!(still_running, state);
6321
6322 let ended = advance_interrupt(state, &[], &[task_with_id("running")]);
6329 assert_eq!(
6330 ended,
6331 Interrupt::Resuming {
6332 parked: vec!["running".to_owned()]
6333 }
6334 );
6335 }
6336
6337 #[test]
6338 fn resuming_ends_the_instant_a_parked_task_is_seen_in_flight() {
6339 let state = Interrupt::Resuming {
6340 parked: vec!["running".to_owned()],
6341 };
6342 let still_waiting = advance_interrupt(state.clone(), &[], &[task_with_id("running")]);
6343 assert_eq!(still_waiting, state);
6344
6345 let dispatched = advance_interrupt(state, &["running".to_owned()], &[]);
6346 assert_eq!(dispatched, Interrupt::Idle);
6347 }
6348
6349 #[test]
6355 fn an_interrupt_task_that_stops_being_runnable_abandons_the_wait_without_losing_the_parked_run()
6356 {
6357 let state = Interrupt::Parking {
6358 parked: vec!["running".to_owned()],
6359 interrupt_task: "marked".to_owned(),
6360 };
6361 let next = advance_interrupt(state, &[], &[]);
6364 assert_eq!(
6365 next,
6366 Interrupt::Resuming {
6367 parked: vec!["running".to_owned()]
6368 },
6369 "abandoning the interrupt must not abandon the resume it owes"
6370 );
6371 }
6372
6373 #[test]
6376 fn resuming_abandons_a_parked_task_that_stops_being_runnable() {
6377 let state = Interrupt::Resuming {
6378 parked: vec!["running".to_owned()],
6379 };
6380 let next = advance_interrupt(state, &[], &[]);
6381 assert_eq!(
6382 next,
6383 Interrupt::Idle,
6384 "nothing is left to wait for; the loop must not stay wedged"
6385 );
6386 }
6387
6388 #[test]
6389 fn disabled_by_config_the_sequence_can_never_leave_idle() {
6390 let marked = interrupt_task("marked");
6391 let next = advance_interrupt_tick(
6392 false,
6393 Interrupt::Idle,
6394 &["running".to_owned()],
6395 std::slice::from_ref(&marked),
6396 );
6397 assert_eq!(
6398 next,
6399 Interrupt::Idle,
6400 "an unmarked, unconfigured daemon must behave exactly as before"
6401 );
6402 }
6403
6404 #[test]
6405 fn the_gate_blocks_everyone_while_something_parked_is_still_in_flight() {
6406 let state = Interrupt::Parking {
6407 parked: vec!["running".to_owned()],
6408 interrupt_task: "marked".to_owned(),
6409 };
6410 let candidates = vec![interrupt_task("marked"), task()];
6411 let allowed = interrupt_gate(&state, &["running".to_owned()], candidates);
6412 assert!(
6413 allowed.is_empty(),
6414 "nothing may dispatch - not even the interrupt task itself - \
6415 until the parked run has actually stopped"
6416 );
6417 }
6418
6419 #[test]
6433 fn urgent_gains_no_exemption_from_an_active_interrupt_sequence() {
6434 for state in [
6435 Interrupt::Parking {
6436 parked: vec!["running".to_owned()],
6437 interrupt_task: "marked".to_owned(),
6438 },
6439 Interrupt::Running {
6440 parked: vec!["running".to_owned()],
6441 interrupt_task: "marked".to_owned(),
6442 },
6443 Interrupt::Resuming {
6444 parked: vec!["running".to_owned()],
6445 },
6446 ] {
6447 let candidates = vec![
6448 interrupt_task("marked"),
6449 urgent_task("hot"),
6450 task_with_id("ordinary"),
6451 ];
6452 let allowed = interrupt_gate(&state, &["running".to_owned()], candidates);
6453 assert!(
6454 !allowed.iter().any(|t| t.id == "hot"),
6455 "an urgent candidate must wait out the same gate as anything \
6456 else while the run it would run alongside has not actually \
6457 left flight, for state {state:?}: {allowed:?}"
6458 );
6459 }
6460 }
6461
6462 #[test]
6468 fn a_task_marked_both_urgent_and_interrupt_is_admitted_once_the_gate_itself_says_so() {
6469 let state = Interrupt::Resuming {
6470 parked: vec!["hot".to_owned()],
6471 };
6472 let candidates = vec![urgent_task("hot"), task()];
6473 let allowed = interrupt_gate(&state, &[], candidates);
6474 assert_eq!(
6475 allowed.iter().filter(|t| t.id == "hot").count(),
6476 1,
6477 "the gate's own decision is unaffected by the urgent flag: {allowed:?}"
6478 );
6479 }
6480
6481 #[test]
6482 fn the_gate_lets_only_the_interrupt_task_through_once_parked_work_has_stopped() {
6483 let state = Interrupt::Parking {
6484 parked: vec!["running".to_owned()],
6485 interrupt_task: "marked".to_owned(),
6486 };
6487 let other = task();
6488 let candidates = vec![interrupt_task("marked"), other.clone()];
6489 let allowed = interrupt_gate(&state, &[], candidates);
6490 assert_eq!(allowed.len(), 1);
6491 assert_eq!(allowed[0].id, "marked");
6492 }
6493
6494 #[test]
6495 fn the_gate_blocks_everyone_while_the_interrupt_task_itself_is_in_flight() {
6496 let state = Interrupt::Running {
6497 parked: vec!["running".to_owned()],
6498 interrupt_task: "marked".to_owned(),
6499 };
6500 let candidates = vec![task(), task()];
6501 let allowed = interrupt_gate(&state, &["marked".to_owned()], candidates);
6502 assert!(allowed.is_empty());
6503 }
6504
6505 #[test]
6512 fn the_gate_offers_at_most_one_candidate_while_resuming_even_with_two_parked() {
6513 let state = Interrupt::Resuming {
6514 parked: vec!["a".to_owned(), "c".to_owned()],
6515 };
6516 let candidates = vec![task_with_id("a"), task_with_id("c"), task_with_id("other")];
6517 let allowed = interrupt_gate(&state, &[], candidates);
6518 assert_eq!(
6519 allowed.len(),
6520 1,
6521 "at most one candidate may be offered while resuming: {allowed:?}"
6522 );
6523 assert_eq!(allowed[0].id, "a");
6524 }
6525
6526 #[test]
6527 fn the_gate_offers_nothing_while_resuming_if_no_parked_task_is_runnable() {
6528 let state = Interrupt::Resuming {
6529 parked: vec!["a".to_owned()],
6530 };
6531 let allowed = interrupt_gate(&state, &[], vec![task_with_id("other")]);
6532 assert!(allowed.is_empty());
6533 }
6534
6535 #[test]
6541 fn a_full_sequence_never_gates_two_runs_through_at_once_and_resumes_exactly_one() {
6542 let running = task(); let marked = interrupt_task("marked");
6544
6545 let mut state = Interrupt::Idle;
6546 let in_flight = vec![running.id.clone()];
6548 state = advance_interrupt_tick(true, state, &in_flight, std::slice::from_ref(&marked));
6549 let gated = interrupt_gate(&state, &in_flight, vec![marked.clone(), running.clone()]);
6550 assert!(gated.is_empty(), "still waiting on `running` to park");
6551
6552 state = advance_interrupt_tick(true, state, &[], &[marked.clone(), running.clone()]);
6554 let gated = interrupt_gate(&state, &[], vec![marked.clone(), running.clone()]);
6555 assert_eq!(
6556 gated.iter().map(|t| t.id.as_str()).collect::<Vec<_>>(),
6557 vec!["marked"],
6558 "only the interrupt task may be offered to the dispatcher now"
6559 );
6560
6561 state = advance_interrupt_tick(
6563 true,
6564 state,
6565 &["marked".to_owned()],
6566 std::slice::from_ref(&running),
6567 );
6568 let gated = interrupt_gate(
6569 &state,
6570 &["marked".to_owned()],
6571 vec![marked.clone(), running.clone()],
6572 );
6573 assert!(
6574 gated.is_empty(),
6575 "the parked run must not be offered back while the interrupt \
6576 task is still running"
6577 );
6578
6579 let other = task_with_id("other");
6583 state = advance_interrupt_tick(true, state, &[], &[running.clone(), other.clone()]);
6584 assert_eq!(
6585 state,
6586 Interrupt::Resuming {
6587 parked: vec![running.id.clone()]
6588 }
6589 );
6590 let gated = interrupt_gate(&state, &[], vec![other.clone(), running.clone()]);
6591 assert_eq!(
6592 gated.iter().map(|t| t.id.as_str()).collect::<Vec<_>>(),
6593 vec![running.id.as_str()],
6594 "exactly the parked run resumes - not the unrelated task, even \
6595 though it was offered first"
6596 );
6597
6598 state = advance_interrupt_tick(
6602 true,
6603 state,
6604 std::slice::from_ref(&running.id),
6605 std::slice::from_ref(&other),
6606 );
6607 assert_eq!(state, Interrupt::Idle);
6608 let gated = interrupt_gate(
6609 &state,
6610 std::slice::from_ref(&running.id),
6611 vec![other.clone()],
6612 );
6613 assert_eq!(
6614 gated.iter().map(|t| t.id.as_str()).collect::<Vec<_>>(),
6615 vec![other.id.as_str()],
6616 "ordinary dispatch is unrestricted again"
6617 );
6618 }
6619
6620 #[test]
6621 fn every_run_status_settles_the_task_it_came_from() {
6622 let table = [
6624 (RunStatus::Merged, TaskStatus::Done, 1),
6625 (RunStatus::Ready, TaskStatus::Done, 1),
6626 (RunStatus::Stalled, TaskStatus::Failed, 0),
6627 (RunStatus::Blocked, TaskStatus::Failed, 1),
6628 (RunStatus::Failed, TaskStatus::Failed, 1),
6629 (RunStatus::VerifiedNoop, TaskStatus::Held, 1),
6630 (RunStatus::Prep, TaskStatus::Failed, 1),
6631 (RunStatus::Implementing, TaskStatus::Failed, 1),
6632 (RunStatus::Judging, TaskStatus::Failed, 1),
6633 (RunStatus::Deliberating, TaskStatus::Failed, 1),
6634 (RunStatus::Voting, TaskStatus::Failed, 1),
6635 (RunStatus::Reviewing, TaskStatus::Failed, 1),
6636 (RunStatus::Gating, TaskStatus::Failed, 1),
6637 ];
6638 for (run, want, attempts) in table {
6639 let mut t = task();
6640 t.start("20260902-000000-aaaa".to_owned());
6641 settle(
6642 &mut t,
6643 Verdict {
6644 status: run,
6645 left_pr: false,
6646 parked: false,
6647 quota_hit: matches!(run, RunStatus::Stalled),
6648 no_viable_candidates: false,
6649 },
6650 "why",
6651 2,
6652 );
6653 assert_eq!(t.status, want, "task status after {}", label(run));
6654 assert_eq!(t.attempts, attempts, "attempts after {}", label(run));
6655 }
6656 }
6657
6658 #[test]
6659 fn a_quota_stall_costs_the_task_no_attempt_but_a_block_does() {
6660 let mut stalled = task();
6661 stalled.start("20260902-000000-aaaa".to_owned());
6662 settle(
6663 &mut stalled,
6664 Verdict {
6665 status: RunStatus::Stalled,
6666 left_pr: false,
6667 parked: false,
6668 quota_hit: true,
6669 no_viable_candidates: false,
6670 },
6671 "quota",
6672 1,
6673 );
6674 assert_eq!(stalled.attempts, 0);
6675 assert!(
6676 stalled.status.runnable(),
6677 "a machine problem must leave the task in line"
6678 );
6679
6680 let mut blocked = task();
6681 blocked.start("20260902-000000-aaaa".to_owned());
6682 settle(
6683 &mut blocked,
6684 Verdict {
6685 status: RunStatus::Blocked,
6686 left_pr: false,
6687 parked: false,
6688 quota_hit: false,
6689 no_viable_candidates: false,
6690 },
6691 "findings open",
6692 1,
6693 );
6694 assert_eq!(blocked.attempts, 1);
6695 assert_eq!(
6696 blocked.status,
6697 TaskStatus::Held,
6698 "the last attempt hands the task to a human"
6699 );
6700 }
6701
6702 #[test]
6703 fn a_run_that_opened_a_pull_request_is_never_re_competed() {
6704 let mut delivered = task();
6707 delivered.start("20260903-080619-01c2".to_owned());
6708 settle(
6709 &mut delivered,
6710 Verdict {
6711 status: RunStatus::Blocked,
6712 left_pr: true,
6713 parked: false,
6714 quota_hit: false,
6715 no_viable_candidates: false,
6716 },
6717 "no check status",
6718 4,
6719 );
6720 assert_eq!(
6721 delivered.status,
6722 TaskStatus::Held,
6723 "a pull request waiting on CI or a person is not a retryable failure"
6724 );
6725 assert!(
6726 !delivered.status.runnable(),
6727 "the loop must not pick this task up again"
6728 );
6729 assert_eq!(
6730 delivered.last_error.as_deref(),
6731 Some("no check status"),
6732 "the operator needs to be told what the gate was waiting for"
6733 );
6734
6735 let mut empty_handed = task();
6738 empty_handed.start("20260903-080619-01c2".to_owned());
6739 settle(
6740 &mut empty_handed,
6741 Verdict {
6742 status: RunStatus::Blocked,
6743 left_pr: false,
6744 parked: false,
6745 quota_hit: false,
6746 no_viable_candidates: false,
6747 },
6748 "findings open",
6749 4,
6750 );
6751 assert_eq!(empty_handed.status, TaskStatus::Failed);
6752 assert!(empty_handed.status.runnable());
6753 }
6754
6755 #[test]
6756 fn a_verified_noop_run_hands_off_rather_than_closing_or_auto_retrying() {
6757 let mut noop = task();
6763 noop.start("20260912-131304-391f".to_owned());
6764 settle(
6765 &mut noop,
6766 Verdict {
6767 status: RunStatus::VerifiedNoop,
6768 left_pr: false,
6769 parked: false,
6770 quota_hit: false,
6771 no_viable_candidates: true,
6772 },
6773 "candidate A: already fixed by b32cfc4, on main",
6774 4,
6775 );
6776 assert_eq!(
6777 noop.status,
6778 TaskStatus::Held,
6779 "an unverified claim is a request for a human, not a failure"
6780 );
6781 assert!(
6782 !noop.status.runnable(),
6783 "the loop must not requeue this on the same unverified claim"
6784 );
6785 assert_eq!(noop.attempts, 1);
6790 }
6791
6792 #[test]
6793 fn parking_costs_the_task_no_attempt_and_leaves_it_in_line() {
6794 let mut parked = task();
6799 parked.start("20260903-183634-2d98".to_owned());
6800 settle(
6801 &mut parked,
6802 Verdict {
6803 status: RunStatus::Implementing,
6804 left_pr: false,
6805 quota_hit: false,
6806 parked: true,
6807 no_viable_candidates: false,
6808 },
6809 "parked after `implementing`",
6810 2,
6811 );
6812 assert_eq!(parked.attempts, 0, "a park is refunded");
6813 assert!(
6814 parked.status.runnable(),
6815 "and the task stays in line so the next loop resumes its run"
6816 );
6817 assert_eq!(parked.status, TaskStatus::Parked);
6818 assert_eq!(
6819 parked.park_reason.as_deref(),
6820 Some("parked after `implementing`"),
6821 "the card says where it stopped"
6822 );
6823 assert_eq!(parked.last_error, None, "nothing failed");
6824
6825 let mut broken = task();
6829 broken.start("20260903-183634-2d98".to_owned());
6830 settle(
6831 &mut broken,
6832 Verdict {
6833 status: RunStatus::Implementing,
6834 left_pr: false,
6835 quota_hit: false,
6836 parked: false,
6837 no_viable_candidates: false,
6838 },
6839 "returned mid-flight",
6840 2,
6841 );
6842 assert_eq!(broken.attempts, 1);
6843 }
6844
6845 #[test]
6846 fn only_a_rate_limit_buys_the_task_its_attempt_back() {
6847 let mut flaky = task();
6852 flaky.start("20260903-123023-e633".to_owned());
6853 settle(
6854 &mut flaky,
6855 Verdict {
6856 status: RunStatus::Stalled,
6857 left_pr: false,
6858 parked: false,
6859 quota_hit: false,
6860 no_viable_candidates: false,
6861 },
6862 "verdict rests on 1 of 3 judges",
6863 2,
6864 );
6865 assert_eq!(
6866 flaky.attempts, 1,
6867 "flakiness spends an attempt, so `max_attempts` still bounds it"
6868 );
6869 assert!(flaky.status.runnable(), "and it is still worth retrying");
6870
6871 let mut limited = task();
6873 limited.start("20260903-123023-e633".to_owned());
6874 settle(
6875 &mut limited,
6876 Verdict {
6877 status: RunStatus::Stalled,
6878 left_pr: false,
6879 parked: false,
6880 quota_hit: true,
6881 no_viable_candidates: false,
6882 },
6883 "judge-2, judge-3 out of quota",
6884 2,
6885 );
6886 assert_eq!(limited.attempts, 0, "a quota window is refunded");
6887 assert!(limited.status.runnable());
6888
6889 let mut worn = task();
6892 for _ in 0..2 {
6893 worn.release();
6894 }
6895 worn.start("20260903-123023-e633".to_owned());
6896 worn.attempts = 2;
6897 settle(
6898 &mut worn,
6899 Verdict {
6900 status: RunStatus::Stalled,
6901 left_pr: false,
6902 parked: false,
6903 quota_hit: false,
6904 no_viable_candidates: false,
6905 },
6906 "no quorum again",
6907 2,
6908 );
6909 assert_eq!(worn.status, TaskStatus::Held);
6910 assert!(!worn.status.runnable());
6911 }
6912
6913 #[test]
6914 fn a_quota_wipeout_that_leaves_nothing_to_judge_also_costs_no_attempt() {
6915 let mut wiped_out = task();
6922 wiped_out.start("20260907-025000-a1b2".to_owned());
6923 settle(
6924 &mut wiped_out,
6925 Verdict {
6926 status: RunStatus::Failed,
6927 left_pr: false,
6928 parked: false,
6929 quota_hit: true,
6930 no_viable_candidates: true,
6931 },
6932 "no candidate produced a change; nothing to judge",
6933 2,
6934 );
6935 assert_eq!(wiped_out.attempts, 0, "a total quota wipeout is refunded");
6936 assert!(
6937 wiped_out.status.runnable(),
6938 "a machine problem must leave the task in line"
6939 );
6940
6941 let mut partial_progress = task();
6947 partial_progress.start("20260907-025500-c3d4".to_owned());
6948 settle(
6949 &mut partial_progress,
6950 Verdict {
6951 status: RunStatus::Failed,
6952 left_pr: false,
6953 parked: false,
6954 quota_hit: true,
6955 no_viable_candidates: false,
6956 },
6957 "gate failed on the winning candidate",
6958 2,
6959 );
6960 assert_eq!(
6961 partial_progress.attempts, 1,
6962 "a candidate that actually produced a change spends the attempt \
6963 even though some other seat hit its quota"
6964 );
6965 assert!(partial_progress.status.runnable());
6966 }
6967
6968 #[test]
6969 fn reclaim_refunds_a_recovered_quota_wipeout_the_same_way_a_live_settle_does() {
6970 let mut t = task();
6976 t.start("20260907-025000-a1b2".to_owned());
6977 let mut state = run_state(RunStatus::Failed);
6978 state.quota.push(QuotaLoss {
6979 seat: "cand-a".to_owned(),
6980 node: "implement".to_owned(),
6981 at: Timestamp::now(),
6982 reset: None,
6983 });
6984 assert!(
6985 state.viable().is_empty(),
6986 "no candidate was added, so nothing is viable"
6987 );
6988 reclaim(&mut t, Some(state), 2, "en");
6989 assert_eq!(t.attempts, 0, "a recovered quota wipeout is refunded");
6990 assert!(t.status.runnable());
6991 }
6992
6993 #[test]
6994 fn a_held_task_is_never_offered_to_the_loop() {
6995 let dir = tempfile::tempdir().unwrap();
6996 let queue = Queue::at(dir.path().to_path_buf());
6997 for (n, priority) in [(1, 0), (2, 5), (3, 5)] {
6998 let mut t = task();
6999 t.id = format!("2026090{n}-000000-000{n}");
7000 t.priority = priority;
7001 queue.put(&mut t).unwrap();
7002 }
7003 let mut held = task();
7004 held.id = "20260909-000000-9999".to_owned();
7005 held.priority = 99;
7006 held.hold_machine(None);
7007 queue.put(&mut held).unwrap();
7008
7009 let order: Vec<String> = runnable(&queue).into_iter().map(|t| t.id).collect();
7010 assert_eq!(order.len(), 3);
7011 assert!(!order.contains(&held.id));
7012 assert_eq!(
7013 order.first().cloned(),
7014 queue.next_runnable().map(|t| t.id),
7015 "the loop's first candidate is exactly what the queue offers"
7016 );
7017 assert_eq!(
7018 order,
7019 vec![
7020 "20260902-000000-0002".to_owned(),
7021 "20260903-000000-0003".to_owned(),
7022 "20260901-000000-0001".to_owned(),
7023 ],
7024 "priority first, then oldest, so nothing starves"
7025 );
7026 }
7027
7028 #[test]
7029 fn sweep_removes_an_old_unparseable_lock_and_keeps_a_live_one() {
7030 let dir = tempfile::tempdir().unwrap();
7031 let queue = Queue::at(dir.path().to_path_buf());
7032 let mut old = task();
7033 old.id = "20260101-000000-old0".to_owned();
7034 queue.put(&mut old).unwrap();
7035 let mut fresh = task();
7036 fresh.id = "20260101-000000-new0".to_owned();
7037 queue.put(&mut fresh).unwrap();
7038
7039 std::fs::write(dir.path().join(format!("{}.lock", old.id)), "not a pid").unwrap();
7043 std::thread::sleep(Duration::from_millis(60));
7044 let live = queue.claim(&fresh.id).unwrap();
7045
7046 let swept = sweep_stale_claims(&queue, Duration::from_millis(50));
7047 assert_eq!(swept, vec![old.id.clone()]);
7048 assert!(
7049 queue.claim(&old.id).is_ok(),
7050 "an unparseable lock older than the threshold is swept"
7051 );
7052 assert!(
7053 queue.claim(&fresh.id).is_err(),
7054 "a live pid protects its lock regardless of age"
7055 );
7056 drop(live);
7057 }
7058
7059 #[test]
7060 fn an_old_lock_whose_pid_is_still_alive_is_never_swept_by_age_alone() {
7061 let dir = tempfile::tempdir().unwrap();
7071 let queue = Queue::at(dir.path().to_path_buf());
7072 let mut t = task();
7073 t.id = "20260101-000000-live".to_owned();
7074 queue.put(&mut t).unwrap();
7075
7076 let claim = queue.claim(&t.id).unwrap();
7077 std::thread::sleep(Duration::from_millis(60));
7078
7079 let swept = sweep_stale_claims(&queue, Duration::from_millis(50));
7080 assert!(
7081 swept.is_empty(),
7082 "a lock naming a live pid must never be swept by age, no matter how old: {swept:?}"
7083 );
7084 assert!(
7085 queue.claim(&t.id).is_err(),
7086 "the lock still protects its task"
7087 );
7088 drop(claim);
7089 }
7090
7091 fn injected_dead_pid() -> u32 {
7094 std::process::id().checked_add(1).unwrap_or(1)
7095 }
7096
7097 #[test]
7098 fn a_lock_naming_a_dead_pid_is_swept_at_once_regardless_of_age() {
7099 let dir = tempfile::tempdir().unwrap();
7100 let queue = Queue::at(dir.path().to_path_buf());
7101 let mut t = task();
7102 t.id = "20260101-000000-dead".to_owned();
7103 queue.put(&mut t).unwrap();
7104 let dead_pid = injected_dead_pid();
7105
7106 std::fs::write(
7111 dir.path().join(format!("{}.lock", t.id)),
7112 dead_pid.to_string(),
7113 )
7114 .unwrap();
7115
7116 let swept = sweep_stale_claims_with(&queue, Duration::from_secs(6 * 60 * 60), |pid| {
7117 pid != dead_pid
7118 });
7119 assert_eq!(
7120 swept,
7121 vec![t.id.clone()],
7122 "a dead owner is reclaimed immediately, not after STALE_CLAIM"
7123 );
7124 assert!(queue.claim(&t.id).is_ok(), "the task is claimable again");
7125 }
7126
7127 #[test]
7128 fn sweeping_on_every_poll_catches_a_lock_that_appears_after_the_first_sweep() {
7129 let dir = tempfile::tempdir().unwrap();
7130 let queue = Queue::at(dir.path().to_path_buf());
7131 let mut t = task();
7132 t.id = "20260101-000000-late".to_owned();
7133 queue.put(&mut t).unwrap();
7134 let dead_pid = injected_dead_pid();
7135
7136 assert!(
7139 sweep_stale_claims(&queue, Duration::from_secs(6 * 60 * 60)).is_empty(),
7140 "nothing has claimed the task yet"
7141 );
7142
7143 std::fs::write(
7146 dir.path().join(format!("{}.lock", t.id)),
7147 dead_pid.to_string(),
7148 )
7149 .unwrap();
7150
7151 let swept = sweep_stale_claims_with(&queue, Duration::from_secs(6 * 60 * 60), |pid| {
7155 pid != dead_pid
7156 });
7157 assert_eq!(swept, vec![t.id.clone()]);
7158 }
7159
7160 #[test]
7161 fn a_running_task_behind_a_dead_daemons_lock_recovers_once_swept_and_keeps_its_history() {
7162 crate::run::pin_test_home();
7167 let dir = tempfile::tempdir().unwrap();
7168 let queue = Queue::at(dir.path().to_path_buf());
7169 let mut t = task();
7170 t.id = "20260101-000000-crsh".to_owned();
7171 t.status = TaskStatus::Running;
7172 t.attempts = 1;
7173 t.runs.push("20260904-000000-4043".to_owned());
7177 queue.put(&mut t).unwrap();
7178 let dead_pid = injected_dead_pid();
7179
7180 std::fs::write(
7183 dir.path().join(format!("{}.lock", t.id)),
7184 dead_pid.to_string(),
7185 )
7186 .unwrap();
7187
7188 assert!(reclaim_orphaned_running(&queue, 2).is_empty());
7194 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Running);
7195
7196 let swept = sweep_stale_claims_with(&queue, Duration::from_secs(6 * 60 * 60), |pid| {
7197 pid != dead_pid
7198 });
7199 assert_eq!(swept, vec![t.id.clone()]);
7200
7201 let reclaimed = reclaim_orphaned_running(&queue, 2);
7202 assert_eq!(reclaimed, vec![t.id.clone()]);
7203 let after = queue.get(&t.id).unwrap();
7204 assert_eq!(
7205 after.status,
7206 TaskStatus::Held,
7207 "no run.json to recover from, so a human is asked"
7208 );
7209 assert_eq!(
7210 after.runs,
7211 vec!["20260904-000000-4043".to_owned()],
7212 "the crashed run's id is kept as evidence, not discarded"
7213 );
7214 }
7215
7216 #[test]
7217 fn a_lock_is_kept_when_the_process_query_is_unavailable() {
7218 let dir = tempfile::tempdir().unwrap();
7219 let queue = Queue::at(dir.path().to_path_buf());
7220 let mut t = task();
7221 t.id = "20260101-000000-unknown".to_owned();
7222 queue.put(&mut t).unwrap();
7223 let dead_pid = injected_dead_pid();
7224 std::fs::write(
7225 dir.path().join(format!("{}.lock", t.id)),
7226 dead_pid.to_string(),
7227 )
7228 .unwrap();
7229
7230 let swept = sweep_stale_claims_with(&queue, Duration::ZERO, |_| true);
7231 assert!(swept.is_empty(), "an unknown pid must keep its lock");
7232 assert!(queue.claim(&t.id).is_err(), "the lock remains protective");
7233 }
7234
7235 fn run_state_in(status: RunStatus, language: &str) -> RunState {
7236 let mut s = run_state(status);
7237 s.config.graph.language = language.to_owned();
7238 s
7239 }
7240
7241 fn unstarted_verdict(status: RunStatus) -> Verdict {
7242 Verdict {
7243 status,
7244 left_pr: false,
7245 quota_hit: false,
7246 parked: false,
7247 no_viable_candidates: false,
7248 }
7249 }
7250
7251 #[test]
7252 fn an_already_in_base_run_finishes_the_task_without_spending_an_attempt() {
7253 let mut t = task();
7254 t.attempts = 1;
7255 settle_in(
7256 &mut t,
7257 unstarted_verdict(RunStatus::AlreadyInBase),
7258 "already in main",
7259 1,
7260 phrases("en"),
7261 );
7262 assert_eq!(t.status, TaskStatus::Done);
7263 assert_eq!(t.attempts, 0);
7264 }
7265
7266 #[test]
7267 fn settle_renders_the_non_terminal_reason_in_the_configured_language() {
7268 let reason = |language: &str| {
7269 let mut t = task();
7270 settle_in(
7271 &mut t,
7272 unstarted_verdict(RunStatus::Judging),
7273 "boom",
7274 1,
7275 phrases(language),
7276 );
7277 t.last_error.or(t.hold_reason).unwrap_or_default()
7278 };
7279 assert!(
7280 reason("en").starts_with("the graph stopped at `"),
7281 "{}",
7282 reason("en")
7283 );
7284 assert!(reason("ja").starts_with("グラフが終端状態に達しないまま"));
7285 assert!(reason("日本語").contains("boom"));
7286 assert_eq!(reason("fr"), reason("en"));
7287 }
7288
7289 #[test]
7290 fn describe_follows_the_run_language_and_keeps_the_run_id() {
7291 let en = describe(&run_state_in(RunStatus::Stalled, "en"));
7292 assert!(en.starts_with("the judging panel lost its quorum"), "{en}");
7293 let ja = describe(&run_state_in(RunStatus::Stalled, "ja"));
7294 assert!(ja.starts_with("審査パネルが定足数を失いました"), "{ja}");
7295 assert!(ja.contains("[run "), "{ja}");
7296 let ended = describe(&run_state_in(RunStatus::Failed, "jp"));
7297 assert!(ended.starts_with("run 終了: "), "{ended}");
7298 let mut de = run_state_in(RunStatus::Failed, "de");
7299 let mut en = run_state_in(RunStatus::Failed, "en");
7300 de.id = "same".to_owned();
7301 en.id = "same".to_owned();
7302 assert_eq!(describe(&de), describe(&en));
7303 }
7304
7305 #[test]
7306 fn handover_refusals_follow_the_language_and_keep_the_detail_apart() {
7307 use crate::handover::Refused;
7308 let cases = [
7309 Refused::Foreign {
7310 branch: "b".into(),
7311 path: "/w/x".into(),
7312 why: "made by hand".into(),
7313 },
7314 Refused::Unsafe {
7315 branch: "b".into(),
7316 path: "/w/x".into(),
7317 why: "its worktree has uncommitted changes (a.rs)".into(),
7318 },
7319 Refused::ReleaseFailed {
7320 branch: "b".into(),
7321 path: "/w/x".into(),
7322 run: "ab12".into(),
7323 },
7324 ];
7325 for r in &cases {
7326 let en = (phrases("en").handover_refused)(r);
7327 assert_eq!(en, r.to_string());
7328 let ja = (phrases("ja").handover_refused)(r);
7329 assert!(
7330 !ja.contains("is checked out") && !ja.contains("try again"),
7331 "{ja}"
7332 );
7333 assert!(ja.contains("`b`") && ja.contains("/w/x"), "{ja}");
7334 if let Refused::Foreign { why, .. } | Refused::Unsafe { why, .. } = r {
7335 assert!(ja.contains(&format!("(詳細: {why})")), "{ja}");
7336 }
7337 }
7338 assert!(phrases("en").handover_hint.contains("release the task"));
7339 assert!(phrases("ja").handover_hint.contains("解放"));
7340 }
7341
7342 #[test]
7343 fn describe_translates_the_unanswered_reviewer_stop_only_in_ja() {
7344 let build = |lang: &str| {
7345 let mut s = run_state_in(RunStatus::Blocked, lang);
7346 s.id = "same".to_owned();
7347 s.config.graph.review_rounds = 3;
7348 let mut r = review_round(3);
7349 r.expected = 3;
7350 r.answered = 1;
7351 s.reviews.push(r);
7352 s.event(
7353 "review",
7354 "2 reviewer seat(s) never answered after 3 rounds; refusing to call it clean",
7355 );
7356 s
7357 };
7358 let en = describe(&build("en"));
7359 assert!(en.contains("(review: 2 reviewer seat(s) never answered after 3 rounds; refusing to call it clean)"), "{en}");
7360 let ja = describe(&build("ja"));
7361 assert!(ja.contains("2 席のレビュアーが 3 ラウンド"), "{ja}");
7362 assert!(!ja.contains("never answered"), "{ja}");
7363 assert!(ja.contains("[run "), "{ja}");
7364
7365 let mut failed = build("ja");
7368 failed.reviews[0].e2e.push(crate::run::CommandOutcome {
7369 command: "cargo test".to_owned(),
7370 code: Some(1),
7371 output_tail: String::new(),
7372 duration_ms: 0,
7373 resource_blocked: false,
7374 });
7375 failed.event("review", "stopped; e2e failed: cargo test");
7376 let ja = describe(&failed);
7377 assert!(ja.contains("stopped; e2e failed: cargo test"), "{ja}");
7378 assert!(!ja.contains("席のレビュアー"), "{ja}");
7379 }
7380
7381 #[test]
7382 fn refusals_and_recovery_prose_follow_the_language() {
7383 let t = held_task_with("r1");
7384 let q = action_question(
7385 "r1",
7386 ask::ChoiceAction::Resume {
7387 run: "r1".to_owned(),
7388 },
7389 );
7390 let refuse = |p: &Phrases| match decide_action(&t, &q, p, |_: &str| bail!("gone")) {
7391 ActionDecision::Refuse(s) => s,
7392 other => panic!("{other:?}"),
7393 };
7394 assert!(refuse(phrases("en")).contains("could not be read: gone"));
7395 assert!(refuse(phrases("ja")).contains("読み込めませんでした: gone"));
7396 assert_eq!(refuse(phrases("fr")), refuse(phrases("en")));
7397
7398 let mut held = task();
7399 reclaim(&mut held, None, 2, "ja");
7400 assert!(held.hold_reason.unwrap().contains("保留にしました"));
7401 let mut held = task();
7402 reclaim(&mut held, None, 2, "xx");
7403 assert!(held.hold_reason.unwrap().contains("held for a human"));
7404 }
7405
7406 fn run_state(status: RunStatus) -> RunState {
7407 let mut state = RunState::new(
7408 PathBuf::from("/repo"),
7409 "main".to_owned(),
7410 "abc1234def".to_owned(),
7411 "add retries".to_owned(),
7412 Config::default(),
7413 );
7414 state.status = status;
7415 state
7416 }
7417
7418 fn candidate(label: char, summary: &str, empty: bool, failed: Option<&str>) -> Candidate {
7419 Candidate {
7420 index: 0,
7421 label,
7422 agent: "claude".to_owned(),
7423 branch: format!("magi/x/{label}"),
7424 worktree: PathBuf::from("/repo"),
7425 summary: summary.to_owned(),
7426 stat: String::new(),
7427 files: 0,
7428 commits: usize::from(!empty),
7429 empty,
7430 failed: failed.map(str::to_owned),
7431 verified_noop: None,
7432 duration_ms: 0,
7433 folded: false,
7434 }
7435 }
7436
7437 #[test]
7438 fn diagnostic_names_the_failing_gate_checks_and_their_output() {
7439 let mut state = run_state(RunStatus::Blocked);
7440 state.gate = vec![
7441 CommandOutcome {
7442 command: "cargo make check".to_owned(),
7443 code: Some(0),
7444 output_tail: "ok".to_owned(),
7445 duration_ms: 0,
7446 resource_blocked: false,
7447 },
7448 CommandOutcome {
7449 command: "cargo test".to_owned(),
7450 code: Some(101),
7451 output_tail: "thread 'x' panicked: assertion failed".to_owned(),
7452 duration_ms: 0,
7453 resource_blocked: false,
7454 },
7455 ];
7456 let d = diagnostic(&state).expect("a failing gate must produce a diagnostic");
7457 assert!(d.contains("cargo test"), "{d}");
7458 assert!(
7459 !d.contains("cargo make check"),
7460 "a passing check is not a diagnostic: {d}"
7461 );
7462 assert!(d.contains("assertion failed"), "{d}");
7463 }
7464
7465 #[test]
7466 fn diagnostic_names_the_checks_the_fixer_gave_up_in_front_of() {
7467 let mut state = run_state(RunStatus::Blocked);
7468 state.event(
7469 "land",
7470 "stopped: the fixer produced no commit while 2 check(s) were failing \
7471 (build, lint); stopping instead of looping on an unchanged tree",
7472 );
7473 let d = diagnostic(&state).expect("a stalled land loop must produce a diagnostic");
7474 assert!(d.contains("build"), "{d}");
7475 assert!(d.contains("lint"), "{d}");
7476 assert!(d.contains("fixer produced no commit"), "{d}");
7477 }
7478
7479 #[test]
7480 fn describe_never_leaves_a_verified_noop_reading_as_a_bare_status_code() {
7481 let state = run_state(RunStatus::VerifiedNoop);
7486 let d = describe(&state);
7487 assert!(
7488 d.contains("agent-verified no-op"),
7489 "expected the display label, not the wire spelling: {d}"
7490 );
7491 assert!(!d.contains("verified_noop"), "{d}");
7492 }
7493
7494 #[test]
7495 fn diagnostic_carries_a_candidates_own_final_word_when_none_was_viable() {
7496 let mut state = run_state(RunStatus::Failed);
7502 state.candidates = vec![candidate(
7503 'A',
7504 "opened pull request #42, merged it, tagged v1.2.3 and published the release",
7505 true,
7506 None,
7507 )];
7508 let d = diagnostic(&state).expect("an empty candidate with a summary must be surfaced");
7509 assert!(d.contains("candidate A"), "{d}");
7510 assert!(d.contains("tagged v1.2.3"), "{d}");
7511 }
7512
7513 #[test]
7514 fn diagnostic_falls_back_to_a_candidates_failure_reason_when_it_has_no_summary() {
7515 let mut state = run_state(RunStatus::Failed);
7516 state.candidates = vec![candidate('A', "", true, Some("agent timed out"))];
7517 let d = diagnostic(&state).expect("a candidate's own failure reason must be surfaced");
7518 assert!(d.contains("candidate A"), "{d}");
7519 assert!(d.contains("agent timed out"), "{d}");
7520 }
7521
7522 #[test]
7523 fn diagnostic_is_none_when_nothing_recognisable_explains_the_hold() {
7524 let mut state = run_state(RunStatus::Failed);
7527 state.candidates = vec![candidate('A', "did the work", false, None)];
7528 assert!(diagnostic(&state).is_none());
7529 }
7530
7531 #[test]
7532 fn diagnostic_is_bounded_however_much_a_run_printed() {
7533 let mut state = run_state(RunStatus::Blocked);
7534 state.gate = vec![
7535 CommandOutcome {
7536 command: "cargo test".to_owned(),
7537 code: Some(101),
7538 output_tail: "x".repeat(50_000),
7539 duration_ms: 0,
7540 resource_blocked: false,
7541 },
7542 CommandOutcome {
7543 command: "cargo clippy".to_owned(),
7544 code: Some(1),
7545 output_tail: "y".repeat(50_000),
7546 duration_ms: 0,
7547 resource_blocked: false,
7548 },
7549 ];
7550 state.candidates = vec![
7551 candidate('A', &"z".repeat(50_000), true, None),
7552 candidate('B', &"w".repeat(50_000), true, None),
7553 ];
7554 let d = diagnostic(&state).expect("plenty here to diagnose");
7555 assert!(
7556 d.len() <= DIAGNOSTIC_MAX,
7557 "diagnostic grew to {} bytes, unbounded",
7558 d.len()
7559 );
7560 }
7561
7562 #[test]
7563 fn selected_delivery_guards_survive_recovery_failure_accounting() {
7564 for status in [RunStatus::Ready, RunStatus::Blocked, RunStatus::Failed] {
7565 let mut t = task();
7566 let mut state = run_state(status);
7567 t.pr_recovery = Some(crate::pr_recovery::Selection {
7568 url: "url".into(),
7569 branch: format!("magi/task-{}", t.id),
7570 head: "a".repeat(40),
7571 source_head: None,
7572 source_run: None,
7573 source_proof: None,
7574 disposition: Some(crate::pr_recovery::Disposition::Land),
7575 instruction: t.instruction.clone(),
7576 approved_at: Timestamp::now(),
7577 });
7578 state.pr_recovery = t.pr_recovery.clone();
7579 t.start(state.id.clone());
7580 settle_and_diagnose(
7581 &mut t,
7582 Verdict {
7583 status,
7584 left_pr: false,
7585 quota_hit: false,
7586 parked: false,
7587 no_viable_candidates: false,
7588 },
7589 "delivery stopped",
7590 5,
7591 &state,
7592 );
7593 assert_eq!(t.status, TaskStatus::Held);
7594 let reason = t.hold_reason.as_deref().unwrap();
7595 if status == RunStatus::Ready {
7596 assert!(reason.contains("Selected PR delivery is incomplete"));
7597 assert!(t.recovery_budget.is_none());
7598 } else {
7599 assert!(reason.contains("Selected recovery stopped"));
7600 assert_eq!(t.recovery_budget.as_ref().unwrap().failures, 1);
7601 }
7602 assert!(t.pr_recovery.is_some());
7603 }
7604 }
7605
7606 #[test]
7607 fn settle_and_diagnose_attaches_a_diagnostic_only_once_the_task_is_held() {
7608 let mut state = run_state(RunStatus::Blocked);
7609 state.gate = vec![CommandOutcome {
7610 command: "cargo test".to_owned(),
7611 code: Some(101),
7612 output_tail: "assertion failed".to_owned(),
7613 duration_ms: 0,
7614 resource_blocked: false,
7615 }];
7616 let verdict = Verdict {
7617 status: RunStatus::Blocked,
7618 left_pr: false,
7619 quota_hit: false,
7620 parked: false,
7621 no_viable_candidates: false,
7622 };
7623
7624 let mut t = task();
7627 t.start("run-1".to_owned());
7628 settle_and_diagnose(&mut t, verdict, "gate failed", 2, &state);
7629 assert_eq!(t.status, TaskStatus::Failed);
7630 assert!(t.diagnostic.is_none());
7631
7632 t.start("run-2".to_owned());
7635 settle_and_diagnose(&mut t, verdict, "gate failed", 2, &state);
7636 assert_eq!(t.status, TaskStatus::Held);
7637 let d = t.diagnostic.expect("a held task must carry its diagnostic");
7638 assert!(d.contains("cargo test"), "{d}");
7639 }
7640
7641 #[test]
7642 fn a_held_task_names_the_open_question_it_is_waiting_on() {
7643 crate::run::pin_test_home();
7648 let home = crate::run::home();
7649 let state = run_state(RunStatus::VerifiedNoop);
7650 let mut q = ask::Question::new(
7651 state.id.clone(),
7652 "implement".to_owned(),
7653 "impl-A".to_owned(),
7654 "is this really a no-op?".to_owned(),
7655 String::new(),
7656 Vec::new(),
7657 );
7658 Questions::at(home.join("questions")).put(&mut q).unwrap();
7659
7660 let verdict = Verdict {
7661 status: RunStatus::VerifiedNoop,
7662 left_pr: false,
7663 quota_hit: false,
7664 parked: false,
7665 no_viable_candidates: false,
7666 };
7667 let mut t = task();
7668 t.start(state.id.clone());
7669 settle_and_diagnose(&mut t, verdict, "run ended agent-verified no-op", 5, &state);
7670
7671 assert_eq!(t.status, TaskStatus::Held);
7672 let reason = t.hold_reason.expect("a held task must record why");
7673 assert!(
7674 reason.starts_with("run ended agent-verified no-op"),
7675 "the original settle reason must survive unchanged: {reason}"
7676 );
7677 assert!(
7678 reason.contains(q.short()),
7679 "the open question's id must be named so the notice is actionable: {reason}"
7680 );
7681 }
7682
7683 #[test]
7684 fn a_held_task_with_no_open_question_keeps_its_plain_reason() {
7685 crate::run::pin_test_home();
7686 let state = run_state(RunStatus::VerifiedNoop);
7687
7688 let verdict = Verdict {
7689 status: RunStatus::VerifiedNoop,
7690 left_pr: false,
7691 quota_hit: false,
7692 parked: false,
7693 no_viable_candidates: false,
7694 };
7695 let mut t = task();
7696 t.start(state.id.clone());
7697 settle_and_diagnose(&mut t, verdict, "run ended agent-verified no-op", 5, &state);
7698
7699 assert_eq!(t.status, TaskStatus::Held);
7700 assert_eq!(
7701 t.hold_reason.as_deref(),
7702 Some("run ended agent-verified no-op"),
7703 "nothing to append when the question was already answered or never asked"
7704 );
7705 }
7706
7707 #[test]
7708 fn supersede_prior_runs_rewrites_an_earlier_blocked_attempt_once_a_later_one_lands() {
7709 crate::run::pin_test_home();
7710 let mut first = run_state(RunStatus::Blocked);
7711 first.id = "20260101-000000-sup1".to_owned();
7712 first.save().unwrap();
7713 let mut second = run_state(RunStatus::Merged);
7714 second.id = "20260101-000000-sup2".to_owned();
7715 second.save().unwrap();
7716
7717 let mut t = task();
7718 t.runs = vec![first.id.clone(), second.id.clone()];
7719 t.status = TaskStatus::Done;
7720
7721 supersede_prior_runs(&t, &crate::run::home());
7722
7723 assert_eq!(
7724 RunState::load(&first.id).unwrap().status,
7725 RunStatus::Superseded,
7726 "the first attempt's Blocked no longer needs anyone's attention"
7727 );
7728 assert_eq!(
7729 RunState::load(&second.id).unwrap().status,
7730 RunStatus::Merged,
7731 "the run that actually succeeded is left exactly as it was"
7732 );
7733 }
7734
7735 #[test]
7736 fn supersede_prior_runs_leaves_a_manually_resumed_attempt_alone() {
7737 crate::run::pin_test_home();
7743 let mut first = run_state(RunStatus::Blocked);
7744 first.id = "20260101-000000-sup9".to_owned();
7745 first.driver_pid = Some(std::process::id());
7748 first.driver_started_at = Some(
7749 crate::proc::process_started_at(std::process::id())
7750 .expect("this test process's own start time must be queryable"),
7751 );
7752 first.save().unwrap();
7753 let mut second = run_state(RunStatus::Merged);
7754 second.id = "20260101-000000-supa".to_owned();
7755 second.save().unwrap();
7756
7757 let mut t = task();
7758 t.runs = vec![first.id.clone(), second.id.clone()];
7759 t.status = TaskStatus::Done;
7760
7761 supersede_prior_runs(&t, &crate::run::home());
7762
7763 assert_eq!(
7764 RunState::load(&first.id).unwrap().status,
7765 RunStatus::Blocked,
7766 "a live driver_pid means something is still actually working this run, \
7767 even though no daemon claims it - rewriting under it would just be \
7768 undone the next time that process saves"
7769 );
7770 }
7771
7772 #[test]
7773 fn resweep_catches_up_a_run_left_live_once_its_manual_process_is_no_longer_driving_it() {
7774 let dir = tempfile::tempdir().unwrap();
7780 let home = dir.path().to_path_buf();
7781 let queue = Queue::at(dir.path().join("queue"));
7782
7783 let mut first = run_state(RunStatus::Blocked);
7784 first.id = "20260101-000000-supd".to_owned();
7785 first.driver_pid = Some(std::process::id());
7786 first.driver_started_at = Some(
7787 crate::proc::process_started_at(std::process::id())
7788 .expect("this test process's own start time must be queryable"),
7789 );
7790 first.save_under(&home).unwrap();
7791 let mut second = run_state(RunStatus::Merged);
7792 second.id = "20260101-000000-supe".to_owned();
7793 second.save_under(&home).unwrap();
7794
7795 let mut t = task();
7796 t.runs = vec![first.id.clone(), second.id.clone()];
7797 t.status = TaskStatus::Done;
7798 queue.put(&mut t).unwrap();
7799
7800 resweep_superseded_attempts(&queue, &home);
7801 assert_eq!(
7802 RunState::load_under(&first.id, &home).unwrap().status,
7803 RunStatus::Blocked,
7804 "still live on the first pass, so still untouched"
7805 );
7806
7807 let mut stale = RunState::load_under(&first.id, &home).unwrap();
7813 stale.driver_started_at = Some("1".to_owned());
7814 stale.save_under(&home).unwrap();
7815
7816 resweep_superseded_attempts(&queue, &home);
7817 assert_eq!(
7818 RunState::load_under(&first.id, &home).unwrap().status,
7819 RunStatus::Superseded,
7820 "the second pass catches up what the first one correctly skipped"
7821 );
7822 }
7823
7824 #[test]
7825 fn supersede_prior_runs_leaves_concurrent_blocked_attempts_alone_while_the_task_is_not_done() {
7826 crate::run::pin_test_home();
7827 let mut first = run_state(RunStatus::Blocked);
7828 first.id = "20260101-000000-sup3".to_owned();
7829 first.save().unwrap();
7830 let mut second = run_state(RunStatus::Blocked);
7831 second.id = "20260101-000000-sup4".to_owned();
7832 second.save().unwrap();
7833
7834 let mut t = task();
7835 t.runs = vec![first.id.clone(), second.id.clone()];
7836 t.status = TaskStatus::Failed;
7840
7841 supersede_prior_runs(&t, &crate::run::home());
7842
7843 assert_eq!(
7844 RunState::load(&first.id).unwrap().status,
7845 RunStatus::Blocked
7846 );
7847 assert_eq!(
7848 RunState::load(&second.id).unwrap().status,
7849 RunStatus::Blocked
7850 );
7851 }
7852
7853 #[test]
7854 fn supersede_prior_runs_does_nothing_when_the_task_was_closed_by_hand() {
7855 crate::run::pin_test_home();
7859 let mut first = run_state(RunStatus::Blocked);
7860 first.id = "20260101-000000-sup5".to_owned();
7861 first.save().unwrap();
7862
7863 let mut t = task();
7864 t.runs = vec![first.id.clone()];
7865 t.status = TaskStatus::Done;
7866
7867 supersede_prior_runs(&t, &crate::run::home());
7868
7869 assert_eq!(
7870 RunState::load(&first.id).unwrap().status,
7871 RunStatus::Blocked,
7872 "a single-attempt task has no earlier run to supersede"
7873 );
7874 }
7875
7876 #[test]
7877 fn supersede_prior_runs_does_nothing_when_the_last_recorded_attempt_never_landed() {
7878 crate::run::pin_test_home();
7885 let mut first = run_state(RunStatus::Blocked);
7886 first.id = "20260101-000000-supb".to_owned();
7887 first.save().unwrap();
7888 let mut second = run_state(RunStatus::Failed);
7889 second.id = "20260101-000000-supc".to_owned();
7890 second.save().unwrap();
7891
7892 let mut t = task();
7893 t.runs = vec![first.id.clone(), second.id.clone()];
7894 t.status = TaskStatus::Done;
7895
7896 supersede_prior_runs(&t, &crate::run::home());
7897
7898 assert_eq!(
7899 RunState::load(&first.id).unwrap().status,
7900 RunStatus::Blocked,
7901 "the task's last attempt never landed, so there is nothing here \
7902 actually superseding it"
7903 );
7904 }
7905
7906 #[test]
7907 fn supersede_prior_runs_leaves_a_failed_or_verified_noop_attempt_as_is() {
7908 crate::run::pin_test_home();
7912 let mut failed = run_state(RunStatus::Failed);
7913 failed.id = "20260101-000000-sup6".to_owned();
7914 failed.save().unwrap();
7915 let mut noop = run_state(RunStatus::VerifiedNoop);
7916 noop.id = "20260101-000000-sup7".to_owned();
7917 noop.save().unwrap();
7918 let mut winner = run_state(RunStatus::Ready);
7919 winner.id = "20260101-000000-sup8".to_owned();
7920 winner.save().unwrap();
7921
7922 let mut t = task();
7923 t.runs = vec![failed.id.clone(), noop.id.clone(), winner.id.clone()];
7924 t.status = TaskStatus::Done;
7925
7926 supersede_prior_runs(&t, &crate::run::home());
7927
7928 assert_eq!(
7929 RunState::load(&failed.id).unwrap().status,
7930 RunStatus::Failed
7931 );
7932 assert_eq!(
7933 RunState::load(&noop.id).unwrap().status,
7934 RunStatus::VerifiedNoop
7935 );
7936 }
7937
7938 fn approval_question(run: &str) -> ask::Question {
7939 ask::Question::new(
7940 run.to_owned(),
7941 land::APPROVAL_NODE.to_owned(),
7942 "land".to_owned(),
7943 "merge?".to_owned(),
7944 String::new(),
7945 vec!["merge".to_owned(), "hold".to_owned()],
7946 )
7947 }
7948
7949 #[test]
7950 fn land_resume_state_leaves_a_fresh_open_question_waiting() {
7951 crate::run::pin_test_home();
7952 let mut state = run_state(RunStatus::Landing);
7953 state.id = "20260101-000000-fre1".to_owned();
7954 state.parked = true;
7955 state.save().unwrap();
7956 ask::Questions::open()
7957 .put(&mut approval_question(&state.id))
7958 .unwrap();
7959
7960 let mut t = task();
7961 t.runs.push(state.id.clone());
7962 assert_eq!(
7963 land_resume_state(&t),
7964 LandResume::StillWaiting,
7965 "nobody has answered and the timeout has not passed"
7966 );
7967 }
7968
7969 #[test]
7970 fn land_resume_state_waits_on_a_withheld_text_question_then_resumes() {
7971 crate::run::pin_test_home();
7972 let mut state = run_state(RunStatus::Gating);
7973 state.id = "20260101-000000-gt01".to_owned();
7974 state.parked = true;
7975 state.config.graph.answer_timeout = 60;
7976 state.github_text = Some(crate::run::GithubTextAsk {
7977 rewrite: None,
7978 fingerprint: "f".into(),
7979 title: true,
7980 body: false,
7981 categories: vec!["title-language".into()],
7982 question: None,
7983 asks: 1,
7984 resolved: false,
7985 chosen_title: None,
7986 choice: None,
7987 });
7988 state.save().unwrap();
7989 let store = ask::Questions::open();
7990 let mut q = ask::Question::new(
7991 state.id.clone(),
7992 crate::github_text::ASK_NODE.to_owned(),
7993 crate::github_text::ASK_SEAT.to_owned(),
7994 "s".into(),
7995 String::new(),
7996 vec!["use fallback".into()],
7997 );
7998 store.put(&mut q).unwrap();
7999 let mut t = task();
8000 t.runs.push(state.id.clone());
8001 assert_eq!(land_resume_state(&t), LandResume::StillWaiting);
8002 q.asked_at = Timestamp::now() - jiff::SignedDuration::from_secs(120);
8003 store.put(&mut q).unwrap();
8004 assert_eq!(land_resume_state(&t), LandResume::Ready);
8005 assert!(!store.get(&q.id).unwrap().status.open());
8006 state.github_text.as_mut().unwrap().resolved = true;
8008 state.save().unwrap();
8009 assert_eq!(land_resume_state(&t), LandResume::NotLanding);
8010 }
8011
8012 #[test]
8013 fn land_resume_state_abandons_a_question_that_outlived_answer_timeout() {
8014 crate::run::pin_test_home();
8019 let mut state = run_state(RunStatus::Landing);
8020 state.id = "20260101-000000-exp1".to_owned();
8021 state.parked = true;
8022 state.config.graph.answer_timeout = 60;
8023 state.save().unwrap();
8024
8025 let store = ask::Questions::open();
8026 let mut q = approval_question(&state.id);
8027 q.asked_at = Timestamp::now() - jiff::SignedDuration::from_secs(120);
8028 store.put(&mut q).unwrap();
8029
8030 let mut t = task();
8031 t.runs.push(state.id.clone());
8032 assert_eq!(
8033 land_resume_state(&t),
8034 LandResume::Ready,
8035 "an expired question must not be waited on forever"
8036 );
8037
8038 let after = store.get(&q.id).unwrap();
8039 assert!(
8040 !after.status.open(),
8041 "the question is abandoned, not silently ignored"
8042 );
8043 assert!(
8044 after.resolution().is_none(),
8045 "an abandoned question is not read as a decision"
8046 );
8047 }
8048
8049 #[test]
8050 fn reclaim_settles_a_running_task_against_its_last_run() {
8051 let mut t = task();
8052 t.start("20260904-000000-4043".to_owned());
8053 reclaim(&mut t, Some(run_state(RunStatus::Ready)), 2, "en");
8054 assert_eq!(
8055 t.status,
8056 TaskStatus::Done,
8057 "a run that actually finished must not stay `running` forever"
8058 );
8059 }
8060
8061 #[test]
8062 fn reclaim_maps_a_parked_run_to_parked_without_touching_last_error() {
8063 let mut t = task();
8064 t.start("20260904-000000-4043".to_owned());
8065 t.last_error = Some("earlier trouble".to_owned());
8066 let mut state = run_state(RunStatus::Judging);
8067 state.parked = true;
8068 reclaim(&mut t, Some(state), 1, "en");
8069 assert_eq!(t.status, TaskStatus::Parked);
8070 assert_eq!(t.attempts, 0, "the park is refunded, even at max_attempts");
8071 assert_eq!(t.last_error.as_deref(), Some("earlier trouble"));
8072 assert!(t.park_reason.is_some());
8073 assert_eq!(t.runs, ["20260904-000000-4043"], "the same run is kept");
8074 assert!(!t.fresh_start);
8075 assert!(t.status.runnable());
8076 }
8077
8078 #[test]
8079 fn settle_parks_a_task_without_a_failure_and_a_new_start_clears_the_reason() {
8080 let mut t = task();
8081 t.start("20260904-000000-4043".to_owned());
8082 t.last_error = Some("earlier trouble".to_owned());
8083 settle(
8084 &mut t,
8085 Verdict {
8086 status: RunStatus::Judging,
8087 left_pr: false,
8088 quota_hit: false,
8089 parked: true,
8090 no_viable_candidates: false,
8091 },
8092 "parked after `judging`",
8093 1,
8094 );
8095 assert_eq!(t.status, TaskStatus::Parked);
8096 assert_eq!(t.attempts, 0);
8097 assert_eq!(t.last_error.as_deref(), Some("earlier trouble"));
8098 assert_eq!(t.park_reason.as_deref(), Some("parked after `judging`"));
8099 assert_eq!(t.schema, crate::queue::SCHEMA);
8100 assert!(
8101 crate::notices::task_held(&t).is_none(),
8102 "a park is not news"
8103 );
8104 t.start("20260904-000000-4043".to_owned());
8105 assert_eq!(t.park_reason, None);
8106 }
8107
8108 #[test]
8109 fn reclaim_reuses_the_same_retry_policy_as_a_live_settle() {
8110 let mut t = task();
8114 t.start("20260904-000000-4043".to_owned());
8115 reclaim(&mut t, Some(run_state(RunStatus::Blocked)), 2, "en");
8116 assert_eq!(t.status, TaskStatus::Failed);
8117 assert!(t.status.runnable());
8118 }
8119
8120 #[test]
8121 fn a_settled_parked_pass_records_its_cause_and_the_refund() {
8122 let mut t = task();
8123 t.start("20260904-000000-4043".to_owned());
8124 assert_eq!(t.attempts, 1);
8125 let mut run = run_state(RunStatus::Implementing);
8126 run.id = "20260904-000000-4043".to_owned();
8127 run.begin_pass();
8128 let cause = crate::run::ParkCause::Upgrade {
8129 from: "0.121.0".to_owned(),
8130 to: Some("0.121.1".to_owned()),
8131 };
8132 run.end_pass_parked(cause.clone());
8133 run.parked = true;
8134 let verdict = Verdict {
8135 status: run.status,
8136 left_pr: false,
8137 quota_hit: false,
8138 parked: true,
8139 no_viable_candidates: false,
8140 };
8141 settle_and_diagnose(&mut t, verdict, "parked", 2, &run);
8142 assert_eq!(t.attempts, 0);
8143 assert_eq!(t.passes.len(), 1);
8144 assert_eq!(t.passes[0].cost, crate::queue::PassCost::Refunded);
8145 assert_eq!(t.passes[0].parked, Some(cause));
8146 assert_eq!(t.passes[0].pass_index, 0);
8147
8148 t.start("20260904-000000-4043".to_owned());
8151 run.begin_pass();
8152 run.status = RunStatus::Failed;
8153 let verdict = Verdict {
8154 status: RunStatus::Failed,
8155 left_pr: false,
8156 quota_hit: false,
8157 parked: false,
8158 no_viable_candidates: false,
8159 };
8160 settle_and_diagnose(&mut t, verdict, "failed", 2, &run);
8161 assert_eq!(t.passes.len(), 2);
8162 assert_eq!(t.passes[0].cost, crate::queue::PassCost::Refunded);
8163 assert_eq!(t.passes[1].cost, crate::queue::PassCost::Spent);
8164 assert_eq!(t.passes[1].pass_index, 1);
8165 assert_eq!(t.attempts, 1);
8166 }
8167
8168 #[test]
8169 fn reclaim_holds_a_running_task_whose_run_cannot_be_found() {
8170 let mut t = task();
8171 t.start("20260904-000000-4043".to_owned());
8172 reclaim(&mut t, None, 2, "en");
8173 assert_eq!(t.status, TaskStatus::Held);
8174 assert!(
8175 t.last_error
8176 .as_deref()
8177 .is_some_and(|e| e.contains("running")),
8178 "the operator needs to know why this task was held"
8179 );
8180 }
8181
8182 #[test]
8183 fn orphaned_running_tasks_are_reclaimed_but_live_ones_are_left_alone() {
8184 let dir = tempfile::tempdir().unwrap();
8185 let queue = Queue::at(dir.path().to_path_buf());
8186
8187 let mut orphaned = task();
8189 orphaned.id = "20260904-000000-orph".to_owned();
8190 orphaned.status = TaskStatus::Running;
8191 orphaned.attempts = 1;
8192 queue.put(&mut orphaned).unwrap();
8193
8194 let mut alive = task();
8195 alive.id = "20260904-000000-live".to_owned();
8196 alive.status = TaskStatus::Running;
8197 alive.attempts = 1;
8198 queue.put(&mut alive).unwrap();
8199 let _held_by_a_live_daemon = queue.claim(&alive.id).unwrap();
8200
8201 let mut queued = task();
8202 queued.id = "20260904-000000-wait".to_owned();
8203 queue.put(&mut queued).unwrap();
8204
8205 let reclaimed = reclaim_orphaned_running(&queue, 2);
8206 assert_eq!(reclaimed, vec![orphaned.id.clone()]);
8207
8208 assert_eq!(
8209 queue.get(&orphaned.id).unwrap().status,
8210 TaskStatus::Held,
8211 "nothing was driving it and there was no run to recover"
8212 );
8213 assert_eq!(
8214 queue.get(&alive.id).unwrap().status,
8215 TaskStatus::Running,
8216 "a live claim must protect the task it belongs to"
8217 );
8218 assert_eq!(queue.get(&queued.id).unwrap().status, TaskStatus::Queued);
8219 }
8220
8221 fn read_run_under(home: &Path, id: &str) -> RunState {
8227 let body = std::fs::read_to_string(home.join("runs").join(id).join("run.json")).unwrap();
8228 serde_json::from_str(&body).unwrap()
8229 }
8230
8231 #[test]
8232 fn reclaim_abandoned_runs_fails_a_run_whose_active_seats_are_all_provably_dead() {
8233 let dir = tempfile::tempdir().unwrap();
8234 let home = dir.path().to_path_buf();
8235 let now = Timestamp::now();
8236 let overrun_seat = || crate::run::ActiveSeat {
8237 node: "implement".to_owned(),
8238 started_at: now - jiff::SignedDuration::new(21_000, 0),
8239 timeout_secs: 3_600,
8240 attempt: 0,
8241 task: None,
8242 command: None,
8243 index: None,
8244 total: None,
8245 };
8246
8247 let mut dead = run_state(RunStatus::Implementing);
8248 dead.id = "20260101-000000-dead".to_owned();
8249 dead.active.insert("impl-A".to_owned(), overrun_seat());
8250 dead.driver_pid = Some(4242);
8253 dead.save_under(&home).unwrap();
8254
8255 let mut alive = run_state(RunStatus::Implementing);
8258 alive.id = "20260101-000000-aliv".to_owned();
8259 alive.active.insert("impl-A".to_owned(), overrun_seat());
8260 alive.save_under(&home).unwrap();
8261 let mut status = Status::new();
8262 status.current = vec![Current {
8263 task: "20260101-000000-task".to_owned(),
8264 run: alive.id.clone(),
8265 }];
8266 write_status_to(&home.join("daemon.json"), &status).unwrap();
8267
8268 let questions = Questions::at(home.join("questions"));
8272 let mut q = ask::Question::new(
8273 dead.id.clone(),
8274 "implement".to_owned(),
8275 "impl-A".to_owned(),
8276 "Which storage backend?".to_owned(),
8277 String::new(),
8278 vec!["SQLite".to_owned(), "Redis".to_owned()],
8279 );
8280 questions.put(&mut q).unwrap();
8281
8282 let abandoned = reclaim_abandoned_runs_with(
8283 &home,
8284 now,
8285 |pid| if pid == 4242 { Some(false) } else { None },
8286 |_| panic!("a query answering Dead outright needs no identity corroboration"),
8287 );
8288 assert_eq!(abandoned, vec![dead.id.clone()]);
8289
8290 let reloaded = read_run_under(&home, &dead.id);
8291 assert_eq!(reloaded.status, RunStatus::Failed);
8292 assert!(reloaded.active.is_empty());
8293 assert!(
8294 !questions.get(&q.id).unwrap().status.open(),
8295 "the failed run's own open question must be settled in the same pass"
8296 );
8297
8298 let still_alive = read_run_under(&home, &alive.id);
8299 assert_eq!(
8300 still_alive.status,
8301 RunStatus::Implementing,
8302 "a live daemon's claim protects it"
8303 );
8304 assert!(!still_alive.active.is_empty());
8305 }
8306
8307 #[test]
8317 fn reclaim_abandoned_runs_leaves_a_live_manual_run_alone_even_though_no_daemon_claims_it() {
8318 let dir = tempfile::tempdir().unwrap();
8319 let home = dir.path().to_path_buf();
8320 let now = Timestamp::now();
8321
8322 let mut manual = run_state(RunStatus::Reviewing);
8323 manual.id = "20260101-000000-manl".to_owned();
8324 manual.active.insert(
8325 "review-1".to_owned(),
8326 crate::run::ActiveSeat {
8327 node: "review".to_owned(),
8328 started_at: now - jiff::SignedDuration::new(21_000, 0),
8329 timeout_secs: 3_600,
8330 attempt: 0,
8331 task: None,
8332 command: None,
8333 index: None,
8334 total: None,
8335 },
8336 );
8337 manual.driver_pid = Some(4242);
8341 manual.driver_started_at = Some("1790000000".to_owned());
8342 manual.save_under(&home).unwrap();
8343
8344 let abandoned = reclaim_abandoned_runs_with(
8345 &home,
8346 now,
8347 |pid| if pid == 4242 { Some(true) } else { None },
8348 |pid| {
8349 if pid == 4242 {
8350 Some("1790000000".to_owned())
8351 } else {
8352 None
8353 }
8354 },
8355 );
8356 assert!(
8357 abandoned.is_empty(),
8358 "a manual run a real process is still driving must never be reclaimed: {abandoned:?}"
8359 );
8360
8361 let reloaded = read_run_under(&home, &manual.id);
8362 assert_eq!(reloaded.status, RunStatus::Reviewing);
8363 assert!(!reloaded.active.is_empty());
8364 }
8365
8366 #[test]
8367 fn an_already_claimed_task_is_skipped_rather_than_failed() {
8368 let dir = tempfile::tempdir().unwrap();
8369 let queue = Queue::at(dir.path().to_path_buf());
8370 let mut only = task();
8371 queue.put(&mut only).unwrap();
8372
8373 let _elsewhere = queue.claim(&only.id).unwrap();
8374 let candidates = runnable(&queue);
8375 assert_eq!(candidates.len(), 1, "the task is still runnable");
8376 assert!(
8377 queue.claim(&candidates[0].id).is_err(),
8378 "the loop cannot take a claim somebody else holds"
8379 );
8380
8381 let after = queue.get(&only.id).unwrap();
8382 assert_eq!(after.status, TaskStatus::Queued);
8383 assert_eq!(
8384 after.attempts, 0,
8385 "losing the race is not an attempt at the task"
8386 );
8387 assert_eq!(after.last_error, None);
8388 }
8389
8390 #[test]
8391 fn the_status_file_round_trips_and_its_heartbeat_advances() {
8392 let dir = tempfile::tempdir().unwrap();
8393 let path = dir.path().join("daemon.json");
8394
8395 let mut status = Status::new();
8396 status.idle = false;
8397 status.completed = 7;
8398 status.current = vec![Current {
8399 task: "20260902-000000-t111".to_owned(),
8400 run: "20260902-000001-r111".to_owned(),
8401 }];
8402 write_status_to(&path, &status).unwrap();
8403 let first: Status = serde_json::from_str(&std::fs::read_to_string(&path).unwrap()).unwrap();
8404 assert_eq!(first.schema, SCHEMA);
8405 assert_eq!(first.pid, std::process::id());
8406 assert!(!first.idle);
8407 assert_eq!(first.completed, 7);
8408 assert_eq!(first.current, status.current);
8409 assert!(
8410 !path.with_extension("json.tmp").exists(),
8411 "the temp file is renamed, not left behind"
8412 );
8413
8414 std::thread::sleep(Duration::from_millis(5));
8415 status.updated_at = Timestamp::now();
8416 status.polls = 3;
8417 write_status_to(&path, &status).unwrap();
8418 let second: Status =
8419 serde_json::from_str(&std::fs::read_to_string(&path).unwrap()).unwrap();
8420 assert!(
8421 second.updated_at > first.updated_at,
8422 "a reader can only detect staleness if the heartbeat moves"
8423 );
8424 assert_eq!(
8425 second.started_at, first.started_at,
8426 "the start time is not a heartbeat"
8427 );
8428 assert_eq!(second.polls, 3);
8429 }
8430
8431 #[test]
8432 fn reading_counts_as_running_only_while_its_heartbeat_is_fresh() {
8433 let dir = tempfile::tempdir().unwrap();
8434
8435 assert!(read_status(dir.path()).is_none(), "no file, no daemon");
8436
8437 let mut status = Status::new();
8438 status.updated_at = Timestamp::now() - jiff::SignedDuration::from_secs(60);
8439 write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
8440 let stale = read_status(dir.path()).unwrap();
8441 assert!(
8442 !stale.running(Timestamp::now()),
8443 "a minute without a heartbeat is a dead daemon, not a busy one"
8444 );
8445 assert!(stale.age_secs(Timestamp::now()).is_some_and(|s| s >= 55));
8446
8447 status.updated_at = Timestamp::now();
8448 write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
8449 let fresh = read_status(dir.path()).unwrap();
8450 assert!(fresh.running(Timestamp::now()));
8451 }
8452
8453 #[test]
8454 fn only_a_live_daemon_on_this_very_run_counts_as_working_on_it() {
8455 let dir = tempfile::tempdir().unwrap();
8456 let now = Timestamp::now();
8457 let mine = "20260903-080619-01c2";
8458
8459 assert!(
8460 !is_working_on(dir.path(), mine, now),
8461 "no status file means nobody is working on anything"
8462 );
8463
8464 let mut status = Status::new();
8465 status.current = vec![Current {
8466 task: "20260903-080340-0167".to_owned(),
8467 run: mine.to_owned(),
8468 }];
8469 status.updated_at = now;
8470 write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
8471 assert!(is_working_on(dir.path(), mine, now));
8472 assert!(
8473 !is_working_on(dir.path(), "20260903-105039-3cbf", now),
8474 "a daemon busy with one run is not working on another"
8475 );
8476
8477 status.updated_at = now - jiff::SignedDuration::from_secs(600);
8480 write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
8481 assert!(
8482 !is_working_on(dir.path(), mine, now),
8483 "a stale heartbeat is a dead daemon, so its run is a leftover"
8484 );
8485 }
8486
8487 #[test]
8488 fn is_working_on_short_matches_by_the_worktree_bays_own_name() {
8489 let dir = tempfile::tempdir().unwrap();
8490 let now = Timestamp::now();
8491
8492 assert!(
8493 !is_working_on_short(dir.path(), "01c2", now),
8494 "no status file means nobody is working on anything"
8495 );
8496
8497 let mut status = Status::new();
8498 status.current = vec![Current {
8499 task: "20260903-080340-0167".to_owned(),
8500 run: "20260903-080619-01c2".to_owned(),
8501 }];
8502 status.updated_at = now;
8503 write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
8504 assert!(
8505 is_working_on_short(dir.path(), "01c2", now),
8506 "the run's short id is the last block of its full id"
8507 );
8508 assert!(
8509 !is_working_on_short(dir.path(), "3cbf", now),
8510 "a daemon busy with one worktree bay is not working on another"
8511 );
8512 }
8513
8514 #[test]
8515 fn a_newer_status_file_still_yields_a_reading() {
8516 let dir = tempfile::tempdir().unwrap();
8517 std::fs::write(
8520 dir.path().join("daemon.json"),
8521 serde_json::json!({
8522 "schema": 2,
8523 "updated_at": Timestamp::now().to_string(),
8524 "idle": true,
8525 "surprise": { "nested": [1, 2, 3] },
8526 })
8527 .to_string(),
8528 )
8529 .unwrap();
8530
8531 let reading = read_status(dir.path()).expect("a forward-compatible read");
8532 assert!(reading.running(Timestamp::now()));
8533 assert!(reading.idle);
8534 assert!(reading.current.is_empty());
8535 }
8536
8537 #[test]
8538 fn an_older_daemons_single_object_current_still_reads_as_a_one_item_list() {
8539 let dir = tempfile::tempdir().unwrap();
8545 std::fs::write(
8546 dir.path().join("daemon.json"),
8547 serde_json::json!({
8548 "schema": 1,
8549 "pid": 4242,
8550 "updated_at": Timestamp::now().to_string(),
8551 "idle": false,
8552 "current": {"task": "20260902-140501-aaaa", "run": "20260902-140502-bbbb"},
8553 "completed": 3,
8554 "polls": 9,
8555 })
8556 .to_string(),
8557 )
8558 .unwrap();
8559
8560 let reading = read_status(dir.path()).expect("an older shape must still parse");
8561 assert!(reading.running(Timestamp::now()));
8562 assert_eq!(
8563 reading.current,
8564 vec![Current {
8565 task: "20260902-140501-aaaa".to_owned(),
8566 run: "20260902-140502-bbbb".to_owned(),
8567 }]
8568 );
8569 }
8570
8571 #[test]
8572 fn an_absent_or_null_current_reads_as_idle_not_a_parse_failure() {
8573 let dir = tempfile::tempdir().unwrap();
8574 std::fs::write(
8575 dir.path().join("daemon.json"),
8576 serde_json::json!({
8577 "schema": 1,
8578 "updated_at": Timestamp::now().to_string(),
8579 "idle": true,
8580 "current": null,
8581 })
8582 .to_string(),
8583 )
8584 .unwrap();
8585 let with_null = read_status(dir.path()).expect("null must still parse");
8586 assert!(with_null.current.is_empty());
8587
8588 std::fs::write(
8589 dir.path().join("daemon.json"),
8590 serde_json::json!({
8591 "schema": 1,
8592 "updated_at": Timestamp::now().to_string(),
8593 "idle": true,
8594 })
8595 .to_string(),
8596 )
8597 .unwrap();
8598 let absent = read_status(dir.path()).expect("a missing field must still parse");
8599 assert!(absent.current.is_empty());
8600 }
8601
8602 #[test]
8603 fn a_task_without_a_repository_runs_in_the_daemons_default() {
8604 let fallback = Path::new("/default");
8605 let mut blank = task();
8606 blank.repo = PathBuf::new();
8607 assert_eq!(repo_for(&blank, fallback), PathBuf::from("/default"));
8608 let mut dot = task();
8609 dot.repo = PathBuf::from(".");
8610 assert_eq!(repo_for(&dot, fallback), PathBuf::from("/default"));
8611 assert_eq!(
8612 repo_for(&task(), fallback),
8613 PathBuf::from("/repo"),
8614 "a task that names a repository keeps it"
8615 );
8616 }
8617
8618 #[test]
8619 fn task_implementers_edit_inherits_daemon_config_and_merge() {
8620 let dir = tempfile::tempdir().unwrap();
8621 let path = dir.path().join("explicit.toml");
8622 std::fs::write(&path, "[graph]\nimplementers = 3\njudges = 1\n").unwrap();
8623 let opts = Opts {
8624 config: Some(path),
8625 merge: Some("none".into()),
8626 ..Default::default()
8627 };
8628 let mut t = task();
8629 t.set_implementers(
8630 2,
8631 &toml::from_str("[[agents]]\nid = 'mock'\nkind = 'command'\ncommand = ['true']\n")
8632 .unwrap(),
8633 )
8634 .unwrap();
8635 let cfg = prepare_for(dir.path(), &opts, &t).unwrap();
8636 assert_eq!(cfg.graph.implementers, 2);
8637 assert_eq!(cfg.graph.judges, 1);
8638 assert_eq!(cfg.merge.mode, MergeMode::None);
8639 t.set_implementers(1, &cfg).unwrap();
8640 let mut cfg = prepare_for(dir.path(), &opts, &t).unwrap();
8641 apply_task_implementers(&mut cfg, &t);
8642 assert_eq!(cfg.graph.implementers, 1);
8643 assert_eq!(cfg.graph.judges, 1);
8644 assert_eq!(cfg.merge.mode, MergeMode::None);
8645 }
8646
8647 #[test]
8648 fn task_implementers_direct_overrides_stay_isolated_after_edits() {
8649 let dir = tempfile::tempdir().unwrap();
8650 let repo = dir.path().join("repo");
8651 std::fs::create_dir(&repo).unwrap();
8652 std::fs::write(
8653 repo.join("magi.toml"),
8654 "[graph]\njudges = 2\n[merge]\nmode = 'none'\n",
8655 )
8656 .unwrap();
8657 let daemon_config = dir.path().join("daemon.toml");
8658 std::fs::write(
8659 &daemon_config,
8660 "[graph]\njudges = 1\n[merge]\nmode = 'pr'\n",
8661 )
8662 .unwrap();
8663 let opts = Opts {
8664 config: Some(daemon_config),
8665 merge: Some("pr".into()),
8666 ..Default::default()
8667 };
8668 let q = Queue::at(dir.path().join("queue"));
8669 let validation: Config =
8670 toml::from_str("[[agents]]\nid = 'mock'\nkind = 'command'\ncommand = ['true']\n")
8671 .unwrap();
8672 for record in ["{}", r#"{"candidates":2}"#] {
8674 let overrides = serde_json::from_str(record).unwrap();
8675 let filing = crate::direct::Filing {
8676 instruction: "work".into(),
8677 title: "direct".into(),
8678 repo: repo.clone(),
8679 source: crate::queue::Source::Human,
8680 solo: false,
8681 overrides,
8682 review_of: None,
8683 sources: Default::default(),
8684 };
8685 let filed =
8686 crate::direct::file(&q, dir.path(), Timestamp::now(), 1, filing, true).unwrap();
8687 let crate::direct::Filed::Standalone {
8688 mut task,
8689 claim: _claim,
8690 } = filed
8691 else {
8692 panic!("local filing")
8693 };
8694 let before = prepare_for(&repo, &opts, &task).unwrap();
8695 assert_eq!(before.graph.judges, 2, "daemon --config must not leak");
8696 assert_eq!(
8697 before.merge.mode,
8698 MergeMode::None,
8699 "daemon --merge must not leak"
8700 );
8701 for n in [2, 1, 2] {
8702 task.set_implementers(n, &validation).unwrap();
8703 q.put(&mut task).unwrap();
8704 task = q.get(&task.id).unwrap();
8705 assert!(!task.overrides.as_ref().unwrap().inherit_daemon_options);
8706 let mut cfg = prepare_for(&repo, &opts, &task).unwrap();
8707 apply_task_implementers(&mut cfg, &task);
8708 assert_eq!(cfg.graph.implementers, usize::from(n));
8709 assert_eq!(cfg.graph.judges, 2);
8710 assert_eq!(cfg.merge.mode, MergeMode::None);
8711 }
8712 }
8713 }
8714
8715 #[test]
8716 fn task_implementers_mismatch_keeps_explicit_reviews_review_only() {
8717 let cfg: Config =
8718 toml::from_str("[[agents]]\nid = 'mock'\nkind = 'command'\ncommand = ['true']\n")
8719 .unwrap();
8720 let mut t = task();
8721 t.review_of = Some("review-target".into());
8722 t.hold_manual(Some("retry".into()));
8723 let mut previous = RunState::new(
8724 PathBuf::from("/repo"),
8725 "main".into(),
8726 "abc1234".into(),
8727 "review".into(),
8728 cfg.clone(),
8729 );
8730 previous.config.graph.implementers = 2;
8731 for count in [1, 3] {
8732 t.set_implementers(count, &cfg).unwrap();
8733 t.release();
8734 assert_eq!(t.review_of.as_deref(), Some("review-target"));
8735 for proposed in [
8736 Starter::Resume(previous.id.clone()),
8737 Starter::Review("recovery-target".into()),
8738 ] {
8739 assert_eq!(
8740 starter_with_implementers(&t, Some(&previous), proposed),
8741 Starter::Review("review-target".into())
8742 );
8743 }
8744 let mut configured = cfg.clone();
8745 apply_task_implementers(&mut configured, &t);
8746 assert_eq!(configured.graph.implementers, usize::from(count));
8747 assert_eq!(previous.config.graph.implementers, 2);
8748 }
8749 }
8750
8751 #[test]
8752 fn task_implementers_dispatch_and_resume_agree() {
8753 let mut t = task();
8754 let cfg: crate::config::Config =
8755 toml::from_str("[[agents]]\nid = 'mock'\nkind = 'command'\ncommand = ['true']\n")
8756 .unwrap();
8757 let mut run = RunState::new(
8758 PathBuf::from("/repo"),
8759 "main".into(),
8760 "abc1234".into(),
8761 "work".into(),
8762 cfg.clone(),
8763 );
8764 run.config.graph.implementers = 2;
8765 for n in [2, 1, 3, 2] {
8766 t.set_implementers(n, &cfg).unwrap();
8767 let mut dispatched = cfg.clone();
8768 apply_task_implementers(&mut dispatched, &t);
8769 assert_eq!(dispatched.graph.implementers, usize::from(n));
8770 assert_eq!(implementers_match(&t, &run), n == 2);
8771 for proposed in [
8772 Starter::Resume(run.id.clone()),
8773 Starter::Review("old-branch".into()),
8774 ] {
8775 let expected = if n == 2 {
8776 proposed.clone()
8777 } else {
8778 Starter::Start
8779 };
8780 assert_eq!(
8781 starter_with_implementers(&t, Some(&run), proposed),
8782 expected
8783 );
8784 }
8785 assert_eq!(
8786 run.config.graph.implementers, 2,
8787 "history must not be rewritten"
8788 );
8789 }
8790 t.solo = false;
8791 t.overrides = None;
8792 assert!(implementers_match(&t, &run));
8793 }
8794
8795 #[test]
8796 fn a_solo_task_runs_with_one_candidate_and_a_plain_task_keeps_the_configs() {
8797 let mut solo_cfg = Config::default();
8803 solo_cfg.graph.implementers = 3;
8804 let mut solo_task = task();
8805 solo_task.solo = true;
8806 apply_task_implementers(&mut solo_cfg, &solo_task);
8807 assert_eq!(solo_cfg.graph.implementers, 1);
8808
8809 let mut plain_cfg = Config::default();
8810 plain_cfg.graph.implementers = 3;
8811 let plain_task = task();
8812 assert!(!plain_task.solo);
8813 apply_task_implementers(&mut plain_cfg, &plain_task);
8814 assert_eq!(
8815 plain_cfg.graph.implementers, 3,
8816 "a task that did not ask to run alone keeps the config's candidates"
8817 );
8818 }
8819
8820 fn loss(seat: &str, at: &str, reset: Option<&str>) -> QuotaLoss {
8821 QuotaLoss {
8822 seat: seat.into(),
8823 node: "judge".into(),
8824 at: at.parse().unwrap(),
8825 reset: reset.map(str::to_string),
8826 }
8827 }
8828
8829 #[test]
8830 fn a_resumed_run_with_only_old_quota_losses_arms_no_cooldown() {
8831 let old: Vec<QuotaLoss> = (1..=4)
8832 .map(|i| {
8833 loss(
8834 &format!("judge-{i}"),
8835 "2026-09-23T05:23:00Z",
8836 Some("2:40pm (Asia/Tokyo)"),
8837 )
8838 })
8839 .collect();
8840 let fresh = losses_this_attempt(&old, &old);
8841 assert!(fresh.is_empty());
8842 assert_eq!(cooldown_until(&fresh, Timestamp::now()), None);
8843 }
8845
8846 #[test]
8847 fn a_new_quota_loss_during_the_attempt_still_arms_the_cooldown() {
8848 let old = vec![loss("judge-1", "2026-09-23T05:23:00Z", None)];
8849 let now = Timestamp::now();
8850 let mut after = old.clone();
8851 after.push(loss("judge-2", &now.to_string(), None));
8852 let fresh = losses_this_attempt(&old, &after);
8853 assert_eq!(fresh, vec![after[1].clone()]);
8854 let until = cooldown_until(&fresh, now).expect("a fresh loss arms the cooldown");
8855 assert_eq!(
8856 until,
8857 now + jiff::SignedDuration::from_secs(QUOTA_WAIT_FALLBACK.as_secs() as i64)
8858 );
8859 }
8860
8861 #[test]
8862 fn a_recovered_seat_dropping_out_of_the_history_does_not_hide_a_new_loss() {
8863 let before = vec![
8866 loss("judge-1", "2026-09-23T05:23:00Z", None),
8867 loss("judge-2", "2026-09-23T05:24:00Z", None),
8868 ];
8869 let after = vec![
8870 loss("judge-2", "2026-09-23T05:24:00Z", None),
8871 loss("judge-1", "2026-09-24T01:00:00Z", None),
8872 ];
8873 assert_eq!(losses_this_attempt(&before, &after), vec![after[1].clone()]);
8874 }
8875
8876 #[test]
8877 fn merge_overrides_are_parsed_or_refused() {
8878 assert_eq!(merge_mode("none").unwrap(), MergeMode::None);
8879 assert_eq!(merge_mode("local").unwrap(), MergeMode::Local);
8880 assert_eq!(merge_mode("pr").unwrap(), MergeMode::Pr);
8881 assert!(merge_mode("squash").is_err());
8882 }
8883
8884 #[test]
8885 fn quota_wait_uses_a_future_reset_time_capped_and_falls_back_otherwise() {
8886 let now = Timestamp::now();
8887 let fallback = Duration::from_secs(300);
8888 let cap = Duration::from_secs(1800);
8889
8890 assert_eq!(quota_wait(None, now, fallback, cap), fallback);
8892
8893 let soon = now + jiff::SignedDuration::from_secs(600);
8895 assert_eq!(
8896 quota_wait(Some(soon), now, fallback, cap),
8897 Duration::from_secs(600)
8898 );
8899
8900 let past = now - jiff::SignedDuration::from_secs(60);
8903 assert_eq!(quota_wait(Some(past), now, fallback, cap), fallback);
8904
8905 let far = now + jiff::SignedDuration::from_secs(3 * 3600);
8908 assert_eq!(quota_wait(Some(far), now, fallback, cap), cap);
8909 }
8910
8911 #[test]
8912 fn parse_reset_hint_reads_the_claude_cli_shape_and_rolls_a_past_clock_to_tomorrow() {
8913 let now = "2026-09-07T02:50:00Z".parse::<Timestamp>().unwrap();
8914
8915 let at = parse_reset_hint("4:50am (UTC)", now, now).expect("a recognised shape parses");
8916 assert_eq!(at.to_string(), "2026-09-07T04:50:00Z");
8917
8918 let already_past =
8922 parse_reset_hint("1:00am (UTC)", now, now).expect("a recognised shape parses");
8923 assert_eq!(already_past.to_string(), "2026-09-08T01:00:00Z");
8924
8925 assert!(
8926 parse_reset_hint("session limit reached", now, now).is_none(),
8927 "free text with no recognised shape is not guessed at"
8928 );
8929 assert!(
8930 parse_reset_hint("4:50am (Nowhere/Fake)", now, now).is_none(),
8931 "an unresolvable zone name is not guessed at either"
8932 );
8933 }
8934
8935 #[test]
8936 fn parse_reset_hint_reads_the_codex_cli_shape_with_no_year_rollover_needed() {
8937 let now = "2026-09-07T02:50:00Z".parse::<Timestamp>().unwrap();
8938
8939 let at = parse_reset_hint(
8940 "You've hit your usage limit. Visit \
8941 https://chatgpt.com/codex/settings/usage to purchase more \
8942 credits or try again at Sep 19th, 2026 5:10 PM.",
8943 now,
8944 now,
8945 )
8946 .expect("the codex reset wording is a recognised shape");
8947 assert_eq!(at.to_string(), "2026-09-19T17:10:00Z");
8948
8949 let earlier = parse_reset_hint("try again at Jan 2nd, 2026 1:00 AM.", now, now)
8954 .expect("an explicit year needs no rollover");
8955 assert_eq!(earlier.to_string(), "2026-01-02T01:00:00Z");
8956
8957 assert!(
8958 parse_reset_hint("try again at Sep 19th, 26 5:10 PM.", now, now).is_none(),
8959 "a two-digit year is not the documented shape and is not guessed at"
8960 );
8961 assert!(
8962 parse_reset_hint("try again at Sept 19th, 2026 5:10 PM.", now, now).is_none(),
8963 "a four-letter month name is not the documented three-letter abbreviation"
8964 );
8965 assert!(
8966 parse_reset_hint("try again at Sep 19th, 2026 5:10 PM (UTC).", now, now).is_none(),
8967 "an explicit zone on the dated shape is a format nobody has \
8968 documented, and is refused rather than guessed at as UTC"
8969 );
8970 }
8971
8972 #[test]
8973 fn parse_reset_hint_reads_agys_relative_shape_from_when_the_loss_was_recorded() {
8974 let now = "2026-09-24T12:00:00Z".parse::<Timestamp>().unwrap();
8975 let recorded = "2026-09-24T08:00:00Z".parse::<Timestamp>().unwrap();
8976
8977 let at = parse_reset_hint("in 1h2m49s", now, recorded).expect("agy's shape parses");
8978 assert_eq!(at.as_second() - recorded.as_second(), 3769);
8979
8980 let partial = parse_reset_hint("in 45m", now, recorded).expect("units are optional");
8981 assert_eq!(partial.as_second() - recorded.as_second(), 45 * 60);
8982
8983 for bad in ["in ", "in 45", "in 3x", "in m", "in 1h junk", "1h2m"] {
8984 assert!(
8985 parse_reset_hint(bad, now, recorded).is_none(),
8986 "{bad:?} must not be guessed at"
8987 );
8988 }
8989 }
8990
8991 fn idle_loop(dir: &Path) -> (Opts, Queue, PathBuf, PathBuf, PathBuf) {
8995 let config = dir.join("magi.toml");
8996 std::fs::write(
8997 &config,
8998 "[disk]\nmin_free_bytes = 0\nauto_fold = false\ncache_limit_bytes = 0\n",
8999 )
9000 .unwrap();
9001 let opts = Opts {
9002 poll: Duration::from_secs(30),
9003 config: Some(config),
9004 repo: dir.join("repo"),
9008 ..Opts::default()
9009 };
9010 let home = dir.join("home");
9019 let worktrees = dir.join("wt");
9020 (
9021 opts,
9022 Queue::at(dir.join("queue")),
9023 home.join("daemon.json"),
9024 home,
9025 worktrees,
9026 )
9027 }
9028
9029 #[test]
9030 fn a_stop_is_idempotent_and_once_set_stays_set() {
9031 let stop = Stop::new();
9032 assert!(!stop.stopped());
9033
9034 stop.stop();
9035 assert!(stop.stopped());
9036 stop.stop();
9037 assert!(stop.stopped(), "a second stop is not a toggle");
9038
9039 let shared = stop.clone();
9040 assert!(
9041 shared.stopped(),
9042 "a clone is the same stop; that is how the loop and its caller share one"
9043 );
9044 }
9045
9046 #[test]
9047 fn only_a_stop_with_a_run_in_flight_reads_as_finishing() {
9048 let stop = Stop::new();
9049 stop.enter();
9050 assert!(
9051 !stop.finishing(),
9052 "a busy loop nobody has asked to stop is just running"
9053 );
9054
9055 stop.stop();
9056 assert!(
9057 stop.finishing(),
9058 "a stop asked for mid-run has not landed until the run is settled"
9059 );
9060
9061 stop.exit();
9062 assert!(
9063 !stop.finishing(),
9064 "once the run is settled the stop has landed and there is nothing to finish"
9065 );
9066 }
9067
9068 #[test]
9069 fn finishing_stays_true_until_the_last_of_several_runs_exits() {
9070 let stop = Stop::new();
9071 stop.enter();
9072 stop.enter();
9073 stop.stop();
9074 assert!(stop.finishing(), "two runs still in flight");
9075
9076 stop.exit();
9077 assert!(
9078 stop.finishing(),
9079 "one run finished, but a sibling is still working"
9080 );
9081
9082 stop.exit();
9083 assert!(
9084 !stop.finishing(),
9085 "the last run out is what actually lands the stop"
9086 );
9087 }
9088
9089 #[tokio::test]
9090 async fn a_loop_already_asked_to_stop_returns_without_waiting_out_a_poll() {
9091 let dir = tempfile::tempdir().unwrap();
9092 let (opts, queue, status_file, home, worktrees) = idle_loop(dir.path());
9093 let stop = Stop::new();
9094 stop.stop();
9095
9096 let began = std::time::Instant::now();
9097 tokio::time::timeout(
9098 Duration::from_secs(2),
9099 drive(&opts, &queue, &status_file, &home, &worktrees, &stop),
9100 )
9101 .await
9102 .expect("a stopped loop must return, not sit out its poll interval")
9103 .expect("the loop's own setup and teardown must not fail");
9104 assert!(
9105 began.elapsed() < opts.poll,
9106 "returned only after {:?}, which is a poll interval, not a stop",
9107 began.elapsed()
9108 );
9109 }
9110
9111 #[tokio::test]
9112 async fn a_stop_while_idle_wakes_the_wait_instead_of_sleeping_it_out() {
9113 let dir = tempfile::tempdir().unwrap();
9114 let (opts, queue, status_file, home, worktrees) = idle_loop(dir.path());
9115 let stop = Stop::new();
9116
9117 let asker = {
9120 let stop = stop.clone();
9121 tokio::spawn(async move {
9122 tokio::time::sleep(Duration::from_millis(20)).await;
9123 stop.stop();
9124 })
9125 };
9126
9127 let began = std::time::Instant::now();
9128 tokio::time::timeout(
9129 Duration::from_secs(2),
9130 drive(&opts, &queue, &status_file, &home, &worktrees, &stop),
9131 )
9132 .await
9133 .expect("a stop asked for while idle must wake the wait")
9134 .expect("the loop's own setup and teardown must not fail");
9135 asker.await.unwrap();
9136 assert!(
9137 began.elapsed() < opts.poll,
9138 "returned only after {:?}, so the stop waited on the sleep",
9139 began.elapsed()
9140 );
9141 }
9142
9143 #[tokio::test]
9144 async fn a_stopped_loop_leaves_no_status_file_claiming_it_is_running() {
9145 let dir = tempfile::tempdir().unwrap();
9146 let (opts, queue, status_file, home, worktrees) = idle_loop(dir.path());
9147 let stop = Stop::new();
9148 stop.stop();
9149
9150 tokio::time::timeout(
9151 Duration::from_secs(2),
9152 drive(&opts, &queue, &status_file, &home, &worktrees, &stop),
9153 )
9154 .await
9155 .expect("a stopped loop must return")
9156 .expect("the loop's own setup and teardown must not fail");
9157
9158 assert!(
9159 home.is_dir(),
9160 "the loop did publish a status file, so its removal is the teardown and not an absence"
9161 );
9162 assert!(
9163 !status_file.exists(),
9164 "a stopped loop clears its status file"
9165 );
9166 assert!(
9167 read_status(&home).is_none(),
9168 "a reader must see no daemon at all, not a heartbeat that merely stopped"
9169 );
9170 }
9171
9172 #[tokio::test]
9173 async fn once_runs_startup_housekeeping_before_an_empty_queue_exits() {
9174 let dir = tempfile::tempdir().unwrap();
9175 let (mut opts, queue, status_file, home, worktrees) = idle_loop(dir.path());
9176 opts.once = true;
9177
9178 let mut settled = RunState::new(
9179 dir.path().join("repo"),
9180 "main".to_owned(),
9181 "abc1234".to_owned(),
9182 "fixture".to_owned(),
9183 Config::default(),
9184 );
9185 settled.status = RunStatus::Ready;
9186 let run_dir = home.join("runs").join(&settled.id);
9187 std::fs::create_dir_all(&run_dir).unwrap();
9188 std::fs::write(
9189 run_dir.join("run.json"),
9190 serde_json::to_string_pretty(&settled).unwrap(),
9191 )
9192 .unwrap();
9193 let questions = Questions::at(home.join("questions"));
9194 let mut question = ask::Question::new(
9195 settled.id.clone(),
9196 "review".to_owned(),
9197 "reviewer-1".to_owned(),
9198 "Continue?".to_owned(),
9199 String::new(),
9200 Vec::new(),
9201 );
9202 questions.put(&mut question).unwrap();
9203
9204 drive(&opts, &queue, &status_file, &home, &worktrees, &Stop::new())
9205 .await
9206 .unwrap();
9207
9208 assert_eq!(
9209 questions.get(&question.id).unwrap().status,
9210 ask::QuestionStatus::Abandoned,
9211 "an empty --once drain still performs startup question cleanup"
9212 );
9213 }
9214
9215 #[test]
9216 fn cache_check_due_fires_immediately_then_waits_out_its_own_interval() {
9217 let t0 = "2026-09-15T00:00:00Z".parse::<Timestamp>().unwrap();
9218
9219 assert!(
9220 cache_check_due(None, t0, CACHE_CHECK_INTERVAL_SECS),
9221 "never checked before: due at once"
9222 );
9223
9224 let one_sec_later = t0 + jiff::SignedDuration::from_secs(1);
9225 assert!(
9226 !cache_check_due(Some(t0), one_sec_later, CACHE_CHECK_INTERVAL_SECS),
9227 "well inside the interval: not due yet"
9228 );
9229
9230 let at_the_edge = t0 + jiff::SignedDuration::from_secs(CACHE_CHECK_INTERVAL_SECS as i64);
9231 assert!(
9232 !cache_check_due(Some(t0), at_the_edge, CACHE_CHECK_INTERVAL_SECS),
9233 "exactly at the edge: not yet due, same convention as `clean::due`"
9234 );
9235
9236 let past_it = t0 + jiff::SignedDuration::from_secs(CACHE_CHECK_INTERVAL_SECS as i64 + 1);
9237 assert!(
9238 cache_check_due(Some(t0), past_it, CACHE_CHECK_INTERVAL_SECS),
9239 "past the interval: due again"
9240 );
9241 }
9242
9243 fn cache_check_opts(dir: &Path, cache_dir: &Path, limit_bytes: u64) -> Opts {
9248 let config = dir.join("magi.toml");
9249 std::fs::write(
9255 &config,
9256 format!(
9257 "[disk]\nmin_free_bytes = 0\nauto_fold = false\ncache_limit_bytes = {limit_bytes}\n\n\
9258 [verify]\ngate = ['CARGO_TARGET_DIR={} cargo make check']\n",
9259 cache_dir.display()
9260 ),
9261 )
9262 .unwrap();
9263 Opts {
9264 config: Some(config),
9265 repo: dir.join("repo"),
9266 ..Opts::default()
9267 }
9268 }
9269
9270 #[tokio::test]
9271 async fn maybe_prune_cache_between_runs_reprunes_only_once_its_own_interval_elapses() {
9272 let dir = tempfile::tempdir().unwrap();
9273 let home = dir.path().join("home");
9274 let cache_dir = dir.path().join("cache");
9275 std::fs::create_dir_all(&cache_dir).unwrap();
9276 std::fs::write(cache_dir.join("a"), vec![0u8; 10]).unwrap();
9277 let opts = cache_check_opts(dir.path(), &cache_dir, 1);
9278
9279 let running = Stop::new();
9282 let mut last_checked = None;
9283 let t0 = "2026-09-15T00:00:00Z".parse::<Timestamp>().unwrap();
9284 maybe_prune_cache_between_runs(&opts.repo, &opts, &home, &running, &mut last_checked, t0)
9285 .await;
9286 assert_eq!(
9287 crate::disk::dir_size(&cache_dir),
9288 0,
9289 "over the cap on the first check ever: pruned at once, no idle queue required"
9290 );
9291 assert_eq!(last_checked, Some(t0));
9292
9293 std::fs::write(cache_dir.join("b"), vec![0u8; 10]).unwrap();
9295 let too_soon = t0 + jiff::SignedDuration::from_secs(1);
9296 maybe_prune_cache_between_runs(
9297 &opts.repo,
9298 &opts,
9299 &home,
9300 &running,
9301 &mut last_checked,
9302 too_soon,
9303 )
9304 .await;
9305 assert_eq!(
9306 crate::disk::dir_size(&cache_dir),
9307 10,
9308 "too soon since the last check: left alone rather than rescanned every call"
9309 );
9310 assert_eq!(
9311 last_checked,
9312 Some(t0),
9313 "an idle check does not reset the clock"
9314 );
9315
9316 let due_again = t0 + jiff::SignedDuration::from_secs(CACHE_CHECK_INTERVAL_SECS as i64 + 1);
9318 maybe_prune_cache_between_runs(
9319 &opts.repo,
9320 &opts,
9321 &home,
9322 &running,
9323 &mut last_checked,
9324 due_again,
9325 )
9326 .await;
9327 assert_eq!(
9328 crate::disk::dir_size(&cache_dir),
9329 0,
9330 "due again: pruned back under the cap"
9331 );
9332 }
9333
9334 #[tokio::test]
9342 async fn a_stop_already_asked_for_skips_the_between_runs_cache_walk() {
9343 let dir = tempfile::tempdir().unwrap();
9344 let home = dir.path().join("home");
9345 let cache_dir = dir.path().join("cache");
9346 std::fs::create_dir_all(&cache_dir).unwrap();
9347 std::fs::write(cache_dir.join("a"), vec![0u8; 10]).unwrap();
9348 let opts = cache_check_opts(dir.path(), &cache_dir, 1);
9349
9350 let stop = Stop::new();
9351 stop.stop();
9352 assert!(
9353 !stop.finishing(),
9354 "no run is in flight at a between-runs boundary, so nothing else \
9355 would tell the operator this stop had not taken effect yet"
9356 );
9357
9358 let mut last_checked = None;
9359 let t0 = "2026-09-15T00:00:00Z".parse::<Timestamp>().unwrap();
9360 maybe_prune_cache_between_runs(&opts.repo, &opts, &home, &stop, &mut last_checked, t0)
9361 .await;
9362 assert_eq!(
9363 crate::disk::dir_size(&cache_dir),
9364 10,
9365 "over its cap, and due for the first check ever, but a stop outranks \
9366 it: the cap is a standing policy the next start measures again"
9367 );
9368 assert_eq!(
9369 last_checked, None,
9370 "a check that never happened must not claim the interval"
9371 );
9372 }
9373
9374 #[tokio::test]
9385 async fn cache_prune_reaches_a_queue_that_never_goes_idle() {
9386 let dir = tempfile::tempdir().unwrap();
9387 let cache_dir = dir.path().join("cache");
9388 std::fs::create_dir_all(&cache_dir).unwrap();
9389 std::fs::write(cache_dir.join("stale"), vec![0u8; 4096]).unwrap();
9390
9391 let mut opts = cache_check_opts(dir.path(), &cache_dir, 1);
9392 opts.poll = Duration::from_millis(20);
9393 opts.max_attempts = 1_000;
9394
9395 let queue = Queue::at(dir.path().join("queue"));
9396 for i in 0..128 {
9397 let mut task = Task::new(
9398 format!("queued work {i}"),
9399 format!("queued work {i}"),
9400 opts.repo.clone(),
9401 Source::Human,
9402 );
9403 queue.put(&mut task).unwrap();
9404 }
9405
9406 let home = dir.path().join("home");
9407 let worktrees = dir.path().join("wt");
9408 let status_file = home.join("daemon.json");
9409 let stop = Stop::new();
9410 let stopper = {
9411 let stop = stop.clone();
9412 let queue = Queue::at(dir.path().join("queue"));
9413 tokio::spawn(async move {
9414 for _ in 0..2_000 {
9418 if queue.list().iter().filter(|t| t.attempts == 1).count() >= 2 {
9419 break;
9420 }
9421 tokio::time::sleep(Duration::from_millis(5)).await;
9422 }
9423 stop.stop();
9424 })
9425 };
9426
9427 tokio::time::timeout(
9428 Duration::from_secs(20),
9429 drive(&opts, &queue, &status_file, &home, &worktrees, &stop),
9430 )
9431 .await
9432 .expect("the loop must not hang on a queue that keeps producing failing work")
9433 .expect("the loop's own setup and teardown must not fail");
9434 stopper.await.unwrap();
9435
9436 let after = queue.list();
9437 let completed = after.iter().filter(|t| t.attempts == 1).count();
9438 assert!(
9439 completed >= 2,
9440 "the harness must have processed multiple tasks to exercise a busy queue \
9441 (got {completed} completed task(s))"
9442 );
9443 assert!(
9444 after.iter().any(|t| t.status.runnable()),
9445 "unprocessed tasks must remain: the external stop, rather than an \
9446 exhausted backlog, must end the test"
9447 );
9448 assert!(
9449 after
9450 .iter()
9451 .filter(|t| t.attempts > 0)
9452 .all(|t| { t.attempts == 1 && t.status == TaskStatus::Held }),
9453 "a busy backlog must not retry a task's unchanged startup failure"
9454 );
9455
9456 assert_eq!(
9457 crate::disk::dir_size(&cache_dir),
9458 0,
9459 "an oversized cache must not be left to grow unboundedly just because the \
9460 queue kept the loop busy the whole time"
9461 );
9462 }
9463
9464 #[test]
9465 fn task_question_reconciliation_keeps_references_and_retires_manual_releases() {
9466 let dir = tempfile::tempdir().unwrap();
9467 let queue = Queue::at(dir.path().join("queue"));
9468 let questions = Questions::at(dir.path().join("questions"));
9469 let mut task = task();
9470 queue.put(&mut task).unwrap();
9471
9472 let mut task_question = ask::Question::new(
9473 task.id.clone(),
9474 crate::conduct::NODE.to_owned(),
9475 "conduct".to_owned(),
9476 "Which backend?".to_owned(),
9477 String::new(),
9478 Vec::new(),
9479 );
9480 questions.put(&mut task_question).unwrap();
9481 task.block(vec![task_question.id.clone()], None);
9482 queue.put(&mut task).unwrap();
9483
9484 let mut run_question = ask::Question::new(
9485 "20260101-000000-run1".to_owned(),
9486 "review".to_owned(),
9487 "reviewer-1".to_owned(),
9488 "Run question".to_owned(),
9489 String::new(),
9490 Vec::new(),
9491 );
9492 questions.put(&mut run_question).unwrap();
9493
9494 let mut coincidental = ask::Question::new(
9499 task.id.clone(),
9500 "review".to_owned(),
9501 "reviewer-1".to_owned(),
9502 "Unrelated review question".to_owned(),
9503 String::new(),
9504 Vec::new(),
9505 );
9506 questions.put(&mut coincidental).unwrap();
9507
9508 reconcile_task_questions(&queue, &questions);
9509 assert!(questions.get(&task_question.id).unwrap().status.open());
9510 assert!(questions.get(&run_question.id).unwrap().status.open());
9511 assert!(questions.get(&coincidental.id).unwrap().status.open());
9512
9513 task.release();
9514 queue.put(&mut task).unwrap();
9515 reconcile_task_questions(&queue, &questions);
9516 assert_eq!(
9517 questions.get(&task_question.id).unwrap().status,
9518 ask::QuestionStatus::Abandoned
9519 );
9520 assert!(
9521 questions.get(&run_question.id).unwrap().status.open(),
9522 "run questions remain the run janitor's responsibility"
9523 );
9524 assert!(
9525 questions.get(&coincidental.id).unwrap().status.open(),
9526 "a non-conductor question must not be abandoned just because its \
9527 run id coincides with a task id"
9528 );
9529 }
9530
9531 #[test]
9532 fn a_freshly_started_running_task_is_never_stalled() {
9533 let dir = tempfile::tempdir().unwrap();
9534 let mut t = task();
9535 t.start("run-1".to_owned());
9536 assert!(!is_stalled(&t, dir.path(), Timestamp::now()));
9539 }
9540
9541 #[test]
9542 fn a_long_running_task_with_no_live_daemon_is_stalled() {
9543 let dir = tempfile::tempdir().unwrap();
9544 let mut t = task();
9545 t.start("run-1".to_owned());
9546 t.updated_at = Timestamp::now()
9547 - jiff::SignedDuration::from_secs(STALLED_RUNNING.as_secs() as i64 + 60);
9548 assert!(is_stalled(&t, dir.path(), Timestamp::now()));
9549 assert_eq!(
9550 stalled_tasks(
9551 &Queue::at(dir.path().join("q")),
9552 dir.path(),
9553 Timestamp::now()
9554 )
9555 .len(),
9556 0,
9557 "the task was never written to this queue"
9558 );
9559 }
9560
9561 #[test]
9562 fn a_long_running_task_a_live_daemon_still_names_is_not_stalled() {
9563 let dir = tempfile::tempdir().unwrap();
9564 let mut t = task();
9565 t.id = "20260903-080340-0167".to_owned();
9566 t.start("20260903-080619-01c2".to_owned());
9567 t.updated_at = Timestamp::now()
9568 - jiff::SignedDuration::from_secs(STALLED_RUNNING.as_secs() as i64 + 60);
9569
9570 let mut status = Status::new();
9571 status.current = vec![Current {
9572 task: t.id.clone(),
9573 run: "20260903-080619-01c2".to_owned(),
9574 }];
9575 write_status_to(&dir.path().join("daemon.json"), &status).unwrap();
9576
9577 assert!(
9578 !is_stalled(&t, dir.path(), Timestamp::now()),
9579 "a live daemon's own heartbeat rules out stalled, however long the task has run"
9580 );
9581 }
9582
9583 fn backdate_task(queue: &Queue, id: &str, seconds_ago: i64) {
9587 let path = queue.path_of(id);
9588 let body = std::fs::read_to_string(&path).unwrap();
9589 let mut v: serde_json::Value = serde_json::from_str(&body).unwrap();
9590 let old = Timestamp::now() - jiff::SignedDuration::from_secs(seconds_ago);
9591 v["updated_at"] = serde_json::Value::String(old.to_string());
9592 std::fs::write(&path, serde_json::to_string_pretty(&v).unwrap()).unwrap();
9593 }
9594
9595 #[test]
9596 fn stalled_tasks_still_reaches_a_task_reclaim_could_not_claim_yet() {
9597 let dir = tempfile::tempdir().unwrap();
9610 let queue = Queue::at(dir.path().join("queue"));
9611 let home = dir.path().join("home");
9612
9613 let mut t = task();
9614 t.id = "20260101-000001-lock".to_owned();
9615 t.start("run-1".to_owned());
9616 queue.put(&mut t).unwrap();
9617 backdate_task(&queue, &t.id, STALLED_RUNNING.as_secs() as i64 + 60);
9618 std::fs::write(
9619 dir.path().join("queue").join(format!("{}.lock", t.id)),
9620 "not a pid",
9621 )
9622 .unwrap();
9623
9624 let now = Timestamp::now();
9625 assert!(
9626 reclaim_orphaned_running(&queue, 2).is_empty(),
9627 "the unparseable lock is still well within STALE_CLAIM, so the claim fails \
9628 and reclaim must leave the task alone"
9629 );
9630 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Running);
9631
9632 let stalled = stalled_tasks(&queue, &home, now);
9633 assert_eq!(
9634 stalled.len(),
9635 1,
9636 "reclaim's inability to claim it yet must not hide it from the conductor"
9637 );
9638 assert_eq!(stalled[0].id, t.id);
9639 }
9640
9641 #[test]
9642 fn ordinary_dead_daemon_task_is_shown_stalled_before_reclaim_and_can_be_requeued() {
9643 let dir = tempfile::tempdir().unwrap();
9644 crate::run::set_home(dir.path().join("run-home"));
9645 let queue = Queue::at(dir.path().join("queue"));
9646 let home = dir.path().join("home");
9647 let questions = Questions::at(dir.path().join("questions"));
9648
9649 let mut t = task();
9650 t.id = "20260101-000003-dead".to_owned();
9651 t.start("missing-run".to_owned());
9652 queue.put(&mut t).unwrap();
9653 backdate_task(&queue, &t.id, STALLED_RUNNING.as_secs() as i64 + 60);
9654
9655 let stalled = stalled_tasks(&queue, &home, Timestamp::now());
9658 assert_eq!(
9659 stalled.iter().map(|task| &task.id).collect::<Vec<_>>(),
9660 [&t.id]
9661 );
9662 assert_eq!(reclaim_orphaned_running(&queue, 2), [t.id.clone()]);
9663 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Held);
9664
9665 crate::conduct::apply(
9668 &queue,
9669 &questions,
9670 &crate::conduct::Verdict {
9671 decisions: vec![crate::conduct::Decision {
9672 id: t.id.clone(),
9673 recovery: Some(crate::conduct::Recovery::Requeue),
9674 ..crate::conduct::Decision::default()
9675 }],
9676 },
9677 )
9678 .unwrap();
9679 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Queued);
9680 }
9681
9682 #[test]
9683 fn stalled_tasks_reports_exactly_the_tasks_is_stalled_agrees_on() {
9684 let dir = tempfile::tempdir().unwrap();
9685 let queue = Queue::at(dir.path().join("queue"));
9686 let home = dir.path().join("home");
9687
9688 let mut fresh = task();
9689 fresh.id = "20260101-000001-aaaa".to_owned();
9690 fresh.start("run-1".to_owned());
9691 queue.put(&mut fresh).unwrap();
9692
9693 let mut old = task();
9694 old.id = "20260101-000002-bbbb".to_owned();
9695 old.start("run-2".to_owned());
9696 queue.put(&mut old).unwrap();
9697 backdate_task(&queue, &old.id, STALLED_RUNNING.as_secs() as i64 + 60);
9698
9699 let stalled = stalled_tasks(&queue, &home, Timestamp::now());
9700 assert_eq!(stalled.len(), 1);
9701 assert_eq!(stalled[0].id, old.id);
9702 }
9703
9704 #[test]
9705 fn queued_and_finished_task_views_partition_by_status() {
9706 let dir = tempfile::tempdir().unwrap();
9707 let queue = Queue::at(dir.path().join("queue"));
9708
9709 let mut queued = task();
9710 queued.id = "20260101-000001-aaaa".to_owned();
9711 queue.put(&mut queued).unwrap();
9712
9713 let mut failed = task();
9714 failed.id = "20260101-000002-bbbb".to_owned();
9715 failed.start("run-1".to_owned());
9716 failed.fail("gate red", 5);
9717 queue.put(&mut failed).unwrap();
9718
9719 let mut held = task();
9720 held.id = "20260101-000003-cccc".to_owned();
9721 held.hold_machine(None);
9722 queue.put(&mut held).unwrap();
9723
9724 let mut running = task();
9725 running.id = "20260101-000004-dddd".to_owned();
9726 running.start("run-2".to_owned());
9727 queue.put(&mut running).unwrap();
9728
9729 let queued_ids: Vec<String> = queued_tasks(&queue).into_iter().map(|t| t.id).collect();
9730 assert_eq!(queued_ids, [queued.id.clone()]);
9731
9732 let mut finished_ids: Vec<String> =
9733 finished_tasks(&queue).into_iter().map(|t| t.id).collect();
9734 finished_ids.sort_unstable();
9735 let mut want = vec![failed.id.clone(), held.id.clone()];
9736 want.sort_unstable();
9737 assert_eq!(finished_ids, want);
9738 }
9739
9740 #[test]
9741 fn dependency_resolution_preserves_existing_short_and_full_running_chains() {
9742 for short in [false, true] {
9743 let dir = tempfile::tempdir().unwrap();
9744 let queue = Queue::at(dir.path().join("queue"));
9745 let questions = ask::Questions::at(dir.path().join("questions"));
9746 let mut running = task();
9747 running.id = "20261013-192213-feb2".into();
9748 running.start("existing-run".into());
9749 queue.put(&mut running).unwrap();
9750 let mut middle = task();
9751 middle.id = "20261013-193611-ca5b".into();
9752 middle.block(
9753 vec![if short {
9754 "feb2".into()
9755 } else {
9756 running.id.clone()
9757 }],
9758 None,
9759 );
9760 queue.put(&mut middle).unwrap();
9761 let mut waiting = task();
9762 waiting.id = "20261013-202849-ee37".into();
9763 waiting.block(
9764 vec![if short {
9765 "ca5b".into()
9766 } else {
9767 middle.id.clone()
9768 }],
9769 Some("wait for implementation".into()),
9770 );
9771 queue.put(&mut waiting).unwrap();
9772 assert_eq!(queue.get("ca5b").unwrap().id, middle.id);
9773 let inv = crate::blockers::Inventory::load(&queue, &questions);
9774 assert_eq!(inv.waits_on(&waiting), ["ca5b (blocked → feb2 running)"]);
9775 assert!(inv.stuck().is_empty());
9776 let report = crate::triage::run_once(&queue, &questions, None, Timestamp::now());
9777 assert!(report.quarantined.is_empty());
9778 assert!(report.asked.is_empty());
9779 resolve_blockers(&queue, &questions);
9780 assert_eq!(queue.get(&waiting.id).unwrap().status, TaskStatus::Blocked);
9781 assert!(questions.list().is_empty());
9782 running.succeed();
9785 queue.put(&mut running).unwrap();
9786 resolve_blockers(&queue, &questions);
9787 assert_eq!(queue.get(&middle.id).unwrap().status, TaskStatus::Queued);
9788 assert_eq!(queue.get(&waiting.id).unwrap().status, TaskStatus::Blocked);
9789 middle = queue.get(&middle.id).unwrap();
9790 middle.succeed();
9791 queue.put(&mut middle).unwrap();
9792 resolve_blockers(&queue, &questions);
9793 let after = queue.get(&waiting.id).unwrap();
9794 assert_eq!(after.status, TaskStatus::Queued);
9795 assert!(after.blocked_by.is_empty());
9796 assert_eq!(after.instruction, waiting.instruction);
9797 }
9798 }
9799
9800 #[test]
9801 fn dependency_resolution_unknown_records_never_hold_ask_or_release() {
9802 for future_schema in [false, true] {
9803 let name = "20261013-193611-ca5b";
9804 let mut future = task();
9805 future.id = name.into();
9806 future.schema = crate::queue::SCHEMA + 1;
9807 let contents = if future_schema {
9808 serde_json::to_string(&future).unwrap()
9809 } else {
9810 "corrupt".into()
9811 };
9812 let dir = tempfile::tempdir().unwrap();
9813 let queue = Queue::at(dir.path().join("queue"));
9814 let questions = ask::Questions::at(dir.path().join("questions"));
9815 let mut waiting = task();
9816 waiting.id = "20261013-202849-ee37".into();
9817 waiting.block(vec!["ca5b".into()], None);
9818 queue.put(&mut waiting).unwrap();
9819 std::fs::write(queue.path_of(name), contents).unwrap();
9820 assert_eq!(queue.resolve_id("ca5b").unwrap(), name);
9821 assert!(
9822 crate::queue::missing_blockers(&queue, &questions, &waiting.blocked_by).is_empty()
9823 );
9824 let report = crate::triage::run_once(&queue, &questions, None, Timestamp::now());
9825 assert!(report.asked.is_empty());
9826 resolve_blockers(&queue, &questions);
9827 assert_eq!(queue.get(&waiting.id).unwrap().status, TaskStatus::Blocked);
9828 assert!(questions.list().is_empty());
9829 }
9830 let dir = tempfile::tempdir().unwrap();
9831 let queue = Queue::at(dir.path().join("queue"));
9832 let questions = ask::Questions::at(dir.path().join("questions"));
9833 let mut waiting = task();
9834 waiting.id = "20261013-202849-ee37".into();
9835 waiting.block(vec!["ca5b".into()], None);
9836 queue.put(&mut waiting).unwrap();
9837 let mut done = task();
9838 done.id = "20261013-193611-ca5b".into();
9839 done.succeed();
9840 queue.put(&mut done).unwrap();
9841 std::fs::write(queue.path_of("20261014-193611-ca5b"), "bad").unwrap();
9844 assert!(queue.get("ca5b").is_err());
9845 assert_eq!(
9846 crate::queue::dependency(&queue, &questions, "ca5b"),
9847 crate::queue::Dependency::Ambiguous
9848 );
9849 resolve_blockers(&queue, &questions);
9850 let report = crate::triage::run_once(&queue, &questions, None, Timestamp::now());
9851 assert!(report.asked.is_empty());
9852 assert_eq!(queue.get(&waiting.id).unwrap().status, TaskStatus::Blocked);
9853 assert_eq!(queue.get(&done.id).unwrap().status, TaskStatus::Done);
9854 }
9855
9856 #[test]
9857 fn dependency_resolution_preserves_independent_consent_and_holds_after_completion() {
9858 for held in [false, true] {
9859 for question in [false, true] {
9860 for reference in ["ca5b", "20261013-193611-ca5b"] {
9861 let dir = tempfile::tempdir().unwrap();
9862 let queue = Queue::at(dir.path().join("queue"));
9863 let questions = ask::Questions::at(dir.path().join("questions"));
9864 let mut prerequisite = task();
9865 prerequisite.id = "20261013-193611-ca5b".into();
9866 prerequisite.start("existing-run".into());
9867 queue.put(&mut prerequisite).unwrap();
9868 let mut waiting = task();
9869 waiting.id = "20261013-202849-ee37".into();
9870 if held {
9871 waiting.hold_machine(Some("waiting for separate consent".into()));
9872 }
9873 queue.put(&mut waiting).unwrap();
9874 let reason = "Deleting saved data requires separate owner consent";
9875 crate::conduct::apply(
9876 &queue,
9877 &questions,
9878 &crate::conduct::Verdict {
9879 decisions: vec![crate::conduct::Decision {
9880 id: waiting.id.clone(),
9881 blocked_by: vec![reference.into()],
9882 question: question.then(|| "May the saved data be deleted?".into()),
9883 recovery: (!question).then_some(crate::conduct::Recovery::Hold),
9884 reason: Some(reason.into()),
9885 ..crate::conduct::Decision::default()
9886 }],
9887 },
9888 )
9889 .unwrap();
9890 let before = queue.get(&waiting.id).unwrap();
9891 if question {
9892 assert_eq!(before.status, TaskStatus::Blocked);
9893 let q = questions.get(&before.blocked_by[0]).unwrap();
9894 assert_eq!(q.summary, "May the saved data be deleted?");
9895 assert!(q.status.open());
9896 } else {
9897 assert_eq!(before.status, TaskStatus::Held);
9898 let recorded_reason = before.hold_reason.as_deref().unwrap();
9899 assert!(recorded_reason.starts_with(reason));
9900 if held {
9901 assert!(recorded_reason.contains("waiting for separate consent"));
9902 }
9903 }
9904 prerequisite.succeed();
9905 queue.put(&mut prerequisite).unwrap();
9906 resolve_blockers(&queue, &questions);
9907 let after = queue.get(&waiting.id).unwrap();
9908 assert_eq!(after.status, before.status);
9909 assert_eq!(after.blocked_by, before.blocked_by);
9910 assert_eq!(after.hold_reason, before.hold_reason);
9911 assert_eq!(after.instruction, waiting.instruction);
9912 if question {
9913 assert!(questions.get(&after.blocked_by[0]).unwrap().status.open());
9914 }
9915 assert!(
9916 queue.next_runnable().is_none(),
9917 "dependency completion is not consent"
9918 );
9919 }
9920 }
9921 }
9922 }
9923
9924 #[test]
9925 fn resolve_blockers_clears_a_done_dependency_and_keeps_an_unresolved_one() {
9926 let dir = tempfile::tempdir().unwrap();
9927 let queue = Queue::at(dir.path().join("queue"));
9928 let questions = ask::Questions::at(dir.path().join("questions"));
9929
9930 let mut dep = task();
9931 dep.id = "20260101-000001-dep0".to_owned();
9932 dep.succeed();
9933 queue.put(&mut dep).unwrap();
9934
9935 let mut still_going = task();
9936 still_going.id = "20260101-000002-dep1".to_owned();
9937 queue.put(&mut still_going).unwrap();
9938
9939 let mut blocked = task();
9940 blocked.id = "20260101-000003-main".to_owned();
9941 blocked.block(
9942 vec![dep.id.clone(), still_going.id.clone()],
9943 Some("waits on both".to_owned()),
9944 );
9945 queue.put(&mut blocked).unwrap();
9946
9947 resolve_blockers(&queue, &questions);
9948
9949 let after = queue.get(&blocked.id).unwrap();
9950 assert_eq!(
9951 after.status,
9952 TaskStatus::Blocked,
9953 "one dependency is still outstanding"
9954 );
9955 assert_eq!(after.blocked_by, [still_going.id.clone()]);
9956 }
9957
9958 #[test]
9959 fn resolve_blockers_carries_an_answers_content_onto_the_task_and_unblocks_it() {
9960 let dir = tempfile::tempdir().unwrap();
9961 let queue = Queue::at(dir.path().join("queue"));
9962 let questions = ask::Questions::at(dir.path().join("questions"));
9963
9964 let mut q = crate::ask::Question::new(
9965 "20260101-000001-main".to_owned(),
9966 crate::conduct::NODE.to_owned(),
9967 "conduct".to_owned(),
9968 "Which backend?".to_owned(),
9969 String::new(),
9970 Vec::new(),
9971 );
9972 questions.put(&mut q).unwrap();
9973 q.answer(crate::ask::Answer::Text("SQLite".to_owned()))
9974 .unwrap();
9975 questions.put(&mut q).unwrap();
9976
9977 let mut blocked = task();
9978 blocked.id = "20260101-000001-main".to_owned();
9979 blocked.block(vec![q.id.clone()], Some("which backend?".to_owned()));
9980 queue.put(&mut blocked).unwrap();
9981
9982 resolve_blockers(&queue, &questions);
9983
9984 let after = queue.get(&blocked.id).unwrap();
9985 assert_eq!(
9986 after.status,
9987 TaskStatus::Queued,
9988 "the only blocker resolved"
9989 );
9990 assert_eq!(after.answers.len(), 1);
9991 assert_eq!(after.answers[0].question, "Which backend?");
9992 assert_eq!(after.answers[0].answer, "SQLite");
9993
9994 let instruction = instruction_for(&after);
9996 assert!(instruction.contains("Which backend?"));
9997 assert!(instruction.contains("SQLite"));
9998 }
9999
10000 #[test]
10001 fn resolve_blockers_holds_a_task_whose_conductor_question_was_abandoned() {
10002 let dir = tempfile::tempdir().unwrap();
10003 let queue = Queue::at(dir.path().join("queue"));
10004 let questions = ask::Questions::at(dir.path().join("questions"));
10005
10006 let mut q = crate::ask::Question::new(
10007 "20260101-000001-main".to_owned(),
10008 crate::conduct::NODE.to_owned(),
10009 "conduct".to_owned(),
10010 "Is the setup done?".to_owned(),
10011 String::new(),
10012 Vec::new(),
10013 );
10014 q.abandon("no answer within 60s of asking");
10015 questions.put(&mut q).unwrap();
10016
10017 let mut blocked = task();
10018 blocked.id = "20260101-000001-main".to_owned();
10019 blocked.block(vec![q.id.clone()], Some("setup?".to_owned()));
10020 queue.put(&mut blocked).unwrap();
10021
10022 resolve_blockers(&queue, &questions);
10023
10024 let after = queue.get(&blocked.id).unwrap();
10025 assert_eq!(
10026 after.status,
10027 TaskStatus::Held,
10028 "never left blocked on nothing"
10029 );
10030 assert!(!after.operator_held(), "a machine hold, for triage");
10031 assert!(
10032 after
10033 .hold_reason
10034 .as_deref()
10035 .unwrap_or_default()
10036 .contains("went unanswered")
10037 );
10038 }
10039
10040 #[test]
10041 fn resolve_blockers_restores_a_held_task_to_held_instead_of_queuing_it() {
10042 let dir = tempfile::tempdir().unwrap();
10048 let queue = Queue::at(dir.path().join("queue"));
10049 let questions = ask::Questions::at(dir.path().join("questions"));
10050
10051 let mut q = crate::ask::Question::new(
10052 "20260101-000001-main".to_owned(),
10053 crate::conduct::NODE.to_owned(),
10054 "conduct".to_owned(),
10055 "How should this be handled?".to_owned(),
10056 String::new(),
10057 Vec::new(),
10058 );
10059 questions.put(&mut q).unwrap();
10060 q.answer(crate::ask::Answer::Text(
10061 "leave it held, a human will look at it later".to_owned(),
10062 ))
10063 .unwrap();
10064 questions.put(&mut q).unwrap();
10065
10066 let mut held = task();
10067 held.id = "20260101-000001-main".to_owned();
10068 held.hold_machine(Some("out of attempts".to_owned()));
10069 held.block(vec![q.id.clone()], Some("what now?".to_owned()));
10070 queue.put(&mut held).unwrap();
10071
10072 resolve_blockers(&queue, &questions);
10073
10074 let after = queue.get(&held.id).unwrap();
10075 assert_eq!(after.status, TaskStatus::Held);
10076 assert_eq!(after.hold_reason.as_deref(), Some("out of attempts"));
10077 assert_eq!(
10078 after.answers[0].answer,
10079 "leave it held, a human will look at it later"
10080 );
10081 }
10082
10083 #[test]
10084 fn resolve_blockers_releases_a_task_whose_dependency_was_deleted_on_purpose() {
10085 let dir = tempfile::tempdir().unwrap();
10086 let queue = Queue::at(dir.path().join("queue"));
10087 let questions = ask::Questions::at(dir.path().join("questions"));
10088 let mut dep = task();
10089 dep.id = "20260101-000001-gone".to_owned();
10090 queue.put(&mut dep).unwrap();
10091 let mut blocked = task();
10092 blocked.id = "20260101-000003-main".to_owned();
10093 blocked.block(vec![dep.id.clone()], None);
10094 queue.put(&mut blocked).unwrap();
10095
10096 let claim = queue.claim(&blocked.id).unwrap();
10098 queue.remove(&dep.id, false, &questions).unwrap();
10099 drop(claim);
10100 resolve_blockers(&queue, &questions);
10101
10102 let after = queue.get(&blocked.id).unwrap();
10103 assert_eq!(after.status, TaskStatus::Queued);
10104 assert!(after.blocked_by.is_empty());
10105 }
10106
10107 #[test]
10108 fn resolve_blockers_holds_a_task_whose_dependency_was_deleted() {
10109 let dir = tempfile::tempdir().unwrap();
10115 let queue = Queue::at(dir.path().join("queue"));
10116 let questions = ask::Questions::at(dir.path().join("questions"));
10117
10118 let mut still_going = task();
10119 still_going.id = "20260101-000002-dep1".to_owned();
10120 queue.put(&mut still_going).unwrap();
10121
10122 let mut blocked = task();
10123 blocked.id = "20260101-000003-main".to_owned();
10124 blocked.block(
10125 vec!["20260101-000001-gone".to_owned(), still_going.id.clone()],
10126 Some("waits on both".to_owned()),
10127 );
10128 queue.put(&mut blocked).unwrap();
10129
10130 resolve_blockers(&queue, &questions);
10131
10132 let after = queue.get(&blocked.id).unwrap();
10133 assert_eq!(
10134 after.status,
10135 TaskStatus::Held,
10136 "a missing dependency must not leave the task blocked forever"
10137 );
10138 assert_eq!(after.hold_source, Some(crate::queue::HoldSource::Machine));
10139 assert!(after.blocked_by.is_empty());
10140 let reason = after.hold_reason.as_deref().unwrap_or_default();
10141 assert!(
10142 reason.contains("20260101-000001-gone"),
10143 "the missing id must be named so an operator can tell what happened: {reason}"
10144 );
10145 assert!(
10146 reason.contains(&still_going.id),
10147 "the still-valid dependency must not silently vanish from the record: {reason}"
10148 );
10149 }
10150
10151 #[test]
10152 fn instruction_for_is_unchanged_without_any_answers() {
10153 let t = task();
10154 assert_eq!(instruction_for(&t), t.instruction);
10155 }
10156
10157 #[test]
10158 fn task_attachments_are_absolute_and_a_missing_file_is_an_error() {
10159 let dir = tempfile::tempdir().unwrap();
10160 let q = Queue::at(dir.path().join("queue"));
10161 let src = dir.path().join("shot.png");
10162 std::fs::write(&src, "x").unwrap();
10163 let mut t = task();
10164 q.attach(&mut t, &[src]).unwrap();
10165 let paths = task_attachments(&q, &t).unwrap();
10166 assert_eq!(paths.len(), 1);
10167 assert!(paths[0].is_absolute() && paths[0].is_file());
10168 std::fs::remove_file(&paths[0]).unwrap();
10169 let err = task_attachments(&q, &t).unwrap_err().to_string();
10170 assert!(err.contains("shot.png"), "{err}");
10171 }
10172
10173 #[tokio::test]
10174 async fn config_failure_notices_distinguish_boundaries_and_targets_after_restart() {
10175 let dir = tempfile::tempdir().unwrap();
10176 let queue_path = dir.path().join("queue");
10177 let queue = Queue::at(queue_path.clone());
10178 let mut t = task();
10179 t.source = Source::Agent {
10180 run: "originating-chat".into(),
10181 node: crate::queue::CHAT_NODE.into(),
10182 };
10183 let fetch = anyhow::anyhow!("network unavailable").context(crate::config::ConfigFailure {
10184 stage: "config-fetch",
10185 target: "origin/main".into(),
10186 });
10187 record_config_failure(&mut t, &fetch);
10188 t.hold_machine(Some(format!("{fetch:#}")));
10189 let fetch_key = t.chat_notice_due().unwrap();
10190 t.chat_notices.push(fetch_key.clone());
10191 queue.put(&mut t).unwrap();
10192
10193 let queue = Queue::at(queue_path);
10194 let mut t = queue.get(&t.id).unwrap();
10195 let a = dir.path().join("a.toml");
10196 let b = dir.path().join("b.toml");
10197 for (path, text, new_cause) in [
10198 (&a, "[graph", true),
10199 (&a, "[verify", false),
10200 (&a, "[graph]\nimplementers = 'invalid-type'", true),
10201 (&b, "[graph]\nimplementers = 'invalid-type'", true),
10202 ] {
10203 t.release_operator("test", None);
10204 std::fs::write(path, text).unwrap();
10205 let error = Config::discover_fetched(dir.path(), Some(path))
10206 .await
10207 .unwrap_err();
10208 record_config_failure(&mut t, &error);
10209 t.hold_machine(Some(format!("{error:#}")));
10210 let due = t.chat_notice_due();
10211 assert_eq!(due.is_some(), new_cause, "{error:#}");
10212 if let Some(key) = due {
10213 assert_ne!(key, fetch_key);
10214 t.chat_notices.push(key);
10215 }
10216 }
10217 t.succeed();
10218 assert_eq!(t.chat_notice_due().as_deref(), Some("done"));
10219 }
10220
10221 #[test]
10222 fn different_missing_attachments_have_distinct_persisted_notice_causes() {
10223 let dir = tempfile::tempdir().unwrap();
10224 let queue_path = dir.path().join("queue");
10225 let queue = Queue::at(queue_path.clone());
10226 let mut t = task();
10227 t.source = crate::queue::Source::Agent {
10228 run: "originating-chat".into(),
10229 node: crate::queue::CHAT_NODE.into(),
10230 };
10231 let a = dir.path().join("a.png");
10232 let b = dir.path().join("b.png");
10233 std::fs::write(&a, "a").unwrap();
10234 std::fs::write(&b, "b").unwrap();
10235 queue.attach(&mut t, &[a, b]).unwrap();
10236 let paths = task_attachments(&queue, &t).unwrap();
10237 for path in &paths {
10238 std::fs::remove_file(path).unwrap();
10239 }
10240 let missing = task_attachments(&queue, &t).unwrap_err();
10241 t.attempts += 1;
10242 let reason = missing.to_string();
10243 t.fail_startup_target("startup-attachments", missing.name, reason, 1);
10244 let first = t.chat_notice_due().expect("first attachment notice");
10245 t.chat_notices.push(first.clone());
10246 queue.put(&mut t).unwrap();
10247
10248 let queue = Queue::at(queue_path.clone());
10249 let mut t = queue.get(&t.id).unwrap();
10250 assert!(t.chat_notice_due().is_none(), "restart does not replay");
10251 t.release_operator("test", None);
10252 t.attempts += 1;
10253 let missing = task_attachments(&queue, &t).unwrap_err();
10254 t.fail_startup_target(
10255 "startup-attachments",
10256 missing.name,
10257 "same missing attachment with regenerated wording",
10258 1,
10259 );
10260 assert!(t.chat_notice_due().is_none(), "same object stays quiet");
10261
10262 std::fs::write(&paths[0], "restored").unwrap();
10263 t.release_operator("test", None);
10264 t.attempts += 1;
10265 let missing = task_attachments(&queue, &t).unwrap_err();
10266 let reason = missing.to_string();
10267 t.fail_startup_target("startup-attachments", missing.name, reason, 1);
10268 let second = t
10269 .chat_notice_due()
10270 .expect("different attachment must notify");
10271 assert_ne!(first, second);
10272 assert!(t.hold_reason.as_ref().unwrap().contains("b.png"));
10273 }
10274
10275 #[test]
10276 fn resumed_instruction_is_unchanged_without_any_answers() {
10277 let t = task();
10278 assert_eq!(resumed_instruction(&t.instruction, &t), t.instruction);
10279 }
10280
10281 #[test]
10282 fn resumed_instruction_carries_a_new_answer_onto_the_old_run() {
10283 let mut t = task();
10284 t.record_answer("Which backend?".to_owned(), "SQLite".to_owned());
10285 let old = t.instruction.clone();
10289
10290 let refreshed = resumed_instruction(&old, &t);
10291 assert!(refreshed.starts_with(&old), "the original text is kept");
10292 assert!(refreshed.contains("Which backend?"));
10293 assert!(refreshed.contains("SQLite"));
10294 }
10295
10296 #[test]
10297 fn resumed_instruction_keeps_an_original_answers_heading() {
10298 let mut t = task();
10299 t.instruction = "Context\n\n# Operator answers\n\nThis is part of the task.".to_owned();
10300 t.record_answer("Which backend?".to_owned(), "SQLite".to_owned());
10301
10302 let refreshed = resumed_instruction(&t.instruction, &t);
10303
10304 assert!(
10305 refreshed.starts_with(&t.instruction),
10306 "an answers heading in the original instruction is not the appended block"
10307 );
10308 assert_eq!(refreshed.matches(ANSWERS_HEADER).count(), 2);
10309 assert!(refreshed.contains("Which backend?"));
10310 assert!(refreshed.contains("SQLite"));
10311
10312 let repeated = resumed_instruction(&refreshed, &t);
10313 assert_eq!(
10314 repeated, refreshed,
10315 "only the final appended block is refreshed"
10316 );
10317 }
10318
10319 #[test]
10320 fn resumed_instruction_does_not_duplicate_across_repeated_resumes() {
10321 let mut t = task();
10322 t.record_answer("Which backend?".to_owned(), "SQLite".to_owned());
10323
10324 let once = resumed_instruction(&t.instruction, &t);
10328 let twice = resumed_instruction(&once, &t);
10329 assert_eq!(once, twice);
10330 assert_eq!(once.matches("Which backend?").count(), 1);
10331
10332 t.record_answer("Which cache?".to_owned(), "Redis".to_owned());
10334 let refreshed = resumed_instruction(&once, &t);
10335 assert_eq!(refreshed.matches(ANSWERS_HEADER).count(), 1);
10336 assert!(refreshed.contains("Which backend?"));
10337 assert!(refreshed.contains("Which cache?"));
10338 }
10339
10340 #[test]
10341 fn prepare_instruction_covers_all_three_starters() {
10342 let mut t = task();
10343 t.record_answer("Which backend?".to_owned(), "SQLite".to_owned());
10344
10345 assert_eq!(
10348 prepare_instruction(&Starter::Start, None, &t),
10349 Some(instruction_for(&t))
10350 );
10351
10352 let old = t.instruction.clone();
10355 assert_eq!(
10356 prepare_instruction(&Starter::Resume("some-run".to_owned()), Some(&old), &t),
10357 Some(resumed_instruction(&old, &t))
10358 );
10359
10360 assert_eq!(
10364 prepare_instruction(&Starter::Review("magi/eba2/A".to_owned()), Some(&old), &t),
10365 None
10366 );
10367 }
10368
10369 #[test]
10370 fn choose_starter_prefers_review_over_resume_when_the_branch_survived() {
10371 assert_eq!(
10372 choose_starter(Some("magi/eba2/A"), true, Some("some-run")),
10373 Starter::Review("magi/eba2/A".to_owned())
10374 );
10375 }
10376
10377 #[test]
10378 fn choose_starter_falls_back_to_start_when_the_review_branch_is_gone() {
10379 assert_eq!(
10380 choose_starter(Some("magi/eba2/A"), false, Some("some-run")),
10381 Starter::Start,
10382 "a vanished review branch must not fall back to resuming the old run either"
10383 );
10384 }
10385
10386 #[test]
10387 fn a_refused_handover_retries_as_a_review_of_the_same_branch() {
10388 let mut t = task();
10389 t.start("old-run".to_owned());
10390 t.hold_for_handover(Some("magi/eba2/A".to_owned()), "checked out".to_owned());
10391 t.release();
10392 let branch = t.review_branch.take();
10393 assert_eq!(
10394 choose_starter(branch.as_deref(), true, Some("old-run")),
10395 Starter::Review("magi/eba2/A".to_owned()),
10396 "a review wins over resuming the old run"
10397 );
10398 }
10399
10400 #[test]
10401 fn choose_starter_resumes_or_starts_when_there_is_no_review_choice_at_all() {
10402 assert_eq!(
10403 choose_starter(None, false, Some("some-run")),
10404 Starter::Resume("some-run".to_owned())
10405 );
10406 assert_eq!(choose_starter(None, false, None), Starter::Start);
10407 }
10408
10409 #[test]
10410 fn an_explicit_release_forces_a_fresh_competition_even_with_a_resumable_run() {
10411 let mut released = task();
10412 released.start("stalled-run".to_owned());
10413 released.requeue();
10414 let unfinished = (!released.fresh_start)
10415 .then(|| Some("stalled-run".to_owned()))
10416 .flatten();
10417 assert_eq!(
10418 choose_starter(None, false, unfinished.as_deref()),
10419 Starter::Start,
10420 "release keeps run history but must not resume it"
10421 );
10422 assert_eq!(released.runs, ["stalled-run"]);
10423 }
10424
10425 fn stopped_public_run(status: RunStatus) -> RunState {
10426 let mut state = run_state(status);
10427 state.driver_exited = true;
10428 state.config.graph.land = true;
10429 state.config.graph.review_rounds = 1;
10430 state.reviews = vec![review_round(1)];
10431 state.merge = Some(crate::run::MergeOutcome {
10432 mode: MergeMode::Pr,
10433 ok: false,
10434 detail: "landing stopped".into(),
10435 empty: false,
10436 });
10437 state.pr = Some(crate::run::PrRecord {
10438 url: "https://example.test/pull/1".into(),
10439 number: 1,
10440 state: "open".into(),
10441 checks: "red".into(),
10442 round: 1,
10443 rounds: 1,
10444 red_at_merge: Vec::new(),
10445 });
10446 state
10447 }
10448
10449 #[test]
10450 fn stopped_public_owner_resumes_on_release_choice_and_control_after_review_exhaustion() {
10451 for status in [RunStatus::Blocked, RunStatus::Gating] {
10452 let home = tempfile::tempdir().unwrap();
10453 let mut state = stopped_public_run(status);
10454 state.save_under(home.path()).unwrap();
10455 assert!(crate::handover::has_resume_checkpoint(&state, home.path()));
10456 assert!(!exhausted_review_budget(&state));
10457 let mut task = held_task_with(&state.id);
10458 task.release();
10459 let unfinished = unfinished_run_with(&task.runs, task.short(), |_| Ok(state.clone()));
10460 assert_eq!(
10461 choose_starter(None, false, unfinished.as_deref()),
10462 Starter::Resume(state.id.clone())
10463 );
10464 let question = action_question(&state.id, resume_action(&state.id));
10465 assert_eq!(
10466 decide_action(&task, &question, &PHRASES_EN, |_| Ok(state.clone())),
10467 ActionDecision::Resume(state.id.clone())
10468 );
10469 assert_eq!(
10471 crate::resume::validate(home.path(), &state.id).unwrap().id,
10472 state.id
10473 );
10474 let queue = Queue::at(home.path().join("queue"));
10475 let mut controlled = Task::new(
10476 "resume land".into(),
10477 state.instruction.clone(),
10478 state.repo.clone(),
10479 Source::Agent {
10480 run: "talk".into(),
10481 node: crate::queue::CHAT_NODE.into(),
10482 },
10483 );
10484 controlled.runs.push(state.id.clone());
10485 controlled.hold_machine(Some("land stopped".into()));
10486 queue.put(&mut controlled).unwrap();
10487 let resumed =
10488 crate::chat_control::resume(&queue, home.path(), "talk", &state.id).unwrap();
10489 assert_eq!(resumed.status, TaskStatus::Queued);
10490 assert_eq!(
10491 resumed.resume_override.unwrap().pinned_run,
10492 Some(state.id.clone())
10493 );
10494 assert_eq!(resumed.attempts, 0);
10495 }
10496 }
10497
10498 #[test]
10499 fn exhausted_gate_failure_with_clean_or_disabled_reviews_is_not_resumable() {
10500 for review_rounds in [0, 1] {
10501 let home = tempfile::tempdir().unwrap();
10502 let mut state = run_state(RunStatus::Blocked);
10503 state.config.graph.review_rounds = review_rounds;
10504 if review_rounds > 0 {
10505 let mut review = review_round(1);
10506 review.clean = true;
10507 state.reviews = vec![review];
10508 }
10509 state.gate_ran = true;
10510 state.gate = vec![crate::run::CommandOutcome {
10511 command: "verify".into(),
10512 code: Some(1),
10513 output_tail: "gate failed after repair budget exhausted".into(),
10514 duration_ms: 0,
10515 resource_blocked: false,
10516 }];
10517 state.save_under(home.path()).unwrap();
10518 assert!(!crate::handover::has_resume_checkpoint(&state, home.path()));
10519 assert!(exhausted_review_budget(&state));
10520 let mut task = held_task_with(&state.id);
10521 task.release();
10522 assert!(unfinished_run_with(&task.runs, task.short(), |_| Ok(state.clone())).is_none());
10523 let question = action_question(&state.id, resume_action(&state.id));
10524 assert_ne!(
10525 decide_action(&task, &question, &PHRASES_EN, |_| Ok(state.clone())),
10526 ActionDecision::Resume(state.id.clone())
10527 );
10528 assert!(crate::resume::validate(home.path(), &state.id).is_err());
10529
10530 state.gate_ran = false;
10532 state.gate.clear();
10533 assert!(!exhausted_review_budget(&state));
10534 assert_eq!(
10535 unfinished_run_with(&task.runs, task.short(), |_| Ok(state.clone())),
10536 Some(state.id)
10537 );
10538 }
10539 }
10540
10541 #[test]
10542 fn resource_blocked_verification_resumes_the_same_run() {
10543 for (kind, stage) in [
10544 ("cache", "clean"),
10545 ("coordination", "clean"),
10546 ("cache", "review-e2e"),
10547 ("coordination", "no-reviews"),
10548 ("coordination", "handoff"),
10549 ] {
10550 let home = tempfile::tempdir().unwrap();
10551 let mut state = run_state(RunStatus::Blocked);
10552 state.config.graph.review_rounds = 1;
10553 let mut review = review_round(1);
10554 review.clean = true;
10555 state.reviews = vec![review];
10556 if stage == "review-e2e" {
10557 let review = &mut state.reviews[0];
10558 review.clean = false;
10559 review.blocking = 1;
10560 review.e2e = vec![crate::run::CommandOutcome {
10561 command: "verify".into(),
10562 code: None,
10563 output_tail: "resource wait expired".into(),
10564 duration_ms: 0,
10565 resource_blocked: true,
10566 }];
10567 } else if stage == "no-reviews" {
10568 state.config.graph.review_rounds = 0;
10569 state.reviews.clear();
10570 } else if stage == "handoff" {
10571 let review = &mut state.reviews[0];
10574 review.clean = false;
10575 review.blocking = 1;
10576 review.e2e = vec![crate::run::CommandOutcome {
10577 command: "verify".into(),
10578 code: Some(0),
10579 output_tail: String::new(),
10580 duration_ms: 0,
10581 resource_blocked: false,
10582 }];
10583 }
10584 state.gate_ran = false;
10585 state.resource_block = Some(crate::run::ResourceBlock {
10586 node: if stage == "review-e2e" {
10587 "review"
10588 } else {
10589 "gate"
10590 }
10591 .into(),
10592 kind: kind.into(),
10593 detail: "resource wait expired".into(),
10594 at: jiff::Timestamp::now(),
10595 });
10596 state.save_under(home.path()).unwrap();
10597 assert!(!crate::handover::has_resume_checkpoint(&state, home.path()));
10598 assert!(!exhausted_review_budget(&state));
10599 let mut task = held_task_with(&state.id);
10600 task.release();
10601 let unfinished = unfinished_run_with(&task.runs, task.short(), |_| Ok(state.clone()));
10602 assert_eq!(
10603 choose_starter(None, false, unfinished.as_deref()),
10604 Starter::Resume(state.id.clone())
10605 );
10606 let question = action_question(&state.id, resume_action(&state.id));
10607 assert_eq!(
10608 decide_action(&task, &question, &PHRASES_EN, |_| Ok(state.clone())),
10609 ActionDecision::Resume(state.id.clone())
10610 );
10611 assert_eq!(
10612 crate::resume::validate(home.path(), &state.id).unwrap().id,
10613 state.id
10614 );
10615 let queue = Queue::at(home.path().join("queue"));
10616 let mut controlled = Task::new(
10617 "resume verification".into(),
10618 state.instruction.clone(),
10619 state.repo.clone(),
10620 Source::Agent {
10621 run: "talk".into(),
10622 node: crate::queue::CHAT_NODE.into(),
10623 },
10624 );
10625 controlled.runs.push(state.id.clone());
10626 controlled.hold_machine(Some("verification resource unavailable".into()));
10627 queue.put(&mut controlled).unwrap();
10628 let resumed =
10629 crate::chat_control::resume(&queue, home.path(), "talk", &state.id).unwrap();
10630 assert_eq!(resumed.status, TaskStatus::Queued);
10631 assert_eq!(
10632 resumed.resume_override.unwrap().pinned_run,
10633 Some(state.id.clone())
10634 );
10635 assert_eq!(resumed.attempts, 0);
10636
10637 state.config.graph.review_rounds = 1;
10639 let mut unresolved = review_round(1);
10640 unresolved.blocking = 1;
10641 state.reviews = vec![unresolved];
10642 state.resource_block = None;
10643 state.save_under(home.path()).unwrap();
10644 assert!(exhausted_review_budget(&state));
10645 assert!(crate::resume::validate(home.path(), &state.id).is_err());
10646 }
10647 }
10648
10649 #[tokio::test]
10650 async fn retained_public_owner_wins_over_an_existing_successor_and_unknown_history_holds() {
10651 let home = tempfile::tempdir().unwrap();
10652 let mut owner = stopped_public_run(RunStatus::Blocked);
10653 let repo = tempfile::tempdir().unwrap();
10654 crate::git::git(repo.path(), &["init"]).await.unwrap();
10655 owner.repo = crate::git::toplevel(repo.path()).await.unwrap();
10656 owner.id = "20260101-000000-aaaa".into();
10657 owner.save_under(home.path()).unwrap();
10658 let mut successor = run_state(RunStatus::Blocked);
10659 successor.id = "20260101-000001-bbbb".into();
10660 successor.save_under(home.path()).unwrap();
10661 let runs = vec![owner.id.clone(), successor.id.clone()];
10662 assert_eq!(
10663 public_resume_owner(home.path(), &runs, &owner.repo)
10664 .await
10665 .unwrap()
10666 .unwrap()
10667 .id,
10668 owner.id
10669 );
10670 let subdir = owner.repo.join("nested");
10671 std::fs::create_dir(&subdir).unwrap();
10672 assert_eq!(
10673 public_resume_owner(home.path(), &runs, &subdir)
10674 .await
10675 .unwrap()
10676 .unwrap()
10677 .id,
10678 owner.id
10679 );
10680 #[cfg(unix)]
10681 {
10682 let alias = home.path().join("repo-alias");
10683 std::os::unix::fs::symlink(&owner.repo, &alias).unwrap();
10684 assert_eq!(
10685 public_resume_owner(home.path(), &runs, &alias.join("nested"))
10686 .await
10687 .unwrap()
10688 .unwrap()
10689 .id,
10690 owner.id
10691 );
10692 }
10693 let different = tempfile::tempdir().unwrap();
10694 crate::git::git(different.path(), &["init"]).await.unwrap();
10695 assert!(
10696 public_resume_owner(home.path(), &runs, different.path())
10697 .await
10698 .is_err()
10699 );
10700 let lock = crate::pr_head::owner_lock(home.path(), &owner.id).unwrap();
10701 assert!(
10702 public_resume_owner(home.path(), &runs, &owner.repo)
10703 .await
10704 .is_err()
10705 );
10706 drop(lock);
10707 std::fs::remove_file(
10708 home.path()
10709 .join("runs")
10710 .join(&successor.id)
10711 .join("run.json"),
10712 )
10713 .unwrap();
10714 assert!(
10715 public_resume_owner(home.path(), &runs, &owner.repo)
10716 .await
10717 .is_err()
10718 );
10719 }
10720
10721 #[test]
10722 fn an_ordinary_release_keeps_a_resumable_run_available() {
10723 let mut released = task();
10724 released.start("stalled-run".to_owned());
10725 released.release();
10726 let unfinished = (!released.fresh_start)
10727 .then(|| Some("stalled-run".to_owned()))
10728 .flatten();
10729 assert_eq!(
10730 choose_starter(None, false, unfinished.as_deref()),
10731 Starter::Resume("stalled-run".to_owned()),
10732 "manual release must preserve the normal resume path"
10733 );
10734 }
10735
10736 #[test]
10737 fn a_blocked_run_that_spent_every_review_round_has_exhausted_its_budget() {
10738 let mut state = run_state(RunStatus::Blocked);
10739 state.config.graph.review_rounds = 3;
10740 state.reviews = vec![review_round(1), review_round(2), review_round(3)];
10741 assert!(exhausted_review_budget(&state));
10742
10743 state.reviews.pop();
10745 assert!(!exhausted_review_budget(&state));
10746
10747 let mut stalled = run_state(RunStatus::Stalled);
10750 stalled.config.graph.review_rounds = 1;
10751 stalled.reviews = vec![review_round(1)];
10752 assert!(!exhausted_review_budget(&stalled));
10753 }
10754
10755 fn review_round(round: usize) -> crate::run::ReviewRound {
10756 crate::run::ReviewRound {
10757 round,
10758 head: "deadbeef".to_owned(),
10759 verified_head: None,
10760 verified_at: None,
10761 reviews: Vec::new(),
10762 e2e: Vec::new(),
10763 verify_retried: false,
10764 e2e_deferred: false,
10765 e2e_defer_reason: None,
10766 fix: None,
10767 blocking: 0,
10768 answered: 1,
10769 expected: 1,
10770 clean: false,
10771 progressed: true,
10772 vote_split: false,
10773 reconsideration: Vec::new(),
10774 verdict: None,
10775 }
10776 }
10777
10778 #[test]
10779 fn awaiting_resume_is_a_failed_task_whose_last_run_parked_and_can_be_resumed() {
10780 let mut run = RunState::new(
10781 PathBuf::from("/repo"),
10782 "main".to_owned(),
10783 "abc1234def".to_owned(),
10784 "add retries".to_owned(),
10785 Config::default(),
10786 );
10787 run.status = RunStatus::Judging;
10788 run.parked = true;
10789 let mut task = Task::new(
10790 "add retries".to_owned(),
10791 "add retries".to_owned(),
10792 PathBuf::from("/repo"),
10793 crate::queue::Source::Human,
10794 );
10795 task.status = TaskStatus::Failed;
10796 task.runs = vec![run.id.clone()];
10797 let with = |t: &Task, r: &RunState| awaiting_resume_with(t, |_| Ok(r.clone()));
10798 assert!(with(&task, &run), "parked after judging is the case");
10799 let mut parked_task = task.clone();
10800 parked_task.status = TaskStatus::Parked;
10801 assert!(with(&parked_task, &run), "the Parked status is the case");
10802
10803 let mut not_parked = run.clone();
10804 not_parked.parked = false;
10805 not_parked.status = RunStatus::Stalled;
10806 assert!(!with(&task, ¬_parked), "a stall is the conductor's");
10807
10808 let mut fresh = task.clone();
10809 fresh.fresh_start = true;
10810 assert!(!with(&fresh, &run), "a requeue asked for a new competition");
10811
10812 let mut review = task.clone();
10813 review.review_branch = Some("magi/x/A".to_owned());
10814 assert!(!with(&review, &run), "review is ranked before resume");
10815
10816 let mut held = task.clone();
10817 held.status = TaskStatus::Held;
10818 assert!(!with(&held, &run), "a hold stays visible to the conductor");
10819
10820 let mut released = run.clone();
10821 released.released_to = Some("20260901-000000-new1".to_owned());
10822 assert!(!with(&task, &released), "nothing left to resume into");
10823
10824 assert!(!awaiting_resume_with(&task, |_| anyhow::bail!(
10825 "unreadable"
10826 )));
10827 }
10828
10829 #[test]
10830 fn unfinished_run_never_offers_a_run_whose_worktree_was_released() {
10831 let mut released = RunState::new(
10832 PathBuf::from("/repo"),
10833 "main".to_owned(),
10834 "abc1234def".to_owned(),
10835 "add retries".to_owned(),
10836 Config::default(),
10837 );
10838 released.status = RunStatus::Blocked;
10839 assert_eq!(
10840 unfinished_run_with(&[released.id.clone()], "t", |_| Ok(released.clone())),
10841 Some(released.id.clone())
10842 );
10843 released.released_to = Some("20260901-000000-new1".to_owned());
10844 assert_eq!(
10845 unfinished_run_with(&[released.id.clone()], "t", |_| Ok(released.clone())),
10846 None,
10847 "there is nothing left to resume it into"
10848 );
10849 }
10850
10851 #[test]
10852 fn unfinished_run_skips_a_round_exhausted_blocked_run_so_requeue_means_a_fresh_competition() {
10853 let mut exhausted = RunState::new(
10862 PathBuf::from("/repo"),
10863 "main".to_owned(),
10864 "abc1234def".to_owned(),
10865 "add retries".to_owned(),
10866 Config::default(),
10867 );
10868 exhausted.status = RunStatus::Blocked;
10869 exhausted.config.graph.review_rounds = 1;
10870 exhausted.reviews = vec![review_round(1)];
10871
10872 assert_eq!(
10873 unfinished_run_with(&[exhausted.id.clone()], "t", |_| Ok(exhausted.clone())),
10874 None,
10875 "an exhausted `Blocked` run must not be offered as resumable"
10876 );
10877
10878 let mut has_budget_left = RunState::new(
10881 PathBuf::from("/repo"),
10882 "main".to_owned(),
10883 "abc1234def".to_owned(),
10884 "add retries".to_owned(),
10885 Config::default(),
10886 );
10887 has_budget_left.status = RunStatus::Blocked;
10888 has_budget_left.config.graph.review_rounds = 3;
10889 has_budget_left.reviews = vec![review_round(1)];
10890
10891 assert_eq!(
10892 unfinished_run_with(&[has_budget_left.id.clone()], "t", |_| {
10893 Ok(has_budget_left.clone())
10894 }),
10895 Some(has_budget_left.id.clone())
10896 );
10897 }
10898
10899 #[test]
10900 fn unfinished_run_resumes_a_run_whose_task_head_publication_is_pending() {
10901 let mut run = RunState::new(
10902 PathBuf::from("/repo"),
10903 "main".to_owned(),
10904 "abc1234def".to_owned(),
10905 "add retries".to_owned(),
10906 Config::default(),
10907 );
10908 run.status = RunStatus::Blocked;
10909 run.config.graph.review_rounds = 1;
10910 run.reviews = vec![review_round(1)];
10911 run.pr_publication_pending = true;
10912 assert_eq!(
10913 unfinished_run_with(&[run.id.clone()], "t", |_| Ok(run.clone())),
10914 Some(run.id.clone()),
10915 "a spent review budget must not turn a failed publication into a re-competition"
10916 );
10917 }
10918
10919 #[test]
10920 fn retained_attempt_bound_survives_changed_bases_releases_and_duplicate_history() {
10921 let mut task = Task::new(
10922 "recover work".into(),
10923 "recover work".into(),
10924 PathBuf::from("/repo"),
10925 crate::queue::Source::Human,
10926 );
10927 let mut history = std::collections::BTreeMap::new();
10928 for index in 0..8 {
10929 let mut run = run_state(RunStatus::Blocked);
10930 run.id = format!("20260101-000000-{index:04}");
10931 run.base_commit = format!("upstream-{index}");
10932 if index > 0 {
10934 run.candidates.clear();
10935 }
10936 task.runs.extend([run.id.clone(), run.id.clone()]);
10937 history.insert(run.id.clone(), run);
10938 task.requeue();
10939 }
10940 let check = |task: &Task, starter: &Starter, limit| {
10941 retained_attempt_hold(task, starter, limit, |id| Ok(history[id].clone()))
10942 };
10943 assert!(
10944 check(&task, &Starter::Start, 8)
10945 .unwrap()
10946 .contains("8 unretired")
10947 );
10948 assert!(check(&task, &Starter::Start, 9).is_none());
10949 assert!(check(&task, &Starter::Review("existing".into()), 8).is_some());
10950 task.release_operator("owner", None);
10951 assert_eq!(task.attempts, 0);
10952 let restored: Task = serde_json::from_slice(&serde_json::to_vec(&task).unwrap()).unwrap();
10953 assert!(check(&restored, &Starter::Start, 8).is_some());
10954 assert!(check(&restored, &Starter::Resume(task.runs[0].clone()), 8).is_none());
10955 assert!(check(&restored, &Starter::Start, 0).is_none());
10956 }
10957
10958 #[test]
10959 fn retained_attempt_bound_requires_verified_retirement_not_completion_labels() {
10960 let mut task = Task::new(
10961 "recover work".into(),
10962 "recover work".into(),
10963 PathBuf::from("/repo"),
10964 crate::queue::Source::Human,
10965 );
10966 let mut run = run_state(RunStatus::Superseded);
10967 run.candidates.clear();
10968 task.runs.push(run.id.clone());
10969 assert!(retained_attempt_hold(&task, &Starter::Start, 1, |_| Ok(run.clone())).is_some());
10970 run.retired_to = Some("authority".into());
10971 assert!(retained_attempt_hold(&task, &Starter::Start, 1, |_| Ok(run.clone())).is_none());
10972 run.retired_to = None;
10973 run.released_to = Some("authority".into());
10974 assert!(retained_attempt_hold(&task, &Starter::Start, 1, |_| Ok(run.clone())).is_none());
10975 assert!(
10976 retained_attempt_hold(&task, &Starter::Start, 8, |_| anyhow::bail!("unreadable"))
10977 .unwrap()
10978 .contains("Cannot verify")
10979 );
10980 assert!(
10982 retained_attempt_hold(&task, &Starter::Resume(run.id), 1, |_| {
10983 panic!("resume must not load historical attempts")
10984 })
10985 .is_none()
10986 );
10987 }
10988
10989 #[test]
10990 fn retired_resume_requires_a_valid_same_repository_authority() {
10991 let mut source = run_state(RunStatus::Blocked);
10992 let mut empty = run_state(RunStatus::Blocked);
10993 empty.retired_to = Some(source.id.clone());
10994 let runs = vec![source.id.clone(), empty.id.clone()];
10995 let resolve = |source: &RunState, empty: &RunState| {
10996 unfinished_run_with(&runs, "fixture", |id| {
10997 if id == source.id {
10998 Ok(source.clone())
10999 } else if id == empty.id {
11000 Ok(empty.clone())
11001 } else {
11002 bail!("unreadable authority")
11003 }
11004 })
11005 };
11006 assert_eq!(resolve(&source, &empty), Some(source.id.clone()));
11007 assert_eq!(
11008 choose_starter(None, false, resolve(&source, &empty).as_deref()),
11009 Starter::Resume(source.id.clone())
11010 );
11011 source.retired_to = Some(empty.id.clone());
11012 assert!(
11013 resolve(&source, &empty).is_none(),
11014 "cycles do not revive obsolete runs"
11015 );
11016 source.retired_to = None;
11017 source.repo = PathBuf::from("/another-repository");
11018 assert!(resolve(&source, &empty).is_none());
11019 source.repo = empty.repo.clone();
11020 source.released_to = Some("later".into());
11021 assert!(resolve(&source, &empty).is_none());
11022 empty.retired_to = Some("unlisted-run".into());
11023 assert!(resolve(&source, &empty).is_none());
11024 }
11025
11026 #[test]
11027 fn unfinished_run_never_falls_back_to_an_older_resumable_run() {
11028 let mut older_stalled = RunState::new(
11036 PathBuf::from("/repo"),
11037 "main".to_owned(),
11038 "abc1234def".to_owned(),
11039 "add retries".to_owned(),
11040 Config::default(),
11041 );
11042 older_stalled.status = RunStatus::Stalled;
11043
11044 let mut newest_exhausted = RunState::new(
11045 PathBuf::from("/repo"),
11046 "main".to_owned(),
11047 "abc1234def".to_owned(),
11048 "add retries".to_owned(),
11049 Config::default(),
11050 );
11051 newest_exhausted.status = RunStatus::Blocked;
11052 newest_exhausted.config.graph.review_rounds = 1;
11053 newest_exhausted.reviews = vec![review_round(1)];
11054
11055 assert_eq!(
11056 unfinished_run_with(
11057 &[older_stalled.id.clone(), newest_exhausted.id.clone()],
11058 "t",
11059 |_| Ok(newest_exhausted.clone())
11060 ),
11061 None,
11062 "the newest run is exhausted, so nothing here is worth resuming - \
11063 least of all the older, already-superseded run"
11064 );
11065 }
11066
11067 #[test]
11068 fn unfinished_run_warns_and_skips_a_run_it_cannot_read() {
11069 assert_eq!(
11070 unfinished_run_with(&["20260101-000000-gone".to_owned()], "t", |_| {
11071 Err(anyhow::anyhow!("fixture is absent"))
11072 }),
11073 None
11074 );
11075 }
11076
11077 fn action_question(run: &str, action: ask::ChoiceAction) -> ask::Question {
11078 let mut q = ask::Question::new(
11079 run.to_owned(),
11080 "implement".to_owned(),
11081 "impl-A".to_owned(),
11082 "continue?".to_owned(),
11083 String::new(),
11084 vec!["resume で続行する".to_owned(), "other".to_owned()],
11085 );
11086 q.actions.insert("resume で続行する".to_owned(), action);
11087 q.answer(ask::Answer::Choice("resume で続行する".to_owned()))
11088 .unwrap();
11089 q
11090 }
11091
11092 fn held_task_with(run: &str) -> Task {
11093 let mut t = task();
11094 t.runs = vec![run.to_owned()];
11095 t.hold_machine(Some("waiting for magi resume to be executed".to_owned()));
11096 t
11097 }
11098
11099 fn resume_action(run: &str) -> ask::ChoiceAction {
11100 ask::ChoiceAction::Resume { run: run.into() }
11101 }
11102
11103 #[test]
11104 fn explicitly_resumed_older_run_remains_the_release_and_question_target() {
11105 for parked in [false, true] {
11106 let mut t = task();
11107 t.start("A".into());
11108 t.start("B".into());
11109 assert_eq!(t.runs, vec!["A", "B"]);
11110 t.start("A".into());
11111 let mut resumed = run_state(RunStatus::Blocked);
11112 resumed.id = "A".into();
11113 resumed.parked = parked;
11114 resumed.driver_exited = true;
11115 t.hold_machine(Some("resumed run stopped".into()));
11116 let q = action_question("A", resume_action("A"));
11117 assert_eq!(action_standing(&t, &q), ActionStanding::Pending);
11118 assert_eq!(
11119 decide_action(&t, &q, &PHRASES_EN, |id| {
11120 assert_eq!(id, "A");
11121 Ok(resumed.clone())
11122 }),
11123 ActionDecision::Resume("A".into())
11124 );
11125 assert_eq!(
11126 action_standing(&t, &action_question("B", resume_action("B"))),
11127 ActionStanding::Stale
11128 );
11129 t.release_operator("cli", None);
11130 let unfinished = unfinished_run_with(&t.runs, t.short(), |id| {
11131 assert_eq!(id, "A");
11132 Ok(resumed.clone())
11133 });
11134 assert_eq!(
11135 choose_starter(None, false, unfinished.as_deref()),
11136 Starter::Resume("A".into())
11137 );
11138 assert_eq!(t.runs, vec!["B", "A"]);
11139 }
11140 }
11141
11142 #[test]
11143 fn merge_repair_dispatch_refuses_obsolete_bindings_and_unsupported_owners() {
11144 let mut state = crate::land::tests::repair_state(true, true);
11145 state.status = RunStatus::Landing;
11146 state.parked = true;
11147 state.driver_exited = true;
11148 let mut q = ask::Question::new(
11149 state.id.clone(),
11150 land::APPROVAL_NODE.into(),
11151 "land".into(),
11152 "Merge?".into(),
11153 String::new(),
11154 vec![land::APPROVE.into(), land::HOLD.into()],
11155 );
11156 q.id = state.id.clone();
11157 state.land_approval = Some(crate::run::LandApproval {
11158 question: q.id.clone(),
11159 head: "current-head".into(),
11160 });
11161 assert!(land::upgrade_repair_choice(&mut q, &state));
11162 q.answer(ask::Answer::Choice(land::REPAIR.into())).unwrap();
11163 let mut task = held_task_with(&state.id);
11164 task.status = TaskStatus::Parked;
11165 assert_eq!(
11166 decide_action(&task, &q, &PHRASES_EN, |_| Ok(state.clone())),
11167 ActionDecision::Resume(state.id.clone())
11168 );
11169 q.asked_at = "2026-01-01T00:00:00Z".parse().unwrap();
11170 q.answered_at = Some("2026-01-01T00:00:20Z".parse().unwrap());
11171 task.operator_decided_at = Some("2026-01-01T00:00:10Z".parse().unwrap());
11172 assert_eq!(
11173 decide_action(&task, &q, &PHRASES_EN, |_| Ok(state.clone())),
11174 ActionDecision::Resume(state.id.clone()),
11175 "a repair answer newer than the manual hold resumes the existing owner"
11176 );
11177 task.operator_decided_at = Some("2026-01-01T00:00:30Z".parse().unwrap());
11178 for answered_at in [q.answered_at, None] {
11179 let mut older = q.clone();
11180 older.answered_at = answered_at;
11181 let refused = decide_action(&task, &older, &PHRASES_EN, |_| {
11182 panic!("a later manual hold must prevent dispatch")
11183 });
11184 let ActionDecision::Refuse(reason) = refused else {
11185 panic!("expected an actionable refusal, got {refused:?}");
11186 };
11187 assert!(reason.contains("later operator decision"));
11188 assert!(reason.contains(&format!("magi task release {}", task.id)));
11189 }
11190 task.operator_decided_at = None;
11191 let mut superseded = state.clone();
11192 superseded.land_approval.as_mut().unwrap().question = "fresh-decision".into();
11193 assert_eq!(
11194 decide_action(&task, &q, &PHRASES_EN, |_| Ok(superseded)),
11195 ActionDecision::Stale
11196 );
11197 for case in 1..4 {
11198 let mut changed = state.clone();
11199 match case {
11200 1 => changed.status = RunStatus::Blocked,
11201 2 => changed.released_to = Some("new-owner".into()),
11202 _ => changed.land_approval.as_mut().unwrap().head.clear(),
11203 }
11204 assert!(matches!(
11205 decide_action(&task, &q, &PHRASES_EN, |_| Ok(changed)),
11206 ActionDecision::Refuse(_)
11207 ));
11208 }
11209 assert!(matches!(
11210 decide_action(&task, &q, &PHRASES_EN, |_| anyhow::bail!("missing owner")),
11211 ActionDecision::Refuse(_)
11212 ));
11213 task.runs.push("replacement".into());
11214 task.operator_decided_at = Some("2026-01-01T00:00:30Z".parse().unwrap());
11215 assert_eq!(
11216 decide_action(&task, &q, &PHRASES_EN, |_| panic!(
11217 "old owner must not load"
11218 )),
11219 ActionDecision::Stale
11220 );
11221 }
11222
11223 #[test]
11224 fn merge_repair_dispatch_obeys_capacity_and_quota_while_hold_stays_decision_only() {
11225 let mut state = crate::land::tests::repair_state(true, true);
11226 state.status = RunStatus::Landing;
11227 state.parked = true;
11228 state.driver_exited = true;
11229 let mut q = ask::Question::new(
11230 state.id.clone(),
11231 land::APPROVAL_NODE.into(),
11232 "land".into(),
11233 "Merge?".into(),
11234 String::new(),
11235 vec![land::APPROVE.into(), land::HOLD.into()],
11236 );
11237 q.id = state.id.clone();
11238 state.land_approval = Some(crate::run::LandApproval {
11239 question: q.id.clone(),
11240 head: "current-head".into(),
11241 });
11242 state.save().unwrap();
11243 land::upgrade_repair_choice(&mut q, &state);
11244 q.answer(ask::Answer::Choice(land::REPAIR.into())).unwrap();
11245 Questions::open().put(&mut q).unwrap();
11246 let tmp = tempfile::tempdir().unwrap();
11247 let queue = Queue::at(tmp.path().join("queue"));
11248 let mut task = held_task_with(&state.id);
11249 task.status = TaskStatus::Parked;
11250 queue.put(&mut task).unwrap();
11251 let empty = Arc::new(tokio::sync::Semaphore::new(0));
11252 let free = Arc::new(tokio::sync::Semaphore::new(1));
11253 assert!(claim_dispatch(&queue, &task, false, &empty, &free).is_none());
11254 assert!(claim_dispatch(&queue, &task, true, &free, &free).is_none());
11255 let claim = claim_dispatch(&queue, &task, false, &free, &free).unwrap();
11256 assert!(claim.permit.is_some());
11257 assert!(claim_dispatch(&queue, &task, false, &free, &free).is_none());
11258 drop(claim);
11259 Questions::open()
11260 .update(&q.id, |q| {
11261 q.answer = Some(ask::Answer::Choice(land::HOLD.into()));
11262 Ok(())
11263 })
11264 .unwrap();
11265 let claim = claim_dispatch(&queue, &task, true, &empty, &empty).unwrap();
11266 assert!(claim.permit.is_none());
11267 assert!(RunState::load(&state.id).unwrap().merge_repairs.is_empty());
11268 }
11269
11270 #[test]
11271 fn decide_action_resumes_only_the_latest_resumable_run() {
11272 let t = held_task_with("r1");
11273 let q = action_question("r1", resume_action("r1"));
11274 let load = |s: RunState| move |_: &str| Ok(s);
11275 assert_eq!(
11276 decide_action(&t, &q, &PHRASES_EN, load(run_state(RunStatus::Blocked))),
11277 ActionDecision::Resume("r1".into())
11278 );
11279 let q_other = action_question("r1", resume_action("r0"));
11281 assert!(matches!(
11282 decide_action(
11283 &t,
11284 &q_other,
11285 &PHRASES_EN,
11286 load(run_state(RunStatus::Blocked))
11287 ),
11288 ActionDecision::Refuse(_)
11289 ));
11290 assert!(matches!(
11292 decide_action(&t, &q, &PHRASES_EN, load(run_state(RunStatus::Ready))),
11293 ActionDecision::Refuse(_)
11294 ));
11295 let mut released = run_state(RunStatus::Blocked);
11297 released.released_to = Some("elsewhere".into());
11298 assert!(matches!(
11299 decide_action(&t, &q, &PHRASES_EN, load(released)),
11300 ActionDecision::Refuse(_)
11301 ));
11302 assert!(matches!(
11304 decide_action(&t, &q, &PHRASES_EN, |_: &str| bail!("gone")),
11305 ActionDecision::Refuse(_)
11306 ));
11307 }
11308
11309 #[test]
11310 fn decide_action_ignores_a_question_about_an_earlier_run() {
11311 let mut t = held_task_with("r1");
11312 t.runs.push("r2".to_owned());
11313 let never = |_: &str| -> Result<RunState> { bail!("not read") };
11314 assert_eq!(
11315 decide_action(
11316 &t,
11317 &action_question("r1", ask::ChoiceAction::Done),
11318 &PHRASES_EN,
11319 never
11320 ),
11321 ActionDecision::Stale
11322 );
11323 }
11324
11325 #[test]
11326 fn decide_action_maps_requeue_and_done_and_never_acts_twice() {
11327 let mut t = held_task_with("r1");
11328 let never = |_: &str| -> Result<RunState> { bail!("not read") };
11329 assert_eq!(
11330 decide_action(
11331 &t,
11332 &action_question("r1", ask::ChoiceAction::Requeue),
11333 &PHRASES_EN,
11334 never
11335 ),
11336 ActionDecision::Requeue
11337 );
11338 let done_q = action_question("r1", ask::ChoiceAction::Done);
11339 assert_eq!(
11340 decide_action(&t, &done_q, &PHRASES_EN, never),
11341 ActionDecision::Done
11342 );
11343 t.mark_action_applied(&done_q.id);
11344 assert_eq!(
11345 decide_action(&t, &done_q, &PHRASES_EN, never),
11346 ActionDecision::Skip
11347 );
11348
11349 let mut plain = action_question("r1", ask::ChoiceAction::Done);
11351 plain.actions.clear();
11352 assert_eq!(
11353 decide_action(&held_task_with("r1"), &plain, &PHRASES_EN, never),
11354 ActionDecision::Skip
11355 );
11356 let mut running = held_task_with("r1");
11358 running.status = TaskStatus::Running;
11359 assert_eq!(
11360 decide_action(
11361 &running,
11362 &action_question("r1", ask::ChoiceAction::Done),
11363 &PHRASES_EN,
11364 never
11365 ),
11366 ActionDecision::Skip
11367 );
11368 }
11369
11370 #[test]
11371 fn merge_repair_rejected_after_manual_hold_keeps_the_conductor_guard() {
11372 crate::run::pin_test_home();
11373 for legacy in [false, true] {
11374 let dir = tempfile::tempdir().unwrap();
11375 let queue = Queue::at(dir.path().join("queue"));
11376 let questions = Questions::at(dir.path().join("questions"));
11377 let run = "20260101-000000-hold";
11378 let mut task = held_task_with(run);
11379 task.status = TaskStatus::Parked;
11380 let mut q = ask::Question::new(
11381 run.into(),
11382 land::APPROVAL_NODE.into(),
11383 "land".into(),
11384 "Repair?".into(),
11385 String::new(),
11386 vec![land::REPAIR.into()],
11387 );
11388 q.actions.insert(land::REPAIR.into(), resume_action(run));
11389 q.answer(ask::Answer::Choice(land::REPAIR.into())).unwrap();
11390 q.asked_at = "2026-01-01T00:00:00Z".parse().unwrap();
11392 q.answered_at = Some("2026-01-01T00:00:01Z".parse().unwrap());
11393 questions.put(&mut q).unwrap();
11394 task.hold_manual(Some("Keep this PR paused".into()));
11395 if legacy {
11396 task.hold_source = None;
11397 }
11398 queue.put(&mut task).unwrap();
11399
11400 apply_task_choice_actions(&queue, &questions, dir.path());
11401 let after = queue.get(&task.id).unwrap();
11402 assert!(after.operator_held());
11403 assert_eq!(after.hold_source, task.hold_source);
11404 assert_eq!(after.operator_decided_at, task.operator_decided_at);
11405 assert_eq!(after.runs, task.runs);
11406 assert_eq!(after.instruction, task.instruction);
11407 assert!(after.resume_override.is_none());
11408 assert!(after.action_applied(&q.id));
11409 let reason = after.hold_reason.as_deref().unwrap();
11410 assert!(reason.starts_with("Keep this PR paused"));
11411 assert!(reason.contains("later operator decision"));
11412 assert!(reason.contains(&format!("magi task release {}", task.id)));
11413
11414 for recovery in [
11415 crate::conduct::Recovery::Review,
11416 crate::conduct::Recovery::Requeue,
11417 ] {
11418 crate::conduct::apply(
11419 &queue,
11420 &questions,
11421 &crate::conduct::Verdict {
11422 decisions: vec![crate::conduct::Decision {
11423 id: task.id.clone(),
11424 recovery: Some(recovery),
11425 ..Default::default()
11426 }],
11427 },
11428 )
11429 .unwrap();
11430 }
11431 apply_task_choice_actions(&queue, &questions, dir.path());
11434 let still_held = queue.get(&task.id).unwrap();
11435 assert!(still_held.operator_held());
11436 assert_eq!(still_held.hold_reason, after.hold_reason);
11437 assert_eq!(still_held.runs, task.runs);
11438 assert!(still_held.review_branch.is_none());
11439 assert!(!still_held.fresh_start);
11440 assert!(queue.next_runnable().is_none());
11441 }
11442 }
11443
11444 #[test]
11445 fn conductor_driver_choices_apply_from_both_queued_and_held_tasks() {
11446 for held in [false, true] {
11447 for retry in [false, true] {
11448 let dir = tempfile::tempdir().unwrap();
11449 let queue = Queue::at(dir.path().join("queue"));
11450 let questions = Questions::at(dir.path().join("questions"));
11451 let home = dir.path().join("home");
11452 let mut t = task();
11453 if held {
11454 t.hold_machine(Some("driver state unknown".into()));
11455 }
11456 queue.put(&mut t).unwrap();
11457 let verdict = serde_json::from_value(serde_json::json!({
11458 "decisions": [{
11459 "id": t.id,
11460 "question": "Driver state unknown; retry the checks or hold?",
11461 "choices": ["daemon: recheck and retry", "daemon: hold the task"],
11462 "actions": {
11463 "daemon: recheck and retry": {"do": "requeue"},
11464 "daemon: hold the task": {"do": "hold"}
11465 }
11466 }]
11467 }))
11468 .unwrap();
11469 crate::conduct::apply(&queue, &questions, &verdict).unwrap();
11470 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Blocked);
11471 let mut q = questions.list().pop().unwrap();
11472 let choice = if retry {
11473 "daemon: recheck and retry"
11474 } else {
11475 "daemon: hold the task"
11476 };
11477 assert_eq!(q.actions.len(), 2);
11478 q.answer(ask::Answer::Choice(choice.into())).unwrap();
11479 questions.put(&mut q).unwrap();
11480
11481 questions.beat(&q.id, ask::WaiterKind::Deputy);
11483 for _ in 0..2 {
11484 resolve_blockers(&queue, &questions);
11485 apply_task_choice_actions(&queue, &questions, &home);
11486 let waiting = queue.get(&t.id).unwrap();
11487 assert_eq!(waiting.status, TaskStatus::Blocked);
11488 assert_eq!(waiting.blocked_by, [q.id.clone()]);
11489 assert!(waiting.answers.is_empty());
11490 assert!(!waiting.action_applied(&q.id));
11491 assert!(runnable(&queue).is_empty());
11492 let free = Arc::new(tokio::sync::Semaphore::new(1));
11493 assert!(claim_dispatch(&queue, &t, false, &free, &free).is_none());
11494 }
11495 std::fs::remove_file(questions.root().join(format!("{}.lease", q.id))).unwrap();
11496
11497 resolve_blockers(&queue, &questions);
11499 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Blocked);
11500 apply_task_choice_actions(&queue, &questions, &home);
11501 let after = queue.get(&t.id).unwrap();
11502 assert_eq!(
11503 after.status,
11504 if retry {
11505 TaskStatus::Queued
11506 } else {
11507 TaskStatus::Held
11508 }
11509 );
11510 assert!(after.action_applied(&q.id));
11511 assert_eq!(after.operator_held(), !retry);
11512 assert_eq!(after.answers.len(), 1);
11513
11514 let mut changed = after;
11516 if retry {
11517 changed.hold_manual(Some("later hold".into()));
11518 } else {
11519 changed.release_operator("test", None);
11520 }
11521 let expected = changed.status;
11522 queue.put(&mut changed).unwrap();
11523 apply_task_choice_actions(&queue, &questions, &home);
11524 assert_eq!(queue.get(&t.id).unwrap().status, expected);
11525 }
11526 }
11527 }
11528
11529 #[test]
11530 fn apply_choice_actions_releases_the_task_pinned_to_its_run_and_only_once() {
11531 let dir = tempfile::tempdir().unwrap();
11532 let queue = Queue::at(dir.path().join("queue"));
11533 let questions = Questions::at(dir.path().join("questions"));
11534 let home = dir.path().join("home");
11535 let mut state = run_state(RunStatus::Blocked);
11536 state.id = "20260101-000000-act1".to_owned();
11537 state.save_under(&home).unwrap();
11538
11539 let mut t = held_task_with(&state.id);
11540 queue.put(&mut t).unwrap();
11541 let mut q = action_question(&state.id, resume_action(&state.id));
11542 questions.put(&mut q).unwrap();
11543
11544 apply_task_choice_actions(&queue, &questions, &home);
11545 let after = queue.get(&t.id).unwrap();
11546 assert_eq!(after.status, TaskStatus::Queued);
11547 assert!(!after.fresh_start);
11548 assert!(after.action_applied(&q.id));
11549 let pin = after.resume_override.clone().unwrap();
11550 assert_eq!(pin.pinned_run.as_deref(), Some(state.id.as_str()));
11551 assert!(pin.forced);
11552
11553 let mut again = queue.get(&t.id).unwrap();
11555 again.hold_machine(Some("later".into()));
11556 queue.put(&mut again).unwrap();
11557 apply_task_choice_actions(&queue, &questions, &home);
11558 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Held);
11559 }
11560
11561 #[test]
11562 fn a_delivered_answer_is_still_acted_on_exactly_once() {
11563 let dir = tempfile::tempdir().unwrap();
11564 let queue = Queue::at(dir.path().join("queue"));
11565 let questions = Questions::at(dir.path().join("questions"));
11566 let home = dir.path().join("home");
11567 let mut state = run_state(RunStatus::Blocked);
11568 state.id = "20260101-000000-act2".to_owned();
11569 state.save_under(&home).unwrap();
11570
11571 let mut t = held_task_with(&state.id);
11572 queue.put(&mut t).unwrap();
11573 let mut q = action_question(&state.id, resume_action(&state.id));
11574 q.answer_delivered = true;
11575 questions.put(&mut q).unwrap();
11576
11577 apply_task_choice_actions(&queue, &questions, &home);
11580 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Queued);
11581 assert!(queue.get(&t.id).unwrap().action_applied(&q.id));
11582
11583 let mut again = queue.get(&t.id).unwrap();
11585 again.hold_machine(Some("later".into()));
11586 queue.put(&mut again).unwrap();
11587 apply_task_choice_actions(&queue, &questions, &home);
11588 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Held);
11589 }
11590
11591 #[test]
11592 fn a_fresh_asker_defers_the_action_until_it_goes_quiet() {
11593 let dir = tempfile::tempdir().unwrap();
11594 let queue = Queue::at(dir.path().join("queue"));
11595 let questions = Questions::at(dir.path().join("questions"));
11596 let home = dir.path().join("home");
11597 let mut state = run_state(RunStatus::Blocked);
11598 state.id = "20260101-000000-act3".to_owned();
11599 state.save_under(&home).unwrap();
11600 let mut t = held_task_with(&state.id);
11601 queue.put(&mut t).unwrap();
11602 let mut q = action_question(&state.id, ask::ChoiceAction::Requeue);
11603 questions.put(&mut q).unwrap();
11604
11605 questions.beat(&q.id, crate::ask::WaiterKind::Asker);
11606 apply_task_choice_actions(&queue, &questions, &home);
11607 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Held);
11608
11609 std::fs::remove_file(questions.root().join(format!("{}.lease", q.id))).unwrap();
11610 apply_task_choice_actions(&queue, &questions, &home);
11611 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Queued);
11612 }
11613
11614 #[test]
11615 fn action_standing_tells_the_reasons_apart() {
11616 let mut t = held_task_with("r1");
11617 let q = action_question("r1", ask::ChoiceAction::Requeue);
11618 assert_eq!(action_standing(&t, &q), ActionStanding::Pending);
11619 let mut plain = q.clone();
11620 plain.actions.clear();
11621 assert_eq!(action_standing(&t, &plain), ActionStanding::NoAction);
11622
11623 t.mark_action_applied(&q.id);
11625 assert_eq!(action_standing(&t, &q), ActionStanding::Applied);
11626
11627 let mut t = held_task_with("r1");
11629 t.start("r2".to_owned());
11630 assert_eq!(action_standing(&t, &q), ActionStanding::Stale);
11631 let q2 = action_question("r2", ask::ChoiceAction::Requeue);
11633 assert_eq!(action_standing(&t, &q2), ActionStanding::Busy);
11634 t.status = TaskStatus::Blocked;
11635 assert_eq!(action_standing(&t, &q2), ActionStanding::Busy);
11636
11637 let mut c = action_question("r0", ask::ChoiceAction::Requeue);
11639 c.node = crate::conduct::NODE.to_owned();
11640 assert_eq!(action_standing(&t, &c), ActionStanding::Busy);
11641 assert!(!ActionStanding::Busy.daemon_owns());
11642 }
11643
11644 #[test]
11645 fn a_running_task_waits_and_a_dead_asker_with_a_cwd_is_still_actioned() {
11646 let dir = tempfile::tempdir().unwrap();
11647 let queue = Queue::at(dir.path().join("queue"));
11648 let questions = Questions::at(dir.path().join("questions"));
11649 let home = dir.path().join("home");
11650 let mut state = run_state(RunStatus::Blocked);
11651 state.id = "20260101-000000-act4".to_owned();
11652 state.save_under(&home).unwrap();
11653 let mut t = held_task_with(&state.id);
11654 t.status = TaskStatus::Running;
11655 queue.put(&mut t).unwrap();
11656 let mut q = action_question(&state.id, ask::ChoiceAction::Done);
11657 q.cwd = Some(dir.path().display().to_string());
11658 questions.put(&mut q).unwrap();
11659
11660 let running = queue.get(&t.id).unwrap();
11663 assert_eq!(action_standing(&running, &q), ActionStanding::Busy);
11664 assert!(matches!(
11665 crate::waiter::decide_owned(&q, None, false, 86_400, Timestamp::now(), false),
11666 crate::waiter::Action::Deliver(_)
11667 ));
11668 apply_task_choice_actions(&queue, &questions, &home);
11670 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Running);
11671 let mut back = queue.get(&t.id).unwrap();
11672 back.hold_machine(Some("later".into()));
11673 queue.put(&mut back).unwrap();
11674 apply_task_choice_actions(&queue, &questions, &home);
11675 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Done);
11676 let mut again = queue.get(&t.id).unwrap();
11677 again.hold_machine(Some("again".into()));
11678 queue.put(&mut again).unwrap();
11679 apply_task_choice_actions(&queue, &questions, &home);
11680 assert_eq!(queue.get(&t.id).unwrap().status, TaskStatus::Held);
11681 }
11682}