38ceab0913
kei-replay <dna> parses DNA, looks up ledger row, loads task.toml from worktree, re-runs compose_prompt, recomputes body hash, reports match/drift. kei-replay diff <dna-1> <dna-2> flags every changed facet between two DNAs. 6 cubes (main/lib/replay/diff/ledger_lookup), all ≤114 LOC. Direct SQLite access in ledger_lookup.rs (kei-ledger has no lib.rs). v4 schema-compatible (reads id/dna/worktree_path/spec_sha). Tests: 6/6 (≥4 required): happy path, missing DNA, drift detection, diff differing bodies, diff identical, explicit --task override. Workspace Cargo.toml: +kei-replay member. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
114 lines
3.3 KiB
Rust
114 lines
3.3 KiB
Rust
//! kei-replay — CLI dispatcher.
|
|
//!
|
|
//! Commands:
|
|
//! kei-replay <dna> — reconstruct; print task.toml + prompt
|
|
//! kei-replay <dna> --verify — also fail non-zero on body-hash drift
|
|
//! kei-replay diff <a> <b> — compare two DNAs, print facet report
|
|
|
|
use clap::{Parser, Subcommand};
|
|
use kei_replay::{diff, ledger_lookup, replay};
|
|
use std::path::PathBuf;
|
|
use std::process::ExitCode;
|
|
|
|
#[derive(Parser)]
|
|
#[command(
|
|
name = "kei-replay",
|
|
version,
|
|
about = "Reconstruct agent spawn from DNA — replay / verify / diff"
|
|
)]
|
|
struct Cli {
|
|
/// Override ledger DB path (default: $KEI_LEDGER_DB or ~/.claude/agents/ledger.sqlite)
|
|
#[arg(long, global = true)]
|
|
db: Option<PathBuf>,
|
|
|
|
#[command(subcommand)]
|
|
cmd: Cmd,
|
|
}
|
|
|
|
#[derive(Subcommand)]
|
|
enum Cmd {
|
|
/// Reconstruct the spawn for a DNA string.
|
|
Replay {
|
|
/// DNA string: role::caps::scope::body-nonce
|
|
dna: String,
|
|
/// Repo root holding _roles/ and _capabilities/ (default: cwd)
|
|
#[arg(long)]
|
|
kit_root: Option<PathBuf>,
|
|
/// Explicit task.toml path (skips ledger lookup for the file path)
|
|
#[arg(long)]
|
|
task: Option<PathBuf>,
|
|
/// Fail with exit 2 when recomputed body hash differs from DNA.
|
|
#[arg(long)]
|
|
verify: bool,
|
|
},
|
|
/// Diff two DNA strings facet-by-facet.
|
|
Diff { left: String, right: String },
|
|
}
|
|
|
|
fn main() -> ExitCode {
|
|
let cli = Cli::parse();
|
|
match cli.cmd {
|
|
Cmd::Replay { dna, kit_root, task, verify } => {
|
|
run_replay(cli.db, dna, kit_root, task, verify)
|
|
}
|
|
Cmd::Diff { left, right } => run_diff(left, right),
|
|
}
|
|
}
|
|
|
|
fn run_replay(
|
|
db_cli: Option<PathBuf>,
|
|
dna: String,
|
|
kit_root: Option<PathBuf>,
|
|
task: Option<PathBuf>,
|
|
verify: bool,
|
|
) -> ExitCode {
|
|
let db = db_cli.unwrap_or_else(ledger_lookup::default_db_path);
|
|
let kit = kit_root.unwrap_or_else(|| std::env::current_dir().unwrap_or_else(|_| ".".into()));
|
|
let result = replay::replay(&db, &dna, task.as_deref(), &kit);
|
|
let r = match result {
|
|
Ok(r) => r,
|
|
Err(e) => {
|
|
eprintln!("replay failed: {e}");
|
|
return ExitCode::from(1);
|
|
}
|
|
};
|
|
print_replay(&r);
|
|
if verify && !r.body_hash_matches {
|
|
eprintln!(
|
|
"DRIFT: DNA body_hash={} but recomputed={} — task.toml differs from original spawn",
|
|
r.dna.body_hash, r.recomputed_body_hash
|
|
);
|
|
return ExitCode::from(2);
|
|
}
|
|
ExitCode::SUCCESS
|
|
}
|
|
|
|
fn print_replay(r: &replay::Replay) {
|
|
println!("=== task.toml ===");
|
|
print!("{}", r.task_toml_text);
|
|
if !r.task_toml_text.ends_with('\n') {
|
|
println!();
|
|
}
|
|
println!("=== composed prompt ===");
|
|
println!("{}", r.composed_prompt);
|
|
println!("=== integrity ===");
|
|
println!("dna.body_hash = {}", r.dna.body_hash);
|
|
println!("recomputed body_hash = {}", r.recomputed_body_hash);
|
|
println!(
|
|
"match = {}",
|
|
if r.body_hash_matches { "yes" } else { "NO (drift)" }
|
|
);
|
|
}
|
|
|
|
fn run_diff(left: String, right: String) -> ExitCode {
|
|
match diff::diff(&left, &right) {
|
|
Ok(d) => {
|
|
println!("{}", d.render());
|
|
ExitCode::SUCCESS
|
|
}
|
|
Err(e) => {
|
|
eprintln!("diff failed: {e}");
|
|
ExitCode::from(1)
|
|
}
|
|
}
|
|
}
|