1use std::path::{Path, PathBuf};
16use std::sync::Arc;
17
18use agentplane::model::ModelProvider;
19use serde::{Deserialize, Serialize};
20use serde_json::{Value, json};
21use speccoder_core::acceptance::{Decided, PreFrozenAcceptance, Producer, SealedRef, VisibleTest};
22use speccoder_core::actions::Via;
23use speccoder_core::context::{ContextEntry, ContextKind};
24use speccoder_core::ladder::Rung;
25use speccoder_core::oracle::{
26 Ambiguity, AuthoredBy, Expectation, HeldOutSummary, OracleDiscarded, OracleOp, OracleRejected,
27 OracleState, OracleTest, OracleView, Reading, Resolution, Untestable,
28};
29use speccoder_core::snapshot::ProjectSnapshot;
30use speccoder_core::verify::{RedClass, classify_red};
31use speccoder_core::{CommandData, CommandResult, CommandStatus};
32
33use crate::format::{Adapter, OracleRequirement};
34use crate::integrity::sha256_hex;
35use crate::{sandbox, substrate, tools, worktree};
36
37pub const DIR: &str = "oracle";
39
40pub const LIMIT: usize = 4;
42
43#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
45pub struct Draft {
46 pub requirement: DraftRequirement,
48 pub samples: u8,
50 pub tests: Vec<DraftTest>,
52 pub discarded: Vec<OracleDiscarded>,
54 pub rejected: Vec<OracleRejected>,
56 pub expectations: Samples<Expectation>,
58 pub untestable: Samples<Untestable>,
60 pub ambiguities: Vec<Ambiguity>,
62 pub held_out: HeldOutSummary,
64 pub sealed: Option<SealedRef>,
66 pub notes: Vec<String>,
68 pub runs: Vec<String>,
70 pub context: Vec<ContextEntry>,
73}
74
75#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
77pub struct DraftRequirement {
78 pub key: String,
80 pub title: String,
82 pub text: String,
84 pub sha256: String,
86}
87
88#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
90pub struct DraftTest {
91 pub path: String,
93 pub sha256: String,
95 pub scenarios: Vec<String>,
97 pub argv: Vec<String>,
99 pub red_class: RedClass,
101 pub authored_by: AuthoredBy,
103 pub edited_by: Option<String>,
105}
106
107#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
109#[serde(bound(deserialize = "T: Deserialize<'de>"))]
110pub struct Samples<T> {
111 pub sample1: Vec<T>,
113 pub sample2: Vec<T>,
115}
116
117impl<T> Default for Samples<T> {
118 fn default() -> Self {
119 Self {
120 sample1: Vec::new(),
121 sample2: Vec::new(),
122 }
123 }
124}
125
126#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
128pub struct OracleEntry {
129 pub requirement_sha256: String,
131 pub manifest_sha256: String,
133 #[serde(default, skip_serializing_if = "Option::is_none")]
135 pub held_out_sha256: Option<String>,
136 pub by: String,
138 pub via: Via,
140 pub at: u64,
142}
143
144#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
147pub struct Frozen {
148 pub requirement_sha256: String,
150 pub wording: String,
152 pub manifest_sha256: String,
154 pub tests: Vec<FrozenTest>,
156}
157
158#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
160pub struct FrozenTest {
161 pub path: String,
163 pub sha256: String,
165 pub argv: Vec<String>,
167 pub scenarios: Vec<String>,
169}
170
171pub type Loaded = (Draft, Vec<(String, String)>);
173
174type Kept = (String, String, Vec<String>, Vec<String>);
176
177#[derive(Debug, Clone, PartialEq, Eq)]
179pub struct Tampered(pub String);
180
181fn slug(key: &str) -> String {
182 key.chars()
183 .map(|c| {
184 if c.is_ascii_alphanumeric() || matches!(c, '-' | '_' | '.') {
185 c
186 } else {
187 '_'
188 }
189 })
190 .collect()
191}
192
193#[must_use]
195pub fn dir(project_state: &Path, unit: &str, key: &str) -> PathBuf {
196 project_state.join(DIR).join(unit).join(slug(key))
197}
198
199fn now() -> u64 {
200 std::time::SystemTime::now()
201 .duration_since(std::time::UNIX_EPOCH)
202 .map_or(0, |d| d.as_secs())
203}
204
205fn digest_of(text: &str) -> String {
206 format!("sha256:{}", sha256_hex(text.as_bytes()))
207}
208
209fn write_atomic(file: &Path, text: &str) -> Result<(), String> {
210 if let Some(parent) = file.parent() {
211 std::fs::create_dir_all(parent).map_err(|e| format!("{}: {e}", parent.display()))?;
212 }
213 let tmp = file.with_extension("json.tmp");
214 std::fs::write(&tmp, text).map_err(|e| format!("{}: {e}", tmp.display()))?;
215 std::fs::rename(&tmp, file).map_err(|e| format!("{}: {e}", file.display()))
216}
217
218pub fn store(
221 project_state: &Path,
222 unit: &str,
223 draft: &Draft,
224 contents: &[(String, String)],
225) -> Result<(), String> {
226 let d = dir(project_state, unit, &draft.requirement.key);
227 let _ = std::fs::remove_dir_all(d.join("files"));
228 for t in &draft.tests {
229 let content = contents
230 .iter()
231 .find(|(p, _)| *p == t.path)
232 .map(|(_, c)| c)
233 .ok_or_else(|| format!("{} has no content", t.path))?;
234 if digest_of(content) != t.sha256 {
235 return Err(format!("{} does not match its digest", t.path));
236 }
237 let file = d.join("files").join(&t.path);
238 if let Some(parent) = file.parent() {
239 std::fs::create_dir_all(parent).map_err(|e| format!("{}: {e}", parent.display()))?;
240 }
241 std::fs::write(&file, content).map_err(|e| format!("{}: {e}", file.display()))?;
242 }
243 let text = serde_json::to_string_pretty(draft).map_err(|e| e.to_string())?;
244 write_atomic(&d.join("draft.json"), &text)
245}
246
247pub fn load(project_state: &Path, unit: &str, key: &str) -> Result<Option<Loaded>, Tampered> {
250 let d = dir(project_state, unit, key);
251 let Ok(text) = std::fs::read_to_string(d.join("draft.json")) else {
252 return Ok(None);
253 };
254 let draft: Draft = serde_json::from_str(&text)
255 .map_err(|e| Tampered(format!("the oracle draft of {key} is not valid: {e}")))?;
256 let mut contents = Vec::new();
257 for t in &draft.tests {
258 let content = std::fs::read_to_string(d.join("files").join(&t.path))
259 .map_err(|_| Tampered(format!("the test {} of {key} is missing", t.path)))?;
260 if digest_of(&content) != t.sha256 {
261 return Err(Tampered(format!("the test {} of {key} changed", t.path)));
262 }
263 contents.push((t.path.clone(), content));
264 }
265 Ok(Some((draft, contents)))
266}
267
268#[must_use]
270pub fn load_frozen(project_state: &Path, unit: &str, key: &str) -> Option<Frozen> {
271 std::fs::read_to_string(dir(project_state, unit, key).join("frozen.json"))
272 .ok()
273 .and_then(|t| serde_json::from_str(&t).ok())
274}
275
276#[must_use]
279pub fn manifest_digest<'a>(
280 tests: impl IntoIterator<Item = (&'a str, &'a str, &'a [String])>,
281) -> String {
282 let mut lines: Vec<String> = tests
283 .into_iter()
284 .map(|(path, sha, argv)| {
285 format!(
286 "{path}={sha} argv={}",
287 serde_json::to_string(argv).unwrap_or_default()
288 )
289 })
290 .collect();
291 lines.sort();
292 digest_of(&lines.join("\n"))
293}
294
295impl Draft {
296 pub fn red(&self) -> impl Iterator<Item = &DraftTest> {
298 self.tests.iter().filter(|t| t.red_class.is_red())
299 }
300
301 #[must_use]
303 pub fn manifest(&self) -> Option<String> {
304 let red: Vec<&DraftTest> = self.red().collect();
305 (!red.is_empty()).then(|| {
306 manifest_digest(
307 red.iter()
308 .map(|t| (t.path.as_str(), t.sha256.as_str(), t.argv.as_slice())),
309 )
310 })
311 }
312
313 pub fn open_ambiguities(&self) -> impl Iterator<Item = &Ambiguity> {
315 self.ambiguities.iter().filter(|a| a.open())
316 }
317
318 fn all_notes(&self) -> Vec<String> {
320 self.rejected
321 .iter()
322 .map(|r| format!("Rejected `{}`: {}", r.path, r.reason))
323 .chain(self.notes.iter().cloned())
324 .collect()
325 }
326}
327
328#[derive(Debug, Clone, Copy)]
330pub struct ViewConfig {
331 pub samples: u8,
333 pub strength_target: Option<u8>,
335 pub held_out: bool,
337 pub acceptance: bool,
339}
340
341impl ViewConfig {
342 #[must_use]
344 pub fn at(root: &Path) -> Self {
345 let rc = crate::verify::find_config(root)
346 .and_then(|d| std::fs::read_to_string(d.join(crate::verify::config::FILE)).ok())
347 .and_then(|t| substrate::run_config::parse(&t).ok())
348 .unwrap_or_default();
349 Self::of(&rc)
350 }
351
352 #[must_use]
354 pub fn of(rc: &substrate::run_config::RunConfig) -> Self {
355 Self {
356 samples: rc.oracle_samples,
357 strength_target: rc.strength.min.map(|m| {
358 let percent = (m * 100.0).round().clamp(0.0, 100.0);
360 format!("{percent:.0}").parse().unwrap_or(0)
361 }),
362 held_out: rc.held_out,
363 acceptance: rc.acceptance,
364 }
365 }
366}
367
368fn test_view(t: &DraftTest, content: &str) -> OracleTest {
369 OracleTest {
370 path: t.path.clone(),
371 content: content.to_owned(),
372 scenarios: t.scenarios.clone(),
373 red_class: crate::pipeline::class_word(t.red_class),
374 red: t.red_class.is_red(),
375 authored_by: t.authored_by,
376 edited_by: t.edited_by.clone(),
377 }
378}
379
380#[must_use]
384pub fn view(
385 project_state: &Path,
386 unit: &str,
387 req: &OracleRequirement,
388 entry: Option<&OracleEntry>,
389 cfg: ViewConfig,
390) -> OracleView {
391 let digest = req.digest();
392 let mut v = OracleView {
393 req: req.key.clone(),
394 title: req.title.clone(),
395 scenarios: req.scenarios.clone(),
396 state: OracleState::None,
397 frozen_on: None,
398 frozen_by: None,
399 frozen_at: None,
400 tests: Vec::new(),
401 discarded: Vec::new(),
402 rejected: Vec::new(),
403 uncovered: req.scenarios.iter().map(|s| s.id.clone()).collect(),
404 untestable: Vec::new(),
405 ambiguities: Vec::new(),
406 held_out: HeldOutSummary {
407 state: if cfg.held_out { "absent" } else { "off" }.to_owned(),
408 count: 0,
409 manifest_sha256: None,
410 reason: None,
411 discarded: Vec::new(),
412 },
413 strength_target: cfg.strength_target,
414 calls: cfg.samples,
415 ambiguity_check: cfg.samples >= 2,
416 manifest_sha256: None,
417 error: (!cfg.acceptance)
418 .then(|| "acceptance tests are off ([acceptance] generate = false)".to_owned()),
419 };
420 let loaded = load(project_state, unit, &req.key);
421 let draft = match loaded {
422 Ok(d) => d,
423 Err(t) => {
424 v.state = OracleState::Failed;
425 v.error = Some(t.0);
426 None
427 }
428 };
429 let current = draft
430 .as_ref()
431 .filter(|(d, _)| d.requirement.sha256 == digest);
432 if let Some((d, contents)) = current {
433 let content = |p: &str| {
434 contents
435 .iter()
436 .find(|(q, _)| q == p)
437 .map(|(_, c)| c.as_str())
438 .unwrap_or_default()
439 };
440 v.tests = d
441 .tests
442 .iter()
443 .map(|t| test_view(t, content(&t.path)))
444 .collect();
445 v.discarded.clone_from(&d.discarded);
446 v.rejected.clone_from(&d.rejected);
447 v.untestable.clone_from(&d.untestable.sample1);
448 v.ambiguities.clone_from(&d.ambiguities);
449 v.held_out = if cfg.held_out {
450 d.held_out.clone()
451 } else {
452 HeldOutSummary {
453 state: "off".to_owned(),
454 count: 0,
455 manifest_sha256: None,
456 reason: None,
457 discarded: Vec::new(),
458 }
459 };
460 v.manifest_sha256 = d.manifest();
461 let covered: Vec<&String> = d.red().flat_map(|t| t.scenarios.iter()).collect();
462 v.uncovered.retain(|s| !covered.contains(&s));
463 v.state = if d.open_ambiguities().next().is_some() {
464 OracleState::Ambiguous
465 } else if d.red().next().is_none() {
466 OracleState::NotRed
467 } else {
468 OracleState::Written
469 };
470 } else if draft.is_some() {
471 v.error = Some("written for an earlier wording: write them again".to_owned());
472 }
473 if let Some(e) = entry {
474 if e.requirement_sha256 == digest {
475 v.frozen_by = Some(e.by.clone());
476 v.frozen_at = Some(e.at);
477 v.state = match crate::pre_frozen::load(project_state, unit, &req.key) {
478 Ok(Some((rec, files))) => {
479 let drafted = current.map(|(d, _)| d);
482 v.tests = rec
483 .visible
484 .iter()
485 .map(|t| {
486 let same = drafted.and_then(|d| {
487 d.tests
488 .iter()
489 .find(|x| x.path == t.path && x.sha256 == t.sha256)
490 });
491 OracleTest {
492 path: t.path.clone(),
493 content: files
494 .iter()
495 .find(|(p, _)| *p == t.path)
496 .map(|(_, c)| c.clone())
497 .unwrap_or_default(),
498 scenarios: t.cites.clone(),
499 red_class: same.map_or_else(
500 || crate::pipeline::class_word(RedClass::Assertion),
501 |x| crate::pipeline::class_word(x.red_class),
502 ),
503 red: true,
504 authored_by: same.map_or(AuthoredBy::Model, |x| x.authored_by),
505 edited_by: same.and_then(|x| x.edited_by.clone()),
506 }
507 })
508 .collect();
509 v.manifest_sha256 = Some(e.manifest_sha256.clone());
510 let covered: Vec<&String> =
511 rec.visible.iter().flat_map(|t| t.cites.iter()).collect();
512 v.uncovered = req
513 .scenarios
514 .iter()
515 .map(|s| s.id.clone())
516 .filter(|s| !covered.contains(&s))
517 .collect();
518 if cfg.held_out {
519 v.held_out = match &rec.held_out_sealed {
520 Some(h) => HeldOutSummary {
521 state: "kept".to_owned(),
522 count: h.count,
523 manifest_sha256: Some(format!("sha256:{}", h.manifest_sha256)),
524 reason: None,
525 discarded: Vec::new(),
526 },
527 None => HeldOutSummary {
528 state: "absent".to_owned(),
529 count: 0,
530 manifest_sha256: None,
531 reason: Some("none_kept".to_owned()),
532 discarded: Vec::new(),
533 },
534 };
535 }
536 OracleState::Frozen
537 }
538 Ok(None) => OracleState::FrozenElsewhere,
539 Err(t) => {
540 v.error = Some(format!("the frozen set was tampered with: {}", t.0));
541 OracleState::Frozen
542 }
543 };
544 } else if current.is_none() {
545 v.state = OracleState::Lapsed;
546 v.frozen_on = load_frozen(project_state, unit, &req.key).map(|f| f.wording);
547 v.frozen_by = Some(e.by.clone());
548 v.frozen_at = Some(e.at);
549 }
550 } else if let Ok(Some((rec, _))) = crate::pre_frozen::load(project_state, unit, &req.key)
551 && rec.producer == Producer::Run
552 && rec.is_current(&digest)
553 {
554 v.state = OracleState::Frozen;
556 v.frozen_by = Some(rec.decided.by.clone());
557 v.frozen_at = Some(rec.decided.at);
558 }
559 v
560}
561
562pub fn open_ambiguities(
569 project_state: &Path,
570 unit: &str,
571 requirements: &[OracleRequirement],
572) -> Result<Vec<(String, String)>, Tampered> {
573 let mut out = Vec::new();
574 for r in requirements {
575 let Some((d, _)) = load(project_state, unit, &r.key)? else {
576 continue;
577 };
578 if d.requirement.sha256 != r.digest() {
579 continue;
580 }
581 out.extend(
582 d.open_ambiguities()
583 .map(|a| (d.requirement.key.clone(), a.scenario.clone())),
584 );
585 }
586 Ok(out)
587}
588
589#[must_use]
591pub fn rung(format: &dyn Adapter, root: &Path, unit: &str) -> Rung {
592 speccoder_core::ladder::rung_of(unit, &format.standing_units(root))
593}
594
595#[derive(Debug, Clone)]
597pub struct Request<'a> {
598 pub req: &'a str,
600 pub op: OracleOp,
602 pub path: Option<&'a str>,
604 pub scenario: Option<&'a str>,
606 pub reading: Option<Reading>,
608 pub text: Option<&'a str>,
610}
611
612impl Request<'_> {
613 #[must_use]
615 pub fn needs_model(&self) -> bool {
616 matches!(self.op, OracleOp::Write | OracleOp::Again)
617 || (self.op == OracleOp::Reading && self.reading == Some(Reading::B))
618 }
619}
620
621struct Ctx {
623 root: PathBuf,
624 feature: String,
625 dir: String,
626 format: &'static dyn Adapter,
627 snapshot: ProjectSnapshot,
628 plane_dir: PathBuf,
629 config: substrate::run_config::RunConfig,
630 sandbox: sandbox::Config,
631 worktree: crate::prepare::Config,
632 state: PathBuf,
634 policy: tools::policy::Policy,
635 model: Option<substrate::model_config::ModelConfig>,
636}
637
638fn refuse(status: CommandStatus, code: &str, message: &str) -> CommandResult {
639 CommandResult::new("review", status, message).with_code(code)
640}
641
642fn open(path: &Path, feature: Option<&str>, state: &Path) -> Result<Ctx, Box<CommandResult>> {
643 let inspected = crate::inspect(path, feature);
644 let Some(CommandData::Inspect(snapshot)) = inspected.data() else {
645 return Err(Box::new(CommandResult::carried("review", &inspected)));
646 };
647 let snapshot = (**snapshot).clone();
648 let Some(f) = snapshot.feature.as_ref() else {
649 return Err(Box::new(CommandResult::new(
650 "review",
651 CommandStatus::SelectionNeeded,
652 inspected.message(),
653 )));
654 };
655 let root = PathBuf::from(&snapshot.root);
656 let config_root = crate::verify::find_config(&root).ok_or_else(|| {
657 Box::new(refuse(
658 CommandStatus::ConfigError,
659 "CONFIG-MISSING",
660 "the project has no speccoder.toml",
661 ))
662 })?;
663 let text =
664 std::fs::read_to_string(config_root.join(crate::verify::config::FILE)).unwrap_or_default();
665 let config = substrate::run_config::parse(&text).map_err(|e| {
666 Box::new(refuse(
667 CommandStatus::ConfigError,
668 "CONFIG-INVALID",
669 &format!("speccoder.toml: {e}"),
670 ))
671 })?;
672 let sandbox = sandbox::parse(&text).map_err(|e| {
673 Box::new(refuse(
674 CommandStatus::ConfigError,
675 "CONFIG-INVALID",
676 &format!("speccoder.toml: {e}"),
677 ))
678 })?;
679 let worktree_config = crate::prepare::parse(&text).map_err(|e| {
680 Box::new(refuse(
681 CommandStatus::ConfigError,
682 "CONFIG-INVALID",
683 &format!("speccoder.toml: {e}"),
684 ))
685 })?;
686 let format = crate::format::of(&snapshot);
687 let policy =
688 tools::policy::Policy::new(&crate::verify::check_argvs(&config_root), &config.allow)
689 .with_prefixes(config.allow_prefix.clone())
690 .rules(format.rules());
691 let model = substrate::model_config::parse(&text).ok().flatten();
692 Ok(Ctx {
693 feature: f.feature.name.clone(),
694 dir: f.feature.dir.clone(),
695 plane_dir: worktree::project_state_dir(state, &root),
696 root,
697 format,
698 snapshot,
699 config,
700 sandbox,
701 worktree: worktree_config,
702 state: state.to_path_buf(),
703 policy,
704 model,
705 })
706}
707
708fn project_in(root: &Path, worktree_root: &Path) -> PathBuf {
710 let canonical = |p: &Path| std::fs::canonicalize(p).unwrap_or_else(|_| p.to_path_buf());
711 let top = crate::git::Git::at(root)
712 .toplevel()
713 .ok()
714 .flatten()
715 .map_or_else(|| root.to_path_buf(), |t| canonical(&t));
716 match canonical(root).strip_prefix(&top) {
717 Ok(rel) if !rel.as_os_str().is_empty() => worktree_root.join(rel),
718 _ => worktree_root.to_path_buf(),
719 }
720}
721
722pub(crate) struct UnitLock(PathBuf);
726
727impl UnitLock {
728 pub(crate) fn take(project_state: &Path, unit: &str) -> Result<Self, String> {
729 let path = project_state.join(DIR).join(format!("{unit}.lock"));
730 if let Some(parent) = path.parent() {
731 std::fs::create_dir_all(parent).map_err(|e| format!("{}: {e}", parent.display()))?;
732 }
733 if std::fs::metadata(&path)
735 .ok()
736 .and_then(|m| m.modified().ok())
737 .and_then(|m| m.elapsed().ok())
738 .is_some_and(|age| age.as_secs() > 3600)
739 {
740 let _ = std::fs::remove_file(&path);
741 }
742 std::fs::OpenOptions::new()
743 .write(true)
744 .create_new(true)
745 .open(&path)
746 .map(|_| Self(path))
747 .map_err(|_| {
748 format!("{unit}'s tests are being written or frozen elsewhere right now: look again when that ends")
749 })
750 }
751}
752
753impl Drop for UnitLock {
754 fn drop(&mut self) {
755 let _ = std::fs::remove_file(&self.0);
756 }
757}
758
759fn stage(options: &crate::RunOptions, req: &str, stage: &str, n: u32, of: u32) {
761 if let Some(p) = &options.progress {
762 p.event(&crate::progress::Event::Oracle {
763 step: crate::progress::OracleEvent::Stage {
764 requirement: req.to_owned(),
765 stage: stage.to_owned(),
766 n,
767 of,
768 },
769 });
770 }
771}
772
773#[must_use]
776pub fn tests(
777 path: &Path,
778 feature: Option<&str>,
779 request: &Request<'_>,
780 via: Via,
781 state: Option<&Path>,
782 options: &crate::RunOptions,
783) -> CommandResult {
784 let Some(state) = state.map(Path::to_path_buf).or_else(worktree::state_dir) else {
785 return refuse(
786 CommandStatus::ConfigError,
787 "NO-STATE-DIR",
788 "no state directory (set SPECCODER_STATE_DIR or HOME)",
789 );
790 };
791 if !request.needs_model() {
792 return tests_with(path, feature, request, via, &state, None, options);
793 }
794 let ctx = match open(path, feature, &state) {
795 Ok(c) => c,
796 Err(r) => return *r,
797 };
798 if let Err(r) = check(&ctx, request) {
799 return *r;
800 }
801 let Some(model) = ctx.model.as_ref() else {
802 return refuse(
803 CommandStatus::ConfigError,
804 "CONFIG",
805 "speccoder.toml has no [model] section: the test author needs a model",
806 );
807 };
808 match substrate::model_config::open_driver(model) {
809 Ok((provider, _)) => {
810 tests_with(path, feature, request, via, &state, Some(provider), options)
811 }
812 Err(e) => substrate::model_config::refused("review", &e),
813 }
814}
815
816#[must_use]
818pub fn tests_with(
819 path: &Path,
820 feature: Option<&str>,
821 request: &Request<'_>,
822 via: Via,
823 state: &Path,
824 provider: Option<Arc<dyn ModelProvider>>,
825 options: &crate::RunOptions,
826) -> CommandResult {
827 let ctx = match open(path, feature, state) {
828 Ok(c) => c,
829 Err(r) => return *r,
830 };
831 let req = match check(&ctx, request) {
832 Ok(r) => r,
833 Err(r) => return *r,
834 };
835 let _lock = match UnitLock::take(&ctx.plane_dir, &ctx.feature) {
836 Ok(l) => l,
837 Err(e) => return refuse(CommandStatus::GitPrecondition, "RUN-ACTIVE", &e),
838 };
839 let who = crate::approvals::Who::local_in(&ctx.root, via);
840 let done = match request.op {
841 OracleOp::Write => write(&ctx, &req, provider, options),
842 OracleOp::Reject => reject(&ctx, &req, request, &who.name),
843 OracleOp::Again => again(&ctx, &req, request, &who.name, provider, options),
844 OracleOp::Edit => edit(&ctx, &req, request, &who.name),
845 OracleOp::Reading => reading(&ctx, &req, request, &who.name, provider, options),
846 };
847 match done {
848 Ok(message) => {
849 let r = crate::review::report_in(path, feature, Some(state));
850 let status = r.status();
851 r.reworded(status, &message)
852 }
853 Err(r) => *r,
854 }
855}
856
857fn check(ctx: &Ctx, request: &Request<'_>) -> Result<OracleRequirement, Box<CommandResult>> {
860 let unit_rung = rung(ctx.format, &ctx.root, &ctx.feature);
861 if unit_rung != Rung::Feature {
862 return Err(Box::new(
863 refuse(
864 CommandStatus::UsageError,
865 "REVIEW-TESTS-OWN-PIPELINE",
866 &format!(
867 "{} is a {} unit: its tests are written and confirmed by its own pipeline",
868 ctx.feature,
869 if unit_rung == Rung::Quick {
870 "quick-change"
871 } else {
872 "bug"
873 }
874 ),
875 )
876 .with_fix(vec![if unit_rung == Rung::Quick {
877 "speccoder do \"…\"".to_owned()
878 } else {
879 format!("speccoder bug confirm {}/<task>", ctx.feature)
880 }]),
881 ));
882 }
883 if !ctx.config.acceptance {
884 return Err(Box::new(refuse(
885 CommandStatus::ConfigError,
886 "ACCEPTANCE-OFF",
887 "acceptance tests are off ([acceptance] generate = false): no tests are written at review",
888 )));
889 }
890 let requirements = ctx.format.oracle_requirements(&ctx.snapshot);
891 let Some(req) = requirements
892 .iter()
893 .find(|r| r.key.eq_ignore_ascii_case(request.req))
894 .cloned()
895 else {
896 return Err(Box::new(refuse(
897 CommandStatus::UsageError,
898 "REVIEW-NO-SCENARIOS",
899 &format!(
900 "the spec has no requirement {}; it names {}",
901 request.req,
902 if requirements.is_empty() {
903 "none with scenarios".to_owned()
904 } else {
905 requirements
906 .iter()
907 .map(|r| r.key.as_str())
908 .collect::<Vec<_>>()
909 .join(", ")
910 }
911 ),
912 )));
913 };
914 if req.scenarios.is_empty() {
915 return Err(Box::new(refuse(
916 CommandStatus::UsageError,
917 "REVIEW-NO-SCENARIOS",
918 &format!(
919 "{} has no acceptance scenarios to write tests from",
920 req.key
921 ),
922 )));
923 }
924 let (entries, _) = crate::review::oracle_entries(&ctx.root, &ctx.dir)
927 .map_err(|e| Box::new(refuse(CommandStatus::NeedsHuman, "REVIEW-UNREADABLE", &e)))?;
928 if let Some(e) = entries
929 .get(&req.key)
930 .filter(|e| e.requirement_sha256 == req.digest())
931 {
932 return Err(Box::new(refuse(
933 CommandStatus::UsageError,
934 "ORACLE-FROZEN",
935 &format!(
936 "{}'s tests were frozen with the approval by {}: the run uses them as frozen — reword {} to lapse the freeze, then write its tests again",
937 req.key, e.by, req.key
938 ),
939 )));
940 }
941 Ok(req)
942}
943
944type Outcome = Result<String, Box<CommandResult>>;
945
946fn failed(message: impl Into<String>) -> Box<CommandResult> {
947 Box::new(refuse(
948 CommandStatus::RunFailed,
949 "ORACLE-AUTHOR-FAILED",
950 &message.into(),
951 ))
952}
953
954fn usage(message: &str) -> Box<CommandResult> {
955 Box::new(refuse(CommandStatus::UsageError, "USAGE", message))
956}
957
958fn existing(ctx: &Ctx, req: &OracleRequirement) -> Result<Loaded, Box<CommandResult>> {
959 match load(&ctx.plane_dir, &ctx.feature, &req.key) {
960 Ok(Some((d, c))) if d.requirement.sha256 == req.digest() => Ok((d, c)),
961 Ok(Some(_)) => Err(usage(&format!(
962 "{}'s tests were written for an earlier wording: write them again (`speccoder review tests {} --write`)",
963 req.key, req.key
964 ))),
965 Ok(None) => Err(usage(&format!(
966 "{} has no tests yet: `speccoder review tests {} --write`",
967 req.key, req.key
968 ))),
969 Err(t) => Err(Box::new(refuse(
970 CommandStatus::NeedsHuman,
971 "ORACLE-TAMPERED",
972 &t.0,
973 ))),
974 }
975}
976
977enum Held<'a, T> {
980 Own(T),
981 Lent(&'a T),
982}
983
984impl<T> std::ops::Deref for Held<'_, T> {
985 type Target = T;
986 fn deref(&self) -> &T {
987 match self {
988 Self::Own(t) => t,
989 Self::Lent(t) => t,
990 }
991 }
992}
993
994struct Author<'a> {
995 ctx: &'a Ctx,
996 rt: Held<'a, tokio::runtime::Runtime>,
997 plane: Held<'a, substrate::Plane>,
998 bundle: Value,
999 instructions: Vec<crate::instructions::Instruction>,
1000 test_layout: String,
1002 spent: std::cell::RefCell<Option<Value>>,
1004}
1005
1006impl<'a> Author<'a> {
1007 fn open(
1008 ctx: &'a Ctx,
1009 req: &OracleRequirement,
1010 provider: Option<Arc<dyn ModelProvider>>,
1011 options: &crate::RunOptions,
1012 ) -> Result<Self, Box<CommandResult>> {
1013 let (Some(provider), Some(model)) = (provider, ctx.model.as_ref()) else {
1014 return Err(Box::new(refuse(
1015 CommandStatus::ConfigError,
1016 "CONFIG",
1017 "the test author needs a model ([model] in speccoder.toml)",
1018 )));
1019 };
1020 let rt = crate::pipeline::tokio_runtime().map_err(failed)?;
1021 let plane = rt
1022 .block_on(async {
1023 substrate::Plane::open_with(
1024 &ctx.plane_dir,
1025 model,
1026 provider,
1027 None,
1028 substrate::Hooks {
1029 format: Some(ctx.format),
1030 progress: options.progress.clone(),
1031 ..substrate::Hooks::default()
1032 },
1033 )
1034 })
1035 .map_err(|e| failed(e.0))?;
1036 Ok(Self::with(ctx, req, Held::Own(rt), Held::Own(plane)))
1037 }
1038
1039 fn lent(
1041 ctx: &'a Ctx,
1042 req: &OracleRequirement,
1043 rt: &'a tokio::runtime::Runtime,
1044 plane: &'a substrate::Plane,
1045 ) -> Self {
1046 Self::with(ctx, req, Held::Lent(rt), Held::Lent(plane))
1047 }
1048
1049 fn with(
1050 ctx: &'a Ctx,
1051 req: &OracleRequirement,
1052 rt: Held<'a, tokio::runtime::Runtime>,
1053 plane: Held<'a, substrate::Plane>,
1054 ) -> Self {
1055 let unit_dir = ctx.root.join(&ctx.dir);
1056 let [constitution, _, plan, _] = ctx.format.convergence_texts(&ctx.root, &unit_dir);
1057 let bundle = json!({
1058 "feature": ctx.feature,
1059 "constitution": constitution,
1060 "plan": plan,
1061 "story": req.text,
1062 "task": {
1063 "id": format!("req-{}", req.key),
1064 "title": format!("Acceptance tests for {}: {}", req.key, req.title),
1065 "phase": "Review",
1066 "subsection": "",
1067 "story": req.key,
1068 "parallel": false,
1069 },
1070 "dependencies": [],
1071 });
1072 let instructions =
1073 crate::instructions::collect(&ctx.root, &ctx.root, &[], &ctx.config.instructions);
1074 Self {
1075 ctx,
1076 rt,
1077 plane,
1078 bundle,
1079 instructions,
1080 test_layout: crate::layout::text_at(&ctx.root),
1081 spent: std::cell::RefCell::new(None),
1082 }
1083 }
1084
1085 fn context(
1087 &self,
1088 req: &OracleRequirement,
1089 notes: &[String],
1090 ) -> speccoder_core::context::ContextManifest {
1091 let mut m = crate::context::Manifest::new();
1092 m.push(
1093 ContextKind::Story,
1094 "person",
1095 "/requirement",
1096 req.text.as_bytes(),
1097 );
1098 for s in &req.scenarios {
1099 m.push(
1100 ContextKind::Scenario,
1101 "person",
1102 &format!("/scenarios/{}", s.id),
1103 s.text.as_bytes(),
1104 );
1105 }
1106 m.push(
1107 ContextKind::Constitution,
1108 "worktree",
1109 "/bundle/constitution",
1110 self.bundle["constitution"]
1111 .as_str()
1112 .unwrap_or_default()
1113 .as_bytes(),
1114 );
1115 m.push(
1116 ContextKind::Plan,
1117 "worktree",
1118 "/bundle/plan",
1119 self.bundle["plan"].as_str().unwrap_or_default().as_bytes(),
1120 );
1121 m.push(
1122 ContextKind::TestLayout,
1123 "worktree",
1124 "/test_layout",
1125 self.test_layout.as_bytes(),
1126 );
1127 for (i, n) in notes.iter().enumerate() {
1128 let kind = if n.starts_with("Rejected `") {
1129 ContextKind::Rejection
1130 } else if n.starts_with("Comment ") {
1131 ContextKind::Comment
1132 } else {
1133 ContextKind::Note
1134 };
1135 m.push(kind, "person", &format!("/notes/{i}"), n.as_bytes());
1136 }
1137 m.finish()
1138 }
1139
1140 #[allow(clippy::too_many_arguments)]
1142 fn call(
1143 &self,
1144 req: &OracleRequirement,
1145 files: &[String],
1146 manifest: &str,
1147 notes: &[String],
1148 refused: &[String],
1149 again: Option<(&str, &str)>,
1150 sample: u8,
1151 ) -> Result<(Value, String, Vec<ContextEntry>), String> {
1152 let context = self.context(req, notes);
1153 let suite = crate::pipeline::acceptance::suite(&self.ctx.root, &req.text, &[]);
1154 let interface = crate::pipeline::acceptance::interface(&self.ctx.root, &req.text);
1155 let commands: Vec<String> = if crate::layout::via_command(&req.text) {
1156 crate::layout::commands(&self.ctx.root)
1157 .into_iter()
1158 .map(|c| c.shown)
1159 .collect()
1160 } else {
1161 Vec::new()
1162 };
1163 let input = substrate::acceptance::review_input(&substrate::acceptance::ReviewParts {
1164 bundle: self.bundle.clone(),
1165 requirement: json!({ "key": req.key, "title": req.title, "text": req.text }),
1166 scenarios: &req.scenarios,
1167 files,
1168 manifest,
1169 notes,
1170 again,
1171 instructions: &self.instructions,
1172 context: &context,
1173 sample,
1174 test_layout: Some(&self.test_layout),
1175 suite: &suite,
1176 interface: &interface,
1177 commands: &commands,
1178 refused,
1179 });
1180 let task = format!("req-{}", req.key);
1181 let outcome = self
1182 .rt
1183 .block_on(self.plane.acceptance(&self.ctx.feature, &task, input))
1184 .map_err(|e| format!("the test author failed: {}", e.0))?;
1185 let output = outcome
1186 .output
1187 .as_ref()
1188 .map(|o| o.peek().clone())
1189 .unwrap_or_default();
1190 let answer = output["answer"].clone();
1191 if !matches!(outcome.status, agentplane::prelude::RunStatus::Succeeded) || answer.is_null()
1192 {
1193 let status = format!("{:?}", outcome.status);
1194 return Err(substrate::provider_refusal(&status).map_or_else(
1195 || {
1196 format!(
1197 "the test author did not answer: {}",
1198 substrate::why_not(&outcome.status)
1199 )
1200 },
1201 |r| format!("the model provider refused the request: {r}"),
1202 ));
1203 }
1204 let total =
1205 crate::pipeline::commit::add_usage(self.spent.borrow().as_ref(), output.get("usage"));
1206 *self.spent.borrow_mut() = total;
1207 Ok((answer, outcome.run_id.to_string(), context.entries))
1208 }
1209}
1210
1211fn expectations(
1217 answer: &Value,
1218 req: &OracleRequirement,
1219) -> Result<(Vec<Expectation>, Vec<Untestable>), String> {
1220 let canonical = |id: &str| {
1221 req.scenarios
1222 .iter()
1223 .find(|s| s.id.eq_ignore_ascii_case(id.trim()))
1224 .map(|s| s.id.clone())
1225 .ok_or_else(|| {
1226 format!(
1227 "the test author's expectations name `{id}`, which {} does not have",
1228 req.key
1229 )
1230 })
1231 };
1232 let list = |field: &str| -> Result<Value, String> {
1233 match &answer[field] {
1234 Value::Null => Ok(Value::Array(Vec::new())),
1235 v @ Value::Array(_) => Ok(v.clone()),
1236 _ => Err(format!("the test author's `{field}` is not a list")),
1237 }
1238 };
1239 let mut e: Vec<Expectation> = serde_json::from_value(list("expectations")?)
1240 .map_err(|x| format!("the test author's expectations cannot be read: {x}"))?;
1241 let mut u: Vec<Untestable> = serde_json::from_value(list("untestable")?)
1242 .map_err(|x| format!("the test author's untestable list cannot be read: {x}"))?;
1243 for x in &mut e {
1244 x.scenario = canonical(&x.scenario)?;
1245 }
1246 for x in &mut u {
1247 x.scenario = canonical(&x.scenario)?;
1248 }
1249 if e.is_empty() && u.is_empty() {
1250 return Err(
1251 "the test author stated no expectation for any scenario: the readings cannot be compared"
1252 .to_owned(),
1253 );
1254 }
1255 Ok((e, u))
1256}
1257
1258struct Bench {
1261 root: PathBuf,
1264 _scratch: Option<worktree::Scratch>,
1265 project: PathBuf,
1266 runner: sandbox::Runner,
1267 tmp: PathBuf,
1268}
1269
1270impl Bench {
1271 fn open(ctx: &Ctx) -> Result<Self, Box<CommandResult>> {
1272 let tmp = ctx.plane_dir.join("tmp");
1273 let scratch = worktree::scratch(&ctx.root, &tmp)
1274 .map_err(|e| failed(format!("the scratch worktree could not be made: {}", e.0)))?;
1275 let project = project_in(&ctx.root, &scratch.path);
1276 let (runner, notice) = sandbox::runner(&scratch.path, &ctx.sandbox)
1277 .map_err(|e| Box::new(refuse(CommandStatus::ConfigError, "CONFIG", &e)))?;
1278 if !runner.is_sandboxed() {
1281 return Err(Box::new(
1282 refuse(
1283 CommandStatus::ConfigError,
1284 "SANDBOX-UNAVAILABLE",
1285 &format!(
1286 "the red check runs model-written tests and needs the OS sandbox: {}",
1287 notice.unwrap_or_else(|| "commands would run unsandboxed".to_owned())
1288 ),
1289 )
1290 .with_fix(vec![
1291 "set [sandbox] mode = \"auto\" on a machine with a sandbox backend (speccoder doctor)".to_owned(),
1292 ]),
1293 ));
1294 }
1295 crate::include::copy(&scratch.path, &project, &ctx.root, &ctx.worktree.include).map_err(
1297 |e| {
1298 Box::new(refuse(
1299 CommandStatus::ConfigError,
1300 "PREPARE-FAILED",
1301 &format!("the files [worktree] include names could not be copied: {e}"),
1302 ))
1303 },
1304 )?;
1305 if ctx.worktree.prepare && crate::prepare::needed(&project) {
1306 crate::prepare::prepare(&crate::prepare::Prepare {
1307 worktree: &scratch.path,
1308 project: &project,
1309 config: &ctx.worktree,
1310 runner: &runner,
1311 state: Some(&ctx.state),
1312 })
1313 .map_err(|f| {
1314 Box::new(
1315 refuse(CommandStatus::ConfigError, "PREPARE-FAILED", &f.message)
1316 .with_fix(f.fix)
1317 .with_commands(f.commands),
1318 )
1319 })?;
1320 }
1321 Ok(Self {
1322 root: scratch.path.clone(),
1323 _scratch: Some(scratch),
1324 project,
1325 runner,
1326 tmp,
1327 })
1328 }
1329
1330 fn here(ctx: &Ctx, wt_root: &Path, runner: sandbox::Runner) -> Self {
1333 Self {
1334 root: wt_root.to_path_buf(),
1335 _scratch: None,
1336 project: ctx.root.clone(),
1337 runner,
1338 tmp: ctx.plane_dir.join("tmp"),
1339 }
1340 }
1341
1342 fn files(&self) -> (Vec<String>, String) {
1343 let files: Vec<String> = worktree::checkpoint(&self.root, &self.tmp)
1344 .and_then(|c| worktree::files(&self.root, &c))
1345 .unwrap_or_default()
1346 .into_iter()
1347 .take(400)
1348 .collect();
1349 let manifest = crate::pipeline::MANIFESTS
1350 .iter()
1351 .find_map(|m| {
1352 std::fs::read_to_string(self.project.join(m))
1353 .ok()
1354 .map(|t| format!("{m}:\n{t}"))
1355 })
1356 .map(|t| t.chars().take(8192).collect::<String>())
1357 .unwrap_or_default();
1358 (files, manifest)
1359 }
1360
1361 fn red(
1367 &self,
1368 tests: &[(String, String, Vec<String>)],
1369 ) -> Result<Vec<Result<RedClass, &'static str>>, String> {
1370 let mut out = Vec::new();
1371 for (path, content, argv) in tests {
1372 let full = self.project.join(path);
1373 let _ = std::fs::remove_file(&full);
1374 let run = || crate::pipeline::acceptance_run(&self.project, argv, &self.runner);
1375 let (bare, summary) = run();
1376 if classify_red(&bare, summary.as_ref()).is_red() {
1377 out.push(Err("red_without_it"));
1378 continue;
1379 }
1380 if let Some(parent) = full.parent() {
1381 std::fs::create_dir_all(parent).map_err(|e| format!("{path}: {e}"))?;
1382 }
1383 std::fs::write(&full, content).map_err(|e| format!("{path}: {e}"))?;
1384 let (first, summary) = run();
1385 let class = classify_red(&first, summary.as_ref());
1386 let class = if !class.is_red() {
1387 class
1388 } else {
1389 let (second, summary) = run();
1390 if classify_red(&second, summary.as_ref()).is_red() {
1391 class
1392 } else {
1393 RedClass::Flaky
1394 }
1395 };
1396 std::fs::remove_file(&full).map_err(|e| format!("{path}: {e}"))?;
1397 out.push(Ok(class));
1398 }
1399 Ok(out)
1400 }
1401}
1402
1403fn keep(
1405 ctx: &Ctx,
1406 bench: &Bench,
1407 answer: &Value,
1408 req: &OracleRequirement,
1409 taken: &[(String, Vec<String>)],
1410) -> (Vec<Kept>, Vec<OracleDiscarded>) {
1411 let (valid, refused) =
1412 crate::pipeline::validate_answer(answer, &bench.project, &ctx.policy, &req.scenarios, None);
1413 let mut discarded: Vec<OracleDiscarded> = refused
1414 .into_iter()
1415 .map(|r| OracleDiscarded {
1416 path: r.path,
1417 why: r.why.to_owned(),
1418 })
1419 .collect();
1420 let set_argv: Vec<String> = crate::pipeline::acceptance::argv_of(&answer["argv"]);
1421 let layout = crate::layout::detect(&bench.project, &crate::verify::check_argvs(&ctx.root));
1422 let commands = crate::layout::commands(&bench.project);
1424 let via_command = !commands.is_empty() && crate::layout::via_command(&req.text);
1425 let mut kept = Vec::new();
1426 for f in valid {
1427 let mut discard = |why: &str| {
1428 discarded.push(OracleDiscarded {
1429 path: f.path.clone(),
1430 why: why.to_owned(),
1431 });
1432 };
1433 if f.cited.is_empty() {
1434 discard("cites_no_scenario");
1435 continue;
1436 }
1437 if via_command && !commands.iter().any(|c| c.run_by(&f.content)) {
1438 discard("no_command");
1439 continue;
1440 }
1441 if taken.iter().any(|(p, _)| *p == f.path)
1442 || kept.iter().any(|(p, ..): &(String, _, _, _)| *p == f.path)
1443 {
1444 discard("exists");
1445 continue;
1446 }
1447 let argv = if f.argv.is_empty() {
1448 set_argv.clone()
1449 } else {
1450 f.argv.clone()
1451 };
1452 if taken.iter().any(|(_, a)| *a == argv)
1455 || kept
1456 .iter()
1457 .any(|(.., a): &(_, _, _, Vec<String>)| *a == argv)
1458 {
1459 discard("command_shared");
1460 continue;
1461 }
1462 if argv.is_empty()
1463 || !crate::pipeline::test_command_allowed(
1464 &ctx.policy,
1465 Some(&layout),
1466 &argv,
1467 &[f.path.as_str()],
1468 )
1469 {
1470 discard("command_not_allowed");
1471 continue;
1472 }
1473 if kept.len() + taken.len() >= LIMIT {
1474 discard("limit");
1475 continue;
1476 }
1477 kept.push((f.path, f.content, f.cited, argv));
1478 }
1479 (kept, discarded)
1480}
1481
1482fn write(
1485 ctx: &Ctx,
1486 req: &OracleRequirement,
1487 provider: Option<Arc<dyn ModelProvider>>,
1488 options: &crate::RunOptions,
1489) -> Outcome {
1490 let samples = ctx.config.oracle_samples.clamp(1, 2);
1491 let author = Author::open(ctx, req, provider, options)?;
1492 let bench = Bench::open(ctx)?;
1493 let earlier = load(&ctx.plane_dir, &ctx.feature, &req.key).map_err(|t| {
1496 Box::new(
1497 refuse(
1498 CommandStatus::NeedsHuman,
1499 "ORACLE-TAMPERED",
1500 &format!(
1501 "{}: writing anew would lose the rejections and notes it holds",
1502 t.0
1503 ),
1504 )
1505 .with_fix(vec![format!(
1506 "→ restore it, or remove {} to start {}'s tests over",
1507 dir(&ctx.plane_dir, &ctx.feature, &req.key).display(),
1508 req.key
1509 )]),
1510 )
1511 })?;
1512 let (rejected, mut notes) = earlier
1513 .as_ref()
1514 .filter(|(d, _)| d.requirement.sha256 == req.digest())
1515 .map(|(d, _)| (d.rejected.clone(), d.notes.clone()))
1516 .unwrap_or_default();
1517 notes.dedup();
1518 let told: Vec<String> = rejected
1519 .iter()
1520 .map(|r| format!("Rejected `{}`: {}", r.path, r.reason))
1521 .chain(notes.iter().cloned())
1522 .collect();
1523 let (draft, contents) = draft(
1524 ctx,
1525 req,
1526 &author,
1527 &bench,
1528 samples,
1529 &told,
1530 &[],
1531 rejected,
1532 notes,
1533 true,
1534 options,
1535 )?;
1536 store(&ctx.plane_dir, &ctx.feature, &draft, &contents).map_err(failed)?;
1537 let red = draft.red().count();
1538 let open = draft.open_ambiguities().count();
1539 Ok(format!(
1540 "wrote {} test(s) for {}: {red} red{}{}",
1541 draft.tests.len(),
1542 req.key,
1543 if draft.held_out.state == "kept" {
1544 format!(", {} held out", draft.held_out.count)
1545 } else {
1546 String::new()
1547 },
1548 if open > 0 {
1549 format!(
1550 "; {} — {open} scenario(s) read two ways: choose a reading or reword (`speccoder review tests {} --reading A|B <scenario>`)",
1551 req.key, req.key
1552 )
1553 } else {
1554 String::new()
1555 }
1556 ))
1557}
1558
1559#[allow(clippy::too_many_arguments)]
1562fn draft(
1563 ctx: &Ctx,
1564 req: &OracleRequirement,
1565 author: &Author<'_>,
1566 bench: &Bench,
1567 samples: u8,
1568 told: &[String],
1569 refused: &[String],
1570 rejected: Vec<OracleRejected>,
1571 notes: Vec<String>,
1572 with_held_out: bool,
1573 options: &crate::RunOptions,
1574) -> Result<Loaded, Box<CommandResult>> {
1575 let (files, manifest) = bench.files();
1576 let of = u32::from(samples);
1577 stage(options, &req.key, "writing", 1, of);
1578 let (first, run1, mut context) = author
1579 .call(req, &files, &manifest, told, refused, None, 1)
1580 .map_err(failed)?;
1581 let read = |answer: &Value| {
1584 let got = expectations(answer, req);
1585 if samples >= 2 {
1586 got.map_err(failed)
1587 } else {
1588 Ok(got.unwrap_or_default())
1589 }
1590 };
1591 let (e1, u1) = read(&first)?;
1592 let mut runs = vec![run1];
1593 let (mut e2, mut u2) = (Vec::new(), Vec::new());
1594 if samples >= 2 {
1595 stage(options, &req.key, "writing", 2, of);
1596 let (second, run2, more) = author
1597 .call(req, &files, &manifest, told, refused, None, 2)
1598 .map_err(failed)?;
1599 (e2, u2) = read(&second)?;
1600 runs.push(run2);
1601 for c in more {
1602 if !context.contains(&c) {
1603 context.push(c);
1604 }
1605 }
1606 stage(options, &req.key, "comparing", 1, 1);
1607 }
1608 let ambiguities = if samples >= 2 {
1609 speccoder_core::oracle::compare(&e1, &u1, &e2, &u2)
1610 } else {
1611 Vec::new()
1612 };
1613 let (kept, mut discarded) = keep(ctx, bench, &first, req, &[]);
1614 stage(options, &req.key, "checking_red", 1, 1);
1615 let plan: Vec<(String, String, Vec<String>)> = kept
1616 .iter()
1617 .map(|(p, c, _, a)| (p.clone(), c.clone(), a.clone()))
1618 .collect();
1619 let classes = bench.red(&plan).map_err(failed)?;
1620 let mut tests: Vec<DraftTest> = Vec::new();
1621 let mut contents: Vec<(String, String)> = Vec::new();
1622 for ((path, content, cited, argv), class) in kept.into_iter().zip(classes) {
1623 match class {
1624 Ok(red_class) => {
1625 tests.push(DraftTest {
1626 path: path.clone(),
1627 sha256: digest_of(&content),
1628 scenarios: cited,
1629 argv,
1630 red_class,
1631 authored_by: AuthoredBy::Model,
1632 edited_by: None,
1633 });
1634 contents.push((path, content));
1635 }
1636 Err(why) => discarded.push(OracleDiscarded {
1637 path,
1638 why: why.to_owned(),
1639 }),
1640 }
1641 }
1642 let (held_out, sealed) = if with_held_out {
1644 held_out(ctx, author, bench, req, &tests, options)
1645 } else {
1646 (
1647 HeldOutSummary {
1648 state: "off".to_owned(),
1649 count: 0,
1650 manifest_sha256: None,
1651 reason: None,
1652 discarded: Vec::new(),
1653 },
1654 None,
1655 )
1656 };
1657 let draft = Draft {
1658 requirement: DraftRequirement {
1659 key: req.key.clone(),
1660 title: req.title.clone(),
1661 text: req.text.clone(),
1662 sha256: req.digest(),
1663 },
1664 samples,
1665 tests,
1666 discarded,
1667 rejected,
1668 expectations: Samples {
1669 sample1: e1,
1670 sample2: e2,
1671 },
1672 untestable: Samples {
1673 sample1: u1,
1674 sample2: u2,
1675 },
1676 ambiguities,
1677 held_out,
1678 sealed,
1679 notes,
1680 runs,
1681 context,
1682 };
1683 Ok((draft, contents))
1684}
1685
1686fn held_out(
1690 ctx: &Ctx,
1691 author: &Author<'_>,
1692 bench: &Bench,
1693 req: &OracleRequirement,
1694 tests: &[DraftTest],
1695 options: &crate::RunOptions,
1696) -> (HeldOutSummary, Option<SealedRef>) {
1697 let summary = |state: &str, reason: Option<&str>| HeldOutSummary {
1698 state: state.to_owned(),
1699 count: 0,
1700 manifest_sha256: None,
1701 reason: reason.map(str::to_owned),
1702 discarded: Vec::new(),
1703 };
1704 if !ctx.config.held_out {
1705 return (summary("off", None), None);
1706 }
1707 let red: Vec<&DraftTest> = tests.iter().filter(|t| t.red_class.is_red()).collect();
1708 if red.is_empty() {
1709 return (summary("absent", Some("not_red")), None);
1710 }
1711 stage(options, &req.key, "held_out", 1, 1);
1712 let visible = crate::pipeline::PreparedAcceptance {
1713 report: json!({ "files": red.iter().map(|t| t.path.clone()).collect::<Vec<_>>() }),
1714 argv: red.first().map(|t| t.argv.clone()),
1715 more_argv: Vec::new(),
1716 red: true,
1717 red_class: Some(RedClass::Assertion),
1718 held_out: None,
1719 generation_failed: None,
1720 skip_cause: None,
1721 };
1722 let task = format!("req-{}", req.key);
1723 let store = crate::held_out::sealed_dir(&ctx.plane_dir, &ctx.feature, &req.key);
1724 let (report, sha) = crate::pipeline::prepare_held_out(
1725 &crate::pipeline::AcceptanceSetup {
1726 rt: &author.rt,
1727 plane: &author.plane,
1728 project: &bench.project,
1729 worktree: &bench.root,
1730 tmp: &bench.tmp,
1731 policy: &ctx.policy,
1732 feature: &ctx.feature,
1733 task: &task,
1734 bundle: author.bundle.clone(),
1735 scenarios: &req.scenarios,
1736 instructions: &author.instructions,
1737 runner: &bench.runner,
1738 layout: None,
1739 told: &[],
1740 related: &[],
1741 oracle: true,
1742 },
1743 &visible,
1744 &req.scenarios,
1745 &store,
1746 false,
1747 );
1748 let total =
1750 crate::pipeline::commit::add_usage(author.spent.borrow().as_ref(), report.get("usage"));
1751 *author.spent.borrow_mut() = total;
1752 match sha {
1753 Some(sha) => {
1754 let count = report["kept"]
1755 .as_u64()
1756 .and_then(|n| usize::try_from(n).ok())
1757 .unwrap_or(0);
1758 (
1759 HeldOutSummary {
1760 state: "kept".to_owned(),
1761 count,
1762 manifest_sha256: Some(format!("sha256:{sha}")),
1763 reason: None,
1764 discarded: Vec::new(),
1765 },
1766 Some(SealedRef {
1767 key: req.key.clone(),
1768 manifest_sha256: sha,
1769 count,
1770 }),
1771 )
1772 }
1773 None => (
1774 HeldOutSummary {
1775 discarded: report["discarded"]
1776 .as_array()
1777 .into_iter()
1778 .flatten()
1779 .filter_map(|d| d["reason"].as_str().map(str::to_owned))
1780 .collect(),
1781 ..summary(
1782 "absent",
1783 Some(report["reason"].as_str().unwrap_or("none_kept")),
1784 )
1785 },
1786 None,
1787 ),
1788 }
1789}
1790
1791fn reject(ctx: &Ctx, req: &OracleRequirement, r: &Request<'_>, by: &str) -> Outcome {
1792 let Some(path) = r.path else {
1793 return Err(usage("name the test to reject (--reject PATH)"));
1794 };
1795 let Some(reason) = r.text.map(str::trim).filter(|t| !t.is_empty()) else {
1796 return Err(usage(
1797 "a rejection needs a reason: why is this test wrong? (--reason \"…\")",
1798 ));
1799 };
1800 let (mut d, contents) = existing(ctx, req)?;
1801 let Some(at) = d.tests.iter().position(|t| t.path == path) else {
1802 return Err(usage(&format!("{} has no test {path}", req.key)));
1803 };
1804 d.tests.remove(at);
1805 d.rejected.push(OracleRejected {
1806 path: path.to_owned(),
1807 reason: reason.to_owned(),
1808 by: by.to_owned(),
1809 at: now(),
1810 });
1811 store(&ctx.plane_dir, &ctx.feature, &d, &contents).map_err(failed)?;
1812 Ok(format!(
1813 "rejected {path} of {}: \"{reason}\" — later author calls for {} are told",
1814 req.key, req.key
1815 ))
1816}
1817
1818fn edit(ctx: &Ctx, req: &OracleRequirement, r: &Request<'_>, by: &str) -> Outcome {
1819 let Some(path) = r.path else {
1820 return Err(usage("name the test to edit (--edit PATH)"));
1821 };
1822 let Some(content) = r.text.filter(|t| !t.trim().is_empty()) else {
1823 return Err(usage("an edit needs the test's new content (--from FILE)"));
1824 };
1825 let (mut d, mut contents) = existing(ctx, req)?;
1826 let Some(at) = d.tests.iter().position(|t| t.path == path) else {
1827 return Err(usage(&format!("{} has no test {path}", req.key)));
1828 };
1829 let bench = Bench::open(ctx)?;
1830 let plan = [(
1831 path.to_owned(),
1832 content.to_owned(),
1833 d.tests[at].argv.clone(),
1834 )];
1835 let checked = bench
1836 .red(&plan)
1837 .map_err(failed)?
1838 .pop()
1839 .unwrap_or(Ok(RedClass::Unclassified));
1840 let class = checked.unwrap_or(RedClass::Unclassified);
1842 let t = &mut d.tests[at];
1843 t.sha256 = digest_of(content);
1844 t.red_class = class;
1845 t.authored_by = AuthoredBy::Person;
1846 t.edited_by = Some(by.to_owned());
1847 if let Some(c) = contents.iter_mut().find(|(p, _)| p == path) {
1848 c.1 = content.to_owned();
1849 }
1850 store(&ctx.plane_dir, &ctx.feature, &d, &contents).map_err(failed)?;
1851 Ok(format!(
1852 "edited {path} of {}: {}",
1853 req.key,
1854 if class.is_red() {
1855 format!("red ({})", crate::pipeline::class_word(class))
1856 } else if checked.is_err() {
1857 "its command is red without it — it decides nothing about this test, which cannot be frozen".to_owned()
1858 } else {
1859 format!(
1860 "not red ({}) — no evidence, it cannot be frozen",
1861 crate::pipeline::class_word(class)
1862 )
1863 }
1864 ))
1865}
1866
1867#[allow(clippy::too_many_arguments)]
1870fn rewrite(
1871 ctx: &Ctx,
1872 req: &OracleRequirement,
1873 mut d: Draft,
1874 mut contents: Vec<(String, String)>,
1875 replaced: &[String],
1876 note: String,
1877 again: Option<(&str, &str)>,
1878 only: Option<&str>,
1879 provider: Option<Arc<dyn ModelProvider>>,
1880 options: &crate::RunOptions,
1881) -> Result<(Draft, Vec<String>), Box<CommandResult>> {
1882 let author = Author::open(ctx, req, provider, options)?;
1883 let bench = Bench::open(ctx)?;
1884 let (files, manifest) = bench.files();
1885 d.notes.push(note);
1886 let told = d.all_notes();
1887 stage(options, &req.key, "writing", 1, 1);
1888 let (answer, run, context) = author
1889 .call(req, &files, &manifest, &told, &[], again, 1)
1890 .map_err(failed)?;
1891 d.runs.push(run);
1892 for c in context {
1893 if !d.context.contains(&c) {
1894 d.context.push(c);
1895 }
1896 }
1897 let keep_paths: Vec<(String, Vec<String>)> = d
1898 .tests
1899 .iter()
1900 .filter(|t| !replaced.contains(&t.path))
1901 .map(|t| (t.path.clone(), t.argv.clone()))
1902 .collect();
1903 let mut answer = answer;
1905 if let Some(files) = answer["files"].as_array_mut() {
1906 files.retain(|f| {
1907 only.is_none_or(|s| {
1908 f["scenarios"]
1909 .as_array()
1910 .is_some_and(|c| c.iter().any(|v| v.as_str() == Some(s)))
1911 })
1912 });
1913 }
1914 let (new, discarded) = keep(ctx, &bench, &answer, req, &keep_paths);
1915 if new.is_empty() {
1916 return Err(failed(format!(
1917 "the test author wrote no usable test{}",
1918 discarded
1919 .first()
1920 .map(|x| format!(" ({} {})", x.path, x.why))
1921 .unwrap_or_default()
1922 )));
1923 }
1924 d.tests.retain(|t| !replaced.contains(&t.path));
1925 contents.retain(|(p, _)| !replaced.contains(p));
1926 d.discarded.extend(discarded);
1927 stage(options, &req.key, "checking_red", 1, 1);
1928 let plan: Vec<(String, String, Vec<String>)> = new
1930 .iter()
1931 .map(|(p, c, _, a)| (p.clone(), c.clone(), a.clone()))
1932 .collect();
1933 let classes = bench.red(&plan).map_err(failed)?;
1934 let mut written = Vec::new();
1935 for ((path, content, cited, argv), class) in new.into_iter().zip(classes) {
1936 let red_class = match class {
1937 Ok(c) => c,
1938 Err(why) => {
1939 d.discarded.push(OracleDiscarded {
1940 path,
1941 why: why.to_owned(),
1942 });
1943 continue;
1944 }
1945 };
1946 d.tests.push(DraftTest {
1947 sha256: digest_of(&content),
1948 scenarios: cited,
1949 argv,
1950 red_class,
1951 authored_by: AuthoredBy::Model,
1952 edited_by: None,
1953 path: path.clone(),
1954 });
1955 contents.push((path.clone(), content));
1956 written.push(path);
1957 }
1958 store(&ctx.plane_dir, &ctx.feature, &d, &contents).map_err(failed)?;
1959 Ok((d, written))
1960}
1961
1962fn again(
1963 ctx: &Ctx,
1964 req: &OracleRequirement,
1965 r: &Request<'_>,
1966 _by: &str,
1967 provider: Option<Arc<dyn ModelProvider>>,
1968 options: &crate::RunOptions,
1969) -> Outcome {
1970 let Some(path) = r.path else {
1971 return Err(usage("name the test to write again (--again PATH)"));
1972 };
1973 let comment = r
1974 .text
1975 .map(str::trim)
1976 .filter(|t| !t.is_empty())
1977 .unwrap_or("write it again");
1978 let (d, contents) = existing(ctx, req)?;
1979 let Some(content) = contents
1980 .iter()
1981 .find(|(p, _)| p == path)
1982 .map(|(_, c)| c.clone())
1983 else {
1984 return Err(usage(&format!("{} has no test {path}", req.key)));
1985 };
1986 let (d, written) = rewrite(
1987 ctx,
1988 req,
1989 d,
1990 contents,
1991 &[path.to_owned()],
1992 format!("Comment on `{path}`: {comment}"),
1993 Some((path, &content)),
1994 None,
1995 provider,
1996 options,
1997 )?;
1998 let red = d
1999 .tests
2000 .iter()
2001 .filter(|t| written.contains(&t.path) && t.red_class.is_red())
2002 .count();
2003 Ok(format!(
2004 "wrote {path} of {} again: {}",
2005 req.key,
2006 if written.is_empty() {
2007 "nothing kept — its command is red without it, so it decides nothing".to_owned()
2008 } else if red == written.len() {
2009 "red".to_owned()
2010 } else {
2011 "not red — no evidence, it cannot be frozen".to_owned()
2012 }
2013 ))
2014}
2015
2016fn reading(
2017 ctx: &Ctx,
2018 req: &OracleRequirement,
2019 r: &Request<'_>,
2020 by: &str,
2021 provider: Option<Arc<dyn ModelProvider>>,
2022 options: &crate::RunOptions,
2023) -> Outcome {
2024 let (Some(scenario), Some(choice)) = (r.scenario, r.reading) else {
2025 return Err(usage(
2026 "name the reading and the scenario (--reading A|B US1-2)",
2027 ));
2028 };
2029 let (mut d, contents) = existing(ctx, req)?;
2030 let Some(item) = d.ambiguities.iter_mut().find(|a| a.scenario == scenario) else {
2031 return Err(usage(&format!(
2032 "{} has no ambiguity on {scenario}",
2033 req.key
2034 )));
2035 };
2036 item.resolution = Some(Resolution {
2037 reading: choice,
2038 by: by.to_owned(),
2039 at: now(),
2040 });
2041 let meant = if choice == Reading::A {
2042 item.a.clone()
2043 } else {
2044 item.b.clone()
2045 };
2046 if choice == Reading::A {
2047 d.notes.push(format!(
2048 "For {scenario} the person means reading A: given {}, expect {}",
2049 meant.given, meant.expect
2050 ));
2051 store(&ctx.plane_dir, &ctx.feature, &d, &contents).map_err(failed)?;
2052 return Ok(format!(
2053 "{scenario} of {}: reading A is meant — the tests stay",
2054 req.key
2055 ));
2056 }
2057 let replaced: Vec<String> = d
2058 .tests
2059 .iter()
2060 .filter(|t| t.scenarios.iter().any(|s| s == scenario))
2061 .map(|t| t.path.clone())
2062 .collect();
2063 let (_, written) = rewrite(
2064 ctx,
2065 req,
2066 d,
2067 contents,
2068 &replaced,
2069 format!(
2070 "For {scenario} the person means reading B: given {}, expect {}",
2071 meant.given, meant.expect
2072 ),
2073 None,
2074 Some(scenario),
2075 provider,
2076 options,
2077 )?;
2078 Ok(format!(
2079 "{scenario} of {}: reading B is meant — wrote {} anew",
2080 req.key,
2081 written.join(", ")
2082 ))
2083}
2084
2085#[derive(Debug, Clone)]
2088pub struct Freezable {
2089 pub req: OracleRequirement,
2091 pub draft: Draft,
2093 pub contents: Vec<(String, String)>,
2095 pub manifest: String,
2097}
2098
2099pub fn freezable(
2102 project_state: &Path,
2103 unit: &str,
2104 requirements: &[OracleRequirement],
2105) -> Result<Vec<Freezable>, Tampered> {
2106 let mut out = Vec::new();
2107 for r in requirements {
2108 let Some((draft, contents)) = load(project_state, unit, &r.key)? else {
2109 continue;
2110 };
2111 if draft.requirement.sha256 != r.digest() || draft.open_ambiguities().next().is_some() {
2112 continue;
2113 }
2114 let Some(manifest) = draft.manifest() else {
2115 continue;
2116 };
2117 let red: Vec<String> = draft.red().map(|t| t.path.clone()).collect();
2118 out.push(Freezable {
2119 req: r.clone(),
2120 contents: contents
2121 .into_iter()
2122 .filter(|(p, _)| red.contains(p))
2123 .collect(),
2124 draft,
2125 manifest,
2126 });
2127 }
2128 Ok(out)
2129}
2130
2131pub fn freeze(
2135 project_state: &Path,
2136 unit: &str,
2137 f: &Freezable,
2138 approval_digest: &str,
2139 decided: Decided,
2140 producer: Producer,
2141) -> Result<PreFrozenAcceptance, String> {
2142 let red: Vec<&DraftTest> = f.draft.red().collect();
2143 freeze_tests(
2144 project_state,
2145 unit,
2146 f,
2147 &red,
2148 approval_digest,
2149 decided,
2150 producer,
2151 )
2152}
2153
2154fn freeze_tests(
2156 project_state: &Path,
2157 unit: &str,
2158 f: &Freezable,
2159 red: &[&DraftTest],
2160 approval_digest: &str,
2161 decided: Decided,
2162 producer: Producer,
2163) -> Result<PreFrozenAcceptance, String> {
2164 let sealed = match &f.draft.sealed {
2165 Some(s) => {
2166 crate::held_out::open_sealed(project_state, unit, s)
2167 .map_err(|t| format!("{}'s held-out set was tampered with: {}", f.req.key, t.0))?;
2168 Some(s.clone())
2169 }
2170 None => None,
2171 };
2172 let record = PreFrozenAcceptance {
2173 unit: unit.to_owned(),
2174 requirement: f.req.key.clone(),
2175 wording_digest: f.draft.requirement.sha256.clone(),
2176 approval_digest: approval_digest.to_owned(),
2177 visible: red
2178 .iter()
2179 .map(|t| VisibleTest {
2180 path: t.path.clone(),
2181 sha256: t.sha256.clone(),
2182 cites: t.scenarios.clone(),
2183 })
2184 .collect(),
2185 held_out_sealed: sealed,
2186 producer,
2187 decided,
2188 };
2189 crate::pre_frozen::store(project_state, &record, &f.contents)?;
2190 let frozen = Frozen {
2191 requirement_sha256: f.draft.requirement.sha256.clone(),
2192 wording: format!(
2193 "{}\n{}",
2194 f.draft.requirement.title, f.draft.requirement.text
2195 ),
2196 manifest_sha256: f.manifest.clone(),
2197 tests: red
2198 .iter()
2199 .map(|t| FrozenTest {
2200 path: t.path.clone(),
2201 sha256: t.sha256.clone(),
2202 argv: t.argv.clone(),
2203 scenarios: t.scenarios.clone(),
2204 })
2205 .collect(),
2206 };
2207 let text = serde_json::to_string_pretty(&frozen).map_err(|e| e.to_string())?;
2208 write_atomic(
2209 &dir(project_state, unit, &f.req.key).join("frozen.json"),
2210 &text,
2211 )?;
2212 Ok(record)
2213}
2214
2215fn discard_words(why: &str) -> &'static str {
2223 match why {
2224 "cites_no_scenario" => "cites no scenario of the list",
2225 "no_command" => {
2226 "never runs the project's command — the story's interface is the command, not a module behind it"
2227 }
2228 "command_shared" => {
2229 "shares its command with another test — each test needs the command that runs only it"
2230 }
2231 "command_not_allowed" => "gives a command the policy does not allow",
2232 "limit" => "is beyond the number of tests one answer may hold",
2233 "red_without_it" => "fails without its own file — its command runs other, failing tests",
2234 _ => crate::pipeline::acceptance::refusal_words(why),
2235 }
2236}
2237
2238fn not_kept(t: &DraftTest, many: bool) -> Option<String> {
2239 if many && t.scenarios.len() > 1 {
2240 return Some(format!(
2241 "it tests {} together — one file per scenario, so each turns green on its own",
2242 t.scenarios.join(", ")
2243 ));
2244 }
2245 (!usable(t)).then(|| {
2246 format!(
2247 "it did not fail on an assertion ({})",
2248 crate::pipeline::class_word(t.red_class)
2249 )
2250 })
2251}
2252
2253fn usable(t: &DraftTest) -> bool {
2254 match t.red_class {
2255 RedClass::Assertion | RedClass::Passed => true,
2256 RedClass::Absent => [".rs", ".go", ".java", ".kt", ".cs", ".swift"]
2257 .iter()
2258 .any(|e| t.path.ends_with(e)),
2259 _ => false,
2260 }
2261}
2262
2263#[derive(Debug)]
2267pub(crate) struct RunOracle {
2268 pub tests: Vec<FrozenTest>,
2271 pub contents: Vec<(String, String)>,
2273 pub contracts: Vec<Expectation>,
2275 pub untestable: Vec<Untestable>,
2277 pub usage: Option<Value>,
2279}
2280
2281#[allow(clippy::too_many_arguments)]
2288pub(crate) fn write_for_run(
2289 project: &Path,
2290 wt_root: &Path,
2291 feature: &str,
2292 state: &Path,
2293 story: &str,
2294 runner: sandbox::Runner,
2295 (rt, plane): (&tokio::runtime::Runtime, &substrate::Plane),
2296 (only, told): (&[String], &[String]),
2297 run: &str,
2298 options: &crate::RunOptions,
2299) -> Result<RunOracle, Box<CommandResult>> {
2300 let ctx = open(project, Some(feature), state)?;
2301 let requirements = ctx.format.oracle_requirements(&ctx.snapshot);
2302 let whole = (story == ctx.feature && !requirements.iter().any(|r| r.key == story)).then(|| {
2304 OracleRequirement {
2305 key: story.to_owned(),
2306 title: ctx.feature.clone(),
2307 text: requirements
2308 .iter()
2309 .map(|r| format!("{}: {}\n{}", r.key, r.title, r.text))
2310 .collect::<Vec<_>>()
2311 .join("\n\n"),
2312 scenarios: requirements
2313 .iter()
2314 .flat_map(|r| r.scenarios.clone())
2315 .collect(),
2316 }
2317 });
2318 let Some(mut req) = whole.or_else(|| requirements.into_iter().find(|r| r.key == story)) else {
2319 return Err(Box::new(refuse(
2320 CommandStatus::NeedsHuman,
2321 "ORACLE-UNWRITTEN",
2322 &format!("{story} has no scenarios to write tests for"),
2323 )));
2324 };
2325 if !only.is_empty() {
2327 req.scenarios.retain(|s| only.contains(&s.id));
2328 }
2329 let _lock = UnitLock::take(&ctx.plane_dir, &ctx.feature)
2330 .map_err(|e| Box::new(refuse(CommandStatus::GitPrecondition, "RUN-ACTIVE", &e)))?;
2331 let author = Author::lent(&ctx, &req, rt, plane);
2332 let many = req.scenarios.len() > 1;
2333 let bench = Bench::here(&ctx, wt_root, runner);
2334 let kept = |d: &Draft| -> Vec<String> {
2335 let mut covered: Vec<String> = d
2336 .tests
2337 .iter()
2338 .filter(|t| not_kept(t, many).is_none())
2339 .flat_map(|t| t.scenarios.iter().cloned())
2340 .collect();
2341 covered.sort();
2342 covered.dedup();
2343 covered
2344 };
2345 let pass = |refused: &[String]| {
2348 draft(
2349 &ctx,
2350 &req,
2351 &author,
2352 &bench,
2353 1,
2354 told,
2355 refused,
2356 Vec::new(),
2357 told.to_vec(),
2358 false,
2359 options,
2360 )
2361 };
2362 let (mut draft, mut contents) = pass(&[])?;
2363 let untestable: Vec<String> = draft
2367 .untestable
2368 .sample1
2369 .iter()
2370 .map(|u| u.scenario.clone())
2371 .collect();
2372 let broke =
2373 !draft.discarded.is_empty() || draft.tests.iter().any(|t| not_kept(t, many).is_some());
2374 let left: Vec<String> = req
2375 .scenarios
2376 .iter()
2377 .map(|s| s.id.clone())
2378 .filter(|id| !kept(&draft).contains(id) && !untestable.contains(id))
2379 .collect();
2380 let opted_out = draft.tests.is_empty() && draft.discarded.is_empty() && !untestable.is_empty();
2381 if (broke && (!left.is_empty() || kept(&draft).is_empty())) || opted_out {
2382 let again: Vec<String> = if left.is_empty() {
2383 req.scenarios.iter().map(|s| s.id.clone()).collect()
2384 } else {
2385 left
2386 };
2387 let refused: Vec<String> = draft
2388 .discarded
2389 .iter()
2390 .map(|d| format!("Not kept `{}`: it {}", d.path, discard_words(&d.why)))
2391 .chain(draft.tests.iter().filter_map(|t| {
2392 not_kept(t, many).map(|why| format!("Not kept `{}`: {why}", t.path))
2393 }))
2394 .chain(opted_out.then(|| {
2395 format!(
2396 "Called untestable: {} — the story's code not existing yet never makes a scenario untestable; test it through the story's interface",
2397 untestable.join(", ")
2398 )
2399 }))
2400 .chain(std::iter::once(format!(
2401 "Write tests again for {} — each must build and be collected without the code it tests existing yet: exercise the story through its interface, import the code under test inside each test",
2402 again.join(", ")
2403 )))
2404 .collect();
2405 let (second, more) = pass(&refused)?;
2406 if kept(&second).len() > kept(&draft).len() {
2407 (draft, contents) = (second, more);
2408 }
2409 }
2410 if only.is_empty() {
2412 let keep: Vec<DraftTest> = draft
2413 .tests
2414 .iter()
2415 .filter(|t| not_kept(t, many).is_none())
2416 .cloned()
2417 .collect();
2418 if !keep.is_empty() {
2419 (draft.held_out, draft.sealed) = held_out(&ctx, &author, &bench, &req, &keep, options);
2420 }
2421 }
2422 if only.is_empty() {
2424 store(&ctx.plane_dir, &ctx.feature, &draft, &contents).map_err(failed)?;
2425 }
2426 let keep: Vec<&DraftTest> = draft
2427 .tests
2428 .iter()
2429 .filter(|t| not_kept(t, many).is_none())
2430 .collect();
2431 if keep.is_empty() && draft.tests.is_empty() && draft.discarded.is_empty() {
2434 return Ok(RunOracle {
2435 tests: Vec::new(),
2436 contents: Vec::new(),
2437 contracts: draft.expectations.sample1.clone(),
2438 untestable: draft.untestable.sample1.clone(),
2439 usage: author.spent.borrow().clone(),
2440 });
2441 }
2442 if keep.is_empty() {
2443 let why: Vec<String> = draft
2444 .discarded
2445 .iter()
2446 .map(|d| format!("`{}` {}", d.path, discard_words(&d.why)))
2447 .chain(
2448 draft
2449 .tests
2450 .iter()
2451 .filter_map(|t| not_kept(t, many).map(|why| format!("`{}`: {why}", t.path))),
2452 )
2453 .collect();
2454 return Err(Box::new(
2455 refuse(
2456 CommandStatus::NeedsHuman,
2457 "ORACLE-UNWRITTEN",
2458 &format!(
2459 "no test written for {story} could be used, asked twice: {}",
2460 why.join("; ")
2461 ),
2462 )
2463 .with_fix(vec![format!("speccoder review tests {story} --write")]),
2464 ));
2465 }
2466 let manifest = manifest_digest(
2467 keep.iter()
2468 .map(|t| (t.path.as_str(), t.sha256.as_str(), t.argv.as_slice())),
2469 );
2470 let paths: Vec<&str> = keep.iter().map(|t| t.path.as_str()).collect();
2471 let kept_contents: Vec<(String, String)> = contents
2472 .iter()
2473 .filter(|(p, _)| paths.contains(&p.as_str()))
2474 .cloned()
2475 .collect();
2476 let f = Freezable {
2477 req: req.clone(),
2478 draft: draft.clone(),
2479 contents: kept_contents.clone(),
2480 manifest,
2481 };
2482 if only.is_empty() {
2485 freeze_tests(
2486 &ctx.plane_dir,
2487 &ctx.feature,
2488 &f,
2489 &keep,
2490 &format!(
2491 "run:{run}{}",
2492 if only.is_empty() { "" } else { ":amendment" }
2493 ),
2494 Decided {
2495 by: "speccoder run".to_owned(),
2496 via: Via::Terminal,
2497 at: now(),
2498 },
2499 Producer::Run,
2500 )
2501 .map_err(failed)?;
2502 }
2503 Ok(RunOracle {
2504 tests: keep
2505 .iter()
2506 .map(|t| FrozenTest {
2507 path: t.path.clone(),
2508 sha256: t.sha256.clone(),
2509 argv: t.argv.clone(),
2510 scenarios: t.scenarios.clone(),
2511 })
2512 .collect(),
2513 contents: kept_contents,
2514 contracts: draft.expectations.sample1.clone(),
2515 untestable: draft.untestable.sample1.clone(),
2516 usage: author.spent.borrow().clone(),
2517 })
2518}
2519
2520pub(crate) fn refreeze(
2527 project_state: &Path,
2528 unit: &str,
2529 key: &str,
2530 tests: Vec<FrozenTest>,
2531) -> Result<(), String> {
2532 let mut frozen = load_frozen(project_state, unit, key)
2533 .ok_or_else(|| format!("{key} has no frozen oracle here"))?;
2534 frozen.manifest_sha256 = manifest_digest(
2535 tests
2536 .iter()
2537 .map(|t| (t.path.as_str(), t.sha256.as_str(), t.argv.as_slice())),
2538 );
2539 frozen.tests = tests;
2540 let text = serde_json::to_string_pretty(&frozen).map_err(|e| e.to_string())?;
2541 write_atomic(&dir(project_state, unit, key).join("frozen.json"), &text)
2542}
2543
2544#[derive(Debug)]
2546pub(crate) enum TaskReuse {
2547 Ready {
2549 record: Box<PreFrozenAcceptance>,
2551 files: Vec<(String, String)>,
2553 tests: Vec<crate::pipeline::ReusedTest>,
2555 manifest: String,
2557 },
2558 Reused {
2560 req: String,
2562 },
2563 Elsewhere {
2565 req: String,
2567 },
2568 Lapsed {
2570 req: String,
2572 frozen: String,
2574 current: String,
2576 },
2577 Tampered(String),
2579 Unreadable(String),
2581}
2582
2583pub(crate) fn for_task(
2588 project_state: &Path,
2589 project: &Path,
2590 wt_root: &Path,
2591 format: &dyn Adapter,
2592 snapshot: &ProjectSnapshot,
2593 scenarios: &[speccoder_core::spec::ScenarioRef],
2594) -> Option<TaskReuse> {
2595 let f = snapshot.feature.as_ref()?;
2596 let unit = f.feature.name.as_str();
2597 let (entries, approval) = match crate::review::oracle_entries(project, &f.feature.dir) {
2598 Ok(x) => x,
2599 Err(e) => return Some(TaskReuse::Unreadable(e)),
2601 };
2602 if entries.is_empty() {
2603 return None;
2604 }
2605 let ids: Vec<&str> = scenarios.iter().map(|s| s.id.as_str()).collect();
2606 let requirements = format.oracle_requirements(snapshot);
2607 let (req, entry) = requirements.iter().find_map(|r| {
2608 let e = entries.get(&r.key)?;
2609 r.scenarios
2610 .iter()
2611 .any(|s| ids.contains(&s.id.as_str()))
2612 .then_some((r, e))
2613 })?;
2614 let digest = req.digest();
2615 if entry.requirement_sha256 != digest {
2616 return Some(TaskReuse::Lapsed {
2617 req: req.key.clone(),
2618 frozen: entry.requirement_sha256.clone(),
2619 current: digest,
2620 });
2621 }
2622 let (record, files) = match crate::pre_frozen::load(project_state, unit, &req.key) {
2623 Ok(Some(x)) => x,
2624 Ok(None) => {
2625 return Some(TaskReuse::Elsewhere {
2626 req: req.key.clone(),
2627 });
2628 }
2629 Err(t) => return Some(TaskReuse::Tampered(t.0)),
2630 };
2631 let needle = format!("Acceptance: review manifest={} ", entry.manifest_sha256);
2635 let reused = crate::git::Git::at(wt_root)
2636 .run(&[
2637 "log",
2638 "--format=%H",
2639 "--fixed-strings",
2640 &format!("--grep={needle}"),
2641 "HEAD",
2642 ])
2643 .map(|out| !out.is_empty());
2644 match reused {
2645 Ok(true) => {
2646 return Some(TaskReuse::Reused {
2647 req: req.key.clone(),
2648 });
2649 }
2650 Ok(false) => {}
2651 Err(e) => {
2652 return Some(TaskReuse::Unreadable(format!(
2653 "the worktree's history cannot be read: {}",
2654 e.0
2655 )));
2656 }
2657 }
2658 if !record.is_current(&digest) {
2659 return Some(TaskReuse::Lapsed {
2660 req: req.key.clone(),
2661 frozen: record.wording_digest.clone(),
2662 current: digest,
2663 });
2664 }
2665 let Some(frozen) = load_frozen(project_state, unit, &req.key) else {
2666 return Some(TaskReuse::Tampered(format!(
2667 "{}'s frozen commands are missing from the state directory",
2668 req.key
2669 )));
2670 };
2671 let manifest = manifest_digest(
2672 frozen
2673 .tests
2674 .iter()
2675 .map(|t| (t.path.as_str(), t.sha256.as_str(), t.argv.as_slice())),
2676 );
2677 let same_files = frozen.tests.len() == record.visible.len()
2678 && frozen.tests.iter().all(|t| {
2679 record
2680 .visible
2681 .iter()
2682 .any(|v| v.path == t.path && v.sha256 == t.sha256)
2683 });
2684 if manifest != entry.manifest_sha256 || manifest != frozen.manifest_sha256 || !same_files {
2685 return Some(TaskReuse::Tampered(format!(
2686 "{}'s frozen tests no longer match the manifest review.json names",
2687 req.key
2688 )));
2689 }
2690 if approval.as_deref() != Some(record.approval_digest.as_str()) {
2693 return Some(TaskReuse::Tampered(format!(
2694 "{}'s frozen set names another approval than review.json{}",
2695 req.key,
2696 if approval.is_none() {
2697 " (which records none)"
2698 } else {
2699 ""
2700 }
2701 )));
2702 }
2703 let sealed = record
2704 .held_out_sealed
2705 .as_ref()
2706 .map(|h| format!("sha256:{}", h.manifest_sha256));
2707 if sealed != entry.held_out_sha256 {
2708 return Some(TaskReuse::Tampered(format!(
2709 "{}'s frozen held-out set is not the one review.json names",
2710 req.key
2711 )));
2712 }
2713 let tests = frozen
2714 .tests
2715 .iter()
2716 .map(|t| crate::pipeline::ReusedTest {
2717 path: t.path.clone(),
2718 cites: t.scenarios.clone(),
2719 argv: t.argv.clone(),
2720 green: false,
2721 })
2722 .collect();
2723 Some(TaskReuse::Ready {
2724 record: Box::new(record),
2725 files,
2726 tests,
2727 manifest,
2728 })
2729}
2730
2731pub(crate) fn note_lapse(project_state: &Path, unit: &str, req: &str, frozen: &str) -> bool {
2734 let file = project_state.join(DIR).join(unit).join("lapsed.json");
2735 let mut seen: Vec<String> = std::fs::read_to_string(&file)
2736 .ok()
2737 .and_then(|t| serde_json::from_str(&t).ok())
2738 .unwrap_or_default();
2739 let key = format!("{req}={frozen}");
2740 if seen.contains(&key) {
2741 return false;
2742 }
2743 seen.push(key);
2744 let _ = write_atomic(&file, &serde_json::to_string(&seen).unwrap_or_default());
2745 true
2746}
2747
2748#[must_use]
2750pub fn autopilot_denials(
2751 task: &str,
2752 tool_calls: &[Value],
2753) -> Vec<speccoder_core::oracle::AutopilotDenial> {
2754 tool_calls
2755 .iter()
2756 .enumerate()
2757 .filter(|(_, c)| {
2758 c["decision"] == "deny" && c["reason"].as_str() == Some(tools::policy::AUTOPILOT_DENIED)
2759 })
2760 .map(|(i, c)| {
2761 let object: Option<tools::policy::Object> =
2762 serde_json::from_value(c["object"].clone()).ok();
2763 speccoder_core::oracle::AutopilotDenial {
2764 task: task.to_owned(),
2765 turn: c["turn"]
2766 .as_u64()
2767 .and_then(|n| u32::try_from(n).ok())
2768 .unwrap_or_else(|| u32::try_from(i + 1).unwrap_or(u32::MAX)),
2769 class: object
2770 .as_ref()
2771 .map(|o| o.class().as_str().to_owned())
2772 .unwrap_or_default(),
2773 object: object.as_ref().map_or_else(
2774 || {
2775 c["arguments"]["path"]
2776 .as_str()
2777 .map(str::to_owned)
2778 .or_else(|| {
2779 c["arguments"]["argv"].as_array().map(|a| {
2780 a.iter()
2781 .filter_map(Value::as_str)
2782 .collect::<Vec<_>>()
2783 .join(" ")
2784 })
2785 })
2786 .unwrap_or_else(|| c["tool"].as_str().unwrap_or_default().to_owned())
2787 },
2788 tools::policy::Object::describe,
2789 ),
2790 reason: tools::policy::AUTOPILOT_DENIED.to_owned(),
2791 }
2792 })
2793 .collect()
2794}
2795
2796#[must_use]
2798pub fn last_denials(
2799 project_state: &Path,
2800 unit: &str,
2801) -> Vec<speccoder_core::oracle::AutopilotDenial> {
2802 std::fs::read_to_string(project_state.join(DIR).join(unit).join("autopilot.json"))
2803 .ok()
2804 .and_then(|t| serde_json::from_str(&t).ok())
2805 .unwrap_or_default()
2806}
2807
2808pub(crate) fn record_denials(
2810 project_state: &Path,
2811 unit: &str,
2812 denials: &[speccoder_core::oracle::AutopilotDenial],
2813) {
2814 let file = project_state.join(DIR).join(unit).join("autopilot.json");
2815 if denials.is_empty() {
2816 let _ = std::fs::remove_file(file);
2817 return;
2818 }
2819 let _ = write_atomic(
2820 &file,
2821 &serde_json::to_string_pretty(denials).unwrap_or_default(),
2822 );
2823}
2824
2825#[cfg(test)]
2826mod tests {
2827 use super::*;
2828
2829 fn draft(content: &str) -> Draft {
2830 Draft {
2831 requirement: DraftRequirement {
2832 key: "US1".to_owned(),
2833 title: "Greet".to_owned(),
2834 text: "### User Story 1 - Greet".to_owned(),
2835 sha256: "sha256:w".to_owned(),
2836 },
2837 samples: 2,
2838 tests: vec![DraftTest {
2839 path: "tests/acceptance_us1.rs".to_owned(),
2840 sha256: digest_of(content),
2841 scenarios: vec!["US1-1".to_owned()],
2842 argv: vec!["cargo".to_owned(), "test".to_owned()],
2843 red_class: RedClass::Assertion,
2844 authored_by: AuthoredBy::Model,
2845 edited_by: None,
2846 }],
2847 discarded: Vec::new(),
2848 rejected: Vec::new(),
2849 expectations: Samples::default(),
2850 untestable: Samples::default(),
2851 ambiguities: Vec::new(),
2852 held_out: HeldOutSummary {
2853 state: "absent".to_owned(),
2854 count: 0,
2855 manifest_sha256: None,
2856 reason: None,
2857 discarded: Vec::new(),
2858 },
2859 sealed: None,
2860 notes: Vec::new(),
2861 runs: Vec::new(),
2862 context: Vec::new(),
2863 }
2864 }
2865
2866 #[test]
2867 fn a_draft_round_trips_and_a_changed_test_is_tampering() {
2868 let state = tempfile::tempdir().unwrap_or_else(|e| panic!("{e}"));
2869 let d = draft("#[test] fn t() {}");
2870 let files = vec![(
2871 "tests/acceptance_us1.rs".to_owned(),
2872 "#[test] fn t() {}".to_owned(),
2873 )];
2874 store(state.path(), "001-greet", &d, &files).unwrap_or_else(|e| panic!("{e}"));
2875 let (back, contents) = load(state.path(), "001-greet", "US1")
2876 .unwrap_or_else(|e| panic!("{e:?}"))
2877 .unwrap_or_else(|| panic!("stored"));
2878 assert_eq!(back, d);
2879 assert_eq!(contents, files);
2880 std::fs::write(
2881 dir(state.path(), "001-greet", "US1").join("files/tests/acceptance_us1.rs"),
2882 "changed",
2883 )
2884 .unwrap_or_else(|e| panic!("{e}"));
2885 assert!(load(state.path(), "001-greet", "US1").is_err());
2886 assert!(store(state.path(), "001-greet", &d, &[]).is_err());
2887 assert!(
2888 load(state.path(), "001-greet", "US2")
2889 .unwrap_or_default()
2890 .is_none()
2891 );
2892 }
2893
2894 #[test]
2895 fn the_manifest_names_paths_contents_and_commands() {
2896 let argv = vec!["cargo".to_owned()];
2897 let other = vec!["make".to_owned()];
2898 let a = manifest_digest([("a", "sha256:1", argv.as_slice())]);
2899 assert_eq!(a, manifest_digest([("a", "sha256:1", argv.as_slice())]));
2900 assert_ne!(a, manifest_digest([("a", "sha256:2", argv.as_slice())]));
2901 assert_ne!(a, manifest_digest([("a", "sha256:1", other.as_slice())]));
2902 let both = manifest_digest([("a", "s", argv.as_slice()), ("b", "s", argv.as_slice())]);
2903 assert_eq!(
2904 both,
2905 manifest_digest([("b", "s", argv.as_slice()), ("a", "s", argv.as_slice())])
2906 );
2907 let mut d = draft("x");
2908 assert!(d.manifest().is_some());
2909 d.tests[0].red_class = RedClass::Passed;
2910 assert!(d.manifest().is_none());
2911 assert_eq!(slug("cap: a / b"), "cap__a___b");
2912 }
2913
2914 #[test]
2915 fn a_lapse_is_noted_once() {
2916 let state = tempfile::tempdir().unwrap_or_else(|e| panic!("{e}"));
2917 assert!(note_lapse(state.path(), "u", "US1", "sha256:a"));
2918 assert!(!note_lapse(state.path(), "u", "US1", "sha256:a"));
2919 assert!(note_lapse(state.path(), "u", "US1", "sha256:b"));
2920 }
2921}