heddle-cli 0.2.0

An AI-native version control system
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

// SPDX-License-Identifier: Apache-2.0
use anyhow::Result;

use super::{
    HarnessActorProbe, HarnessAttachHints, HarnessProbeInput, HarnessProbeResult, ProbeSource,
    argv_value, csv_paths, parse_u64,
};

pub(crate) struct ClaudeCodeProbe;

impl HarnessActorProbe for ClaudeCodeProbe {
    fn harness_name(&self) -> &'static str {
        "claude-code"
    }

    fn matches(&self, input: &HarnessProbeInput) -> bool {
        input.explicit_harness.as_deref() == Some(self.harness_name())
            || input.probe_metadata.contains_key("session_id")
            || input.probe_metadata.contains_key("agent_id")
            || input.env_hints.contains_key("CLAUDECODE")
            || input
                .argv
                .as_ref()
                .and_then(|argv| argv.first())
                .is_some_and(|program| program.to_ascii_lowercase().contains("claude"))
    }

    fn probe(&self, input: &HarnessProbeInput) -> Result<HarnessProbeResult> {
        let metadata = &input.probe_metadata;
        let argv = input.argv.as_deref().unwrap_or(&[]);
        let session_id = metadata
            .get("session_id")
            .cloned()
            .or_else(|| argv_value(argv, "--session-id"));
        let agent_id = metadata.get("agent_id").cloned();
        let transcript_path = metadata
            .get("transcript_path")
            .cloned()
            .or_else(|| input.env_hints.get("CLAUDE_TRANSCRIPT_PATH").cloned());
        let probe_source = if metadata.get("hook_event").is_some() {
            ProbeSource::HookPayload
        } else if metadata.get("status_line").is_some() {
            ProbeSource::StatusPayload
        } else if session_id.is_some() {
            ProbeSource::AppProtocol
        } else {
            ProbeSource::ArgvEnv
        };
        Ok(HarnessProbeResult {
            harness: Some("claude-code".to_string()),
            provider: Some("anthropic".to_string()),
            model: metadata
                .get("model")
                .cloned()
                .or_else(|| argv_value(argv, "--model")),
            thinking_level: metadata
                .get("effort")
                .cloned()
                .or_else(|| argv_value(argv, "--effort")),
            native_actor_key: agent_id
                .clone()
                .map(|id| format!("claude-code:agent:{id}"))
                .or_else(|| {
                    session_id
                        .clone()
                        .map(|id| format!("claude-code:session:{id}"))
                }),
            native_parent_actor_key: agent_id
                .as_ref()
                .and(session_id.as_ref())
                .map(|session| format!("claude-code:session:{session}")),
            native_instance_key: transcript_path
                .clone()
                .map(|path| format!("claude-code:transcript:{path}")),
            usage_totals: proto::UsageTotals {
                input_tokens: parse_u64(metadata.get("input_tokens")),
                output_tokens: parse_u64(metadata.get("output_tokens")),
                reasoning_tokens: parse_u64(metadata.get("reasoning_tokens")),
                cache_creation_tokens: None,
                cache_read_tokens: None,
                tool_calls: metadata.get("tool_calls").and_then(|v| v.parse().ok()),
                cost_micros_usd: parse_u64(metadata.get("cost_micros_usd")),
            },
            touched_paths: csv_paths(metadata.get("touched_paths")),
            transcript_refs: transcript_path
                .map(|path| {
                    vec![proto::TranscriptAttachmentRef {
                        attachment_id: path,
                        kind: Some("transcript_path".to_string()),
                        summary: None,
                    }]
                })
                .unwrap_or_default(),
            attach_hints: HarnessAttachHints {
                root_actor: agent_id.is_none(),
            },
            confidence: Some(if metadata.get("hook_event").is_some() {
                0.96
            } else if metadata.get("status_line").is_some() {
                0.9
            } else if session_id.is_some() {
                0.75
            } else {
                0.55
            }),
            probe_source: Some(probe_source.as_str().to_string()),
            ..HarnessProbeResult::default()
        })
    }
}