1use crate::backlog;
9use crate::cli::{Cli, Command, GrantCommand, QuestionCommand, RevisionCommand, TicketCommand};
10use crate::output::{self, ansi};
11use crate::planning_tui::PlanningOutcome;
12use crate::tail::{self, EventRenderer};
13use anyhow::{anyhow, bail, Context, Result};
14use kranz_engine::backend::AgentBackend;
15use kranz_engine::backend_claude::ClaudeBackend;
16use kranz_engine::config;
17use kranz_engine::control;
18use kranz_engine::corpus_export;
19use kranz_engine::cost;
20use kranz_engine::event_log::{EventLog, LockForce};
21use kranz_engine::mission_catalog;
22use kranz_engine::orchestrator::{MissionEngine, PlanRequest};
23use kranz_engine::paths::MissionPaths;
24use kranz_engine::reducer;
25use kranz_engine::trace_export;
26use kranz_engine::types::{ControlCommand, MissionConfig, MissionState, MissionStatus};
27use std::io::{IsTerminal, Write};
28use std::path::{Path, PathBuf};
29use std::sync::atomic::{AtomicBool, Ordering};
30use std::sync::Arc;
31use std::time::{Duration, SystemTime};
32
33pub async fn run_cli(cli: Cli) -> Result<i32> {
36 let lock_force = cli.lock_force();
37 let repo = match cli.repo {
38 Some(repo) => repo,
39 None => std::env::current_dir().context("cannot determine the current directory")?,
40 };
41 if cli.dangerously_allow_all {
42 print_danger_banner();
43 }
44
45 match cli.command {
46 Command::Licenses => {
47 let mut out = std::io::stdout().lock();
48 out.write_all(include_bytes!("../LICENSE"))?;
49 out.write_all(b"\n\n")?;
50 out.write_all(include_bytes!("../assets/THIRD_PARTY_NOTICES.txt"))?;
51 out.write_all(b"\n\nEmbedded dashboard dependency notices\n")?;
52 out.write_all(include_bytes!(
53 "../assets/dashboard/dist/THIRD_PARTY_NOTICES.txt"
54 ))?;
55 Ok(0)
56 }
57 Command::Init {
58 gates,
59 register,
60 id,
61 display_name,
62 } => {
63 let options = crate::init::InitOptions {
64 gates,
65 registration: register.then_some(crate::init::Registration { id, display_name }),
66 global_config: kranz_engine::paths::global_config(),
67 };
68 let report = crate::init::initialize(&repo, &options)?;
69 print!("{}", crate::init::render(&report));
70 Ok(0)
71 }
72 Command::Plan { goal } => {
73 let cfg = load_config(&repo, cli.dangerously_allow_all)?;
74 cmd_plan(repo, goal, cfg, cli.mission.as_deref(), lock_force)
75 .await
76 .map_err(augment_limit_hint)
77 }
78 Command::Run => {
79 let mission = select_mission(&repo, cli.mission.as_deref())?;
80 cmd_run(repo, mission, lock_force, cli.dangerously_allow_all)
81 .await
82 .map_err(augment_limit_hint)
83 }
84 Command::Status { json } => {
85 let mission = select_mission(&repo, cli.mission.as_deref())?;
86 let state = load_state(&repo, &mission)?;
87 if json {
88 println!("{}", serde_json::to_string_pretty(&state)?);
89 } else {
90 print!("{}", output::render_status(&state));
91 }
92 Ok(0)
93 }
94 Command::SandboxProbe { json } => {
95 let report = kranz_engine::sandbox_windows::probe();
96 if json {
97 println!("{}", serde_json::to_string_pretty(&report)?);
98 } else {
99 print!("{}", report.render_text());
100 }
101 Ok(0)
102 }
103 Command::SandboxPrepare { targets } => {
104 for target in targets {
105 eprintln!(
106 "Preparing AppContainer host metadata: target={}",
107 target.display()
108 );
109 let changed = kranz_engine::sandbox_windows::prepare_appcontainer_host(&target)
110 .map_err(anyhow::Error::msg)?;
111 let result = if changed {
112 "applied"
113 } else {
114 "already prepared"
115 };
116 println!(
117 "AppContainer host preparation {result}: target={} mask=0x00120088 inheritance=none",
118 target.display()
119 );
120 }
121 eprintln!("Preparing AppContainer profile-parent metadata (derived from USERPROFILE)");
125 let (profile_parent, profile_changed) =
126 kranz_engine::sandbox_windows::prepare_appcontainer_profile_parent()
127 .map_err(anyhow::Error::msg)?;
128 let profile_result = if profile_changed {
129 "applied"
130 } else {
131 "already prepared"
132 };
133 println!(
134 "AppContainer profile-parent preparation {profile_result}: target={} mask=0x00120088 inheritance=none",
135 profile_parent.display()
136 );
137 eprintln!("Preparing AppContainer null-device metadata: target=\\Device\\Null");
138 kranz_engine::sandbox_windows::prepare_appcontainer_null_device()
139 .map_err(anyhow::Error::msg)?;
140 println!(
141 "AppContainer null-device preparation applied: target=\\Device\\Null inheritance=none"
142 );
143 Ok(0)
144 }
145 Command::Outcomes {
146 json,
147 all,
148 window_days,
149 } => {
150 if all {
151 let config = kranz_engine::paths::global_config()
154 .ok_or_else(|| anyhow::anyhow!("cannot locate the home directory"))?;
155 let report =
156 crate::merged_costs::assess_all(&config, window_days, chrono::Utc::now());
157 if json {
158 println!("{}", serde_json::to_string_pretty(&report)?);
159 } else {
160 print!("{}", crate::merged_costs::render_org(&report));
161 }
162 } else {
163 let outcomes = kranz_engine::outcomes::compute_outcomes(&repo)?;
164 if json {
165 println!("{}", output::render_outcomes_json(&outcomes)?);
166 } else {
167 print!("{}", output::render_outcomes(&outcomes));
168 }
169 }
170 Ok(0)
171 }
172 Command::EscalationMetrics { json } => {
173 let metrics = kranz_engine::escalation_metrics::compute_escalation_metrics(&repo)?;
174 if json {
175 println!("{}", output::render_escalation_metrics_json(&metrics)?);
176 } else {
177 print!("{}", output::render_escalation_metrics(&metrics));
178 }
179 Ok(0)
180 }
181 Command::Provenance { mission_id, json } => {
182 let mission = select_mission(&repo, mission_id.as_deref().or(cli.mission.as_deref()))?;
183 let chain = kranz_engine::provenance::compute_provenance(&repo, &mission)?;
184 if json {
185 println!("{}", output::render_provenance_json(&chain)?);
186 } else {
187 print!("{}", output::render_provenance(&chain));
188 }
189 Ok(0)
190 }
191 Command::GateScores { gate, json } => {
192 let series = kranz_engine::gate_scores::compute_gate_score_series(&repo, &gate)?;
193 if json {
194 println!("{}", output::render_gate_score_series_json(&series)?);
195 } else {
196 print!("{}", output::render_gate_score_series(&series));
197 }
198 Ok(0)
199 }
200 Command::EvidenceBundle { mission_id, out } => {
201 let mission = select_mission(&repo, mission_id.as_deref().or(cli.mission.as_deref()))?;
202 let out = out.unwrap_or_else(|| PathBuf::from(format!("evidence-bundle-{mission}")));
203 let outcome =
204 kranz_engine::evidence_bundle::export_evidence_bundle(&repo, &mission, &out)?;
205 println!(
206 "evidence bundle for mission {mission} written to {} ({} files; {} resolved artefacts, {} unresolved)",
207 outcome.out_dir.display(),
208 outcome.files_written,
209 outcome.resolved_artefacts,
210 outcome.unresolved_artefacts,
211 );
212 Ok(0)
213 }
214 Command::ExportTraces {
215 mission_id,
216 all,
217 out,
218 } => {
219 let jsonl = if all {
220 cmd_export_traces_all(&repo)
221 } else {
222 let mission =
223 select_mission(&repo, mission_id.as_deref().or(cli.mission.as_deref()))?;
224 cmd_export_traces(&repo, &mission)?
225 };
226 match out {
227 Some(path) => {
231 kranz_engine::trace_export::write_export_output(&path, jsonl.as_bytes())
232 .with_context(|| {
233 format!("writing export-traces output to {}", path.display())
234 })?
235 }
236 None => print!("{jsonl}"),
237 }
238 Ok(0)
239 }
240 Command::ExportCorpus {
241 mission_id,
242 all,
243 out,
244 } => {
245 let jsonl = if all {
246 cmd_export_corpus_all(&repo)
247 } else {
248 let mission =
249 select_mission(&repo, mission_id.as_deref().or(cli.mission.as_deref()))?;
250 cmd_export_corpus(&repo, &mission)?
251 };
252 match out {
253 Some(path) => {
255 kranz_engine::trace_export::write_export_output(&path, jsonl.as_bytes())
256 .with_context(|| {
257 format!("writing export-corpus output to {}", path.display())
258 })?
259 }
260 None => print!("{jsonl}"),
261 }
262 Ok(0)
263 }
264 Command::Pause => {
265 let mission = select_control_mission(&repo, cli.mission.as_deref())?;
266 cmd_pause(&repo, &mission)?;
267 println!("pause queued for mission {mission} (takes effect between worker runs)");
268 if let Some(hint) = control_queue_hint(&repo, &mission) {
269 println!("{hint}");
270 }
271 Ok(0)
272 }
273 Command::Resume => {
274 let mission = select_control_mission(&repo, cli.mission.as_deref())?;
275 cmd_resume(&repo, &mission)?;
276 println!("resume queued for mission {mission} (takes effect between worker runs)");
277 if let Some(hint) = control_queue_hint(&repo, &mission) {
278 println!("{hint}");
279 }
280 Ok(0)
281 }
282 Command::Msg { text, interrupt } => {
283 let mission = select_control_mission(&repo, cli.mission.as_deref())?;
284 cmd_msg(&repo, &mission, &text, interrupt)?;
285 if interrupt {
286 println!(
287 "message queued for mission {mission} with --interrupt: the current worker \
288 run will be aborted (recorded as partial) before the message is injected"
289 );
290 } else {
291 println!(
292 "message queued for mission {mission}; it is processed between worker runs"
293 );
294 }
295 if let Some(hint) = control_queue_hint(&repo, &mission) {
296 println!("{hint}");
297 }
298 Ok(0)
299 }
300 Command::Revise { id, instructions } => {
301 let instructions = instructions.join(" ");
302 cmd_request_revision(&repo, &id, &instructions)?;
303 println!("revision request queued for mission {id}");
304 if let Some(hint) = control_queue_hint(&repo, &id) {
305 println!("{hint}");
306 }
307 Ok(0)
308 }
309 Command::Revision { command } => {
310 match command {
311 RevisionCommand::Approve { id, revision } => {
312 cmd_approve_revision(&repo, &id, revision)?;
313 println!("revision {revision} approval queued for mission {id}");
314 if let Some(hint) = control_queue_hint(&repo, &id) {
315 println!("{hint}");
316 }
317 }
318 RevisionCommand::Reject { id, revision } => {
319 cmd_reject_revision(&repo, &id, revision)?;
320 println!("revision {revision} rejection queued for mission {id}");
321 if let Some(hint) = control_queue_hint(&repo, &id) {
322 println!("{hint}");
323 }
324 }
325 }
326 Ok(0)
327 }
328 Command::Grant { command } => {
329 match command {
330 GrantCommand::Approve { id, command } => {
331 cmd_approve_grant(&repo, &id, &command)?;
332 println!("grant approval for `{command}` queued for mission {id}");
333 if let Some(hint) = control_queue_hint(&repo, &id) {
334 println!("{hint}");
335 }
336 }
337 GrantCommand::Deny {
338 id,
339 command,
340 reason,
341 } => {
342 cmd_deny_grant(&repo, &id, &command, &reason)?;
343 println!("grant denial for `{command}` queued for mission {id}");
344 if let Some(hint) = control_queue_hint(&repo, &id) {
345 println!("{hint}");
346 }
347 }
348 }
349 Ok(0)
350 }
351 Command::Question { command } => {
352 match command {
353 QuestionCommand::List { id } => {
354 print!("{}", cmd_list_questions(&repo, &id)?);
355 }
356 QuestionCommand::Answer {
357 id,
358 question_id,
359 answer,
360 option,
361 } => {
362 cmd_answer_question(&repo, &id, &question_id, &answer, option)?;
363 println!("answer for question {question_id} queued for mission {id}");
364 if let Some(hint) = control_queue_hint(&repo, &id) {
365 println!("{hint}");
366 }
367 }
368 }
369 Ok(0)
370 }
371 Command::Missions => {
372 print!("{}", cmd_missions(&repo)?);
373 Ok(0)
374 }
375 Command::Abandon { id, reason } => {
376 let mission = select_mission(&repo, id.as_deref().or(cli.mission.as_deref()))?;
379 let reason = reason.as_deref().unwrap_or("abandoned by operator");
380 cmd_abandon(&repo, &mission, reason, lock_force)?;
381 println!("mission {mission} ABANDONED ({reason})");
382 Ok(0)
383 }
384 Command::Clean { yes, all } => cmd_clean(&repo, yes, all),
385 Command::Ticket { command } => dispatch_ticket(&repo, command, cli.mission.as_deref()),
386 Command::Draft {
387 slug,
388 yes,
389 from_mission,
390 } => backlog::cmd_draft(
391 repo,
392 &slug,
393 yes,
394 from_mission.as_deref(),
395 cli.dangerously_allow_all,
396 )
397 .await
398 .map_err(augment_limit_hint),
399 Command::Decompose { goal, yes } => {
400 backlog::cmd_decompose(repo, &goal, yes, cli.dangerously_allow_all)
401 .await
402 .map_err(augment_limit_hint)
403 }
404 Command::Exec {
405 file,
406 yes: _,
407 max_cycles,
408 enqueue,
409 enqueue_source,
410 enqueue_external_ref,
411 push,
412 allow_unvalidated,
413 } => crate::exec::cmd_exec(
414 repo,
415 file,
416 crate::exec::ExecOptions {
417 max_cycles,
418 enqueue,
419 enqueue_source: enqueue_source.zip(enqueue_external_ref).map(
420 |(producer, external_ref)| crate::exec::ExternalEnqueueSource {
421 producer,
422 external_ref,
423 },
424 ),
425 push,
426 dangerously_allow_all: cli.dangerously_allow_all,
427 allow_unvalidated,
428 },
429 )
430 .await
431 .map_err(augment_limit_hint),
432 Command::Queue { remove } => match remove {
433 Some(mission_id) => {
434 print!("{}", backlog::cmd_queue_remove(&repo, &mission_id)?);
435 Ok(0)
436 }
437 None => {
438 print!("{}", backlog::cmd_queue(&repo));
439 Ok(0)
440 }
441 },
442 Command::KnowledgeRefresh { json } => cmd_knowledge_refresh(&repo, json),
443 Command::Scan { staged, range } => cmd_scan(&repo, staged, range.as_deref()),
444 Command::DomainLint { seed_config, json } => {
445 cmd_domain_lint(&repo, seed_config.as_deref(), json)
446 }
447 Command::HookGuard { config } => {
448 let mut stdin = std::io::stdin();
449 Ok(crate::hook_guard::run_hook_guard(&config, &mut stdin))
450 }
451 Command::HookStatus { config } => {
452 let mut stdin = std::io::stdin();
453 Ok(crate::hook_status::run_hook_status(
454 &config,
455 &mut stdin,
456 &crate::hook_status::post_signal,
457 )
458 .await)
459 }
460 Command::Ready { json, all } => {
461 if all {
462 let config = kranz_engine::paths::global_config()
463 .ok_or_else(|| anyhow::anyhow!("cannot locate the home directory"))?;
464 let report = crate::ready::assess_all(&config);
465 if json {
466 println!("{}", serde_json::to_string_pretty(&report)?);
467 } else {
468 print!("{}", crate::ready::render_org(&report));
469 }
470 } else {
471 let report = crate::ready::assess(&repo);
472 if json {
473 println!("{}", serde_json::to_string_pretty(&report)?);
474 } else {
475 print!("{}", crate::ready::render(&report));
476 }
477 }
478 Ok(0)
479 }
480 Command::Work { once, expect } => backlog::cmd_work(repo, once, expect)
481 .await
482 .map_err(augment_limit_hint),
483 Command::Serve {
484 port,
485 host,
486 insecure_lan,
487 read_auth,
488 open,
489 dashboard,
490 token,
491 read_token,
492 slack,
493 } => {
494 cmd_serve(
495 repo,
496 host,
497 port,
498 insecure_lan,
499 read_auth,
500 open,
501 dashboard,
502 token,
503 read_token,
504 slack,
505 )
506 .await
507 }
508 Command::Release { url, token } => {
509 let mission = select_mission(&repo, cli.mission.as_deref())?;
510 cmd_release(&repo, &mission, &url, token).await
511 }
512 Command::Config { command } => {
513 crate::config_cmd::cmd_config(&repo, command, cli.mission.as_deref())
514 }
515 Command::Pack { command } => match command {
516 crate::cli::PackCommand::Lint { dir } => cmd_pack_lint(&repo, &dir),
517 },
518 Command::Standards { command } => match command {
519 crate::cli::StandardsCommand::Metrics { json } => {
520 let report = kranz_engine::standards_metrics::compute(&repo)?;
521 if json {
522 println!("{}", serde_json::to_string_pretty(&report)?);
523 } else {
524 print!("{}", crate::output::render_standards_metrics(&report));
525 }
526 Ok(0)
527 }
528 crate::cli::StandardsCommand::Lint { dir, against } => {
529 cmd_standards_lint(&repo, &dir, against.as_deref())
530 }
531 crate::cli::StandardsCommand::Waive {
532 rule,
533 revision,
534 finding,
535 reason,
536 expires,
537 } => {
538 let mission = select_mission(&repo, cli.mission.as_deref())?;
539 cmd_standards_waive(
540 &repo,
541 &mission,
542 &rule,
543 revision,
544 finding.as_deref(),
545 &reason,
546 &expires,
547 lock_force,
548 )
549 }
550 crate::cli::StandardsCommand::Attest { rule, reason } => {
551 let mission = select_mission(&repo, cli.mission.as_deref())?;
552 cmd_standards_attest(&repo, &mission, &rule, &reason, lock_force)
553 }
554 },
555 Command::Otel {
556 endpoint,
557 from_start,
558 } => crate::otel::run_otel(repo, cli.mission.clone(), endpoint, from_start).await,
559 }
560}
561
562fn dispatch_ticket(repo: &Path, command: TicketCommand, mission: Option<&str>) -> Result<i32> {
566 match command {
567 TicketCommand::List => {
568 print!("{}", backlog::cmd_ticket_list(repo));
569 Ok(0)
570 }
571 TicketCommand::Ready { include_deferred } => {
572 print!("{}", backlog::cmd_ticket_ready(repo, include_deferred));
573 Ok(0)
574 }
575 TicketCommand::Show { slug } => {
576 print!("{}", backlog::cmd_ticket_show(repo, &slug)?);
577 Ok(0)
578 }
579 TicketCommand::New { slug, title, goal } => {
580 let path = backlog::cmd_ticket_new(repo, &slug, &title, goal.as_deref())?;
581 println!("created ticket '{slug}' at {}", path.display());
582 Ok(0)
583 }
584 TicketCommand::ImportOpenspec { path, slug } => {
585 let written = crate::openspec::import_change(repo, &path, slug.as_deref())?;
586 println!("imported {} as {}", path.display(), written.display());
587 println!(
588 "acceptance criteria are still prose: replace the placeholder in \
589 '## Acceptance hints' with commands that can fail"
590 );
591 Ok(0)
592 }
593 TicketCommand::Note { slug, text } => {
594 print!(
595 "{}",
596 crate::ticket_notes::cmd_ticket_note(repo, &slug, &text.join(" "))?
597 );
598 Ok(0)
599 }
600 TicketCommand::Notes { slug } => {
601 print!("{}", crate::ticket_notes::cmd_ticket_notes(repo, &slug)?);
602 Ok(0)
603 }
604 TicketCommand::Queue {
605 slug,
606 mission: explicit,
607 force,
608 } => {
609 backlog::cmd_ticket_queue(repo, &slug, explicit.as_deref().or(mission), force)
611 }
612 TicketCommand::Approve {
613 slug,
614 mission: explicit,
615 force,
616 } => {
617 backlog::cmd_ticket_approve(repo, &slug, explicit.as_deref().or(mission), force)
620 }
621 TicketCommand::MigrateState { yes } => backlog::cmd_ticket_migrate_state(repo, yes),
622 }
623}
624
625pub fn load_config(repo: &Path, dangerously_allow_all: bool) -> Result<MissionConfig> {
632 let mut cfg = config::load(repo)?;
633 if dangerously_allow_all {
634 cfg.dangerously_allow_all = true;
635 }
636 config::validate(&cfg)?;
637 Ok(cfg)
638}
639
640pub(crate) fn build_backend(cfg: &MissionConfig) -> Result<Arc<dyn AgentBackend>> {
643 let backend = ClaudeBackend::discover(cfg.claude_binary.as_deref())?;
644 Ok(Arc::new(backend))
645}
646
647pub fn select_mission(repo: &Path, explicit: Option<&str>) -> Result<String> {
650 if let Some(id) = explicit {
651 require_mission(repo, id)?;
652 return Ok(id.to_string());
653 }
654 let ids = MissionPaths::list_missions(repo);
655 match ids.len() {
656 0 => bail!(
657 "no missions found under {}; create one with `kranz plan \"<goal>\"`",
658 repo.join(".kranz").join("missions").display()
659 ),
660 1 => Ok(ids.into_iter().next().expect("len checked")),
661 _ => {
662 let mut best: Option<(SystemTime, String)> = None;
663 for id in ids {
664 let events = MissionPaths::new(repo, &id).events_file();
665 let mtime = std::fs::metadata(&events)
666 .and_then(|m| m.modified())
667 .unwrap_or(SystemTime::UNIX_EPOCH);
668 let newer = match &best {
669 Some((t, _)) => mtime >= *t,
670 None => true,
671 };
672 if newer {
673 best = Some((mtime, id));
674 }
675 }
676 Ok(best.expect("non-empty list").1)
677 }
678 }
679}
680
681pub fn select_control_mission(repo: &Path, explicit: Option<&str>) -> Result<String> {
693 if explicit.is_some() {
694 return Ok(control::resolve_active_mission(repo, explicit)?);
695 }
696 let mission = select_mission(repo, None)?;
697 let status = load_state(repo, &mission)?.mission.status;
698 if mission_catalog::is_terminal_status(status) {
699 bail!(
700 "mission {mission} is {status:?}; control commands apply only to active \
701 missions (a terminal mission's inbox is never drained — see \
702 `kranz missions`)"
703 );
704 }
705 Ok(mission)
706}
707
708pub fn control_queue_hint(repo: &Path, mission_id: &str) -> Option<String> {
713 let paths = MissionPaths::new(repo, mission_id);
714 (!mission_catalog::mission_lock_is_live(&paths)).then(|| {
715 format!(
716 "note: mission {mission_id} is not currently running — the command is \
717 queued and applies when the mission next runs"
718 )
719 })
720}
721
722pub fn select_planning_mission(repo: &Path, explicit: Option<&str>) -> Result<String> {
726 if let Some(id) = explicit {
727 require_mission(repo, id)?;
728 return Ok(id.to_string());
729 }
730 let mut best: Option<(SystemTime, String)> = None;
731 for id in MissionPaths::list_missions(repo) {
732 let Ok(state) = load_state(repo, &id) else {
733 continue; };
735 if state.mission.status != MissionStatus::Planning {
736 continue;
737 }
738 let events = MissionPaths::new(repo, &id).events_file();
739 let mtime = std::fs::metadata(&events)
740 .and_then(|m| m.modified())
741 .unwrap_or(SystemTime::UNIX_EPOCH);
742 if best.as_ref().is_none_or(|(t, _)| mtime >= *t) {
743 best = Some((mtime, id));
744 }
745 }
746 best.map(|(_, id)| id).ok_or_else(|| {
747 anyhow!(
748 "no mission is currently in planning under {} — start one with \
749 `kranz plan \"<goal>\"`",
750 repo.join(".kranz").join("missions").display()
751 )
752 })
753}
754
755pub fn augment_limit_hint(e: anyhow::Error) -> anyhow::Error {
759 let msg = format!("{e:#}").to_ascii_lowercase();
760 if [
761 "session limit",
762 "usage limit",
763 "rate limit",
764 "hit your limit",
765 ]
766 .iter()
767 .any(|s| msg.contains(s))
768 {
769 e.context(
770 "this is your Claude subscription's usage window, not a Kranz failure. \
771 The mission and its conversation are saved: when the limit resets, \
772 `kranz plan` (no goal) resumes planning and `kranz run` resumes execution",
773 )
774 } else {
775 e
776 }
777}
778
779fn require_mission(repo: &Path, mission_id: &str) -> Result<MissionPaths> {
781 if !MissionPaths::is_safe_id(mission_id) {
782 bail!(
783 "invalid mission id '{mission_id}': ids cannot contain path separators, '..', or drive designators"
784 );
785 }
786 let paths = MissionPaths::new(repo, mission_id);
787 if !paths.events_file().is_file() {
788 bail!(
789 "mission '{mission_id}' not found under {} (see `kranz missions`)",
790 paths.missions_dir().display()
791 );
792 }
793 Ok(paths)
794}
795
796pub fn load_state(repo: &Path, mission_id: &str) -> Result<MissionState> {
799 let paths = require_mission(repo, mission_id)?;
800 let events = EventLog::read_events(&paths.events_file())
801 .with_context(|| format!("reading the event log of mission '{mission_id}'"))?;
802 let state = reducer::fold(&events)
803 .with_context(|| format!("folding the event log of mission '{mission_id}'"))?;
804 Ok(state)
805}
806
807pub fn cmd_export_traces(repo: &Path, mission_id: &str) -> Result<String> {
812 let paths = require_mission(repo, mission_id)?;
813 let events = EventLog::read_events(&paths.events_file())
814 .with_context(|| format!("reading the event log of mission '{mission_id}'"))?;
815 let state = reducer::fold(&events)
816 .with_context(|| format!("folding the event log of mission '{mission_id}'"))?;
817 let pairs = trace_export::export_validated_traces(&state, &events);
818 Ok(trace_export::to_jsonl(&pairs))
819}
820
821pub fn cmd_export_traces_all(repo: &Path) -> String {
826 let mut pairs = Vec::new();
827 for mission_id in MissionPaths::list_missions(repo) {
828 let paths = MissionPaths::new(repo, &mission_id);
829 let Ok(events) = EventLog::read_events(&paths.events_file()) else {
830 continue;
831 };
832 let Ok(state) = reducer::fold(&events) else {
833 continue;
834 };
835 pairs.extend(trace_export::export_validated_traces(&state, &events));
836 }
837 trace_export::to_jsonl(&pairs)
838}
839
840pub fn cmd_export_corpus(repo: &Path, mission_id: &str) -> Result<String> {
847 let paths = require_mission(repo, mission_id)?;
848 paths.require_no_follow()?;
849 let events = EventLog::read_events(&paths.events_file())
850 .with_context(|| format!("reading the event log of mission '{mission_id}'"))?;
851 let records = corpus_export::export_corpus(&paths.mission_dir(), mission_id, &events)
852 .with_context(|| format!("deriving the training corpus of mission '{mission_id}'"))?;
853 Ok(corpus_export::to_jsonl(&records))
854}
855
856pub fn cmd_export_corpus_all(repo: &Path) -> String {
862 let mut records = Vec::new();
863 for mission_id in MissionPaths::list_missions(repo) {
864 let paths = MissionPaths::new(repo, &mission_id);
865 if paths.require_no_follow().is_err() {
866 continue;
867 }
868 let Ok(events) = EventLog::read_events(&paths.events_file()) else {
869 continue;
870 };
871 let Ok(mission_records) =
872 corpus_export::export_corpus(&paths.mission_dir(), &mission_id, &events)
873 else {
874 continue;
875 };
876 records.extend(mission_records);
877 }
878 corpus_export::to_jsonl(&records)
879}
880
881pub fn print_danger_banner() {
883 eprintln!(
884 "\n\
885 ============================================================\n\
886 !! --dangerously-allow-all IS SET !!\n\
887 !! !!\n\
888 !! Permission gating is BYPASSED for every agent !!\n\
889 !! session (bypassPermissions). Workers can run ANY !!\n\
890 !! command: file writes, network access, git push, !!\n\
891 !! package publishes, sudo. !!\n\
892 !! !!\n\
893 !! Only use this on a sandboxed, disposable checkout. !!\n\
894 ============================================================\n"
895 );
896}
897
898struct StdinLines {
910 rx: Arc<tokio::sync::Mutex<tokio::sync::mpsc::UnboundedReceiver<String>>>,
911}
912
913impl StdinLines {
914 fn spawn() -> Self {
915 static CHANNEL: std::sync::OnceLock<
916 Arc<tokio::sync::Mutex<tokio::sync::mpsc::UnboundedReceiver<String>>>,
917 > = std::sync::OnceLock::new();
918 let rx = CHANNEL
919 .get_or_init(|| {
920 let (tx, rx) = tokio::sync::mpsc::unbounded_channel();
921 std::thread::spawn(move || {
922 let mut buf = String::new();
923 loop {
924 buf.clear();
925 match std::io::stdin().read_line(&mut buf) {
926 Ok(0) | Err(_) => break, Ok(_) => {
928 let line = buf.trim_end_matches(['\r', '\n']).to_string();
929 if tx.send(line).is_err() {
930 break;
931 }
932 }
933 }
934 }
935 });
936 Arc::new(tokio::sync::Mutex::new(rx))
937 })
938 .clone();
939 StdinLines { rx }
940 }
941
942 async fn next(&mut self) -> Option<String> {
944 self.rx.lock().await.recv().await
945 }
946
947 fn drain(&mut self) -> usize {
949 let Ok(mut rx) = self.rx.try_lock() else {
952 return 0;
953 };
954 let mut n = 0;
955 while rx.try_recv().is_ok() {
956 n += 1;
957 }
958 n
959 }
960
961 fn drain_noisily(&mut self, tty: bool) {
965 if !std::io::stdin().is_terminal() {
966 return;
967 }
968 let n = self.drain();
969 if n > 0 {
970 let (dim, reset) = if tty {
971 (ansi::DIM, ansi::RESET)
972 } else {
973 ("", "")
974 };
975 eprintln!(
976 "{dim}(ignored {n} line(s) typed while the orchestrator was working — \
977 the prompt below wants fresh input){reset}"
978 );
979 }
980 }
981}
982
983async fn cmd_plan(
991 repo: PathBuf,
992 goal: Option<String>,
993 cfg: MissionConfig,
994 explicit_mission: Option<&str>,
995 force_lock: LockForce,
996) -> Result<i32> {
997 let backend = build_backend(&cfg)?;
998 let (mut engine, intro) = match goal {
999 Some(goal) => {
1000 let engine = MissionEngine::create(backend, repo.clone(), &goal, cfg)?;
1001 let intro = format!("mission {} created (planning)", engine.mission_id());
1002 (engine, intro)
1003 }
1004 None => {
1005 let mission = select_planning_mission(&repo, explicit_mission)?;
1006 let engine = MissionEngine::resume(backend, repo.clone(), &mission, force_lock)?;
1007 if engine.state().mission.status != MissionStatus::Planning {
1008 return Err(anyhow!(
1009 "mission {mission} is {:?}, not in planning — use 'kranz run' \
1010 to execute it, or 'kranz plan \"<goal>\"' to start a new mission",
1011 engine.state().mission.status
1012 ));
1013 }
1014 let intro = format!(
1015 "resuming planning for mission {mission} — the conversation continues \
1016 where it left off"
1017 );
1018 (engine, intro)
1019 }
1020 };
1021
1022 if std::io::stdin().is_terminal() && std::io::stdout().is_terminal() {
1028 let mission_id = engine.mission_id().to_string();
1029 return match crate::planning_tui::run(engine, intro).await? {
1033 PlanningOutcome::ApprovedRun => start_run_after_plan(repo, mission_id).await,
1034 PlanningOutcome::ApprovedExit | PlanningOutcome::NotApproved => Ok(0),
1035 };
1036 }
1037 println!("{intro}");
1038 let tty = std::io::stdout().is_terminal();
1039
1040 let color = std::io::stderr().is_terminal();
1043 let stop = Arc::new(AtomicBool::new(false));
1044 let printer = tokio::spawn(tail::tail_events(
1045 engine.paths().events_file(),
1046 engine.state().last_seq,
1047 EventRenderer::planning(engine.state(), color),
1048 Arc::clone(&stop),
1049 ));
1050 let mut approved = false;
1051 let mut run_now = false;
1052 let mut stdin_lines = StdinLines::spawn();
1053
1054 println!("talk to the orchestrator to shape the plan:");
1055 println!(" /plan request the plan + cost estimate and review it for approval");
1056 println!(" /quit exit planning (Ctrl-D works too)");
1057
1058 loop {
1059 stdin_lines.drain_noisily(tty);
1060 if tty {
1061 print!("you> ");
1062 let _ = std::io::stdout().flush();
1063 }
1064 let Some(line) = stdin_lines.next().await else {
1065 break; };
1067 let line = line.trim().to_string();
1068 if line.is_empty() {
1069 continue;
1070 }
1071 match line.as_str() {
1072 "/quit" => break,
1073 "/plan" => {
1074 let request = engine.request_plan().await;
1075 if let Some(seed) = engine.take_seed_reply() {
1078 print_orchestrator_reply(&seed, tty);
1079 }
1080 let plan = match request {
1081 Ok(PlanRequest::Ready(plan)) => plan,
1082 Ok(PlanRequest::NotReady(text)) => {
1083 print_orchestrator_reply(&text, tty);
1086 println!(
1087 "the orchestrator isn't ready to emit the plan yet — answer it \
1088 above, then /plan again."
1089 );
1090 continue;
1091 }
1092 Ok(PlanRequest::WrongPlan { reason }) => {
1093 print_orchestrator_reply(&reason, tty);
1096 println!(
1097 "the orchestrator believes a plan here is likely WRONG — reframe \
1098 the goal or fix the premise above, then /plan again."
1099 );
1100 continue;
1101 }
1102 Err(e) => {
1103 eprintln!(
1104 "kranz: plan request failed: {:#}",
1105 augment_limit_hint(e.into())
1106 );
1107 continue;
1108 }
1109 };
1110 println!("{}", output::render_plan(&plan));
1111 let calibration = cost::calibrate(&repo);
1114 let estimate = cost::estimate(&plan, &engine.state().config, &calibration.params);
1115 let estimate = cost::apply_shape(estimate, &plan, &calibration);
1116 println!(
1117 "{}",
1118 output::render_cost_estimate(&estimate, calibration.missions_used)
1119 );
1120
1121 stdin_lines.drain_noisily(tty);
1122 print!("approve? [y/N] ");
1123 let _ = std::io::stdout().flush();
1124 let answer = stdin_lines.next().await.unwrap_or_default();
1125 if matches!(answer.trim().to_ascii_lowercase().as_str(), "y" | "yes") {
1126 match engine.approve_plan(plan) {
1127 Ok(()) => {
1128 let branch = engine.state().mission.mission_branch.clone();
1129 approved = true;
1130 if std::io::stdin().is_terminal() {
1135 println!("plan approved and committed on {branch}.");
1136 stdin_lines.drain_noisily(tty);
1137 print!("start execution now? [Y/n] ");
1138 let _ = std::io::stdout().flush();
1139 let reply = stdin_lines.next().await;
1140 if run_now_answer(reply.as_deref()) {
1141 run_now = true;
1142 } else {
1143 println!("run 'kranz run' to execute.");
1144 }
1145 } else {
1146 println!(
1147 "plan approved and committed on {branch}. \
1148 run 'kranz run' to execute."
1149 );
1150 }
1151 break;
1152 }
1153 Err(e) => eprintln!("kranz: plan approval failed: {e}"),
1154 }
1155 } else {
1156 println!("not approved — back to the conversation.");
1157 }
1158 }
1159 _ if line.starts_with('/') => {
1160 println!("unknown command {line}; use /plan or /quit");
1161 }
1162 _ => {
1163 let result = engine.planning_turn(&line).await;
1164 if let Some(seed) = engine.take_seed_reply() {
1167 print_orchestrator_reply(&seed, tty);
1168 }
1169 match result {
1170 Ok(reply) => print_orchestrator_reply(&reply, tty),
1171 Err(e) => eprintln!(
1172 "kranz: orchestrator turn failed: {:#}",
1173 augment_limit_hint(e.into())
1174 ),
1175 }
1176 }
1177 }
1178 }
1179 if !approved {
1180 println!(
1181 "leaving planning; mission {} was not approved. Resume anytime with `kranz plan`.",
1182 engine.mission_id()
1183 );
1184 }
1185 let mission_id = engine.mission_id().to_string();
1186 drop(engine);
1189 stop.store(true, Ordering::Relaxed);
1190 let _ = printer.await;
1191 if run_now {
1192 return start_run_after_plan(repo, mission_id).await;
1195 }
1196 Ok(0)
1197}
1198
1199pub fn run_now_answer(answer: Option<&str>) -> bool {
1204 match answer {
1205 None => false,
1206 Some(text) => !matches!(text.trim().to_ascii_lowercase().as_str(), "n" | "no"),
1207 }
1208}
1209
1210async fn start_run_after_plan(repo: PathBuf, mission_id: String) -> Result<i32> {
1214 println!(
1215 "starting mission {mission_id} — live event feed follows \
1216 (Ctrl-C safe; resume with 'kranz run')"
1217 );
1218 run_mission_loop(repo, mission_id, LockForce::No, true).await
1219}
1220
1221fn print_orchestrator_reply(text: &str, tty: bool) {
1223 let prefix = if tty {
1224 format!("{}orchestrator>{} ", ansi::DIM, ansi::RESET)
1225 } else {
1226 "orchestrator> ".to_string()
1227 };
1228 for line in text.lines() {
1229 println!("{prefix}{line}");
1230 }
1231}
1232
1233async fn cmd_run(
1240 repo: PathBuf,
1241 mission: String,
1242 force_lock: LockForce,
1243 dangerously_allow_all: bool,
1244) -> Result<i32> {
1245 if dangerously_allow_all {
1248 let paths = require_mission(&repo, &mission)?;
1249 control::enqueue(
1250 &paths,
1251 &ControlCommand::ConfigChange {
1252 patch: serde_json::json!({ "dangerouslyAllowAll": true }),
1253 },
1254 )?;
1255 }
1256 run_mission_loop(repo, mission, force_lock, true).await
1257}
1258
1259pub(crate) async fn run_mission_loop(
1263 repo: PathBuf,
1264 mission: String,
1265 force_lock: LockForce,
1266 interactive: bool,
1270) -> Result<i32> {
1271 let cfg = load_config(&repo, false)?;
1272 let backend = build_backend(&cfg)?;
1273 run_mission_loop_with_backend(repo, mission, force_lock, interactive, backend).await
1274}
1275
1276async fn run_mission_loop_with_backend(
1280 repo: PathBuf,
1281 mission: String,
1282 force_lock: LockForce,
1283 interactive: bool,
1284 backend: Arc<dyn AgentBackend>,
1285) -> Result<i32> {
1286 let paths = require_mission(&repo, &mission)?;
1287
1288 loop {
1289 let mut engine =
1290 MissionEngine::resume(Arc::clone(&backend), repo.clone(), &mission, force_lock)?;
1291
1292 let color = std::io::stderr().is_terminal();
1294 let renderer = EventRenderer::seeded(engine.state(), color);
1295 let stop = Arc::new(AtomicBool::new(false));
1296 let printer = tokio::spawn(tail::tail_events(
1297 engine.paths().events_file(),
1298 engine.state().last_seq,
1299 renderer,
1300 Arc::clone(&stop),
1301 ));
1302
1303 let run_result = engine.run().await;
1304 drop(engine);
1307 stop.store(true, Ordering::Relaxed);
1308 let _ = printer.await;
1309
1310 let status = run_result?;
1311 if let Err(e) = kranz_engine::work::reconcile_ticket_for_mission(&repo, &mission) {
1315 eprintln!("kranz run: warning: failed to reconcile linked ticket: {e}");
1316 }
1317
1318 match status {
1319 MissionStatus::Complete => {
1320 println!("mission {mission} COMPLETE");
1321 return Ok(0);
1322 }
1323 MissionStatus::Blocked => {
1324 eprintln!(
1325 "\n\
1326 ==================== MILESTONE BLOCKED ====================\n\
1327 A milestone is blocked (fix-cycle cap reached or blocked by\n\
1328 the orchestrator). Inspect it with `kranz status`.\n\
1329 ==========================================================="
1330 );
1331 if interactive && std::io::stdin().is_terminal() && std::io::stdout().is_terminal()
1335 {
1336 print!(
1337 "guidance for the orchestrator (what to do about the block; \
1338 empty line or Ctrl-D exits)\nguidance> "
1339 );
1340 let _ = std::io::stdout().flush();
1341 let mut lines = StdinLines::spawn();
1342 if let Some(text) = lines.next().await {
1343 let text = text.trim().to_string();
1344 if !text.is_empty() {
1345 control::enqueue(
1346 &paths,
1347 &ControlCommand::Msg {
1348 text,
1349 interrupt: false,
1350 },
1351 )?;
1352 println!("guidance queued — resuming the mission…");
1353 continue;
1354 }
1355 }
1356 }
1357 eprintln!(
1358 "unblock later by sending guidance via `kranz msg \"<text>\"` \
1359 and re-running `kranz run`."
1360 );
1361 println!("mission {mission} BLOCKED");
1362 return Ok(2);
1363 }
1364 MissionStatus::Failed => {
1365 println!("mission {mission} FAILED");
1366 return Ok(1);
1367 }
1368 other => {
1369 println!(
1370 "mission {mission} ended as {}",
1371 output::mission_status_label(other)
1372 );
1373 return Ok(1);
1374 }
1375 }
1376 }
1377}
1378
1379pub fn cmd_pause(repo: &Path, mission_id: &str) -> Result<PathBuf> {
1385 let paths = require_mission(repo, mission_id)?;
1386 Ok(control::enqueue(&paths, &ControlCommand::Pause)?)
1387}
1388
1389pub fn cmd_resume(repo: &Path, mission_id: &str) -> Result<PathBuf> {
1391 let paths = require_mission(repo, mission_id)?;
1392 Ok(control::enqueue(&paths, &ControlCommand::Resume)?)
1393}
1394
1395pub fn cmd_msg(repo: &Path, mission_id: &str, text: &str, interrupt: bool) -> Result<PathBuf> {
1397 let paths = require_mission(repo, mission_id)?;
1398 let cmd = ControlCommand::Msg {
1399 text: text.to_string(),
1400 interrupt,
1401 };
1402 Ok(control::enqueue(&paths, &cmd)?)
1403}
1404
1405pub fn cmd_request_revision(repo: &Path, mission_id: &str, instructions: &str) -> Result<PathBuf> {
1407 let instructions = instructions.trim();
1408 if instructions.is_empty() {
1409 bail!("revision instructions must not be empty");
1410 }
1411 let paths = require_revisable_mission(repo, mission_id)?;
1412 Ok(control::enqueue(
1413 &paths,
1414 &ControlCommand::RequestRevision {
1415 instructions: instructions.to_string(),
1416 },
1417 )?)
1418}
1419
1420pub fn cmd_approve_revision(repo: &Path, mission_id: &str, revision: u32) -> Result<PathBuf> {
1422 let paths = require_pending_revision(repo, mission_id, revision)?;
1423 Ok(control::enqueue(
1424 &paths,
1425 &ControlCommand::ApproveRevision { revision },
1426 )?)
1427}
1428
1429pub fn cmd_reject_revision(repo: &Path, mission_id: &str, revision: u32) -> Result<PathBuf> {
1431 let paths = require_pending_revision(repo, mission_id, revision)?;
1432 Ok(control::enqueue(
1433 &paths,
1434 &ControlCommand::RejectRevision { revision },
1435 )?)
1436}
1437
1438pub fn cmd_approve_grant(repo: &Path, mission_id: &str, command: &str) -> Result<PathBuf> {
1440 let paths = require_pending_grant(repo, mission_id, command)?;
1441 Ok(control::enqueue(
1442 &paths,
1443 &ControlCommand::ApproveGrant {
1444 command: command.to_string(),
1445 },
1446 )?)
1447}
1448
1449pub fn cmd_deny_grant(
1451 repo: &Path,
1452 mission_id: &str,
1453 command: &str,
1454 reason: &str,
1455) -> Result<PathBuf> {
1456 let paths = require_pending_grant(repo, mission_id, command)?;
1457 Ok(control::enqueue(
1458 &paths,
1459 &ControlCommand::DenyGrant {
1460 command: command.to_string(),
1461 reason: reason.to_string(),
1462 },
1463 )?)
1464}
1465
1466pub fn cmd_list_questions(repo: &Path, mission_id: &str) -> Result<String> {
1471 let mission_id = control::resolve_active_mission(repo, Some(mission_id))?;
1472 let state = load_state(repo, &mission_id)?;
1473 if state.pending_questions.is_empty() {
1474 return Ok(format!("mission {mission_id} has no open questions\n"));
1475 }
1476 let mut out = String::new();
1477 for q in &state.pending_questions {
1478 out.push_str(&format!(
1479 "{} ({}): {}\n",
1480 q.question_id,
1481 q.feature_id.as_deref().unwrap_or("mission"),
1482 q.text
1483 ));
1484 if q.options.is_empty() {
1485 out.push_str(" free-text answer expected\n");
1486 } else {
1487 for (index, option) in q.options.iter().enumerate() {
1488 out.push_str(&format!(" [{index}] {option}\n"));
1489 }
1490 }
1491 }
1492 Ok(out)
1493}
1494
1495pub fn cmd_answer_question(
1498 repo: &Path,
1499 mission_id: &str,
1500 question_id: &str,
1501 answer: &str,
1502 option: Option<u32>,
1503) -> Result<PathBuf> {
1504 let paths = require_pending_question(repo, mission_id, question_id, option, answer)?;
1505 Ok(control::enqueue(
1506 &paths,
1507 &ControlCommand::AnswerQuestion {
1508 question_id: question_id.to_string(),
1509 answer: answer.to_string(),
1510 option,
1511 },
1512 )?)
1513}
1514
1515fn require_pending_question(
1521 repo: &Path,
1522 mission_id: &str,
1523 question_id: &str,
1524 option: Option<u32>,
1525 answer: &str,
1526) -> Result<MissionPaths> {
1527 let mission_id = control::resolve_active_mission(repo, Some(mission_id))?;
1528 let state = load_state(repo, &mission_id)?;
1529 let Some(pending) = state
1530 .pending_questions
1531 .iter()
1532 .find(|q| q.question_id == question_id)
1533 else {
1534 bail!("mission {mission_id} has no open question '{question_id}'");
1535 };
1536 if let Some(index) = option {
1537 match pending.options.get(index as usize) {
1538 Some(expected) if expected == answer => {}
1539 Some(expected) => bail!(
1540 "answer `{answer}` does not match option {index} (`{expected}`) of question '{question_id}'"
1541 ),
1542 None => bail!(
1543 "question '{question_id}' has no option {index} (it offered {})",
1544 pending.options.len()
1545 ),
1546 }
1547 }
1548 Ok(MissionPaths::new(repo, &mission_id))
1549}
1550
1551fn require_revisable_mission(repo: &Path, mission_id: &str) -> Result<MissionPaths> {
1552 let mission_id = control::resolve_active_mission(repo, Some(mission_id))?;
1553 let state = load_state(repo, &mission_id)?;
1554 if state.mission.status == MissionStatus::Planning {
1555 bail!("mission {mission_id} has no approved plan to revise yet");
1556 }
1557 Ok(MissionPaths::new(repo, &mission_id))
1558}
1559
1560fn require_pending_revision(repo: &Path, mission_id: &str, revision: u32) -> Result<MissionPaths> {
1561 let paths = require_revisable_mission(repo, mission_id)?;
1562 let state = load_state(repo, mission_id)?;
1563 match state.pending_revision {
1564 Some(pending) if pending.revision == revision => Ok(paths),
1565 Some(pending) => bail!(
1566 "mission {mission_id} is awaiting revision {}, not {revision}",
1567 pending.revision
1568 ),
1569 None => bail!("mission {mission_id} has no pending revision"),
1570 }
1571}
1572
1573fn require_pending_grant(repo: &Path, mission_id: &str, command: &str) -> Result<MissionPaths> {
1577 let mission_id = control::resolve_active_mission(repo, Some(mission_id))?;
1578 let state = load_state(repo, &mission_id)?;
1579 match state.pending_grant_request {
1580 Some(pending) if pending.command == command => Ok(MissionPaths::new(repo, &mission_id)),
1581 Some(pending) => bail!(
1582 "mission {mission_id} is awaiting a grant for `{}`, not `{command}`",
1583 pending.command
1584 ),
1585 None => bail!("mission {mission_id} has no pending grant request"),
1586 }
1587}
1588
1589pub fn cmd_knowledge_refresh(repo: &Path, json: bool) -> Result<i32> {
1593 let report = kranz_engine::knowledge::refresh_knowledge(repo);
1594 if json {
1595 println!("{}", serde_json::to_string_pretty(&report)?);
1596 } else if report.findings.is_empty() {
1597 println!("knowledge refresh: no notes");
1598 } else {
1599 for finding in &report.findings {
1600 let labels: Vec<String> = finding
1601 .verdicts
1602 .iter()
1603 .map(|v| match v {
1604 kranz_engine::knowledge::RefreshVerdict::Ok => "ok".into(),
1605 kranz_engine::knowledge::RefreshVerdict::AlreadyStale => "already-stale".into(),
1606 kranz_engine::knowledge::RefreshVerdict::Unverified => "unverified".into(),
1607 kranz_engine::knowledge::RefreshVerdict::InvalidMetadata { field, value } => {
1608 format!(
1609 "invalid-metadata:{field}:{}",
1610 value.as_deref().unwrap_or("missing")
1611 )
1612 }
1613 kranz_engine::knowledge::RefreshVerdict::InvalidCitation { citation } => {
1614 format!("invalid-citation:{citation}")
1615 }
1616 kranz_engine::knowledge::RefreshVerdict::PathMissing { path } => {
1617 format!("path-missing:{path}")
1618 }
1619 kranz_engine::knowledge::RefreshVerdict::PathDrifted { path } => {
1620 format!("path-drifted:{path}")
1621 }
1622 kranz_engine::knowledge::RefreshVerdict::CommandSkipped { command } => {
1623 format!("command-skipped:{command}")
1624 }
1625 kranz_engine::knowledge::RefreshVerdict::ProbeFailed { target, error } => {
1626 format!("probe-failed:{target}:{error}")
1627 }
1628 })
1629 .collect();
1630 println!(
1631 "{} ({}) [{}] {}",
1632 finding.rel_path,
1633 finding.title,
1634 finding.freshness,
1635 labels.join(", ")
1636 );
1637 }
1638 if report.check_needed() {
1639 println!("knowledge refresh: check-needed");
1640 } else {
1641 println!("knowledge refresh: ok");
1642 }
1643 }
1644 Ok(if report.check_needed() { 1 } else { 0 })
1645}
1646
1647pub fn cmd_scan(repo: &Path, staged: bool, range: Option<&str>) -> Result<i32> {
1648 if staged && range.is_some() {
1649 bail!("choose either --staged or --range, not both");
1650 }
1651 let git = kranz_engine::git_ops::GitRepo::open(repo)?;
1652 let diff = if staged {
1653 git.diff_staged()?
1654 } else if let Some(range) = range {
1655 git.diff_range(range)?
1656 } else {
1657 git.diff_range("HEAD")?
1658 };
1659 let allowed = std::fs::read_to_string(repo.join(kranz_engine::scrub::SECRET_ALLOWLIST_PATH))
1660 .ok()
1661 .map(|text| kranz_engine::scrub::read_allowlist_text(&text))
1662 .unwrap_or_default();
1663 let findings = kranz_engine::scrub::filter_allowed(
1664 kranz_engine::scrub::scan_unified_diff(&diff),
1665 &allowed,
1666 );
1667 if findings.is_empty() {
1668 println!("secret scan passed");
1669 Ok(0)
1670 } else {
1671 println!(
1672 "secret scan failed; add a fingerprint to {} only for a reviewed false positive:\n{}",
1673 kranz_engine::scrub::SECRET_ALLOWLIST_PATH,
1674 kranz_engine::scrub::format_findings(&findings)
1675 );
1676 Ok(2)
1677 }
1678}
1679
1680pub fn cmd_domain_lint(repo: &Path, seed_config: Option<&Path>, json: bool) -> Result<i32> {
1691 use kranz_engine::domain_lint as dl;
1692 let config_path = repo.join(dl::DENYLIST_PATH);
1693
1694 if let Some(terms_file) = seed_config {
1695 let terms = std::fs::read_to_string(terms_file)
1696 .with_context(|| format!("read terms file {}", terms_file.display()))?;
1697 let existing = std::fs::read_to_string(&config_path).ok();
1698 let config = dl::seed_config(existing.as_deref(), &terms)?;
1699 if let Some(parent) = config_path.parent() {
1701 std::fs::create_dir_all(parent)
1702 .with_context(|| format!("create {}", parent.display()))?;
1703 }
1704 std::fs::write(&config_path, &config)
1705 .with_context(|| format!("write {}", config_path.display()))?;
1706 let denylist = dl::load_denylist(&config)?;
1707 println!(
1709 "seeded {} ({} hashed terms, {})",
1710 dl::DENYLIST_PATH,
1711 denylist.term_count(),
1712 if existing.is_some() {
1713 "salt preserved"
1714 } else {
1715 "fresh salt"
1716 }
1717 );
1718 warn_if_terms_file_unprotected(repo, terms_file);
1719 return Ok(0);
1720 }
1721
1722 let config_text = std::fs::read_to_string(&config_path).with_context(|| {
1723 format!(
1724 "read {} — seed it with `kranz domain-lint --seed-config <terms-file>` (docs/domain-lint.md)",
1725 dl::DENYLIST_PATH
1726 )
1727 })?;
1728 let denylist = dl::load_denylist(&config_text)?;
1729 let allowed = std::fs::read_to_string(repo.join(dl::ALLOWLIST_PATH))
1730 .ok()
1731 .map(|text| kranz_engine::scrub::read_allowlist_text(&text))
1732 .unwrap_or_default();
1733 let report = dl::lint_tree(repo, &denylist, &allowed)?;
1734
1735 if json {
1736 println!(
1737 "{}",
1738 serde_json::to_string_pretty(&serde_json::json!({
1739 "passed": report.is_clean(),
1740 "filesScanned": report.files_scanned,
1741 "filesSkipped": report.files_skipped,
1742 "findings": report.findings,
1743 }))?
1744 );
1745 }
1746 if report.is_clean() {
1747 if !json {
1748 println!(
1749 "domain lint passed ({} files scanned)",
1750 report.files_scanned
1751 );
1752 }
1753 Ok(0)
1754 } else {
1755 if !json {
1756 println!(
1757 "domain lint failed: {} unwaived hit(s); add a fingerprint to {} only for a reviewed false positive:",
1758 report.findings.len(),
1759 dl::ALLOWLIST_PATH
1760 );
1761 for finding in &report.findings {
1762 println!("{} {}:{}", finding.fingerprint, finding.path, finding.line);
1763 }
1764 }
1765 Ok(1)
1766 }
1767}
1768
1769fn warn_if_terms_file_unprotected(repo: &Path, terms_file: &Path) {
1774 let (Ok(repo), Ok(terms_file)) = (repo.canonicalize(), terms_file.canonicalize()) else {
1775 return;
1776 };
1777 let Ok(relative) = terms_file.strip_prefix(&repo) else {
1778 return; };
1780 let ignored = std::process::Command::new("git")
1781 .args(["check-ignore", "-q", "--"])
1782 .arg(relative)
1783 .current_dir(&repo)
1784 .status()
1785 .map(|status| status.success())
1786 .unwrap_or(true);
1788 if !ignored {
1789 eprintln!(
1790 "warning: {} is inside the repo and NOT gitignored — move it outside the repo or use {} (gitignored)",
1791 relative.display(),
1792 kranz_engine::domain_lint::TERMS_LOCAL_PATH
1793 );
1794 }
1795}
1796
1797fn cmd_pack_lint(repo: &Path, dir: &Path) -> Result<i32> {
1808 let trust = kranz_engine::pack::standards::trust_for_dir(repo, dir);
1809 match kranz_engine::pack::Pack::load_with_trust(dir, trust) {
1810 Ok(Some(pack)) => {
1811 print!("{}", kranz_engine::pack::render_lint(&pack));
1812 Ok(0)
1813 }
1814 Ok(None) => {
1815 println!(
1816 "no pack at {} (no {}) — nothing to lint",
1817 dir.display(),
1818 kranz_engine::pack::PACK_MANIFEST
1819 );
1820 Ok(0)
1821 }
1822 Err(err) => {
1823 eprintln!("invalid pack at {}: {err}", dir.display());
1824 Ok(1)
1825 }
1826 }
1827}
1828
1829fn cmd_standards_lint(repo: &Path, dir: &Path, against: Option<&str>) -> Result<i32> {
1836 let trust = kranz_engine::pack::standards::trust_for_dir(repo, dir);
1837 let pack = match kranz_engine::pack::Pack::load_with_trust(dir, trust) {
1838 Ok(Some(pack)) => pack,
1839 Ok(None) => {
1840 println!(
1841 "no pack at {} (no {}) — nothing to lint",
1842 dir.display(),
1843 kranz_engine::pack::PACK_MANIFEST
1844 );
1845 return Ok(0);
1846 }
1847 Err(err) => {
1848 eprintln!("invalid pack at {}: {err}", dir.display());
1849 return Ok(1);
1850 }
1851 };
1852 let Some(manifest) = &pack.standards else {
1853 println!(
1854 "pack `{}` (schema {}) at {} declares no [standards] root — nothing to lint",
1855 pack.name,
1856 pack.schema,
1857 dir.display()
1858 );
1859 return Ok(0);
1860 };
1861 print!(
1862 "{}",
1863 kranz_engine::pack::standards::render_manifest(manifest, trust)
1864 );
1865 if let Some(refname) = against {
1866 let Some(pack_rel) = kranz_engine::pack::standards::repo_relative_dir(repo, dir) else {
1870 eprintln!(
1871 "--against reads the base pack from tracked git blobs in {}; {} is outside \
1872 the repo — external packs have no base history to compare against (D-A)",
1873 repo.display(),
1874 dir.display()
1875 );
1876 return Ok(1);
1877 };
1878 let git = kranz_engine::git_ops::GitRepo::open(repo)?;
1879 let base = match kranz_engine::pack::standards::load_at_ref(&git, refname, &pack_rel) {
1880 Ok(base) => base,
1881 Err(err) => {
1882 eprintln!("cannot load the base standards at `{refname}`: {err}");
1883 return Ok(1);
1884 }
1885 };
1886 let errors = kranz_engine::pack::standards::check_transitions(base.as_ref(), manifest);
1887 print!(
1888 "{}",
1889 kranz_engine::pack::standards::render_transition_report(
1890 refname,
1891 base.as_ref(),
1892 &errors
1893 )
1894 );
1895 if !errors.is_empty() {
1896 return Ok(1);
1897 }
1898 }
1899 Ok(0)
1900}
1901
1902#[allow(clippy::too_many_arguments)]
1912fn cmd_standards_waive(
1913 repo: &Path,
1914 mission_id: &str,
1915 rule: &str,
1916 revision: Option<u64>,
1917 finding: Option<&str>,
1918 reason: &str,
1919 expires: &str,
1920 force_lock: LockForce,
1921) -> Result<i32> {
1922 let expires_at = match chrono::DateTime::parse_from_rfc3339(expires) {
1923 Ok(parsed) => parsed.with_timezone(&chrono::Utc),
1924 Err(err) => {
1925 eprintln!("waiver refused: --expires must be an RFC 3339 instant: {err}");
1926 return Ok(1);
1927 }
1928 };
1929 let request = kranz_engine::standards_waiver::WaiverRequest {
1930 rule_id: rule.to_string(),
1931 revision,
1932 finding_subject: finding.map(str::to_string),
1933 reason: reason.to_string(),
1934 expires_at,
1935 };
1936 let outcome = match kranz_engine::standards_waiver::approve_standards_waiver(
1937 repo, mission_id, &request, "cli", force_lock,
1938 ) {
1939 Ok(outcome) => outcome,
1940 Err(kranz_engine::error::EngineError::LockHeld(e)) => {
1941 eprintln!(
1942 "waiver refused: an engine still holds mission '{mission_id}'s lock — stop \
1943 the running mission first (a waiver against a live mission would race the \
1944 runner's own appends).\n (underlying: {e})"
1945 );
1946 return Ok(1);
1947 }
1948 Err(e) => {
1949 eprintln!("waiver refused: {e}");
1950 return Ok(1);
1951 }
1952 };
1953 let pinned = &outcome.rule;
1956 println!(
1957 "recorded standards.waiver.approved (seq {})",
1958 outcome.event.seq
1959 );
1960 println!("mission: {mission_id}");
1961 println!(
1962 "rule: {} r{} — {}, {}; checker {}; waivable: {}",
1963 pinned.id,
1964 pinned.revision,
1965 pinned.level,
1966 pinned.effective_status,
1967 pinned.checker.as_deref().unwrap_or("-"),
1968 pinned.waivable
1969 );
1970 println!(" statement: {}", pinned.statement);
1971 println!(
1972 "finding: {} (run {})\n evidence: {}",
1973 outcome.finding_subject, outcome.run_id, outcome.finding_evidence
1974 );
1975 println!(" fingerprint: sha256:{}", outcome.finding_fingerprint);
1976 if pinned.when_paths.is_empty() {
1977 println!(
1978 "affected paths: the whole mission diff ({} path(s)) — the rule is unscoped",
1979 outcome.affected_paths.len()
1980 );
1981 } else if outcome.affected_paths.is_empty() {
1982 println!(
1983 "affected paths: (none — the rule's when-paths match no changed path; the \
1984 waiver binds the empty scoped diff)"
1985 );
1986 } else {
1987 println!("affected paths: {}", outcome.affected_paths.join(", "));
1988 }
1989 println!(
1990 "diff digest: sha256:{} (covers the affected-path diff at the mission branch tip)",
1991 outcome.diff_digest
1992 );
1993 println!(
1994 "approver: {} via cli\nreason: {reason}\nexpires: {}",
1995 kranz_engine::standards_waiver::LOCAL_OPERATOR,
1996 expires_at.to_rfc3339()
1997 );
1998 Ok(0)
1999}
2000
2001fn cmd_standards_attest(
2006 repo: &Path,
2007 mission_id: &str,
2008 rule: &str,
2009 reason: &str,
2010 force_lock: LockForce,
2011) -> Result<i32> {
2012 let record = match kranz_engine::standards_attestation::approve_attestation(
2013 repo, mission_id, rule, reason, "cli", force_lock,
2014 ) {
2015 Ok(record) => record,
2016 Err(kranz_engine::error::EngineError::LockHeld(e)) => {
2017 eprintln!(
2018 "attestation refused: an engine still holds mission '{mission_id}'s lock — stop \
2019 the running mission first.\n (underlying: {e})"
2020 );
2021 return Ok(1);
2022 }
2023 Err(e) => {
2024 eprintln!("attestation refused: {e}");
2025 return Ok(1);
2026 }
2027 };
2028 println!(
2029 "recorded standards.attestation.approved (seq {})",
2030 record.seq
2031 );
2032 println!("mission: {mission_id}");
2033 println!("rule: {} r{}", record.rule_id, record.rule_revision);
2034 if record.paths.is_empty() {
2035 println!("affected paths: (none)");
2036 } else {
2037 println!("affected paths: {}", record.paths.join(", "));
2038 }
2039 println!("diff digest: sha256:{}", record.diff_digest);
2040 println!(
2041 "approver: {} via {}\nreason: {}",
2042 record.approver, record.surface, record.reason
2043 );
2044 Ok(0)
2045}
2046
2047pub fn cmd_missions(repo: &Path) -> Result<String> {
2054 let index_contents =
2055 std::fs::read_to_string(MissionPaths::new(repo, "_").missions_dir().join("index.md"))
2056 .unwrap_or_default();
2057 let mut ids = MissionPaths::list_missions(repo);
2058 for id in mission_catalog::mission_index_ids(&index_contents) {
2059 if !ids.contains(&id) {
2060 ids.push(id);
2061 }
2062 }
2063 ids.sort();
2064 if ids.is_empty() {
2065 return Ok("no missions\n".to_string());
2066 }
2067 let ticket_of: std::collections::HashMap<String, String> =
2070 kranz_engine::ticket::Ticket::list(repo)
2071 .into_iter()
2072 .filter_map(|t| {
2073 kranz_engine::ticket::Ticket::mission_for(repo, &t.slug).map(|m| (m, t.slug))
2074 })
2075 .collect();
2076 let mut out = String::new();
2077 for id in ids {
2078 let ticket = ticket_of
2079 .get(&id)
2080 .map(|s| format!(" [ticket: {s}]"))
2081 .unwrap_or_default();
2082 let paths = MissionPaths::new(repo, &id);
2083 if !paths.events_file().is_file() {
2084 out.push_str(&format!(
2085 "{id} {:<10} deleted mission (no data recorded)\n",
2086 "DELETED"
2087 ));
2088 continue;
2089 }
2090 if let Err(error) = paths.require_no_follow() {
2093 out.push_str(&format!("{id} {:<10} (unreadable: {error})\n", "FAILED"));
2094 continue;
2095 }
2096 match load_state(repo, &id) {
2097 Ok(state) => out.push_str(&format!(
2098 "{id} {:<10} {}{ticket}\n",
2099 output::mission_status_label(state.mission.status),
2100 state.mission.goal
2101 )),
2102 Err(e) => out.push_str(&format!("{id} {:<10} (unreadable: {e:#})\n", "FAILED")),
2103 }
2104 }
2105 Ok(out)
2106}
2107
2108pub fn cmd_abandon(
2116 repo: &Path,
2117 mission_id: &str,
2118 reason: &str,
2119 force_lock: LockForce,
2120) -> Result<()> {
2121 require_mission(repo, mission_id)?;
2122 mission_catalog::abandon_mission(repo, mission_id, reason, force_lock).map_err(|e| {
2123 if matches!(e, kranz_engine::error::EngineError::LockHeld(_)) {
2124 anyhow!(
2125 "cannot abandon mission '{mission_id}' — an engine still holds its lock. \
2126 Stop the running `kranz run` first. If the holder is a crashed leftover, \
2127 pass --force-lock (steals unless the holder is provably alive); a provably \
2128 LIVE holder that you have verified to be a zombie or foreign process \
2129 additionally requires --dangerously-steal-live-lock.\n (underlying: {e})"
2130 )
2131 } else {
2132 anyhow::Error::new(e).context(format!("abandoning mission '{mission_id}'"))
2133 }
2134 })
2135}
2136
2137#[derive(Debug, Clone, PartialEq, Eq)]
2139pub struct CleanEntry {
2140 pub id: String,
2141 pub status_label: String,
2142 pub goal: String,
2143}
2144
2145pub fn select_cleanable(repo: &Path, all: bool) -> Vec<CleanEntry> {
2153 let mut out = Vec::new();
2154 for id in MissionPaths::list_missions(repo) {
2155 let paths = MissionPaths::new(repo, &id);
2156 if mission_catalog::mission_lock_is_live(&paths) {
2158 continue;
2159 }
2160 let Ok(state) = load_state(repo, &id) else {
2161 continue; };
2163 let has_plan = paths.plan_file().is_file();
2164 if mission_catalog::cleanable_class(state.mission.status, has_plan).is_cleaned(all) {
2165 out.push(CleanEntry {
2166 id,
2167 status_label: output::mission_status_label(state.mission.status).to_string(),
2168 goal: state.mission.goal,
2169 });
2170 }
2171 }
2172 out
2173}
2174
2175pub fn render_clean_listing(entries: &[CleanEntry]) -> String {
2178 if entries.is_empty() {
2179 return "nothing to clean\n".to_string();
2180 }
2181 let mut out = String::new();
2182 for e in entries {
2183 out.push_str(&format!("{:<10} {} {}\n", e.status_label, e.id, e.goal));
2184 }
2185 out
2186}
2187
2188fn cmd_clean(repo: &Path, yes: bool, all: bool) -> Result<i32> {
2194 let entries = select_cleanable(repo, all);
2195 if entries.is_empty() {
2196 print!("{}", render_clean_listing(&entries));
2197 return Ok(0);
2198 }
2199
2200 print!("{}", render_clean_listing(&entries));
2201 println!(
2202 "\n{} mission director{} above would be removed.{}",
2203 entries.len(),
2204 if entries.len() == 1 { "y" } else { "ies" },
2205 if all {
2206 ""
2207 } else {
2208 " (Complete missions are kept; pass --all to include them.)"
2209 }
2210 );
2211
2212 if !yes && !confirm_clean()? {
2213 println!("clean aborted; nothing removed.");
2214 return Ok(0);
2215 }
2216
2217 let removed = remove_missions(repo, &entries, true);
2218 println!(
2219 "cleaned {} mission director{}",
2220 removed.len(),
2221 if removed.len() == 1 { "y" } else { "ies" }
2222 );
2223 Ok(0)
2224}
2225
2226pub fn remove_missions(repo: &Path, entries: &[CleanEntry], verbose: bool) -> Vec<String> {
2232 let mut removed = Vec::new();
2233 for e in entries {
2234 let paths = MissionPaths::new(repo, &e.id);
2235 if mission_catalog::mission_lock_is_live(&paths) {
2241 eprintln!("kranz: skipping {} — became live since listing", e.id);
2242 continue;
2243 }
2244 let dir = paths.mission_dir();
2245 match std::fs::remove_dir_all(&dir) {
2246 Ok(()) => {
2247 if verbose {
2248 println!("removed {}", dir.display());
2249 }
2250 mission_catalog::prune_mission_index_file(repo, &e.id);
2251 removed.push(e.id.clone());
2252 }
2253 Err(err) => eprintln!("kranz: could not remove {}: {err}", dir.display()),
2254 }
2255 }
2256 removed
2257}
2258
2259fn confirm_clean() -> Result<bool> {
2262 use std::io::BufRead;
2263 print!("proceed? [y/N] ");
2264 std::io::stdout().flush().ok();
2265 let mut line = String::new();
2266 let n = std::io::stdin().lock().read_line(&mut line)?;
2267 if n == 0 {
2268 return Ok(false); }
2270 Ok(matches!(
2271 line.trim().to_ascii_lowercase().as_str(),
2272 "y" | "yes"
2273 ))
2274}
2275
2276pub(crate) fn refuse_non_loopback_without_insecure_lan(
2286 bind: std::net::IpAddr,
2287 insecure_lan: bool,
2288) -> Result<()> {
2289 if !bind.is_loopback() && !insecure_lan {
2290 anyhow::bail!(
2291 "refusing to bind {bind}: non-loopback binds expose the API on the \
2292 network (reads require the read-only or mutation token; POSTs \
2293 require the mutation token). \
2294 Re-run with `--insecure-lan` if you intentionally trust this \
2295 network (LAN/tailnet), or keep the default `--host 127.0.0.1`."
2296 );
2297 }
2298 Ok(())
2299}
2300
2301pub(crate) fn effective_require_read_token(bind_is_loopback: bool, read_auth: bool) -> bool {
2307 !bind_is_loopback || read_auth
2308}
2309
2310#[allow(clippy::too_many_arguments)]
2320async fn cmd_serve(
2321 repo: PathBuf,
2322 host: String,
2323 port: u16,
2324 insecure_lan: bool,
2325 read_auth: bool,
2326 open: bool,
2327 dashboard: Option<PathBuf>,
2328 token: Option<String>,
2329 read_token: Option<String>,
2330 slack: bool,
2331) -> Result<i32> {
2332 let bind: std::net::IpAddr = host
2333 .parse()
2334 .map_err(|e| anyhow!("--host '{host}' is not an IP address: {e}"))?;
2335 refuse_non_loopback_without_insecure_lan(bind, insecure_lan)?;
2336 if !bind.is_loopback() {
2337 eprintln!(
2338 "WARNING: binding {bind} with --insecure-lan — the API is reachable \
2339 beyond this machine. Reads require the read-only or mutation token; \
2340 POSTs require the mutation token. Use only on a network you trust."
2341 );
2342 }
2343 if read_auth && effective_require_read_token(bind.is_loopback(), read_auth) {
2344 eprintln!(
2345 "--read-auth: GETs and the WS upgrade now require the read-only or \
2346 mutation token, including on loopback; POSTs still require mutation \
2347 authority."
2348 );
2349 }
2350 let listener = kranz_server::bind_listener(bind, port)
2353 .await
2354 .map_err(|e| anyhow!("failed to bind {bind}:{port}: {e}"))?;
2355 let local_addr = listener
2356 .local_addr()
2357 .map_err(|e| anyhow!("failed to read bound address: {e}"))?;
2358 let multi_host = Arc::new(kranz_server::MultiRepoHost::from_global_config(
2362 kranz_engine::paths::global_config().as_deref(),
2363 repo.clone(),
2364 )?);
2365 multi_host.ensure_auto_work_started();
2368
2369 if slack {
2373 if multi_host.uses_operator_catalog() {
2374 let catalog = multi_repo_slack_catalog(&multi_host)?;
2375 tokio::spawn(async move {
2376 let never = std::future::pending::<()>();
2377 if let Err(error) = kranz_slack::serve_slack_catalog(catalog, never).await {
2378 tracing::error!(%error, "slack catalog bridge exited with an error");
2379 }
2380 });
2381 } else {
2382 let context = single_repo_slack_context(&multi_host)?;
2383 let repo_slack = context.root().to_path_buf();
2384 let hosted = context.host().ok_or_else(|| {
2385 anyhow!(
2386 "cannot start Slack bridge for unavailable repository '{}': {}",
2387 context.id(),
2388 context.unavailable_reason().unwrap_or("unavailable")
2389 )
2390 })?;
2391 let bridge_host: kranz_slack::SharedHost =
2392 Arc::new(crate::host_bridge::HostedPlanning(hosted.clone()));
2393 tokio::spawn(async move {
2394 let never = std::future::pending::<()>();
2398 if let Err(e) =
2399 kranz_slack::serve_slack(&repo_slack, Some(bridge_host), never).await
2400 {
2401 tracing::error!(error = %e, "slack bridge exited with an error");
2402 }
2403 });
2404 }
2405 }
2406
2407 let dashboard_assets = resolve_dashboard_assets(&repo, dashboard);
2408 let display_host = match local_addr.ip() {
2411 std::net::IpAddr::V6(v6) => format!("[{v6}]"),
2412 std::net::IpAddr::V4(v4) => v4.to_string(),
2413 };
2414 let url = format!("http://{display_host}:{}/", local_addr.port());
2415 let token = token
2416 .or_else(|| std::env::var("KRANZ_TOKEN").ok())
2417 .unwrap_or_else(kranz_server::generate_token);
2418 let read_token = read_token
2419 .or_else(|| std::env::var("KRANZ_READ_TOKEN").ok())
2420 .unwrap_or_else(kranz_server::generate_token);
2421
2422 println!("kranz server on {url}");
2423 println!("mutation token: {token}");
2424 println!("read token: {read_token} (GETs/WS only — safe for dashboards and agents)");
2425 if open {
2426 println!(
2427 "opening the bare URL; no token rides in the opener's argv. Paste the \
2428 mutation token above when the dashboard asks for one"
2429 );
2430 }
2431 match &dashboard_assets {
2432 Some(DashboardAssets::Embedded) => println!(
2433 "serving embedded dashboard ({})",
2434 crate::embedded_dashboard::EMBEDDED_DASHBOARD_SOURCE
2435 ),
2436 Some(DashboardAssets::Dir(dir)) => println!("serving dashboard from {}", dir.display()),
2437 None => println!(
2438 "no dashboard build found (--dashboard, $KRANZ_DASHBOARD_DIST, \
2439 <repo>/apps/dashboard/dist, installed asset dirs, or the kranz checkout) \
2440 and no embedded dashboard is available; serving API only"
2441 ),
2442 }
2443
2444 let static_assets = dashboard_assets.map(|assets| match assets {
2445 DashboardAssets::Dir(dir) => kranz_server::DashboardStatic::Dir(dir),
2446 DashboardAssets::Embedded => {
2447 kranz_server::DashboardStatic::Embedded(crate::embedded_dashboard::EMBEDDED_DASHBOARD)
2448 }
2449 });
2450
2451 if open {
2452 let url = url.clone();
2477 tokio::spawn(async move {
2478 tokio::time::sleep(Duration::from_millis(600)).await;
2479 open_browser(&url);
2480 });
2481 }
2482
2483 let shutdown = async {
2484 if let Err(e) = tokio::signal::ctrl_c().await {
2485 tracing::error!(error = %e, "failed to install ctrl-c handler");
2486 }
2487 };
2488 let result = serve_multi_with_token_cleanup(
2489 &repo,
2490 multi_host,
2491 listener,
2492 static_assets,
2493 token,
2494 read_token,
2495 read_auth,
2496 shutdown,
2497 )
2498 .await;
2499 match result {
2500 Ok(()) => Ok(0),
2501 Err(e) => Err(anyhow!("server failed: {e}")),
2502 }
2503}
2504
2505fn single_repo_slack_context(
2506 multi_host: &kranz_server::MultiRepoHost,
2507) -> Result<Arc<kranz_server::RepoContext>> {
2508 multi_host
2509 .compatibility_context()
2510 .ok_or_else(|| anyhow!("--slack requires one healthy configured repository"))
2511}
2512
2513fn multi_repo_slack_catalog(
2514 multi_host: &kranz_server::MultiRepoHost,
2515) -> Result<kranz_slack::SlackCatalog> {
2516 let global_config = kranz_engine::paths::global_config()
2517 .ok_or_else(|| anyhow!("cannot locate the operator config directory for Slack affinity"))?;
2518 let affinity_path = global_config
2519 .parent()
2520 .expect("global config has a parent")
2521 .join("slack")
2522 .join("thread-affinity.json");
2523 let repos = multi_host
2524 .contexts()
2525 .map(|context| {
2526 let host = context.host().map(|host| {
2527 Arc::new(crate::host_bridge::HostedPlanning(host.clone()))
2528 as kranz_slack::SharedHost
2529 });
2530 kranz_slack::SlackRepo {
2531 id: context.id().to_string(),
2532 root: context.root().to_path_buf(),
2533 display_name: context
2534 .config()
2535 .display_name
2536 .clone()
2537 .unwrap_or_else(|| context.id().to_string()),
2538 routes: context
2539 .config()
2540 .slack
2541 .channels
2542 .iter()
2543 .map(|route| kranz_slack::SlackRoute {
2544 team_id: route.team.clone(),
2545 channel_id: route.channel.clone(),
2546 })
2547 .collect(),
2548 allow_users: context.config().slack.allow_users.clone(),
2549 available: context.is_healthy(),
2550 host,
2551 unavailable_reason: context.unavailable_reason().map(str::to_string),
2552 default: multi_host.default_repo() == Some(context.id()),
2553 }
2554 })
2555 .collect();
2556 kranz_slack::SlackCatalog::new(repos, affinity_path)
2557}
2558
2559#[allow(clippy::too_many_arguments)]
2560async fn serve_multi_with_token_cleanup(
2561 repo: &Path,
2562 multi_host: Arc<kranz_server::MultiRepoHost>,
2563 listener: tokio::net::TcpListener,
2564 static_assets: Option<kranz_server::DashboardStatic>,
2565 token: String,
2566 read_token: String,
2567 read_auth: bool,
2568 shutdown: impl std::future::Future<Output = ()> + Send + 'static,
2569) -> anyhow::Result<()> {
2570 let (token_file, read_token_file) = if multi_host.uses_operator_catalog() {
2571 let address = listener
2572 .local_addr()
2573 .context("cannot resolve the bound address for operator token storage")?;
2574 let token_file = write_operator_serve_token(address, &token)
2575 .context("cannot securely store the operator serve token")?;
2576 let read_token_file = write_operator_serve_read_token(address, &read_token)
2577 .context("cannot securely store the operator serve read token")?;
2578 (token_file, read_token_file)
2579 } else {
2580 let token_file =
2581 write_serve_token(repo, &token).context("cannot securely store the serve token")?;
2582 let read_token_file = write_serve_read_token(repo, &read_token)
2583 .context("cannot securely store the serve read token")?;
2584 (token_file, read_token_file)
2585 };
2586 let result = kranz_server::serve_multi_on_listener(
2587 multi_host,
2588 listener,
2589 static_assets,
2590 kranz_server::MutationAuthority::new(token)?,
2591 Some(read_token),
2592 read_auth,
2593 shutdown,
2594 )
2595 .await;
2596 remove_token_file(&token_file);
2597 remove_token_file(&read_token_file);
2598 result
2599}
2600
2601#[cfg(test)]
2607async fn serve_with_token_cleanup(
2608 repo: &Path,
2609 host: Arc<kranz_server::MissionHost>,
2610 listener: tokio::net::TcpListener,
2611 static_assets: Option<kranz_server::DashboardStatic>,
2612 token: String,
2613 read_token: String,
2614 shutdown: impl std::future::Future<Output = ()> + Send + 'static,
2615) -> anyhow::Result<()> {
2616 let token_file = write_serve_token(repo, &token)?;
2620 let read_token_file = write_serve_read_token(repo, &read_token)?;
2621
2622 let result = kranz_server::serve_on_listener(
2623 host,
2624 listener,
2625 static_assets,
2626 kranz_server::MutationAuthority::new(token)?,
2627 shutdown,
2628 )
2629 .await;
2630
2631 remove_token_file(&token_file);
2632 remove_token_file(&read_token_file);
2633
2634 result
2635}
2636
2637fn write_serve_token(repo: &Path, token: &str) -> std::io::Result<PathBuf> {
2643 write_token_file(&repo.join(".kranz").join("serve.token"), token)
2644}
2645
2646fn write_serve_read_token(repo: &Path, read_token: &str) -> std::io::Result<PathBuf> {
2652 write_token_file(&repo.join(".kranz").join("serve.read.token"), read_token)
2653}
2654
2655fn write_operator_serve_token(
2660 address: std::net::SocketAddr,
2661 token: &str,
2662) -> std::io::Result<PathBuf> {
2663 let global = kranz_engine::paths::global_config().ok_or_else(|| {
2664 std::io::Error::new(
2665 std::io::ErrorKind::NotFound,
2666 "cannot resolve operator config directory",
2667 )
2668 })?;
2669 let path = operator_serve_token_path(&global, address);
2670 write_token_file(&path, token)
2671}
2672
2673fn write_operator_serve_read_token(
2675 address: std::net::SocketAddr,
2676 read_token: &str,
2677) -> std::io::Result<PathBuf> {
2678 let global = kranz_engine::paths::global_config().ok_or_else(|| {
2679 std::io::Error::new(
2680 std::io::ErrorKind::NotFound,
2681 "cannot resolve operator config directory",
2682 )
2683 })?;
2684 let path = operator_serve_read_token_path(&global, address);
2685 write_token_file(&path, read_token)
2686}
2687
2688fn operator_serve_read_token_path(global_config: &Path, address: std::net::SocketAddr) -> PathBuf {
2690 operator_serve_token_path(global_config, address).with_extension("read.token")
2691}
2692
2693fn operator_serve_token_path(global_config: &Path, address: std::net::SocketAddr) -> PathBuf {
2694 let endpoint = match address.ip() {
2695 std::net::IpAddr::V4(ip) => format!("v4-{:08x}-{}", u32::from(ip), address.port()),
2696 std::net::IpAddr::V6(ip) => format!("v6-{:032x}-{}", u128::from(ip), address.port()),
2697 };
2698 global_config
2699 .parent()
2700 .unwrap_or_else(|| Path::new("."))
2701 .join("serve")
2702 .join(format!("{endpoint}.token"))
2703}
2704
2705fn legacy_operator_serve_token_path(global_config: &Path, port: u16) -> PathBuf {
2709 global_config
2710 .parent()
2711 .unwrap_or_else(|| Path::new("."))
2712 .join("serve")
2713 .join(format!("{port}.token"))
2714}
2715
2716fn write_token_file(path: &Path, token: &str) -> std::io::Result<PathBuf> {
2717 let dir = path.parent().ok_or_else(|| {
2718 std::io::Error::new(std::io::ErrorKind::InvalidInput, "token path has no parent")
2719 })?;
2720 std::fs::create_dir_all(dir)?;
2721 let tmp = dir.join(format!(
2726 ".{}.tmp-{}",
2727 path.file_name()
2728 .and_then(|name| name.to_str())
2729 .unwrap_or("serve.token"),
2730 uuid::Uuid::new_v4().as_simple()
2731 ));
2732 let write_tmp = || -> std::io::Result<()> {
2733 let mut options = std::fs::OpenOptions::new();
2734 options.create_new(true).write(true);
2735 #[cfg(unix)]
2736 {
2737 use std::os::unix::fs::OpenOptionsExt;
2738 options.mode(0o600);
2739 }
2740 let mut file = options.open(&tmp)?;
2741 file.write_all(token.as_bytes())?;
2742 file.flush()?;
2743 Ok(())
2744 };
2745 if let Err(error) = write_tmp() {
2746 let _ = std::fs::remove_file(&tmp);
2747 return Err(error);
2748 }
2749 #[cfg(unix)]
2750 {
2751 use std::os::unix::fs::PermissionsExt;
2752 std::fs::set_permissions(&tmp, std::fs::Permissions::from_mode(0o600))?;
2753 }
2754 if let Err(error) = std::fs::rename(&tmp, path) {
2755 let _ = std::fs::remove_file(&tmp);
2756 return Err(error);
2757 }
2758 Ok(path.to_path_buf())
2759}
2760
2761fn remove_token_file(path: &Path) {
2762 if let Err(err) = std::fs::remove_file(path) {
2763 if err.kind() != std::io::ErrorKind::NotFound {
2764 eprintln!("kranz: could not remove {}: {err}", path.display());
2765 }
2766 }
2767}
2768
2769#[derive(Debug, Clone, PartialEq, Eq)]
2770enum DashboardAssets {
2771 Dir(PathBuf),
2772 Embedded,
2773}
2774
2775#[derive(Debug, Clone, Default)]
2776struct DashboardResolutionInputs {
2777 env_dist: Option<PathBuf>,
2778 home: Option<PathBuf>,
2779 exe: Option<PathBuf>,
2780 manifest_dir: Option<PathBuf>,
2781 embedded_available: bool,
2782}
2783
2784impl DashboardResolutionInputs {
2785 fn runtime() -> Self {
2786 Self {
2787 env_dist: std::env::var_os("KRANZ_DASHBOARD_DIST").map(PathBuf::from),
2788 home: std::env::var_os(if cfg!(windows) { "USERPROFILE" } else { "HOME" })
2789 .map(PathBuf::from),
2790 exe: std::env::current_exe().ok(),
2791 manifest_dir: Some(PathBuf::from(env!("CARGO_MANIFEST_DIR"))),
2792 embedded_available: !crate::embedded_dashboard::EMBEDDED_DASHBOARD.is_empty(),
2793 }
2794 }
2795}
2796
2797fn resolve_dashboard_assets(repo: &Path, explicit: Option<PathBuf>) -> Option<DashboardAssets> {
2798 resolve_dashboard_assets_from(repo, explicit, &DashboardResolutionInputs::runtime())
2799}
2800
2801fn resolve_dashboard_assets_from(
2809 repo: &Path,
2810 explicit: Option<PathBuf>,
2811 inputs: &DashboardResolutionInputs,
2812) -> Option<DashboardAssets> {
2813 if let Some(d) = explicit {
2814 return Some(DashboardAssets::Dir(d));
2817 }
2818
2819 if let Some(d) = first_dashboard_dir(dashboard_dir_candidates(repo, inputs)) {
2820 return Some(DashboardAssets::Dir(d));
2821 }
2822
2823 inputs
2824 .embedded_available
2825 .then_some(DashboardAssets::Embedded)
2826}
2827
2828pub fn resolve_dashboard_dist(repo: &Path, explicit: Option<PathBuf>) -> Option<PathBuf> {
2831 match resolve_dashboard_assets(repo, explicit) {
2832 Some(DashboardAssets::Dir(dir)) => Some(dir),
2833 Some(DashboardAssets::Embedded) | None => None,
2834 }
2835}
2836
2837fn dashboard_dir_candidates(repo: &Path, inputs: &DashboardResolutionInputs) -> Vec<PathBuf> {
2838 let mut candidates = Vec::new();
2839
2840 candidates.extend(inputs.env_dist.clone());
2841 candidates.push(repo.join("apps").join("dashboard").join("dist"));
2842
2843 if let Some(home) = &inputs.home {
2844 candidates.push(home.join(".kranz").join("dashboard").join("dist"));
2845 candidates.push(home.join(".kranz").join("dashboard"));
2846 }
2847
2848 if let Some(exe) = &inputs.exe {
2849 candidates.extend(installed_dashboard_dirs(exe));
2850 candidates.extend(source_checkout_dist_from_exe(exe));
2851 }
2852
2853 if let Some(manifest_dir) = &inputs.manifest_dir {
2854 candidates.extend(source_checkout_dist_from_manifest(manifest_dir));
2855 candidates.push(manifest_dir.join("assets").join("dashboard").join("dist"));
2856 }
2857
2858 candidates
2859}
2860
2861fn first_dashboard_dir(candidates: Vec<PathBuf>) -> Option<PathBuf> {
2862 candidates
2863 .into_iter()
2864 .find(|d| d.join("index.html").is_file())
2865}
2866
2867fn installed_dashboard_dirs(exe: &Path) -> Vec<PathBuf> {
2868 let Some(bin_dir) = exe.parent() else {
2869 return Vec::new();
2870 };
2871 let mut dirs = vec![
2872 bin_dir.join("dashboard").join("dist"),
2873 bin_dir.join("dashboard"),
2874 ];
2875 if let Some(prefix) = bin_dir.parent() {
2876 dirs.push(
2877 prefix
2878 .join("share")
2879 .join("kranz")
2880 .join("dashboard")
2881 .join("dist"),
2882 );
2883 dirs.push(prefix.join("share").join("kranz").join("dashboard"));
2884 }
2885 dirs
2886}
2887
2888fn source_checkout_dist_from_exe(exe: &Path) -> Option<PathBuf> {
2889 let profile_dir = exe.parent()?;
2890 let target_dir = profile_dir.parent()?;
2891 if target_dir.file_name()? != "target" {
2892 return None;
2893 }
2894 Some(
2895 target_dir
2896 .parent()?
2897 .join("apps")
2898 .join("dashboard")
2899 .join("dist"),
2900 )
2901}
2902
2903fn source_checkout_dist_from_manifest(manifest_dir: &Path) -> Option<PathBuf> {
2904 Some(
2905 manifest_dir
2906 .parent()?
2907 .parent()?
2908 .join("apps")
2909 .join("dashboard")
2910 .join("dist"),
2911 )
2912}
2913
2914fn open_browser(url: &str) {
2916 #[cfg(target_os = "macos")]
2917 let mut command = {
2918 let mut c = std::process::Command::new("open");
2919 c.arg(url);
2920 c
2921 };
2922 #[cfg(target_os = "windows")]
2923 let mut command = {
2924 let mut c = std::process::Command::new("cmd");
2925 c.args(["/C", "start", "", url]);
2927 c
2928 };
2929 #[cfg(not(any(target_os = "macos", target_os = "windows")))]
2930 let mut command = {
2931 let mut c = std::process::Command::new("xdg-open");
2932 c.arg(url);
2933 c
2934 };
2935
2936 match command.spawn() {
2937 Ok(mut child) => {
2938 std::thread::spawn(move || {
2940 let _ = child.wait();
2941 });
2942 }
2943 Err(e) => eprintln!("kranz: could not open the browser: {e}"),
2944 }
2945}
2946
2947fn resolve_release_token(
2959 repo: &Path,
2960 url: &str,
2961 flag: Option<String>,
2962) -> Result<String, ReleaseTokenError> {
2963 let parsed = reqwest::Url::parse(url).ok();
2964 let operator_lookup = parsed
2965 .as_ref()
2966 .and_then(|url| {
2967 kranz_engine::paths::global_config()
2968 .as_deref()
2969 .map(|global| operator_token_for_url(global, url))
2970 })
2971 .unwrap_or(OperatorTokenLookup::Absent);
2972 let allow_repo_fallback = parsed.as_ref().is_some_and(automatic_repo_token_allowed);
2973 resolve_release_token_from_sources(repo, operator_lookup, allow_repo_fallback, flag)
2974}
2975
2976#[derive(Debug, Clone, PartialEq, Eq)]
2981enum OperatorTokenLookup {
2982 Absent,
2983 Found(String),
2984 Legacy(String),
2985 Ambiguous,
2986}
2987
2988#[derive(Debug, Clone, PartialEq, Eq)]
2989enum ReleaseTokenError {
2990 Absent,
2991 Ambiguous,
2992}
2993
2994fn resolve_release_token_from_sources(
2995 repo: &Path,
2996 operator_lookup: OperatorTokenLookup,
2997 allow_repo_fallback: bool,
2998 flag: Option<String>,
2999) -> Result<String, ReleaseTokenError> {
3000 if let Some(flag) = flag {
3001 return Ok(flag);
3002 }
3003 if let Ok(env) = std::env::var("KRANZ_TOKEN") {
3004 return Ok(env);
3005 }
3006 match operator_lookup {
3007 OperatorTokenLookup::Found(authority) => Ok(authority),
3008 OperatorTokenLookup::Legacy(authority) => {
3013 let repo_authority = allow_repo_fallback
3014 .then(|| read_token_file(&repo.join(".kranz").join("serve.token")))
3015 .flatten();
3016 Ok(repo_authority.unwrap_or(authority))
3017 }
3018 OperatorTokenLookup::Ambiguous => Err(ReleaseTokenError::Ambiguous),
3019 OperatorTokenLookup::Absent => allow_repo_fallback
3020 .then(|| read_token_file(&repo.join(".kranz").join("serve.token")))
3021 .flatten()
3022 .ok_or(ReleaseTokenError::Absent),
3023 }
3024}
3025
3026fn scan_operator_token_addresses(
3027 global_config: &Path,
3028 addresses: &[std::net::SocketAddr],
3029) -> OperatorTokenLookup {
3030 let mut authority = None;
3031 for address in addresses {
3032 if let Some(found) = read_token_file(&operator_serve_token_path(global_config, *address)) {
3033 if authority.is_some() {
3034 return OperatorTokenLookup::Ambiguous;
3037 }
3038 authority = Some(found);
3039 }
3040 }
3041 match authority {
3042 Some(authority) => OperatorTokenLookup::Found(authority),
3043 None => OperatorTokenLookup::Absent,
3044 }
3045}
3046
3047fn operator_token_for_url(global_config: &Path, url: &reqwest::Url) -> OperatorTokenLookup {
3048 let Some(port) = url.port_or_known_default() else {
3049 return OperatorTokenLookup::Absent;
3050 };
3051 let Some(host) = normalized_url_host(url) else {
3052 return OperatorTokenLookup::Absent;
3053 };
3054 let endpoint_lookup = if host.eq_ignore_ascii_case("localhost") {
3058 let primary = [
3059 std::net::SocketAddr::from((std::net::Ipv4Addr::LOCALHOST, port)),
3060 std::net::SocketAddr::from((std::net::Ipv6Addr::LOCALHOST, port)),
3061 ];
3062 match scan_operator_token_addresses(global_config, &primary) {
3063 OperatorTokenLookup::Absent => scan_operator_token_addresses(
3064 global_config,
3065 &[
3066 std::net::SocketAddr::from((std::net::Ipv4Addr::UNSPECIFIED, port)),
3067 std::net::SocketAddr::from((std::net::Ipv6Addr::UNSPECIFIED, port)),
3068 ],
3069 ),
3070 other => other,
3071 }
3072 } else {
3073 let Ok(ip) = host.parse::<std::net::IpAddr>() else {
3074 return OperatorTokenLookup::Absent;
3075 };
3076 if !ip.is_loopback() {
3079 return OperatorTokenLookup::Absent;
3080 }
3081 let primary = [std::net::SocketAddr::new(ip, port)];
3082 match scan_operator_token_addresses(global_config, &primary) {
3083 OperatorTokenLookup::Absent => {
3084 let unspecified = match ip {
3085 std::net::IpAddr::V4(_) => {
3086 std::net::IpAddr::V4(std::net::Ipv4Addr::UNSPECIFIED)
3087 }
3088 std::net::IpAddr::V6(_) => {
3089 std::net::IpAddr::V6(std::net::Ipv6Addr::UNSPECIFIED)
3090 }
3091 };
3092 scan_operator_token_addresses(
3093 global_config,
3094 &[std::net::SocketAddr::new(unspecified, port)],
3095 )
3096 }
3097 other => other,
3098 }
3099 };
3100 match endpoint_lookup {
3101 OperatorTokenLookup::Absent => {
3102 match read_token_file(&legacy_operator_serve_token_path(global_config, port)) {
3104 Some(token) => OperatorTokenLookup::Legacy(token),
3105 None => OperatorTokenLookup::Absent,
3106 }
3107 }
3108 other => other,
3109 }
3110}
3111
3112fn automatic_repo_token_allowed(url: &reqwest::Url) -> bool {
3113 normalized_url_host(url).is_some_and(|host| {
3114 host.eq_ignore_ascii_case("localhost")
3115 || host
3116 .parse::<std::net::IpAddr>()
3117 .is_ok_and(|ip| ip.is_loopback())
3118 })
3119}
3120
3121fn normalized_url_host(url: &reqwest::Url) -> Option<&str> {
3122 let host = url.host_str()?;
3123 Some(
3124 host.strip_prefix('[')
3125 .and_then(|host| host.strip_suffix(']'))
3126 .unwrap_or(host),
3127 )
3128}
3129
3130fn read_token_file(path: &Path) -> Option<String> {
3131 let contents = std::fs::read_to_string(path).ok()?;
3132 let authority = contents.trim_end().to_string();
3133 if authority.is_empty() {
3134 None
3135 } else {
3136 Some(authority)
3137 }
3138}
3139
3140fn release_http_client() -> Result<reqwest::Client> {
3141 reqwest::Client::builder()
3142 .redirect(reqwest::redirect::Policy::none())
3143 .connect_timeout(Duration::from_secs(5))
3144 .timeout(Duration::from_secs(30))
3145 .build()
3146 .context("building release HTTP client")
3147}
3148
3149fn release_repo_id_is_valid(id: &str) -> bool {
3150 let mut chars = id.chars();
3151 let valid_first = chars.next().is_some_and(|ch| ch.is_ascii_alphanumeric());
3152 let valid_rest = chars.all(|ch| ch.is_ascii_alphanumeric() || matches!(ch, '-' | '_'));
3153 valid_first && valid_rest
3154}
3155
3156async fn cmd_release(
3157 repo: &Path,
3158 mission_id: &str,
3159 url: &str,
3160 token: Option<String>,
3161) -> Result<i32> {
3162 let token = match resolve_release_token(repo, url, token) {
3163 Ok(token) => token,
3164 Err(ReleaseTokenError::Ambiguous) => {
3165 bail!(
3166 "multiple ~/.kranz/serve/<endpoint>.token files match {url}; \
3167 pass --token for the intended serve instead of guessing"
3168 );
3169 }
3170 Err(ReleaseTokenError::Absent) => {
3171 bail!(
3172 "no mutation token available — pass --token, set $KRANZ_TOKEN, or use a local URL matching \
3173 a live ~/.kranz/serve/<endpoint>.token / single-repo .kranz/serve.token \
3174 (the token `kranz serve` prints on startup)"
3175 );
3176 }
3177 };
3178
3179 let client = release_http_client()?;
3180 let repo_id = resolve_release_repo_id(repo, url, &client, &token).await?;
3181 if let Some(id) = repo_id.as_deref() {
3182 if !release_repo_id_is_valid(id) {
3183 bail!("live serve returned an invalid repository id '{id}'; refusing release");
3184 }
3185 }
3186 let endpoint = release_endpoint(url, mission_id, repo_id.as_deref());
3187
3188 let response = client
3189 .post(&endpoint)
3190 .header("x-kranz-token", token)
3191 .json(&serde_json::json!({}))
3192 .send()
3193 .await
3194 .map_err(|e| {
3195 if e.is_connect() {
3196 anyhow!("no kranz serve reachable at {url} — is it running?")
3197 } else {
3198 anyhow::Error::new(e).context(format!("releasing mission '{mission_id}'"))
3199 }
3200 })?;
3201
3202 match response.status() {
3203 reqwest::StatusCode::OK => {
3204 let body: serde_json::Value = response.json().await.unwrap_or_default();
3205 let released = body
3206 .get("released")
3207 .and_then(serde_json::Value::as_bool)
3208 .unwrap_or(false);
3209 if released {
3210 println!("mission {mission_id} released — the lock is now free");
3211 } else {
3212 println!("mission {mission_id} was already free (no lock held)");
3213 }
3214 Ok(0)
3215 }
3216 reqwest::StatusCode::CONFLICT => {
3217 eprintln!("kranz: mission {mission_id} has a turn in flight — try again shortly");
3218 Ok(1)
3219 }
3220 reqwest::StatusCode::NOT_FOUND => {
3221 eprintln!("kranz: unknown mission '{mission_id}' at {url}");
3222 Ok(1)
3223 }
3224 reqwest::StatusCode::UNAUTHORIZED => {
3225 eprintln!(
3226 "kranz: the token was missing or invalid — check --token / $KRANZ_TOKEN \
3227 against the token `kranz serve` printed on startup"
3228 );
3229 Ok(1)
3230 }
3231 other => {
3232 let body = response.text().await.unwrap_or_default();
3233 eprintln!("kranz: release failed ({other}): {body}");
3234 Ok(1)
3235 }
3236 }
3237}
3238
3239async fn resolve_release_repo_id(
3244 repo: &Path,
3245 url: &str,
3246 client: &reqwest::Client,
3247 token: &str,
3248) -> Result<Option<String>> {
3249 live_release_repo_id(repo, url, client, token).await
3250}
3251
3252fn release_repo_id_from_summaries(
3253 repo: &Path,
3254 repos: &[kranz_server::RepoSummary],
3255) -> Result<Option<String>> {
3256 if repos.is_empty() {
3257 bail!("the live serve returned an empty repository catalog; refusing an unscoped release");
3258 }
3259 let root = std::fs::canonicalize(repo).unwrap_or_else(|_| repo.to_path_buf());
3260 let root_str = root.to_string_lossy();
3261 let matches: Vec<&kranz_server::RepoSummary> = repos
3262 .iter()
3263 .filter(|entry| {
3264 let entry_root = PathBuf::from(&entry.root);
3265 let entry_canon =
3266 std::fs::canonicalize(&entry_root).unwrap_or_else(|_| entry_root.clone());
3267 entry_canon == root || entry.root == root_str
3268 })
3269 .collect();
3270 match matches.as_slice() {
3271 [one] => Ok(Some(one.id.clone())),
3272 [] => Err(anyhow!(
3273 "selected repository '{}' is not present in the live serve catalog; refusing an unscoped release",
3274 repo.display()
3275 )),
3276 _ => Err(anyhow!(
3277 "selected repository '{}' matches multiple live serve catalog entries; refusing release",
3278 repo.display()
3279 )),
3280 }
3281}
3282
3283async fn live_release_repo_id(
3284 repo: &Path,
3285 url: &str,
3286 client: &reqwest::Client,
3287 token: &str,
3288) -> Result<Option<String>> {
3289 let base = url.trim_end_matches('/');
3290 let mut request = client.get(format!("{base}/api/repos"));
3294 let loopback_catalog = reqwest::Url::parse(url)
3295 .ok()
3296 .is_some_and(|parsed| automatic_repo_token_allowed(&parsed));
3297 if !loopback_catalog {
3298 request = request.header("x-kranz-token", token);
3299 }
3300 let send_error = |e: reqwest::Error| {
3301 if e.is_connect() {
3302 anyhow!("no kranz serve reachable at {url} — is it running?")
3303 } else {
3304 anyhow::Error::new(e).context("listing live serve repositories")
3305 }
3306 };
3307 let mut response = request.send().await.map_err(send_error)?;
3308 if loopback_catalog && response.status() == reqwest::StatusCode::UNAUTHORIZED {
3313 response = client
3314 .get(format!("{base}/api/repos"))
3315 .header("x-kranz-token", token)
3316 .send()
3317 .await
3318 .map_err(send_error)?;
3319 }
3320 if !response.status().is_success() {
3321 bail!(
3322 "cannot list repositories at {url}: HTTP {}",
3323 response.status()
3324 );
3325 }
3326 let repos: Vec<kranz_server::RepoSummary> = response
3327 .json()
3328 .await
3329 .context("parsing GET /api/repos response")?;
3330 release_repo_id_from_summaries(repo, &repos)
3331}
3332
3333fn release_endpoint(url: &str, mission_id: &str, repo_id: Option<&str>) -> String {
3334 let base = url.trim_end_matches('/');
3335 match repo_id {
3336 Some(repo_id) => {
3337 format!("{base}/api/repos/{repo_id}/missions/{mission_id}/release")
3338 }
3339 None => format!("{base}/api/missions/{mission_id}/release"),
3340 }
3341}
3342
3343#[cfg(test)]
3344mod tests {
3345 use super::*;
3346 use std::fs;
3347
3348 static KRANZ_TOKEN_ENV_LOCK: std::sync::Mutex<()> = std::sync::Mutex::new(());
3353
3354 #[tokio::test]
3357 #[allow(clippy::await_holding_lock)]
3361 async fn release_without_a_token_errors_clearly() {
3362 let _guard = KRANZ_TOKEN_ENV_LOCK.lock().unwrap();
3363 unsafe {
3365 std::env::remove_var("KRANZ_TOKEN");
3366 }
3367 let tmp = tempfile::tempdir().unwrap();
3368 let repo = tmp.path().to_path_buf();
3369 let err = cmd_release(&repo, "m-1", "http://192.0.2.1:4560", None)
3373 .await
3374 .unwrap_err();
3375 let msg = err.to_string();
3376 assert!(msg.contains("--token"), "{msg}");
3377 assert!(msg.contains("KRANZ_TOKEN"), "{msg}");
3378 }
3379
3380 #[test]
3383 fn release_reads_token_file_when_flag_and_env_absent() {
3384 let _guard = KRANZ_TOKEN_ENV_LOCK.lock().unwrap();
3385 unsafe {
3387 std::env::remove_var("KRANZ_TOKEN");
3388 }
3389 let tmp = tempfile::tempdir().unwrap();
3390 let repo = tmp.path().to_path_buf();
3391 write_serve_token(&repo, "file-token").unwrap();
3392
3393 assert_eq!(
3394 resolve_release_token_from_sources(&repo, OperatorTokenLookup::Absent, true, None),
3395 Ok("file-token".to_string())
3396 );
3397 }
3398
3399 #[test]
3400 fn release_reads_token_file_but_flag_overrides() {
3401 let _guard = KRANZ_TOKEN_ENV_LOCK.lock().unwrap();
3402 unsafe {
3404 std::env::remove_var("KRANZ_TOKEN");
3405 }
3406 let tmp = tempfile::tempdir().unwrap();
3407 let repo = tmp.path().to_path_buf();
3408 write_serve_token(&repo, "file-token").unwrap();
3409
3410 assert_eq!(
3411 resolve_release_token_from_sources(
3412 &repo,
3413 OperatorTokenLookup::Absent,
3414 true,
3415 Some("flag-token".to_string()),
3416 ),
3417 Ok("flag-token".to_string())
3418 );
3419 }
3420
3421 #[test]
3422 fn release_reads_token_file_but_env_overrides() {
3423 let _guard = KRANZ_TOKEN_ENV_LOCK.lock().unwrap();
3424 unsafe {
3426 std::env::set_var("KRANZ_TOKEN", "env-token");
3427 }
3428 let tmp = tempfile::tempdir().unwrap();
3429 let repo = tmp.path().to_path_buf();
3430 write_serve_token(&repo, "file-token").unwrap();
3431
3432 let result =
3433 resolve_release_token_from_sources(&repo, OperatorTokenLookup::Absent, true, None);
3434 unsafe {
3435 std::env::remove_var("KRANZ_TOKEN");
3436 }
3437 assert_eq!(result, Ok("env-token".to_string()));
3438 }
3439
3440 #[test]
3443 fn release_reads_token_file_none_present_yields_none() {
3444 let _guard = KRANZ_TOKEN_ENV_LOCK.lock().unwrap();
3445 unsafe {
3447 std::env::remove_var("KRANZ_TOKEN");
3448 }
3449 let tmp = tempfile::tempdir().unwrap();
3450 let repo = tmp.path().to_path_buf();
3451
3452 assert_eq!(
3453 resolve_release_token_from_sources(&repo, OperatorTokenLookup::Absent, true, None),
3454 Err(ReleaseTokenError::Absent)
3455 );
3456 }
3457
3458 #[test]
3459 fn release_prefers_operator_process_token_over_repo_compatibility_token() {
3460 let _guard = KRANZ_TOKEN_ENV_LOCK.lock().unwrap();
3461 unsafe {
3462 std::env::remove_var("KRANZ_TOKEN");
3463 }
3464 let tmp = tempfile::tempdir().unwrap();
3465 let repo = tmp.path().join("repo");
3466 write_serve_token(&repo, "repo-token").unwrap();
3467 let global = tmp.path().join("operator").join("config.json");
3468 let address = std::net::SocketAddr::from((std::net::Ipv4Addr::LOCALHOST, 4560));
3469 let operator = operator_serve_token_path(&global, address);
3470 write_token_file(&operator, "operator-token").unwrap();
3471 let url = reqwest::Url::parse("http://127.0.0.1:4560").unwrap();
3472
3473 assert_eq!(
3474 resolve_release_token_from_sources(
3475 &repo,
3476 operator_token_for_url(&global, &url),
3477 true,
3478 None,
3479 ),
3480 Ok("operator-token".to_string())
3481 );
3482 }
3483
3484 #[test]
3485 fn ambiguous_operator_token_does_not_fall_through_to_repo_file() {
3486 let _guard = KRANZ_TOKEN_ENV_LOCK.lock().unwrap();
3487 unsafe {
3488 std::env::remove_var("KRANZ_TOKEN");
3489 }
3490 let tmp = tempfile::tempdir().unwrap();
3491 let repo = tmp.path().join("repo");
3492 write_serve_token(&repo, "repo-token").unwrap();
3493
3494 assert_eq!(
3495 resolve_release_token_from_sources(&repo, OperatorTokenLookup::Ambiguous, true, None,),
3496 Err(ReleaseTokenError::Ambiguous)
3497 );
3498 }
3499
3500 #[test]
3501 fn operator_token_discovery_is_ambiguous_for_dual_localhost_endpoint_files() {
3502 let tmp = tempfile::tempdir().unwrap();
3503 let global = tmp.path().join(".kranz").join("config.json");
3504 let v4 = std::net::SocketAddr::from((std::net::Ipv4Addr::LOCALHOST, 4560));
3505 let v6 = std::net::SocketAddr::from((std::net::Ipv6Addr::LOCALHOST, 4560));
3506 write_token_file(&operator_serve_token_path(&global, v4), "v4-token").unwrap();
3507 write_token_file(&operator_serve_token_path(&global, v6), "v6-token").unwrap();
3508 let url = reqwest::Url::parse("http://localhost:4560").unwrap();
3509
3510 assert_eq!(
3511 operator_token_for_url(&global, &url),
3512 OperatorTokenLookup::Ambiguous
3513 );
3514 }
3515
3516 #[test]
3517 fn operator_token_path_is_scoped_by_full_bound_endpoint() {
3518 let global = Path::new("/operator/.kranz/config.json");
3519 let a =
3520 operator_serve_token_path(global, std::net::SocketAddr::from(([127, 0, 0, 1], 4560)));
3521 let b =
3522 operator_serve_token_path(global, std::net::SocketAddr::from(([127, 0, 0, 2], 4560)));
3523 assert_ne!(a, b);
3524 assert_eq!(
3525 a,
3526 Path::new("/operator/.kranz/serve/v4-7f000001-4560.token")
3527 );
3528 }
3529
3530 #[test]
3531 fn operator_token_discovery_handles_ipv6_url_brackets() {
3532 let tmp = tempfile::tempdir().unwrap();
3533 let global = tmp.path().join(".kranz").join("config.json");
3534 let address = std::net::SocketAddr::from((std::net::Ipv6Addr::LOCALHOST, 4560));
3535 write_token_file(&operator_serve_token_path(&global, address), "ipv6-token").unwrap();
3536 let url = reqwest::Url::parse("http://[::1]:4560").unwrap();
3537
3538 assert_eq!(
3539 operator_token_for_url(&global, &url),
3540 OperatorTokenLookup::Found("ipv6-token".to_string())
3541 );
3542 }
3543
3544 #[test]
3545 fn automatic_token_discovery_refuses_remote_domain_urls() {
3546 let _guard = KRANZ_TOKEN_ENV_LOCK.lock().unwrap();
3547 unsafe {
3548 std::env::remove_var("KRANZ_TOKEN");
3549 }
3550 let tmp = tempfile::tempdir().unwrap();
3551 let repo = tmp.path().join("repo");
3552 write_serve_token(&repo, "repo-token").unwrap();
3553
3554 assert_eq!(
3555 resolve_release_token(&repo, "https://example.com:4560", None),
3556 Err(ReleaseTokenError::Absent)
3557 );
3558 }
3559
3560 #[test]
3561 fn operator_token_discovery_refuses_non_loopback_ip_literals() {
3562 let tmp = tempfile::tempdir().unwrap();
3563 let global = tmp.path().join(".kranz").join("config.json");
3564 let address = std::net::SocketAddr::from(([203, 0, 113, 10], 4560));
3565 write_token_file(&operator_serve_token_path(&global, address), "remote-token").unwrap();
3566 let url = reqwest::Url::parse("http://203.0.113.10:4560").unwrap();
3567
3568 assert_eq!(
3569 operator_token_for_url(&global, &url),
3570 OperatorTokenLookup::Absent
3571 );
3572 }
3573
3574 #[test]
3575 fn operator_token_discovery_prefers_exact_loopback_over_stale_unspecified() {
3576 let tmp = tempfile::tempdir().unwrap();
3577 let global = tmp.path().join(".kranz").join("config.json");
3578 let loopback = std::net::SocketAddr::from((std::net::Ipv4Addr::LOCALHOST, 4560));
3579 let unspecified = std::net::SocketAddr::from((std::net::Ipv4Addr::UNSPECIFIED, 4560));
3580 write_token_file(&operator_serve_token_path(&global, loopback), "live-token").unwrap();
3581 write_token_file(
3582 &operator_serve_token_path(&global, unspecified),
3583 "stale-token",
3584 )
3585 .unwrap();
3586 let url = reqwest::Url::parse("http://127.0.0.1:4560").unwrap();
3587
3588 assert_eq!(
3589 operator_token_for_url(&global, &url),
3590 OperatorTokenLookup::Found("live-token".to_string())
3591 );
3592 }
3593
3594 #[test]
3595 fn operator_token_discovery_falls_back_to_legacy_port_token() {
3596 let tmp = tempfile::tempdir().unwrap();
3597 let global = tmp.path().join(".kranz").join("config.json");
3598 write_token_file(
3599 &legacy_operator_serve_token_path(&global, 4560),
3600 "legacy-token",
3601 )
3602 .unwrap();
3603 let url = reqwest::Url::parse("http://127.0.0.1:4560").unwrap();
3604
3605 assert_eq!(
3606 operator_token_for_url(&global, &url),
3607 OperatorTokenLookup::Legacy("legacy-token".to_string())
3608 );
3609 }
3610
3611 #[test]
3612 fn stale_legacy_operator_token_does_not_mask_live_repo_token() {
3613 let _guard = KRANZ_TOKEN_ENV_LOCK.lock().unwrap();
3614 unsafe {
3615 std::env::remove_var("KRANZ_TOKEN");
3616 }
3617 let tmp = tempfile::tempdir().unwrap();
3618 let repo = tmp.path().join("repo");
3619 write_serve_token(&repo, "live-repo-token").unwrap();
3620 let global = tmp.path().join(".kranz").join("config.json");
3621 write_token_file(
3622 &legacy_operator_serve_token_path(&global, 4560),
3623 "stale-legacy-token",
3624 )
3625 .unwrap();
3626 let url = reqwest::Url::parse("http://127.0.0.1:4560").unwrap();
3627 let operator = operator_token_for_url(&global, &url);
3628
3629 assert_eq!(
3630 resolve_release_token_from_sources(&repo, operator, true, None),
3631 Ok("live-repo-token".to_string())
3632 );
3633 }
3634
3635 #[test]
3636 fn operator_token_discovery_prefers_endpoint_file_over_legacy_port_token() {
3637 let tmp = tempfile::tempdir().unwrap();
3638 let global = tmp.path().join(".kranz").join("config.json");
3639 let loopback = std::net::SocketAddr::from((std::net::Ipv4Addr::LOCALHOST, 4560));
3640 write_token_file(
3641 &operator_serve_token_path(&global, loopback),
3642 "endpoint-token",
3643 )
3644 .unwrap();
3645 write_token_file(
3646 &legacy_operator_serve_token_path(&global, 4560),
3647 "legacy-token",
3648 )
3649 .unwrap();
3650 let url = reqwest::Url::parse("http://127.0.0.1:4560").unwrap();
3651
3652 assert_eq!(
3653 operator_token_for_url(&global, &url),
3654 OperatorTokenLookup::Found("endpoint-token".to_string())
3655 );
3656 }
3657
3658 #[test]
3659 fn release_repo_id_from_summaries_refuses_duplicate_root() {
3660 let tmp = tempfile::tempdir().unwrap();
3661 let repo = tmp.path().join("repo");
3662 std::fs::create_dir_all(&repo).unwrap();
3663 let root = repo.to_string_lossy().into_owned();
3664 let summaries = vec![
3665 kranz_server::RepoSummary {
3666 id: "alpha".to_string(),
3667 root: root.clone(),
3668 display_name: "alpha".to_string(),
3669 group: None,
3670 pinned: false,
3671 is_default: true,
3672 status: "healthy".to_string(),
3673 error: None,
3674 activity: kranz_server::RepoActivity::default(),
3675 },
3676 kranz_server::RepoSummary {
3677 id: "beta".to_string(),
3678 root,
3679 display_name: "beta".to_string(),
3680 group: None,
3681 pinned: false,
3682 is_default: false,
3683 status: "healthy".to_string(),
3684 error: None,
3685 activity: kranz_server::RepoActivity::default(),
3686 },
3687 ];
3688 let error = release_repo_id_from_summaries(&repo, &summaries).unwrap_err();
3689 assert!(error.to_string().contains("matches multiple"));
3690 }
3691
3692 #[test]
3693 fn slack_operator_catalog_is_not_rejected_by_legacy_context_helper() {
3694 fn init_git(root: &Path) {
3695 std::fs::create_dir_all(root).unwrap();
3696 let status = std::process::Command::new("git")
3697 .args(["init", "-q"])
3698 .arg(root)
3699 .status()
3700 .unwrap();
3701 assert!(status.success());
3702 }
3703
3704 let tmp = tempfile::tempdir().unwrap();
3705 let a = tmp.path().join("a");
3706 init_git(&a);
3707 let multi = kranz_server::MultiRepoHost::from_config(kranz_server::HostConfig {
3708 default_repo: Some("a".to_string()),
3709 max_concurrent_repos: 1,
3710 repos: vec![kranz_server::RepoConfig {
3711 id: "a".to_string(),
3712 root: a,
3713 display_name: None,
3714 group: None,
3715 pinned: false,
3716 slack: kranz_server::RepoSlackConfig::default(),
3717 }],
3718 })
3719 .unwrap();
3720 let context = single_repo_slack_context(&multi).unwrap();
3721 assert_eq!(context.id(), "a");
3722 }
3723
3724 #[test]
3725 fn release_endpoint_is_repo_scoped_when_catalog_id_is_known() {
3726 assert_eq!(
3727 release_endpoint("http://127.0.0.1:4560/", "same-id", Some("repo-b")),
3728 "http://127.0.0.1:4560/api/repos/repo-b/missions/same-id/release"
3729 );
3730 }
3731
3732 #[test]
3733 fn release_repo_id_from_summaries_matches_live_catalog_root() {
3734 let tmp = tempfile::tempdir().unwrap();
3735 let repo = tmp.path().join("repo");
3736 std::fs::create_dir_all(&repo).unwrap();
3737 let summaries = vec![kranz_server::RepoSummary {
3738 id: "alpha".to_string(),
3739 root: repo.to_string_lossy().into_owned(),
3740 display_name: "alpha".to_string(),
3741 group: None,
3742 pinned: false,
3743 is_default: true,
3744 status: "healthy".to_string(),
3745 error: None,
3746 activity: kranz_server::RepoActivity::default(),
3747 }];
3748 assert_eq!(
3749 release_repo_id_from_summaries(&repo, &summaries)
3750 .unwrap()
3751 .as_deref(),
3752 Some("alpha")
3753 );
3754 }
3755
3756 #[test]
3757 fn release_repo_id_from_summaries_refuses_unmatched_root() {
3758 let tmp = tempfile::tempdir().unwrap();
3759 let repo = tmp.path().join("local");
3760 let other = tmp.path().join("other");
3761 std::fs::create_dir_all(&repo).unwrap();
3762 std::fs::create_dir_all(&other).unwrap();
3763 let summaries = vec![kranz_server::RepoSummary {
3764 id: "other".to_string(),
3765 root: other.to_string_lossy().into_owned(),
3766 display_name: "other".to_string(),
3767 group: None,
3768 pinned: false,
3769 is_default: false,
3770 status: "healthy".to_string(),
3771 error: None,
3772 activity: kranz_server::RepoActivity::default(),
3773 }];
3774 let error = release_repo_id_from_summaries(&repo, &summaries).unwrap_err();
3775 assert!(error
3776 .to_string()
3777 .contains("not present in the live serve catalog"));
3778 }
3779
3780 #[test]
3781 fn release_repo_id_from_summaries_refuses_empty_catalog() {
3782 let tmp = tempfile::tempdir().unwrap();
3783 let error = release_repo_id_from_summaries(tmp.path(), &[]).unwrap_err();
3784 assert!(error.to_string().contains("empty repository catalog"));
3785 }
3786
3787 #[tokio::test]
3788 async fn live_release_repo_lookup_authenticates_protected_catalog() {
3789 let tmp = tempfile::tempdir().unwrap();
3790 let repo = tmp.path().join("repo");
3791 std::fs::create_dir_all(&repo).unwrap();
3792 let status = std::process::Command::new("git")
3793 .args(["init", "-q"])
3794 .arg(&repo)
3795 .status()
3796 .unwrap();
3797 assert!(status.success());
3798
3799 let catalog = Arc::new(kranz_server::MultiRepoHost::single(repo.clone()).unwrap());
3800 let listener = tokio::net::TcpListener::bind((std::net::Ipv4Addr::LOCALHOST, 0))
3801 .await
3802 .unwrap();
3803 let address = listener.local_addr().unwrap();
3804 let app = kranz_server::router_with_multi_repo_host_and_addr(
3806 catalog,
3807 None,
3808 kranz_server::MutationAuthority::new("catalog-token").unwrap(),
3809 Some(address),
3810 true,
3811 false,
3812 );
3813 let server = tokio::spawn(async move {
3814 axum::serve(listener, app).await.unwrap();
3815 });
3816 let client = release_http_client().unwrap();
3817 let url = format!("http://{address}");
3818
3819 let repo_id = live_release_repo_id(&repo, &url, &client, "catalog-token")
3820 .await
3821 .unwrap();
3822
3823 server.abort();
3824 let _ = server.await;
3825 assert_eq!(repo_id.as_deref(), Some("repo"));
3826 }
3827
3828 #[tokio::test]
3829 async fn live_release_repo_lookup_retries_with_token_when_read_gated() {
3830 let tmp = tempfile::tempdir().unwrap();
3831 let repo = tmp.path().join("repo");
3832 std::fs::create_dir_all(&repo).unwrap();
3833 let status = std::process::Command::new("git")
3834 .args(["init", "-q"])
3835 .arg(&repo)
3836 .status()
3837 .unwrap();
3838 assert!(status.success());
3839
3840 let catalog = Arc::new(kranz_server::MultiRepoHost::single(repo.clone()).unwrap());
3841 let listener = tokio::net::TcpListener::bind((std::net::Ipv4Addr::LOCALHOST, 0))
3842 .await
3843 .unwrap();
3844 let address = listener.local_addr().unwrap();
3845 let app = kranz_server::router_with_multi_repo_host_and_addr(
3850 catalog,
3851 None,
3852 kranz_server::MutationAuthority::new("catalog-token").unwrap(),
3853 Some(address),
3854 false,
3855 true,
3856 );
3857 let server = tokio::spawn(async move {
3858 axum::serve(listener, app).await.unwrap();
3859 });
3860 let client = release_http_client().unwrap();
3861 let url = format!("http://{address}");
3862
3863 let repo_id = live_release_repo_id(&repo, &url, &client, "catalog-token")
3864 .await
3865 .unwrap();
3866
3867 server.abort();
3868 let _ = server.await;
3869 assert_eq!(repo_id.as_deref(), Some("repo"));
3870 }
3871
3872 #[test]
3873 fn empty_token_files_are_ignored() {
3874 let tmp = tempfile::tempdir().unwrap();
3875 let path = tmp.path().join("serve.token");
3876 std::fs::write(&path, "").unwrap();
3877 assert_eq!(read_token_file(&path), None);
3878 }
3879
3880 #[cfg(unix)]
3881 #[test]
3882 fn serve_token_file_is_written_with_owner_only_permissions() {
3883 use std::os::unix::fs::PermissionsExt;
3884 let tmp = tempfile::tempdir().unwrap();
3885 let repo = tmp.path().to_path_buf();
3886 let path = write_serve_token(&repo, "secret").unwrap();
3887 let mode = std::fs::metadata(&path).unwrap().permissions().mode();
3888 assert_eq!(mode & 0o777, 0o600);
3889 }
3890
3891 #[cfg(unix)]
3892 #[test]
3893 fn serve_read_token_file_is_written_with_owner_only_permissions() {
3894 use std::os::unix::fs::PermissionsExt;
3895 let tmp = tempfile::tempdir().unwrap();
3896 let repo = tmp.path().to_path_buf();
3897 let path = write_serve_read_token(&repo, "read-secret").unwrap();
3898 assert!(path.ends_with(".kranz/serve.read.token"));
3899 let mode = std::fs::metadata(&path).unwrap().permissions().mode();
3900 assert_eq!(mode & 0o777, 0o600);
3901 }
3902
3903 #[test]
3904 fn operator_read_token_path_sits_beside_the_operator_token() {
3905 let global = Path::new("/home/op/.kranz/config.json");
3906 let address = std::net::SocketAddr::from(([127, 0, 0, 1], 4560));
3907 let mutation = operator_serve_token_path(global, address);
3908 let read = operator_serve_read_token_path(global, address);
3909 assert_eq!(read, mutation.with_extension("read.token"));
3910 assert!(read.to_string_lossy().ends_with(".read.token"));
3911 }
3912
3913 #[cfg(unix)]
3914 #[test]
3915 fn existing_token_permissions_are_hardened_before_replacement() {
3916 use std::os::unix::fs::PermissionsExt;
3917 let tmp = tempfile::tempdir().unwrap();
3918 let path = tmp.path().join("serve.token");
3919 std::fs::write(&path, "old-token").unwrap();
3920 std::fs::set_permissions(&path, std::fs::Permissions::from_mode(0o644)).unwrap();
3921
3922 write_token_file(&path, "new-token").unwrap();
3923
3924 assert_eq!(std::fs::read_to_string(&path).unwrap(), "new-token");
3925 let mode = std::fs::metadata(&path).unwrap().permissions().mode();
3926 assert_eq!(mode & 0o777, 0o600);
3927 }
3928
3929 #[tokio::test]
3930 async fn serve_token_file_is_removed_after_graceful_shutdown() {
3931 let tmp = tempfile::tempdir().unwrap();
3932 let repo = tmp.path().to_path_buf();
3933 let path = repo.join(".kranz").join("serve.token");
3934 let read_path = repo.join(".kranz").join("serve.read.token");
3935
3936 let host = std::sync::Arc::new(kranz_server::MissionHost::new(repo.clone()));
3937 let listener =
3938 kranz_server::bind_listener(std::net::IpAddr::V4(std::net::Ipv4Addr::LOCALHOST), 0)
3939 .await
3940 .unwrap();
3941 serve_with_token_cleanup(
3942 &repo,
3943 host,
3944 listener,
3945 None,
3946 "tok".to_string(),
3947 "read-tok".to_string(),
3948 std::future::ready(()),
3949 )
3950 .await
3951 .unwrap();
3952
3953 assert!(!path.exists());
3954 assert!(!read_path.exists());
3955 }
3956
3957 fn dashboard_at(path: PathBuf) -> PathBuf {
3958 fs::create_dir_all(&path).unwrap();
3959 fs::write(path.join("index.html"), "<!doctype html>").unwrap();
3960 path
3961 }
3962
3963 fn inputs() -> DashboardResolutionInputs {
3964 DashboardResolutionInputs {
3965 embedded_available: false,
3966 ..Default::default()
3967 }
3968 }
3969
3970 #[test]
3971 fn dashboard_resolution_honors_explicit_path_verbatim() {
3972 let tmp = tempfile::tempdir().unwrap();
3973 let repo = tmp.path().join("repo");
3974 let explicit = tmp.path().join("missing-dashboard");
3975
3976 assert_eq!(
3977 resolve_dashboard_assets_from(&repo, Some(explicit.clone()), &inputs()),
3978 Some(DashboardAssets::Dir(explicit))
3979 );
3980 }
3981
3982 #[test]
3983 fn dashboard_resolution_env_precedes_repo_and_invalid_env_is_skipped() {
3984 let tmp = tempfile::tempdir().unwrap();
3985 let repo = tmp.path().join("repo");
3986 let repo_dist = dashboard_at(repo.join("apps").join("dashboard").join("dist"));
3987 let env_dist = dashboard_at(tmp.path().join("env-dist"));
3988
3989 let mut with_env = inputs();
3990 with_env.env_dist = Some(env_dist.clone());
3991 assert_eq!(
3992 resolve_dashboard_assets_from(&repo, None, &with_env),
3993 Some(DashboardAssets::Dir(env_dist))
3994 );
3995
3996 let mut with_invalid_env = inputs();
3997 with_invalid_env.env_dist = Some(tmp.path().join("missing-env-dist"));
3998 assert_eq!(
3999 resolve_dashboard_assets_from(&repo, None, &with_invalid_env),
4000 Some(DashboardAssets::Dir(repo_dist))
4001 );
4002 }
4003
4004 #[test]
4005 fn dashboard_resolution_finds_installed_asset_dirs() {
4006 let tmp = tempfile::tempdir().unwrap();
4007 let repo = tmp.path().join("repo");
4008 let exe = tmp.path().join("prefix").join("bin").join("kranz");
4009 let installed = dashboard_at(
4010 tmp.path()
4011 .join("prefix")
4012 .join("share")
4013 .join("kranz")
4014 .join("dashboard")
4015 .join("dist"),
4016 );
4017
4018 let mut inputs = inputs();
4019 inputs.exe = Some(exe);
4020 assert_eq!(
4021 resolve_dashboard_assets_from(&repo, None, &inputs),
4022 Some(DashboardAssets::Dir(installed))
4023 );
4024 }
4025
4026 #[test]
4027 fn dashboard_resolution_finds_checkout_used_to_build_installed_binary() {
4028 let tmp = tempfile::tempdir().unwrap();
4029 let repo = tmp.path().join("mission-repo");
4030 let checkout = tmp.path().join("kranz");
4031 let manifest_dir = checkout.join("crates").join("cli");
4032 let checkout_dist = dashboard_at(checkout.join("apps").join("dashboard").join("dist"));
4033
4034 let mut inputs = inputs();
4035 inputs.manifest_dir = Some(manifest_dir);
4036 inputs.exe = Some(tmp.path().join("cargo-home").join("bin").join("kranz"));
4037 assert_eq!(
4038 resolve_dashboard_assets_from(&repo, None, &inputs),
4039 Some(DashboardAssets::Dir(checkout_dist))
4040 );
4041 }
4042
4043 #[test]
4044 fn dashboard_resolution_falls_back_to_embedded_assets() {
4045 let tmp = tempfile::tempdir().unwrap();
4046 let repo = tmp.path().join("repo");
4047 let mut inputs = inputs();
4048 inputs.embedded_available = true;
4049
4050 assert_eq!(
4051 resolve_dashboard_assets_from(&repo, None, &inputs),
4052 Some(DashboardAssets::Embedded)
4053 );
4054 }
4055
4056 #[test]
4057 fn embedded_dashboard_bundle_contains_index() {
4058 assert!(
4059 crate::embedded_dashboard::EMBEDDED_DASHBOARD
4060 .iter()
4061 .any(|file| file.path == "index.html"),
4062 "embedded dashboard source: {}",
4063 crate::embedded_dashboard::EMBEDDED_DASHBOARD_SOURCE
4064 );
4065 }
4066
4067 #[test]
4068 fn serve_refuses_non_loopback_without_insecure_lan() {
4069 let bind: std::net::IpAddr = "0.0.0.0".parse().unwrap();
4070 let err = refuse_non_loopback_without_insecure_lan(bind, false).unwrap_err();
4071 let msg = err.to_string();
4072 assert!(
4073 msg.contains("refusing to bind") && msg.contains("--insecure-lan"),
4074 "unexpected error: {msg}"
4075 );
4076 }
4077
4078 #[test]
4079 fn serve_allows_non_loopback_with_insecure_lan() {
4080 let bind: std::net::IpAddr = "0.0.0.0".parse().unwrap();
4081 refuse_non_loopback_without_insecure_lan(bind, true).unwrap();
4082 }
4083
4084 #[test]
4085 fn serve_allows_loopback_without_insecure_lan() {
4086 let bind: std::net::IpAddr = "127.0.0.1".parse().unwrap();
4087 refuse_non_loopback_without_insecure_lan(bind, false).unwrap();
4088 }
4089
4090 #[test]
4091 fn read_auth_on_loopback_requires_read_token() {
4092 assert!(effective_require_read_token(true, true));
4093 }
4094
4095 #[test]
4096 fn read_auth_off_loopback_bind_does_not_require_read_token() {
4097 assert!(!effective_require_read_token(true, false));
4098 }
4099
4100 #[test]
4101 fn read_auth_non_loopback_always_requires_read_token() {
4102 assert!(effective_require_read_token(false, true));
4103 assert!(effective_require_read_token(false, false));
4104 }
4105
4106 fn reconcile_turn(reply: &str) -> Vec<kranz_engine::backend::AgentEvent> {
4111 vec![
4112 kranz_engine::backend_mock::mock_text(reply),
4113 kranz_engine::backend_mock::mock_result_text(reply),
4114 ]
4115 }
4116
4117 fn reconcile_worker_pass() -> kranz_engine::backend_mock::MockScript {
4118 kranz_engine::backend_mock::MockScript::single_shot_json(&serde_json::json!({
4119 "result": "pass",
4120 "summary": "implemented and tested",
4121 "filesTouched": ["delivered.txt"],
4122 "testsAdded": [],
4123 "testEvidence": "all green",
4124 "commits": []
4125 }))
4126 .writes_file("delivered.txt", "delivered by the mock worker\n")
4127 }
4128
4129 fn reconcile_plan_json() -> serde_json::Value {
4130 serde_json::json!({
4131 "goal": "ship the demo",
4132 "validationContract": [],
4133 "milestones": [{
4134 "title": "M1",
4135 "features": [{
4136 "title": "F1",
4137 "spec": "build the thing",
4138 "validationCriteria": ["it works"]
4139 }]
4140 }]
4141 })
4142 }
4143
4144 #[tokio::test]
4153 async fn reconcile_on_terminal_after_cli_run_marks_ticket_done() {
4154 let tmp = tempfile::tempdir().unwrap();
4155 let repo = tmp.path().to_path_buf();
4156 let status = std::process::Command::new("git")
4157 .args(["init", "-b", "main"])
4158 .current_dir(&repo)
4159 .status()
4160 .unwrap();
4161 assert!(status.success());
4162 std::process::Command::new("git")
4163 .args(["config", "user.name", "test"])
4164 .current_dir(&repo)
4165 .status()
4166 .unwrap();
4167 std::process::Command::new("git")
4168 .args(["config", "user.email", "test@example.com"])
4169 .current_dir(&repo)
4170 .status()
4171 .unwrap();
4172 std::fs::write(repo.join("README.md"), "seed\n").unwrap();
4173 std::process::Command::new("git")
4174 .args(["add", "-A"])
4175 .current_dir(&repo)
4176 .status()
4177 .unwrap();
4178 std::process::Command::new("git")
4179 .args(["commit", "-m", "seed"])
4180 .current_dir(&repo)
4181 .status()
4182 .unwrap();
4183 let repo = std::fs::canonicalize(&repo).unwrap();
4184
4185 let judgement = serde_json::json!({
4186 "decision": "complete",
4187 "guidance": "",
4188 "summary": "worker did the job"
4189 });
4190 let orch_setup = kranz_engine::backend_mock::MockScript::streaming(vec![
4194 kranz_engine::backend_mock::mock_init("orch-session"),
4195 kranz_engine::backend_mock::mock_result_text("seed-hi"),
4196 ])
4197 .responding(vec![
4198 reconcile_turn("let's scope the demo"),
4199 reconcile_turn(&reconcile_plan_json().to_string()),
4200 ]);
4201 let orch_run = kranz_engine::backend_mock::MockScript::streaming(vec![
4202 kranz_engine::backend_mock::mock_init("orch-session-2"),
4203 kranz_engine::backend_mock::mock_result_text("ack"),
4204 ])
4205 .responding(vec![
4206 reconcile_turn("ack"),
4207 reconcile_turn(
4208 &serde_json::json!({"action": "commit-as-is", "note": "worker delivered files"})
4209 .to_string(),
4210 ),
4211 reconcile_turn(&judgement.to_string()),
4212 reconcile_turn("NONE"),
4213 reconcile_turn("NONE"),
4217 reconcile_turn("NONE"),
4218 reconcile_turn("NONE"),
4219 ]);
4220 let backend: Arc<dyn AgentBackend> =
4221 Arc::new(kranz_engine::backend_mock::MockBackend::with_scripts(vec![
4222 orch_setup,
4223 kranz_engine::backend_mock::MockScript::single_shot("ok"),
4228 reconcile_worker_pass(),
4231 orch_run,
4232 ]));
4233
4234 let cfg = MissionConfig {
4235 skip_scrutiny: true,
4236 skip_functional: true,
4237 ..Default::default()
4238 };
4239 let mut engine =
4240 MissionEngine::create(Arc::clone(&backend), repo.clone(), "ship the demo", cfg)
4241 .unwrap();
4242 let mission_id = engine.mission_id().to_string();
4243 engine.planning_turn("ship the demo").await.unwrap();
4244 let request = engine.request_plan().await.unwrap();
4245 let plan = match request {
4246 PlanRequest::Ready(plan) => plan,
4247 PlanRequest::NotReady(text) => panic!("expected a ready plan, got: {text}"),
4248 PlanRequest::WrongPlan { reason } => {
4249 panic!("expected a ready plan, got a wrong-plan escalation: {reason}")
4250 }
4251 };
4252 engine.approve_plan(plan).unwrap();
4253 drop(engine);
4254
4255 kranz_engine::ticket::Ticket::record_mission(&repo, "my-ticket", &mission_id).unwrap();
4258 kranz_engine::ticket::Ticket::write_state(
4259 &repo,
4260 "my-ticket",
4261 kranz_engine::ticket::TicketState::Failed,
4262 None,
4263 )
4264 .unwrap();
4265 assert_eq!(
4266 kranz_engine::ticket::Ticket::read_state(&repo, "my-ticket"),
4267 kranz_engine::ticket::TicketState::Failed
4268 );
4269
4270 let exit_code =
4271 run_mission_loop_with_backend(repo.clone(), mission_id, LockForce::No, false, backend)
4272 .await
4273 .unwrap();
4274 assert_eq!(exit_code, 0);
4275
4276 assert_eq!(
4277 kranz_engine::ticket::Ticket::read_state(&repo, "my-ticket"),
4278 kranz_engine::ticket::TicketState::Done,
4279 "run_mission_loop must reconcile the linked ticket to Done on Complete"
4280 );
4281 }
4282}