Skip to main content

cli/harness/
mod.rs

1// SPDX-License-Identifier: Apache-2.0
2use std::{
3    collections::{BTreeMap, BTreeSet},
4    fs::{self, OpenOptions},
5    io::{BufRead, BufReader, BufWriter, Write},
6    path::{Path, PathBuf},
7};
8
9use anyhow::{Result, anyhow};
10use base64::Engine as _;
11use chrono::Utc;
12use heddle_core::{
13    ExplicitAgentBind, SessionAttachFacts, SessionLookupFact, SessionPolicy, TokenSidFact,
14    WorktreeSessionFact, decide_session_attach, first_value_string, map_from_pairs,
15    merge_string_vec, opencode_tool_name, opencode_tool_status,
16    parse_relay_payload as core_parse_relay_payload,
17    should_rotate_segment as pure_should_rotate_segment, value_array_join, value_cost_micros,
18    value_cost_micros_u64, value_string, value_string_array, value_u64, value_u64_string,
19};
20use objects::{
21    fs_atomic::write_file_atomic,
22    object::{
23        ContentHash, DiffKind, NativeToolCallRefV1, Session, StateId, ThreadName, TimelineBranchId,
24        TimelineLabel, TimelineOperationBodyV1, TimelineOperationEnvelope, TimelineStepId,
25        TimelineToolCallStatus, TimelineToolPayloadMetadata, ToolCallFinishedV1, ToolCallStartedV1,
26        Tree,
27    },
28    store::{ActorPresence, ActorPresenceStore, ActorPresenceStatus, AgentUsageSummary, ObjectStore},
29};
30use oplog::OpLogRecorder;
31use refs::Head;
32use repo::{
33    Repository, SessionManager, Thread, ThreadFreshness, ThreadIntegrationPolicy, ThreadManager,
34    ThreadMode, ThreadState, TimelineStore, TimelineView,
35};
36use serde::{Deserialize, Serialize};
37use serde_json::Value;
38use wire::{
39    HarnessIdentity, ProgressCheckpoint, SessionDiffSummary, SessionReportEnvelope,
40    TranscriptAttachmentRef, UsageTotals, WorktreeChangeBaseline,
41};
42
43mod claude_hook;
44mod probe;
45
46use self::probe::{HarnessProbeInput, HarnessProbeResult, probe_harness_actor};
47use crate::{
48    cli::{
49        Cli,
50        commands::{
51            snapshot::{
52                SnapshotAgentOverrides, create_snapshot, summarize_confidence,
53                summarize_verification,
54            },
55            worktree_cmd::helpers::{prepare_worktree_target, write_isolated_checkout},
56        },
57        style, worktree_status_options,
58    },
59    config::{
60        HarnessMode, HarnessTranscriptMode, HarnessTransport, UserConfig, UserHarnessOverride,
61        UserHarnessRootThreadPolicy, UserHarnessSubagentThreadPolicy, UserThreadWorkspaceMode,
62    },
63};
64
65pub(crate) fn probe_current_process_harness(
66    repo: &Repository,
67    current_provider: Option<String>,
68    current_model: Option<String>,
69    current_policy: Option<String>,
70) -> Result<HarnessProbeResult> {
71    probe_harness_actor(&HarnessProbeInput {
72        argv: detected_harness_argv().or_else(|| Some(std::env::args().collect())),
73        env_hints: harness_env_hints(),
74        explicit_harness: None,
75        explicit_provider: None,
76        explicit_model: None,
77        explicit_thinking_level: None,
78        explicit_policy: None,
79        probe_metadata: BTreeMap::new(),
80        current_provider,
81        current_model,
82        current_policy,
83        repo_root: repo.root().display().to_string(),
84    })
85}
86
87fn harness_env_hints() -> BTreeMap<String, String> {
88    std::env::vars()
89        .filter(|(key, value)| {
90            !value.trim().is_empty()
91                && (key.starts_with("HEDDLE_AGENT_")
92                    || key.starts_with("CODEX_")
93                    || key.starts_with("CLAUDE")
94                    || key.starts_with("ANTHROPIC_")
95                    || key.starts_with("OPENAI_")
96                    || key.starts_with("OPENCODE_")
97                    || key.starts_with("AIDER_")
98                    || matches!(
99                        key.as_str(),
100                        "MODEL" | "REASONING_EFFORT" | "THINKING_LEVEL"
101                    ))
102        })
103        .collect()
104}
105
106fn detected_harness_argv() -> Option<Vec<String>> {
107    detected_harness_argv_impl()
108}
109
110#[cfg(target_os = "linux")]
111fn detected_harness_argv_impl() -> Option<Vec<String>> {
112    let mut pid = std::process::id();
113    for _ in 0..8 {
114        let stat = fs::read_to_string(format!("/proc/{pid}/stat")).ok()?;
115        let ppid = stat.split_whitespace().nth(3)?.parse::<u32>().ok()?;
116        if ppid == 0 || ppid == pid {
117            return None;
118        }
119        pid = ppid;
120        let raw = fs::read(format!("/proc/{pid}/cmdline")).ok()?;
121        let argv = raw
122            .split(|byte| *byte == 0)
123            .filter(|part| !part.is_empty())
124            .map(|part| String::from_utf8_lossy(part).to_string())
125            .collect::<Vec<_>>();
126        let program = argv.first().map(|arg| arg.to_ascii_lowercase())?;
127        if ["codex", "claude", "opencode", "aider"]
128            .iter()
129            .any(|needle| program.contains(needle))
130        {
131            return Some(argv);
132        }
133    }
134    None
135}
136
137#[cfg(not(target_os = "linux"))]
138fn detected_harness_argv_impl() -> Option<Vec<String>> {
139    None
140}
141
142pub fn cmd_harness_bridge(cli: &Cli) -> Result<()> {
143    let repo = cli.open_repo()?;
144    let mut runtime = init_harness_runtime(&repo)?;
145
146    let stdin = std::io::stdin();
147    let stdout = std::io::stdout();
148    let reader = BufReader::new(stdin.lock());
149    let mut writer = BufWriter::new(stdout.lock());
150
151    for line in reader.lines() {
152        let line = line?;
153        if line.trim().is_empty() {
154            continue;
155        }
156        let response = match serde_json::from_str::<BridgeRequest>(&line) {
157            Ok(request) => runtime.handle_request(request),
158            Err(err) => BridgeResponse::error(
159                None,
160                "invalid_request",
161                format!("failed to parse request: {err}"),
162            ),
163        };
164        serde_json::to_writer(&mut writer, &response)?;
165        writer.write_all(b"\n")?;
166        writer.flush()?;
167    }
168
169    Ok(())
170}
171
172pub(crate) fn relay_harness_event(
173    repo: &Repository,
174    harness: &str,
175    event: &str,
176    payload: &str,
177) -> Result<()> {
178    let mut runtime = init_harness_runtime(repo)?;
179    let (json, warning) = parse_relay_payload(payload);
180    if let Some(warning) = warning {
181        eprintln!("{}", style::warn(&warning));
182    }
183    match harness {
184        "codex" => relay_codex(&mut runtime, event, &json),
185        "claude-code" => relay_claude(&mut runtime, event, &json),
186        "opencode" => relay_opencode(&mut runtime, event, &json),
187        other => Err(anyhow!("unsupported harness relay: {other}")),
188    }
189}
190
191fn init_harness_runtime(repo: &Repository) -> Result<HarnessBridgeRuntime> {
192    let (user_config, warning) = load_harness_user_config(UserConfig::default_path());
193    if let Some(warning) = warning {
194        eprintln!("{}", style::warn(&warning));
195    }
196    Ok(HarnessBridgeRuntime::new(
197        Repository::open(repo.root())?,
198        user_config,
199    ))
200}
201
202fn load_harness_user_config(default_path: Option<PathBuf>) -> (UserConfig, Option<String>) {
203    let Some(path) = default_path else {
204        return (UserConfig::default(), None);
205    };
206    match UserConfig::load(&path) {
207        Ok(config) => (config, None),
208        Err(err) if is_not_found(&err) => (UserConfig::default(), None),
209        Err(err) => {
210            let warning = format!(
211                "warning: failed to load user config from {}: {err}; continuing with defaults",
212                path.display()
213            );
214            (UserConfig::default(), Some(warning))
215        }
216    }
217}
218
219fn is_not_found(err: &anyhow::Error) -> bool {
220    err.downcast_ref::<std::io::Error>()
221        .is_some_and(|io| io.kind() == std::io::ErrorKind::NotFound)
222}
223
224fn parse_relay_payload(payload: &str) -> (Value, Option<String>) {
225    core_parse_relay_payload(payload)
226}
227
228struct HarnessBridgeRuntime {
229    repo: Repository,
230    user_config: UserConfig,
231    reports: SessionReportStore,
232}
233
234struct RegistryEntryRequest<'a> {
235    heddle_session_id: &'a str,
236    thread_name: Option<&'a str>,
237    thread_id: Option<&'a str>,
238    identity: &'a ResolvedIdentity,
239    probe: &'a HarnessProbeResult,
240    attach: &'a ResolvedAttachment,
241    client_instance_id: Option<&'a str>,
242    requested_entry: Option<&'a ActorPresence>,
243}
244
245struct CanonicalActorSessionRequest<'a> {
246    tentative_session: Session,
247    tentative_owns_session: bool,
248    entry: &'a ActorPresence,
249    probe: &'a HarnessProbeResult,
250    attach: &'a mut ResolvedAttachment,
251}
252
253struct AttachmentResolutionInput<'a> {
254    requested_entry: Option<&'a ActorPresence>,
255    explicit_heddle_session_id: Option<&'a str>,
256    client_instance_id: Option<&'a str>,
257    probe: &'a HarnessProbeResult,
258    token_claims: Option<&'a TokenClaims>,
259}
260
261fn relay_codex(runtime: &mut HarnessBridgeRuntime, _event: &str, payload: &Value) -> Result<()> {
262    let metadata = map_from_pairs([
263        (
264            "client_name",
265            value_string(payload, &["client"]).or_else(|| value_string(payload, &["client_name"])),
266        ),
267        ("model", value_string(payload, &["model"])),
268        (
269            "model_provider",
270            value_string(payload, &["model_provider"])
271                .or_else(|| value_string(payload, &["provider"])),
272        ),
273        (
274            "model_reasoning_effort",
275            value_string(payload, &["reasoning_effort"]),
276        ),
277    ]);
278    let opened = runtime.open_session(OpenSessionParams {
279        harness: Some("codex".to_string()),
280        summary: value_string(payload, &["message"]),
281        probe_metadata: metadata,
282        ..OpenSessionParams::default()
283    })?;
284    runtime.update_progress(UpdateProgressParams {
285        heddle_session_id: opened.heddle_session_id,
286        summary: value_string(payload, &["message"]),
287        harness: Some("codex".to_string()),
288        ..UpdateProgressParams::default()
289    })?;
290    Ok(())
291}
292
293fn relay_claude(runtime: &mut HarnessBridgeRuntime, event: &str, payload: &Value) -> Result<()> {
294    let metadata = map_from_pairs([
295        ("session_id", value_string(payload, &["session_id"])),
296        ("agent_id", value_string(payload, &["agent_id"])),
297        ("session_name", value_string(payload, &["session_name"])),
298        (
299            "transcript_path",
300            value_string(payload, &["transcript_path"]),
301        ),
302        (
303            "model",
304            value_string(payload, &["model", "id"]).or_else(|| value_string(payload, &["model"])),
305        ),
306        (
307            "model_display_name",
308            value_string(payload, &["model", "display_name"]),
309        ),
310        ("effort", value_string(payload, &["effort"])),
311        ("hook_event", Some(event.to_string())),
312        (
313            "status_line",
314            (event == "StatusLine").then(|| "1".to_string()),
315        ),
316        (
317            "touched_paths",
318            value_array_join(payload, &["tool_response", "filePaths"])
319                .or_else(|| value_string(payload, &["file_path"])),
320        ),
321        (
322            "input_tokens",
323            value_u64_string(payload, &["context_window", "total_input_tokens"]),
324        ),
325        (
326            "output_tokens",
327            value_u64_string(payload, &["context_window", "total_output_tokens"]),
328        ),
329        (
330            "cost_micros_usd",
331            value_cost_micros(payload, &["cost", "total_cost_usd"]),
332        ),
333    ]);
334    let opened = runtime.open_session(OpenSessionParams {
335        harness: Some("claude-code".to_string()),
336        model: value_string(payload, &["model", "display_name"])
337            .or_else(|| value_string(payload, &["model", "id"]))
338            .or_else(|| value_string(payload, &["model"])),
339        summary: value_string(payload, &["message"]).or_else(|| value_string(payload, &["reason"])),
340        probe_metadata: metadata.clone(),
341        ..OpenSessionParams::default()
342    })?;
343    match event {
344        "SessionEnd" => {
345            runtime.close_session(CloseSessionParams {
346                heddle_session_id: opened.heddle_session_id,
347                summary: value_string(payload, &["reason"])
348                    .or_else(|| value_string(payload, &["stop_hook_active"])),
349                outcome: Some("completed".to_string()),
350                ..CloseSessionParams::default()
351            })?;
352        }
353        "StatusLine" => {
354            runtime.update_progress(UpdateProgressParams {
355                heddle_session_id: opened.heddle_session_id.clone(),
356                harness: Some("claude-code".to_string()),
357                status: Some("StatusLine".to_string()),
358                message: value_string(payload, &["session_name"])
359                    .or_else(|| value_string(payload, &["cwd"]))
360                    .or_else(|| value_string(payload, &["workspace", "current_dir"])),
361                probe_metadata: metadata.clone(),
362                ..UpdateProgressParams::default()
363            })?;
364            runtime.record_usage(RecordUsageParams {
365                heddle_session_id: opened.heddle_session_id,
366                input_tokens: value_u64(payload, &["context_window", "total_input_tokens"]),
367                output_tokens: value_u64(payload, &["context_window", "total_output_tokens"]),
368                reasoning_tokens: value_u64(payload, &["context_window", "total_reasoning_tokens"]),
369                cache_creation_tokens: None,
370                cache_read_tokens: None,
371                tool_calls: None,
372                cost_micros_usd: value_cost_micros_u64(payload, &["cost", "total_cost_usd"]),
373            })?;
374        }
375        "Stop" => {
376            runtime.update_progress(UpdateProgressParams {
377                heddle_session_id: opened.heddle_session_id,
378                harness: Some("claude-code".to_string()),
379                status: Some("Stop".to_string()),
380                message: value_string(payload, &["message"])
381                    .or_else(|| value_string(payload, &["result"]))
382                    .or_else(|| value_string(payload, &["stop_reason"])),
383                probe_metadata: metadata,
384                ..UpdateProgressParams::default()
385            })?;
386            if let Err(err) = claude_hook::handle_stop_capture(
387                &runtime.repo,
388                &runtime.user_config,
389                payload,
390                "Claude Code turn",
391            ) {
392                tracing::warn!(?err, "heddle Stop hook capture failed");
393            }
394        }
395        "SubagentStop" => {
396            runtime.update_progress(UpdateProgressParams {
397                heddle_session_id: opened.heddle_session_id,
398                harness: Some("claude-code".to_string()),
399                status: Some("SubagentStop".to_string()),
400                touched_paths: csv_from_value(metadata.get("touched_paths")),
401                probe_metadata: metadata,
402                ..UpdateProgressParams::default()
403            })?;
404            if let Err(err) = claude_hook::handle_stop_capture(
405                &runtime.repo,
406                &runtime.user_config,
407                payload,
408                "Claude Code subagent turn",
409            ) {
410                tracing::warn!(?err, "heddle SubagentStop hook capture failed");
411            }
412            if let Err(err) = claude_hook::mark_subagent_complete(&runtime.repo, payload) {
413                tracing::debug!(?err, "heddle SubagentStop mark-complete failed");
414            }
415        }
416        "SubagentStart" => {
417            // open_session above has already created (or reattached) the
418            // child `ActorPresence` with `native_parent_actor_key` pointing at
419            // the parent session via the claude-code probe. The explicit
420            // branch exists so the relay's behaviour is traceable in tests
421            // and logs, and to preserve room for future subagent-specific
422            // bookkeeping.
423            runtime.update_progress(UpdateProgressParams {
424                heddle_session_id: opened.heddle_session_id,
425                harness: Some("claude-code".to_string()),
426                status: Some("SubagentStart".to_string()),
427                touched_paths: csv_from_value(metadata.get("touched_paths")),
428                probe_metadata: metadata,
429                ..UpdateProgressParams::default()
430            })?;
431        }
432        "UserPromptSubmit" => {
433            runtime.update_progress(UpdateProgressParams {
434                heddle_session_id: opened.heddle_session_id.clone(),
435                harness: Some("claude-code".to_string()),
436                status: Some("UserPromptSubmit".to_string()),
437                touched_paths: csv_from_value(metadata.get("touched_paths")),
438                probe_metadata: metadata,
439                ..UpdateProgressParams::default()
440            })?;
441            if let Err(err) = claude_hook::handle_user_prompt_segment_rotate(
442                &runtime.repo,
443                &opened.heddle_session_id,
444                payload,
445            ) {
446                tracing::debug!(?err, "heddle UserPromptSubmit segment rotation failed");
447            }
448        }
449        "PreToolUse" => {
450            runtime.update_progress(UpdateProgressParams {
451                heddle_session_id: opened.heddle_session_id,
452                harness: Some("claude-code".to_string()),
453                status: Some("PreToolUse".to_string()),
454                touched_paths: csv_from_value(metadata.get("touched_paths")),
455                probe_metadata: metadata,
456                ..UpdateProgressParams::default()
457            })?;
458            if let Err(err) = claude_hook::handle_pre_tool_use(&runtime.repo, payload) {
459                tracing::debug!(?err, "heddle PreToolUse context inject skipped");
460            }
461        }
462        _ => {
463            runtime.update_progress(UpdateProgressParams {
464                heddle_session_id: opened.heddle_session_id,
465                harness: Some("claude-code".to_string()),
466                status: Some(event.to_string()),
467                touched_paths: csv_from_value(metadata.get("touched_paths")),
468                probe_metadata: metadata,
469                ..UpdateProgressParams::default()
470            })?;
471        }
472    }
473    Ok(())
474}
475
476fn relay_opencode(runtime: &mut HarnessBridgeRuntime, event: &str, payload: &Value) -> Result<()> {
477    let metadata = map_from_pairs([
478        (
479            "session_id",
480            value_string(payload, &["sessionID"])
481                .or_else(|| value_string(payload, &["session_id"])),
482        ),
483        (
484            "parent_id",
485            value_string(payload, &["parentID"]).or_else(|| value_string(payload, &["parent_id"])),
486        ),
487        (
488            "client_name",
489            value_string(payload, &["client"]).or_else(|| std::env::var("OPENCODE_CLIENT").ok()),
490        ),
491        ("model", value_string(payload, &["model"])),
492        ("provider", value_string(payload, &["provider"])),
493        ("hook_event", Some(event.to_string())),
494        (
495            "touched_paths",
496            value_string(payload, &["file", "path"]).or_else(|| value_string(payload, &["path"])),
497        ),
498    ]);
499    let opened = runtime.open_session(OpenSessionParams {
500        harness: Some("opencode".to_string()),
501        model: value_string(payload, &["model"]),
502        provider: value_string(payload, &["provider"]),
503        probe_metadata: metadata.clone(),
504        ..OpenSessionParams::default()
505    })?;
506    let session_id = opened.heddle_session_id.clone();
507    runtime.update_progress(UpdateProgressParams {
508        heddle_session_id: session_id.clone(),
509        harness: Some("opencode".to_string()),
510        status: Some(event.to_string()),
511        touched_paths: csv_from_value(metadata.get("touched_paths")),
512        probe_metadata: metadata,
513        ..UpdateProgressParams::default()
514    })?;
515    if let Err(err) = record_opencode_timeline_event(runtime, event, payload, &opened) {
516        tracing::debug!(?err, event, "heddle OpenCode timeline recording skipped");
517    }
518    Ok(())
519}
520
521#[derive(Clone, Copy, Debug, PartialEq, Eq)]
522enum TimelineToolEvent {
523    Started,
524    Finished,
525}
526
527trait HarnessTimelineExtractor {
528    fn timeline_event(&self, event: &str) -> Option<TimelineToolEvent>;
529    fn native_tool_call(&self, payload: &Value) -> Option<NativeToolCallRefV1>;
530    fn tool_name(&self, payload: &Value) -> String;
531    fn tool_status(&self, payload: &Value) -> TimelineToolCallStatus;
532    fn payload_metadata(&self, event: &str, payload: &Value)
533    -> Result<TimelineToolPayloadMetadata>;
534    fn touched_paths(&self, payload: &Value) -> Vec<String>;
535    fn capture_intent(&self, native: &NativeToolCallRefV1, payload: &Value) -> String;
536
537    fn timeline_thread(
538        &self,
539        runtime: &HarnessBridgeRuntime,
540        opened: &OpenSessionResult,
541    ) -> Result<String> {
542        if let Some(report) = runtime.reports.load(&opened.heddle_session_id)?
543            && let Some(thread) = report.thread
544        {
545            return Ok(thread);
546        }
547        match runtime.repo.head_ref()? {
548            Head::Attached { thread } => Ok(thread.to_string()),
549            Head::Detached { .. } => Ok("main".to_string()),
550        }
551    }
552
553    fn stable_step_id(&self, native: &NativeToolCallRefV1) -> TimelineStepId {
554        let key = format!(
555            "{}\0{}\0{}\0{}",
556            native.harness,
557            native.session_id.as_deref().unwrap_or(""),
558            native.message_id.as_deref().unwrap_or(""),
559            native.tool_call_id
560        );
561        let hash =
562            ContentHash::compute_typed("timeline-native-tool-call-v1", key.as_bytes()).to_hex();
563        TimelineStepId::new(format!("tls-{}", &hash[..24]))
564    }
565
566    fn started_labels(&self, _payload: &Value) -> Vec<TimelineLabel> {
567        vec![TimelineLabel::ExternalSideEffectsUnknown]
568    }
569
570    fn finished_labels(&self, changed: bool, _payload: &Value) -> Vec<TimelineLabel> {
571        if changed {
572            vec![
573                TimelineLabel::RepoReversible,
574                TimelineLabel::ExternalSideEffectsUnknown,
575            ]
576        } else {
577            vec![TimelineLabel::ExternalSideEffectsUnknown]
578        }
579    }
580}
581
582struct OpenCodeTimelineExtractor;
583
584impl HarnessTimelineExtractor for OpenCodeTimelineExtractor {
585    fn timeline_event(&self, event: &str) -> Option<TimelineToolEvent> {
586        match event {
587            "tool.execute.before" => Some(TimelineToolEvent::Started),
588            "tool.execute.after" => Some(TimelineToolEvent::Finished),
589            _ => None,
590        }
591    }
592
593    fn native_tool_call(&self, payload: &Value) -> Option<NativeToolCallRefV1> {
594        opencode_native_tool_call(payload)
595    }
596
597    fn tool_name(&self, payload: &Value) -> String {
598        opencode_tool_name(payload)
599    }
600
601    fn tool_status(&self, payload: &Value) -> TimelineToolCallStatus {
602        opencode_tool_status(payload)
603    }
604
605    fn payload_metadata(
606        &self,
607        event: &str,
608        payload: &Value,
609    ) -> Result<TimelineToolPayloadMetadata> {
610        opencode_payload_metadata(event, payload)
611    }
612
613    fn touched_paths(&self, payload: &Value) -> Vec<String> {
614        opencode_touched_paths(payload)
615    }
616
617    fn capture_intent(&self, native: &NativeToolCallRefV1, payload: &Value) -> String {
618        format!(
619            "OpenCode {} tool call {}",
620            self.tool_name(payload),
621            native.tool_call_id
622        )
623    }
624}
625
626fn record_opencode_timeline_event(
627    runtime: &mut HarnessBridgeRuntime,
628    event: &str,
629    payload: &Value,
630    opened: &OpenSessionResult,
631) -> Result<()> {
632    record_timeline_event(runtime, event, payload, opened, &OpenCodeTimelineExtractor)
633}
634
635fn record_timeline_event<E: HarnessTimelineExtractor>(
636    runtime: &mut HarnessBridgeRuntime,
637    event: &str,
638    payload: &Value,
639    opened: &OpenSessionResult,
640    extractor: &E,
641) -> Result<()> {
642    match extractor.timeline_event(event) {
643        Some(TimelineToolEvent::Started) => {
644            record_timeline_tool_started(runtime, event, payload, opened, extractor)
645        }
646        Some(TimelineToolEvent::Finished) => {
647            record_timeline_tool_finished(runtime, event, payload, opened, extractor)
648        }
649        None => Ok(()),
650    }
651}
652
653fn record_timeline_tool_started<E: HarnessTimelineExtractor>(
654    runtime: &mut HarnessBridgeRuntime,
655    event: &str,
656    payload: &Value,
657    opened: &OpenSessionResult,
658    extractor: &E,
659) -> Result<()> {
660    let Some(native) = extractor.native_tool_call(payload) else {
661        return Ok(());
662    };
663    let Some(before_state) = current_state_id(&runtime.repo)? else {
664        return Ok(());
665    };
666    let thread = extractor.timeline_thread(runtime, opened)?;
667    let store = TimelineStore::open(runtime.repo.heddle_dir())?;
668    let _record_guard = store.lock_recording(&thread)?;
669    let view = TimelineView::rebuild(&store)?;
670    let step_id = extractor.stable_step_id(&native);
671    let (branch_id, parent_step_id) = timeline_position_for_new_tool_step(&view, &thread, &step_id);
672    let envelope = TimelineOperationEnvelope::new(
673        TimelineOperationBodyV1::ToolCallStarted(ToolCallStartedV1 {
674            thread,
675            step_id,
676            branch_id,
677            parent_step_id,
678            native,
679            tool_name: extractor.tool_name(payload),
680            before_state,
681            payload: Some(extractor.payload_metadata(event, payload)?),
682            started_at_ms: Utc::now().timestamp_millis(),
683        }),
684        extractor.started_labels(payload),
685    );
686    store.write_operation(&envelope)?;
687    Ok(())
688}
689
690fn record_timeline_tool_finished<E: HarnessTimelineExtractor>(
691    runtime: &mut HarnessBridgeRuntime,
692    event: &str,
693    payload: &Value,
694    opened: &OpenSessionResult,
695    extractor: &E,
696) -> Result<()> {
697    let Some(native) = extractor.native_tool_call(payload) else {
698        return Ok(());
699    };
700    let Some(fallback_state) = current_state_id(&runtime.repo)? else {
701        return Ok(());
702    };
703    let thread = extractor.timeline_thread(runtime, opened)?;
704    let store = TimelineStore::open(runtime.repo.heddle_dir())?;
705    let _record_guard = store.lock_recording(&thread)?;
706    let before_view = TimelineView::rebuild(&store)?;
707    let step_id = extractor.stable_step_id(&native);
708    let (branch_id, _) = timeline_position_for_new_tool_step(&before_view, &thread, &step_id);
709    let before_state = before_view
710        .step(&thread, &step_id)
711        .and_then(|step| step.before_state)
712        .unwrap_or(fallback_state);
713    let has_worktree_changes_before_capture = !collect_worktree_changes(&runtime.repo)?.is_empty();
714    let mut capture_failed = false;
715    let capture_state = if !has_worktree_changes_before_capture {
716        None
717    } else {
718        let intent = extractor.capture_intent(&native, payload);
719        match create_snapshot(
720            &runtime.repo,
721            &runtime.user_config,
722            Some(intent),
723            None,
724            SnapshotAgentOverrides {
725                provider: opened.provider.clone(),
726                model: opened.model.clone(),
727                session: native.session_id.clone(),
728                segment: None,
729                policy: None,
730                no_policy: false,
731                no_agent: false,
732            },
733        ) {
734            Ok(_) => runtime.repo.head()?,
735            Err(err) => {
736                capture_failed = true;
737                tracing::warn!(?err, "heddle timeline tool capture failed");
738                None
739            }
740        }
741    };
742    let after_state = current_state_id(&runtime.repo)?.unwrap_or(fallback_state);
743    let mut touched_paths = extractor.touched_paths(payload);
744    merge_string_vec(
745        &mut touched_paths,
746        changed_paths_between_states(&runtime.repo, before_state, after_state)?,
747    );
748    let changed = before_state != after_state;
749    let mut labels = extractor.finished_labels(changed, payload);
750    if capture_failed {
751        merge_timeline_labels(&mut labels, vec![TimelineLabel::CaptureFailed]);
752    }
753    let envelope = TimelineOperationEnvelope::new(
754        TimelineOperationBodyV1::ToolCallFinished(ToolCallFinishedV1 {
755            thread,
756            step_id,
757            branch_id,
758            native,
759            status: extractor.tool_status(payload),
760            before_state,
761            after_state,
762            capture_state,
763            capture_oplog_batch_id: None,
764            changed,
765            touched_paths,
766            payload: Some(extractor.payload_metadata(event, payload)?),
767            finished_at_ms: Utc::now().timestamp_millis(),
768        }),
769        labels,
770    );
771    store.write_operation(&envelope)?;
772    Ok(())
773}
774
775fn opencode_native_tool_call(payload: &Value) -> Option<NativeToolCallRefV1> {
776    let tool_call_id = first_value_string(
777        payload,
778        &[
779            &["toolCallID"],
780            &["tool_call_id"],
781            &["toolCallId"],
782            &["callID"],
783            &["call_id"],
784            &["tool", "callID"],
785            &["tool", "call_id"],
786            &["tool", "id"],
787            &["toolCall", "id"],
788            &["tool_call", "id"],
789            &["id"],
790        ],
791    )?;
792    Some(NativeToolCallRefV1 {
793        harness: "opencode".to_string(),
794        session_id: value_string(payload, &["sessionID"])
795            .or_else(|| value_string(payload, &["session_id"])),
796        message_id: value_string(payload, &["messageID"])
797            .or_else(|| value_string(payload, &["message_id"]))
798            .or_else(|| value_string(payload, &["message", "id"])),
799        tool_call_id,
800    })
801}
802
803fn timeline_position_for_new_tool_step(
804    view: &TimelineView,
805    thread: &str,
806    step_id: &TimelineStepId,
807) -> (TimelineBranchId, Option<TimelineStepId>) {
808    let branch_id = view
809        .status(thread)
810        .and_then(|status| status.current_branch_id.clone())
811        .unwrap_or_else(|| TimelineBranchId::new("tlb-main"));
812    let parent_step_id = view
813        .status(thread)
814        .and_then(|status| status.current_step_id.clone())
815        .filter(|current| current != step_id);
816    (branch_id, parent_step_id)
817}
818
819fn current_state_id(repo: &Repository) -> Result<Option<StateId>> {
820    Ok(repo
821        .current_state()?
822        .map(|state| state.state_id)
823        .or(repo.head()?))
824}
825
826// opencode_tool_name / opencode_tool_status: heddle_core::harness_json
827
828fn opencode_payload_metadata(event: &str, payload: &Value) -> Result<TimelineToolPayloadMetadata> {
829    let tool_name = opencode_tool_name(payload);
830    let tool_call_id = opencode_native_tool_call(payload)
831        .map(|native| native.tool_call_id)
832        .unwrap_or_default();
833    let raw = serde_json::to_vec(payload)?;
834    let hash = ContentHash::compute_typed("timeline-tool-payload", &raw);
835    let summary = if tool_call_id.is_empty() {
836        format!("OpenCode {event}: {tool_name}")
837    } else {
838        format!("OpenCode {event}: {tool_name} ({tool_call_id})")
839    };
840    Ok(TimelineToolPayloadMetadata {
841        summary: Some(summary),
842        hash: Some(hash),
843    })
844}
845
846fn opencode_touched_paths(payload: &Value) -> Vec<String> {
847    let mut paths = Vec::new();
848    for path in [
849        value_string(payload, &["file", "path"]),
850        value_string(payload, &["path"]),
851        value_string(payload, &["tool", "path"]),
852        value_string(payload, &["tool", "input", "file_path"]),
853        value_string(payload, &["input", "file_path"]),
854    ]
855    .into_iter()
856    .flatten()
857    {
858        if !path.trim().is_empty() && !paths.contains(&path) {
859            paths.push(path);
860        }
861    }
862    for value_path in [
863        &["paths"][..],
864        &["files"][..],
865        &["tool", "input", "paths"][..],
866        &["input", "paths"][..],
867    ] {
868        if let Some(items) = value_string_array(payload, value_path) {
869            merge_string_vec(&mut paths, items);
870        }
871    }
872    paths
873}
874
875fn merge_timeline_labels(target: &mut Vec<TimelineLabel>, incoming: Vec<TimelineLabel>) {
876    for label in incoming {
877        if !target.contains(&label) {
878            target.push(label);
879        }
880    }
881}
882
883fn csv_from_value(value: Option<&String>) -> Vec<String> {
884    value
885        .map(|value| {
886            value
887                .split(',')
888                .map(|item| item.trim().to_string())
889                .filter(|item| !item.is_empty())
890                .collect()
891        })
892        .unwrap_or_default()
893}
894
895impl HarnessBridgeRuntime {
896    fn new(repo: Repository, user_config: UserConfig) -> Self {
897        let reports = SessionReportStore::new(repo.root());
898        Self {
899            repo,
900            user_config,
901            reports,
902        }
903    }
904
905    fn handle_request(&mut self, request: BridgeRequest) -> BridgeResponse {
906        let response = match request.method.as_str() {
907            "open_session" => self
908                .decode_params::<OpenSessionParams>(request.params)
909                .and_then(|params| self.open_session(params))
910                .and_then(to_json_value),
911            "update_progress" => self
912                .decode_params::<UpdateProgressParams>(request.params)
913                .and_then(|params| self.update_progress(params))
914                .and_then(to_json_value),
915            "record_usage" => self
916                .decode_params::<RecordUsageParams>(request.params)
917                .and_then(|params| self.record_usage(params))
918                .and_then(to_json_value),
919            "record_touched_paths" => self
920                .decode_params::<RecordTouchedPathsParams>(request.params)
921                .and_then(|params| self.record_touched_paths(params))
922                .and_then(to_json_value),
923            "close_session" => self
924                .decode_params::<CloseSessionParams>(request.params)
925                .and_then(|params| self.close_session(params))
926                .and_then(to_json_value),
927            "flush_reports" => self
928                .decode_params::<FlushReportsParams>(request.params)
929                .and_then(|params| self.flush_reports(params))
930                .and_then(to_json_value),
931            other => Err(anyhow!("unknown method '{other}'")),
932        };
933
934        match response {
935            Ok(result) => BridgeResponse::ok(request.id, result),
936            Err(err) => BridgeResponse::error(request.id, "bridge_error", err.to_string()),
937        }
938    }
939
940    fn decode_params<T: for<'de> Deserialize<'de>>(&self, value: Value) -> Result<T> {
941        serde_json::from_value(value).map_err(|err| anyhow!(err))
942    }
943
944    fn open_session(&mut self, params: OpenSessionParams) -> Result<OpenSessionResult> {
945        if self.user_config.harness.mode == HarnessMode::Off {
946            return Err(anyhow!("harness integration is disabled in user config"));
947        }
948
949        let requested_transport = params
950            .transport
951            .unwrap_or(self.user_config.harness.transport);
952        let transcript_mode = params
953            .transcript_mode
954            .unwrap_or(self.user_config.harness.transcript);
955        let env_hints = merged_env_hints(&params.env_hints);
956        let token_claims = active_token_claims();
957        let current_session = SessionManager::new(self.repo.root()).get_current_session()?;
958        let current_segment = current_session
959            .as_ref()
960            .and_then(|session| session.current_segment());
961        let probe = probe_harness_actor(&HarnessProbeInput {
962            argv: params.argv.clone(),
963            env_hints: env_hints.clone(),
964            explicit_harness: params.harness.clone(),
965            explicit_provider: params.provider.clone(),
966            explicit_model: params.model.clone(),
967            explicit_thinking_level: params.thinking_level.clone(),
968            explicit_policy: params.policy.clone(),
969            probe_metadata: params.probe_metadata.clone(),
970            current_provider: current_segment.map(|segment| segment.provider.clone()),
971            current_model: current_segment.map(|segment| segment.model.clone()),
972            current_policy: current_segment.and_then(|segment| segment.policy_id.clone()),
973            repo_root: self.repo.root().display().to_string(),
974        })?;
975        let identity = resolve_identity(
976            &self.repo,
977            &self.user_config,
978            IdentityHints {
979                harness: params.harness.clone(),
980                provider: params.provider.clone(),
981                model: params.model.clone(),
982                thinking_level: params.thinking_level.clone(),
983                policy: params.policy.clone(),
984                probe: probe.clone(),
985            },
986        )?;
987        let registry = ActorPresenceStore::new(self.repo.heddle_dir());
988        let requested_entry = resolve_requested_registry_entry(
989            &registry,
990            params.agent_session_id.as_deref(),
991            params.client_instance_id.as_deref(),
992        )?;
993
994        if self.user_config.harness.mode == HarnessMode::Required
995            && (identity.harness.is_none()
996                || identity.provider.is_none()
997                || identity.model.is_none())
998        {
999            return Err(anyhow!(
1000                "harness mode is 'required' but harness/provider/model could not be resolved"
1001            ));
1002        }
1003
1004        let mut sessions = SessionManager::new(self.repo.root());
1005        let principal = self.repo.get_principal()?;
1006        let mut attach = resolve_actor_attachment(
1007            &registry,
1008            &self.repo,
1009            &mut sessions,
1010            AttachmentResolutionInput {
1011                requested_entry: requested_entry.as_ref(),
1012                explicit_heddle_session_id: params.heddle_session_id.as_deref(),
1013                client_instance_id: params.client_instance_id.as_deref(),
1014                probe: &probe,
1015                token_claims: token_claims.as_ref(),
1016            },
1017        )?;
1018        let (session, owns_session) = match &attach.target {
1019            AttachTarget::ExistingSession(session) => {
1020                let segment_id = session.current_segment_id.clone().unwrap_or_default();
1021                sessions.set_current_session(&session.id, &segment_id)?;
1022                (session.clone(), false)
1023            }
1024            AttachTarget::CreateNew {
1025                _because_claimed: _,
1026            } => {
1027                let session = sessions.start_session(
1028                    principal,
1029                    identity
1030                        .provider
1031                        .clone()
1032                        .unwrap_or_else(|| "unknown".to_string()),
1033                    identity
1034                        .model
1035                        .clone()
1036                        .unwrap_or_else(|| "unknown".to_string()),
1037                    identity.policy.clone(),
1038                )?;
1039                (session, true)
1040            }
1041        };
1042
1043        let (thread_name, thread_id) =
1044            self.resolve_harness_thread_binding(&params, &probe, &identity)?;
1045        let entry = self.ensure_registry_entry(RegistryEntryRequest {
1046            heddle_session_id: &session.id,
1047            thread_name: thread_name.as_deref(),
1048            thread_id: thread_id.as_deref(),
1049            identity: &identity,
1050            probe: &probe,
1051            attach: &attach,
1052            client_instance_id: params.client_instance_id.as_deref(),
1053            requested_entry: requested_entry.as_ref(),
1054        })?;
1055        let (session, owns_session) = self.reuse_canonical_actor_session(
1056            &mut sessions,
1057            CanonicalActorSessionRequest {
1058                tentative_session: session,
1059                tentative_owns_session: owns_session,
1060                entry: &entry,
1061                probe: &probe,
1062                attach: &mut attach,
1063            },
1064        )?;
1065
1066        let mut segment_id = session.current_segment_id.clone().unwrap_or_default();
1067        if should_rotate_segment(&session, &identity) {
1068            let segment = sessions.add_segment(
1069                &session.id,
1070                identity
1071                    .provider
1072                    .clone()
1073                    .unwrap_or_else(|| "unknown".to_string()),
1074                identity
1075                    .model
1076                    .clone()
1077                    .unwrap_or_else(|| "unknown".to_string()),
1078                identity.policy.clone(),
1079            )?;
1080            segment_id = segment.id;
1081        }
1082
1083        let base_state = self
1084            .repo
1085            .current_state()?
1086            .map(|state| state.state_id.to_string_full())
1087            .or_else(|| {
1088                self.repo
1089                    .head()
1090                    .ok()
1091                    .flatten()
1092                    .map(|id| id.to_string_full())
1093            });
1094        let worktree_changes_at_open = capture_worktree_change_snapshot(&self.repo)?;
1095        let opened_at = Utc::now().to_rfc3339();
1096        let mut report = SessionReportEnvelope {
1097            version: 1,
1098            heddle_session_id: session.id.clone(),
1099            heddle_segment_id: (!segment_id.is_empty()).then_some(segment_id.clone()),
1100            agent_session_id: Some(entry.session_id.clone()),
1101            client_instance_id: entry.client_instance_id.clone(),
1102            native_actor_key: entry.native_actor_key.clone(),
1103            native_parent_actor_key: entry.native_parent_actor_key.clone(),
1104            native_instance_key: entry.native_instance_key.clone(),
1105            repo_root: self.repo.root().display().to_string(),
1106            thread: thread_name.clone(),
1107            thread_id,
1108            task: params.task.clone(),
1109            summary: params.summary.clone(),
1110            opened_at,
1111            closed_at: None,
1112            base_state_at_open: base_state.clone(),
1113            worktree_changes_at_open,
1114            head_state_at_close: None,
1115            transport_mode: transport_mode_name(requested_transport).to_string(),
1116            transcript_mode: transcript_mode_name(transcript_mode).to_string(),
1117            outcome: None,
1118            harness: identity.to_transport_identity(),
1119            progress: Vec::new(),
1120            usage: UsageTotals::default(),
1121            touched_paths: Vec::new(),
1122            changed_paths: Vec::new(),
1123            diff_summary: None,
1124            transcript_refs: Vec::new(),
1125            last_progress_at: None,
1126            report_flush_state: Some("pending-local".to_string()),
1127            attach_reason: Some(attach.attach_reason.clone()),
1128            attach_precedence: attach.precedence.clone(),
1129            winning_attach_rule: Some(attach.winning_rule.clone()),
1130            probe_source: probe.probe_source.clone(),
1131            probe_confidence: probe.confidence,
1132            pending_flush: true,
1133            last_flushed_at: None,
1134            owns_session,
1135        };
1136        merge_unique_paths(&mut report.touched_paths, probe.touched_paths.clone());
1137        merge_usage(&mut report.usage, &probe.usage_totals);
1138        if transcript_mode != HarnessTranscriptMode::Off {
1139            report.transcript_refs = probe.transcript_refs.clone();
1140        }
1141        self.reports.save(&report)?;
1142        self.sync_registry_from_report(&report, ActorPresenceStatus::Active)?;
1143        if matches!(requested_transport, HarnessTransport::Direct) {
1144            enqueue_report(&self.reports, &mut report)?;
1145            self.sync_registry_from_report(&report, ActorPresenceStatus::Active)?;
1146        }
1147
1148        Ok(OpenSessionResult {
1149            heddle_session_id: report.heddle_session_id.clone(),
1150            heddle_segment_id: report.heddle_segment_id.clone(),
1151            agent_session_id: report.agent_session_id.clone(),
1152            created_session: owns_session,
1153            harness: report.harness.harness.clone(),
1154            provider: report.harness.provider.clone(),
1155            model: report.harness.model.clone(),
1156            thinking_level: report.harness.thinking_level.clone(),
1157            report_flush_state: report.report_flush_state.clone(),
1158            attach_reason: report.attach_reason.clone(),
1159        })
1160    }
1161
1162    fn update_progress(&mut self, params: UpdateProgressParams) -> Result<SessionMutationResult> {
1163        let mut report = self
1164            .reports
1165            .load(&params.heddle_session_id)?
1166            .ok_or_else(|| anyhow!("session report not found for {}", params.heddle_session_id))?;
1167        let current_session = SessionManager::new(self.repo.root()).get_current_session()?;
1168        let current_segment = current_session
1169            .as_ref()
1170            .and_then(|session| session.current_segment());
1171        let probe = probe_harness_actor(&HarnessProbeInput {
1172            argv: params.argv.clone(),
1173            env_hints: merged_env_hints(&params.env_hints),
1174            explicit_harness: params.harness.clone(),
1175            explicit_provider: params.provider.clone(),
1176            explicit_model: params.model.clone(),
1177            explicit_thinking_level: params.thinking_level.clone(),
1178            explicit_policy: params.policy.clone(),
1179            probe_metadata: params.probe_metadata.clone(),
1180            current_provider: current_segment.map(|segment| segment.provider.clone()),
1181            current_model: current_segment.map(|segment| segment.model.clone()),
1182            current_policy: current_segment.and_then(|segment| segment.policy_id.clone()),
1183            repo_root: self.repo.root().display().to_string(),
1184        })?;
1185        let identity = resolve_identity(
1186            &self.repo,
1187            &self.user_config,
1188            IdentityHints {
1189                harness: params.harness.clone(),
1190                provider: params.provider.clone(),
1191                model: params.model.clone(),
1192                thinking_level: params.thinking_level.clone(),
1193                policy: params.policy.clone(),
1194                probe: probe.clone(),
1195            },
1196        )?;
1197        self.ensure_segment_for_report(&mut report, &identity)?;
1198        if report.harness.harness.is_none() {
1199            report.harness.harness = identity.harness.clone();
1200        }
1201        if report.harness.provider.is_none() {
1202            report.harness.provider = identity.provider.clone();
1203        }
1204        if report.harness.model.is_none() {
1205            report.harness.model = identity.model.clone();
1206        }
1207        if report.harness.thinking_level.is_none() {
1208            report.harness.thinking_level = identity.thinking_level.clone();
1209        }
1210        if report.harness.policy.is_none() {
1211            report.harness.policy = identity.policy.clone();
1212        }
1213        if report.native_actor_key.is_none() {
1214            report.native_actor_key = probe.native_actor_key.clone();
1215        }
1216        if report.native_parent_actor_key.is_none() {
1217            report.native_parent_actor_key = probe.native_parent_actor_key.clone();
1218        }
1219        if report.native_instance_key.is_none() {
1220            report.native_instance_key = probe.native_instance_key.clone();
1221        }
1222        if report.probe_source.is_none() {
1223            report.probe_source = probe.probe_source.clone();
1224        }
1225        if report.probe_confidence.is_none() {
1226            report.probe_confidence = probe.confidence;
1227        }
1228
1229        let recorded_at = Utc::now().to_rfc3339();
1230        let checkpoint = ProgressCheckpoint {
1231            status: params.status.clone(),
1232            message: params.message.clone(),
1233            completed_steps: params.completed_steps,
1234            total_steps: params.total_steps,
1235            touched_paths: normalize_paths(
1236                params
1237                    .touched_paths
1238                    .into_iter()
1239                    .chain(probe.touched_paths)
1240                    .collect::<Vec<_>>(),
1241            ),
1242            recorded_at: recorded_at.clone(),
1243        };
1244        merge_unique_paths(
1245            &mut report.touched_paths,
1246            checkpoint.touched_paths.iter().cloned(),
1247        );
1248        merge_usage(&mut report.usage, &probe.usage_totals);
1249        if report.transcript_mode != "off" && report.transcript_refs.is_empty() {
1250            report.transcript_refs = probe.transcript_refs;
1251        }
1252        report.progress.push(checkpoint);
1253        if let Some(summary) = params.summary {
1254            report.summary = Some(summary);
1255        }
1256        report.last_progress_at = Some(recorded_at);
1257        mark_pending_flush(&mut report);
1258        self.persist_report(report)
1259    }
1260
1261    fn resolve_harness_thread_binding(
1262        &self,
1263        params: &OpenSessionParams,
1264        probe: &HarnessProbeResult,
1265        identity: &ResolvedIdentity,
1266    ) -> Result<(Option<String>, Option<String>)> {
1267        if let Some(thread) = params.thread.clone() {
1268            let thread_id = thread_id_for_name(&self.repo, Some(&thread))?;
1269            return Ok((Some(thread), thread_id));
1270        }
1271
1272        let current_attached = match self.repo.head_ref()? {
1273            Head::Attached { thread } => Some(thread.to_string()),
1274            Head::Detached { .. } => None,
1275        };
1276
1277        if !probe.attach_hints.root_actor
1278            && self.user_config.harness.threading.subagent
1279                == UserHarnessSubagentThreadPolicy::CreateChild
1280            && let Some(parent_thread) =
1281                resolve_parent_thread_for_subagent(&self.repo, probe, current_attached.as_deref())?
1282            && can_create_harness_thread(&self.repo, Some(&parent_thread), Some(&parent_thread))?
1283        {
1284            let name = allocate_thread_name(
1285                &self.repo,
1286                &format!(
1287                    "{}/{}",
1288                    parent_thread,
1289                    sanitize_name(&preferred_thread_slug(params, probe, identity))
1290                ),
1291            )?;
1292            self.ensure_harness_thread(
1293                &name,
1294                Some(&parent_thread),
1295                Some(&parent_thread),
1296                params.task.clone(),
1297            )?;
1298            let thread_id = thread_id_for_name(&self.repo, Some(&name))?;
1299            return Ok((Some(name), thread_id));
1300        }
1301
1302        if probe.attach_hints.root_actor
1303            && self.user_config.harness.threading.root_actor
1304                == UserHarnessRootThreadPolicy::CreateNew
1305            && let Some(current) = current_attached.clone()
1306            && can_create_harness_thread(&self.repo, Some(&current), None)?
1307        {
1308            let name = allocate_thread_name(
1309                &self.repo,
1310                &format!(
1311                    "{}/{}",
1312                    current,
1313                    sanitize_name(&preferred_thread_slug(params, probe, identity))
1314                ),
1315            )?;
1316            self.ensure_harness_thread(&name, Some(&current), None, params.task.clone())?;
1317            let thread_id = thread_id_for_name(&self.repo, Some(&name))?;
1318            return Ok((Some(name), thread_id));
1319        }
1320
1321        let thread_id = thread_id_for_name(&self.repo, current_attached.as_deref())?;
1322        Ok((current_attached, thread_id))
1323    }
1324
1325    fn ensure_harness_thread(
1326        &self,
1327        name: &str,
1328        target_thread: Option<&str>,
1329        parent_thread: Option<&str>,
1330        task: Option<String>,
1331    ) -> Result<()> {
1332        let manager = ThreadManager::new(self.repo.heddle_dir());
1333        if manager.load(name)?.is_some() {
1334            return Ok(());
1335        }
1336
1337        let base_state = self
1338            .resolve_harness_thread_base_state(target_thread, parent_thread)?
1339            .ok_or_else(|| anyhow!("No current state to start a thread from"))?;
1340        let tn = ThreadName::new(name);
1341        if self.repo.refs().get_thread(&tn)?.is_none() {
1342            self.repo
1343                .refs()
1344                .set_thread_cas(&tn, refs::RefExpectation::Missing, &base_state)?;
1345            // Harness writes the ThreadManager record later in this
1346            // function (after materializing); no record exists to
1347            // snapshot at recording time. `None` matches the pattern
1348            // used by `cmd_start` / agent reservation. heddle#23 r2.
1349            self.repo.oplog().record_thread_create(
1350                &tn,
1351                &base_state,
1352                None,
1353                Some(&self.repo.op_scope()),
1354            )?;
1355        }
1356
1357        let workspace_mode = self
1358            .user_config
1359            .harness
1360            .threading
1361            .workspace_default
1362            .unwrap_or(UserThreadWorkspaceMode::Materialized);
1363        let thread_mode = match workspace_mode {
1364            UserThreadWorkspaceMode::Materialized | UserThreadWorkspaceMode::Auto => {
1365                ThreadMode::Materialized
1366            }
1367            UserThreadWorkspaceMode::Virtualized => ThreadMode::Virtualized,
1368            UserThreadWorkspaceMode::Solid => ThreadMode::Solid,
1369        };
1370        let path = match thread_mode {
1371            ThreadMode::Solid | ThreadMode::Materialized => {
1372                default_private_thread_path(&self.repo, name)
1373            }
1374            // Harness-managed light workspaces still need mount lifecycle
1375            // wiring before they can become the default execution root.
1376            ThreadMode::Virtualized => default_private_thread_path(&self.repo, name),
1377        };
1378        let abs_path = prepare_worktree_target(&self.repo, &path, Some(name))?.path;
1379        write_isolated_checkout(&self.repo, &abs_path, &base_state, Some(name))?;
1380
1381        let base_state_obj = self
1382            .repo
1383            .store()
1384            .get_state(&base_state)?
1385            .ok_or_else(|| anyhow!("Base state '{}' not found", base_state.short()))?;
1386        let thread = Thread {
1387            id: name.to_string(),
1388            thread: name.to_string(),
1389            target_thread: target_thread.map(ToString::to_string),
1390            parent_thread: parent_thread.map(ToString::to_string),
1391            mode: thread_mode.clone(),
1392            state: ThreadState::Active,
1393            base_state: base_state.short(),
1394            base_root: base_state_obj.tree.short(),
1395            current_state: Some(base_state.short()),
1396            merged_state: None,
1397            task,
1398            execution_path: abs_path.clone(),
1399            materialized_path: match thread_mode {
1400                ThreadMode::Solid => Some(abs_path),
1401                // See note above: harness can't currently produce
1402                // Virtualized, so defaulting to None matches the
1403                // Lightweight branch.
1404                ThreadMode::Materialized | ThreadMode::Virtualized => None,
1405            },
1406            changed_paths: vec![],
1407            impact_categories: vec![],
1408            heavy_impact_paths: vec![],
1409            promotion_suggested: false,
1410            freshness: if target_thread.is_some() {
1411                ThreadFreshness::Current
1412            } else {
1413                ThreadFreshness::Unknown
1414            },
1415            verification_summary: summarize_verification(base_state_obj.verification.as_ref()),
1416            confidence_summary: summarize_confidence(base_state_obj.confidence),
1417            integration_policy_result: ThreadIntegrationPolicy::default(),
1418            created_at: Utc::now(),
1419            updated_at: Utc::now(),
1420            ephemeral: None,
1421            // Mark this as harness-created so `heddle thread list`
1422            // hides it by default and `heddle thread cleanup --auto`
1423            // can sweep it once stale. (Item 2.2 of the heddle 6→8
1424            // plan.)
1425            auto: true,
1426            // The harness's create-on-rotate path doesn't materialize
1427            // a heavy checkout, so there's nothing to redirect.
1428            shared_target_dir: None,
1429        };
1430        manager.save(&thread)?;
1431        Ok(())
1432    }
1433
1434    fn resolve_harness_thread_base_state(
1435        &self,
1436        target_thread: Option<&str>,
1437        parent_thread: Option<&str>,
1438    ) -> Result<Option<objects::object::StateId>> {
1439        resolve_harness_thread_base_state(&self.repo, target_thread, parent_thread)
1440    }
1441
1442    fn record_usage(&mut self, params: RecordUsageParams) -> Result<SessionMutationResult> {
1443        let mut report = self
1444            .reports
1445            .load(&params.heddle_session_id)?
1446            .ok_or_else(|| anyhow!("session report not found for {}", params.heddle_session_id))?;
1447        if let Some(input) = params.input_tokens {
1448            report.usage.input_tokens = Some(max_u64(report.usage.input_tokens, input));
1449        }
1450        if let Some(output) = params.output_tokens {
1451            report.usage.output_tokens = Some(max_u64(report.usage.output_tokens, output));
1452        }
1453        if let Some(reasoning) = params.reasoning_tokens {
1454            report.usage.reasoning_tokens = Some(max_u64(report.usage.reasoning_tokens, reasoning));
1455        }
1456        if let Some(cache_creation) = params.cache_creation_tokens {
1457            report.usage.cache_creation_tokens =
1458                Some(max_u64(report.usage.cache_creation_tokens, cache_creation));
1459        }
1460        if let Some(cache_read) = params.cache_read_tokens {
1461            report.usage.cache_read_tokens =
1462                Some(max_u64(report.usage.cache_read_tokens, cache_read));
1463        }
1464        if let Some(tool_calls) = params.tool_calls {
1465            report.usage.tool_calls = Some(max_u32(report.usage.tool_calls, tool_calls));
1466        }
1467        if let Some(cost) = params.cost_micros_usd {
1468            report.usage.cost_micros_usd = Some(max_u64(report.usage.cost_micros_usd, cost));
1469        }
1470        mark_pending_flush(&mut report);
1471        self.persist_report(report)
1472    }
1473
1474    fn record_touched_paths(
1475        &mut self,
1476        params: RecordTouchedPathsParams,
1477    ) -> Result<SessionMutationResult> {
1478        let mut report = self
1479            .reports
1480            .load(&params.heddle_session_id)?
1481            .ok_or_else(|| anyhow!("session report not found for {}", params.heddle_session_id))?;
1482        merge_unique_paths(&mut report.touched_paths, normalize_paths(params.paths));
1483        mark_pending_flush(&mut report);
1484        self.persist_report(report)
1485    }
1486
1487    fn close_session(&mut self, params: CloseSessionParams) -> Result<CloseSessionResult> {
1488        let mut report = self
1489            .reports
1490            .load(&params.heddle_session_id)?
1491            .ok_or_else(|| anyhow!("session report not found for {}", params.heddle_session_id))?;
1492        report.closed_at = Some(Utc::now().to_rfc3339());
1493        report.outcome = params.outcome.clone();
1494        if let Some(summary) = params.summary {
1495            report.summary = Some(summary);
1496        }
1497        if let Some(transcript_refs) = params.transcript_refs {
1498            report.transcript_refs = transcript_refs;
1499        }
1500        let final_diff = compute_final_diff(
1501            &self.repo,
1502            report.base_state_at_open.as_deref(),
1503            &report.worktree_changes_at_open,
1504        )?;
1505        report.head_state_at_close = final_diff.head_state;
1506        report.changed_paths = final_diff.changed_paths;
1507        report.diff_summary = Some(final_diff.diff_summary);
1508        mark_pending_flush(&mut report);
1509        if report.owns_session {
1510            let mut sessions = SessionManager::new(self.repo.root());
1511            if let Ok(Some(session)) = sessions.get_session(&report.heddle_session_id)
1512                && session.is_active()
1513            {
1514                let _ = sessions.end_session(Some(&report.heddle_session_id));
1515            }
1516        }
1517
1518        let transport = params
1519            .transport
1520            .unwrap_or(self.user_config.harness.transport);
1521        if matches!(transport, HarnessTransport::Direct | HarnessTransport::End) {
1522            enqueue_report(&self.reports, &mut report)?;
1523        } else {
1524            self.reports.save(&report)?;
1525        }
1526        self.sync_registry_from_report(&report, ActorPresenceStatus::Complete)?;
1527        Ok(CloseSessionResult {
1528            heddle_session_id: report.heddle_session_id,
1529            changed_paths: report.changed_paths,
1530            diff_summary: report.diff_summary.unwrap_or_default(),
1531            report_flush_state: report.report_flush_state,
1532        })
1533    }
1534
1535    fn flush_reports(&mut self, params: FlushReportsParams) -> Result<FlushReportsResult> {
1536        let mut flushed = 0usize;
1537        let session_ids = match params.heddle_session_id {
1538            Some(session_id) => vec![session_id],
1539            None => self.reports.list_pending()?,
1540        };
1541        for session_id in session_ids {
1542            let Some(mut report) = self.reports.load(&session_id)? else {
1543                continue;
1544            };
1545            if !report.pending_flush {
1546                continue;
1547            }
1548            enqueue_report(&self.reports, &mut report)?;
1549            let status = if report.closed_at.is_some() {
1550                ActorPresenceStatus::Complete
1551            } else {
1552                ActorPresenceStatus::Active
1553            };
1554            self.sync_registry_from_report(&report, status)?;
1555            flushed += 1;
1556        }
1557        Ok(FlushReportsResult { flushed })
1558    }
1559
1560    fn persist_report(
1561        &mut self,
1562        mut report: SessionReportEnvelope,
1563    ) -> Result<SessionMutationResult> {
1564        let transport = transport_from_report(&report, self.user_config.harness.transport);
1565        match transport {
1566            HarnessTransport::Direct => {
1567                enqueue_report(&self.reports, &mut report)?;
1568            }
1569            HarnessTransport::Spool | HarnessTransport::End => {
1570                self.reports.save(&report)?;
1571            }
1572        }
1573        self.sync_registry_from_report(&report, ActorPresenceStatus::Active)?;
1574        Ok(SessionMutationResult {
1575            heddle_session_id: report.heddle_session_id,
1576            heddle_segment_id: report.heddle_segment_id,
1577            report_flush_state: report.report_flush_state,
1578        })
1579    }
1580
1581    fn ensure_segment_for_report(
1582        &self,
1583        report: &mut SessionReportEnvelope,
1584        identity: &ResolvedIdentity,
1585    ) -> Result<()> {
1586        let mut sessions = SessionManager::new(self.repo.root());
1587        let Some(session) = sessions.get_session(&report.heddle_session_id)? else {
1588            return Ok(());
1589        };
1590        if !session.is_active() || !should_rotate_segment(&session, identity) {
1591            return Ok(());
1592        }
1593        let segment = sessions.add_segment(
1594            &report.heddle_session_id,
1595            identity
1596                .provider
1597                .clone()
1598                .unwrap_or_else(|| "unknown".to_string()),
1599            identity
1600                .model
1601                .clone()
1602                .unwrap_or_else(|| "unknown".to_string()),
1603            identity.policy.clone(),
1604        )?;
1605        report.heddle_segment_id = Some(segment.id);
1606        if identity.provider.is_some() {
1607            report.harness.provider = identity.provider.clone();
1608        }
1609        if identity.model.is_some() {
1610            report.harness.model = identity.model.clone();
1611        }
1612        if identity.policy.is_some() {
1613            report.harness.policy = identity.policy.clone();
1614        }
1615        if identity.thinking_level.is_some() {
1616            report.harness.thinking_level = identity.thinking_level.clone();
1617        }
1618        Ok(())
1619    }
1620
1621    fn ensure_registry_entry(&self, request: RegistryEntryRequest<'_>) -> Result<ActorPresence> {
1622        let RegistryEntryRequest {
1623            heddle_session_id,
1624            thread_name,
1625            thread_id,
1626            identity,
1627            probe,
1628            attach,
1629            client_instance_id,
1630            requested_entry,
1631        } = request;
1632        let registry = ActorPresenceStore::new(self.repo.heddle_dir());
1633        let fallback_entry = if client_instance_id.is_some()
1634            || probe.native_actor_key.is_some()
1635            || probe.native_instance_key.is_some()
1636        {
1637            None
1638        } else {
1639            find_matching_registry_entry(&registry, &self.repo, heddle_session_id, thread_name)?
1640        };
1641        if let Some(entry) = requested_entry
1642            .cloned()
1643            .or_else(|| attach.matched_entry.clone())
1644            .or(fallback_entry)
1645        {
1646            return registry
1647                .update_entry(&entry.session_id, |existing| {
1648                    if client_instance_id.is_some() {
1649                        existing.client_instance_id = client_instance_id.map(ToString::to_string);
1650                    }
1651                    if probe.native_actor_key.is_some() {
1652                        existing.native_actor_key = probe.native_actor_key.clone();
1653                    }
1654                    if probe.native_parent_actor_key.is_some() {
1655                        existing.native_parent_actor_key = probe.native_parent_actor_key.clone();
1656                    }
1657                    if probe.native_instance_key.is_some() {
1658                        existing.native_instance_key = probe.native_instance_key.clone();
1659                    }
1660                    existing.heddle_session_id = Some(heddle_session_id.to_string());
1661                    existing.thread_id = thread_id.map(ToString::to_string);
1662                    if let Some(thread_name) = thread_name {
1663                        existing.thread = thread_name.to_string();
1664                    }
1665                    existing.path = Some(self.repo.root().to_path_buf());
1666                    if identity.provider.is_some() {
1667                        existing.provider = identity.provider.clone();
1668                    }
1669                    if identity.model.is_some() {
1670                        existing.model = identity.model.clone();
1671                    }
1672                    if identity.harness.is_some() {
1673                        existing.harness = identity.harness.clone();
1674                    }
1675                    if identity.thinking_level.is_some() {
1676                        existing.thinking_level = identity.thinking_level.clone();
1677                    }
1678                    existing.attach_reason = Some(attach.attach_reason.clone());
1679                    existing.attach_precedence = attach.precedence.clone();
1680                    existing.winning_attach_rule = Some(attach.winning_rule.clone());
1681                    existing.probe_source = probe.probe_source.clone();
1682                    existing.probe_confidence = probe.confidence;
1683                    existing.status = ActorPresenceStatus::Active;
1684                    existing.completed_at = None;
1685                })?
1686                .ok_or_else(|| anyhow!("registry entry disappeared during update"));
1687        }
1688
1689        if client_instance_id.is_none() && probe.native_actor_key.is_some() {
1690            let (entry, _) = registry.find_or_create_active_entry(
1691                |entry| {
1692                    claude_actor_compatible(entry, probe, self.repo.root())
1693                        && entry.native_actor_key == probe.native_actor_key
1694                },
1695                |existing| {
1696                    if client_instance_id.is_some() {
1697                        existing.client_instance_id = client_instance_id.map(ToString::to_string);
1698                    }
1699                    if existing.heddle_session_id.is_none() {
1700                        existing.heddle_session_id = Some(heddle_session_id.to_string());
1701                    }
1702                    existing.thread_id = thread_id.map(ToString::to_string);
1703                    if let Some(thread_name) = thread_name {
1704                        existing.thread = thread_name.to_string();
1705                    }
1706                    existing.path = Some(self.repo.root().to_path_buf());
1707                    if identity.provider.is_some() {
1708                        existing.provider = identity.provider.clone();
1709                    }
1710                    if identity.model.is_some() {
1711                        existing.model = identity.model.clone();
1712                    }
1713                    if identity.harness.is_some() {
1714                        existing.harness = identity.harness.clone();
1715                    }
1716                    if identity.thinking_level.is_some() {
1717                        existing.thinking_level = identity.thinking_level.clone();
1718                    }
1719                    if probe.native_parent_actor_key.is_some() {
1720                        existing.native_parent_actor_key = probe.native_parent_actor_key.clone();
1721                    }
1722                    if probe.native_instance_key.is_some() {
1723                        existing.native_instance_key = probe.native_instance_key.clone();
1724                    }
1725                    existing.attach_reason = Some(attach.attach_reason.clone());
1726                    existing.attach_precedence = attach.precedence.clone();
1727                    existing.winning_attach_rule = Some(attach.winning_rule.clone());
1728                    existing.probe_source = probe.probe_source.clone();
1729                    existing.probe_confidence = probe.confidence;
1730                    existing.status = ActorPresenceStatus::Active;
1731                    existing.completed_at = None;
1732                },
1733                |session_id| {
1734                    Ok(ActorPresence {
1735                        session_id: session_id.to_string(),
1736                        client_instance_id: client_instance_id.map(ToString::to_string),
1737                        native_actor_key: probe.native_actor_key.clone(),
1738                        native_parent_actor_key: probe.native_parent_actor_key.clone(),
1739                        native_instance_key: probe.native_instance_key.clone(),
1740                        heddle_session_id: Some(heddle_session_id.to_string()),
1741                        thread_id: thread_id.map(ToString::to_string),
1742                        thread: thread_name.unwrap_or("detached").to_string(),
1743                        anchor_state: self.repo.head()?.map(|id| id.to_string_full()),
1744                        anchor_root: None,
1745                        path: Some(self.repo.root().to_path_buf()),
1746                        base_state: self.repo.head()?.map(|id| id.short()).unwrap_or_default(),
1747                        started_at: Utc::now(),
1748                        provider: identity.provider.clone(),
1749                        model: identity.model.clone(),
1750                        harness: identity.harness.clone(),
1751                        thinking_level: identity.thinking_level.clone(),
1752                        usage_summary: AgentUsageSummary::default(),
1753                        last_progress_at: None,
1754                        report_flush_state: Some("pending-local".to_string()),
1755                        attach_reason: Some(attach.attach_reason.clone()),
1756                        task_assignment_id: None,
1757                        attach_precedence: attach.precedence.clone(),
1758                        winning_attach_rule: Some(attach.winning_rule.clone()),
1759                        probe_source: probe.probe_source.clone(),
1760                        probe_confidence: probe.confidence,
1761                        status: ActorPresenceStatus::Active,
1762                        completed_at: None,
1763                        context_queries: vec![],
1764                    })
1765                },
1766            )?;
1767            return Ok(entry);
1768        }
1769
1770        Ok(registry.create_generated_entry(|session_id| {
1771            Ok(ActorPresence {
1772                session_id: session_id.to_string(),
1773                client_instance_id: client_instance_id.map(ToString::to_string),
1774                native_actor_key: probe.native_actor_key.clone(),
1775                native_parent_actor_key: probe.native_parent_actor_key.clone(),
1776                native_instance_key: probe.native_instance_key.clone(),
1777                heddle_session_id: Some(heddle_session_id.to_string()),
1778                thread_id: thread_id.map(ToString::to_string),
1779                thread: thread_name.unwrap_or("detached").to_string(),
1780                anchor_state: self.repo.head()?.map(|id| id.to_string_full()),
1781                anchor_root: None,
1782                path: Some(self.repo.root().to_path_buf()),
1783                base_state: self.repo.head()?.map(|id| id.short()).unwrap_or_default(),
1784                started_at: Utc::now(),
1785                provider: identity.provider.clone(),
1786                model: identity.model.clone(),
1787                harness: identity.harness.clone(),
1788                thinking_level: identity.thinking_level.clone(),
1789                usage_summary: AgentUsageSummary::default(),
1790                last_progress_at: None,
1791                report_flush_state: Some("pending-local".to_string()),
1792                attach_reason: Some(attach.attach_reason.clone()),
1793                task_assignment_id: None,
1794                attach_precedence: attach.precedence.clone(),
1795                winning_attach_rule: Some(attach.winning_rule.clone()),
1796                probe_source: probe.probe_source.clone(),
1797                probe_confidence: probe.confidence,
1798                status: ActorPresenceStatus::Active,
1799                completed_at: None,
1800                context_queries: vec![],
1801            })
1802        })?)
1803    }
1804
1805    fn reuse_canonical_actor_session(
1806        &self,
1807        sessions: &mut SessionManager,
1808        request: CanonicalActorSessionRequest<'_>,
1809    ) -> Result<(Session, bool)> {
1810        let CanonicalActorSessionRequest {
1811            tentative_session,
1812            tentative_owns_session,
1813            entry,
1814            probe,
1815            attach,
1816        } = request;
1817        let Some(canonical_session_id) = entry.heddle_session_id.as_deref() else {
1818            return Ok((tentative_session, tentative_owns_session));
1819        };
1820        if canonical_session_id == tentative_session.id {
1821            return Ok((tentative_session, tentative_owns_session));
1822        }
1823
1824        if tentative_owns_session
1825            && let Ok(Some(session)) = sessions.get_session(&tentative_session.id)
1826            && session.is_active()
1827        {
1828            let _ = sessions.end_session(Some(&tentative_session.id));
1829        }
1830
1831        let canonical_session = sessions
1832            .get_session(canonical_session_id)?
1833            .ok_or_else(|| anyhow!("session not found: {canonical_session_id}"))?;
1834        let canonical_segment_id = canonical_session
1835            .current_segment_id
1836            .clone()
1837            .unwrap_or_default();
1838        sessions.set_current_session(canonical_session_id, &canonical_segment_id)?;
1839
1840        if let Some(native_actor_key) = probe
1841            .native_actor_key
1842            .as_deref()
1843            .or(entry.native_actor_key.as_deref())
1844        {
1845            attach.precedence.push(format!(
1846                "post-create-native-actor-key:{native_actor_key}:matched"
1847            ));
1848            attach.attach_reason = format!(
1849                "reused existing native actor {} on Heddle session {}",
1850                native_actor_key, canonical_session_id
1851            );
1852            attach.winning_rule = "native-actor-key-post-create".to_string();
1853        }
1854
1855        Ok((canonical_session, false))
1856    }
1857
1858    fn sync_registry_from_report(
1859        &self,
1860        report: &SessionReportEnvelope,
1861        status: ActorPresenceStatus,
1862    ) -> Result<()> {
1863        let registry = ActorPresenceStore::new(self.repo.heddle_dir());
1864        let entry = if let Some(agent_session_id) = &report.agent_session_id {
1865            registry.update_entry(agent_session_id, |entry| {
1866                if report.client_instance_id.is_some() {
1867                    entry.client_instance_id = report.client_instance_id.clone();
1868                }
1869                if report.native_actor_key.is_some() {
1870                    entry.native_actor_key = report.native_actor_key.clone();
1871                }
1872                if report.native_parent_actor_key.is_some() {
1873                    entry.native_parent_actor_key = report.native_parent_actor_key.clone();
1874                }
1875                if report.native_instance_key.is_some() {
1876                    entry.native_instance_key = report.native_instance_key.clone();
1877                }
1878                entry.heddle_session_id = Some(report.heddle_session_id.clone());
1879                entry.path = Some(self.repo.root().to_path_buf());
1880                entry.harness = report.harness.harness.clone();
1881                entry.provider = report.harness.provider.clone();
1882                entry.model = report.harness.model.clone();
1883                entry.thinking_level = report.harness.thinking_level.clone();
1884                entry.usage_summary = usage_to_summary(&report.usage);
1885                entry.last_progress_at =
1886                    report.last_progress_at.as_deref().and_then(parse_timestamp);
1887                entry.report_flush_state = report.report_flush_state.clone();
1888                entry.attach_reason = report.attach_reason.clone();
1889                entry.attach_precedence = report.attach_precedence.clone();
1890                entry.winning_attach_rule = report.winning_attach_rule.clone();
1891                entry.probe_source = report.probe_source.clone();
1892                entry.probe_confidence = report.probe_confidence;
1893                entry.status = status.clone();
1894                entry.completed_at = match status {
1895                    ActorPresenceStatus::Active => None,
1896                    ActorPresenceStatus::Abandoned | ActorPresenceStatus::Complete | ActorPresenceStatus::Merged => {
1897                        Some(Utc::now())
1898                    }
1899                };
1900            })?
1901        } else {
1902            None
1903        };
1904
1905        if entry.is_none() {
1906            let resolved = self.ensure_registry_entry(RegistryEntryRequest {
1907                heddle_session_id: &report.heddle_session_id,
1908                thread_name: report.thread.as_deref(),
1909                thread_id: report.thread_id.as_deref(),
1910                identity: &ResolvedIdentity {
1911                    harness: report.harness.harness.clone(),
1912                    provider: report.harness.provider.clone(),
1913                    model: report.harness.model.clone(),
1914                    thinking_level: report.harness.thinking_level.clone(),
1915                    policy: report.harness.policy.clone(),
1916                },
1917                probe: &HarnessProbeResult {
1918                    native_actor_key: report.native_actor_key.clone(),
1919                    native_parent_actor_key: report.native_parent_actor_key.clone(),
1920                    native_instance_key: report.native_instance_key.clone(),
1921                    probe_source: report.probe_source.clone(),
1922                    confidence: report.probe_confidence,
1923                    ..HarnessProbeResult::default()
1924                },
1925                attach: &ResolvedAttachment {
1926                    target: AttachTarget::CreateNew {
1927                        _because_claimed: false,
1928                    },
1929                    matched_entry: None,
1930                    attach_reason: report.attach_reason.clone().unwrap_or_else(|| {
1931                        format!(
1932                            "created actor for Heddle session {}",
1933                            report.heddle_session_id
1934                        )
1935                    }),
1936                    precedence: report.attach_precedence.clone(),
1937                    winning_rule: report
1938                        .winning_attach_rule
1939                        .clone()
1940                        .unwrap_or_else(|| "report-sync".to_string()),
1941                },
1942                client_instance_id: report.client_instance_id.as_deref(),
1943                requested_entry: None,
1944            })?;
1945            let mut report = report.clone();
1946            report.agent_session_id = Some(resolved.session_id);
1947            self.reports.save(&report)?;
1948        }
1949        Ok(())
1950    }
1951}
1952
1953#[derive(Debug, Clone, Default)]
1954struct ResolvedIdentity {
1955    harness: Option<String>,
1956    provider: Option<String>,
1957    model: Option<String>,
1958    thinking_level: Option<String>,
1959    policy: Option<String>,
1960}
1961
1962impl ResolvedIdentity {
1963    fn to_transport_identity(&self) -> HarnessIdentity {
1964        HarnessIdentity {
1965            harness: self.harness.clone(),
1966            provider: self.provider.clone(),
1967            model: self.model.clone(),
1968            thinking_level: self.thinking_level.clone(),
1969            policy: self.policy.clone(),
1970        }
1971    }
1972}
1973
1974struct IdentityHints {
1975    harness: Option<String>,
1976    provider: Option<String>,
1977    model: Option<String>,
1978    thinking_level: Option<String>,
1979    policy: Option<String>,
1980    probe: HarnessProbeResult,
1981}
1982
1983fn resolve_identity(
1984    repo: &Repository,
1985    user_config: &UserConfig,
1986    hints: IdentityHints,
1987) -> Result<ResolvedIdentity> {
1988    let current_session = SessionManager::new(repo.root()).get_current_session()?;
1989    let current_segment = current_session
1990        .as_ref()
1991        .and_then(|session| session.current_segment());
1992    let token_claims = if user_config.harness.auto_infer {
1993        active_token_claims()
1994    } else {
1995        None
1996    };
1997    let harness_override = resolved_harness_override(
1998        user_config,
1999        hints.harness.as_deref(),
2000        hints.probe.harness.as_deref(),
2001    );
2002
2003    Ok(ResolvedIdentity {
2004        harness: hints.harness.or(hints.probe.harness),
2005        provider: hints
2006            .provider
2007            .or(hints.probe.provider)
2008            .or_else(|| current_segment.map(|segment| segment.provider.clone()))
2009            .or_else(|| {
2010                token_claims
2011                    .as_ref()
2012                    .and_then(|claims| claims.agent_provider.clone())
2013            })
2014            .or_else(|| harness_override.and_then(|entry| entry.provider.clone()))
2015            .or_else(|| user_config.agent.provider.clone()),
2016        model: hints
2017            .model
2018            .or(hints.probe.model)
2019            .or_else(|| current_segment.map(|segment| segment.model.clone()))
2020            .or_else(|| {
2021                token_claims
2022                    .as_ref()
2023                    .and_then(|claims| claims.agent_model.clone())
2024            })
2025            .or_else(|| harness_override.and_then(|entry| entry.model.clone()))
2026            .or_else(|| user_config.agent.model.clone()),
2027        thinking_level: hints
2028            .thinking_level
2029            .or(hints.probe.thinking_level)
2030            .or_else(|| harness_override.and_then(|entry| entry.thinking_level.clone())),
2031        policy: hints
2032            .policy
2033            .or(hints.probe.policy)
2034            .or_else(|| current_segment.and_then(|segment| segment.policy_id.clone()))
2035            .or_else(|| harness_override.and_then(|entry| entry.policy.clone()))
2036            .or_else(|| user_config.agent.default_policy.clone()),
2037    })
2038}
2039
2040fn resolved_harness_override<'a>(
2041    user_config: &'a UserConfig,
2042    explicit: Option<&str>,
2043    fingerprint: Option<&str>,
2044) -> Option<&'a UserHarnessOverride> {
2045    explicit
2046        .and_then(|name| user_config.harness.harnesses.get(name))
2047        .or_else(|| fingerprint.and_then(|name| user_config.harness.harnesses.get(name)))
2048}
2049
2050enum AttachTarget {
2051    ExistingSession(objects::object::Session),
2052    CreateNew { _because_claimed: bool },
2053}
2054
2055struct ResolvedAttachment {
2056    target: AttachTarget,
2057    matched_entry: Option<ActorPresence>,
2058    attach_reason: String,
2059    precedence: Vec<String>,
2060    winning_rule: String,
2061}
2062
2063fn resolve_actor_attachment(
2064    registry: &ActorPresenceStore,
2065    repo: &Repository,
2066    sessions: &mut SessionManager,
2067    input: AttachmentResolutionInput<'_>,
2068) -> Result<ResolvedAttachment> {
2069    let AttachmentResolutionInput {
2070        requested_entry,
2071        explicit_heddle_session_id,
2072        client_instance_id,
2073        probe,
2074        token_claims,
2075    } = input;
2076
2077    // CLI owns FS/registry/session I/O; pure core owns attach/create precedence.
2078    let mut sessions_by_id: BTreeMap<String, Session> = BTreeMap::new();
2079    let mut matched_by_session: BTreeMap<String, ActorPresence> = BTreeMap::new();
2080    let mut facts = SessionAttachFacts {
2081        root_actor: probe.attach_hints.root_actor,
2082        ..SessionAttachFacts::default()
2083    };
2084
2085    if let Some(entry) = requested_entry
2086        && let Some(bound_session_id) = entry.heddle_session_id.as_deref()
2087    {
2088        let session = sessions
2089            .get_session(bound_session_id)?
2090            .ok_or_else(|| anyhow!("session not found: {bound_session_id}"))?;
2091        if !session.is_active() {
2092            return Err(anyhow!("session is not active: {bound_session_id}"));
2093        }
2094        matched_by_session.insert(session.id.clone(), entry.clone());
2095        sessions_by_id.insert(session.id.clone(), session);
2096        facts.explicit_agent = Some(ExplicitAgentBind {
2097            agent_session_id: entry.session_id.clone(),
2098            heddle_session_id: bound_session_id.to_string(),
2099        });
2100    }
2101
2102    if facts.explicit_agent.is_none()
2103        && let Some(session_id) = explicit_heddle_session_id
2104    {
2105        ensure_requested_entry_matches_session(requested_entry, session_id)?;
2106        let session = sessions
2107            .get_session(session_id)?
2108            .ok_or_else(|| anyhow!("session not found: {session_id}"))?;
2109        if !session.is_active() {
2110            return Err(anyhow!("session is not active: {session_id}"));
2111        }
2112        sessions_by_id.insert(session.id.clone(), session);
2113        facts.explicit_heddle_session_id = Some(session_id.to_string());
2114    }
2115
2116    if client_instance_id.is_none()
2117        && let Some(native_actor_key) = probe.native_actor_key.as_deref()
2118    {
2119        if let Some(entry) = registry.find_active_by_native_actor_key(native_actor_key)?
2120            && claude_actor_compatible(&entry, probe, repo.root())
2121            && let Some(bound_session_id) = entry.heddle_session_id.clone()
2122        {
2123            let session = sessions
2124                .get_session(&bound_session_id)?
2125                .ok_or_else(|| anyhow!("session not found: {bound_session_id}"))?;
2126            if session.is_active() {
2127                matched_by_session.insert(session.id.clone(), entry);
2128                sessions_by_id.insert(session.id.clone(), session);
2129                facts.native_actor = SessionLookupFact::Hit {
2130                    key: native_actor_key.to_string(),
2131                    session_id: bound_session_id,
2132                };
2133            } else {
2134                facts.native_actor = SessionLookupFact::Miss {
2135                    key: native_actor_key.to_string(),
2136                };
2137            }
2138        } else {
2139            facts.native_actor = SessionLookupFact::Miss {
2140                key: native_actor_key.to_string(),
2141            };
2142        }
2143    }
2144
2145    if let Some(client_instance_id) = client_instance_id {
2146        if let Some(entry) = registry.find_active_by_client_instance_id(client_instance_id)?
2147            && let Some(bound_session_id) = entry.heddle_session_id.clone()
2148        {
2149            let session = sessions
2150                .get_session(&bound_session_id)?
2151                .ok_or_else(|| anyhow!("session not found: {bound_session_id}"))?;
2152            if session.is_active() {
2153                matched_by_session.insert(session.id.clone(), entry);
2154                sessions_by_id.insert(session.id.clone(), session);
2155                facts.client_instance = SessionLookupFact::Hit {
2156                    key: client_instance_id.to_string(),
2157                    session_id: bound_session_id,
2158                };
2159            } else {
2160                facts.client_instance = SessionLookupFact::Miss {
2161                    key: client_instance_id.to_string(),
2162                };
2163            }
2164        } else {
2165            facts.client_instance = SessionLookupFact::Miss {
2166                key: client_instance_id.to_string(),
2167            };
2168        }
2169    }
2170
2171    if let Some(native_instance_key) = probe.native_instance_key.as_deref() {
2172        if let Some(entry) =
2173            registry.find_active_by_native_instance_key_at_path(native_instance_key, repo.root())?
2174            && claude_actor_compatible(&entry, probe, repo.root())
2175            && let Some(bound_session_id) = entry.heddle_session_id.clone()
2176        {
2177            let session = sessions
2178                .get_session(&bound_session_id)?
2179                .ok_or_else(|| anyhow!("session not found: {bound_session_id}"))?;
2180            if session.is_active() {
2181                matched_by_session.insert(session.id.clone(), entry);
2182                sessions_by_id.insert(session.id.clone(), session);
2183                facts.native_instance = SessionLookupFact::Hit {
2184                    key: native_instance_key.to_string(),
2185                    session_id: bound_session_id,
2186                };
2187            } else {
2188                facts.native_instance = SessionLookupFact::Miss {
2189                    key: native_instance_key.to_string(),
2190                };
2191            }
2192        } else {
2193            facts.native_instance = SessionLookupFact::Miss {
2194                key: native_instance_key.to_string(),
2195            };
2196        }
2197    }
2198
2199    if probe.attach_hints.root_actor
2200        && let Some(current) = sessions.get_current_session()?
2201        && current.is_active()
2202    {
2203        let claimed = session_claimed_by_other(
2204            registry,
2205            &current.id,
2206            requested_entry,
2207            client_instance_id,
2208            probe.native_actor_key.as_deref(),
2209        )?;
2210        sessions_by_id
2211            .entry(current.id.clone())
2212            .or_insert_with(|| current.clone());
2213        facts.current_worktree = if claimed {
2214            WorktreeSessionFact::Claimed {
2215                session_id: current.id.clone(),
2216            }
2217        } else {
2218            WorktreeSessionFact::Available {
2219                session_id: current.id.clone(),
2220            }
2221        };
2222    }
2223
2224    if let Some(claims) = token_claims
2225        && let Some(token_sid) = claims.sid.as_deref()
2226        && let Some(session) = sessions.get_session(token_sid)?
2227        && session.is_active()
2228    {
2229        let claimed = session_claimed_by_other(
2230            registry,
2231            &session.id,
2232            requested_entry,
2233            client_instance_id,
2234            probe.native_actor_key.as_deref(),
2235        )?;
2236        let session_id = session.id.clone();
2237        sessions_by_id.insert(session_id.clone(), session);
2238        facts.token_sid = if claimed {
2239            TokenSidFact::Claimed { session_id }
2240        } else {
2241            TokenSidFact::Available { session_id }
2242        };
2243    }
2244
2245    let decision = decide_session_attach(&facts);
2246    match decision.policy {
2247        SessionPolicy::AttachExisting { session_id, .. } => {
2248            let session = sessions_by_id
2249                .remove(&session_id)
2250                .or_else(|| sessions.get_session(&session_id).ok().flatten())
2251                .ok_or_else(|| anyhow!("session not found: {session_id}"))?;
2252            Ok(ResolvedAttachment {
2253                target: AttachTarget::ExistingSession(session),
2254                matched_entry: matched_by_session.remove(&session_id),
2255                attach_reason: decision.attach_reason,
2256                precedence: decision.precedence,
2257                winning_rule: decision.winning_rule.to_string(),
2258            })
2259        }
2260        SessionPolicy::CreateNew {
2261            because_claimed, ..
2262        } => Ok(ResolvedAttachment {
2263            target: AttachTarget::CreateNew {
2264                _because_claimed: because_claimed,
2265            },
2266            matched_entry: None,
2267            attach_reason: decision.attach_reason,
2268            precedence: decision.precedence,
2269            winning_rule: decision.winning_rule.to_string(),
2270        }),
2271    }
2272}
2273
2274fn claude_actor_compatible(
2275    entry: &ActorPresence,
2276    probe: &HarnessProbeResult,
2277    repo_root: &Path,
2278) -> bool {
2279    let Some(native_actor_key) = probe.native_actor_key.as_deref() else {
2280        return true;
2281    };
2282    if !native_actor_key.starts_with("claude-code:") {
2283        return true;
2284    }
2285    if native_actor_key.starts_with("claude-code:agent:") {
2286        return entry.native_actor_key.as_deref() == Some(native_actor_key);
2287    }
2288    if let Some(native_instance_key) = probe.native_instance_key.as_deref() {
2289        return entry.native_actor_key.as_deref() == Some(native_actor_key)
2290            && entry.native_instance_key.as_deref() == Some(native_instance_key);
2291    }
2292    let same_repo = entry
2293        .path
2294        .as_ref()
2295        .map(|path| path.canonicalize().unwrap_or_else(|_| path.clone()))
2296        .unwrap_or_default()
2297        == repo_root
2298            .canonicalize()
2299            .unwrap_or_else(|_| repo_root.to_path_buf());
2300    entry.native_actor_key.as_deref() == Some(native_actor_key)
2301        && same_repo
2302        && probe.confidence.unwrap_or_default() >= 0.9
2303}
2304
2305fn decode_token_claims(token: &str) -> Option<TokenClaims> {
2306    let payload = token.split('.').nth(1)?;
2307    let decoded = base64::engine::general_purpose::URL_SAFE_NO_PAD
2308        .decode(payload.as_bytes())
2309        .ok()?;
2310    serde_json::from_slice(&decoded).ok()
2311}
2312
2313/// Decode the claims of the locally active bearer token, if any, resolved
2314/// through the single credential precedence (`HEDDLE_CREDENTIAL` → keystore).
2315/// Best-effort session-identity inference; returns `None` when unauthenticated
2316/// or built without the `client` feature.
2317fn active_token_claims() -> Option<TokenClaims> {
2318    #[cfg(feature = "client")]
2319    {
2320        crate::client::resolve_active_bearer()
2321            .ok()
2322            .flatten()
2323            .and_then(|token| decode_token_claims(&token.id))
2324    }
2325    #[cfg(not(feature = "client"))]
2326    {
2327        None
2328    }
2329}
2330
2331#[derive(Debug, Deserialize)]
2332struct TokenClaims {
2333    #[serde(default)]
2334    sid: Option<String>,
2335    #[serde(default)]
2336    agent_provider: Option<String>,
2337    #[serde(default)]
2338    agent_model: Option<String>,
2339}
2340
2341fn should_rotate_segment(session: &objects::object::Session, identity: &ResolvedIdentity) -> bool {
2342    let Some(segment) = session.current_segment() else {
2343        return false;
2344    };
2345    pure_should_rotate_segment(
2346        Some(segment.provider.as_str()),
2347        Some(segment.model.as_str()),
2348        identity.provider.as_deref(),
2349        identity.model.as_deref(),
2350    )
2351}
2352
2353fn thread_id_for_name(repo: &Repository, thread_name: Option<&str>) -> Result<Option<String>> {
2354    let Some(thread_name) = thread_name else {
2355        return Ok(None);
2356    };
2357    Ok(ThreadManager::new(repo.heddle_dir())
2358        .load(thread_name)?
2359        .map(|thread| thread.id))
2360}
2361
2362fn can_create_harness_thread(
2363    repo: &Repository,
2364    target_thread: Option<&str>,
2365    parent_thread: Option<&str>,
2366) -> Result<bool> {
2367    Ok(resolve_harness_thread_base_state(repo, target_thread, parent_thread)?.is_some())
2368}
2369
2370fn resolve_harness_thread_base_state(
2371    repo: &Repository,
2372    target_thread: Option<&str>,
2373    parent_thread: Option<&str>,
2374) -> Result<Option<objects::object::StateId>> {
2375    if let Some(head_state) = repo.head()? {
2376        return Ok(Some(head_state));
2377    }
2378
2379    for thread_name in [parent_thread, target_thread].into_iter().flatten() {
2380        if let Some(state) = resolve_named_thread_base_state(repo, thread_name)? {
2381            return Ok(Some(state));
2382        }
2383    }
2384
2385    Ok(None)
2386}
2387
2388fn resolve_named_thread_base_state(
2389    repo: &Repository,
2390    thread_name: &str,
2391) -> Result<Option<objects::object::StateId>> {
2392    if let Some(thread) = ThreadManager::new(repo.heddle_dir()).load(thread_name)?
2393        && let Some(state_spec) = thread
2394            .current_state
2395            .as_deref()
2396            .or(Some(thread.base_state.as_str()))
2397        && let Some(state_id) = repo
2398            .resolve_state(state_spec)?
2399            .or_else(|| objects::object::StateId::parse(state_spec).ok())
2400    {
2401        return Ok(Some(state_id));
2402    }
2403
2404    Ok(repo.refs().get_thread(&ThreadName::new(thread_name))?)
2405}
2406
2407fn resolve_parent_thread_for_subagent(
2408    repo: &Repository,
2409    probe: &HarnessProbeResult,
2410    current_attached: Option<&str>,
2411) -> Result<Option<String>> {
2412    if let Some(parent_key) = probe.native_parent_actor_key.as_deref() {
2413        let registry = ActorPresenceStore::new(repo.heddle_dir());
2414        if let Some(entry) = registry.find_active_by_native_actor_key(parent_key)? {
2415            return Ok(Some(entry.thread));
2416        }
2417    }
2418    Ok(current_attached.map(ToString::to_string))
2419}
2420
2421fn preferred_thread_slug(
2422    params: &OpenSessionParams,
2423    probe: &HarnessProbeResult,
2424    identity: &ResolvedIdentity,
2425) -> String {
2426    params
2427        .task
2428        .clone()
2429        .or_else(|| params.summary.clone())
2430        .or_else(|| probe.native_actor_key.as_deref().map(native_key_slug))
2431        .or_else(|| probe.native_instance_key.as_deref().map(native_key_slug))
2432        .or_else(|| identity.harness.clone())
2433        .unwrap_or_else(|| "work".to_string())
2434}
2435
2436fn native_key_slug(value: &str) -> String {
2437    value
2438        .rsplit(':')
2439        .next()
2440        .map(ToString::to_string)
2441        .unwrap_or_else(|| value.to_string())
2442}
2443
2444fn allocate_thread_name(repo: &Repository, base: &str) -> Result<String> {
2445    if ThreadManager::new(repo.heddle_dir()).load(base)?.is_none()
2446        && repo.refs().get_thread(&ThreadName::new(base))?.is_none()
2447    {
2448        return Ok(base.to_string());
2449    }
2450    for idx in 2..1000 {
2451        let candidate = format!("{base}-{idx}");
2452        if ThreadManager::new(repo.heddle_dir())
2453            .load(&candidate)?
2454            .is_none()
2455            && repo
2456                .refs()
2457                .get_thread(&ThreadName::new(&candidate))?
2458                .is_none()
2459        {
2460            return Ok(candidate);
2461        }
2462    }
2463    Err(anyhow!(
2464        "could not allocate a unique thread name from '{base}'"
2465    ))
2466}
2467
2468fn default_private_thread_path(repo: &Repository, name: &str) -> PathBuf {
2469    // Route through the ONE canonical `thread_manifest::thread_dir`
2470    // derivation `heddle start` and the per-thread `manifest.toml` sidecar
2471    // use — NOT a harness-local re-sanitisation. Harness subagent/root-actor
2472    // names are commonly slash-namespaced (`parent/task`); a local
2473    // `sanitize_name` flattened `parent/task` and `parent-task` onto the same
2474    // `.heddle/threads/parent-task/<repo-name>`, colliding two distinct threads and
2475    // diverging from the manifest/checkout layout (heddle#572 r2).
2476    repo.managed_checkout_path(name)
2477}
2478
2479fn sanitize_name(name: &str) -> String {
2480    let mut out = String::new();
2481    let mut last_dash = false;
2482    for ch in name.chars() {
2483        if ch.is_ascii_alphanumeric() {
2484            out.push(ch.to_ascii_lowercase());
2485            last_dash = false;
2486        } else if !last_dash {
2487            out.push('-');
2488            last_dash = true;
2489        }
2490    }
2491    out.trim_matches('-').to_string()
2492}
2493
2494fn resolve_requested_registry_entry(
2495    registry: &ActorPresenceStore,
2496    agent_session_id: Option<&str>,
2497    client_instance_id: Option<&str>,
2498) -> Result<Option<ActorPresence>> {
2499    if let Some(agent_session_id) = agent_session_id {
2500        let entry = registry
2501            .load(agent_session_id)?
2502            .ok_or_else(|| anyhow!("agent session not found: {agent_session_id}"))?;
2503        if entry.status != ActorPresenceStatus::Active {
2504            return Err(anyhow!("agent session is not active: {agent_session_id}"));
2505        }
2506        return Ok(Some(entry));
2507    }
2508
2509    if let Some(client_instance_id) = client_instance_id {
2510        return Ok(registry.find_active_by_client_instance_id(client_instance_id)?);
2511    }
2512
2513    Ok(None)
2514}
2515
2516fn ensure_requested_entry_matches_session(
2517    requested_entry: Option<&ActorPresence>,
2518    heddle_session_id: &str,
2519) -> Result<()> {
2520    if let Some(entry) = requested_entry
2521        && let Some(bound_session_id) = entry.heddle_session_id.as_deref()
2522        && bound_session_id != heddle_session_id
2523    {
2524        return Err(anyhow!(
2525            "requested agent is already bound to a different heddle session: {}",
2526            entry.session_id
2527        ));
2528    }
2529    Ok(())
2530}
2531
2532fn session_claimed_by_other(
2533    registry: &ActorPresenceStore,
2534    heddle_session_id: &str,
2535    requested_entry: Option<&ActorPresence>,
2536    client_instance_id: Option<&str>,
2537    native_actor_key: Option<&str>,
2538) -> Result<bool> {
2539    if requested_entry.is_none() && client_instance_id.is_none() && native_actor_key.is_none() {
2540        return Ok(false);
2541    }
2542
2543    let Some(existing) = registry.find_active_by_heddle_session_id(heddle_session_id)? else {
2544        return Ok(false);
2545    };
2546    if let Some(requested) = requested_entry {
2547        return Ok(requested.session_id != existing.session_id);
2548    }
2549    if let Some(client_instance_id) = client_instance_id
2550        && existing.client_instance_id.as_deref() == Some(client_instance_id)
2551    {
2552        return Ok(false);
2553    }
2554    if let Some(native_actor_key) = native_actor_key
2555        && existing.native_actor_key.as_deref() == Some(native_actor_key)
2556    {
2557        return Ok(false);
2558    }
2559    Ok(true)
2560}
2561
2562fn find_matching_registry_entry(
2563    registry: &ActorPresenceStore,
2564    repo: &Repository,
2565    heddle_session_id: &str,
2566    thread_name: Option<&str>,
2567) -> Result<Option<ActorPresence>> {
2568    if let Some(entry) = registry.find_active_by_heddle_session_id(heddle_session_id)? {
2569        return Ok(Some(entry));
2570    }
2571    let canonical_root = repo
2572        .root()
2573        .canonicalize()
2574        .unwrap_or_else(|_| repo.root().to_path_buf());
2575    Ok(registry
2576        .list()?
2577        .into_iter()
2578        .filter(|entry| entry.status == ActorPresenceStatus::Active)
2579        .find(|entry| {
2580            entry
2581                .path
2582                .as_ref()
2583                .map(|path| path.canonicalize().unwrap_or_else(|_| path.clone()) == canonical_root)
2584                .unwrap_or(false)
2585                || thread_name.is_some_and(|thread| entry.thread == thread)
2586        }))
2587}
2588
2589fn merged_env_hints(extra: &BTreeMap<String, String>) -> BTreeMap<String, String> {
2590    let mut merged: BTreeMap<String, String> = std::env::vars()
2591        .filter(|(key, _)| inherited_harness_hint(key))
2592        .collect();
2593    for (key, value) in extra {
2594        merged.insert(key.clone(), value.clone());
2595    }
2596    merged
2597}
2598
2599fn inherited_harness_hint(key: &str) -> bool {
2600    if matches!(
2601        key,
2602        "OPENAI_MODEL"
2603            | "ANTHROPIC_MODEL"
2604            | "CLAUDE_MODEL"
2605            | "MODEL"
2606            | "OPENAI_REASONING_EFFORT"
2607            | "REASONING_EFFORT"
2608            | "THINKING_LEVEL"
2609            | "PROMPT_POLICY"
2610    ) {
2611        return false;
2612    }
2613
2614    key.starts_with("HEDDLE_")
2615        || key.starts_with("CODEX_")
2616        || key == "CLAUDECODE"
2617        || key.starts_with("OPENCODE_")
2618}
2619
2620fn to_json_value<T: Serialize>(value: T) -> Result<Value> {
2621    serde_json::to_value(value).map_err(|err| anyhow!(err))
2622}
2623
2624fn normalize_paths<I>(paths: I) -> Vec<String>
2625where
2626    I: IntoIterator<Item = String>,
2627{
2628    let mut ordered = BTreeSet::new();
2629    for path in paths {
2630        let normalized = path.trim().replace('\\', "/");
2631        if !normalized.is_empty() {
2632            ordered.insert(normalized);
2633        }
2634    }
2635    ordered.into_iter().collect()
2636}
2637
2638fn merge_unique_paths<I>(target: &mut Vec<String>, paths: I)
2639where
2640    I: IntoIterator<Item = String>,
2641{
2642    let mut merged: BTreeSet<String> = target.iter().cloned().collect();
2643    merged.extend(paths);
2644    *target = merged.into_iter().collect();
2645}
2646
2647fn max_u64(current: Option<u64>, candidate: u64) -> u64 {
2648    current
2649        .map(|value| value.max(candidate))
2650        .unwrap_or(candidate)
2651}
2652
2653fn max_u32(current: Option<u32>, candidate: u32) -> u32 {
2654    current
2655        .map(|value| value.max(candidate))
2656        .unwrap_or(candidate)
2657}
2658
2659fn merge_usage(target: &mut UsageTotals, incoming: &UsageTotals) {
2660    if let Some(input) = incoming.input_tokens {
2661        target.input_tokens = Some(max_u64(target.input_tokens, input));
2662    }
2663    if let Some(output) = incoming.output_tokens {
2664        target.output_tokens = Some(max_u64(target.output_tokens, output));
2665    }
2666    if let Some(reasoning) = incoming.reasoning_tokens {
2667        target.reasoning_tokens = Some(max_u64(target.reasoning_tokens, reasoning));
2668    }
2669    if let Some(cache_creation) = incoming.cache_creation_tokens {
2670        target.cache_creation_tokens = Some(max_u64(target.cache_creation_tokens, cache_creation));
2671    }
2672    if let Some(cache_read) = incoming.cache_read_tokens {
2673        target.cache_read_tokens = Some(max_u64(target.cache_read_tokens, cache_read));
2674    }
2675    if let Some(tool_calls) = incoming.tool_calls {
2676        target.tool_calls = Some(max_u32(target.tool_calls, tool_calls));
2677    }
2678    if let Some(cost) = incoming.cost_micros_usd {
2679        target.cost_micros_usd = Some(max_u64(target.cost_micros_usd, cost));
2680    }
2681}
2682
2683fn parse_timestamp(value: &str) -> Option<chrono::DateTime<Utc>> {
2684    chrono::DateTime::parse_from_rfc3339(value)
2685        .ok()
2686        .map(|dt| dt.with_timezone(&Utc))
2687}
2688
2689fn transport_from_report(
2690    report: &SessionReportEnvelope,
2691    fallback: HarnessTransport,
2692) -> HarnessTransport {
2693    match report.transport_mode.as_str() {
2694        "spool" => HarnessTransport::Spool,
2695        "direct" => HarnessTransport::Direct,
2696        "end" => HarnessTransport::End,
2697        _ => fallback,
2698    }
2699}
2700
2701fn mark_pending_flush(report: &mut SessionReportEnvelope) {
2702    report.pending_flush = true;
2703    report.report_flush_state = Some("pending-local".to_string());
2704}
2705
2706fn enqueue_report(store: &SessionReportStore, report: &mut SessionReportEnvelope) -> Result<()> {
2707    store.append_outbox(report)?;
2708    report.pending_flush = false;
2709    let flushed_at = Utc::now().to_rfc3339();
2710    report.last_flushed_at = Some(flushed_at);
2711    report.report_flush_state = Some("queued-local".to_string());
2712    store.save(report)?;
2713    Ok(())
2714}
2715
2716fn usage_to_summary(usage: &UsageTotals) -> AgentUsageSummary {
2717    AgentUsageSummary {
2718        input_tokens: usage.input_tokens,
2719        output_tokens: usage.output_tokens,
2720        reasoning_tokens: usage.reasoning_tokens,
2721        tool_calls: usage.tool_calls,
2722        cost_micros_usd: usage.cost_micros_usd,
2723    }
2724}
2725
2726fn transcript_mode_name(mode: HarnessTranscriptMode) -> &'static str {
2727    match mode {
2728        HarnessTranscriptMode::Off => "off",
2729        HarnessTranscriptMode::Summary => "summary",
2730        HarnessTranscriptMode::Full => "full",
2731    }
2732}
2733
2734fn transport_mode_name(mode: HarnessTransport) -> &'static str {
2735    match mode {
2736        HarnessTransport::Spool => "spool",
2737        HarnessTransport::Direct => "direct",
2738        HarnessTransport::End => "end",
2739    }
2740}
2741
2742struct FinalDiff {
2743    changed_paths: Vec<String>,
2744    diff_summary: SessionDiffSummary,
2745    head_state: Option<String>,
2746}
2747
2748fn compute_final_diff(
2749    repo: &Repository,
2750    base_state: Option<&str>,
2751    worktree_baseline: &[WorktreeChangeBaseline],
2752) -> Result<FinalDiff> {
2753    let mut changes: BTreeMap<String, DiffKind> = BTreeMap::new();
2754
2755    let head_state = repo.head()?;
2756    if let (Some(base_spec), Some(head_id)) = (base_state, head_state) {
2757        let base_id = repo
2758            .resolve_state(base_spec)?
2759            .or_else(|| objects::object::StateId::parse(base_spec).ok());
2760        if let Some(base_id) = base_id
2761            && base_id != head_id
2762        {
2763            let Some(base_state_obj) = repo.store().get_state(&base_id)? else {
2764                return Err(anyhow!("base state not found: {base_spec}"));
2765            };
2766            let Some(head_state_obj) = repo.store().get_state(&head_id)? else {
2767                return Err(anyhow!("head state not found: {}", head_id.short()));
2768            };
2769            for change in repo.diff_trees(&base_state_obj.tree, &head_state_obj.tree)? {
2770                changes.insert(change.path, change.kind);
2771            }
2772        }
2773    }
2774
2775    let baseline_paths: BTreeSet<(String, String)> = worktree_baseline
2776        .iter()
2777        .map(|change| (change.path.clone(), change.kind.clone()))
2778        .collect();
2779    for (path, kind) in collect_worktree_changes(repo)? {
2780        let kind_name = diff_kind_name(kind);
2781        if !baseline_paths.contains(&(path.clone(), kind_name.to_string())) {
2782            changes.insert(path, kind);
2783        }
2784    }
2785
2786    let diff_summary = SessionDiffSummary {
2787        changed_file_count: changes.len() as u32,
2788        added_files: changes
2789            .values()
2790            .filter(|kind| **kind == DiffKind::Added)
2791            .count() as u32,
2792        modified_files: changes
2793            .values()
2794            .filter(|kind| **kind == DiffKind::Modified)
2795            .count() as u32,
2796        deleted_files: changes
2797            .values()
2798            .filter(|kind| **kind == DiffKind::Deleted)
2799            .count() as u32,
2800    };
2801
2802    Ok(FinalDiff {
2803        changed_paths: changes.into_keys().collect(),
2804        diff_summary,
2805        head_state: head_state.map(|id| id.to_string_full()),
2806    })
2807}
2808
2809fn capture_worktree_change_snapshot(repo: &Repository) -> Result<Vec<WorktreeChangeBaseline>> {
2810    Ok(collect_worktree_changes(repo)?
2811        .into_iter()
2812        .map(|(path, kind)| WorktreeChangeBaseline {
2813            path,
2814            kind: diff_kind_name(kind).to_string(),
2815        })
2816        .collect())
2817}
2818
2819fn collect_worktree_changes(repo: &Repository) -> Result<BTreeMap<String, DiffKind>> {
2820    let status_options = worktree_status_options(Some(repo.config()));
2821    let worktree_tree = match repo.current_state()? {
2822        Some(state) => repo.require_tree(&state.tree)?,
2823        None => Tree::new(),
2824    };
2825    let status = repo.compare_worktree_cached_with_options(&worktree_tree, &status_options)?;
2826    let mut changes = BTreeMap::new();
2827    for path in status.added {
2828        changes.insert(path.display().to_string(), DiffKind::Added);
2829    }
2830    for path in status.modified {
2831        changes.insert(path.display().to_string(), DiffKind::Modified);
2832    }
2833    for path in status.deleted {
2834        changes.insert(path.display().to_string(), DiffKind::Deleted);
2835    }
2836    Ok(changes)
2837}
2838
2839fn changed_paths_between_states(
2840    repo: &Repository,
2841    before_state: StateId,
2842    after_state: StateId,
2843) -> Result<Vec<String>> {
2844    if before_state == after_state {
2845        return Ok(Vec::new());
2846    }
2847    let Some(before_state_obj) = repo.store().get_state(&before_state)? else {
2848        return Err(anyhow!(
2849            "timeline before state not found: {}",
2850            before_state.short()
2851        ));
2852    };
2853    let Some(after_state_obj) = repo.store().get_state(&after_state)? else {
2854        return Err(anyhow!(
2855            "timeline after state not found: {}",
2856            after_state.short()
2857        ));
2858    };
2859    let mut paths = BTreeSet::new();
2860    for change in repo.diff_trees(&before_state_obj.tree, &after_state_obj.tree)? {
2861        paths.insert(change.path);
2862    }
2863    Ok(paths.into_iter().collect())
2864}
2865
2866fn diff_kind_name(kind: DiffKind) -> &'static str {
2867    match kind {
2868        DiffKind::Added => "added",
2869        DiffKind::Modified => "modified",
2870        DiffKind::Deleted => "deleted",
2871        DiffKind::Unchanged => "unchanged",
2872    }
2873}
2874
2875struct SessionReportStore {
2876    dir: PathBuf,
2877}
2878
2879impl SessionReportStore {
2880    fn new(repo_root: &Path) -> Self {
2881        Self {
2882            dir: repo_root.join(".heddle/state").join("session-reports"),
2883        }
2884    }
2885
2886    fn session_path(&self, heddle_session_id: &str) -> PathBuf {
2887        self.dir.join(format!("{heddle_session_id}.json"))
2888    }
2889
2890    fn outbox_path(&self) -> PathBuf {
2891        self.dir.join("outbox.jsonl")
2892    }
2893
2894    fn load(&self, heddle_session_id: &str) -> Result<Option<SessionReportEnvelope>> {
2895        let path = self.session_path(heddle_session_id);
2896        if !path.exists() {
2897            return Ok(None);
2898        }
2899        let bytes = fs::read(path)?;
2900        Ok(Some(serde_json::from_slice(&bytes)?))
2901    }
2902
2903    fn save(&self, report: &SessionReportEnvelope) -> Result<()> {
2904        fs::create_dir_all(&self.dir)?;
2905        let path = self.session_path(&report.heddle_session_id);
2906        let bytes = serde_json::to_vec_pretty(report)?;
2907        write_file_atomic(&path, &bytes)?;
2908        Ok(())
2909    }
2910
2911    fn append_outbox(&self, report: &SessionReportEnvelope) -> Result<()> {
2912        fs::create_dir_all(&self.dir)?;
2913        let mut file = OpenOptions::new()
2914            .create(true)
2915            .append(true)
2916            .open(self.outbox_path())?;
2917        serde_json::to_writer(&mut file, report)?;
2918        file.write_all(b"\n")?;
2919        file.flush()?;
2920        Ok(())
2921    }
2922
2923    fn list_pending(&self) -> Result<Vec<String>> {
2924        if !self.dir.exists() {
2925            return Ok(Vec::new());
2926        }
2927        let mut ids = Vec::new();
2928        for entry in fs::read_dir(&self.dir)? {
2929            let entry = entry?;
2930            let path = entry.path();
2931            if path.extension().and_then(|ext| ext.to_str()) != Some("json") {
2932                continue;
2933            }
2934            let bytes = fs::read(&path)?;
2935            let report: SessionReportEnvelope = serde_json::from_slice(&bytes)?;
2936            if report.pending_flush {
2937                ids.push(report.heddle_session_id);
2938            }
2939        }
2940        ids.sort();
2941        Ok(ids)
2942    }
2943}
2944
2945#[derive(Debug, Deserialize)]
2946struct BridgeRequest {
2947    #[serde(default)]
2948    id: Option<String>,
2949    method: String,
2950    #[serde(default)]
2951    params: Value,
2952}
2953
2954#[derive(Debug, Serialize)]
2955struct BridgeResponse {
2956    #[serde(default)]
2957    id: Option<String>,
2958    ok: bool,
2959    #[serde(skip_serializing_if = "Option::is_none")]
2960    result: Option<Value>,
2961    #[serde(skip_serializing_if = "Option::is_none")]
2962    error: Option<BridgeError>,
2963}
2964
2965impl BridgeResponse {
2966    fn ok(id: Option<String>, result: Value) -> Self {
2967        Self {
2968            id,
2969            ok: true,
2970            result: Some(result),
2971            error: None,
2972        }
2973    }
2974
2975    fn error(id: Option<String>, code: impl Into<String>, message: impl Into<String>) -> Self {
2976        Self {
2977            id,
2978            ok: false,
2979            result: None,
2980            error: Some(BridgeError {
2981                code: code.into(),
2982                message: message.into(),
2983            }),
2984        }
2985    }
2986}
2987
2988#[derive(Debug, Serialize)]
2989struct BridgeError {
2990    code: String,
2991    message: String,
2992}
2993
2994#[derive(Debug, Clone, Deserialize, Default)]
2995struct OpenSessionParams {
2996    #[serde(default)]
2997    heddle_session_id: Option<String>,
2998    #[serde(default)]
2999    agent_session_id: Option<String>,
3000    #[serde(default)]
3001    client_instance_id: Option<String>,
3002    #[serde(default)]
3003    thread: Option<String>,
3004    #[serde(default)]
3005    task: Option<String>,
3006    #[serde(default)]
3007    summary: Option<String>,
3008    #[serde(default)]
3009    harness: Option<String>,
3010    #[serde(default)]
3011    provider: Option<String>,
3012    #[serde(default)]
3013    model: Option<String>,
3014    #[serde(default)]
3015    thinking_level: Option<String>,
3016    #[serde(default)]
3017    policy: Option<String>,
3018    #[serde(default)]
3019    transport: Option<HarnessTransport>,
3020    #[serde(default)]
3021    transcript_mode: Option<HarnessTranscriptMode>,
3022    #[serde(default)]
3023    argv: Option<Vec<String>>,
3024    #[serde(default)]
3025    env_hints: BTreeMap<String, String>,
3026    #[serde(default)]
3027    probe_metadata: BTreeMap<String, String>,
3028}
3029
3030#[derive(Debug, Clone, Deserialize, Default)]
3031struct UpdateProgressParams {
3032    heddle_session_id: String,
3033    #[serde(default)]
3034    status: Option<String>,
3035    #[serde(default)]
3036    message: Option<String>,
3037    #[serde(default)]
3038    completed_steps: Option<u32>,
3039    #[serde(default)]
3040    total_steps: Option<u32>,
3041    #[serde(default)]
3042    touched_paths: Vec<String>,
3043    #[serde(default)]
3044    summary: Option<String>,
3045    #[serde(default)]
3046    harness: Option<String>,
3047    #[serde(default)]
3048    provider: Option<String>,
3049    #[serde(default)]
3050    model: Option<String>,
3051    #[serde(default)]
3052    thinking_level: Option<String>,
3053    #[serde(default)]
3054    policy: Option<String>,
3055    #[serde(default)]
3056    argv: Option<Vec<String>>,
3057    #[serde(default)]
3058    env_hints: BTreeMap<String, String>,
3059    #[serde(default)]
3060    probe_metadata: BTreeMap<String, String>,
3061}
3062
3063#[derive(Debug, Clone, Deserialize, Default)]
3064struct RecordUsageParams {
3065    heddle_session_id: String,
3066    #[serde(default)]
3067    input_tokens: Option<u64>,
3068    #[serde(default)]
3069    output_tokens: Option<u64>,
3070    #[serde(default)]
3071    reasoning_tokens: Option<u64>,
3072    #[serde(default)]
3073    cache_creation_tokens: Option<u64>,
3074    #[serde(default)]
3075    cache_read_tokens: Option<u64>,
3076    #[serde(default)]
3077    tool_calls: Option<u32>,
3078    #[serde(default)]
3079    cost_micros_usd: Option<u64>,
3080}
3081
3082#[derive(Debug, Clone, Deserialize, Default)]
3083struct RecordTouchedPathsParams {
3084    heddle_session_id: String,
3085    #[serde(default)]
3086    paths: Vec<String>,
3087}
3088
3089#[derive(Debug, Clone, Deserialize, Default)]
3090struct CloseSessionParams {
3091    heddle_session_id: String,
3092    #[serde(default)]
3093    outcome: Option<String>,
3094    #[serde(default)]
3095    summary: Option<String>,
3096    #[serde(default)]
3097    transcript_refs: Option<Vec<TranscriptAttachmentRef>>,
3098    #[serde(default)]
3099    transport: Option<HarnessTransport>,
3100}
3101
3102#[derive(Debug, Clone, Deserialize, Default)]
3103struct FlushReportsParams {
3104    #[serde(default)]
3105    heddle_session_id: Option<String>,
3106}
3107
3108#[derive(Debug, Serialize)]
3109struct OpenSessionResult {
3110    heddle_session_id: String,
3111    heddle_segment_id: Option<String>,
3112    agent_session_id: Option<String>,
3113    created_session: bool,
3114    harness: Option<String>,
3115    provider: Option<String>,
3116    model: Option<String>,
3117    thinking_level: Option<String>,
3118    report_flush_state: Option<String>,
3119    attach_reason: Option<String>,
3120}
3121
3122#[derive(Debug, Serialize)]
3123struct SessionMutationResult {
3124    heddle_session_id: String,
3125    heddle_segment_id: Option<String>,
3126    report_flush_state: Option<String>,
3127}
3128
3129#[derive(Debug, Serialize)]
3130struct CloseSessionResult {
3131    heddle_session_id: String,
3132    changed_paths: Vec<String>,
3133    diff_summary: SessionDiffSummary,
3134    report_flush_state: Option<String>,
3135}
3136
3137#[derive(Debug, Serialize)]
3138struct FlushReportsResult {
3139    flushed: usize,
3140}
3141
3142#[cfg(test)]
3143mod tests {
3144    #[cfg(unix)]
3145    use std::os::unix::fs::PermissionsExt;
3146
3147    use super::*;
3148
3149    fn init_repo() -> (tempfile::TempDir, Repository) {
3150        let temp = tempfile::TempDir::new().unwrap();
3151        let repo = Repository::init_default(temp.path()).unwrap();
3152        (temp, repo)
3153    }
3154
3155    #[test]
3156    fn harness_config_load_missing_path_defaults_without_warning() {
3157        let temp = tempfile::TempDir::new().unwrap();
3158        let missing = temp.path().join("missing-config.toml");
3159
3160        let (config, warning) = load_harness_user_config(Some(missing));
3161
3162        assert_eq!(config.harness.transport, HarnessTransport::Spool);
3163        assert!(warning.is_none());
3164    }
3165
3166    #[test]
3167    fn harness_config_load_malformed_path_warns_and_defaults() {
3168        let temp = tempfile::TempDir::new().unwrap();
3169        let path = temp.path().join("config.toml");
3170        std::fs::write(&path, "[harness\ntransport = \"direct\"\n").unwrap();
3171
3172        let (config, warning) = load_harness_user_config(Some(path.clone()));
3173
3174        assert_eq!(config.harness.transport, HarnessTransport::Spool);
3175        let warning = warning.expect("malformed config should produce a warning");
3176        assert!(warning.contains("failed to load user config"));
3177        assert!(warning.contains(&path.display().to_string()));
3178        assert!(warning.contains("continuing with defaults"));
3179    }
3180
3181    #[test]
3182    fn harness_config_load_valid_path_loads_without_warning() {
3183        let temp = tempfile::TempDir::new().unwrap();
3184        let path = temp.path().join("config.toml");
3185        std::fs::write(
3186            &path,
3187            "[harness]\ntransport = \"direct\"\ntranscript = \"summary\"\n",
3188        )
3189        .unwrap();
3190
3191        let (config, warning) = load_harness_user_config(Some(path));
3192
3193        assert_eq!(config.harness.transport, HarnessTransport::Direct);
3194        assert_eq!(config.harness.transcript, HarnessTranscriptMode::Summary);
3195        assert!(warning.is_none());
3196    }
3197
3198    #[test]
3199    fn relay_payload_parse_invalid_json_warns_and_uses_null() {
3200        let (value, warning) = parse_relay_payload("{not-json");
3201
3202        assert_eq!(value, Value::Null);
3203        let warning = warning.expect("invalid JSON should produce a warning");
3204        assert!(warning.contains("failed to parse harness relay payload as JSON"));
3205        assert!(warning.contains("continuing with null payload"));
3206    }
3207
3208    #[test]
3209    fn relay_payload_parse_empty_payload_uses_null_without_warning() {
3210        let (value, warning) = parse_relay_payload("  \n");
3211
3212        assert_eq!(value, Value::Null);
3213        assert!(warning.is_none());
3214    }
3215
3216    #[test]
3217    fn relay_payload_parse_valid_json_without_warning() {
3218        let (value, warning) = parse_relay_payload(r#"{"message":"hello"}"#);
3219
3220        assert_eq!(value["message"], "hello");
3221        assert!(warning.is_none());
3222    }
3223
3224    /// Harness subagent/root-actor checkout paths must use the SAME canonical
3225    /// managed checkout path derivation `start` and the per-thread manifest use
3226    /// — for the slash-namespaced names the harness commonly mints
3227    /// (`parent/task`). Before this, a harness-local `sanitize_name` flattened
3228    /// `parent/task` and `parent-task` onto the same
3229    /// `.heddle/threads/parent-task/<repo-name>`, colliding distinct threads
3230    /// (heddle#572 r2).
3231    #[test]
3232    fn harness_default_path_matches_canonical_thread_dir() {
3233        let (_temp, repo) = init_repo();
3234        for id in ["foo", "parent/task", "feature/foo", "team@scope"] {
3235            let harness_path = default_private_thread_path(&repo, id);
3236            let canonical = repo.managed_checkout_path(id);
3237            assert_eq!(
3238                harness_path, canonical,
3239                "harness default must match the canonical thread_dir for {id:?}"
3240            );
3241        }
3242    }
3243
3244    #[test]
3245    fn inherited_harness_hints_exclude_ambient_model_identity() {
3246        assert!(!inherited_harness_hint("OPENAI_MODEL"));
3247        assert!(!inherited_harness_hint("ANTHROPIC_MODEL"));
3248        assert!(!inherited_harness_hint("CLAUDE_MODEL"));
3249        assert!(!inherited_harness_hint("MODEL"));
3250        assert!(!inherited_harness_hint("OPENAI_REASONING_EFFORT"));
3251        assert!(inherited_harness_hint("HEDDLE_AGENT_MODEL"));
3252        assert!(inherited_harness_hint("CODEX_SANDBOX"));
3253        assert!(inherited_harness_hint("CLAUDECODE"));
3254    }
3255
3256    #[test]
3257    fn open_session_creates_or_attaches() {
3258        let (_temp, repo) = init_repo();
3259        let user_config = UserConfig::default();
3260        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3261
3262        let created = runtime
3263            .open_session(OpenSessionParams {
3264                harness: Some("codex".to_string()),
3265                provider: Some("openai".to_string()),
3266                model: Some("gpt-5.4".to_string()),
3267                ..OpenSessionParams::default()
3268            })
3269            .unwrap();
3270        assert!(created.created_session);
3271
3272        let attached = runtime
3273            .open_session(OpenSessionParams {
3274                harness: Some("codex".to_string()),
3275                provider: Some("openai".to_string()),
3276                model: Some("gpt-5.4".to_string()),
3277                ..OpenSessionParams::default()
3278            })
3279            .unwrap();
3280        assert!(!attached.created_session);
3281        assert_eq!(created.heddle_session_id, attached.heddle_session_id);
3282    }
3283
3284    #[test]
3285    fn same_client_instance_reattaches_to_its_existing_session() {
3286        let (_temp, repo) = init_repo();
3287        let user_config = UserConfig::default();
3288        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3289
3290        let first = runtime
3291            .open_session(OpenSessionParams {
3292                client_instance_id: Some("client-a".to_string()),
3293                harness: Some("codex".to_string()),
3294                provider: Some("openai".to_string()),
3295                model: Some("gpt-5.4".to_string()),
3296                ..OpenSessionParams::default()
3297            })
3298            .unwrap();
3299        let second = runtime
3300            .open_session(OpenSessionParams {
3301                client_instance_id: Some("client-b".to_string()),
3302                harness: Some("codex".to_string()),
3303                provider: Some("openai".to_string()),
3304                model: Some("gpt-5.4".to_string()),
3305                ..OpenSessionParams::default()
3306            })
3307            .unwrap();
3308        let reopened = runtime
3309            .open_session(OpenSessionParams {
3310                client_instance_id: Some("client-a".to_string()),
3311                harness: Some("codex".to_string()),
3312                provider: Some("openai".to_string()),
3313                model: Some("gpt-5.4".to_string()),
3314                ..OpenSessionParams::default()
3315            })
3316            .unwrap();
3317
3318        assert_ne!(first.heddle_session_id, second.heddle_session_id);
3319        assert_eq!(first.heddle_session_id, reopened.heddle_session_id);
3320        assert_eq!(first.agent_session_id, reopened.agent_session_id);
3321    }
3322
3323    #[test]
3324    fn different_client_instances_do_not_share_the_current_session() {
3325        let (_temp, repo) = init_repo();
3326        let user_config = UserConfig::default();
3327        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3328
3329        let first = runtime
3330            .open_session(OpenSessionParams {
3331                client_instance_id: Some("client-a".to_string()),
3332                harness: Some("codex".to_string()),
3333                provider: Some("openai".to_string()),
3334                model: Some("gpt-5.4".to_string()),
3335                ..OpenSessionParams::default()
3336            })
3337            .unwrap();
3338        let second = runtime
3339            .open_session(OpenSessionParams {
3340                client_instance_id: Some("client-b".to_string()),
3341                harness: Some("codex".to_string()),
3342                provider: Some("openai".to_string()),
3343                model: Some("gpt-5.4".to_string()),
3344                ..OpenSessionParams::default()
3345            })
3346            .unwrap();
3347
3348        assert_ne!(first.heddle_session_id, second.heddle_session_id);
3349        assert_ne!(first.agent_session_id, second.agent_session_id);
3350    }
3351
3352    #[test]
3353    fn provider_model_change_creates_segment() {
3354        let (_temp, repo) = init_repo();
3355        let user_config = UserConfig::default();
3356        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3357
3358        let opened = runtime
3359            .open_session(OpenSessionParams {
3360                harness: Some("claude-code".to_string()),
3361                provider: Some("anthropic".to_string()),
3362                model: Some("claude-sonnet".to_string()),
3363                ..OpenSessionParams::default()
3364            })
3365            .unwrap();
3366        runtime
3367            .update_progress(UpdateProgressParams {
3368                heddle_session_id: opened.heddle_session_id.clone(),
3369                provider: Some("openai".to_string()),
3370                model: Some("gpt-5.4".to_string()),
3371                ..UpdateProgressParams::default()
3372            })
3373            .unwrap();
3374
3375        let report = runtime
3376            .reports
3377            .load(&opened.heddle_session_id)
3378            .unwrap()
3379            .unwrap();
3380        let expected_segment = format!("{}-seg-2", opened.heddle_session_id);
3381        assert_eq!(
3382            report.heddle_segment_id.as_deref(),
3383            Some(expected_segment.as_str())
3384        );
3385    }
3386
3387    #[test]
3388    fn blank_agent_model_hint_falls_through_to_detected_model_without_segment_rotation() {
3389        let (_temp, repo) = init_repo();
3390        let user_config = UserConfig::default();
3391        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3392        let blank_model_env = BTreeMap::from([
3393            ("HEDDLE_AGENT_PROVIDER".to_string(), "anthropic".to_string()),
3394            ("HEDDLE_AGENT_MODEL".to_string(), String::new()),
3395        ]);
3396
3397        let opened = runtime
3398            .open_session(OpenSessionParams {
3399                harness: Some("claude-code".to_string()),
3400                env_hints: blank_model_env.clone(),
3401                probe_metadata: BTreeMap::from([
3402                    ("session_id".to_string(), "claude-sess-blank".to_string()),
3403                    ("model".to_string(), "claude-opus-4-8[1m]".to_string()),
3404                ]),
3405                ..OpenSessionParams::default()
3406            })
3407            .unwrap();
3408        assert_eq!(opened.model.as_deref(), Some("claude-opus-4-8[1m]"));
3409
3410        let original_segment = opened.heddle_segment_id.clone();
3411        runtime
3412            .update_progress(UpdateProgressParams {
3413                heddle_session_id: opened.heddle_session_id.clone(),
3414                env_hints: blank_model_env,
3415                probe_metadata: BTreeMap::from([
3416                    ("session_id".to_string(), "claude-sess-blank".to_string()),
3417                    ("model".to_string(), "claude-opus-4-8[1m]".to_string()),
3418                ]),
3419                ..UpdateProgressParams::default()
3420            })
3421            .unwrap();
3422
3423        let report = runtime
3424            .reports
3425            .load(&opened.heddle_session_id)
3426            .unwrap()
3427            .unwrap();
3428        assert_eq!(report.harness.model.as_deref(), Some("claude-opus-4-8[1m]"));
3429        assert_eq!(report.heddle_segment_id, original_segment);
3430    }
3431
3432    #[test]
3433    fn close_session_captures_changed_paths_from_status_and_hints() {
3434        let (temp, repo) = init_repo();
3435        let config = UserConfig::default();
3436        let mut runtime = HarnessBridgeRuntime::new(repo, config);
3437
3438        let opened = runtime
3439            .open_session(OpenSessionParams {
3440                harness: Some("codex".to_string()),
3441                provider: Some("openai".to_string()),
3442                model: Some("gpt-5.4".to_string()),
3443                ..OpenSessionParams::default()
3444            })
3445            .unwrap();
3446        std::fs::write(temp.path().join("src.txt"), "hello\n").unwrap();
3447        runtime
3448            .record_touched_paths(RecordTouchedPathsParams {
3449                heddle_session_id: opened.heddle_session_id.clone(),
3450                paths: vec!["src.txt".to_string(), "notes.md".to_string()],
3451            })
3452            .unwrap();
3453        let closed = runtime
3454            .close_session(CloseSessionParams {
3455                heddle_session_id: opened.heddle_session_id.clone(),
3456                outcome: Some("completed".to_string()),
3457                ..CloseSessionParams::default()
3458            })
3459            .unwrap();
3460        let report = runtime
3461            .reports
3462            .load(&opened.heddle_session_id)
3463            .unwrap()
3464            .unwrap();
3465        assert!(closed.changed_paths.iter().any(|path| path == "src.txt"));
3466        assert!(!closed.changed_paths.iter().any(|path| path == "notes.md"));
3467        assert!(report.touched_paths.iter().any(|path| path == "src.txt"));
3468        assert!(report.touched_paths.iter().any(|path| path == "notes.md"));
3469        assert_eq!(
3470            closed.diff_summary.changed_file_count,
3471            closed.changed_paths.len() as u32
3472        );
3473    }
3474
3475    #[test]
3476    fn flush_reports_moves_pending_report_to_outbox() {
3477        let (_temp, repo) = init_repo();
3478        let user_config = UserConfig::default();
3479        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3480
3481        let opened = runtime
3482            .open_session(OpenSessionParams {
3483                harness: Some("codex".to_string()),
3484                provider: Some("openai".to_string()),
3485                model: Some("gpt-5.4".to_string()),
3486                ..OpenSessionParams::default()
3487            })
3488            .unwrap();
3489        let flushed = runtime
3490            .flush_reports(FlushReportsParams {
3491                heddle_session_id: Some(opened.heddle_session_id.clone()),
3492            })
3493            .unwrap();
3494        assert_eq!(flushed.flushed, 1);
3495        let report = runtime
3496            .reports
3497            .load(&opened.heddle_session_id)
3498            .unwrap()
3499            .unwrap();
3500        assert!(!report.pending_flush);
3501        assert_eq!(report.report_flush_state.as_deref(), Some("queued-local"));
3502        assert!(runtime.reports.outbox_path().exists());
3503    }
3504
3505    #[test]
3506    fn explicit_overrides_beat_fingerprint_and_user_defaults() {
3507        let (_temp, repo) = init_repo();
3508        let mut user_config = UserConfig::default();
3509        user_config.harness.harnesses.insert(
3510            "codex".to_string(),
3511            UserHarnessOverride {
3512                provider: Some("openai".to_string()),
3513                model: Some("gpt-default".to_string()),
3514                thinking_level: Some("medium".to_string()),
3515                policy: Some("default".to_string()),
3516            },
3517        );
3518        let identity = resolve_identity(
3519            &repo,
3520            &user_config,
3521            IdentityHints {
3522                harness: Some("codex".to_string()),
3523                provider: Some("openai".to_string()),
3524                model: Some("gpt-5.4".to_string()),
3525                thinking_level: Some("high".to_string()),
3526                policy: Some("custom".to_string()),
3527                probe: HarnessProbeResult::default(),
3528            },
3529        )
3530        .unwrap();
3531        assert_eq!(identity.model.as_deref(), Some("gpt-5.4"));
3532        assert_eq!(identity.thinking_level.as_deref(), Some("high"));
3533        assert_eq!(identity.policy.as_deref(), Some("custom"));
3534    }
3535
3536    #[test]
3537    fn transcript_mode_defaults_to_off_and_keeps_refs_empty() {
3538        let (_temp, repo) = init_repo();
3539        let user_config = UserConfig::default();
3540        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3541
3542        let opened = runtime
3543            .open_session(OpenSessionParams {
3544                harness: Some("codex".to_string()),
3545                provider: Some("openai".to_string()),
3546                model: Some("gpt-5.4".to_string()),
3547                ..OpenSessionParams::default()
3548            })
3549            .unwrap();
3550        let report = runtime
3551            .reports
3552            .load(&opened.heddle_session_id)
3553            .unwrap()
3554            .unwrap();
3555        assert_eq!(report.transcript_mode, "off");
3556        assert!(report.transcript_refs.is_empty());
3557    }
3558
3559    #[test]
3560    fn codex_thread_probe_reattaches_same_actor() {
3561        let (_temp, repo) = init_repo();
3562        let user_config = UserConfig::default();
3563        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3564
3565        let first = runtime
3566            .open_session(OpenSessionParams {
3567                harness: Some("codex".to_string()),
3568                probe_metadata: BTreeMap::from([
3569                    ("thread_id".to_string(), "thr_123".to_string()),
3570                    ("client_name".to_string(), "codex-tui".to_string()),
3571                ]),
3572                ..OpenSessionParams::default()
3573            })
3574            .unwrap();
3575        let second = runtime
3576            .open_session(OpenSessionParams {
3577                harness: Some("codex".to_string()),
3578                probe_metadata: BTreeMap::from([
3579                    ("thread_id".to_string(), "thr_123".to_string()),
3580                    ("client_name".to_string(), "codex-tui".to_string()),
3581                ]),
3582                ..OpenSessionParams::default()
3583            })
3584            .unwrap();
3585
3586        assert_eq!(first.agent_session_id, second.agent_session_id);
3587        assert_eq!(first.heddle_session_id, second.heddle_session_id);
3588    }
3589
3590    #[test]
3591    fn opencode_child_session_creates_distinct_actor_with_parent_key() {
3592        let (_temp, repo) = init_repo();
3593        let user_config = UserConfig::default();
3594        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3595
3596        let root = runtime
3597            .open_session(OpenSessionParams {
3598                harness: Some("opencode".to_string()),
3599                probe_metadata: BTreeMap::from([("session_id".to_string(), "root-1".to_string())]),
3600                ..OpenSessionParams::default()
3601            })
3602            .unwrap();
3603        let child = runtime
3604            .open_session(OpenSessionParams {
3605                harness: Some("opencode".to_string()),
3606                probe_metadata: BTreeMap::from([
3607                    ("session_id".to_string(), "child-1".to_string()),
3608                    ("parent_id".to_string(), "root-1".to_string()),
3609                ]),
3610                ..OpenSessionParams::default()
3611            })
3612            .unwrap();
3613
3614        assert_ne!(root.agent_session_id, child.agent_session_id);
3615        let report = runtime
3616            .reports
3617            .load(&child.heddle_session_id)
3618            .unwrap()
3619            .unwrap();
3620        assert_eq!(
3621            report.native_parent_actor_key.as_deref(),
3622            Some("opencode:session:root-1")
3623        );
3624    }
3625
3626    #[test]
3627    fn claude_resume_with_new_session_id_does_not_steal_existing_actor() {
3628        let (_temp, repo) = init_repo();
3629        let user_config = UserConfig::default();
3630        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3631
3632        let first = runtime
3633            .open_session(OpenSessionParams {
3634                harness: Some("claude-code".to_string()),
3635                probe_metadata: BTreeMap::from([
3636                    ("session_id".to_string(), "sess-old".to_string()),
3637                    (
3638                        "transcript_path".to_string(),
3639                        "/tmp/claude/session-a.jsonl".to_string(),
3640                    ),
3641                ]),
3642                ..OpenSessionParams::default()
3643            })
3644            .unwrap();
3645        let resumed = runtime
3646            .open_session(OpenSessionParams {
3647                harness: Some("claude-code".to_string()),
3648                probe_metadata: BTreeMap::from([
3649                    ("session_id".to_string(), "sess-new".to_string()),
3650                    (
3651                        "transcript_path".to_string(),
3652                        "/tmp/claude/session-a.jsonl".to_string(),
3653                    ),
3654                ]),
3655                ..OpenSessionParams::default()
3656            })
3657            .unwrap();
3658
3659        assert_ne!(first.agent_session_id, resumed.agent_session_id);
3660        assert_ne!(first.heddle_session_id, resumed.heddle_session_id);
3661    }
3662
3663    #[test]
3664    fn explicit_claude_harness_beats_generic_session_id_probe_match() {
3665        let (_temp, repo) = init_repo();
3666        let user_config = UserConfig::default();
3667        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3668
3669        let opened = runtime
3670            .open_session(OpenSessionParams {
3671                harness: Some("claude-code".to_string()),
3672                probe_metadata: BTreeMap::from([
3673                    ("session_id".to_string(), "claude-sess-1".to_string()),
3674                    ("hook_event".to_string(), "SubagentStop".to_string()),
3675                ]),
3676                ..OpenSessionParams::default()
3677            })
3678            .unwrap();
3679        let report = runtime
3680            .reports
3681            .load(&opened.heddle_session_id)
3682            .unwrap()
3683            .unwrap();
3684        assert_eq!(
3685            report.native_actor_key.as_deref(),
3686            Some("claude-code:session:claude-sess-1")
3687        );
3688        assert_eq!(report.harness.harness.as_deref(), Some("claude-code"));
3689    }
3690
3691    #[test]
3692    fn same_native_actor_key_reuses_existing_actor_after_tentative_session_creation() {
3693        let (_temp, repo) = init_repo();
3694        let user_config = UserConfig::default();
3695        let runtime = HarnessBridgeRuntime::new(repo, user_config);
3696        let principal = runtime.repo.get_principal().unwrap();
3697        let mut sessions = SessionManager::new(runtime.repo.root());
3698        let existing_session = sessions
3699            .start_session(
3700                principal.clone(),
3701                "anthropic".to_string(),
3702                "claude-opus-4-7[1m]".to_string(),
3703                None,
3704            )
3705            .unwrap();
3706        let tentative_session = sessions
3707            .start_session(
3708                principal,
3709                "anthropic".to_string(),
3710                "claude-opus-4-7[1m]".to_string(),
3711                None,
3712            )
3713            .unwrap();
3714
3715        let registry = ActorPresenceStore::new(runtime.repo.heddle_dir());
3716        let existing_entry = registry
3717            .create_generated_entry(|session_id| {
3718                Ok(ActorPresence {
3719                    session_id: session_id.to_string(),
3720                    client_instance_id: None,
3721                    native_actor_key: Some(
3722                        "claude-code:session:282396d3-554a-48aa-a9a8-8d1f0bd15fa5".to_string(),
3723                    ),
3724                    native_parent_actor_key: None,
3725                    native_instance_key: Some(
3726                        "claude-code:transcript:/tmp/claude/282396d3.jsonl".to_string(),
3727                    ),
3728                    heddle_session_id: Some(existing_session.id.clone()),
3729                    thread_id: None,
3730                    thread: "detached".to_string(),
3731                    anchor_state: None,
3732                    anchor_root: None,
3733                    path: Some(runtime.repo.root().to_path_buf()),
3734                    base_state: String::new(),
3735                    started_at: Utc::now(),
3736                    provider: Some("anthropic".to_string()),
3737                    model: Some("claude-opus-4-7[1m]".to_string()),
3738                    harness: Some("claude-code".to_string()),
3739                    thinking_level: None,
3740                    usage_summary: AgentUsageSummary::default(),
3741                    last_progress_at: None,
3742                    report_flush_state: Some("pending-local".to_string()),
3743                    attach_reason: None,
3744                    task_assignment_id: None,
3745                    attach_precedence: vec![],
3746                    winning_attach_rule: None,
3747                    probe_source: Some("hook_payload".to_string()),
3748                    probe_confidence: Some(1.0),
3749                    status: ActorPresenceStatus::Active,
3750                    completed_at: None,
3751                    context_queries: vec![],
3752                })
3753            })
3754            .unwrap();
3755
3756        let probe = HarnessProbeResult {
3757            harness: Some("claude-code".to_string()),
3758            provider: Some("anthropic".to_string()),
3759            model: Some("claude-opus-4-7[1m]".to_string()),
3760            native_actor_key: Some(
3761                "claude-code:session:282396d3-554a-48aa-a9a8-8d1f0bd15fa5".to_string(),
3762            ),
3763            native_instance_key: Some(
3764                "claude-code:transcript:/tmp/claude/282396d3.jsonl".to_string(),
3765            ),
3766            probe_source: Some("hook_payload".to_string()),
3767            confidence: Some(1.0),
3768            ..HarnessProbeResult::default()
3769        };
3770        let identity = ResolvedIdentity {
3771            harness: Some("claude-code".to_string()),
3772            provider: Some("anthropic".to_string()),
3773            model: Some("claude-opus-4-7[1m]".to_string()),
3774            thinking_level: None,
3775            policy: None,
3776        };
3777        let mut attach = ResolvedAttachment {
3778            target: AttachTarget::CreateNew {
3779                _because_claimed: false,
3780            },
3781            matched_entry: None,
3782            attach_reason:
3783                "started new Heddle session because no compatible native actor match was found"
3784                    .to_string(),
3785            precedence: vec!["native-actor-key:miss".to_string()],
3786            winning_rule: "create-new-session".to_string(),
3787        };
3788
3789        let resolved_entry = runtime
3790            .ensure_registry_entry(RegistryEntryRequest {
3791                heddle_session_id: &tentative_session.id,
3792                thread_name: None,
3793                thread_id: None,
3794                identity: &identity,
3795                probe: &probe,
3796                attach: &attach,
3797                client_instance_id: None,
3798                requested_entry: None,
3799            })
3800            .unwrap();
3801        assert_eq!(resolved_entry.session_id, existing_entry.session_id);
3802        assert_eq!(
3803            resolved_entry.heddle_session_id.as_deref(),
3804            Some(existing_session.id.as_str())
3805        );
3806
3807        let (canonical_session, owns_session) = runtime
3808            .reuse_canonical_actor_session(
3809                &mut sessions,
3810                CanonicalActorSessionRequest {
3811                    tentative_session: tentative_session.clone(),
3812                    tentative_owns_session: true,
3813                    entry: &resolved_entry,
3814                    probe: &probe,
3815                    attach: &mut attach,
3816                },
3817            )
3818            .unwrap();
3819        assert_eq!(canonical_session.id, existing_session.id);
3820        assert!(!owns_session);
3821        assert!(
3822            attach
3823                .precedence
3824                .iter()
3825                .any(|step| step.starts_with("post-create-native-actor-key:"))
3826        );
3827        assert_eq!(attach.winning_rule, "native-actor-key-post-create");
3828        assert!(
3829            !sessions
3830                .get_session(&tentative_session.id)
3831                .unwrap()
3832                .unwrap()
3833                .is_active()
3834        );
3835    }
3836
3837    #[test]
3838    fn close_session_does_not_blame_preexisting_dirty_worktree() {
3839        let (temp, repo) = init_repo();
3840        std::fs::write(temp.path().join("preexisting.txt"), "already dirty\n").unwrap();
3841        let user_config = UserConfig::default();
3842        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
3843
3844        let opened = runtime
3845            .open_session(OpenSessionParams {
3846                harness: Some("claude-code".to_string()),
3847                provider: Some("anthropic".to_string()),
3848                model: Some("claude-opus-4-7[1m]".to_string()),
3849                ..OpenSessionParams::default()
3850            })
3851            .unwrap();
3852        let closed = runtime
3853            .close_session(CloseSessionParams {
3854                heddle_session_id: opened.heddle_session_id.clone(),
3855                outcome: Some("completed".to_string()),
3856                ..CloseSessionParams::default()
3857            })
3858            .unwrap();
3859        let report = runtime
3860            .reports
3861            .load(&opened.heddle_session_id)
3862            .unwrap()
3863            .unwrap();
3864
3865        assert!(
3866            report
3867                .worktree_changes_at_open
3868                .iter()
3869                .any(|change| change.path == "preexisting.txt")
3870        );
3871        assert!(
3872            !closed
3873                .changed_paths
3874                .iter()
3875                .any(|path| path == "preexisting.txt")
3876        );
3877        assert_eq!(closed.diff_summary.changed_file_count, 0);
3878    }
3879
3880    #[test]
3881    fn timeline_state_delta_paths_ignore_uncaptured_worktree_changes() {
3882        let (temp, repo) = init_repo();
3883        let repo_root = repo.root().to_path_buf();
3884        std::fs::write(repo_root.join("tracked.txt"), b"one\n").unwrap();
3885        let before = repo.snapshot(Some("seed".into()), None).unwrap();
3886        std::fs::write(repo_root.join("tracked.txt"), b"two\n").unwrap();
3887        let after = repo.snapshot(Some("advance".into()), None).unwrap();
3888        std::fs::write(temp.path().join("ambient.txt"), b"not in the state delta\n").unwrap();
3889
3890        assert_eq!(
3891            changed_paths_between_states(&repo, before.state_id, after.state_id).unwrap(),
3892            vec!["tracked.txt"]
3893        );
3894    }
3895
3896    #[test]
3897    fn relay_claude_stop_captures_state_with_agent_attribution() {
3898        let (temp, repo) = init_repo();
3899        let repo_root = repo.root().to_path_buf();
3900
3901        // Establish HEAD with an initial snapshot.
3902        std::fs::write(repo_root.join("seed.txt"), b"hello").unwrap();
3903        let _ = repo.snapshot(Some("seed".into()), None).unwrap();
3904
3905        // Make a dirty change that the Stop hook should capture.
3906        std::fs::write(repo_root.join("seed.txt"), b"hello, heddle").unwrap();
3907
3908        drop(repo);
3909
3910        let fresh_repo = Repository::open(temp.path()).unwrap();
3911        let user_config = UserConfig {
3912            principal: Some(crate::config::UserPrincipalConfig {
3913                name: "Ada Lovelace".to_string(),
3914                email: "ada@example.com".to_string(),
3915            }),
3916            ..UserConfig::default()
3917        };
3918        let mut runtime = HarnessBridgeRuntime::new(fresh_repo, user_config);
3919        let payload = serde_json::json!({
3920            "session_id": "claude-sess-123",
3921            "transcript_path": "/tmp/claude/x.jsonl",
3922            "model": {
3923                "id": "claude-opus-4-7",
3924                "display_name": "Claude Opus 4.7",
3925            },
3926            "message": "hook-driven capture test",
3927            "hook_event_name": "Stop",
3928        });
3929        relay_claude(&mut runtime, "Stop", &payload).unwrap();
3930        drop(runtime);
3931
3932        let verify = Repository::open(temp.path()).unwrap();
3933        let head_id = verify.head().unwrap().expect("HEAD after Stop capture");
3934        let state = verify
3935            .store()
3936            .get_state(&head_id)
3937            .unwrap()
3938            .expect("state for HEAD");
3939        let agent = state.attribution.agent.expect("agent attribution on state");
3940        assert_eq!(agent.provider, "anthropic");
3941        assert_eq!(agent.model, "Claude Opus 4.7");
3942        assert_eq!(
3943            state.intent.as_deref(),
3944            Some("hook-driven capture test"),
3945            "intent should be pulled from payload message",
3946        );
3947    }
3948
3949    #[test]
3950    fn relay_claude_stop_is_idempotent_when_clean() {
3951        let (temp, repo) = init_repo();
3952        let repo_root = repo.root().to_path_buf();
3953        std::fs::write(repo_root.join("seed.txt"), b"hello").unwrap();
3954        let seed = repo.snapshot(Some("seed".into()), None).unwrap();
3955        drop(repo);
3956
3957        let fresh_repo = Repository::open(temp.path()).unwrap();
3958        let mut runtime = HarnessBridgeRuntime::new(fresh_repo, UserConfig::default());
3959        let payload = serde_json::json!({
3960            "session_id": "claude-sess-clean",
3961            "model": {"id": "claude-sonnet-4-6"},
3962        });
3963        relay_claude(&mut runtime, "Stop", &payload).unwrap();
3964        drop(runtime);
3965
3966        let verify = Repository::open(temp.path()).unwrap();
3967        let head_id = verify.head().unwrap().expect("HEAD preserved");
3968        assert_eq!(
3969            head_id, seed.state_id,
3970            "no change expected when worktree is clean",
3971        );
3972    }
3973
3974    #[test]
3975    fn relay_claude_pre_tool_use_ignores_non_file_tool() {
3976        let (temp, repo) = init_repo();
3977        drop(repo);
3978        let fresh_repo = Repository::open(temp.path()).unwrap();
3979        let mut runtime = HarnessBridgeRuntime::new(fresh_repo, UserConfig::default());
3980        let payload = serde_json::json!({
3981            "session_id": "claude-sess-bash",
3982            "tool_name": "Bash",
3983            "tool_input": {"command": "ls"},
3984        });
3985        // Should succeed without writing any stdout or erroring.
3986        relay_claude(&mut runtime, "PreToolUse", &payload).unwrap();
3987    }
3988
3989    #[test]
3990    fn relay_opencode_tool_execute_before_records_timeline_step() {
3991        let (_temp, repo) = init_repo();
3992        let root = repo.root().to_path_buf();
3993        std::fs::write(root.join("seed.txt"), b"hello").unwrap();
3994        let seed = repo.snapshot(Some("seed".into()), None).unwrap();
3995        let mut runtime = HarnessBridgeRuntime::new(repo, UserConfig::default());
3996        let payload = opencode_tool_payload("call-1");
3997
3998        relay_opencode(&mut runtime, "tool.execute.before", &payload).unwrap();
3999
4000        let store = TimelineStore::open(runtime.repo.heddle_dir()).unwrap();
4001        let view = TimelineView::rebuild(&store).unwrap();
4002        let steps = view.steps_for_thread("main");
4003        assert_eq!(steps.len(), 1);
4004        let step = steps[0];
4005        assert_eq!(step.native.as_ref().unwrap().harness, "opencode");
4006        assert_eq!(step.native.as_ref().unwrap().tool_call_id, "call-1");
4007        assert_eq!(step.tool_name.as_deref(), Some("bash"));
4008        assert_eq!(step.before_state, Some(seed.state_id));
4009        assert!(step.status.is_none());
4010        assert!(step.payload_summary.as_deref().unwrap().contains("call-1"));
4011        assert!(step.payload_hash.is_some());
4012        assert!(
4013            step.labels
4014                .contains(&TimelineLabel::ExternalSideEffectsUnknown)
4015        );
4016    }
4017
4018    #[test]
4019    fn relay_opencode_tool_execute_after_captures_dirty_worktree() {
4020        let (_temp, repo) = init_repo();
4021        let root = repo.root().to_path_buf();
4022        std::fs::write(root.join("tracked.txt"), b"one\n").unwrap();
4023        let seed = repo.snapshot(Some("seed".into()), None).unwrap();
4024        let user_config = UserConfig {
4025            principal: Some(crate::config::UserPrincipalConfig {
4026                name: "Ada Lovelace".to_string(),
4027                email: "ada@example.com".to_string(),
4028            }),
4029            ..UserConfig::default()
4030        };
4031        let mut runtime = HarnessBridgeRuntime::new(repo, user_config);
4032        let payload = opencode_tool_payload("call-2");
4033
4034        relay_opencode(&mut runtime, "tool.execute.before", &payload).unwrap();
4035        std::fs::write(root.join("tracked.txt"), b"two\n").unwrap();
4036        relay_opencode(&mut runtime, "tool.execute.after", &payload).unwrap();
4037
4038        let head = runtime.repo.head().unwrap().expect("capture advanced HEAD");
4039        assert_ne!(head, seed.state_id);
4040        let store = TimelineStore::open(runtime.repo.heddle_dir()).unwrap();
4041        let view = TimelineView::rebuild(&store).unwrap();
4042        let steps = view.steps_for_thread("main");
4043        assert_eq!(steps.len(), 1, "before/after should merge by native id");
4044        let step = steps[0];
4045        assert_eq!(step.operation_ids.len(), 2);
4046        assert_eq!(step.status, Some(TimelineToolCallStatus::Succeeded));
4047        assert_eq!(step.before_state, Some(seed.state_id));
4048        assert_eq!(step.after_state, Some(head));
4049        assert_eq!(step.capture_state, Some(head));
4050        assert_eq!(step.changed, Some(true));
4051        assert!(step.touched_paths.contains(&"tracked.txt".to_string()));
4052        assert!(step.labels.contains(&TimelineLabel::RepoReversible));
4053        assert!(
4054            step.labels
4055                .contains(&TimelineLabel::ExternalSideEffectsUnknown)
4056        );
4057        assert!(!step.payload_summary.as_deref().unwrap().contains("SECRET"));
4058        assert!(step.payload_hash.is_some());
4059    }
4060
4061    #[cfg(unix)]
4062    #[test]
4063    fn relay_opencode_tool_execute_after_records_capture_failed_without_ambient_paths() {
4064        let (_temp, repo) = init_repo();
4065        let root = repo.root().to_path_buf();
4066        std::fs::write(root.join("seed.txt"), b"seed\n").unwrap();
4067        let seed = repo.snapshot(Some("seed".into()), None).unwrap();
4068        let mut runtime = HarnessBridgeRuntime::new(repo, UserConfig::default());
4069        let mut payload = opencode_tool_payload("call-capture-failed");
4070        payload["tool"]["input"]["file_path"] = serde_json::json!("hinted.txt");
4071        let hooks_dir = root.join(".heddle/hooks");
4072        std::fs::create_dir_all(&hooks_dir).unwrap();
4073        let hook_path = hooks_dir.join("pre-snapshot");
4074        std::fs::write(&hook_path, "#!/bin/sh\nexit 1\n").unwrap();
4075        let mut perms = std::fs::metadata(&hook_path).unwrap().permissions();
4076        perms.set_mode(0o755);
4077        std::fs::set_permissions(&hook_path, perms).unwrap();
4078
4079        relay_opencode(&mut runtime, "tool.execute.before", &payload).unwrap();
4080        std::fs::write(root.join("ambient.txt"), b"dirty but uncaptured\n").unwrap();
4081        relay_opencode(&mut runtime, "tool.execute.after", &payload).unwrap();
4082
4083        assert_eq!(
4084            runtime.repo.head().unwrap(),
4085            Some(seed.state_id),
4086            "capture failure must not advance HEAD"
4087        );
4088        let store = TimelineStore::open(runtime.repo.heddle_dir()).unwrap();
4089        let view = TimelineView::rebuild(&store).unwrap();
4090        let steps = view.steps_for_thread("main");
4091        assert_eq!(steps.len(), 1, "before/after should merge by native id");
4092        let step = steps[0];
4093        assert_eq!(step.operation_ids.len(), 2);
4094        assert_eq!(step.before_state, Some(seed.state_id));
4095        assert_eq!(step.after_state, Some(seed.state_id));
4096        assert_eq!(step.capture_state, None);
4097        assert_eq!(step.changed, Some(false));
4098        assert!(step.labels.contains(&TimelineLabel::CaptureFailed));
4099        assert!(
4100            !step.labels.contains(&TimelineLabel::RepoReversible),
4101            "failed captures are not repo-reversible"
4102        );
4103        assert_eq!(step.touched_paths, vec!["hinted.txt"]);
4104    }
4105
4106    #[test]
4107    fn relay_opencode_tool_execute_missing_tool_id_does_not_fail_or_record_timeline() {
4108        let (_temp, repo) = init_repo();
4109        let root = repo.root().to_path_buf();
4110        std::fs::write(root.join("seed.txt"), b"hello").unwrap();
4111        let _ = repo.snapshot(Some("seed".into()), None).unwrap();
4112        let mut runtime = HarnessBridgeRuntime::new(repo, UserConfig::default());
4113        let payload = serde_json::json!({
4114            "sessionID": "opencode-session",
4115            "model": "gpt-5.4",
4116            "provider": "openai",
4117            "tool": {"name": "bash"},
4118        });
4119
4120        relay_opencode(&mut runtime, "tool.execute.before", &payload).unwrap();
4121
4122        let store = TimelineStore::open(runtime.repo.heddle_dir()).unwrap();
4123        let view = TimelineView::rebuild(&store).unwrap();
4124        assert!(view.steps_for_thread("main").is_empty());
4125        let report_count = std::fs::read_dir(root.join(".heddle/state/session-reports"))
4126            .unwrap()
4127            .count();
4128        assert!(
4129            report_count > 0,
4130            "session progress should still be recorded"
4131        );
4132    }
4133
4134    fn opencode_tool_payload(call_id: &str) -> Value {
4135        serde_json::json!({
4136            "sessionID": "opencode-session",
4137            "messageID": "message-1",
4138            "toolCallID": call_id,
4139            "model": "gpt-5.4",
4140            "provider": "openai",
4141            "tool": {
4142                "name": "bash",
4143                "input": {
4144                    "command": "echo SECRET",
4145                    "file_path": "tracked.txt"
4146                }
4147            },
4148            "status": "success"
4149        })
4150    }
4151
4152    #[test]
4153    fn relay_claude_subagent_start_creates_child_entry_with_parent_key() {
4154        let (temp, repo) = init_repo();
4155        drop(repo);
4156        let fresh_repo = Repository::open(temp.path()).unwrap();
4157        let mut runtime = HarnessBridgeRuntime::new(fresh_repo, UserConfig::default());
4158        let payload = serde_json::json!({
4159            "session_id": "parent-claude-sess",
4160            "agent_id": "child-subagent-xyz",
4161            "model": {"id": "claude-sonnet-4-6"},
4162        });
4163        relay_claude(&mut runtime, "SubagentStart", &payload).unwrap();
4164        drop(runtime);
4165
4166        let verify = Repository::open(temp.path()).unwrap();
4167        let registry = ActorPresenceStore::new(verify.heddle_dir());
4168        let child = registry
4169            .find_active_by_native_actor_key("claude-code:agent:child-subagent-xyz")
4170            .unwrap()
4171            .expect("subagent ActorPresence should exist after SubagentStart");
4172        assert_eq!(
4173            child.native_parent_actor_key.as_deref(),
4174            Some("claude-code:session:parent-claude-sess"),
4175            "subagent must carry parent session linkage",
4176        );
4177        assert_eq!(child.status, ActorPresenceStatus::Active);
4178    }
4179
4180    #[test]
4181    fn relay_claude_subagent_stop_marks_child_entry_complete() {
4182        let (temp, repo) = init_repo();
4183        let repo_root = repo.root().to_path_buf();
4184        drop(repo);
4185
4186        // Start: create the child entry.
4187        let fresh = Repository::open(temp.path()).unwrap();
4188        let mut runtime = HarnessBridgeRuntime::new(fresh, UserConfig::default());
4189        let start_payload = serde_json::json!({
4190            "session_id": "parent-sess",
4191            "agent_id": "worker-1",
4192            "model": {"id": "claude-sonnet-4-6"},
4193        });
4194        relay_claude(&mut runtime, "SubagentStart", &start_payload).unwrap();
4195        drop(runtime);
4196
4197        // Dirty the worktree so SubagentStop also captures a state.
4198        std::fs::write(
4199            repo_root.join("child-output.txt"),
4200            b"subagent produced this",
4201        )
4202        .unwrap();
4203
4204        let fresh = Repository::open(temp.path()).unwrap();
4205        let mut runtime = HarnessBridgeRuntime::new(fresh, UserConfig::default());
4206        let stop_payload = serde_json::json!({
4207            "session_id": "parent-sess",
4208            "agent_id": "worker-1",
4209            "model": {
4210                "id": "claude-sonnet-4-6",
4211                "display_name": "Claude Sonnet 4.6",
4212            },
4213        });
4214        relay_claude(&mut runtime, "SubagentStop", &stop_payload).unwrap();
4215        drop(runtime);
4216
4217        let verify = Repository::open(temp.path()).unwrap();
4218        let registry = ActorPresenceStore::new(verify.heddle_dir());
4219        let child = registry
4220            .list()
4221            .unwrap()
4222            .into_iter()
4223            .find(|e| e.native_actor_key.as_deref() == Some("claude-code:agent:worker-1"))
4224            .expect("child entry should still exist");
4225        assert_eq!(
4226            child.status,
4227            ActorPresenceStatus::Complete,
4228            "SubagentStop should mark the child entry Complete",
4229        );
4230    }
4231
4232    #[test]
4233    fn relay_claude_user_prompt_submit_rotates_segment() {
4234        let (temp, repo) = init_repo();
4235        drop(repo);
4236
4237        let fresh = Repository::open(temp.path()).unwrap();
4238        let mut runtime = HarnessBridgeRuntime::new(fresh, UserConfig::default());
4239        // SessionStart establishes the Heddle session + initial segment.
4240        let session_payload = serde_json::json!({
4241            "session_id": "claude-prompt-sess",
4242            "model": {"id": "claude-opus-4-7", "display_name": "Claude Opus 4.7"},
4243        });
4244        relay_claude(&mut runtime, "SessionStart", &session_payload).unwrap();
4245        let sessions_before = SessionManager::new(runtime.repo.root())
4246            .list_sessions(true)
4247            .unwrap();
4248        let initial_segments = sessions_before
4249            .iter()
4250            .find(|s| !s.segments.is_empty())
4251            .map(|s| s.segments.len())
4252            .unwrap_or(0);
4253
4254        // UserPromptSubmit should force a new segment.
4255        let prompt_payload = serde_json::json!({
4256            "session_id": "claude-prompt-sess",
4257            "model": {"id": "claude-opus-4-7", "display_name": "Claude Opus 4.7"},
4258            "prompt": "write a new feature",
4259        });
4260        relay_claude(&mut runtime, "UserPromptSubmit", &prompt_payload).unwrap();
4261        drop(runtime);
4262
4263        let verify = Repository::open(temp.path()).unwrap();
4264        let sessions_after = SessionManager::new(verify.root())
4265            .list_sessions(true)
4266            .unwrap();
4267        let rotated = sessions_after
4268            .iter()
4269            .any(|s| s.segments.len() > initial_segments);
4270        assert!(
4271            rotated,
4272            "UserPromptSubmit must add at least one segment beyond the SessionStart baseline \
4273             (initial={initial_segments}, sessions_after={:?})",
4274            sessions_after
4275                .iter()
4276                .map(|s| (s.id.clone(), s.segments.len()))
4277                .collect::<Vec<_>>(),
4278        );
4279    }
4280}