clustervision 0.18.0

clustervision CLI — search, read, and port AI coding-agent sessions across harnesses.
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

//! The pretty Session envelopes around staged message frames: the archive's (with its reusable
//! stable prefix) and the CLI's (optionally as an OpenSession document). The writer itself is
//! [`cv_core::projection::write_frames`].
use anyhow::Result;
use cv_core::bytes::ByteSource;
use cv_core::projection::{write_frames, Framing, Prefix};
use cv_core::Session;
use std::{io::Write, path::Path};

pub(crate) fn write_session(
    writer: &mut impl Write,
    session: &Session,
    spool: &ByteSource,
    old: Option<(&Path, &Prefix)>,
    stable: Option<usize>,
    total: usize,
) -> Result<Option<Prefix>> {
    anyhow::ensure!(
        session.messages.is_empty(),
        "JSON header unexpectedly retained messages"
    );
    let framing = Framing {
        pretty: true,
        old,
        stable,
        total: Some(total),
    };
    write_frames(writer, session, "messages", spool, framing)
}

pub(crate) fn write_document(
    writer: &mut impl Write,
    session: &Session,
    spool: &ByteSource,
    total: usize,
    document: bool,
) -> Result<()> {
    anyhow::ensure!(
        session.messages.is_empty(),
        "JSON header unexpectedly retained messages"
    );
    let framing = Framing {
        pretty: true,
        total: Some(total),
        ..Framing::default()
    };
    if document {
        write_frames(
            writer,
            &cv_core::harness::opensession::document(session),
            "messages",
            spool,
            framing,
        )?;
    } else {
        write_frames(writer, session, "messages", spool, framing)?;
    }
    Ok(())
}