csusage-adapter-kimi 0.0.2

Coding agent CLI usage reports (ccusage fork with Claude Science support)
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
111
112
113
114
115

use std::{
    collections::HashSet,
    env,
    path::{Component, Path, PathBuf},
};

use crate::{Result, collect_files_with_extension};

pub(super) const KIMI_DATA_DIR_ENV: &str = "KIMI_DATA_DIR";
const KIMI_SESSIONS_DIR_NAME: &str = "sessions";
const KIMI_WIRE_FILE_NAME: &str = "wire.jsonl";

fn paths() -> Result<Vec<PathBuf>> {
    let mut paths = Vec::new();
    let mut seen = HashSet::new();
    if let Ok(env_paths) = env::var(KIMI_DATA_DIR_ENV) {
        for raw in env_paths
            .split(',')
            .map(str::trim)
            .filter(|path| !path.is_empty())
        {
            let path = PathBuf::from(raw);
            if path.is_dir() && seen.insert(path.clone()) {
                paths.push(path);
            }
        }
        return Ok(paths);
    }

    if let Some(home) = crate::home::home_dir() {
        for dir in [".kimi", ".kimi-code"] {
            let path = home.join(dir);
            if path.is_dir() && seen.insert(path.clone()) {
                paths.push(path);
            }
        }
    }
    Ok(paths)
}

pub(super) fn discover_wire_files() -> Result<Vec<PathBuf>> {
    let mut files = Vec::new();
    for kimi_path in paths()? {
        let sessions_path = kimi_path.join(KIMI_SESSIONS_DIR_NAME);
        let mut candidates = Vec::new();
        collect_files_with_extension(&sessions_path, "jsonl", &mut candidates);
        files.extend(
            candidates
                .into_iter()
                .filter(|file| is_kimi_wire_file(&sessions_path, file)),
        );
    }
    files.sort();
    files.dedup();
    Ok(files)
}

fn is_kimi_wire_file(sessions_path: &Path, file_path: &Path) -> bool {
    if file_path.file_name().and_then(|name| name.to_str()) != Some(KIMI_WIRE_FILE_NAME) {
        return false;
    }
    let Ok(relative) = file_path.strip_prefix(sessions_path) else {
        return false;
    };
    // Old format: sessions/<group>/<session>/wire.jsonl             → 3 components
    // New format: sessions/<workspace>/<session>/agents/<agent>/wire.jsonl → 5 components
    matches!(
        relative
            .components()
            .filter(|component| matches!(component, Component::Normal(_)))
            .count(),
        3 | 5
    )
}

#[cfg(test)]
mod tests {
    use super::*;
    use csusage_test_support::{EnvVarGuard, fs_fixture};

    #[test]
    fn discovers_wire_jsonl_files_under_sessions_group_session() {
        let fixture = fs_fixture!({
            "sessions/group/session/wire.jsonl": "{}\n",
            "sessions/group/session/other.jsonl": "{}\n",
            "sessions/nested/path/session/wire.jsonl": "{}\n",
        });
        let _cleanup = EnvVarGuard::set(KIMI_DATA_DIR_ENV, fixture.root());
        let files = discover_wire_files().unwrap();

        assert_eq!(
            files,
            vec![fixture.path("sessions/group/session/wire.jsonl")]
        );
    }

    #[test]
    fn discovers_both_old_and_new_layouts_and_skips_non_wire_files() {
        let fixture = fs_fixture!({
            "sessions/ws/session-c/agents/agent-1/wire.jsonl": "{}\n",
            "sessions/ws/session-c/agents/agent-1/other.jsonl": "{}\n",
            "sessions/group/session-d/wire.jsonl": "{}\n",
        });
        let _cleanup = EnvVarGuard::set(KIMI_DATA_DIR_ENV, fixture.root());
        let files = discover_wire_files().unwrap();

        assert_eq!(
            files,
            vec![
                fixture.path("sessions/group/session-d/wire.jsonl"),
                fixture.path("sessions/ws/session-c/agents/agent-1/wire.jsonl"),
            ]
        );
    }
}