Skip to main content

claudectl_core/
runtime.rs

1//! UI ↔ runtime contract.
2//!
3//! Tracking: issue #274 of the workspace-refactor epic (#279).
4//!
5//! The TUI today reaches deep into brain / coord / bus / rules internals. This
6//! module defines the **read-only** boundary it should reach through instead,
7//! so a future `claudectl-tui` crate (#275) can be extracted and iterated on
8//! without recompiling brain or the bus.
9//!
10//! ## Why traits, why core-owned DTOs
11//!
12//! Each view is a trait, not a concrete struct, so:
13//!
14//! - The binary crate can hand the TUI a real implementation backed by SQLite
15//!   / the engine / the bus DB. A future remote frontend can hand it an HTTP
16//!   client. Tests hand it a fixture.
17//! - Adding a method to a trait is a contract change reviewable in one PR;
18//!   adding a method to a concrete struct ripples through every caller.
19//!
20//! Each DTO (`SessionSnapshot`, `LeaseSummary`, …) is owned by `core` so the
21//! traits don't drag `brain::DecisionRecord` / `coord::Lease` upward into the
22//! TUI's dependency surface. Conversion happens once, in the wrapper.
23//!
24//! ## What's in scope here
25//!
26//! Read-only views only. Side-effecting paths (`terminate_session`,
27//! `inject_prompt`, `log_decision`) deserve a separate `Actions` trait once
28//! the TUI's write surface is mapped. Adding it speculatively now would
29//! violate the "only add methods existing call sites need" rule the epic
30//! committed to.
31//!
32//! ## What's NOT in scope
33//!
34//! - The TUI doesn't yet call through these traits — that's #275.
35//! - Brain's review/scorecard surface — it's heavier and worth its own
36//!   trait once #275 stabilizes the basic shape.
37//! - Pub/sub claim protocol — #283 builds on top of these traits, not the
38//!   other way around.
39
40use serde::{Deserialize, Serialize};
41use std::sync::Arc;
42
43// ============================================================================
44// Sessions
45// ============================================================================
46
47/// One running Claude Code session, as observed by the runtime. Minimal
48/// projection of the binary-crate `ClaudeSession`; only fields the TUI
49/// renders.
50#[derive(Debug, Clone, Serialize, Deserialize)]
51pub struct SessionSnapshot {
52    pub session_id: String,
53    pub pid: u32,
54    pub cwd: String,
55    pub project_name: String,
56    pub status: String,
57    pub cost_usd: f64,
58    pub context_tokens: u64,
59    pub context_max: u64,
60    pub last_message_ts: u64,
61}
62
63/// Read access to the live session roster.
64pub trait SessionSource: Send + Sync {
65    /// Snapshot of every running Claude Code session. Order is the
66    /// implementor's choice; the TUI sorts again client-side.
67    fn list(&self) -> Vec<SessionSnapshot>;
68
69    /// Fetch a specific session by its ID. `None` when the session has
70    /// exited or never existed.
71    fn detail(&self, session_id: &str) -> Option<SessionSnapshot> {
72        self.list().into_iter().find(|s| s.session_id == session_id)
73    }
74}
75
76// ============================================================================
77// Brain
78// ============================================================================
79
80/// Mirrors the binary's `brain::GateMode` without depending on the brain
81/// crate. Persisted as the lowercased label.
82#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
83#[serde(rename_all = "snake_case")]
84pub enum BrainGateMode {
85    On,
86    Off,
87    Auto,
88}
89
90impl BrainGateMode {
91    /// Canonical lowercase label — the form persisted to
92    /// `~/.claudectl/brain/gate-mode` and emitted by the TUI status messages.
93    pub fn as_str(&self) -> &'static str {
94        match self {
95            Self::On => "on",
96            Self::Off => "off",
97            Self::Auto => "auto",
98        }
99    }
100}
101
102impl std::fmt::Display for BrainGateMode {
103    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
104        f.write_str(self.as_str())
105    }
106}
107
108/// A single past brain decision, projected for display.
109///
110/// The first six fields are the common shape used by `BrainView::recent_decisions`.
111/// The remaining fields support the Brain Review surface (`BrainReviewView`); they
112/// are `Option`-wrapped + `#[serde(default)]` so older `BrainView` callers can
113/// keep treating them as opaque.
114#[derive(Debug, Clone, Serialize, Deserialize)]
115pub struct DecisionSummary {
116    pub id: String,
117    pub timestamp: String,
118    pub action: String,
119    pub confidence: Option<f64>,
120    pub project: Option<String>,
121    pub tool: Option<String>,
122    /// PID of the session this decision belongs to. Used by counterfactual
123    /// analysis to pair decisions with their subsequent outcome from the
124    /// same session.
125    #[serde(default)]
126    pub pid: u32,
127
128    /// Tool input string when the decision was about a specific command.
129    #[serde(default)]
130    pub command: Option<String>,
131    /// Brain's free-form rationale for the suggestion.
132    #[serde(default)]
133    pub reasoning: Option<String>,
134    /// What the user did with the suggestion — `"accept"`, `"reject"`,
135    /// `"deny_rule_override"`, etc.
136    #[serde(default)]
137    pub user_action: Option<String>,
138    /// Why the user overrode the brain (if applicable).
139    #[serde(default)]
140    pub override_reason: Option<String>,
141    /// Wall-clock latency of the brain decision in milliseconds.
142    #[serde(default)]
143    pub brain_decision_ms: Option<u64>,
144    /// Whether the operator has marked this decision as canonical (teaching
145    /// material). `None` for records written before the field existed.
146    #[serde(default)]
147    pub canonical: Option<bool>,
148    /// Cache hit flag — served from the few-shot store without an LLM call.
149    /// `None` before instrumentation.
150    #[serde(default)]
151    pub cache_hit: Option<bool>,
152    /// Cost in USD when this decision was made (context snapshot).
153    #[serde(default)]
154    pub cost_usd: Option<f64>,
155    /// Model that produced the suggestion.
156    #[serde(default)]
157    pub model: Option<String>,
158    /// Resolved outcome category, when known. `"success" | "error" |
159    /// "test_failed" | "skipped"` etc. Mirrors the variants of the binary's
160    /// `brain::decisions::DecisionOutcome` enum, flattened to a string so
161    /// the contract doesn't pull the enum upward.
162    #[serde(default)]
163    pub outcome_kind: Option<String>,
164    /// Free-form detail for failure outcomes (the failing command for
165    /// `test_failed`, the error message for `error`).
166    #[serde(default)]
167    pub outcome_detail: Option<String>,
168    /// Epoch seconds when the brain suggestion was first surfaced. Used by
169    /// time-to-correct analysis. `None` for records pre-instrumentation or
170    /// passive observations.
171    #[serde(default)]
172    pub suggested_at: Option<u64>,
173    /// Epoch seconds when the user acted on the suggestion. `None` for
174    /// passive observations or records still in flight.
175    #[serde(default)]
176    pub resolved_at: Option<u64>,
177    /// One-line, human-readable explanation of *why* the brain reached this
178    /// decision (#372): source, confidence, rule/few-shot provenance. Rendered
179    /// inline in the review detail panel. Empty for records that carry no
180    /// derivable cause.
181    #[serde(default)]
182    pub why: String,
183    /// Where the decision originated (#372): `"llm"`, `"rule"`, `"few_shot"`.
184    #[serde(default)]
185    pub decision_source: Option<String>,
186    /// The auto-rule that fired, when this was a rule decision (#372).
187    #[serde(default)]
188    pub rule_name: Option<String>,
189}
190
191impl DecisionSummary {
192    /// Whether the user agreed with the brain (or the call was auto-executed).
193    /// Mirrors `brain::decisions::DecisionRecord::is_positive`.
194    pub fn is_positive(&self) -> bool {
195        matches!(
196            self.user_action.as_deref(),
197            Some("accept" | "auto" | "user_approve" | "rule_approve")
198        )
199    }
200
201    /// Whether the user disagreed with the brain. Mirrors
202    /// `brain::decisions::DecisionRecord::is_negative`.
203    pub fn is_negative(&self) -> bool {
204        matches!(
205            self.user_action.as_deref(),
206            Some("reject" | "deny_rule_override" | "rule_deny" | "conflict_deny")
207        )
208    }
209}
210
211/// Read access to the brain's decision history and current mode.
212pub trait BrainView: Send + Sync {
213    fn gate_mode(&self) -> BrainGateMode;
214
215    /// Most recent `n` decisions, newest first.
216    fn recent_decisions(&self, n: usize) -> Vec<DecisionSummary>;
217
218    /// Total count of brain decisions on disk. Drives the "decisions: N"
219    /// status line.
220    fn decision_count(&self) -> usize;
221}
222
223/// One entry in the Brain Review queue — a decision worth showing the operator
224/// for canonical-marking review, with a reason and a priority score.
225#[derive(Debug, Clone, Serialize, Deserialize)]
226pub struct ReviewItemSummary {
227    pub decision: DecisionSummary,
228    /// Free-form rationale for why this decision was queued for review.
229    pub reason: String,
230    /// Priority score (higher = more important to review first).
231    pub score: f64,
232}
233
234/// Read access to the Brain Review surface — the full decision log plus the
235/// review-queue projection. Separate trait from `BrainView` because:
236///
237/// - The Brain Review screen is the only TUI consumer; the dashboard's status
238///   bar only needs `BrainView::recent_decisions(n)`.
239/// - `all_decisions()` returns the whole log (can be thousands of records);
240///   it's a heavier surface than the lightweight `BrainView` methods and
241///   worth gating behind its own trait so callers don't accidentally invoke it.
242pub trait BrainReviewView: Send + Sync {
243    /// Every decision on disk, newest first. Used by the Brain Review screen
244    /// to render the full history list.
245    fn all_decisions(&self) -> Vec<DecisionSummary>;
246
247    /// Priority-ordered review queue. Built from `all_decisions()` plus the
248    /// brain's queue-building heuristics (counterfactual hits, Critical-tier
249    /// safety cases, high-confidence misses).
250    fn review_queue(&self) -> Vec<ReviewItemSummary>;
251}
252
253// ============================================================================
254// Coordination layer
255// ============================================================================
256
257#[derive(Debug, Clone, Serialize, Deserialize)]
258pub struct LeaseSummary {
259    pub id: String,
260    pub owner_session_id: String,
261    pub resource_kind: String,
262    pub resource_value: String,
263    pub mode: String,
264    pub acquired_at: String,
265    /// ISO timestamp when the lease expires. `None` for leases held without
266    /// an explicit deadline.
267    #[serde(default)]
268    pub expires_at: Option<String>,
269}
270
271#[derive(Debug, Clone, Serialize, Deserialize)]
272pub struct HandoffSummary {
273    pub id: String,
274    pub from_session_id: String,
275    pub to_session_id: Option<String>,
276    pub task_id: String,
277    pub summary: String,
278    pub priority: String,
279    pub created_at: String,
280}
281
282#[derive(Debug, Clone, Serialize, Deserialize)]
283pub struct InterruptSummary {
284    pub id: String,
285    pub interrupt_type: String,
286    pub priority: String,
287    pub target_session_id: String,
288    pub reason: String,
289    pub created_at: String,
290}
291
292/// A supervisor task lifecycle, projected for the TUI. One row per task in the
293/// coord `tasks` table. `state` is the wire label (`TaskState::as_str`), kept as
294/// a string so the contract doesn't drag the coord state enum upward.
295#[derive(Debug, Clone, Serialize, Deserialize)]
296pub struct TaskSummary {
297    pub id: String,
298    pub name: String,
299    pub state: String,
300    pub role: Option<String>,
301    /// Number of recorded attempts so far.
302    pub attempts: u32,
303    pub max_retries: u32,
304    /// Session id of the most recent attempt, when one has run.
305    pub last_session_id: Option<String>,
306    /// Most recent verifier verdict (e.g. "PASS"/"FAIL"), or `None` if no
307    /// verifier has run for this task yet.
308    #[serde(default)]
309    pub last_verdict: Option<String>,
310    /// Total cost in USD across all attempts.
311    #[serde(default)]
312    pub cost_usd: f64,
313    pub created_at: String,
314    pub updated_at: String,
315}
316
317/// Read access to coordination state (leases, handoffs, interrupts, tasks).
318/// Backed today by the `coord` SQLite store; in tests, by a fixture.
319pub trait CoordView: Send + Sync {
320    fn active_leases(&self) -> Vec<LeaseSummary>;
321    fn pending_handoffs(&self) -> Vec<HandoffSummary>;
322    fn pending_interrupts(&self) -> Vec<InterruptSummary>;
323    /// Supervisor task lifecycles, newest first. Empty when the `coord` feature
324    /// is off so the TUI can render a "no supervisor" state without cfg.
325    fn tasks(&self) -> Vec<TaskSummary>;
326}
327
328// ============================================================================
329// Agent bus
330// ============================================================================
331
332#[derive(Debug, Clone, Serialize, Deserialize)]
333pub struct AgentDirectoryEntry {
334    pub session_id: String,
335    pub pid: u32,
336    pub cwd: String,
337    pub project: String,
338    pub status: String,
339    pub role: Option<String>,
340}
341
342#[derive(Debug, Clone, Serialize, Deserialize)]
343pub struct RoleBinding {
344    pub name: String,
345    pub cwd_selector: String,
346    pub last_session_id: Option<String>,
347    pub last_seen: String,
348    /// PID this role is bound to (#307). `None` for cwd-only bindings.
349    pub pid: Option<u32>,
350}
351
352/// Read access to the agent-bus roster + role table. Disabled implementations
353/// (`bus` feature off in the binary) return empty vectors so the TUI can
354/// render the panel as "no bus" without conditional compilation.
355pub trait BusView: Send + Sync {
356    fn list_agents(&self) -> Vec<AgentDirectoryEntry>;
357    fn list_roles(&self) -> Vec<RoleBinding>;
358}
359
360// ============================================================================
361// Actions (write surface)
362// ============================================================================
363
364/// What the TUI needs to record alongside a user action. Core-owned so the
365/// trait doesn't drag `brain::decisions::log_observation`'s argument list
366/// upward into the TUI surface.
367#[derive(Debug, Clone, Serialize, Deserialize)]
368pub struct ObservationInput {
369    /// PID of the session the observation belongs to.
370    pub session_pid: u32,
371    /// Project label (usually the cwd basename).
372    pub project: String,
373    /// Tool the action targeted, when applicable ("Bash", "Write", …).
374    pub tool: Option<String>,
375    /// The command or input string the user ran/sent.
376    pub command: Option<String>,
377    /// Classification — `"user_approve"`, `"user_input"`, `"rule_approve"`,
378    /// `"rule_deny"`, and friends. Kept as a string so callers can introduce
379    /// new categories without a trait change.
380    pub observed_action: String,
381}
382
383/// Whether a decision was about a session or about an orchestration task.
384/// Mirrors `brain::decisions::DecisionType` without depending on it.
385#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
386#[serde(rename_all = "snake_case")]
387pub enum DecisionScope {
388    Session,
389    Orchestration,
390}
391
392impl DecisionScope {
393    /// Wire label. Matches what `brain::decisions::DecisionType::as_label`
394    /// writes to disk so the round-trip is stable.
395    pub fn as_str(&self) -> &'static str {
396        match self {
397            Self::Session => "session",
398            Self::Orchestration => "orchestration",
399        }
400    }
401}
402
403/// What the TUI needs to record when the user resolves a brain suggestion.
404/// Carries the suggestion that was on screen at the time, plus what the user
405/// did and (optionally) why they overrode the brain.
406#[derive(Debug, Clone, Serialize, Deserialize)]
407pub struct LogDecisionInput {
408    /// PID of the session the decision applies to.
409    pub session_pid: u32,
410    /// Project label.
411    pub project: String,
412    /// Tool name the suggestion targeted.
413    pub tool: Option<String>,
414    /// Command / input string the suggestion targeted.
415    pub command: Option<String>,
416    /// The brain's suggestion the user is resolving.
417    pub suggestion: PendingSuggestion,
418    /// What the user did — `"accept"`, `"reject"`, `"deny_rule_override"`, etc.
419    pub user_action: String,
420    /// Whether this was a session decision or an orchestration decision.
421    pub decision_type: DecisionScope,
422    /// Why the user overrode a brain denial (if applicable).
423    pub override_reason: Option<String>,
424}
425
426/// Side-effecting paths the TUI invokes when the user takes an action
427/// (terminating a session, sending a prompt, flipping the brain mode,
428/// recording an observation, marking a decision canonical for teaching).
429///
430/// Each method is fallible — implementations return a `Result` rather than
431/// silently swallowing errors so the TUI can surface failures in the status
432/// bar instead of pretending an action succeeded.
433///
434/// **Not modeled here (intentionally):**
435///
436/// - `launch::*` — launching a new Claude Code session is wider than a single
437///   call (cwd selection, model defaults, plugin propagation). Will get its
438///   own trait once the TUI's launcher surface is refactored alongside #275.
439/// - `mailbox::deliver_pending` — orchestrator-style mailbox delivery; should
440///   eventually move to an `Orchestrator` trait, not `Actions`.
441/// - `log_decision` (vs. `log_observation`) — currently only called from the
442///   plugin gate path, not the TUI. Stays a direct call until a TUI site
443///   needs it.
444pub trait Actions: Send + Sync {
445    /// Terminate the OS process with the given PID. The TUI uses this for
446    /// "kill session" hotkeys.
447    fn terminate_session(&self, pid: u32) -> Result<(), String>;
448
449    /// Inject text into the session's terminal. Implementations pick the
450    /// right backend (tmux, Kitty, etc.) from the session metadata.
451    ///
452    /// The `text` is sent verbatim — the caller is responsible for any
453    /// trailing newline they want. Sanitization (e.g. neutralizing a leading
454    /// `/`) belongs to the caller; this trait is a transport.
455    fn inject_text(&self, session_id: &str, text: &str) -> Result<(), String>;
456
457    /// Update the persisted brain gate mode. Reflected back through
458    /// `BrainView::gate_mode` on the next call.
459    fn set_gate_mode(&self, mode: BrainGateMode) -> Result<(), String>;
460
461    /// Record an observation about a user action — non-LLM "the user did X."
462    /// Drives the brain's outcome telemetry and preference distillation.
463    fn log_observation(&self, observation: ObservationInput) -> Result<(), String>;
464
465    /// Record a user's accept/reject decision on a brain suggestion. The
466    /// distinction from `log_observation` is that `log_decision` carries
467    /// the actual suggestion that was on screen — the brain pairs it with
468    /// the user's response for outcome telemetry, counterfactual analysis,
469    /// and few-shot retrieval.
470    fn log_decision(&self, input: LogDecisionInput) -> Result<(), String>;
471
472    /// Mark a past brain decision as canonical for teaching. Optional `note`
473    /// is the operator's annotation. Used by the Brain Review surface.
474    fn mark_canonical(&self, decision_id: &str, note: Option<String>) -> Result<(), String>;
475
476    /// Bind an agent-bus role to a `(cwd, pid)` pair. The TUI calls this from
477    /// the new role-bind key (Ctrl+R) so the operator can attach a role to a
478    /// specific running Claude session without dropping to another terminal.
479    /// Returns `Err` when the bus feature is compiled out or the DB write
480    /// fails. See issue #307.
481    fn bind_bus_role(&self, name: &str, cwd: &str, pid: u32) -> Result<(), String>;
482
483    /// Cancel a supervisor task (move it to CANCELLED). Used by the Supervisor
484    /// panel's one-key cancel (#368). `Err` when the `coord` feature is compiled
485    /// out, the task is already terminal, or the store write fails.
486    fn cancel_task(&self, task_id: &str) -> Result<(), String>;
487
488    /// Set or clear the supervisor drain marker. When draining, the reconciler
489    /// stops issuing new task assignments while letting running tasks finish.
490    /// `Err` when the `coord` feature is compiled out.
491    fn set_supervisor_drain(&self, draining: bool) -> Result<(), String>;
492
493    /// Re-queue a failed/cancelled task (move it back to PENDING so the
494    /// reconciler re-assigns it). `Err` for a DONE task (nothing to retry), a
495    /// still-active task, or when the `coord` feature is compiled out.
496    fn retry_task(&self, task_id: &str) -> Result<(), String>;
497
498    /// Approve a NEEDS_HUMAN task — accept the work as-is and move it to DONE,
499    /// overriding the verifier that escalated it. `Err` when the task isn't in
500    /// NEEDS_HUMAN, or the `coord` feature is compiled out.
501    fn approve_task(&self, task_id: &str) -> Result<(), String>;
502
503    /// Move a conversation to a paired peer, so it can be resumed there
504    /// (#510). The TUI calls this from the send hotkey; `transcript` is the
505    /// path session discovery already resolved, which is more reliable than
506    /// re-deriving it from `cwd` and the id.
507    ///
508    /// Returns the session id the transcript actually travels under, which is
509    /// not always `session_id`: discovery can hand back a transcript belonging
510    /// to a different id, and the bytes decide.
511    ///
512    /// Everything checkable without the network — transcript readable, peer
513    /// paired, an address to dial — is checked before returning `Ok`. The
514    /// megabytes themselves go in a detached child, because a median
515    /// transcript is 5.45 MB and the caller is a render loop. So `Ok` means
516    /// "accepted and started", not "delivered"; the ledger behind
517    /// `claudectl relay sessions` is where delivery shows up.
518    ///
519    /// `Err` when the `relay` feature is compiled out.
520    fn send_session_to_peer(
521        &self,
522        session_id: &str,
523        transcript: Option<&std::path::Path>,
524        cwd: &str,
525        peer: &str,
526    ) -> Result<String, String>;
527}
528
529// ============================================================================
530// Hive actions — knowledge-store (`hive`) + transport (`relay`) reads/writes
531// ============================================================================
532
533/// Snapshot of local hive state the overlay reads to render the Hive tab.
534///
535/// Mirrors the binary's same-named struct, lifted here so the TUI can hold
536/// it without depending on the binary's app module.
537#[derive(Debug, Clone, Default)]
538pub struct HiveViewSnapshot {
539    /// Local relay identity (peer ID string). `None` when the `relay`
540    /// feature is compiled out.
541    pub identity: Option<String>,
542    /// Known peers, paired with last-seen address when available.
543    pub peers: Vec<(String, Option<String>)>,
544}
545
546/// Hive + relay surface the TUI needs to render the Skills and Hive panels.
547///
548/// Separate trait from `Actions` because the underlying subsystems are
549/// feature-gated (`hive`, `relay`) — when those features are off the
550/// implementation returns empty/no-op values rather than failing.
551pub trait HiveActions: Send + Sync {
552    /// Semantic keys (`skill:<lowered-name>`) of skills already shared into
553    /// the local hive store. Used by the Skills overlay to mark "already
554    /// shared" rows.
555    fn shared_skill_keys(&self) -> std::collections::HashSet<String>;
556
557    /// Share a discovered skill into the local hive store. Returns the new
558    /// unit ID on success.
559    fn share_skill(&self, skill: &crate::skills::DiscoveredSkill) -> Result<String, String>;
560
561    /// Current view of the local relay identity + known peers for the Hive
562    /// tab.
563    fn hive_view_snapshot(&self) -> HiveViewSnapshot;
564}
565
566// ============================================================================
567// Brain driver (stateful)
568// ============================================================================
569
570/// One pending brain suggestion the user can accept or reject. Mirrors the
571/// binary's `brain::client::BrainSuggestion` without exposing brain types.
572#[derive(Debug, Clone, Serialize, Deserialize)]
573pub struct PendingSuggestion {
574    /// PID of the session this suggestion applies to.
575    pub pid: u32,
576    /// String label of the suggested action — `"approve"`, `"deny"`,
577    /// `"send"`, `"terminate"`. String to avoid pulling `RuleAction`'s enum
578    /// shape into the contract (it's already in `claudectl-core::rules` but
579    /// callers may want to add new categories without bumping the trait).
580    pub action: String,
581    /// Suggested message body (a prompt to inject, a rationale to surface).
582    pub message: Option<String>,
583    /// Why the brain suggested this — used for the detail panel and the
584    /// rejection log.
585    pub reasoning: String,
586    /// 0.0–1.0 confidence the brain assigned.
587    pub confidence: f64,
588    /// Epoch seconds when the suggestion was created. Drives time-to-correct
589    /// telemetry.
590    pub suggested_at: u64,
591}
592
593/// Stateful brain orchestration. Unlike the read-only views, this needs
594/// `&mut` per tick — the engine accumulates pending suggestions, cleans up
595/// after exited sessions, and applies user accept/reject decisions.
596///
597/// Boxed as `Box<dyn BrainDriver>` rather than `Arc<dyn BrainDriver>`
598/// because there is exactly one owner (the TUI's `App`) and every method
599/// needs `&mut self`. Sharing across threads is not a goal here.
600///
601/// Implementations may be `None` — the brain is opt-in (`--brain` flag).
602/// When the field is `None` the TUI renders without brain interaction.
603pub trait BrainDriver: Send {
604    /// Run one tick of the brain loop against the current session set and
605    /// the operator's deny rules. Returns `(pid, status_message)` pairs
606    /// for any actions the brain decided on this tick — the TUI surfaces
607    /// the messages in the status bar.
608    fn tick(
609        &mut self,
610        sessions: &[SessionSnapshot],
611        deny_rules: &[crate::rules::AutoRule],
612    ) -> Vec<(u32, String)>;
613
614    /// Drop pending suggestions for sessions that have exited. Called every
615    /// refresh so the pending map stays bounded by the live session count.
616    fn cleanup(&mut self, sessions: &[SessionSnapshot]);
617
618    /// User accepted the pending suggestion for `pid`. Implementations
619    /// return the log message the TUI should surface (`None` when there's
620    /// no suggestion to accept).
621    fn accept(&mut self, pid: u32) -> Option<String>;
622
623    /// User rejected the pending suggestion for `pid`. Returns the
624    /// suggestion that was rejected — the TUI logs it for replay /
625    /// counterfactual analysis.
626    fn reject(&mut self, pid: u32) -> Option<PendingSuggestion>;
627
628    /// Lookup the pending suggestion for `pid`, if any. Used by detail
629    /// panels and the status bar.
630    fn pending_for(&self, pid: u32) -> Option<PendingSuggestion>;
631
632    /// Total pending suggestion count. Used by the status bar.
633    fn pending_count(&self) -> usize;
634
635    /// Drop every pending suggestion. Called when the brain mode flips
636    /// off or the operator resets state.
637    fn clear_pending(&mut self);
638
639    /// Inject a pending suggestion for `pid`. Used by demo mode to fake
640    /// brain activity for screenshots / recordings. Implementations may
641    /// drop suggestions whose `action` string doesn't map to a known
642    /// engine action (rather than erroring out) — demo is best-effort.
643    fn set_pending(&mut self, suggestion: PendingSuggestion);
644}
645
646// ============================================================================
647// Orchestrator
648// ============================================================================
649
650/// Tick the orchestration layer once per refresh: deliver pending mailbox
651/// messages, deliver pending coordination interrupts, and bookkeep stale
652/// rows in the underlying stores.
653///
654/// Each `deliver_*` method returns `(id, status_message)` tuples the TUI
655/// surfaces in the status bar. `id` is a `u32` for mailbox (matching the
656/// brain's per-PID queue) and a `String` for interrupts (matching the
657/// coord interrupt_id). They differ deliberately — keeping the trait honest
658/// to the underlying surfaces rather than papering over the distinction.
659///
660/// Unlike `CoordView` / `BrainView` which are read-only, this trait is
661/// stateful: each call writes to the brain/coord SQLite stores. It still
662/// boxes as `Arc<dyn>` because no method needs `&mut self` — implementations
663/// open a connection per call.
664///
665/// Implementations may be no-ops when the relevant feature is off (e.g. the
666/// `coord` feature is disabled at compile time), so call sites don't need
667/// `#[cfg(feature = "...")]` guards.
668pub trait Orchestrator: Send + Sync {
669    /// Drain pending mailbox messages addressed to running sessions and
670    /// deliver them to the live terminals. Used by `App::tick` to surface
671    /// `"Brain → sess_X: 'go ahead'"` style messages.
672    fn deliver_mailbox(&self, sessions: &[SessionSnapshot]) -> Vec<(u32, String)>;
673
674    /// Deliver pending coordination interrupts (cross-agent signals) to
675    /// running sessions. Returns `(interrupt_id, status_message)` tuples.
676    fn deliver_interrupts(&self, sessions: &[SessionSnapshot]) -> Vec<(String, String)>;
677
678    /// Expire stale leases and interrupts that have passed their deadline.
679    /// Bookkeeping side-effect; best-effort, no return. Implementations
680    /// log failures rather than propagating them.
681    fn expire_stale(&self);
682}
683
684// ============================================================================
685// Runtime aggregate
686// ============================================================================
687
688/// Single struct the binary builds at startup and hands to the TUI.
689///
690/// All fields are `Arc<dyn ...>` so the TUI doesn't care whether an impl is
691/// a thin SQLite wrapper, a remote HTTP client, or an in-memory mock — they
692/// all share the same shape.
693#[derive(Clone)]
694pub struct Runtime {
695    pub sessions: Arc<dyn SessionSource>,
696    pub brain: Arc<dyn BrainView>,
697    pub coord: Arc<dyn CoordView>,
698    pub bus: Arc<dyn BusView>,
699    pub actions: Arc<dyn Actions>,
700    pub review: Arc<dyn BrainReviewView>,
701    pub orchestrator: Arc<dyn Orchestrator>,
702    pub hive: Arc<dyn HiveActions>,
703}
704
705impl Runtime {
706    // 8-trait composition root; each view is its own arc. Splitting this into
707    // sub-records would just trade one wide ctor for several narrow ones with
708    // the same total fan-out.
709    #[allow(clippy::too_many_arguments)]
710    pub fn new(
711        sessions: Arc<dyn SessionSource>,
712        brain: Arc<dyn BrainView>,
713        coord: Arc<dyn CoordView>,
714        bus: Arc<dyn BusView>,
715        actions: Arc<dyn Actions>,
716        review: Arc<dyn BrainReviewView>,
717        orchestrator: Arc<dyn Orchestrator>,
718        hive: Arc<dyn HiveActions>,
719    ) -> Self {
720        Self {
721            sessions,
722            brain,
723            coord,
724            bus,
725            actions,
726            review,
727            orchestrator,
728            hive,
729        }
730    }
731}
732
733// ============================================================================
734// MockRuntime — for tests in this crate and in claudectl-tui
735// ============================================================================
736
737/// In-memory runtime backed by `Vec`s and interior-mutable counters. Used by
738/// tests in this crate to verify the trait shapes compile and roundtrip
739/// cleanly, and by the future `claudectl-tui` crate's tests to render the
740/// TUI against fixtures without dragging in brain / coord / bus.
741///
742/// `Actions` impls record their calls in `actions_log` so tests can assert
743/// "the TUI invoked the right side-effect" without spying on real I/O.
744#[derive(Default)]
745pub struct MockRuntime {
746    pub sessions: Vec<SessionSnapshot>,
747    pub gate_mode: std::sync::Mutex<Option<BrainGateMode>>,
748    pub decisions: Vec<DecisionSummary>,
749    pub review_queue: Vec<ReviewItemSummary>,
750    pub leases: Vec<LeaseSummary>,
751    pub handoffs: Vec<HandoffSummary>,
752    pub interrupts: Vec<InterruptSummary>,
753    pub tasks: Vec<TaskSummary>,
754    pub agents: Vec<AgentDirectoryEntry>,
755    pub roles: Vec<RoleBinding>,
756    /// Paired peers `hive_view_snapshot` reports, so a test can exercise the
757    /// peer picker without a relay on the machine running the suite.
758    pub peers: Vec<(String, Option<String>)>,
759    pub actions_log: std::sync::Mutex<Vec<MockAction>>,
760}
761
762/// Recorded `Actions` calls. Tests use this to assert side-effect ordering
763/// without needing a real terminal or filesystem.
764#[derive(Debug, Clone, PartialEq, Eq)]
765pub enum MockAction {
766    Terminate {
767        pid: u32,
768    },
769    InjectText {
770        session_id: String,
771        text: String,
772    },
773    SetGateMode(BrainGateMode),
774    LogObservation(ObservationInput),
775    LogDecision(LogDecisionInput),
776    MarkCanonical {
777        decision_id: String,
778        note: Option<String>,
779    },
780    BindBusRole {
781        name: String,
782        cwd: String,
783        pid: u32,
784    },
785    CancelTask {
786        task_id: String,
787    },
788    SetSupervisorDrain {
789        draining: bool,
790    },
791    RetryTask {
792        task_id: String,
793    },
794    ApproveTask {
795        task_id: String,
796    },
797    SendSessionToPeer {
798        session_id: String,
799        transcript: Option<std::path::PathBuf>,
800        cwd: String,
801        peer: String,
802    },
803}
804
805impl PartialEq for ObservationInput {
806    fn eq(&self, other: &Self) -> bool {
807        self.session_pid == other.session_pid
808            && self.project == other.project
809            && self.tool == other.tool
810            && self.command == other.command
811            && self.observed_action == other.observed_action
812    }
813}
814
815impl Eq for ObservationInput {}
816
817impl PartialEq for LogDecisionInput {
818    fn eq(&self, other: &Self) -> bool {
819        self.session_pid == other.session_pid
820            && self.project == other.project
821            && self.tool == other.tool
822            && self.command == other.command
823            && self.user_action == other.user_action
824            && self.decision_type == other.decision_type
825            && self.override_reason == other.override_reason
826            // PendingSuggestion's field-by-field equality
827            && self.suggestion.pid == other.suggestion.pid
828            && self.suggestion.action == other.suggestion.action
829            && self.suggestion.message == other.suggestion.message
830            && self.suggestion.reasoning == other.suggestion.reasoning
831    }
832}
833
834impl Eq for LogDecisionInput {}
835
836impl MockRuntime {
837    pub fn into_runtime(self) -> Runtime {
838        let arc = Arc::new(self);
839        Runtime::new(
840            arc.clone(),
841            arc.clone(),
842            arc.clone(),
843            arc.clone(),
844            arc.clone(),
845            arc.clone(),
846            arc.clone(),
847            arc,
848        )
849    }
850
851    /// Like `into_runtime`, but hands back the `Arc` as well so a test can
852    /// read `actions()` after the TUI has driven the runtime. `into_runtime`
853    /// alone moves the mock out of reach.
854    pub fn into_runtime_shared(self) -> (Arc<Self>, Runtime) {
855        let arc = Arc::new(self);
856        let runtime = Runtime::new(
857            arc.clone(),
858            arc.clone(),
859            arc.clone(),
860            arc.clone(),
861            arc.clone(),
862            arc.clone(),
863            arc.clone(),
864            arc.clone(),
865        );
866        (arc, runtime)
867    }
868
869    pub fn actions(&self) -> Vec<MockAction> {
870        self.actions_log
871            .lock()
872            .expect("actions_log poisoned")
873            .clone()
874    }
875}
876
877impl SessionSource for MockRuntime {
878    fn list(&self) -> Vec<SessionSnapshot> {
879        self.sessions.clone()
880    }
881}
882
883impl BrainView for MockRuntime {
884    fn gate_mode(&self) -> BrainGateMode {
885        self.gate_mode
886            .lock()
887            .expect("gate_mode poisoned")
888            .unwrap_or(BrainGateMode::On)
889    }
890    fn recent_decisions(&self, n: usize) -> Vec<DecisionSummary> {
891        self.decisions.iter().take(n).cloned().collect()
892    }
893    fn decision_count(&self) -> usize {
894        self.decisions.len()
895    }
896}
897
898impl CoordView for MockRuntime {
899    fn active_leases(&self) -> Vec<LeaseSummary> {
900        self.leases.clone()
901    }
902    fn pending_handoffs(&self) -> Vec<HandoffSummary> {
903        self.handoffs.clone()
904    }
905    fn pending_interrupts(&self) -> Vec<InterruptSummary> {
906        self.interrupts.clone()
907    }
908    fn tasks(&self) -> Vec<TaskSummary> {
909        self.tasks.clone()
910    }
911}
912
913impl BusView for MockRuntime {
914    fn list_agents(&self) -> Vec<AgentDirectoryEntry> {
915        self.agents.clone()
916    }
917    fn list_roles(&self) -> Vec<RoleBinding> {
918        self.roles.clone()
919    }
920}
921
922impl BrainReviewView for MockRuntime {
923    fn all_decisions(&self) -> Vec<DecisionSummary> {
924        self.decisions.clone()
925    }
926    fn review_queue(&self) -> Vec<ReviewItemSummary> {
927        self.review_queue.clone()
928    }
929}
930
931impl Orchestrator for MockRuntime {
932    fn deliver_mailbox(&self, _sessions: &[SessionSnapshot]) -> Vec<(u32, String)> {
933        Vec::new()
934    }
935    fn deliver_interrupts(&self, _sessions: &[SessionSnapshot]) -> Vec<(String, String)> {
936        Vec::new()
937    }
938    fn expire_stale(&self) {}
939}
940
941impl HiveActions for MockRuntime {
942    fn shared_skill_keys(&self) -> std::collections::HashSet<String> {
943        std::collections::HashSet::new()
944    }
945    fn share_skill(&self, _skill: &crate::skills::DiscoveredSkill) -> Result<String, String> {
946        Err("MockRuntime does not implement skill sharing".into())
947    }
948    fn hive_view_snapshot(&self) -> HiveViewSnapshot {
949        HiveViewSnapshot {
950            identity: Some("mock-local".into()),
951            peers: self.peers.clone(),
952        }
953    }
954}
955
956impl Actions for MockRuntime {
957    fn terminate_session(&self, pid: u32) -> Result<(), String> {
958        self.actions_log
959            .lock()
960            .expect("actions_log poisoned")
961            .push(MockAction::Terminate { pid });
962        Ok(())
963    }
964    fn inject_text(&self, session_id: &str, text: &str) -> Result<(), String> {
965        self.actions_log
966            .lock()
967            .expect("actions_log poisoned")
968            .push(MockAction::InjectText {
969                session_id: session_id.into(),
970                text: text.into(),
971            });
972        Ok(())
973    }
974    fn set_gate_mode(&self, mode: BrainGateMode) -> Result<(), String> {
975        *self.gate_mode.lock().expect("gate_mode poisoned") = Some(mode);
976        self.actions_log
977            .lock()
978            .expect("actions_log poisoned")
979            .push(MockAction::SetGateMode(mode));
980        Ok(())
981    }
982    fn log_observation(&self, observation: ObservationInput) -> Result<(), String> {
983        self.actions_log
984            .lock()
985            .expect("actions_log poisoned")
986            .push(MockAction::LogObservation(observation));
987        Ok(())
988    }
989    fn log_decision(&self, input: LogDecisionInput) -> Result<(), String> {
990        self.actions_log
991            .lock()
992            .expect("actions_log poisoned")
993            .push(MockAction::LogDecision(input));
994        Ok(())
995    }
996    fn mark_canonical(&self, decision_id: &str, note: Option<String>) -> Result<(), String> {
997        self.actions_log
998            .lock()
999            .expect("actions_log poisoned")
1000            .push(MockAction::MarkCanonical {
1001                decision_id: decision_id.into(),
1002                note,
1003            });
1004        Ok(())
1005    }
1006    fn bind_bus_role(&self, name: &str, cwd: &str, pid: u32) -> Result<(), String> {
1007        self.actions_log
1008            .lock()
1009            .expect("actions_log poisoned")
1010            .push(MockAction::BindBusRole {
1011                name: name.into(),
1012                cwd: cwd.into(),
1013                pid,
1014            });
1015        Ok(())
1016    }
1017    fn cancel_task(&self, task_id: &str) -> Result<(), String> {
1018        self.actions_log
1019            .lock()
1020            .expect("actions_log poisoned")
1021            .push(MockAction::CancelTask {
1022                task_id: task_id.into(),
1023            });
1024        Ok(())
1025    }
1026    fn set_supervisor_drain(&self, draining: bool) -> Result<(), String> {
1027        self.actions_log
1028            .lock()
1029            .expect("actions_log poisoned")
1030            .push(MockAction::SetSupervisorDrain { draining });
1031        Ok(())
1032    }
1033    fn retry_task(&self, task_id: &str) -> Result<(), String> {
1034        self.actions_log
1035            .lock()
1036            .expect("actions_log poisoned")
1037            .push(MockAction::RetryTask {
1038                task_id: task_id.into(),
1039            });
1040        Ok(())
1041    }
1042    fn approve_task(&self, task_id: &str) -> Result<(), String> {
1043        self.actions_log
1044            .lock()
1045            .expect("actions_log poisoned")
1046            .push(MockAction::ApproveTask {
1047                task_id: task_id.into(),
1048            });
1049        Ok(())
1050    }
1051    fn send_session_to_peer(
1052        &self,
1053        session_id: &str,
1054        transcript: Option<&std::path::Path>,
1055        cwd: &str,
1056        peer: &str,
1057    ) -> Result<String, String> {
1058        self.actions_log.lock().expect("actions_log poisoned").push(
1059            MockAction::SendSessionToPeer {
1060                session_id: session_id.into(),
1061                transcript: transcript.map(|p| p.to_path_buf()),
1062                cwd: cwd.into(),
1063                peer: peer.into(),
1064            },
1065        );
1066        Ok(session_id.to_string())
1067    }
1068}
1069
1070#[cfg(test)]
1071mod tests {
1072    use super::*;
1073
1074    fn sample_session(id: &str) -> SessionSnapshot {
1075        SessionSnapshot {
1076            session_id: id.into(),
1077            pid: 12345,
1078            cwd: "/work/proj".into(),
1079            project_name: "proj".into(),
1080            status: "Processing".into(),
1081            cost_usd: 1.23,
1082            context_tokens: 4000,
1083            context_max: 200_000,
1084            last_message_ts: 1_780_000_000,
1085        }
1086    }
1087
1088    #[test]
1089    fn mock_runtime_assembles_and_lists_sessions() {
1090        let mock = MockRuntime {
1091            sessions: vec![sample_session("sess_a"), sample_session("sess_b")],
1092            ..Default::default()
1093        };
1094        let rt = mock.into_runtime();
1095        let listed = rt.sessions.list();
1096        assert_eq!(listed.len(), 2);
1097        assert_eq!(listed[0].session_id, "sess_a");
1098    }
1099
1100    #[test]
1101    fn session_detail_falls_back_to_list_scan() {
1102        let mock = MockRuntime {
1103            sessions: vec![sample_session("sess_a"), sample_session("sess_b")],
1104            ..Default::default()
1105        };
1106        let rt = mock.into_runtime();
1107        assert!(rt.sessions.detail("sess_a").is_some());
1108        assert!(rt.sessions.detail("sess_missing").is_none());
1109    }
1110
1111    #[test]
1112    fn brain_view_returns_recent_decisions_with_cap() {
1113        let mock = MockRuntime {
1114            decisions: (0..5)
1115                .map(|i| DecisionSummary {
1116                    id: format!("dec_{i}"),
1117                    timestamp: format!("2026-06-06T00:00:0{i}Z"),
1118                    action: "approve".into(),
1119                    confidence: Some(0.9),
1120                    project: None,
1121                    tool: None,
1122                    pid: 0,
1123                    command: None,
1124                    reasoning: None,
1125                    user_action: None,
1126                    override_reason: None,
1127                    brain_decision_ms: None,
1128                    canonical: None,
1129                    cache_hit: None,
1130                    cost_usd: None,
1131                    model: None,
1132                    outcome_kind: None,
1133                    outcome_detail: None,
1134                    suggested_at: None,
1135                    resolved_at: None,
1136                    why: String::new(),
1137                    decision_source: None,
1138                    rule_name: None,
1139                })
1140                .collect(),
1141            ..Default::default()
1142        };
1143        let rt = mock.into_runtime();
1144        assert_eq!(rt.brain.recent_decisions(3).len(), 3);
1145        assert_eq!(rt.brain.decision_count(), 5);
1146        assert_eq!(rt.brain.gate_mode(), BrainGateMode::On);
1147    }
1148
1149    #[test]
1150    fn coord_view_reports_empty_state_cleanly() {
1151        let rt = MockRuntime::default().into_runtime();
1152        assert!(rt.coord.active_leases().is_empty());
1153        assert!(rt.coord.pending_handoffs().is_empty());
1154        assert!(rt.coord.pending_interrupts().is_empty());
1155        assert!(rt.coord.tasks().is_empty());
1156    }
1157
1158    #[test]
1159    fn coord_view_returns_task_fixture() {
1160        let mock = MockRuntime {
1161            tasks: vec![TaskSummary {
1162                id: "t1".into(),
1163                name: "ship the thing".into(),
1164                state: "running".into(),
1165                role: Some("backend".into()),
1166                attempts: 2,
1167                max_retries: 3,
1168                last_session_id: Some("sess-abcdef12".into()),
1169                last_verdict: Some("PASS".into()),
1170                cost_usd: 1.25,
1171                created_at: "2026-06-25T10:00:00Z".into(),
1172                updated_at: "2026-06-25T10:05:00Z".into(),
1173            }],
1174            ..Default::default()
1175        };
1176        let rt = mock.into_runtime();
1177        let tasks = rt.coord.tasks();
1178        assert_eq!(tasks.len(), 1);
1179        assert_eq!(tasks[0].state, "running");
1180        assert_eq!(tasks[0].attempts, 2);
1181    }
1182
1183    #[test]
1184    fn bus_view_reports_empty_state_cleanly() {
1185        let rt = MockRuntime::default().into_runtime();
1186        assert!(rt.bus.list_agents().is_empty());
1187        assert!(rt.bus.list_roles().is_empty());
1188    }
1189
1190    /// Smoke test that the trait shapes are usable behind `dyn`. If this
1191    /// compiles, the boxed-trait composition works.
1192    #[test]
1193    fn runtime_implements_clone_and_send() {
1194        fn assert_send<T: Send>() {}
1195        assert_send::<Runtime>();
1196
1197        let rt = MockRuntime::default().into_runtime();
1198        let _rt2 = rt.clone();
1199    }
1200
1201    #[test]
1202    fn actions_terminate_records_in_log() {
1203        let rt = MockRuntime::default().into_runtime();
1204        rt.actions.terminate_session(42).unwrap();
1205        rt.actions.terminate_session(43).unwrap();
1206        // No public mock accessor on Arc<dyn Actions>, but we can downcast
1207        // through the trait by going via the *original* Arc through inspecting
1208        // the trait's behavior — the integration test in the binary covers
1209        // that surface. Here we just assert the calls succeeded.
1210    }
1211
1212    #[test]
1213    fn actions_task_writes_record_in_log() {
1214        let mock = MockRuntime::default();
1215        // MockRuntime itself implements Actions; call directly so we can read
1216        // the recorded log via the public `actions()` accessor.
1217        mock.cancel_task("t-1").unwrap();
1218        mock.set_supervisor_drain(true).unwrap();
1219        mock.retry_task("t-2").unwrap();
1220        mock.approve_task("t-3").unwrap();
1221        let actions = mock.actions();
1222        assert!(matches!(&actions[0], MockAction::CancelTask { task_id } if task_id == "t-1"));
1223        assert!(matches!(
1224            actions[1],
1225            MockAction::SetSupervisorDrain { draining: true }
1226        ));
1227        assert!(matches!(&actions[2], MockAction::RetryTask { task_id } if task_id == "t-2"));
1228        assert!(matches!(&actions[3], MockAction::ApproveTask { task_id } if task_id == "t-3"));
1229    }
1230
1231    #[test]
1232    fn actions_set_gate_mode_round_trips_through_brain_view() {
1233        let mock = MockRuntime::default();
1234        let arc = Arc::new(mock);
1235        let rt = Runtime::new(
1236            arc.clone(),
1237            arc.clone(),
1238            arc.clone(),
1239            arc.clone(),
1240            arc.clone(),
1241            arc.clone(),
1242            arc.clone(),
1243            arc.clone(),
1244        );
1245        // Initial: default On.
1246        assert_eq!(rt.brain.gate_mode(), BrainGateMode::On);
1247        // Flip via Actions; observe via BrainView.
1248        rt.actions.set_gate_mode(BrainGateMode::Off).unwrap();
1249        assert_eq!(rt.brain.gate_mode(), BrainGateMode::Off);
1250        rt.actions.set_gate_mode(BrainGateMode::Auto).unwrap();
1251        assert_eq!(rt.brain.gate_mode(), BrainGateMode::Auto);
1252        // And the log captured both transitions.
1253        let calls = arc.actions();
1254        assert_eq!(
1255            calls,
1256            vec![
1257                MockAction::SetGateMode(BrainGateMode::Off),
1258                MockAction::SetGateMode(BrainGateMode::Auto),
1259            ]
1260        );
1261    }
1262
1263    #[test]
1264    fn actions_log_observation_captures_full_input() {
1265        let mock = MockRuntime::default();
1266        let arc = Arc::new(mock);
1267        let rt = Runtime::new(
1268            arc.clone(),
1269            arc.clone(),
1270            arc.clone(),
1271            arc.clone(),
1272            arc.clone(),
1273            arc.clone(),
1274            arc.clone(),
1275            arc.clone(),
1276        );
1277        let obs = ObservationInput {
1278            session_pid: 12345,
1279            project: "claudectl".into(),
1280            tool: Some("Bash".into()),
1281            command: Some("cargo test".into()),
1282            observed_action: "user_approve".into(),
1283        };
1284        rt.actions.log_observation(obs.clone()).unwrap();
1285        let calls = arc.actions();
1286        assert_eq!(calls, vec![MockAction::LogObservation(obs)]);
1287    }
1288
1289    #[test]
1290    fn actions_inject_text_and_mark_canonical_record_inputs() {
1291        let mock = MockRuntime::default();
1292        let arc = Arc::new(mock);
1293        let rt = Runtime::new(
1294            arc.clone(),
1295            arc.clone(),
1296            arc.clone(),
1297            arc.clone(),
1298            arc.clone(),
1299            arc.clone(),
1300            arc.clone(),
1301            arc.clone(),
1302        );
1303        rt.actions.inject_text("sess_a", "/compact\n").unwrap();
1304        rt.actions
1305            .mark_canonical("dec_42", Some("nice catch".into()))
1306            .unwrap();
1307        let calls = arc.actions();
1308        assert_eq!(
1309            calls,
1310            vec![
1311                MockAction::InjectText {
1312                    session_id: "sess_a".into(),
1313                    text: "/compact\n".into(),
1314                },
1315                MockAction::MarkCanonical {
1316                    decision_id: "dec_42".into(),
1317                    note: Some("nice catch".into()),
1318                },
1319            ]
1320        );
1321    }
1322}