1use std::fmt::Write as _;
23use std::path::{Path, PathBuf};
24use std::time::Duration;
25
26use anyhow::{Context as _, Result, anyhow, bail};
27use serde::{Deserialize, Serialize};
28
29use crate::agent::{self, Invocation, SeatState};
30use crate::ask;
31use crate::config::AgentSpec;
32use crate::git;
33use crate::land;
34use crate::notices::{Link, Notice, Notices};
35use crate::proc::Quiet as _;
36use crate::run::{self, RunState, RunStatus};
37use crate::verdict;
38
39const DECISION_TIMEOUT: Duration = Duration::from_secs(600);
46
47pub fn should_release_bump(status: RunStatus) -> bool {
63 status == RunStatus::Merged
64}
65
66#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
68#[serde(rename_all = "lowercase")]
69pub enum BumpLevel {
70 Major,
72 Minor,
74 Patch,
76}
77
78impl BumpLevel {
79 pub fn as_str(self) -> &'static str {
81 match self {
82 Self::Major => "major",
83 Self::Minor => "minor",
84 Self::Patch => "patch",
85 }
86 }
87
88 fn severity(self) -> u8 {
93 match self {
94 Self::Patch => 0,
95 Self::Minor => 1,
96 Self::Major => 2,
97 }
98 }
99}
100
101#[derive(Debug, Clone, Deserialize)]
108pub struct BumpDecision {
109 pub level: BumpLevel,
111 pub reason: String,
114}
115
116pub fn parse_decision(text: &str) -> Result<BumpDecision> {
120 let decision: BumpDecision = verdict::extract_json(text)?;
121 if decision.reason.trim().is_empty() {
122 bail!("the bump decision carried no reason");
123 }
124 Ok(decision)
125}
126
127#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
130pub struct Version {
131 pub major: u64,
133 pub minor: u64,
135 pub patch: u64,
137}
138
139impl Version {
140 pub fn parse(s: &str) -> Result<Self> {
145 let s = s.trim();
146 let mut parts = s.splitn(3, '.');
147 let major = parts
148 .next()
149 .with_context(|| format!("`{s}` has no major component"))?;
150 let minor = parts
151 .next()
152 .with_context(|| format!("`{s}` has no minor component"))?;
153 let patch = parts
154 .next()
155 .with_context(|| format!("`{s}` has no patch component"))?;
156 let patch_digits: String = patch.chars().take_while(char::is_ascii_digit).collect();
157 Ok(Self {
158 major: major
159 .trim()
160 .parse()
161 .with_context(|| format!("`{major}` is not a number"))?,
162 minor: minor
163 .trim()
164 .parse()
165 .with_context(|| format!("`{minor}` is not a number"))?,
166 patch: patch_digits
167 .parse()
168 .with_context(|| format!("`{patch}` has no numeric patch component"))?,
169 })
170 }
171
172 #[must_use]
175 pub fn bump(self, level: BumpLevel) -> Self {
176 match level {
177 BumpLevel::Major => Self {
178 major: self.major + 1,
179 minor: 0,
180 patch: 0,
181 },
182 BumpLevel::Minor => Self {
183 major: self.major,
184 minor: self.minor + 1,
185 patch: 0,
186 },
187 BumpLevel::Patch => Self {
188 major: self.major,
189 minor: self.minor,
190 patch: self.patch + 1,
191 },
192 }
193 }
194}
195
196impl std::fmt::Display for Version {
197 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
198 write!(f, "{}.{}.{}", self.major, self.minor, self.patch)
199 }
200}
201
202pub fn is_release_only(files: &[String]) -> bool {
211 !files.is_empty() && files.iter().all(|f| f == "Cargo.toml" || f == "Cargo.lock")
212}
213
214fn rewrite_table_version(toml: &str, table: &str, new_version: &str) -> Result<String> {
225 let mut out = String::with_capacity(toml.len() + 8);
226 let mut in_table = false;
227 let mut done = false;
228 for line in toml.split_inclusive('\n') {
229 let trimmed = line.trim();
230 if trimmed.starts_with('[') {
231 in_table = trimmed == table;
232 }
233 if !done && in_table && trimmed.split('=').next().map(str::trim) == Some("version") {
234 let newline = if line.ends_with("\r\n") { "\r\n" } else { "\n" };
235 let _ = write!(out, "version = \"{new_version}\"{newline}");
236 done = true;
237 continue;
238 }
239 out.push_str(line);
240 }
241 if !done {
242 bail!("no `version` field found under `{table}`");
243 }
244 Ok(out)
245}
246
247fn strip_trailing_comment(s: &str) -> &str {
253 let mut in_string = false;
254 let mut quote = '"';
255 for (i, c) in s.char_indices() {
256 if in_string {
257 if c == quote {
258 in_string = false;
259 }
260 } else if c == '"' || c == '\'' {
261 in_string = true;
262 quote = c;
263 } else if c == '#' {
264 return s[..i].trim_end();
265 }
266 }
267 s
268}
269
270fn unquote_key(key: &str) -> &str {
276 for quote in ['"', '\''] {
277 if let Some(inner) = key.strip_prefix(quote).and_then(|k| k.strip_suffix(quote)) {
278 return inner;
279 }
280 }
281 key
282}
283
284fn extract_workspace_dependencies_fragment(toml: &str) -> String {
296 let mut fragment = String::new();
297 let mut capturing = false;
298 for line in toml.split_inclusive('\n') {
299 let header = strip_trailing_comment(line.trim());
300 if header.starts_with('[') {
301 if header == "[workspace.dependencies]" {
302 capturing = true;
303 fragment.push_str("[dependencies]\n");
304 } else if let Some(name) = header
305 .strip_prefix("[workspace.dependencies.")
306 .and_then(|rest| rest.strip_suffix(']'))
307 {
308 capturing = true;
309 let _ = writeln!(fragment, "[dependencies.{name}]");
310 } else {
311 capturing = false;
312 }
313 continue;
314 }
315 if capturing {
316 fragment.push_str(line);
317 }
318 }
319 fragment
320}
321
322fn internal_pin_names(toml: &str) -> Result<Vec<String>> {
340 let fragment = extract_workspace_dependencies_fragment(toml);
341 if fragment.trim().is_empty() {
342 return Ok(Vec::new());
343 }
344 let value: toml::Value =
345 toml::from_str(&fragment).context("failed to parse `[workspace.dependencies]` as TOML")?;
346 let mut names: Vec<String> = value
347 .get("dependencies")
348 .and_then(|d| d.as_table())
349 .into_iter()
350 .flatten()
351 .filter(|(_, dep)| {
352 dep.as_table()
353 .is_some_and(|t| t.contains_key("path") && t.contains_key("version"))
354 })
355 .map(|(name, _)| name.clone())
356 .collect();
357 names.sort();
358 Ok(names)
359}
360
361fn rewrite_quoted_field(line: &str, key: &str, new_value: &str) -> Option<String> {
371 let is_ident = |c: char| c.is_ascii_alphanumeric() || c == '_' || c == '-';
372 let mut search_from = 0;
373 while let Some(rel) = line[search_from..].find(key) {
374 let key_start = search_from + rel;
375 let before_ok = key_start == 0 || !is_ident(line[..key_start].chars().next_back().unwrap());
376 let eq_pos = key_start
377 + key.len()
378 + (line[key_start + key.len()..].len()
379 - line[key_start + key.len()..].trim_start().len());
380 if before_ok && line[eq_pos..].starts_with('=') {
381 let after_eq = &line[eq_pos + 1..];
382 let ws_len = after_eq.len() - after_eq.trim_start().len();
383 let quote_pos = eq_pos + 1 + ws_len;
384 if let Some(quote_char) = line[quote_pos..]
385 .chars()
386 .next()
387 .filter(|c| *c == '"' || *c == '\'')
388 {
389 let value_begin = quote_pos + quote_char.len_utf8();
390 if let Some(end_rel) = line[value_begin..].find(quote_char) {
391 let value_end = value_begin + end_rel;
392 let mut out = String::with_capacity(line.len());
393 out.push_str(&line[..value_begin]);
394 out.push_str(new_value);
395 out.push_str(&line[value_end..]);
396 return Some(out);
397 }
398 }
399 }
400 search_from = key_start + key.len();
401 }
402 None
403}
404
405fn rewrite_workspace_dependency_pins(
420 toml: &str,
421 names: &[String],
422 new_version: &str,
423) -> Result<String> {
424 if names.is_empty() {
425 return Ok(toml.to_owned());
426 }
427 let mut out = String::with_capacity(toml.len() + names.len() * 8);
428 let mut in_flat_table = false;
429 let mut in_named_table: Option<String> = None;
430 let mut rewritten: std::collections::HashSet<String> = std::collections::HashSet::new();
431
432 for line in toml.split_inclusive('\n') {
433 let trimmed = line.trim();
434 if trimmed.starts_with('[') {
435 let header = strip_trailing_comment(trimmed);
436 in_flat_table = header == "[workspace.dependencies]";
437 in_named_table = header
438 .strip_prefix("[workspace.dependencies.")
439 .and_then(|rest| rest.strip_suffix(']'))
440 .map(str::to_owned);
441 out.push_str(line);
442 continue;
443 }
444
445 let key = trimmed.split('=').next().map(str::trim).map(unquote_key);
446
447 if in_flat_table {
448 if let Some(name) = key.and_then(|k| names.iter().find(|n| n.as_str() == k)) {
450 if let Some(rewritten_line) = rewrite_quoted_field(line, "version", new_version) {
451 rewritten.insert(name.clone());
452 out.push_str(&rewritten_line);
453 continue;
454 }
455 } else if let Some(name) = key.and_then(|k| {
456 k.strip_suffix(".version")
457 .and_then(|prefix| names.iter().find(|n| n.as_str() == prefix))
458 }) {
459 if let Some(rewritten_line) = rewrite_quoted_field(line, "version", new_version) {
464 rewritten.insert(name.clone());
465 out.push_str(&rewritten_line);
466 continue;
467 }
468 }
469 } else if let Some(table_name) = &in_named_table {
470 if key == Some("version") && names.iter().any(|n| n == table_name) {
471 if let Some(rewritten_line) = rewrite_quoted_field(line, "version", new_version) {
472 rewritten.insert(table_name.clone());
473 out.push_str(&rewritten_line);
474 continue;
475 }
476 }
477 }
478
479 out.push_str(line);
480 }
481
482 for name in names {
483 if !rewritten.contains(name) {
484 bail!(
485 "could not rewrite the `[workspace.dependencies]` version pin \
486 for `{name}` - its `version` field was not found in a shape \
487 this rewrite understands"
488 );
489 }
490 }
491 Ok(out)
492}
493
494fn verify_pins_rewritten(toml: &str, names: &[String], new_version: &str) -> Result<()> {
502 let fragment = extract_workspace_dependencies_fragment(toml);
503 let value: toml::Value = toml::from_str(&fragment)
504 .context("rewritten `[workspace.dependencies]` failed to parse")?;
505 let deps = value.get("dependencies").and_then(|d| d.as_table());
506 for name in names {
507 let actual = deps
508 .and_then(|d| d.get(name))
509 .and_then(|dep| dep.as_table())
510 .and_then(|t| t.get("version"))
511 .and_then(|v| v.as_str());
512 if actual != Some(new_version) {
513 bail!(
514 "the `[workspace.dependencies]` version pin for `{name}` did \
515 not end up at `{new_version}` after the rewrite"
516 );
517 }
518 }
519 Ok(())
520}
521
522pub fn rewrite_cargo_version(toml: &str, new_version: &str) -> Result<String> {
544 let rewritten = rewrite_table_version(toml, "[package]", new_version)
545 .or_else(|_| rewrite_table_version(toml, "[workspace.package]", new_version))
546 .context("no `version` field found under `[package]` or `[workspace.package]`")?;
547
548 let pin_names = internal_pin_names(&rewritten)?;
549 if pin_names.is_empty() {
550 return Ok(rewritten);
551 }
552
553 let rewritten = rewrite_workspace_dependency_pins(&rewritten, &pin_names, new_version)?;
554 verify_pins_rewritten(&rewritten, &pin_names, new_version)?;
555 Ok(rewritten)
556}
557
558fn version_in_table(toml: &str, table: &str) -> Option<String> {
560 let mut in_table = false;
561 for line in toml.lines() {
562 let trimmed = line.trim();
563 if trimmed.starts_with('[') {
564 in_table = trimmed == table;
565 continue;
566 }
567 if !in_table {
568 continue;
569 }
570 let mut parts = trimmed.splitn(2, '=');
571 let key = parts.next().map(str::trim);
572 let Some(value) = parts.next() else {
573 continue;
574 };
575 if key == Some("version") {
576 return Some(value.trim().trim_matches('"').to_owned());
577 }
578 }
579 None
580}
581
582pub(crate) fn current_version(toml: &str) -> Result<String> {
587 version_in_table(toml, "[package]")
588 .or_else(|| version_in_table(toml, "[workspace.package]"))
589 .context("no `version` field found under `[package]` or `[workspace.package]`")
590}
591
592pub fn decision_prompt(
602 subject: &str,
603 instruction: &str,
604 diffstat: &str,
605 files: &[String],
606 current_version: &str,
607) -> String {
608 let mut s = format!(
609 "A pull request just merged into the base branch. Decide which digit \
610 of this project's `major.minor.patch` version this change earns, so \
611 a release bump can be opened for exactly it.\n\n\
612 Current version: {current_version}\n\n\
613 # Merge subject\n\n{subject}\n\n\
614 # The task that produced it\n\n{instruction}\n\n\
615 # Files changed ({} total)\n\n",
616 files.len()
617 );
618 const MAX_FILES: usize = 50;
619 for f in files.iter().take(MAX_FILES) {
620 let _ = writeln!(s, "- {f}");
621 }
622 if files.len() > MAX_FILES {
623 let _ = writeln!(s, "- ... and {} more", files.len() - MAX_FILES);
624 }
625 let _ = write!(s, "\n# Diffstat\n\n```\n{}\n```\n", diffstat.trim());
626
627 s.push_str(
628 "\n# How to decide\n\n\
629 This project is below version `1.0.0`. At that stage **`minor` is \
630 the digit that carries a breaking change** - do not spend `major` \
631 below `1.0.0`.\n\n\
632 A change is breaking, and earns `minor`, when it changes any of: \
633 the public API reachable from `src/lib.rs`, a CLI subcommand or \
634 flag, an HTTP API route or response shape, a configuration key, or \
635 the on-disk shape of persisted state.\n\n\
636 A user-visible new capability that breaks none of the above also \
637 earns `minor`.\n\n\
638 A fix, an internal refactor, or a dependency update earns `patch`.\n\n\
639 **When it is not obvious which digit applies, choose the larger \
640 one.** An oversized bump costs nothing; a breaking change shipped as \
641 `patch` breaks every downstream update that pins a range.\n\n\
642 # Output\n\n\
643 Reply with exactly one fenced JSON object and nothing that matters \
644 outside it:\n\n\
645 ```json\n\
646 {\"level\": \"major\" | \"minor\" | \"patch\", \"reason\": \"one line\"}\n\
647 ```\n",
648 );
649 let _ = write!(
651 s,
652 "\n{}\n\nThe `reason` goes into a GitHub pull request body, so write it \
653 in English.\n",
654 crate::prompt::GITHUB_ENGLISH_HEADING
655 );
656 s
657}
658
659#[derive(Debug, Clone, Serialize, Deserialize)]
675pub struct PendingBump {
676 pub target_version: String,
678 pub level: BumpLevel,
681 pub branch: String,
684 pub pr_url: String,
687}
688
689pub fn marker_path(home: &Path, repo: &Path) -> PathBuf {
693 let key = repo.to_string_lossy();
694 home.join("bump")
695 .join(format!("{:016x}.json", crate::rng::fnv1a(&key)))
696}
697
698pub fn read_marker(path: &Path) -> Option<PendingBump> {
702 let body = std::fs::read_to_string(path).ok()?;
703 serde_json::from_str(&body).ok()
704}
705
706pub fn write_marker(path: &Path, marker: &PendingBump) -> Result<()> {
710 if let Some(parent) = path.parent() {
711 std::fs::create_dir_all(parent).with_context(|| format!("create {}", parent.display()))?;
712 }
713 let body = serde_json::to_string_pretty(marker).context("serialize pending bump")?;
714 let tmp = path.with_extension("json.tmp");
715 std::fs::write(&tmp, &body).with_context(|| format!("write {}", tmp.display()))?;
716 std::fs::rename(&tmp, path).with_context(|| format!("replace {}", path.display()))?;
717 Ok(())
718}
719
720pub fn clear_marker(path: &Path) {
723 let _ = std::fs::remove_file(path);
724}
725
726#[derive(Debug, Clone, PartialEq, Eq)]
730pub enum Coalesce {
731 Proceed,
734 Skip {
736 target_version: String,
738 },
739}
740
741pub fn coalesce(pending: Option<&PendingBump>, current_version: &str) -> Result<Coalesce> {
743 let Some(pending) = pending else {
744 return Ok(Coalesce::Proceed);
745 };
746 let current = Version::parse(current_version)?;
747 let target = Version::parse(&pending.target_version)?;
748 if current >= target {
749 return Ok(Coalesce::Proceed);
750 }
751 Ok(Coalesce::Skip {
752 target_version: pending.target_version.clone(),
753 })
754}
755
756#[derive(Debug, Clone, Copy, PartialEq, Eq)]
758pub enum PendingAction {
759 AlreadyCovered,
762 Escalate,
765}
766
767pub fn pending_action(pending_level: BumpLevel, decision_level: BumpLevel) -> PendingAction {
776 if decision_level.severity() > pending_level.severity() {
777 PendingAction::Escalate
778 } else {
779 PendingAction::AlreadyCovered
780 }
781}
782
783fn parse_pr_state(json: &str) -> Result<bool> {
785 #[derive(Deserialize)]
786 struct State {
787 state: String,
788 }
789 let parsed: State =
790 serde_json::from_str(json).context("parse `gh pr view --json state` output")?;
791 Ok(parsed.state.eq_ignore_ascii_case("OPEN"))
792}
793
794async fn pr_is_open(repo: &Path, pr_url: &str) -> Result<bool> {
807 let out = tokio::process::Command::new("gh")
808 .args(["pr", "view", pr_url, "--json", "state"])
809 .current_dir(repo)
810 .quiet()
811 .stdin(std::process::Stdio::null())
812 .output()
813 .await
814 .context("spawn gh pr view")?;
815 if !out.status.success() {
816 bail!(
817 "gh pr view {pr_url}: {}",
818 String::from_utf8_lossy(&out.stderr).trim()
819 );
820 }
821 parse_pr_state(&String::from_utf8_lossy(&out.stdout))
822}
823
824const LOCK_STALE_AFTER: Duration = Duration::from_secs(30 * 60);
832
833struct MarkerLock {
845 path: PathBuf,
846}
847
848impl MarkerLock {
849 fn acquire(marker: &Path) -> Result<Option<Self>> {
853 let path = marker.with_extension("lock");
854 if let Some(parent) = path.parent() {
855 std::fs::create_dir_all(parent)
856 .with_context(|| format!("create {}", parent.display()))?;
857 }
858 if Self::try_create(&path)? {
859 return Ok(Some(Self { path }));
860 }
861 if Self::is_stale(&path) {
862 let _ = std::fs::remove_file(&path);
863 if Self::try_create(&path)? {
864 return Ok(Some(Self { path }));
865 }
866 }
867 Ok(None)
868 }
869
870 fn try_create(path: &Path) -> Result<bool> {
871 match std::fs::OpenOptions::new()
872 .write(true)
873 .create_new(true)
874 .open(path)
875 {
876 Ok(_) => Ok(true),
877 Err(e) if e.kind() == std::io::ErrorKind::AlreadyExists => Ok(false),
878 Err(e) => Err(e).with_context(|| format!("create {}", path.display())),
879 }
880 }
881
882 fn is_stale(path: &Path) -> bool {
883 std::fs::metadata(path)
884 .and_then(|m| m.modified())
885 .ok()
886 .and_then(|m| m.elapsed().ok())
887 .is_some_and(|age| age >= LOCK_STALE_AFTER)
888 }
889}
890
891impl Drop for MarkerLock {
892 fn drop(&mut self) {
893 let _ = std::fs::remove_file(&self.path);
894 }
895}
896
897const LOCK_POLL: Duration = Duration::from_secs(5);
899
900const LOCK_WAIT_CEILING: Duration = Duration::from_secs(25 * 60);
913
914async fn wait_for_marker_lock(marker: &Path) -> Result<Option<MarkerLock>> {
917 wait_for_marker_lock_with(marker, LOCK_POLL, LOCK_WAIT_CEILING).await
918}
919
920async fn wait_for_marker_lock_with(
924 marker: &Path,
925 poll: Duration,
926 ceiling: Duration,
927) -> Result<Option<MarkerLock>> {
928 let mut waited = Duration::ZERO;
929 loop {
930 if let Some(lock) = MarkerLock::acquire(marker)? {
931 return Ok(Some(lock));
932 }
933 if waited >= ceiling {
934 return Ok(None);
935 }
936 tokio::time::sleep(poll).await;
937 waited += poll;
938 }
939}
940
941fn level_between(from: Version, to: Version) -> Option<BumpLevel> {
949 if to.major != from.major {
950 Some(BumpLevel::Major)
951 } else if to.minor != from.minor {
952 Some(BumpLevel::Minor)
953 } else if to.patch != from.patch {
954 Some(BumpLevel::Patch)
955 } else {
956 None
957 }
958}
959
960fn parse_open_release_pr(json: &str) -> Result<Option<(String, String)>> {
963 #[derive(Deserialize)]
964 struct Pr {
965 url: String,
966 #[serde(rename = "headRefName")]
967 head_ref_name: String,
968 }
969 let list: Vec<Pr> =
970 serde_json::from_str(json).context("parse `gh pr list --json url,headRefName` output")?;
971 Ok(list
972 .into_iter()
973 .find(|p| p.head_ref_name.starts_with("chore/release-v"))
974 .map(|p| (p.head_ref_name, p.url)))
975}
976
977pub(crate) fn parse_open_release_prs(json: &str) -> Result<Vec<(String, String)>> {
980 #[derive(Deserialize)]
981 struct Pr {
982 url: String,
983 #[serde(rename = "headRefName")]
984 head_ref_name: String,
985 }
986 let list: Vec<Pr> =
987 serde_json::from_str(json).context("parse `gh pr list --json url,headRefName` output")?;
988 Ok(list
989 .into_iter()
990 .filter(|p| p.head_ref_name.starts_with("chore/release-v"))
991 .map(|p| (p.head_ref_name, p.url))
992 .collect())
993}
994
995pub(crate) async fn list_open_release_prs(repo: &Path) -> Result<Vec<(String, String)>> {
997 let out = tokio::process::Command::new("gh")
998 .args([
999 "pr",
1000 "list",
1001 "--state",
1002 "open",
1003 "--limit",
1004 "100",
1005 "--json",
1006 "url,headRefName",
1007 ])
1008 .current_dir(repo)
1009 .env_remove("GH_REPO")
1010 .quiet()
1011 .stdin(std::process::Stdio::null())
1012 .output()
1013 .await
1014 .context("spawn gh pr list")?;
1015 if !out.status.success() {
1016 bail!(
1017 "gh pr list: {}",
1018 String::from_utf8_lossy(&out.stderr).trim()
1019 );
1020 }
1021 parse_open_release_prs(&String::from_utf8_lossy(&out.stdout))
1022}
1023
1024async fn find_open_release_pr(repo: &Path) -> Result<Option<(String, String)>> {
1039 let out = tokio::process::Command::new("gh")
1040 .args(["pr", "list", "--state", "open", "--json", "url,headRefName"])
1041 .current_dir(repo)
1042 .quiet()
1043 .stdin(std::process::Stdio::null())
1044 .output()
1045 .await
1046 .context("spawn gh pr list")?;
1047 if !out.status.success() {
1048 bail!(
1049 "gh pr list: {}",
1050 String::from_utf8_lossy(&out.stderr).trim()
1051 );
1052 }
1053 parse_open_release_pr(&String::from_utf8_lossy(&out.stdout))
1054}
1055
1056pub async fn after_merge(state: &mut RunState, pr_url: &str) -> Result<()> {
1076 after_merge_at(state, pr_url, None, None).await
1077}
1078
1079#[derive(Debug, Default)]
1081struct Progress {
1082 version: Option<String>,
1084 pending_pr: Option<String>,
1086}
1087
1088fn notice_for_failure(progress: &Progress, err: &anyhow::Error) -> (Option<String>, String) {
1094 let chain = format!("{err:#}");
1095 let first = chain.lines().next().unwrap_or_default().trim();
1096 (progress.version.clone(), first.to_owned())
1097}
1098
1099async fn base_has_cargo_toml(repo: &Path, remote: &str, base: &str) -> Result<bool> {
1105 let out = git::git(
1106 repo,
1107 &[
1108 "ls-tree",
1109 "--name-only",
1110 &format!("{remote}/{base}"),
1111 "--",
1112 "Cargo.toml",
1113 ],
1114 )
1115 .await
1116 .context("look for Cargo.toml on the base branch")?;
1117 Ok(!out.trim().is_empty())
1118}
1119
1120async fn after_merge_at(
1124 state: &mut RunState,
1125 pr_url: &str,
1126 home: Option<&Path>,
1127 store: Option<&Notices>,
1128) -> Result<()> {
1129 let mut progress = Progress::default();
1130 let result = after_merge_inner(state, pr_url, home, &mut progress).await;
1131 if let Err(e) = &result {
1132 let (version, reason) = notice_for_failure(&progress, e);
1133 let pending = progress.pending_pr.as_deref();
1134 let store = store.cloned().unwrap_or_else(Notices::open);
1135 report_problem_in(
1136 state,
1137 &store,
1138 pending,
1139 version.as_deref(),
1140 &reason,
1141 pending.is_some(),
1142 )
1143 .await;
1144 }
1145 result
1146}
1147
1148async fn after_merge_inner(
1149 state: &mut RunState,
1150 pr_url: &str,
1151 home: Option<&Path>,
1152 progress: &mut Progress,
1153) -> Result<()> {
1154 if !state.config.merge.release_bump {
1155 return Ok(());
1156 }
1157 let Some(winner) = state.winner().cloned() else {
1158 return Ok(());
1159 };
1160 let repo = state.repo.clone();
1161 let base = state.base_branch.clone();
1162 let remote = state.config.merge.remote.clone();
1163
1164 let files = git::changed_files(&winner.worktree, &base, &winner.branch)
1165 .await
1166 .unwrap_or_default();
1167 if is_release_only(&files) {
1168 state.event(
1169 "bump",
1170 "the merged change touches only the release manifest; not treating it as a trigger",
1171 );
1172 return Ok(());
1173 }
1174
1175 git::fetch(&repo, &remote, &base).await.ok();
1178 if !base_has_cargo_toml(&repo, &remote, &base).await? {
1179 state.event(
1180 "bump",
1181 "release bump: no Cargo.toml on the base branch; release bumps are Rust-only, skipping",
1182 );
1183 return Ok(());
1184 }
1185
1186 let marker = marker_path(&home.map_or_else(run::home, Path::to_path_buf), &repo);
1187 let Some(_lock) = wait_for_marker_lock(&marker).await? else {
1194 state.event(
1195 "bump",
1196 "another release bump decision held the lock past the wait ceiling; skipping this round",
1197 );
1198 return Ok(());
1199 };
1200
1201 git::fetch(&repo, &remote, &base).await.ok();
1202 let cargo_toml = git::git(&repo, &["show", &format!("{remote}/{base}:Cargo.toml")])
1203 .await
1204 .context("read Cargo.toml from the base branch")?;
1205 let base_version = current_version(&cargo_toml)?;
1206
1207 let mut pending = read_marker(&marker);
1208 if let Some(p) = &pending {
1209 match coalesce(Some(p), &base_version)? {
1210 Coalesce::Proceed => {
1211 clear_marker(&marker);
1214 pending = None;
1215 }
1216 Coalesce::Skip { target_version } => {
1217 if !pr_is_open(&repo, &p.pr_url).await.unwrap_or(true) {
1218 state.event(
1219 "bump",
1220 format!(
1221 "the pending release bump to v{target_version} ({}) is no longer \
1222 open; treating it as abandoned",
1223 p.pr_url
1224 ),
1225 );
1226 clear_marker(&marker);
1227 pending = None;
1228 }
1229 }
1234 }
1235 }
1236
1237 if pending.is_none() {
1238 if let Ok(Some((branch, url))) = find_open_release_pr(&repo).await
1243 && let Some(target) = branch
1244 .strip_prefix("chore/release-v")
1245 .and_then(|v| Version::parse(v).ok())
1246 {
1247 let base_parsed = Version::parse(&base_version)?;
1248 if target > base_parsed
1249 && let Some(level) = level_between(base_parsed, target)
1250 {
1251 let adopted = PendingBump {
1252 target_version: target.to_string(),
1253 level,
1254 branch,
1255 pr_url: url,
1256 };
1257 let _ = write_marker(&marker, &adopted);
1260 pending = Some(adopted);
1261 }
1262 }
1263 }
1264
1265 progress.pending_pr = pending.as_ref().map(|p| p.pr_url.clone());
1266 let title = pr_title(&repo, pr_url).await.unwrap_or_default();
1267 let subject = land::merge_subject(
1268 crate::graph::landing_title(state, &title),
1269 &crate::graph::landing_subject_source(state),
1270 );
1271 let stat = git::diff_stat(&winner.worktree, &base, &winner.branch)
1272 .await
1273 .unwrap_or_default();
1274 let prompt = decision_prompt(&subject, &state.instruction, &stat, &files, &base_version);
1275
1276 let spec: AgentSpec = agent::pick(
1282 &state.config.agents,
1283 crate::config::primary(state.config.roles.chatter.as_ref()),
1284 &agent::installed,
1285 )
1286 .context("choose an agent for the release-bump decision")?;
1287 let mut seat = SeatState::new("bump", &spec.id, state.seed);
1288 let artifacts = agent::artifacts_dir(&state.dir());
1289 let out = agent::invoke(
1290 &spec,
1291 &mut seat,
1292 &Invocation {
1293 cwd: &repo,
1294 prompt: &prompt,
1295 timeout: DECISION_TIMEOUT,
1296 allow_write: false,
1299 sessions: false,
1300 artifacts: &artifacts,
1301 stem: "bump-decision",
1302 run: &state.id,
1303 node: "bump",
1304 cache_dir: state.config.cache_dir().as_deref(),
1305 attachments: &[],
1306 writable: &[],
1307 },
1308 )
1309 .await
1310 .context("ask an agent how big the merged change was")?;
1311 if !out.usable() {
1312 bail!(
1313 "the release-bump decision produced nothing usable (exit {:?}, timed out: {})",
1314 out.exit_code,
1315 out.timed_out
1316 );
1317 }
1318 let decision = parse_decision(&out.text).context("parse the release-bump decision")?;
1319
1320 if let Some(p) = pending {
1321 return match pending_action(p.level, decision.level) {
1322 PendingAction::AlreadyCovered => {
1323 state.event(
1324 "bump",
1325 format!(
1326 "a release bump to v{} ({}) already covers at least a {} change; not \
1327 opening another",
1328 p.target_version,
1329 p.pr_url,
1330 decision.level.as_str()
1331 ),
1332 );
1333 Ok(())
1334 }
1335 PendingAction::Escalate => {
1336 progress.version = Some(
1337 Version::parse(&base_version)?
1338 .bump(decision.level)
1339 .to_string(),
1340 );
1341 escalate_pending(state, &repo, &remote, &p, &decision, &base_version, &marker).await
1342 }
1343 };
1344 }
1345
1346 let next = Version::parse(&base_version)?
1347 .bump(decision.level)
1348 .to_string();
1349 progress.version = Some(next.clone());
1350 let branch = format!("chore/release-v{next}");
1351 let worktree = state.dir().join("bump");
1352 let (shared, branch_ref, next_ref, decision_ref) = (&*state, &branch, &next, &decision);
1353 let (pr_url_opened, outcome) =
1354 release_attempt(
1355 &repo,
1356 &remote,
1357 &base,
1358 &worktree,
1359 &branch,
1360 |head| {
1361 let repo = repo.clone();
1362 async move { gh_open_pr_for_head(&repo, &head).await }
1363 },
1364 |wt| async move {
1365 open_bump_pr(shared, &wt, branch_ref, next_ref, decision_ref, pr_url).await
1366 },
1367 )
1368 .await?;
1369 let local = outcome == AutomergeOutcome::LocalGate;
1370 let (automerge_warning, merged_detail) = match outcome {
1371 AutomergeOutcome::Enabled | AutomergeOutcome::LocalGate => (None, None),
1372 AutomergeOutcome::MergedDirectly { detail } => (None, Some(detail)),
1373 AutomergeOutcome::Failed { reason } => (Some(reason), None),
1374 };
1375
1376 let marker_write = write_marker(
1385 &marker,
1386 &PendingBump {
1387 target_version: next.clone(),
1388 level: decision.level,
1389 branch,
1390 pr_url: pr_url_opened.clone(),
1391 },
1392 );
1393 state.event(
1394 "bump",
1395 format!(
1396 "opened a {} release bump to v{next} ({}): {pr_url_opened}",
1397 decision.level.as_str(),
1398 decision.reason
1399 ),
1400 );
1401 if let Err(e) = marker_write {
1402 state.event(
1403 "bump",
1404 format!(
1405 "could not record the pending release bump marker for v{next}: {e:#}; a later \
1406 merge may open a duplicate pull request if it cannot find {pr_url_opened} on \
1407 the forge either"
1408 ),
1409 );
1410 }
1411 state.release_bump = Some(run::ReleaseBump {
1412 pr_url: Some(pr_url_opened.clone()),
1413 version: Some(next.clone()),
1414 automerge_enabled: automerge_warning.is_none() && merged_detail.is_none() && !local,
1415 merged_directly: merged_detail.is_some(),
1416 local,
1417 ..run::ReleaseBump::default()
1418 });
1419 if let Some(detail) = merged_detail {
1420 clear_marker(&marker);
1423 state.event("bump", format!("merged v{next} directly: {detail}"));
1424 }
1425 if let Some(warning) = automerge_warning {
1426 state.event(
1427 "bump",
1428 format!("could not enable automerge on {pr_url_opened}: {warning}; merge it by hand"),
1429 );
1430 report_problem(state, Some(&pr_url_opened), Some(&next), &warning).await;
1431 }
1432 Ok(())
1433}
1434
1435pub const NOTICE_NODE: &str = "release-bump";
1440
1441fn automerge_hint(reason: &str) -> &'static str {
1448 let r = reason.to_lowercase();
1449 if is_clean_status_refusal(reason) {
1450 "merge the release pull request by hand; CI is already green"
1451 } else if r.contains("enablepullrequestautomerge") || r.contains("protected branch rules") {
1452 "merge the release pull request by hand, and enable branch protection with required \
1453 status checks on the base branch so automerge can work next time"
1454 } else {
1455 "merge the release pull request by hand"
1456 }
1457}
1458
1459fn automerge_failure_comment(reason: &str) -> String {
1461 format!(
1462 "magi could not enable automerge on this pull request: {reason}\n\n\
1463 Action required: {}. Until then the release does not happen.",
1464 automerge_hint(reason)
1465 )
1466}
1467
1468#[cfg(test)]
1476fn surface_problem(
1477 state: &mut RunState,
1478 store: &Notices,
1479 pr_url: Option<&str>,
1480 version: Option<&str>,
1481 reason: &str,
1482) -> Result<(Notice, Option<String>)> {
1483 surface_problem_in(state, store, pr_url, version, reason, false)
1484}
1485
1486fn surface_problem_in(
1490 state: &mut RunState,
1491 store: &Notices,
1492 pr_url: Option<&str>,
1493 version: Option<&str>,
1494 reason: &str,
1495 pending: bool,
1496) -> Result<(Notice, Option<String>)> {
1497 let action = if let (true, Some(url)) = (pending, pr_url) {
1498 let ja = crate::lang::is_japanese(&state.config.graph.language);
1499 let first = reason.lines().next().filter(|l| !l.is_empty());
1500 let cause = first.map(|l| format!(" ({l})")).unwrap_or_default();
1501 if ja {
1502 let target = version.map(|v| format!(" v{v}")).unwrap_or_default();
1503 format!(
1504 "既存のリリース PR {url} を{target}へ更新できませんでした{cause}。PR を手で更新してください"
1505 )
1506 } else {
1507 let target = version.map(|v| format!(" to v{v}")).unwrap_or_default();
1508 format!(
1509 "the pending release pull request {url} could not be updated{target}{cause}; update it by hand"
1510 )
1511 }
1512 } else if pr_url.is_some() {
1513 automerge_hint(reason).to_owned()
1514 } else {
1515 let ja = crate::lang::is_japanese(&state.config.graph.language);
1516 let first = reason.lines().next().filter(|l| !l.is_empty());
1517 if ja {
1518 let target = version.map(|v| format!(" (v{v})")).unwrap_or_default();
1519 let cause = first.map(|l| format!(" ({l})")).unwrap_or_default();
1520 format!(
1521 "リリースバンプ{target}は実行されませんでした{cause}。リリース PR を手で開いてください"
1522 )
1523 } else {
1524 let target = version.map(|v| format!(" to v{v}")).unwrap_or_default();
1525 let cause = first.map(|l| format!(" ({l})")).unwrap_or_default();
1526 format!(
1527 "the release bump{target} did not run{cause}; open the release pull request by hand"
1528 )
1529 }
1530 };
1531 let record = state.release_bump.get_or_insert_with(Default::default);
1532 record.pr_url = pr_url.map(str::to_owned).or(record.pr_url.take());
1533 record.version = version.map(str::to_owned).or(record.version.take());
1534 record.automerge_enabled = false;
1535 record.problem = Some(reason.to_owned());
1536 record.action_required = Some(action.clone());
1537
1538 let mut notice = Notice::error(
1539 &format!("release-bump:{}", state.id),
1540 format!(
1541 "Run {} merged, but its release step failed: {action}.",
1542 state.id
1543 ),
1544 );
1545 notice = match pr_url {
1546 Some(url) => notice.link(Link::Url {
1547 url: url.to_owned(),
1548 }),
1549 None => notice.link(Link::Run {
1550 id: state.id.clone(),
1551 }),
1552 };
1553 let notice = store
1554 .raise(notice)
1555 .context("raise the release-bump notification")?;
1556 state.event("bump", format!("needs attention: {action}"));
1557 let comment = pr_url
1558 .filter(|_| !pending)
1559 .map(|_| automerge_failure_comment(reason));
1560 Ok((notice, comment))
1561}
1562
1563pub async fn report_problem(
1568 state: &mut RunState,
1569 pr_url: Option<&str>,
1570 version: Option<&str>,
1571 reason: &str,
1572) {
1573 report_problem_in(state, &Notices::open(), pr_url, version, reason, false).await;
1574}
1575
1576async fn report_problem_in(
1578 state: &mut RunState,
1579 store: &Notices,
1580 pr_url: Option<&str>,
1581 version: Option<&str>,
1582 reason: &str,
1583 pending: bool,
1584) {
1585 match surface_problem_in(state, store, pr_url, version, reason, pending) {
1586 Ok((notice, comment)) => {
1587 if let (Some(url), Some(body)) = (pr_url, comment)
1588 && let Err(e) = gh_pr_comment(
1589 &state.repo,
1590 url,
1591 &crate::scrub::scrub(&body, &crate::scrub::Identity::current()),
1592 )
1593 .await
1594 {
1595 state.event("bump", format!("could not comment on {url}: {e:#}"));
1596 }
1597 let summary = match pr_url {
1600 Some(url) if pending => format!("Release PR could not be updated: {url}"),
1601 Some(url) => format!("Release PR needs a human: {url}"),
1602 None if crate::lang::is_japanese(&state.config.graph.language) => {
1603 "リリースバンプが実行されませんでした".to_owned()
1604 }
1605 None => "Release bump did not run".to_owned(),
1606 };
1607 let q = ask::Question::new(
1608 state.id.clone(),
1609 NOTICE_NODE.to_owned(),
1610 "bump".to_owned(),
1611 summary,
1612 notice.message.clone(),
1613 Vec::new(),
1614 );
1615 if let Err(e) = ask::notify(&state.config.notify, &q).await {
1616 tracing::warn!(
1617 "could not notify about the release bump of {}: {e:#}",
1618 state.id
1619 );
1620 }
1621 }
1622 Err(e) => state.event("bump", format!("could not raise a notice: {e:#}")),
1623 }
1624}
1625
1626pub(crate) async fn gh_pr_comment(cwd: &Path, pr_url: &str, body: &str) -> Result<()> {
1627 let out = tokio::process::Command::new("gh")
1628 .args(["pr", "comment", pr_url, "--body", body])
1629 .current_dir(cwd)
1630 .quiet()
1631 .stdin(std::process::Stdio::null())
1632 .output()
1633 .await
1634 .context("spawn gh pr comment")?;
1635 if out.status.success() {
1636 Ok(())
1637 } else {
1638 bail!(
1639 "gh pr comment: {}",
1640 String::from_utf8_lossy(&out.stderr).trim()
1641 )
1642 }
1643}
1644
1645async fn escalate_pending(
1654 state: &mut RunState,
1655 repo: &Path,
1656 remote: &str,
1657 pending: &PendingBump,
1658 decision: &BumpDecision,
1659 base_version: &str,
1660 marker: &Path,
1661) -> Result<()> {
1662 let next = Version::parse(base_version)?
1663 .bump(decision.level)
1664 .to_string();
1665 let worktree = state.dir().join("bump");
1666 git::worktree_remove(repo, &worktree).await.ok();
1667 let checked_out = git::git_raw(
1668 repo,
1669 &[
1670 "worktree",
1671 "add",
1672 "--force",
1673 &worktree.to_string_lossy(),
1674 &pending.branch,
1675 ],
1676 )
1677 .await?;
1678 if !checked_out.ok() {
1679 bail!(
1680 "checking out the pending release branch {} failed: {}",
1681 pending.branch,
1682 checked_out.stderr
1683 );
1684 }
1685
1686 let pushed: Result<()> = async {
1691 let cargo_toml_path = worktree.join("Cargo.toml");
1692 let toml = tokio::fs::read_to_string(&cargo_toml_path)
1693 .await
1694 .with_context(|| format!("read {}", cargo_toml_path.display()))?;
1695 let rewritten = rewrite_cargo_version(&toml, &next)?;
1696 tokio::fs::write(&cargo_toml_path, rewritten)
1697 .await
1698 .with_context(|| format!("write {}", cargo_toml_path.display()))?;
1699 sync_lockfile(&worktree, state.config.cache_dir().as_deref()).await?;
1700 let committed = git::commit_all(
1701 &worktree,
1702 &format!(
1703 "chore: release v{next} (supersedes v{})",
1704 pending.target_version
1705 ),
1706 )
1707 .await
1708 .context("commit the escalated version bump")?;
1709 if !committed {
1710 bail!("escalating the version bump left nothing to commit");
1711 }
1712 let pushed = git::push(&worktree, remote, &pending.branch).await?;
1713 if !pushed.ok() {
1714 bail!("pushing {} failed: {}", pending.branch, pushed.stderr);
1715 }
1716 Ok(())
1717 }
1718 .await;
1719 if let Err(e) = pushed {
1720 git::worktree_remove(repo, &worktree).await.ok();
1721 return Err(e);
1722 }
1723
1724 let title_warning = match gh_pr_edit_title(
1728 &worktree,
1729 &pending.pr_url,
1730 &crate::scrub::scrub(
1731 &format!("chore: release v{next} ({} bump)", decision.level.as_str()),
1732 &crate::scrub::Identity::current(),
1733 ),
1734 )
1735 .await
1736 {
1737 Ok(()) => None,
1738 Err(e) => Some(e.to_string()),
1739 };
1740 git::worktree_remove(repo, &worktree).await.ok();
1741
1742 let marker_write = write_marker(
1743 marker,
1744 &PendingBump {
1745 target_version: next.clone(),
1746 level: decision.level,
1747 branch: pending.branch.clone(),
1748 pr_url: pending.pr_url.clone(),
1749 },
1750 );
1751 state.event(
1752 "bump",
1753 format!(
1754 "escalated the pending release bump from v{} to v{next} to a {} change ({}): {}",
1755 pending.target_version,
1756 decision.level.as_str(),
1757 decision.reason,
1758 pending.pr_url
1759 ),
1760 );
1761 if let Err(e) = marker_write {
1762 state.event(
1763 "bump",
1764 format!(
1765 "could not update the pending release bump marker to v{next}: {e:#}; a later \
1766 merge may misjudge whether it is already covered"
1767 ),
1768 );
1769 }
1770 if let Some(warning) = title_warning {
1771 state.event(
1772 "bump",
1773 format!(
1774 "pushed v{next} to {} but could not update its title: {warning}; the squashed \
1775 subject may still read the superseded version",
1776 pending.pr_url
1777 ),
1778 );
1779 }
1780 Ok(())
1781}
1782
1783async fn gh_open_pr_for_head(repo: &Path, branch: &str) -> Result<bool> {
1786 let out = tokio::process::Command::new("gh")
1787 .args([
1788 "pr", "list", "--head", branch, "--state", "open", "--json", "url",
1789 ])
1790 .current_dir(repo)
1791 .quiet()
1792 .stdin(std::process::Stdio::null())
1793 .output()
1794 .await
1795 .context("spawn gh pr list")?;
1796 if !out.status.success() {
1797 bail!(
1798 "gh pr list --head {branch}: {}",
1799 String::from_utf8_lossy(&out.stderr).trim()
1800 );
1801 }
1802 let prs: Vec<serde_json::Value> = serde_json::from_slice(&out.stdout)
1803 .with_context(|| format!("parse the pull requests headed by {branch}"))?;
1804 Ok(!prs.is_empty())
1805}
1806
1807async fn release_attempt<T, F, Fut, P, PFut>(
1816 repo: &Path,
1817 remote: &str,
1818 base: &str,
1819 worktree: &Path,
1820 branch: &str,
1821 open_pr: P,
1822 fill: F,
1823) -> Result<T>
1824where
1825 F: FnOnce(PathBuf) -> Fut,
1826 Fut: std::future::Future<Output = Result<T>>,
1827 P: Fn(String) -> PFut,
1828 PFut: std::future::Future<Output = Result<bool>>,
1829{
1830 let start = format!("{remote}/{base}");
1831 git::worktree_remove(repo, worktree).await.ok();
1832 let mut retried = false;
1833 if git::branch_exists(repo, branch).await? {
1834 reclaim_stale_branch(repo, remote, &start, branch, &open_pr).await?;
1835 retried = true;
1836 }
1837 loop {
1838 if let Err(e) = git::git(repo, &["branch", branch, &start]).await {
1843 let e = e.context("create the release branch");
1844 if !git::branch_exists(repo, branch).await.unwrap_or(false) {
1845 return Err(e);
1846 }
1847 if retried {
1849 return Err(anyhow!(
1850 "{e:#}; left {branch} in place: it is not one this attempt created"
1851 ));
1852 }
1853 retried = true;
1854 if let Err(r) = reclaim_stale_branch(repo, remote, &start, branch, &open_pr).await {
1855 return Err(anyhow!("{r:#} (after: {e:#})"));
1856 }
1857 continue;
1858 }
1859 if let Some(parent) = worktree.parent() {
1860 tokio::fs::create_dir_all(parent).await.ok();
1861 }
1862 let path = worktree.to_string_lossy();
1863 if let Err(e) = git::git(repo, &["worktree", "add", &path, branch]).await {
1864 let e = e.context("create the release-bump worktree");
1865 return Err(discard_attempt(repo, remote, worktree, branch, e).await);
1866 }
1867 break;
1868 }
1869 match fill(worktree.to_path_buf()).await {
1870 Ok(v) => {
1871 git::worktree_remove(repo, worktree).await.ok();
1874 Ok(v)
1875 }
1876 Err(e) => Err(discard_attempt(repo, remote, worktree, branch, e).await),
1877 }
1878}
1879
1880async fn discard_attempt(
1882 repo: &Path,
1883 remote: &str,
1884 worktree: &Path,
1885 branch: &str,
1886 cause: anyhow::Error,
1887) -> anyhow::Error {
1888 git::worktree_remove(repo, worktree).await.ok();
1889 if !git::branch_exists(repo, branch).await.unwrap_or(false) {
1890 return cause;
1891 }
1892 let note = match git::remote_has_branch(repo, remote, branch).await {
1895 Ok(false) => match git::branch_delete(repo, branch).await {
1896 Ok(true) => return cause,
1897 _ => format!("could not delete the local branch {branch}"),
1898 },
1899 Ok(true) => format!("left {branch} in place: it was pushed to {remote}"),
1900 Err(e) => format!(
1901 "left {branch} in place: could not tell whether it was pushed to {remote} ({e:#})"
1902 ),
1903 };
1904 anyhow!("{cause:#}; {note}")
1905}
1906
1907async fn reclaim_stale_branch<P, PFut>(
1912 repo: &Path,
1913 remote: &str,
1914 start: &str,
1915 branch: &str,
1916 open_pr: &P,
1917) -> Result<()>
1918where
1919 P: Fn(String) -> PFut,
1920 PFut: std::future::Future<Output = Result<bool>>,
1921{
1922 match git::remote_has_branch(repo, remote, branch).await {
1923 Ok(false) => {}
1924 Ok(true) => bail!("the branch {branch} already exists and is on {remote}; left alone"),
1925 Err(e) => bail!(
1926 "the branch {branch} already exists and could not check whether {remote} has it ({e:#}); left alone"
1927 ),
1928 }
1929 match open_pr(branch.to_string()).await {
1930 Ok(false) => {}
1931 Ok(true) => {
1932 bail!("the branch {branch} already exists and has an open pull request; left alone")
1933 }
1934 Err(e) => bail!(
1935 "the branch {branch} already exists and could not check for an open pull request \
1936 ({e:#}); left alone"
1937 ),
1938 }
1939 match git::commits_ahead(repo, start, branch).await {
1940 Ok(0) => {}
1941 Ok(n) => bail!(
1942 "the branch {branch} already exists with {n} commit(s) not in {start}; left alone"
1943 ),
1944 Err(e) => bail!(
1945 "the branch {branch} already exists and could not compare it with {start} ({e:#}); \
1946 left alone"
1947 ),
1948 }
1949 if let Some(held) = git::worktree_holding(repo, branch).await? {
1950 let same = |a: &Path, b: &Path| match (a.canonicalize(), b.canonicalize()) {
1951 (Ok(a), Ok(b)) => a == b,
1952 _ => a == b,
1953 };
1954 if same(&held, repo) {
1955 bail!(
1956 "the branch {branch} already exists and is checked out in the main checkout; left alone"
1957 );
1958 }
1959 git::worktree_remove(repo, &held).await.ok();
1960 }
1961 if !git::branch_delete(repo, branch).await? {
1962 bail!("the branch {branch} already exists and could not be deleted; left alone");
1963 }
1964 Ok(())
1965}
1966
1967async fn open_bump_pr(
1973 state: &RunState,
1974 worktree: &Path,
1975 branch: &str,
1976 next_version: &str,
1977 decision: &BumpDecision,
1978 source_pr_url: &str,
1979) -> Result<(String, AutomergeOutcome)> {
1980 let cargo_toml_path = worktree.join("Cargo.toml");
1981 let toml = tokio::fs::read_to_string(&cargo_toml_path)
1982 .await
1983 .with_context(|| format!("read {}", cargo_toml_path.display()))?;
1984 let rewritten = rewrite_cargo_version(&toml, next_version)?;
1985 tokio::fs::write(&cargo_toml_path, rewritten)
1986 .await
1987 .with_context(|| format!("write {}", cargo_toml_path.display()))?;
1988
1989 sync_lockfile(worktree, state.config.cache_dir().as_deref()).await?;
1990
1991 let committed = git::commit_all(worktree, &format!("chore: release v{next_version}"))
1992 .await
1993 .context("commit the version bump")?;
1994 if !committed {
1995 bail!("the version bump left nothing to commit");
1996 }
1997
1998 let remote = state.config.merge.remote.clone();
1999 let pushed = git::push(worktree, &remote, branch).await?;
2000 if !pushed.ok() {
2001 bail!("pushing {branch} failed: {}", pushed.stderr);
2002 }
2003
2004 let (title, body) = release_pr(
2005 decision.level.as_str(),
2006 &decision.reason,
2007 next_version,
2008 &state.id,
2009 source_pr_url,
2010 );
2011 let who = crate::scrub::Identity::current();
2012 let (title, body) = (
2013 crate::scrub::scrub(&title, &who),
2014 crate::scrub::scrub(&body, &who),
2015 );
2016 let url = gh_pr_create(worktree, &state.base_branch, branch, &title, &body).await?;
2017 if state.config.release.is_local() {
2021 crate::release_watch::register(&crate::run::home(), &state.repo, &url, &state.id);
2024 return Ok((url, AutomergeOutcome::LocalGate));
2025 }
2026 let outcome = match gh_enable_automerge(worktree, &url).await {
2027 Ok(()) => AutomergeOutcome::Enabled,
2028 Err(e) => {
2029 let reason = e.to_string();
2030 if is_clean_status_refusal(&reason) {
2031 gh_merge_directly(worktree, &url, &title, reason).await
2032 } else {
2033 AutomergeOutcome::Failed { reason }
2034 }
2035 }
2036 };
2037 Ok((url, outcome))
2038}
2039
2040#[derive(Debug, Clone, PartialEq, Eq)]
2042enum AutomergeOutcome {
2043 Enabled,
2045 MergedDirectly { detail: String },
2047 LocalGate,
2049 Failed { reason: String },
2051}
2052
2053fn is_clean_status_refusal(reason: &str) -> bool {
2059 let r = reason.to_lowercase();
2060 r.contains("is in clean status") && r.contains("enablepullrequestautomerge")
2061}
2062
2063pub(crate) fn bump_merge_argv(pr_url: &str, subject: &str) -> Vec<String> {
2066 [
2067 "pr",
2068 "merge",
2069 pr_url,
2070 "--squash",
2071 "--delete-branch",
2072 "--subject",
2073 subject,
2074 ]
2075 .map(str::to_owned)
2076 .to_vec()
2077}
2078
2079fn resolve_direct_merge(
2083 refusal: &str,
2084 argv: &[String],
2085 merge_ok: bool,
2086 stderr: &str,
2087 after: Option<land::PrLifecycle>,
2088) -> AutomergeOutcome {
2089 if merge_ok {
2090 return AutomergeOutcome::MergedDirectly {
2091 detail: format!("automerge was refused ({refusal}); gh {}", argv.join(" ")),
2092 };
2093 }
2094 match land::merged_after_all(argv, stderr, after) {
2095 Some(m) => AutomergeOutcome::MergedDirectly { detail: m.detail },
2096 None => AutomergeOutcome::Failed {
2097 reason: format!("{refusal}; merging directly failed too: {}", stderr.trim()),
2098 },
2099 }
2100}
2101
2102async fn gh_merge_directly(
2106 cwd: &Path,
2107 pr_url: &str,
2108 subject: &str,
2109 refusal: String,
2110) -> AutomergeOutcome {
2111 let argv = bump_merge_argv(pr_url, subject);
2112 let out = match tokio::process::Command::new("gh")
2113 .args(&argv)
2114 .current_dir(cwd)
2115 .quiet()
2116 .stdin(std::process::Stdio::null())
2117 .output()
2118 .await
2119 {
2120 Ok(o) => o,
2121 Err(e) => {
2122 return AutomergeOutcome::Failed {
2123 reason: format!("{refusal}; could not spawn gh to merge directly: {e}"),
2124 };
2125 }
2126 };
2127 let stderr = String::from_utf8_lossy(&out.stderr).into_owned();
2128 let after = if out.status.success() {
2131 None
2132 } else {
2133 land::lifecycle(cwd, pr_url).await.ok()
2134 };
2135 resolve_direct_merge(&refusal, &argv, out.status.success(), &stderr, after)
2136}
2137
2138async fn sync_lockfile(worktree: &Path, cache_dir: Option<&Path>) -> Result<()> {
2146 let mut cmd = tokio::process::Command::new("cargo");
2147 cmd.arg("build").current_dir(worktree).quiet();
2148 if let Some(dir) = cache_dir {
2149 cmd.env("CARGO_TARGET_DIR", dir);
2150 }
2151 let out = cmd
2152 .stdin(std::process::Stdio::null())
2153 .output()
2154 .await
2155 .context("spawn cargo build")?;
2156 if !out.status.success() {
2157 bail!(
2158 "cargo build failed while syncing Cargo.lock: {}",
2159 String::from_utf8_lossy(&out.stderr).trim()
2160 );
2161 }
2162 Ok(())
2163}
2164
2165fn release_pr(
2170 level: &str,
2171 reason: &str,
2172 next_version: &str,
2173 run_id: &str,
2174 source_pr_url: &str,
2175) -> (String, String) {
2176 let title = format!("chore: release v{next_version} ({level} bump)");
2177 let body = format!(
2178 "## Background\n\n\
2179 A change that was just merged is a `{level}` change, so the crate needs a new \
2180 release: {reason}\n\n\
2181 Triggered by magi run `{run_id}`, which landed {source}.\n\n\
2182 ## Change\n\n\
2183 Raises the package version to `v{next_version}` in `Cargo.toml`, with \
2184 `Cargo.lock` following it. Nothing else changes.\n\n\
2185 ## Risk\n\n\
2186 Version-bump-only, so there is nothing here for a reviewer to find. Merging \
2187 it starts the release pipeline (auto-tag, then the release workflow).",
2188 source = source_pr_url,
2189 );
2190 (title, body)
2191}
2192
2193async fn pr_title(repo: &Path, pr_url: &str) -> Result<String> {
2195 let out = tokio::process::Command::new("gh")
2196 .args(["pr", "view", pr_url, "--json", "title"])
2197 .current_dir(repo)
2198 .quiet()
2199 .stdin(std::process::Stdio::null())
2200 .output()
2201 .await
2202 .context("spawn gh pr view")?;
2203 if !out.status.success() {
2204 bail!(
2205 "gh pr view {pr_url}: {}",
2206 String::from_utf8_lossy(&out.stderr).trim()
2207 );
2208 }
2209 #[derive(Deserialize)]
2210 struct Title {
2211 title: String,
2212 }
2213 let parsed: Title = serde_json::from_str(&String::from_utf8_lossy(&out.stdout))
2214 .context("parse `gh pr view --json title` output")?;
2215 Ok(parsed.title)
2216}
2217
2218async fn gh_pr_create(
2219 cwd: &Path,
2220 base: &str,
2221 head: &str,
2222 title: &str,
2223 body: &str,
2224) -> Result<String> {
2225 let out = tokio::process::Command::new("gh")
2226 .args([
2227 "pr", "create", "--base", base, "--head", head, "--title", title, "--body", body,
2228 ])
2229 .current_dir(cwd)
2230 .quiet()
2231 .stdin(std::process::Stdio::null())
2232 .output()
2233 .await
2234 .context("spawn gh pr create")?;
2235 if out.status.success() {
2236 Ok(String::from_utf8_lossy(&out.stdout).trim().to_owned())
2237 } else {
2238 bail!(
2239 "gh pr create: {}",
2240 String::from_utf8_lossy(&out.stderr).trim()
2241 )
2242 }
2243}
2244
2245async fn gh_enable_automerge(cwd: &Path, pr_url: &str) -> Result<()> {
2249 let out = tokio::process::Command::new("gh")
2250 .args([
2251 "pr",
2252 "merge",
2253 pr_url,
2254 "--auto",
2255 "--squash",
2256 "--delete-branch",
2257 ])
2258 .current_dir(cwd)
2259 .quiet()
2260 .stdin(std::process::Stdio::null())
2261 .output()
2262 .await
2263 .context("spawn gh pr merge --auto")?;
2264 if out.status.success() {
2265 Ok(())
2266 } else {
2267 bail!(
2268 "gh pr merge --auto: {}",
2269 String::from_utf8_lossy(&out.stderr).trim()
2270 )
2271 }
2272}
2273
2274async fn gh_pr_edit_title(cwd: &Path, pr_url: &str, title: &str) -> Result<()> {
2278 let out = tokio::process::Command::new("gh")
2279 .args(["pr", "edit", pr_url, "--title", title])
2280 .current_dir(cwd)
2281 .quiet()
2282 .stdin(std::process::Stdio::null())
2283 .output()
2284 .await
2285 .context("spawn gh pr edit")?;
2286 if out.status.success() {
2287 Ok(())
2288 } else {
2289 bail!(
2290 "gh pr edit --title: {}",
2291 String::from_utf8_lossy(&out.stderr).trim()
2292 )
2293 }
2294}
2295
2296#[cfg(test)]
2297mod tests {
2298 use super::*;
2299 use crate::notices::Severity;
2300
2301 #[test]
2302 fn github_facing_bump_text_is_english() {
2303 let (title, body) =
2304 release_pr("minor", "adds a flag", "0.37.0", "ab12", "https://x/pull/1");
2305 assert!(title.is_ascii() && body.is_ascii(), "{title}\n{body}");
2306 assert_eq!(title, "chore: release v0.37.0 (minor bump)");
2307 assert!(
2308 body.contains("## Background") && body.contains("## Change"),
2309 "{body}"
2310 );
2311 let p = decision_prompt("s", "i", "d", &[], "0.36.5");
2312 assert!(p.contains(crate::prompt::GITHUB_ENGLISH_HEADING), "{p}");
2313 }
2314 use crate::config::Config;
2315 use crate::land::PrLifecycle;
2316
2317 #[tokio::test]
2323 async fn a_disabled_config_does_nothing() {
2324 let config = Config {
2325 merge: crate::config::Merge {
2326 release_bump: false,
2327 ..crate::config::Merge::default()
2328 },
2329 ..Config::default()
2330 };
2331 let mut state = RunState::new(
2332 PathBuf::from("/no/such/repo"),
2333 "main".to_owned(),
2334 "0000000000000000000000000000000000000000".to_owned(),
2335 "irrelevant".to_owned(),
2336 config,
2337 );
2338 after_merge(&mut state, "https://example.invalid/pull/1")
2339 .await
2340 .expect("a disabled config must return Ok without touching anything");
2341 assert!(
2342 state.events.is_empty(),
2343 "nothing should happen at all, not even a logged event"
2344 );
2345 }
2346
2347 async fn origin_with(files: &[(&str, &str)]) -> (tempfile::TempDir, PathBuf) {
2349 let dir = tempfile::tempdir().unwrap();
2350 let origin = dir.path().join("origin.git");
2351 let repo = dir.path().join("repo");
2352 let o = origin.to_string_lossy().into_owned();
2353 git::git(dir.path(), &["init", "--bare", "-b", "main", &o])
2354 .await
2355 .unwrap();
2356 tokio::fs::create_dir_all(&repo).await.unwrap();
2357 git::git(&repo, &["init", "-b", "main"]).await.unwrap();
2358 git::git(&repo, &["config", "user.name", "test"])
2359 .await
2360 .unwrap();
2361 git::git(&repo, &["config", "user.email", "test@example.com"])
2362 .await
2363 .unwrap();
2364 for (name, body) in files {
2365 tokio::fs::write(repo.join(name), body).await.unwrap();
2366 }
2367 git::git(&repo, &["add", "-A"]).await.unwrap();
2368 git::git(&repo, &["commit", "-m", "init"]).await.unwrap();
2369 git::git(&repo, &["remote", "add", "origin", &o])
2370 .await
2371 .unwrap();
2372 git::git(&repo, &["push", "origin", "main"]).await.unwrap();
2373 (dir, repo)
2374 }
2375
2376 async fn no_pr(_: String) -> Result<bool> {
2377 Ok(false)
2378 }
2379
2380 async fn with_branch(ahead: bool) -> (tempfile::TempDir, PathBuf) {
2383 let (d, repo) = origin_with(&[("f", "x\n")]).await;
2384 git::git(&repo, &["branch", "b"]).await.unwrap();
2385 if ahead {
2386 git::git(&repo, &["checkout", "-q", "b"]).await.unwrap();
2387 git::git(&repo, &["commit", "--allow-empty", "-m", "wip"])
2388 .await
2389 .unwrap();
2390 git::git(&repo, &["checkout", "-q", "main"]).await.unwrap();
2391 }
2392 (d, repo)
2393 }
2394
2395 async fn attempt(repo: &Path, open_pr: bool, fail_after_push: Option<bool>) -> Result<()> {
2396 let wt = repo.parent().unwrap().join("bump");
2397 release_attempt(
2398 repo,
2399 "origin",
2400 "main",
2401 &wt,
2402 "b",
2403 |_| async move { Ok(open_pr) },
2404 |w| async move {
2405 if fail_after_push == Some(true) {
2406 git::push(&w, "origin", "b").await?;
2407 }
2408 if fail_after_push.is_some() {
2409 bail!("cargo build failed");
2410 }
2411 Ok(())
2412 },
2413 )
2414 .await
2415 }
2416
2417 #[tokio::test]
2418 async fn a_stale_ancestor_branch_is_deleted_and_the_attempt_proceeds() {
2419 let (_d, repo) = with_branch(false).await;
2420 attempt(&repo, false, None).await.unwrap();
2421 assert!(git::branch_exists(&repo, "b").await.unwrap());
2423 }
2424
2425 #[tokio::test]
2426 async fn a_stale_branch_holding_a_leftover_worktree_is_reclaimed() {
2427 let (_d, repo) = with_branch(false).await;
2428 let old = repo.parent().unwrap().join("old");
2429 git::git(&repo, &["worktree", "add", &old.to_string_lossy(), "b"])
2430 .await
2431 .unwrap();
2432 attempt(&repo, false, None).await.unwrap();
2433 assert!(!old.exists());
2434 }
2435
2436 #[tokio::test]
2437 async fn a_stale_branch_with_an_unmerged_commit_is_kept_with_a_reason() {
2438 let (_d, repo) = with_branch(true).await;
2439 let e = attempt(&repo, false, None).await.unwrap_err().to_string();
2440 assert!(e.contains("`b`") || e.contains("branch b"), "{e}");
2441 assert!(e.contains("1 commit(s) not in origin/main"), "{e}");
2442 assert!(git::branch_exists(&repo, "b").await.unwrap());
2443 }
2444
2445 #[tokio::test]
2446 async fn a_stale_branch_on_the_remote_is_kept() {
2447 let (_d, repo) = with_branch(false).await;
2448 git::git(&repo, &["push", "origin", "b"]).await.unwrap();
2449 let e = attempt(&repo, false, None).await.unwrap_err().to_string();
2450 assert!(e.contains("branch b") && e.contains("is on origin"), "{e}");
2451 assert!(git::branch_exists(&repo, "b").await.unwrap());
2452 }
2453
2454 #[tokio::test]
2455 async fn a_stale_branch_with_an_open_pull_request_is_kept() {
2456 let (_d, repo) = with_branch(false).await;
2457 let e = attempt(&repo, true, None).await.unwrap_err().to_string();
2458 assert!(e.contains("open pull request"), "{e}");
2459 assert!(git::branch_exists(&repo, "b").await.unwrap());
2460 }
2461
2462 #[tokio::test]
2463 async fn an_unanswerable_pull_request_check_keeps_the_branch() {
2464 let (_d, repo) = with_branch(false).await;
2465 let wt = repo.parent().unwrap().join("bump");
2466 let e = release_attempt(
2467 &repo,
2468 "origin",
2469 "main",
2470 &wt,
2471 "b",
2472 |_| async { bail!("offline") },
2473 |_| async { Ok(()) },
2474 )
2475 .await
2476 .unwrap_err()
2477 .to_string();
2478 assert!(
2479 e.contains("could not check for an open pull request"),
2480 "{e}"
2481 );
2482 assert!(git::branch_exists(&repo, "b").await.unwrap());
2483 }
2484
2485 #[tokio::test]
2486 async fn a_failed_attempt_removes_its_own_worktree_and_branch() {
2487 let (_d, repo) = origin_with(&[("f", "x\n")]).await;
2488 let wt = repo.parent().unwrap().join("bump");
2489 let e = attempt(&repo, false, Some(false)).await.unwrap_err();
2490 assert!(format!("{e:#}").contains("cargo build failed"));
2491 assert!(!wt.exists());
2492 assert!(!git::branch_exists(&repo, "b").await.unwrap());
2493 }
2494
2495 #[tokio::test]
2496 async fn a_pushed_branch_survives_a_failed_attempt_and_the_reason_says_so() {
2497 let (_d, repo) = origin_with(&[("f", "x\n")]).await;
2498 let e = attempt(&repo, false, Some(true)).await.unwrap_err();
2499 let e = format!("{e:#}");
2500 assert!(
2501 e.contains("cargo build failed") && e.contains("pushed to origin"),
2502 "{e}"
2503 );
2504 assert!(git::branch_exists(&repo, "b").await.unwrap());
2505 }
2506
2507 #[tokio::test]
2508 async fn a_branch_left_by_a_half_done_worktree_add_is_removed() {
2509 let (_d, repo) = origin_with(&[("f", "x\n")]).await;
2512 let blocker = repo.parent().unwrap().join("blocker");
2513 tokio::fs::write(&blocker, "file").await.unwrap();
2514 let wt = blocker.join("bump");
2515 let r = release_attempt(
2516 &repo,
2517 "origin",
2518 "main",
2519 &wt,
2520 "b",
2521 |_| async { bail!("offline") },
2522 |_| async { Ok(()) },
2523 )
2524 .await;
2525 assert!(r.is_err());
2526 assert!(!git::branch_exists(&repo, "b").await.unwrap());
2527 }
2528
2529 #[tokio::test]
2530 async fn the_retry_is_taken_at_most_once() {
2531 let (_d, repo) = with_branch(false).await;
2534 let blocker = repo.parent().unwrap().join("blocker");
2535 tokio::fs::write(&blocker, "file").await.unwrap();
2536 let wt = blocker.join("bump");
2537 let r = release_attempt(&repo, "origin", "main", &wt, "b", no_pr, |_| async {
2538 Ok(())
2539 })
2540 .await;
2541 assert!(r.is_err());
2542 assert!(!git::branch_exists(&repo, "b").await.unwrap());
2543 }
2544
2545 #[tokio::test]
2546 async fn base_has_cargo_toml_tells_rust_from_non_rust() {
2547 let (_d, rust) = origin_with(&[("Cargo.toml", "[package]\nversion = \"0.1.0\"\n")]).await;
2548 assert!(base_has_cargo_toml(&rust, "origin", "main").await.unwrap());
2549 let (_d2, other) = origin_with(&[("README.md", "hi\n")]).await;
2550 assert!(!base_has_cargo_toml(&other, "origin", "main").await.unwrap());
2551 assert!(base_has_cargo_toml(&other, "origin", "nope").await.is_err());
2553 }
2554
2555 #[tokio::test]
2556 async fn a_repo_without_cargo_toml_skips_with_one_event_and_no_lock() {
2557 let (_d, repo) = origin_with(&[("README.md", "hi\n")]).await;
2558 let home = tempfile::tempdir().unwrap();
2559 let mut state = RunState::new(
2560 repo.clone(),
2561 "main".to_owned(),
2562 "0000000000000000000000000000000000000000".to_owned(),
2563 "task".to_owned(),
2564 Config::default(),
2565 );
2566 state.candidates.push(crate::run::Candidate {
2567 index: 0,
2568 label: 'A',
2569 agent: "x".to_owned(),
2570 branch: "main".to_owned(),
2571 worktree: repo.clone(),
2572 summary: String::new(),
2573 stat: String::new(),
2574 files: 1,
2575 commits: 1,
2576 empty: false,
2577 failed: None,
2578 verified_noop: None,
2579 duration_ms: 0,
2580 folded: false,
2581 });
2582 state.tally = Some(
2583 serde_json::from_value(serde_json::json!({
2584 "first_choice": {}, "borda": {}, "winner": "A",
2585 "unanimous_initial": true, "deliberated": false,
2586 "changed_votes": 0, "unanimous_final": true,
2587 }))
2588 .unwrap(),
2589 );
2590 after_merge_at(
2591 &mut state,
2592 "https://example.invalid/pull/1",
2593 Some(home.path()),
2594 None,
2595 )
2596 .await
2597 .expect("a non-Rust repository is not an error");
2598 let bumps: Vec<_> = state.events.iter().filter(|e| e.node == "bump").collect();
2599 assert_eq!(bumps.len(), 1, "{:?}", state.events);
2600 assert_eq!(
2601 bumps[0].message,
2602 "release bump: no Cargo.toml on the base branch; release bumps are Rust-only, skipping"
2603 );
2604 assert!(
2605 std::fs::read_dir(home.path()).unwrap().next().is_none(),
2606 "no marker and no lock may be created"
2607 );
2608 }
2609
2610 fn winner_state(repo: &Path, base: &str) -> RunState {
2611 let mut state = RunState::new(
2612 repo.to_path_buf(),
2613 base.to_owned(),
2614 "0000000000000000000000000000000000000000".to_owned(),
2615 "task".to_owned(),
2616 Config::default(),
2617 );
2618 state.candidates.push(crate::run::Candidate {
2619 index: 0,
2620 label: 'A',
2621 agent: "x".to_owned(),
2622 branch: "main".to_owned(),
2623 worktree: repo.to_path_buf(),
2624 summary: String::new(),
2625 stat: String::new(),
2626 files: 1,
2627 commits: 1,
2628 empty: false,
2629 failed: None,
2630 verified_noop: None,
2631 duration_ms: 0,
2632 folded: false,
2633 });
2634 state.tally = Some(
2635 serde_json::from_value(serde_json::json!({
2636 "first_choice": {}, "borda": {}, "winner": "A",
2637 "unanimous_initial": true, "deliberated": false,
2638 "changed_votes": 0, "unanimous_final": true,
2639 }))
2640 .unwrap(),
2641 );
2642 state
2643 }
2644
2645 #[tokio::test]
2646 async fn a_real_failure_without_a_pr_raises_one_notice_and_a_retry_folds_into_it() {
2647 let (_d, repo) =
2648 origin_with(&[("Cargo.toml", "[package]\nname=\"x\"\nversion=\"0.1.0\"\n")]).await;
2649 let home = tempfile::tempdir().unwrap();
2650 let store = Notices::at(home.path().join("notifications"));
2651 let mut state = winner_state(&repo, "nope");
2653 let url = "https://example.invalid/pull/1";
2654 after_merge_at(&mut state, url, Some(home.path()), Some(&store))
2655 .await
2656 .expect_err("an unresolvable base is a failure");
2657 let listed = store.list();
2658 assert_eq!(listed.len(), 1, "{listed:?}");
2659 assert!(
2660 listed[0].message.contains("did not run"),
2661 "{}",
2662 listed[0].message
2663 );
2664 assert!(
2665 listed[0].message.contains("Cargo.toml"),
2666 "{}",
2667 listed[0].message
2668 );
2669 assert!(state.events.iter().any(|e| e.node == "bump"));
2670 after_merge_at(&mut state, url, Some(home.path()), Some(&store))
2671 .await
2672 .expect_err("still failing");
2673 let listed = store.list();
2674 assert_eq!(listed.len(), 1, "{listed:?}");
2675 assert_eq!(listed[0].count, 2);
2676 }
2677
2678 #[tokio::test]
2679 async fn not_eligible_cases_raise_no_notice() {
2680 let home = tempfile::tempdir().unwrap();
2681 let store = Notices::at(home.path().join("notifications"));
2682 let url = "https://example.invalid/pull/1";
2683 let (_d, repo) = origin_with(&[("README.md", "hi\n")]).await;
2685 let mut state = winner_state(&repo, "main");
2686 after_merge_at(&mut state, url, Some(home.path()), Some(&store))
2687 .await
2688 .unwrap();
2689 let mut state = winner_state(&repo, "nope");
2691 state.config.merge.release_bump = false;
2692 after_merge_at(&mut state, url, Some(home.path()), Some(&store))
2693 .await
2694 .unwrap();
2695 let mut state = winner_state(&repo, "nope");
2697 state.tally = None;
2698 after_merge_at(&mut state, url, Some(home.path()), Some(&store))
2699 .await
2700 .unwrap();
2701 assert!(store.list().is_empty(), "{:?}", store.list());
2702 }
2703
2704 #[test]
2705 fn the_no_pr_notice_follows_the_configured_language() {
2706 let dir = tempfile::tempdir().unwrap();
2707 let store = Notices::at(dir.path().join("notifications"));
2708 let mut state = merged_state();
2709 state.config.graph.language = "ja".to_owned();
2710 let (n, _) = surface_problem(&mut state, &store, None, Some("0.2.0"), "boom").unwrap();
2711 assert!(n.message.contains("実行されませんでした"), "{}", n.message);
2712 assert!(n.message.contains("boom") && n.message.contains("v0.2.0"));
2713 }
2714
2715 #[test]
2716 fn a_failed_escalation_points_at_the_pending_pr_without_commenting() {
2717 let dir = tempfile::tempdir().unwrap();
2718 let store = Notices::at(dir.path().join("notifications"));
2719 let mut state = merged_state();
2720 let url = "https://example.invalid/pull/9";
2721 let (n, comment) = surface_problem_in(
2722 &mut state,
2723 &store,
2724 Some(url),
2725 Some("0.3.0"),
2726 "push failed",
2727 true,
2728 )
2729 .unwrap();
2730 assert!(comment.is_none());
2731 assert!(
2732 n.message.contains(url) && n.message.contains("v0.3.0"),
2733 "{}",
2734 n.message
2735 );
2736 assert!(!n.message.contains("open the release pull request by hand"));
2737 assert!(matches!(n.link, Some(Link::Url { .. })));
2738 }
2739
2740 #[test]
2741 fn a_failure_notice_names_the_pending_pr_and_the_first_line_of_the_cause() {
2742 let err = anyhow!("outer context").context("cargo build failed\nsecond line");
2743 let with_pr = Progress {
2744 version: Some("0.2.0".into()),
2745 pending_pr: Some("https://example.invalid/pull/9".into()),
2746 };
2747 let (v, reason) = notice_for_failure(&with_pr, &err);
2748 assert_eq!(v.as_deref(), Some("0.2.0"));
2749 assert_eq!(reason, "cargo build failed");
2750 let (v, reason) = notice_for_failure(&Progress::default(), &err);
2751 assert_eq!(v, None);
2752 assert_eq!(reason, "cargo build failed");
2753 }
2754
2755 const NO_RULES: &str = "gh pr merge --auto: GraphQL: Pull request Branch does not have \
2756 required protected branch rules (enablePullRequestAutoMerge)";
2757
2758 fn merged_state() -> RunState {
2759 run::pin_test_home();
2762 let mut s = RunState::new(
2763 PathBuf::from("/no/such/repo"),
2764 "main".to_owned(),
2765 "0000000000000000000000000000000000000000".to_owned(),
2766 "task".to_owned(),
2767 Config::default(),
2768 );
2769 s.status = RunStatus::Merged;
2770 s
2771 }
2772
2773 #[test]
2774 fn the_known_automerge_refusal_names_branch_protection() {
2775 assert!(automerge_hint(NO_RULES).contains("branch protection with required"));
2776 let other = automerge_hint("gh: network unreachable");
2777 assert!(!other.contains("branch protection"), "{other}");
2778 let body = automerge_failure_comment(NO_RULES);
2779 assert!(body.contains("enablePullRequestAutoMerge"), "{body}");
2780 assert!(body.contains("Action required"), "{body}");
2781 }
2782
2783 const CLEAN: &str = "gh pr merge --auto: GraphQL: Pull request Pull request is in clean \
2784 status (enablePullRequestAutoMerge)";
2785
2786 #[test]
2787 fn clean_status_refusal_is_matched_narrowly() {
2788 assert!(is_clean_status_refusal(CLEAN));
2789 assert!(!is_clean_status_refusal(NO_RULES));
2790 assert!(!is_clean_status_refusal("gh: network unreachable"));
2791 assert!(!is_clean_status_refusal("Pull request is in clean status"));
2792 assert!(automerge_hint(CLEAN).contains("already green"));
2793 }
2794
2795 #[test]
2796 fn the_direct_merge_argv_matches_the_land_flags() {
2797 let a = bump_merge_argv("https://github.com/o/r/pull/9", "chore: release v1.0.0");
2798 let l = land::merge_argv(9, "chore: release v1.0.0");
2799 assert_eq!(a[..2], l[..2]);
2800 assert_eq!(a[3..], l[3..]);
2801 assert_eq!(a[2], "https://github.com/o/r/pull/9");
2802 }
2803
2804 #[test]
2805 fn a_direct_merge_is_judged_by_the_forge_not_the_exit_code() {
2806 let argv = bump_merge_argv("u", "t");
2807 let merged = |o: &AutomergeOutcome| matches!(o, AutomergeOutcome::MergedDirectly { .. });
2808 assert!(merged(&resolve_direct_merge(CLEAN, &argv, true, "", None)));
2809 let detached = "not on any branch";
2810 assert!(merged(&resolve_direct_merge(
2811 CLEAN,
2812 &argv,
2813 false,
2814 detached,
2815 Some(PrLifecycle::Merged)
2816 )));
2817 for after in [Some(PrLifecycle::Open), None] {
2818 let o = resolve_direct_merge(CLEAN, &argv, false, "boom", after);
2819 match o {
2820 AutomergeOutcome::Failed { reason } => {
2821 assert!(
2822 reason.contains("clean status") && reason.contains("boom"),
2823 "{reason}"
2824 )
2825 }
2826 other => panic!("expected Failed, got {other:?}"),
2827 }
2828 }
2829 }
2830
2831 #[test]
2832 fn a_local_mode_bump_reads_back_and_old_records_are_not_local() {
2833 let old: run::ReleaseBump =
2834 serde_json::from_str(r#"{"pr_url":"u","automerge_enabled":true}"#).unwrap();
2835 assert!(!old.local);
2836 let mut state = merged_state();
2837 state.release_bump = Some(run::ReleaseBump {
2838 pr_url: Some("https://github.com/o/r/pull/9".to_owned()),
2839 version: Some("1.0.0".to_owned()),
2840 local: true,
2841 ..run::ReleaseBump::default()
2842 });
2843 assert!(!state.needs_attention());
2845 let text = crate::report::run(&state);
2846 assert!(text.contains("release.mode = local"), "{text}");
2847 assert!(!text.contains("FAILED"), "{text}");
2848 }
2849
2850 #[test]
2851 fn a_directly_merged_bump_is_not_reported_as_pending_or_failed() {
2852 let mut state = merged_state();
2853 state.release_bump = Some(run::ReleaseBump {
2854 pr_url: Some("https://github.com/o/r/pull/9".to_owned()),
2855 version: Some("1.0.0".to_owned()),
2856 merged_directly: true,
2857 ..run::ReleaseBump::default()
2858 });
2859 assert!(!state.needs_attention());
2860 let text = crate::report::run(&state);
2861 assert!(text.contains("merged directly"), "{text}");
2862 assert!(!text.contains("FAILED"), "{text}");
2863 }
2864
2865 #[test]
2866 fn an_automerge_failure_is_recorded_shown_and_filed_and_survives_settling() {
2867 let dir = tempfile::tempdir().unwrap();
2868 let store = Notices::at(dir.path().join("notifications"));
2869 let questions = ask::Questions::at(dir.path().join("questions"));
2870 let mut state = merged_state();
2871 let url = "https://github.com/o/r/pull/35";
2872
2873 let (n, comment) =
2874 surface_problem(&mut state, &store, Some(url), Some("0.8.0"), NO_RULES).unwrap();
2875
2876 let comment = comment.expect("a PR was opened, so it gets a comment");
2878 assert!(comment.contains("branch protection"), "{comment}");
2879
2880 assert_eq!(state.status, RunStatus::Merged);
2882 assert!(state.needs_attention());
2883 let text = crate::report::run(&state);
2884 assert!(text.contains("release bump"), "{text}");
2885 assert!(text.contains("FAILED"), "{text}");
2886 assert!(text.contains(url), "{text}");
2887 assert!(text.contains("action required"), "{text}");
2888 assert!(crate::report::line(&state).contains("release needs a human"));
2889
2890 assert_eq!(n.severity, Severity::Error);
2892 assert_eq!(
2893 n.link,
2894 Some(Link::Url {
2895 url: url.to_owned()
2896 })
2897 );
2898 assert_eq!(store.list().len(), 1);
2899 assert!(questions.open_for(&state.id).is_empty());
2900
2901 surface_problem(&mut state, &store, Some(url), Some("0.8.0"), NO_RULES).unwrap();
2903 let listed = store.list();
2904 assert_eq!(listed.len(), 1);
2905 assert_eq!(listed[0].count, 2);
2906 }
2907
2908 #[test]
2909 fn a_bump_that_never_ran_is_surfaced_without_a_pr_comment() {
2910 let dir = tempfile::tempdir().unwrap();
2911 let store = Notices::at(dir.path().join("notifications"));
2912 let mut state = merged_state();
2913 let (n, comment) = surface_problem(&mut state, &store, None, None, "no agent").unwrap();
2914 assert!(comment.is_none());
2915 assert!(matches!(n.link, Some(Link::Run { .. })));
2916 assert!(state.needs_attention());
2917 }
2918
2919 #[test]
2920 fn version_parses_and_bumps_each_digit() {
2921 let v = Version::parse("0.4.0").unwrap();
2922 assert_eq!(
2923 v,
2924 Version {
2925 major: 0,
2926 minor: 4,
2927 patch: 0
2928 }
2929 );
2930
2931 assert_eq!(v.bump(BumpLevel::Major).to_string(), "1.0.0");
2932 assert_eq!(v.bump(BumpLevel::Minor).to_string(), "0.5.0");
2933 assert_eq!(v.bump(BumpLevel::Patch).to_string(), "0.4.1");
2934 }
2935
2936 #[test]
2937 fn version_tolerates_a_prerelease_suffix_on_patch() {
2938 let v = Version::parse("1.2.3-rc1").unwrap();
2939 assert_eq!(
2940 v,
2941 Version {
2942 major: 1,
2943 minor: 2,
2944 patch: 3
2945 }
2946 );
2947 }
2948
2949 #[test]
2950 fn version_rejects_garbage() {
2951 assert!(Version::parse("not-a-version").is_err());
2952 assert!(Version::parse("1.2").is_err());
2953 }
2954
2955 #[test]
2956 fn decision_parses_each_level() {
2957 for (json, level) in [
2958 (
2959 r#"{"level":"major","reason":"drops a config key"}"#,
2960 BumpLevel::Major,
2961 ),
2962 (
2963 r#"{"level":"minor","reason":"adds a new flag"}"#,
2964 BumpLevel::Minor,
2965 ),
2966 (
2967 r#"{"level":"patch","reason":"fixes a race"}"#,
2968 BumpLevel::Patch,
2969 ),
2970 ] {
2971 let decision = parse_decision(json).unwrap();
2972 assert_eq!(decision.level, level);
2973 assert!(!decision.reason.is_empty());
2974 }
2975 }
2976
2977 #[test]
2978 fn decision_wrapped_in_a_fence_and_prose_still_parses() {
2979 let text = "Here is my call.\n\n```json\n{\"level\":\"minor\",\"reason\":\"new HTTP route\"}\n```\n\nDone.";
2980 let decision = parse_decision(text).unwrap();
2981 assert_eq!(decision.level, BumpLevel::Minor);
2982 assert_eq!(decision.reason, "new HTTP route");
2983 }
2984
2985 #[test]
2986 fn a_broken_reply_is_an_error_not_a_default() {
2987 assert!(parse_decision("I decline to answer.").is_err());
2988 assert!(parse_decision(r#"{"level":"huge","reason":"go big"}"#).is_err());
2989 assert!(
2990 parse_decision(r#"{"level":"patch","reason":""}"#).is_err(),
2991 "an empty reason must not pass either"
2992 );
2993 assert!(
2994 parse_decision(r#"{"level":"patch"}"#).is_err(),
2995 "a reply with no reason at all must not pass"
2996 );
2997 }
2998
2999 #[test]
3000 fn prompt_states_the_zero_x_rule_and_the_tie_break() {
3001 let prompt = decision_prompt(
3002 "feat: add a phone endpoint",
3003 "add POST /api/widgets",
3004 "1 file changed, 10 insertions(+)",
3005 &["src/web.rs".to_owned()],
3006 "0.8.0",
3007 );
3008 assert!(prompt.contains("0.8.0"), "the current version is stated");
3009 assert!(
3010 prompt.contains("below `1.0.0`")
3011 && prompt.contains("`minor` is the digit that carries a breaking change"),
3012 "the 0.x rule must be explicit: {prompt}"
3013 );
3014 assert!(
3015 prompt.contains("choose the larger"),
3016 "the tie-break toward the bigger digit must be explicit: {prompt}"
3017 );
3018 }
3019
3020 #[test]
3021 fn release_only_diffs_are_recognised() {
3022 assert!(is_release_only(&["Cargo.toml".to_owned()]));
3023 assert!(is_release_only(&[
3024 "Cargo.toml".to_owned(),
3025 "Cargo.lock".to_owned()
3026 ]));
3027 assert!(!is_release_only(&[]));
3028 assert!(!is_release_only(&[
3029 "Cargo.toml".to_owned(),
3030 "src/main.rs".to_owned()
3031 ]));
3032 }
3033
3034 #[test]
3035 fn cargo_version_rewrite_touches_only_the_package_table() {
3036 let toml = "\
3037[package]\n\
3038# a comment mentioning version on purpose\n\
3039name = \"magi-cli\"\n\
3040version = \"0.8.0\"\n\
3041edition = \"2024\"\n\
3042\n\
3043[dependencies]\n\
3044foo = { version = \"1.2.3\" }\n";
3045 let out = rewrite_cargo_version(toml, "0.9.0").unwrap();
3046 assert!(out.contains("version = \"0.9.0\""));
3047 assert!(
3048 out.contains("foo = { version = \"1.2.3\" }"),
3049 "a dependency's own version pin must survive: {out}"
3050 );
3051 assert!(
3052 out.contains("# a comment mentioning version on purpose"),
3053 "unrelated lines, comments included, must be byte-for-byte preserved: {out}"
3054 );
3055 assert_eq!(
3056 out.lines().count(),
3057 toml.lines().count(),
3058 "the rewrite replaces one line, it does not add or remove any"
3059 );
3060 }
3061
3062 #[test]
3063 fn cargo_version_rewrite_fails_without_a_package_table() {
3064 let toml = "[dependencies]\nfoo = \"1\"\n";
3065 assert!(rewrite_cargo_version(toml, "1.0.0").is_err());
3066 }
3067
3068 #[test]
3074 fn cargo_version_rewrite_falls_back_to_workspace_package_without_a_package_table() {
3075 let toml = "\
3076[workspace]\n\
3077members = [\"crates/a\", \"crates/b\"]\n\
3078\n\
3079[workspace.package]\n\
3080version = \"0.45.18\"\n\
3081edition = \"2024\"\n\
3082\n\
3083[workspace.dependencies]\n\
3084foo = { version = \"1.2.3\" }\n";
3085 let out = rewrite_cargo_version(toml, "0.45.19").unwrap();
3086 assert!(out.contains("version = \"0.45.19\""));
3087 assert!(
3088 out.contains("foo = { version = \"1.2.3\" }"),
3089 "a workspace dependency's own version pin must survive: {out}"
3090 );
3091 assert_eq!(
3092 out.lines().count(),
3093 toml.lines().count(),
3094 "the rewrite replaces one line, it does not add or remove any"
3095 );
3096 }
3097
3098 #[test]
3102 fn cargo_version_rewrite_bumps_an_internal_workspace_dependency_pin() {
3103 let toml = "\
3104[workspace]\n\
3105members = [\"crates/kanade-shared\"]\n\
3106\n\
3107[workspace.package]\n\
3108version = \"0.48.2\"\n\
3109\n\
3110[workspace.dependencies]\n\
3111kanade-shared = { path = \"crates/kanade-shared\", version = \"0.48.2\" }\n";
3112 let out = rewrite_cargo_version(toml, "0.48.3").unwrap();
3113 assert!(out.contains("[workspace.package]\nversion = \"0.48.3\"\n"));
3114 assert!(
3115 out.contains(
3116 "kanade-shared = { path = \"crates/kanade-shared\", version = \"0.48.3\" }"
3117 ),
3118 "the internal pin must move with the workspace version: {out}"
3119 );
3120 }
3121
3122 #[test]
3124 fn cargo_version_rewrite_bumps_an_internal_pin_with_version_before_path() {
3125 let toml = "\
3126[workspace.package]\n\
3127version = \"1.0.0\"\n\
3128\n\
3129[workspace.dependencies]\n\
3130inner = { version = \"1.0.0\", path = \"crates/inner\" }\n";
3131 let out = rewrite_cargo_version(toml, "1.0.1").unwrap();
3132 assert!(out.contains("inner = { version = \"1.0.1\", path = \"crates/inner\" }"));
3133 }
3134
3135 #[test]
3138 fn cargo_version_rewrite_bumps_an_internal_pin_in_dotted_table_form() {
3139 let toml = "\
3140[workspace.package]\n\
3141version = \"2.3.0\"\n\
3142\n\
3143[workspace.dependencies.inner]\n\
3144path = \"crates/inner\"\n\
3145version = \"2.3.0\"\n";
3146 let out = rewrite_cargo_version(toml, "2.4.0").unwrap();
3147 assert!(out.contains("[workspace.package]\nversion = \"2.4.0\"\n"));
3148 assert!(out.contains(
3149 "[workspace.dependencies.inner]\npath = \"crates/inner\"\nversion = \"2.4.0\"\n"
3150 ));
3151 }
3152
3153 #[test]
3158 fn cargo_version_rewrite_bumps_an_internal_pin_in_dotted_key_form() {
3159 let toml = "\
3160[workspace.package]\n\
3161version = \"2.3.0\"\n\
3162\n\
3163[workspace.dependencies]\n\
3164inner.path = \"crates/inner\"\n\
3165inner.version = \"2.3.0\"\n";
3166 let out = rewrite_cargo_version(toml, "2.4.0").unwrap();
3167 assert!(out.contains("[workspace.package]\nversion = \"2.4.0\"\n"));
3168 assert!(out.contains("inner.path = \"crates/inner\"\ninner.version = \"2.4.0\"\n"));
3169 }
3170
3171 #[test]
3174 fn cargo_version_rewrite_leaves_a_path_only_workspace_dependency_untouched() {
3175 let toml = "\
3176[workspace.package]\n\
3177version = \"0.1.0\"\n\
3178\n\
3179[workspace.dependencies]\n\
3180internal-only = { path = \"crates/internal-only\" }\n";
3181 let out = rewrite_cargo_version(toml, "0.2.0").unwrap();
3182 assert!(out.contains("internal-only = { path = \"crates/internal-only\" }"));
3183 }
3184
3185 #[test]
3189 fn cargo_version_rewrite_leaves_an_external_dependency_untouched() {
3190 let toml = "\
3191[workspace.package]\n\
3192version = \"0.1.0\"\n\
3193\n\
3194[workspace.dependencies]\n\
3195serde = { version = \"1\", features = [\"derive\"] }\n\
3196inner = { path = \"crates/inner\", version = \"0.1.0\" }\n";
3197 let out = rewrite_cargo_version(toml, "0.2.0").unwrap();
3198 assert!(out.contains("serde = { version = \"1\", features = [\"derive\"] }"));
3199 assert!(out.contains("inner = { path = \"crates/inner\", version = \"0.2.0\" }"));
3200 }
3201
3202 #[test]
3206 fn cargo_version_rewrite_without_workspace_dependencies_table_still_bumps_package() {
3207 let toml = "[workspace.package]\nversion = \"0.9.0\"\nedition = \"2024\"\n";
3208 let out = rewrite_cargo_version(toml, "0.10.0").unwrap();
3209 assert_eq!(
3210 out,
3211 "[workspace.package]\nversion = \"0.10.0\"\nedition = \"2024\"\n"
3212 );
3213 }
3214
3215 #[test]
3218 fn cargo_version_rewrite_handles_a_kanade_shaped_workspace_dependencies_table() {
3219 let toml = "\
3220[workspace.package]\n\
3221version = \"0.48.2\"\n\
3222\n\
3223[workspace.dependencies]\n\
3224anyhow = { version = \"1\" }\n\
3225serde = { version = \"1\", features = [\"derive\"] }\n\
3226kanade-shared = { path = \"crates/kanade-shared\", version = \"0.48.2\" }\n\
3227kanade-core = { path = \"crates/kanade-core\", version = \"0.48.2\" }\n\
3228kanade-internal-tool = { path = \"crates/kanade-internal-tool\" }\n";
3229 let out = rewrite_cargo_version(toml, "0.48.3").unwrap();
3230 assert!(out.contains("anyhow = { version = \"1\" }"));
3231 assert!(out.contains("serde = { version = \"1\", features = [\"derive\"] }"));
3232 assert!(
3233 out.contains(
3234 "kanade-shared = { path = \"crates/kanade-shared\", version = \"0.48.3\" }"
3235 )
3236 );
3237 assert!(
3238 out.contains("kanade-core = { path = \"crates/kanade-core\", version = \"0.48.3\" }")
3239 );
3240 assert!(out.contains("kanade-internal-tool = { path = \"crates/kanade-internal-tool\" }"));
3241 }
3242
3243 #[test]
3247 fn cargo_version_rewrite_bails_on_an_unrepresentable_inline_table() {
3248 let toml = "\
3249[workspace.package]\n\
3250version = \"0.1.0\"\n\
3251\n\
3252[workspace.dependencies]\n\
3253inner = { path = \"crates/inner\",\n\
3254 version = \"0.1.0\" }\n";
3255 assert!(rewrite_cargo_version(toml, "0.2.0").is_err());
3256 }
3257
3258 #[test]
3261 fn cargo_version_rewrite_recognises_a_commented_workspace_dependencies_header() {
3262 let toml = "\
3263[workspace.package]\n\
3264version = \"0.1.0\"\n\
3265\n\
3266[workspace.dependencies] # internal pins\n\
3267inner = { path = \"crates/inner\", version = \"0.1.0\" }\n";
3268 let out = rewrite_cargo_version(toml, "0.2.0").unwrap();
3269 assert!(out.contains("inner = { path = \"crates/inner\", version = \"0.2.0\" }"));
3270 }
3271
3272 #[test]
3275 fn cargo_version_rewrite_bumps_a_quoted_dependency_key() {
3276 let toml = "\
3277[workspace.package]\n\
3278version = \"0.1.0\"\n\
3279\n\
3280[workspace.dependencies]\n\
3281\"inner\" = { path = \"crates/inner\", version = \"0.1.0\" }\n";
3282 let out = rewrite_cargo_version(toml, "0.2.0").unwrap();
3283 assert!(out.contains("\"inner\" = { path = \"crates/inner\", version = \"0.2.0\" }"));
3284 }
3285
3286 #[test]
3290 fn cargo_version_rewrite_bumps_a_literal_string_version_pin() {
3291 let toml = "\
3292[workspace.package]\n\
3293version = \"0.1.0\"\n\
3294\n\
3295[workspace.dependencies]\n\
3296inner = { path = 'crates/inner', version = '0.1.0' }\n";
3297 let out = rewrite_cargo_version(toml, "0.2.0").unwrap();
3298 assert!(out.contains("inner = { path = 'crates/inner', version = '0.2.0' }"));
3299 }
3300
3301 #[test]
3307 fn cargo_version_rewrite_does_not_require_the_whole_file_to_parse() {
3308 let toml = "\
3309[workspace.package]\n\
3310version = \"0.1.0\"\n\
3311\n\
3312[workspace.dependencies]\n\
3313inner = { path = \"crates/inner\", version = \"0.1.0\" }\n\
3314\n\
3315[workspace.package]\n\
3316edition = \"2024\"\n";
3317 let out = rewrite_cargo_version(toml, "0.2.0").unwrap();
3318 assert!(out.contains("inner = { path = \"crates/inner\", version = \"0.2.0\" }"));
3319 assert!(
3320 toml::from_str::<toml::Value>(toml).is_err(),
3321 "the fixture itself must be invalid as a whole file, or this test proves nothing"
3322 );
3323 }
3324
3325 #[test]
3326 fn current_version_prefers_the_package_table_when_both_exist() {
3327 let toml = "[workspace.package]\nversion = \"9.9.9\"\n\n[package]\nversion = \"0.8.0\"\n";
3328 assert_eq!(current_version(toml).unwrap(), "0.8.0");
3329 }
3330
3331 #[test]
3334 fn current_version_falls_back_to_workspace_package_without_a_package_table() {
3335 let toml = "\
3336[workspace]\n\
3337members = [\"crates/a\", \"crates/b\"]\n\
3338\n\
3339[workspace.package]\n\
3340version = \"0.45.18\"\n";
3341 assert_eq!(current_version(toml).unwrap(), "0.45.18");
3342 }
3343
3344 #[test]
3345 fn coalesce_proceeds_with_nothing_pending() {
3346 assert_eq!(coalesce(None, "0.8.0").unwrap(), Coalesce::Proceed);
3347 }
3348
3349 fn test_pending(target_version: &str, level: BumpLevel) -> PendingBump {
3352 PendingBump {
3353 target_version: target_version.to_owned(),
3354 level,
3355 branch: format!("chore/release-v{target_version}"),
3356 pr_url: "https://example.invalid/pull/9".to_owned(),
3357 }
3358 }
3359
3360 #[test]
3361 fn coalesce_skips_while_the_pending_target_is_still_ahead() {
3362 let pending = test_pending("0.9.0", BumpLevel::Minor);
3363 assert_eq!(
3364 coalesce(Some(&pending), "0.8.0").unwrap(),
3365 Coalesce::Skip {
3366 target_version: "0.9.0".to_owned()
3367 }
3368 );
3369 }
3370
3371 #[test]
3372 fn coalesce_treats_a_landed_or_superseded_pending_bump_as_stale() {
3373 let pending = test_pending("0.9.0", BumpLevel::Minor);
3374 assert_eq!(
3376 coalesce(Some(&pending), "0.9.0").unwrap(),
3377 Coalesce::Proceed
3378 );
3379 assert_eq!(
3381 coalesce(Some(&pending), "1.0.0").unwrap(),
3382 Coalesce::Proceed
3383 );
3384 }
3385
3386 #[test]
3387 fn pending_action_escalates_only_for_a_more_severe_decision() {
3388 assert_eq!(
3389 pending_action(BumpLevel::Patch, BumpLevel::Patch),
3390 PendingAction::AlreadyCovered
3391 );
3392 assert_eq!(
3393 pending_action(BumpLevel::Patch, BumpLevel::Minor),
3394 PendingAction::Escalate
3395 );
3396 assert_eq!(
3397 pending_action(BumpLevel::Patch, BumpLevel::Major),
3398 PendingAction::Escalate
3399 );
3400 assert_eq!(
3401 pending_action(BumpLevel::Minor, BumpLevel::Patch),
3402 PendingAction::AlreadyCovered
3403 );
3404 assert_eq!(
3405 pending_action(BumpLevel::Major, BumpLevel::Minor),
3406 PendingAction::AlreadyCovered
3407 );
3408 assert_eq!(
3409 pending_action(BumpLevel::Major, BumpLevel::Major),
3410 PendingAction::AlreadyCovered
3411 );
3412 }
3413
3414 #[test]
3415 fn pr_state_parsing_reads_open_and_not_open() {
3416 assert!(parse_pr_state(r#"{"state":"OPEN"}"#).unwrap());
3417 assert!(!parse_pr_state(r#"{"state":"CLOSED"}"#).unwrap());
3418 assert!(!parse_pr_state(r#"{"state":"MERGED"}"#).unwrap());
3419 }
3420
3421 #[test]
3422 fn a_lock_is_exclusive_until_dropped() {
3423 let dir = tempfile::tempdir().unwrap();
3424 let marker = dir.path().join("bump").join("deadbeefdeadbeef.json");
3425 let first = MarkerLock::acquire(&marker)
3426 .unwrap()
3427 .expect("first attempt takes the lock");
3428 assert!(
3429 MarkerLock::acquire(&marker).unwrap().is_none(),
3430 "a second attempt must be refused while the first holds it"
3431 );
3432 drop(first);
3433 assert!(
3434 MarkerLock::acquire(&marker).unwrap().is_some(),
3435 "dropping the guard releases the lock for the next attempt"
3436 );
3437 }
3438
3439 #[test]
3440 fn a_stale_lock_is_reclaimed() {
3441 let dir = tempfile::tempdir().unwrap();
3442 let marker = dir.path().join("bump").join("deadbeefdeadbeef.json");
3443 let lock_path = marker.with_extension("lock");
3444 std::fs::create_dir_all(lock_path.parent().unwrap()).unwrap();
3445 std::fs::write(&lock_path, b"").unwrap();
3446 let old = std::time::SystemTime::now() - LOCK_STALE_AFTER - Duration::from_secs(1);
3447 std::fs::OpenOptions::new()
3448 .write(true)
3449 .open(&lock_path)
3450 .unwrap()
3451 .set_modified(old)
3452 .unwrap();
3453 assert!(
3454 MarkerLock::acquire(&marker).unwrap().is_some(),
3455 "a lock older than the stale window must be reclaimed rather than block forever"
3456 );
3457 }
3458
3459 #[tokio::test]
3460 async fn a_contended_lock_is_retried_until_the_holder_releases_it() {
3461 let dir = tempfile::tempdir().unwrap();
3462 let marker = dir.path().join("bump").join("deadbeefdeadbeef.json");
3463 let held = MarkerLock::acquire(&marker)
3464 .unwrap()
3465 .expect("seed the contention");
3466 let releaser = tokio::spawn(async move {
3467 tokio::time::sleep(Duration::from_millis(20)).await;
3468 drop(held);
3469 });
3470 let waited =
3471 wait_for_marker_lock_with(&marker, Duration::from_millis(5), Duration::from_secs(5))
3472 .await
3473 .unwrap();
3474 assert!(
3475 waited.is_some(),
3476 "a merge landing behind another's still-running decision must not be dropped - it \
3477 must wait for that decision to finish and then judge against what it left behind"
3478 );
3479 releaser.await.unwrap();
3480 }
3481
3482 #[tokio::test]
3483 async fn a_lock_held_past_the_ceiling_gives_up() {
3484 let dir = tempfile::tempdir().unwrap();
3485 let marker = dir.path().join("bump").join("deadbeefdeadbeef.json");
3486 let _held = MarkerLock::acquire(&marker).unwrap().unwrap();
3487 let waited =
3488 wait_for_marker_lock_with(&marker, Duration::from_millis(2), Duration::from_millis(10))
3489 .await
3490 .unwrap();
3491 assert!(
3492 waited.is_none(),
3493 "a lock genuinely held past the ceiling must eventually give up rather than wait \
3494 forever"
3495 );
3496 }
3497
3498 #[test]
3499 fn level_between_reads_off_the_differing_digit() {
3500 assert_eq!(
3501 level_between(
3502 Version::parse("0.8.0").unwrap(),
3503 Version::parse("1.0.0").unwrap()
3504 ),
3505 Some(BumpLevel::Major)
3506 );
3507 assert_eq!(
3508 level_between(
3509 Version::parse("0.8.0").unwrap(),
3510 Version::parse("0.9.0").unwrap()
3511 ),
3512 Some(BumpLevel::Minor)
3513 );
3514 assert_eq!(
3515 level_between(
3516 Version::parse("0.8.0").unwrap(),
3517 Version::parse("0.8.1").unwrap()
3518 ),
3519 Some(BumpLevel::Patch)
3520 );
3521 assert_eq!(
3522 level_between(
3523 Version::parse("0.8.0").unwrap(),
3524 Version::parse("0.8.0").unwrap()
3525 ),
3526 None
3527 );
3528 }
3529
3530 #[test]
3531 fn open_release_pr_is_found_among_unrelated_pull_requests() {
3532 let json = r#"[
3533 {"url": "https://example.invalid/pull/1", "headRefName": "feat/something"},
3534 {"url": "https://example.invalid/pull/2", "headRefName": "chore/release-v0.9.0"}
3535 ]"#;
3536 let found = parse_open_release_pr(json).unwrap();
3537 assert_eq!(
3538 found,
3539 Some((
3540 "chore/release-v0.9.0".to_owned(),
3541 "https://example.invalid/pull/2".to_owned()
3542 ))
3543 );
3544 }
3545
3546 #[test]
3547 fn no_open_release_pr_reads_as_none_not_an_error() {
3548 let json =
3549 r#"[{"url": "https://example.invalid/pull/1", "headRefName": "feat/something"}]"#;
3550 assert_eq!(parse_open_release_pr(json).unwrap(), None);
3551 assert_eq!(parse_open_release_pr("[]").unwrap(), None);
3552 }
3553
3554 #[test]
3555 fn marker_round_trips_through_disk() {
3556 let dir = tempfile::tempdir().unwrap();
3557 let path = marker_path(dir.path(), Path::new("/repos/magi"));
3558 assert!(read_marker(&path).is_none());
3559
3560 let marker = test_pending("0.9.0", BumpLevel::Patch);
3561 write_marker(&path, &marker).unwrap();
3562 let read_back = read_marker(&path).unwrap();
3563 assert_eq!(read_back.target_version, "0.9.0");
3564 assert_eq!(read_back.level, BumpLevel::Patch);
3565 assert_eq!(read_back.pr_url, marker.pr_url);
3566
3567 clear_marker(&path);
3568 assert!(read_marker(&path).is_none());
3569 }
3570
3571 #[test]
3572 fn different_repos_get_different_marker_files() {
3573 let dir = tempfile::tempdir().unwrap();
3574 let a = marker_path(dir.path(), Path::new("/repos/a"));
3575 let b = marker_path(dir.path(), Path::new("/repos/b"));
3576 assert_ne!(a, b);
3577 }
3578
3579 #[test]
3583 fn a_bump_pull_requests_own_merge_does_not_retrigger() {
3584 let files = vec!["Cargo.toml".to_owned(), "Cargo.lock".to_owned()];
3585 assert!(
3586 is_release_only(&files),
3587 "the bump pull request's own diff must read as release-only"
3588 );
3589 }
3590
3591 #[test]
3592 fn should_release_bump_reads_only_a_merged_status() {
3593 assert!(should_release_bump(RunStatus::Merged));
3594 for other in [RunStatus::Blocked, RunStatus::Ready, RunStatus::Prep] {
3595 assert!(!should_release_bump(other));
3596 }
3597 }
3598
3599 #[test]
3603 fn all_three_merge_paths_report_pr_lifecycle_merged_case_done() {
3604 let pr = land::PrState {
3605 url: "https://github.com/o/r/pull/1".to_owned(),
3606 number: 1,
3607 state: PrLifecycle::Merged,
3608 checks: land::Checks::Green,
3609 failing: Vec::new(),
3610 review_comments: Vec::new(),
3611 blocking: land::Blocking::No,
3612 };
3613 assert_eq!(
3614 land::decide(&pr, 0, 4, Duration::ZERO),
3615 land::Step::Done { merged: true }
3616 );
3617 assert!(should_release_bump(RunStatus::Merged));
3618 }
3619
3620 #[test]
3625 fn all_three_merge_paths_report_pr_lifecycle_merged_case_direct_merge() {
3626 let pr = land::PrState {
3627 url: "https://github.com/o/r/pull/2".to_owned(),
3628 number: 2,
3629 state: PrLifecycle::Open,
3630 checks: land::Checks::Green,
3631 failing: Vec::new(),
3632 review_comments: Vec::new(),
3633 blocking: land::Blocking::No,
3634 };
3635 assert_eq!(land::decide(&pr, 0, 4, Duration::ZERO), land::Step::Merge);
3636 assert!(should_release_bump(RunStatus::Merged));
3639 }
3640
3641 #[test]
3644 fn all_three_merge_paths_report_pr_lifecycle_merged_case_merged_after_all() {
3645 let argv = land::merge_argv(3, "feat: something");
3646 let outcome = land::merged_after_all(
3647 &argv,
3648 "could not determine current branch: not on any branch",
3649 Some(PrLifecycle::Merged),
3650 );
3651 assert!(outcome.is_some(), "the forge's confirmation must win");
3652 assert!(should_release_bump(RunStatus::Merged));
3653
3654 assert!(land::merged_after_all(&argv, "network error", Some(PrLifecycle::Open)).is_none());
3657 assert!(land::merged_after_all(&argv, "network error", None).is_none());
3658 }
3659
3660 #[test]
3662 fn a_close_or_a_give_up_does_not_trigger_a_bump() {
3663 let pr = land::PrState {
3664 url: "https://github.com/o/r/pull/4".to_owned(),
3665 number: 4,
3666 state: PrLifecycle::Closed,
3667 checks: land::Checks::Green,
3668 failing: Vec::new(),
3669 review_comments: Vec::new(),
3670 blocking: land::Blocking::No,
3671 };
3672 assert_eq!(
3673 land::decide(&pr, 0, 4, Duration::ZERO),
3674 land::Step::Done { merged: false }
3675 );
3676 assert!(!should_release_bump(RunStatus::Blocked));
3677 }
3678}