use std::path::{Path, PathBuf};
use std::process::Command;
use std::time::{SystemTime, UNIX_EPOCH};
use anyhow::{bail, Context, Result};
use serde::{Deserialize, Serialize};
pub const RECEIPTS_DIR: &str = "e2e-attestations";
const LEGACY_ATTESTATION: &str = "e2e-attestation.json";
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct Attestation {
pub command: String,
pub ran_at: u64,
pub exit_code: i32,
pub commit: String,
#[serde(default)]
pub branch: String,
}
pub fn branch_slug(branch: &str) -> String {
let mut slug = String::new();
for c in branch.to_lowercase().chars() {
let mapped = if c.is_ascii_lowercase() || c.is_ascii_digit() || c == '.' || c == '_' {
c
} else {
'-'
};
if mapped == '-' && slug.ends_with('-') {
continue;
}
slug.push(mapped);
}
let slug: String = slug.chars().take(80).collect();
let slug = slug.trim_matches(|c| c == '-' || c == '.');
if slug.is_empty() {
"branch".to_string()
} else {
slug.to_string()
}
}
pub(crate) fn current_branch(repo: &Path) -> Result<String> {
git_capture(repo, &["symbolic-ref", "--short", "-q", "HEAD"]).context(
"resolving the current branch — the receipt is keyed by branch, so this \
must run on a checked-out branch (a detached HEAD has none): `git switch <branch>`",
)
}
pub fn attest(repo: &Path, command: &str) -> Result<Attestation> {
let commit = git_capture(repo, &["rev-parse", "HEAD"])
.context("resolving HEAD — `e2e attest` must run inside a git repo with a commit")?;
let branch = current_branch(repo)?;
let status = Command::new("sh")
.arg("-c")
.arg(command)
.current_dir(repo)
.status()
.with_context(|| format!("running e2e command `{command}`"))?;
let exit_code = status.code().unwrap_or(-1);
let ran_at = SystemTime::now()
.duration_since(UNIX_EPOCH)
.map(|d| d.as_secs())
.unwrap_or(0);
let attestation = Attestation {
command: command.to_string(),
ran_at,
exit_code,
commit,
branch: branch.clone(),
};
let dir = repo.join(RECEIPTS_DIR);
std::fs::create_dir_all(&dir).with_context(|| format!("creating {}", dir.display()))?;
for entry in std::fs::read_dir(&dir).with_context(|| format!("reading {}", dir.display()))? {
let path = entry?.path();
if path.extension().is_some_and(|e| e == "json") {
std::fs::remove_file(&path).with_context(|| format!("pruning {}", path.display()))?;
}
}
let path = dir.join(format!("{}.json", branch_slug(&branch)));
let json = serde_json::to_string_pretty(&attestation).context("serializing the receipt")?;
std::fs::write(&path, format!("{json}\n"))
.with_context(|| format!("writing {}", path.display()))?;
git_run(repo, &["add", "-A", "--", RECEIPTS_DIR])?;
if pathspec_matches_tracked(repo, LEGACY_ATTESTATION)? {
git_run(
repo,
&["rm", "-q", "--ignore-unmatch", "--", LEGACY_ATTESTATION],
)?;
}
let message = format!("e2e attestation for {branch}");
git_run(repo, &["commit", "-q", "-m", message.as_str()])?;
Ok(attestation)
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum Verification {
Fresh,
Missing,
}
pub fn verify(repo: &Path) -> Result<Verification> {
verify_scoped(repo, repo)
}
pub fn verify_scoped(repo: &Path, scope: &Path) -> Result<Verification> {
verify_since(repo, scope, None)
}
pub fn verify_since(repo: &Path, scope: &Path, base: Option<&str>) -> Result<Verification> {
verify_extra_scoped(repo, scope, base, &[], &[])
}
pub fn verify_extra_scoped(
repo: &Path,
scope: &Path,
base: Option<&str>,
extra_scopes: &[PathBuf],
excludes: &[PathBuf],
) -> Result<Verification> {
let Some(base) = base else {
return Ok(if has_receipts(repo) {
Verification::Fresh
} else {
Verification::Missing
});
};
validate_scopes(repo, scope, extra_scopes)?;
let mut args: Vec<String> = vec![
"diff".into(),
"--quiet".into(),
format!("{base}...HEAD"),
"--".into(),
relative_pathspec(repo, scope),
];
for extra in extra_scopes {
args.push(format!(":(top){}", extra.display()));
}
args.push(format!(":(exclude){RECEIPTS_DIR}"));
args.push(format!(":(exclude){LEGACY_ATTESTATION}"));
args.push(format!(":(top,exclude,glob)**/{RECEIPTS_DIR}/**"));
args.push(format!(":(top,exclude,glob)**/{LEGACY_ATTESTATION}"));
for exclude in excludes {
args.push(format!(":(top,exclude){}", exclude.display()));
}
let arg_refs: Vec<&str> = args.iter().map(String::as_str).collect();
if !git_diff_changed(repo, &arg_refs)? {
return Ok(Verification::Fresh);
}
let out = git_capture(
repo,
&[
"diff",
"--name-only",
"--diff-filter=ACMRT",
&format!("{base}...HEAD"),
"--",
RECEIPTS_DIR,
],
)?;
Ok(if out.is_empty() {
Verification::Missing
} else {
Verification::Fresh
})
}
fn has_receipts(repo: &Path) -> bool {
let Ok(entries) = std::fs::read_dir(repo.join(RECEIPTS_DIR)) else {
return false;
};
entries
.flatten()
.any(|e| e.path().extension().is_some_and(|ext| ext == "json") && e.path().is_file())
}
fn relative_pathspec(repo: &Path, scope: &Path) -> String {
if scope == repo {
return ".".to_string();
}
match scope.strip_prefix(repo) {
Ok(rel) if !rel.as_os_str().is_empty() => rel.to_string_lossy().into_owned(),
_ => scope.to_string_lossy().into_owned(),
}
}
fn validate_scopes(repo: &Path, scope: &Path, extra_scopes: &[PathBuf]) -> Result<()> {
let scope_spec = relative_pathspec(repo, scope);
if !pathspec_matches_tracked(repo, &scope_spec)? {
bail!(
"e2e verify: --scope `{}` matches no tracked path under `{}` — \
--scope must name `{}` or a directory beneath it that git tracks",
scope.display(),
repo.display(),
repo.display(),
);
}
for extra in extra_scopes {
let extra_spec = format!(":(top){}", extra.display());
if !pathspec_matches_tracked(repo, &extra_spec)? {
bail!(
"e2e verify: --extra-scope `{}` matches no tracked path — \
--extra-scope must name a repo-root-relative directory that git tracks",
extra.display(),
);
}
}
Ok(())
}
fn pathspec_matches_tracked(repo: &Path, pathspec: &str) -> Result<bool> {
let out = Command::new("git")
.args(["ls-files", "--", pathspec])
.current_dir(repo)
.output()
.with_context(|| format!("running `git ls-files -- {pathspec}`"))?;
Ok(out.status.success() && !out.stdout.is_empty())
}
fn git_diff_changed(repo: &Path, args: &[&str]) -> Result<bool> {
let out = Command::new("git")
.args(args)
.current_dir(repo)
.output()
.with_context(|| format!("running `git {}`", args.join(" ")))?;
match out.status.code() {
Some(0) => Ok(false),
Some(1) => Ok(true),
_ => bail!(
"`git {}` failed: {}",
args.join(" "),
String::from_utf8_lossy(&out.stderr).trim()
),
}
}
fn git_capture(repo: &Path, args: &[&str]) -> Result<String> {
let out = Command::new("git")
.args(args)
.current_dir(repo)
.output()
.with_context(|| format!("running `git {}`", args.join(" ")))?;
if !out.status.success() {
bail!(
"`git {}` failed: {}",
args.join(" "),
String::from_utf8_lossy(&out.stderr).trim()
);
}
Ok(String::from_utf8(out.stdout)?.trim().to_string())
}
fn git_run(repo: &Path, args: &[&str]) -> Result<()> {
let status = Command::new("git")
.args(args)
.current_dir(repo)
.status()
.with_context(|| format!("running `git {}`", args.join(" ")))?;
if !status.success() {
bail!("`git {}` failed", args.join(" "));
}
Ok(())
}
#[cfg(test)]
mod tests {
use super::branch_slug;
#[test]
fn slug_lowercases_and_maps_separators() {
assert_eq!(branch_slug("feature/one"), "feature-one");
assert_eq!(branch_slug("Feature/One"), "feature-one");
assert_eq!(
branch_slug("claude/e2e-attestation-conflicts-mrkc1b"),
"claude-e2e-attestation-conflicts-mrkc1b"
);
}
#[test]
fn slug_keeps_dots_and_underscores() {
assert_eq!(branch_slug("v1.2_rc"), "v1.2_rc");
}
#[test]
fn slug_collapses_runs_and_trims_edges() {
assert_eq!(branch_slug("wip//Émil's"), "wip-mil-s");
assert_eq!(branch_slug("--dashes--"), "dashes");
assert_eq!(branch_slug(".hidden."), "hidden");
}
#[test]
fn slug_truncates_to_80() {
let long = "x".repeat(300);
assert_eq!(branch_slug(&long).len(), 80);
}
#[test]
fn slug_never_returns_empty() {
assert_eq!(branch_slug(""), "branch");
assert_eq!(branch_slug("É"), "branch");
}
}