OXYGENHUB

Coverage / crates/oxy-core/src/state.rs 93.3%

2187 of 2344 lines run, 461 of 660 functions · run #348 on paths-for-resolutions-and-amend · run missed

1
//! A state: a dependency-closed set of changes, applied to a tree graph and
2
//! per-node file graphs (spec §8.3), with materialization (spec §9.5, §11).
3
4
use crate::cbor::{self, Value};
5
use crate::graph::{AtomKey, DepOracle, FileGraph, Materialized, Vertex};
6
use crate::hash::KeyId;
7
use crate::hash::{ChangeHash, ObjectRef};
8
use crate::ids::NodeId;
9
use crate::objects::{
10
    self as o, AtomRef, Change, ChangeKind, Edit, Hlc, Mode, NodeKind, ObjectError, TreeOp,
11
};
12
use crate::tree::{TreeGraph, TreeView};
13
use std::collections::{BTreeMap, BTreeSet, HashMap, HashSet};
14
use std::sync::Arc;
15
16
/// Checkpoint payload version 2 (spec §11): file graphs by reference.
17
pub const CHECKPOINT_INDEX_VERSION: u64 = 2;
18
19
/// A file graph encoded in fewer bytes stays inline in a version 2 index.
20
pub const INLINE_GRAPH_LIMIT: usize = 2048;
21
22
#[derive(Debug, Clone, PartialEq, Eq, thiserror::Error)]
23
pub enum StateError {
24
    #[error("change {1} depends on {0}, which is not in the state")]
25
    MissingDep(ChangeHash, ChangeHash),
26
}
27
28
/// One `Replace` edit: `(change, old blob, new blob)`.
29
type ReplaceRecord = (ChangeHash, Option<ObjectRef>, Option<ObjectRef>);
30
31
/// What a state remembers about a change: everything but its operations,
32
/// which live in the graphs once applied (and in the store).
33
#[derive(Clone, Debug, PartialEq, Eq)]
34
pub struct ChangeMeta {
35
    pub change_id: crate::ids::ChangeId,
36
    pub deps: Vec<ChangeHash>,
37
    pub author: KeyId,
38
    pub hlc: Hlc,
39
    pub message: String,
40
    pub kind: ChangeKind,
41
}
42
43
/// The log's order of changes: newest first by clock, then by hash, so
44
/// it is total.
451167×
pub fn log_order(a: (&ChangeHash, &Hlc), b: (&ChangeHash, &Hlc)) -> std::cmp::Ordering {
461167×
    b.1.cmp(a.1).then(b.0.cmp(a.0))
471167×
}
48
49
impl ChangeMeta {
50
    /// In the flat layout of local indexes and caches ([`crate::flat`]).
516524×
    pub(crate) fn write_flat(&self, w: &mut crate::flat::Writer) {
526524×
        w.fixed(&self.change_id.0);
536524×
        w.count(self.deps.len());
546524×
        for d in &self.deps {
555979×
            w.object(d);
565979×
        }
576524×
        w.object(&self.author);
586524×
        w.u64(self.hlc.wall_ms);
596524×
        w.u32(self.hlc.logical);
606524×
        w.bytes(self.message.as_bytes());
616524×
        w.u8(self.kind.tag() as u8);
626524×
    }
63
6411546×
    pub(crate) fn read_flat(r: &mut crate::flat::Reader) -> Result<ChangeMeta, ObjectError> {
6511546×
        let change_id = crate::ids::ChangeId(r.fixed::<16>()?);
6611546×
        let mut deps = Vec::new();
6711546×
        for _ in 0..r.count()? {
6810705×
            deps.push(r.object()?);
69
        }
70
        Ok(ChangeMeta {
7111546×
            change_id,
7211546×
            deps,
7311546×
            author: r.object()?,
74
            hlc: Hlc {
7511546×
                wall_ms: r.u64()?,
7611546×
                logical: r.u32()?,
77
            },
7811546×
            message: r.text()?,
7911546×
            kind: ChangeKind::from_tag(r.u8()? as u64)?,
80
        })
8111546×
    }
82
8316114×
    pub fn of(c: &Change) -> ChangeMeta {
8416114×
        ChangeMeta {
8516114×
            change_id: c.change_id,
8616114×
            deps: c.deps.clone(),
8716114×
            author: c.author,
8816114×
            hlc: c.hlc,
8916114×
            message: c.message.clone(),
9016114×
            kind: c.kind,
9116114×
        }
9216114×
    }
93
}
94
95
#[derive(Default)]
96
pub struct State {
97
    /// Shared between clones, like `files` and `deferred`: a state kept
98
    /// beside another that differs by a few changes shares the rest.
99
    changes: crate::fasthash::FastMap<ChangeHash, Arc<ChangeMeta>>,
100
    tree: TreeGraph,
101
    /// Shared between clones: a clone copies a graph only when an edit
102
    /// changes it (`Arc::make_mut`), so keeping several states that differ
103
    /// by a few changes costs little more than one.
104
    files: crate::fasthash::FastMap<NodeId, Arc<FileGraph>>,
105
    replaces: HashMap<NodeId, Vec<ReplaceRecord>>,
106
    /// Memoised transitive dependencies, bounded by the hashes they hold
107
    /// (each set is the whole history below a change, so keeping one for
108
    /// every change asked about grew with the square of history: #25). A
109
    /// mutex, not a `RefCell`, so a loaded state can be shared across
110
    /// threads by a host that caches it.
111
    closure: std::sync::RwLock<ClosureMemo>,
112
    /// Each change's level: 0 with no dependencies, else one more than its
113
    /// highest dependency. A change can only depend on changes of lower
114
    /// level, which bounds the walk [`DepOracle::depends_on`] does.
115
    levels: std::sync::RwLock<crate::fasthash::FastMap<ChangeHash, u32>>,
116
    /// `None` for a full state. For a history-only state (loaded for its
117
    /// changes and tree), the file graphs not built yet, by node: `files`
118
    /// holds only those loaded with [`State::load_graphs`].
119
    deferred: Option<crate::fasthash::FastMap<NodeId, Arc<Deferred>>>,
120
    /// The tree as readers see it ([`State::view`]), once worked out for
121
    /// a tree with name conflicts; cleared by every change added.
122
    unified_view: std::sync::OnceLock<TreeView>,
123
}
124
125
use crate::objects::TextRef;
126
127
/// A file graph a history-only state has not built: its base from a
128
/// version 2 checkpoint, if any, and the edits of later changes in the
129
/// order they were applied (spec §11). Building it later gives the same
130
/// graph, since graph construction does not depend on order (§8).
131
///
132
/// Read from a state index, it stays in the index, unread, until first
133
/// used: a command that builds a few files reads and checks only their
134
/// histories (oxygen#94).
135
#[derive(Clone, Default)]
136
struct Deferred {
137
    /// The history: `raw` read and decoded, with `more` after it. An
138
    /// error when `raw` does not match its hash.
139
    body: std::sync::OnceLock<Result<DeferredBody, String>>,
140
    /// The history as an index holds it, not read yet.
141
    raw: Option<RawHistory>,
142
    /// Changes added since the index was read, kept beside `raw`, so that
143
    /// adding one neither reads nor decodes the history.
144
    more: Vec<(ChangeHash, Vec<Edit>)>,
145
    more_texts: Vec<TextRef>,
146
    /// Its graph key, once worked out (or read from an index); reset by
147
    /// any edit added.
148
    key: std::sync::OnceLock<[u8; 32]>,
149
}
150
151
/// Where a state index's file histories are: the index file itself, read
152
/// on first use, or bytes in memory.
153
#[derive(Clone)]
154
pub enum Histories {
155
    Bytes(Arc<Vec<u8>>),
156
    /// The file, where its histories start, and how many bytes they take.
157
    File(Arc<std::fs::File>, u64, u64),
158
}
159
160
impl Histories {
16116736×
    fn len(&self) -> u64 {
16216736×
        match self {
1634569×
            Histories::Bytes(b) => b.len() as u64,
16412167×
            Histories::File(_, _, len) => *len,
165
        }
16616736×
    }
167
16821891×
    fn read(&self, at: u64, len: usize) -> std::io::Result<Vec<u8>> {
16921891×
        match self {
1704569×
            Histories::Bytes(b) => Ok(b[at as usize..at as usize + len].to_vec()),
17117322×
            Histories::File(f, start, _) => {
17217322×
                let mut buf = vec![0u8; len];
17317322×
                read_exact_at(f, &mut buf, start + at)?;
17417322×
                Ok(buf)
175
            }
176
        }
17721891×
    }
178
}
179
180
#[cfg(unix)]
18117322×
fn read_exact_at(f: &std::fs::File, buf: &mut [u8], at: u64) -> std::io::Result<()> {
18217322×
    std::os::unix::fs::FileExt::read_exact_at(f, buf, at)
18317322×
}
184
185
#[cfg(windows)]
186
fn read_exact_at(f: &std::fs::File, mut buf: &mut [u8], mut at: u64) -> std::io::Result<()> {
187
    while !buf.is_empty() {
188
        match std::os::windows::fs::FileExt::seek_read(f, buf, at)? {
189
            0 => return Err(std::io::ErrorKind::UnexpectedEof.into()),
190
            n => {
191
                buf = &mut buf[n..];
192
                at += n as u64;
193
            }
194
        }
195
    }
196
    Ok(())
197
}
198
199
/// One file's history in a state index: where it is and its hash.
200
#[derive(Clone)]
201
struct RawHistory {
202
    from: Histories,
203
    at: u64,
204
    len: usize,
205
    sum: [u8; 32],
206
}
207
208
impl RawHistory {
209
    /// The history's bytes, checked against the hash the index holds.
21021891×
    fn bytes(&self) -> Result<Vec<u8>, ObjectError> {
21121891×
        let damaged = |why: String| ObjectError::DamagedIndex(why);
21221891×
        let b = self
21321891×
            .from
21421891×
            .read(self.at, self.len)
21521891×
            .map_err(|e| damaged(format!("a file's history cannot be read: {e}")))?;
21621891×
        if *blake3::hash(&b).as_bytes() != self.sum {
2174×
            return Err(damaged("a file's history does not match its hash".into()));
21821887×
        }
21921887×
        Ok(b)
22021891×
    }
221
}
222
223
/// The closures kept, and how many hashes they hold in all.
224
#[derive(Default)]
225
struct ClosureMemo {
226
    sets: HashMap<ChangeHash, Arc<HashSet<ChangeHash>>>,
227
    held: usize,
228
}
229
230
/// The most hashes the closure memo holds before it starts again: a few
231
/// hundred megabytes at most.
232
const CLOSURE_MEMO_HASHES: usize = 4_000_000;
233
234
#[derive(Clone, Default)]
235
struct DeferredBody {
236
    base: Option<DeferredBase>,
237
    edits: Vec<(ChangeHash, Vec<Edit>)>,
238
    /// The text blob and line count of each insert in `edits` left without
239
    /// its lines, in order: read when the graph is built.
240
    texts: Vec<TextRef>,
241
}
242
243
impl Deferred {
244141×
    fn from_body(body: DeferredBody) -> Deferred {
245141×
        Deferred {
246141×
            body: std::sync::OnceLock::from(Ok(body)),
247141×
            ..Deferred::default()
248141×
        }
249141×
    }
250
251
    /// The history, read and checked on first use.
252103546×
    fn body(&self) -> Result<&DeferredBody, ObjectError> {
253103546×
        self.body
254103546×
            .get_or_init(|| {
25538086×
                let mut b = match &self.raw {
256
                    // Matching its hash, it is what `write_index` wrote: it
257
                    // decodes.
2585477×
                    Some(raw) => raw
2595477×
                        .bytes()
2605477×
                        .and_then(|b| State::read_deferred_body(&b))
2615477×
                        .map_err(|e| e.to_string())?,
26232609×
                    None => DeferredBody::default(),
263
                };
26438085×
                b.edits.extend(self.more.iter().cloned());
26538085×
                b.texts.extend(self.more_texts.iter().cloned());
26638085×
                Ok(b)
26738086×
            })
268103546×
            .as_ref()
269103546×
            .map_err(|e| ObjectError::DamagedIndex(e.clone()))
270103546×
    }
271
272
    /// Add a change's edits to the history: beside the index's encoding
273
    /// while it is unread, so that nothing is read or decoded to add them.
27435748×
    fn add(&mut self, h: ChangeHash, edits: Vec<Edit>, texts: Vec<TextRef>) {
27535748×
        if self.raw.is_some() && !matches!(self.body.get(), Some(Ok(_))) {
2762×
            self.more.push((h, edits));
2772×
            self.more_texts.extend(texts);
278
            // A body found damaged stays so: what was added cannot mend it.
2792×
            return;
28035746×
        }
281
        // No encoding to read, or decoded already: this cannot fail.
28235746×
        let b = self.body_mut().expect("a history in memory");
28335746×
        b.edits.push((h, edits));
28435746×
        b.texts.extend(texts);
28535748×
    }
286
28735746×
    fn body_mut(&mut self) -> Result<&mut DeferredBody, ObjectError> {
28835746×
        self.body()?;
28935746×
        self.raw = None;
29035746×
        self.more.clear();
29135746×
        self.more_texts.clear();
29235746×
        Ok(self
29335746×
            .body
29435746×
            .get_mut()
29535746×
            .and_then(|b| b.as_mut().ok())
29635746×
            .expect("decoded above"))
29735746×
    }
298
29923433×
    fn into_body(mut self) -> Result<DeferredBody, ObjectError> {
30023433×
        self.body()?;
30123432×
        Ok(self
30223432×
            .body
30323432×
            .take()
30423432×
            .and_then(|b| b.ok())
30523432×
            .expect("decoded above"))
30623433×
    }
307
}
308
309
#[derive(Clone)]
310
enum DeferredBase {
311
    Inline(Value),
312
    Blob(ObjectRef),
313
}
314
315
/// A local index of a history-only state (oxygen#41), in the flat layout
316
/// of [`crate::flat`]: its change metadata, tree (with its view), replace
317
/// records and every file's deferred history with its graph key, so a
318
/// command reads it rather than replaying history. Derived only: written
319
/// from a state loaded the long way, and read back as the same state.
320
impl State {
321
    const INDEX_MAGIC: &'static [u8] = b"oxy state index 4\n";
322
323
    /// The index of this state, or `None` when it has anything an index
324
    /// does not hold: a built graph, or text in memory (a change added
325
    /// whole rather than as its skeleton).
326984×
    pub fn write_index(&self) -> Option<Vec<u8>> {
327
        use crate::flat::Writer;
328984×
        let deferred = self.deferred.as_ref()?;
329984×
        if !self.files.is_empty() {
3300×
            return None;
331984×
        }
332
        // The changes first, as a block a reader can take on its own (see
333
        // [`State::read_index_changes`]).
334984×
        let mut w = Writer::default();
335984×
        w.fixed(Self::INDEX_MAGIC);
336
        // In log order, newest first (oxygen#94): the log takes them as
337
        // they come, with no map and no sort.
338984×
        let mut changes: Vec<(&ChangeHash, &Arc<ChangeMeta>)> = self.changes.iter().collect();
339984×
        changes.sort_unstable_by(|a, b| log_order((a.0, &a.1.hlc), (b.0, &b.1.hlc)));
340984×
        w.count(changes.len());
341984×
        for (h, m) in changes {
342268×
            w.object(h);
343268×
            m.write_flat(&mut w);
344268×
        }
345984×
        let changes_block = std::mem::take(&mut w);
346984×
        w.bytes(&changes_block.0);
347
        // The tree and the files' histories as blocks of their own, so a
348
        // reader decodes them side by side.
349984×
        let mut tw = Writer::default();
350984×
        self.tree.write_flat(&mut tw);
351984×
        w.bytes(&tw.0);
352984×
        let mut replaces: Vec<(&NodeId, &Vec<ReplaceRecord>)> = self.replaces.iter().collect();
353984×
        replaces.sort_by_key(|(n, _)| **n);
354984×
        w.count(replaces.len());
355984×
        for (n, recs) in replaces {
3564×
            w.fixed(&n.0);
3574×
            w.count(recs.len());
358
            // In any order: which record is live does not depend on it.
3594×
            let mut recs: Vec<&ReplaceRecord> = recs.iter().collect();
3604×
            recs.sort();
3614×
            for (c, old, new) in recs {
3624×
                w.object(c);
3634×
                w.opt_object(old);
3644×
                w.opt_object(new);
3654×
            }
366
        }
367984×
        let mut front = w;
368
        // The table: per file its key and where its history is, with the
369
        // history's hash, which is checked when it is read (oxygen#94).
370984×
        let mut table = Writer::default();
371984×
        let mut histories: Vec<u8> = Vec::new();
372984×
        let mut nodes: Vec<(&NodeId, &Arc<Deferred>)> = deferred.iter().collect();
373984×
        nodes.sort_by_key(|(n, _)| **n);
374984×
        table.count(nodes.len());
37532810×
        for (n, d) in nodes {
37632810×
            let key = match d.key.get() {
37716414×
                Some(k) => *k,
378
                None => {
37916396×
                    let mut h = blake3::Hasher::new();
38016396×
                    h.update(b"oxy graph key 1\0");
38116396×
                    h.update(&n.0);
38216396×
                    let k = d.graph_key(h).ok()?;
38316396×
                    let _ = d.key.set(k);
38416396×
                    k
385
                }
386
            };
387
            // Unchanged since read: its bytes as they are, checked.
38832810×
            let bytes = match &d.raw {
38916414×
                Some(raw) if d.more.is_empty() => raw.bytes().ok()?,
39016396×
                _ => Self::write_deferred_body(d.body().ok()?)?,
391
            };
39232807×
            table.fixed(&n.0);
39332807×
            table.fixed(&key);
39432807×
            table.u64(histories.len() as u64);
39532807×
            table.u32(u32::try_from(bytes.len()).ok()?);
39632807×
            table.fixed(blake3::hash(&bytes).as_bytes());
39732807×
            histories.extend_from_slice(&bytes);
398
        }
399981×
        front.bytes(&table.0);
400981×
        let mut out = Writer::default();
401981×
        out.u64(front.0.len() as u64);
402981×
        out.fixed(&front.0);
403981×
        out.fixed(&histories);
404981×
        Some(out.0)
405984×
    }
406
407
    /// One file's history as an index encodes it; `None` when it holds
408
    /// text in memory, which an index does not.
40916396×
    fn write_deferred_body(d: &DeferredBody) -> Option<Vec<u8>> {
410
        use crate::flat::Writer;
41133584×
        let atom = |w: &mut Writer, a: &AtomRef| match a {
41216399×
            AtomRef::Start => w.u8(0),
41316505×
            AtomRef::End => w.u8(1),
414680×
            AtomRef::Atom { change, index } => {
415680×
                w.u8(2);
416680×
                w.object(change);
417680×
                w.u32(*index);
418680×
            }
41933584×
        };
42016396×
        let mut w = Writer::default();
4213×
        match &d.base {
42216393×
            None => w.u8(0),
4230×
            Some(DeferredBase::Blob(r)) => {
4240×
                w.u8(1);
4250×
                w.object(r);
4260×
            }
4273×
            Some(DeferredBase::Inline(v)) => {
4283×
                w.u8(2);
4293×
                w.bytes(&cbor::encode(v));
4303×
            }
431
        }
432
        // Each change's edits with the texts of its inserts, sorted by
433
        // change: a graph does not depend on the order changes applied
434
        // in, and the index is then the same however history was read.
43516396×
        let mut texts = d.texts.iter();
43616396×
        let mut per_change: Vec<(&ChangeHash, &Vec<Edit>, Vec<&TextRef>)> = Vec::new();
43716534×
        for (c, edits) in &d.edits {
43816534×
            let n = edits
43916534×
                .iter()
44016792×
                .filter(|e| matches!(e, Edit::Insert { lines, .. } if lines.is_empty()))
44116534×
                .count();
44216534×
            per_change.push((c, edits, texts.by_ref().take(n).collect()));
443
        }
44416396×
        per_change.sort_by_key(|(c, _, _)| **c);
44516396×
        w.count(per_change.len());
44616534×
        for (c, edits, _) in &per_change {
44716534×
            w.object(c);
44816534×
            w.count(edits.len());
44916833×
            for e in edits.iter() {
45016833×
                match e {
451
                    Edit::Insert {
45216792×
                        node,
45316792×
                        after,
45416792×
                        before,
45516792×
                        lines,
456
                    } => {
45716792×
                        if !lines.is_empty() {
4580×
                            return None;
45916792×
                        }
46016792×
                        w.u8(0);
46116792×
                        w.fixed(&node.0);
46216792×
                        atom(&mut w, after);
46316792×
                        atom(&mut w, before);
464
                    }
46541×
                    Edit::Delete { node, atoms } => {
46641×
                        w.u8(1);
46741×
                        w.fixed(&node.0);
46841×
                        w.count(atoms.len());
469287×
                        for r in atoms {
470287×
                            w.object(&r.change);
471287×
                            w.u32(r.from);
472287×
                            w.u32(r.to);
473287×
                        }
474
                    }
4750×
                    Edit::Replace { .. } => return None,
476
                }
477
            }
478
        }
47916396×
        let texts: Vec<&TextRef> = per_change
48016396×
            .iter()
48116534×
            .flat_map(|(_, _, t)| t.iter().copied())
48216396×
            .collect();
48316396×
        w.count(texts.len());
48416792×
        for t in texts {
48516792×
            match t {
4861×
                TextRef::Blob(r, count, form) => {
4871×
                    w.u8(0);
4881×
                    w.object(r);
4891×
                    w.u32(*count);
4901×
                    w.u8(match form {
4910×
                        crate::objects::TextForm::Cbor => 0,
4921×
                        crate::objects::TextForm::Raw => 1,
4930×
                        crate::objects::TextForm::CborFallback => 2,
494
                    });
495
                }
49616791×
                TextRef::Inline { change, start, end } => {
49716791×
                    w.u8(1);
49816791×
                    w.object(change);
49916791×
                    w.u32(*start);
50016791×
                    w.u32(*end);
50116791×
                }
502
            }
503
        }
50416396×
        Some(w.0)
50516396×
    }
506
507
    /// The state an index holds (see [`State::write_index`]).
508238×
    pub fn read_index(bytes: &[u8]) -> Result<State, ObjectError> {
509238×
        let bad = || ObjectError::Schema("state index: truncated".into());
510238×
        let front_len =
511238×
            usize::try_from(crate::flat::Reader::new(bytes).u64()?).map_err(|_| bad())?;
512238×
        let front = bytes.get(8..8 + front_len).ok_or_else(bad)?;
513238×
        let histories = Histories::Bytes(Arc::new(bytes[8 + front_len..].to_vec()));
514238×
        Self::read_index_front(front, histories)
515238×
    }
516
517
    /// An index's front, everything but the files' histories, with where
518
    /// those are: each is read, and checked against the hash the front
519
    /// holds, only when its graph is built (oxygen#94).
5201965×
    pub fn read_index_front(bytes: &[u8], histories: Histories) -> Result<State, ObjectError> {
521
        use crate::flat::Reader;
5221965×
        let bad = |what: &str| ObjectError::Schema(format!("state index: {what}"));
5231965×
        let mut r = Reader::new(bytes);
5241965×
        let mut st = State::new_history_only();
525
        {
5261965×
            let _t = crate::timing::phase("index: changes");
5271965×
            st.changes = Self::read_index_changes(r.bytes()?)?;
528
        }
5291965×
        let tree_bytes = r.bytes()?;
5301965×
        for _ in 0..r.count()? {
5312×
            let n = NodeId(r.fixed::<16>()?);
5322×
            let mut recs = Vec::new();
5332×
            for _ in 0..r.count()? {
5342×
                recs.push((r.object()?, r.opt_object()?, r.opt_object()?));
535
            }
5362×
            st.replaces.insert(n, recs);
537
        }
5381965×
        let deferred_bytes = r.bytes()?;
5391965×
        if !r.at_end() {
5400×
            return Err(bad("trailing bytes"));
5411965×
        }
5421965×
        let (tree, deferred) = rayon::join(
5431965×
            || TreeGraph::read_flat(&mut Reader::new(tree_bytes)),
5441965×
            || Self::read_deferred(deferred_bytes, &histories),
545
        );
5461965×
        st.tree = tree?;
5471965×
        st.deferred = Some(deferred?);
5481965×
        Ok(st)
5491965×
    }
550
551
    /// The changes of a state index: its first block, which a reader can
552
    /// take without the rest.
5531965×
    pub fn read_index_changes(
5541965×
        bytes: &[u8],
5551965×
    ) -> Result<crate::fasthash::FastMap<ChangeHash, Arc<ChangeMeta>>, ObjectError> {
5561965×
        Ok(Self::read_index_changes_listed(bytes)?
5571965×
            .into_iter()
5581965×
            .map(|(h, m)| (h, Arc::new(m)))
5591965×
            .collect())
5601965×
    }
561
562
    /// The same changes as a list in the order the index keeps them, which
563
    /// a current index keeps in log order ([`log_order`]): what the log
564
    /// needs (oxygen#94).
5652134×
    pub fn read_index_changes_listed(
5662134×
        bytes: &[u8],
5672134×
    ) -> Result<Vec<(ChangeHash, ChangeMeta)>, ObjectError> {
568
        use crate::flat::Reader;
5692134×
        let mut r = Reader::new(bytes);
5702134×
        if r.fixed::<18>()?.as_slice() != Self::INDEX_MAGIC {
5710×
            return Err(ObjectError::Schema("state index: not a state index".into()));
5722134×
        }
5732134×
        let n = r.count()?;
5742134×
        let mut changes = Vec::with_capacity(n);
5752134×
        for _ in 0..n {
57610174×
            let h = r.object()?;
57710174×
            changes.push((h, ChangeMeta::read_flat(&mut r)?));
578
        }
5792133×
        if !r.at_end() {
5800×
            return Err(ObjectError::Schema("state index: trailing bytes".into()));
5812133×
        }
5822133×
        Ok(changes)
5832134×
    }
584
585
    /// The deferred block: each node with its graph key and its body,
586
    /// left encoded (see [`Deferred`]).
5871965×
    fn read_deferred(
5881965×
        table: &[u8],
5891965×
        histories: &Histories,
5901965×
    ) -> Result<crate::fasthash::FastMap<NodeId, Arc<Deferred>>, ObjectError> {
591
        use crate::flat::Reader;
5921965×
        let _t = crate::timing::phase("index: deferred");
5931965×
        let mut r = Reader::new(table);
5941965×
        let mut deferred = crate::fasthash::FastMap::default();
5951965×
        let n = r.count()?;
5961965×
        deferred.reserve(n);
5971965×
        for _ in 0..n {
59816736×
            let id = NodeId(r.fixed::<16>()?);
59916736×
            let key = r.fixed::<32>()?;
600
            // The length is of bytes in the histories, not in this table:
601
            // checked against the histories' size below.
60216736×
            let (at, len, sum) = (r.u64()?, r.u32()? as usize, r.fixed::<32>()?);
60316736×
            if at
60416736×
                .checked_add(len as u64)
60516736×
                .is_none_or(|end| end > histories.len())
606
            {
6070×
                return Err(ObjectError::Schema(
6080×
                    "state index: a file's history lies outside it".into(),
6090×
                ));
61016736×
            }
61116736×
            let d = Deferred {
61216736×
                raw: Some(RawHistory {
61316736×
                    from: histories.clone(),
61416736×
                    at,
61516736×
                    len,
61616736×
                    sum,
61716736×
                }),
61816736×
                key: std::sync::OnceLock::from(key),
61916736×
                ..Deferred::default()
62016736×
            };
62116736×
            deferred.insert(id, Arc::new(d));
622
        }
6231965×
        if !r.at_end() {
6240×
            return Err(ObjectError::Schema("state index: trailing bytes".into()));
6251965×
        }
6261965×
        Ok(deferred)
6271965×
    }
628
629
    /// One node's deferred history as the index encodes it.
6305476×
    fn read_deferred_body(bytes: &[u8]) -> Result<DeferredBody, ObjectError> {
631
        use crate::flat::Reader;
6325476×
        let bad = |what: &str| ObjectError::Schema(format!("state index: {what}"));
6335487×
        let node = |r: &mut Reader| -> Result<NodeId, ObjectError> { Ok(NodeId(r.fixed::<16>()?)) };
63410974×
        let atom = |r: &mut Reader| -> Result<AtomRef, ObjectError> {
63510974×
            Ok(match r.u8()? {
6365470×
                0 => AtomRef::Start,
6375487×
                1 => AtomRef::End,
638
                2 => AtomRef::Atom {
63917×
                    change: r.object()?,
64017×
                    index: r.u32()?,
641
                },
6420×
                _ => return Err(bad("atom")),
643
            })
64410974×
        };
6455476×
        let mut r = Reader::new(bytes);
6465476×
        let base = match r.u8()? {
6475470×
            0 => None,
6480×
            1 => Some(DeferredBase::Blob(r.object()?)),
6496×
            _ => Some(DeferredBase::Inline(cbor::decode(r.bytes()?)?)),
650
        };
6515476×
        let mut edits = Vec::new();
6525476×
        for _ in 0..r.count()? {
6535487×
            let c = r.object()?;
6545487×
            let mut es = Vec::new();
6555487×
            for _ in 0..r.count()? {
6565487×
                es.push(match r.u8()? {
657
                    0 => Edit::Insert {
6585487×
                        node: node(&mut r)?,
6595487×
                        after: atom(&mut r)?,
6605487×
                        before: atom(&mut r)?,
6615487×
                        lines: crate::lines::Lines::default(),
662
                    },
663
                    1 => {
6640×
                        let node = node(&mut r)?;
6650×
                        let mut atoms = Vec::new();
6660×
                        for _ in 0..r.count()? {
6670×
                            atoms.push(crate::objects::AtomRange {
6680×
                                change: r.object()?,
6690×
                                from: r.u32()?,
6700×
                                to: r.u32()?,
671
                            });
672
                        }
6730×
                        Edit::Delete { node, atoms }
674
                    }
6750×
                    _ => return Err(bad("edit")),
676
                });
677
            }
6785487×
            edits.push((c, es));
679
        }
6805476×
        let mut texts = Vec::new();
6815476×
        for _ in 0..r.count()? {
6825487×
            texts.push(match r.u8()? {
683
                0 => TextRef::Blob(
6840×
                    r.object()?,
6850×
                    r.u32()?,
6860×
                    match r.u8()? {
6870×
                        0 => crate::objects::TextForm::Cbor,
6880×
                        1 => crate::objects::TextForm::Raw,
6890×
                        _ => crate::objects::TextForm::CborFallback,
690
                    },
691
                ),
692
                _ => TextRef::Inline {
6935487×
                    change: r.object()?,
6945487×
                    start: r.u32()?,
6955487×
                    end: r.u32()?,
696
                },
697
            });
698
        }
6995476×
        if !r.at_end() {
7000×
            return Err(bad("trailing bytes in a file's history"));
7015476×
        }
7025476×
        Ok(DeferredBody { base, edits, texts })
7035476×
    }
704
}
705
706
impl Clone for State {
7071650×
    fn clone(&self) -> State {
7081650×
        State {
7091650×
            changes: self.changes.clone(),
7101650×
            tree: self.tree.clone(),
7111650×
            files: self.files.clone(),
7121650×
            replaces: self.replaces.clone(),
7131650×
            // A cache: the clone builds its own.
7141650×
            closure: Default::default(),
7151650×
            levels: Default::default(),
7161650×
            deferred: self.deferred.clone(),
7171650×
            unified_view: self.unified_view.clone(),
7181650×
        }
7191650×
    }
720
}
721
722
#[derive(Clone, Debug, PartialEq, Eq)]
723
pub enum FileContent {
724
    Text {
725
        bytes: Vec<u8>,
726
        materialized: Materialized,
727
    },
728
    /// Live blobs; more than one is a replace conflict (spec §9.3).
729
    Opaque {
730
        blobs: Vec<ObjectRef>,
731
    },
732
    Dir,
733
}
734
735
#[derive(Clone, Debug, PartialEq, Eq)]
736
pub struct FileState {
737
    pub node: NodeId,
738
    pub kind: NodeKind,
739
    pub mode: Mode,
740
    pub content: FileContent,
741
}
742
743
impl FileState {
74413689×
    pub fn is_conflicted(&self) -> bool {
74513689×
        match &self.content {
74612790×
            FileContent::Text { materialized, .. } => materialized.is_conflicted(),
747250×
            FileContent::Opaque { blobs } => blobs.len() > 1,
748649×
            FileContent::Dir => false,
749
        }
75013689×
    }
751
}
752
753
/// See `State::conflict_regions`.
754
#[derive(Clone, Debug, Default, PartialEq, Eq)]
755
pub struct ConflictRegions {
756
    pub plain: Vec<(AtomKey, Vec<u8>)>,
757
    pub region_atoms: Vec<AtomKey>,
758
    pub contributors: Vec<ChangeHash>,
759
}
760
761
/// The whole tree of a state, materialized.
762
#[derive(Clone, Debug, Default)]
763
pub struct Snapshot {
764
    pub tree: TreeView,
765
    pub files: BTreeMap<String, FileState>,
766
    /// Text files left unrendered by [`State::materialize_except`]: their
767
    /// `bytes` are empty and they are known not to be conflicted.
768
    pub unrendered: BTreeSet<String>,
769
}
770
771
impl Snapshot {
7722124×
    pub fn is_conflicted(&self) -> bool {
77313346×
        self.tree.is_conflicted() || self.files.values().any(|f| f.is_conflicted())
7742124×
    }
775
}
776
77763932×
fn vertex(r: AtomRef) -> Vertex {
77863932×
    match r {
77926885×
        AtomRef::Start => Vertex::Start,
78028136×
        AtomRef::End => Vertex::End,
7818911×
        AtomRef::Atom { change, index } => Vertex::Atom((change, index)),
782
    }
78363932×
}
784
785
/// Replace self references in a tree op with the concrete change hash.
78639295×
pub fn resolve_tree_op(op: &TreeOp, me: ChangeHash) -> TreeOp {
78739295×
    match op {
788924×
        TreeOp::RemoveNode { node, placement } => TreeOp::RemoveNode {
789924×
            node: *node,
790924×
            placement: placement.resolve(me),
791924×
        },
792
        TreeOp::Move {
793154×
            node,
794154×
            old,
795154×
            parent,
796154×
            name,
797154×
        } => TreeOp::Move {
798154×
            node: *node,
799154×
            old: old.resolve(me),
800154×
            parent: *parent,
801154×
            name: name.clone(),
802154×
        },
80322×
        TreeOp::SetMode { node, old, mode } => TreeOp::SetMode {
80422×
            node: *node,
80522×
            old: old.resolve(me),
80622×
            mode: *mode,
80722×
        },
80838195×
        add @ TreeOp::AddNode { .. } => add.clone(),
809
    }
81039295×
}
811
812
/// Apply one change's edits of one file (inserts and deletes; replaces
813
/// live beside the graphs) to that file's graph.
81426432×
fn apply_file_edits(g: &mut FileGraph, h: ChangeHash, edits: Vec<Edit>) {
815
    // The graph's kept order, if any, follows the edits (see `Order`).
81626432×
    let order = g.take_order();
81726432×
    let mut inserts: Vec<(Vertex, Vertex, u32, u32)> = Vec::new();
81826432×
    let mut deleted: Vec<crate::graph::AtomKey> = Vec::new();
81926432×
    let mut next = 0u32;
82028655×
    for e in edits {
82128655×
        match e {
822
            Edit::Insert {
82327291×
                after,
82427291×
                before,
82527291×
                lines,
826
                ..
82727291×
            } => {
82827291×
                let first = next;
82927291×
                let count = lines.len() as u32;
83027291×
                next += count;
83127291×
                let (after, before) = (vertex(after.resolve(h)), vertex(before.resolve(h)));
83227291×
                inserts.push((after, before, first, count));
83327291×
                g.insert_owned(h, first, after, before, lines);
83427291×
            }
8351364×
            Edit::Delete { atoms, .. } => {
8362444×
                for r in atoms {
8372444×
                    let r = r.resolve(h);
8382503×
                    deleted.extend((r.from..r.to).map(|i| (r.change, i)));
8392444×
                    g.delete(h, &r);
840
                }
841
            }
8420×
            Edit::Replace { .. } => {}
843
        }
844
    }
84526432×
    if let Some(o) = order {
84662×
        g.keep_spliced(o, h, &inserts, &deleted);
84726370×
    }
84826432×
}
849
850
/// Build the graphs of deferred entries: their checkpoint bases and the
851
/// texts of inserts read without their lines are loaded in one call, then
852
/// the graphs are built in parallel (they are independent).
8534130×
fn build_deferred(
8544130×
    taken: Vec<(NodeId, Deferred)>,
8554130×
    load: impl FnOnce(&[ObjectRef]) -> Result<Vec<Vec<u8>>, ObjectError>,
8564130×
    read_change: impl Fn(&ChangeHash) -> Result<Arc<Vec<u8>>, ObjectError>,
8574130×
) -> Result<Vec<(NodeId, FileGraph)>, ObjectError> {
8584130×
    let mut taken: Vec<(NodeId, DeferredBody)> = taken
8594130×
        .into_iter()
86023433×
        .map(|(n, d)| Ok((n, d.into_body()?)))
8614130×
        .collect::<Result<_, ObjectError>>()?;
8624129×
    let mut refs: Vec<ObjectRef> = taken
8634129×
        .iter()
8644129×
        .filter_map(|(_, d)| match &d.base {
8656×
            Some(DeferredBase::Blob(r)) => Some(*r),
86623426×
            _ => None,
86723432×
        })
8684129×
        .collect();
869
    // Text blobs of inserts read without their lines, after the bases.
8704129×
    let bases = refs.len();
87123432×
    refs.extend(taken.iter().flat_map(|(_, d)| {
87226623×
        d.texts.iter().filter_map(|t| match t {
87370×
            TextRef::Blob(r, _, _) => Some(*r),
87426553×
            TextRef::Inline { .. } => None,
87526623×
        })
87623432×
    }));
877
    // And the encodings of changes whose inline inserts are needed,
878
    // each read once.
8794129×
    let reading = crate::timing::phase("graphs: read changes and blobs");
8804129×
    let mut changes: HashMap<ChangeHash, Arc<Vec<u8>>> = HashMap::new();
88126623×
    for t in taken.iter().flat_map(|(_, d)| d.texts.iter()) {
88226623×
        if let TextRef::Inline { change, .. } = t {
88326553×
            if !changes.contains_key(change) {
8846665×
                changes.insert(*change, read_change(change)?);
88519888×
            }
88670×
        }
887
    }
8884129×
    let n_changes = changes.len();
8894129×
    drop(reading);
8904129×
    let reading = crate::timing::phase("graphs:   of which blobs");
8914129×
    let mut blobs = if refs.is_empty() {
8924058×
        Vec::new()
893
    } else {
89471×
        load(&refs)?
895
    };
8964126×
    if crate::timing::enabled() {
8972×
        eprintln!(
8982×
            "timing graphs: {} graphs, {n_changes} changes, {} blobs",
8992×
            taken.len(),
9002×
            refs.len()
9012×
        );
9024124×
    }
9034126×
    if blobs.len() != refs.len() {
9040×
        return Err(ObjectError::Schema("graph blobs: wrong count".into()));
9054126×
    }
9064126×
    drop(reading);
9074126×
    let lines = crate::timing::phase("graphs: hand out blobs");
9084126×
    let mut texts = blobs.split_off(bases).into_iter();
9094126×
    let mut blobs = blobs.into_iter();
910
    // Hand each graph its own blobs, in order, then fill in its inserts'
911
    // lines and build it, all in parallel: graphs are independent.
912
    type Work = (NodeId, DeferredBody, Option<Vec<u8>>, Vec<Vec<u8>>);
9134126×
    let mut work: Vec<Work> = Vec::with_capacity(taken.len());
91423429×
    for (node, d) in taken.drain(..) {
91523429×
        let wanted = d
91623429×
            .texts
91723429×
            .iter()
91826621×
            .filter(|t| matches!(t, TextRef::Blob(..)))
91923429×
            .count();
92023429×
        let mine: Vec<Vec<u8>> = texts.by_ref().take(wanted).collect();
92123429×
        if mine.len() != wanted {
9220×
            return Err(ObjectError::Schema("text blobs: wrong count".into()));
92323429×
        }
92423429×
        let bytes = match &d.base {
925
            Some(DeferredBase::Blob(_)) => Some(
9265×
                blobs
9275×
                    .next()
9285×
                    .ok_or_else(|| ObjectError::Schema("graph blobs: wrong count".into()))?,
929
            ),
93023424×
            _ => None,
931
        };
93223429×
        work.push((node, d, bytes, mine));
933
    }
9344126×
    drop(lines);
9354126×
    let _t = crate::timing::phase("graphs: lines and build");
936
    use rayon::prelude::*;
9374126×
    let changes = &changes;
9384126×
    let built: Vec<(NodeId, FileGraph)> = work
9394126×
        .into_par_iter()
94023429×
        .map(|(node, mut d, bytes, mine)| {
941
            // Fill in the lines of inserts read without them, in order.
94223429×
            let mut want = std::mem::take(&mut d.texts).into_iter();
94323429×
            let mut mine = mine.into_iter();
94425969×
            for (_, edits) in d.edits.iter_mut() {
94527749×
                for e in edits.iter_mut() {
94627749×
                    if let Edit::Insert { lines, .. } = e {
94726628×
                        if lines.is_empty() {
94826621×
                            *lines = match want.next() {
94968×
                                Some(TextRef::Blob(_, count, form)) => {
95068×
                                    let content = mine.next().ok_or_else(|| {
9510×
                                        ObjectError::Schema("text blobs: wrong count".into())
9520×
                                    })?;
95368×
                                    crate::objects::text_lines(&content, count, form)?
954
                                }
95526553×
                                Some(TextRef::Inline { change, start, end }) => {
95626553×
                                    let bytes = &changes[&change];
95726553×
                                    let range = bytes
95826553×
                                        .get(start as usize..end as usize)
95926553×
                                        .ok_or_else(|| {
9600×
                                            ObjectError::Schema(
9610×
                                                "an inline insert lies outside its change".into(),
9620×
                                            )
9630×
                                        })?;
96426553×
                                    crate::objects::inline_lines(range)?
965
                                }
966
                                None => {
9670×
                                    return Err(ObjectError::Schema(
9680×
                                        "an insert's text is missing".into(),
9690×
                                    ))
970
                                }
971
                            };
9727×
                        }
9731121×
                    }
974
                }
975
            }
97623429×
            let mut graph = match (d.base.take(), bytes) {
97796×
                (Some(DeferredBase::Inline(v)), _) => FileGraph::from_value(&v)?,
9785×
                (Some(DeferredBase::Blob(_)), Some(bytes)) => {
9795×
                    FileGraph::from_value(&cbor::decode(&bytes)?)?
980
                }
98123328×
                _ => FileGraph::default(),
982
            };
98325969×
            for (h, edits) in std::mem::take(&mut d.edits) {
98425969×
                apply_file_edits(&mut graph, h, edits);
98525969×
            }
98623429×
            Ok((node, graph))
98723429×
        })
9884126×
        .collect::<Result<_, ObjectError>>()?;
9894126×
    Ok(built)
9904130×
}
991
992
impl State {
993721×
    pub fn new() -> Self {
994721×
        Self::default()
995721×
    }
996
997
    /// An empty state for history only: changes and tree are kept, file
998
    /// edits are recorded but no file graph is built until
999
    /// [`State::load_graphs`] asks for it. For log and other commands that
1000
    /// need no file contents, or only a few files.
10013824×
    pub fn new_history_only() -> Self {
10023824×
        State {
10033824×
            deferred: Some(Default::default()),
10043824×
            ..Self::default()
10053824×
        }
10063824×
    }
1007
1008
    /// Whether `node`'s file graph is built (always, in a full state).
100940×
    pub fn is_built(&self, node: &NodeId) -> bool {
101040×
        self.deferred.is_none() || self.files.contains_key(node)
101140×
    }
1012
1013
    /// Whether this state builds file graphs only when asked
1014
    /// ([`State::new_history_only`]), even if it has none waiting.
10153590×
    pub fn is_history_only(&self) -> bool {
10163590×
        self.deferred.is_some()
10173590×
    }
1018
1019
    /// Whether this state holds every file graph.
10203000×
    pub fn is_full(&self) -> bool {
10213000×
        self.deferred.as_ref().is_none_or(|d| d.is_empty())
10223000×
    }
1023
1024
    /// The graph of `node`, which must be loaded: asking a history-only
1025
    /// state for a graph it has not built is a bug, never an empty file.
102626463×
    fn graph(&self, node: &NodeId) -> Option<&FileGraph> {
102726463×
        if let Some(d) = &self.deferred {
102810321×
            assert!(
102910321×
                !d.contains_key(node),
1030
                "file graph {} of a history-only state was not loaded (State::load_graphs)",
10311×
                node.short(12)
1032
            );
103316142×
        }
103426462×
        self.files.get(node).map(|g| g.as_ref())
103526462×
    }
1036
1037
    /// Operations over every file need every graph.
10382993×
    fn require_full(&self, what: &str) {
10392993×
        assert!(
10402993×
            self.is_full(),
1041
            "{what} needs every file graph; this state was loaded for history only"
1042
        );
10432992×
    }
1044
1045
    /// Build the graphs of `nodes` in a history-only state: from their
1046
    /// checkpoint base (blobs read in one batch through `load`) and their
1047
    /// deferred edits. Nodes already built, or with no graph, are skipped.
10484053×
    pub fn load_graphs(
10494053×
        &mut self,
10504053×
        nodes: &[NodeId],
10514053×
        load: impl FnOnce(&[ObjectRef]) -> Result<Vec<Vec<u8>>, ObjectError>,
10524053×
        read_change: impl Fn(&ChangeHash) -> Result<Arc<Vec<u8>>, ObjectError>,
10534053×
    ) -> Result<(), ObjectError> {
10544053×
        let Some(deferred) = self.deferred.as_mut() else {
10550×
            return Ok(());
1056
        };
10574053×
        let taken: Vec<(NodeId, Deferred)> = nodes
10584053×
            .iter()
105923680×
            .filter_map(|n| deferred.remove(n).map(|d| (*n, Arc::unwrap_or_clone(d))))
10604053×
            .collect();
10614053×
        let built = build_deferred(taken, load, read_change)?;
10624049×
        self.files
106323368×
            .extend(built.into_iter().map(|(n, g)| (n, Arc::new(g))));
10644049×
        Ok(())
10654053×
    }
1066
1067
    /// The graphs of `nodes`, built from this history-only state's deferred
1068
    /// edits without building them in it: the state stays history only, so
1069
    /// changes read without their lines can still be added to it and every
1070
    /// graph keeps its key. For a caller that renders a few files of many
1071
    /// states in turn (verifying an import).
107277×
    pub fn build_graphs_detached(
107377×
        &self,
107477×
        nodes: &[NodeId],
107577×
        load: impl FnOnce(&[ObjectRef]) -> Result<Vec<Vec<u8>>, ObjectError>,
107677×
        read_change: impl Fn(&ChangeHash) -> Result<Arc<Vec<u8>>, ObjectError>,
107777×
    ) -> Result<Vec<(NodeId, FileGraph)>, ObjectError> {
107877×
        let Some(deferred) = self.deferred.as_ref() else {
10790×
            return Ok(Vec::new());
1080
        };
1081
        // A file this state has built has no deferred history: its graph
1082
        // is the built one, copied. Built from an empty history instead, it
1083
        // rendered as an empty file, and a merge taking this path recorded
1084
        // the whole file as inserted again, conflicting with itself (#71).
108577×
        let mut built: Vec<(NodeId, FileGraph)> = Vec::new();
108677×
        let mut copies: Vec<(NodeId, Deferred)> = Vec::new();
108777×
        for n in nodes {
108864×
            match (self.files.get(n), deferred.get(n)) {
10893×
                (Some(g), _) => built.push((*n, (**g).clone())),
109061×
                (None, d) => copies.push((*n, d.map(|d| (**d).clone()).unwrap_or_default())),
1091
            }
1092
        }
109377×
        let mut out = build_deferred(copies, load, read_change)?;
109477×
        out.extend(built);
109577×
        Ok(out)
109677×
    }
1097
1098
    /// Render a graph built for this state (see
1099
    /// [`State::build_graphs_detached`]), as [`State::render_file`] does.
110064×
    pub fn render_graph(&self, g: &FileGraph) -> (Vec<u8>, Materialized) {
110164×
        let m = g.materialize(self);
110264×
        let bytes = g.render(&m, &|h| self.label(h));
110364×
        (bytes, m)
110464×
    }
1105
1106
    /// Every file graph a history-only state has not built, deleted files'
1107
    /// included.
110877×
    pub fn unbuilt_nodes(&self) -> Vec<NodeId> {
110977×
        self.deferred
111077×
            .as_ref()
111177×
            .map(|d| d.keys().copied().collect())
111277×
            .unwrap_or_default()
111377×
    }
1114
1115
    /// Drop the built graphs of `nodes`, for a caller rendering a large
1116
    /// tree a batch at a time. The state can no longer render those files,
1117
    /// so it must not be used for them again.
1118452×
    pub fn drop_graphs(&mut self, nodes: &[NodeId]) {
11196018×
        for n in nodes {
11206018×
            self.files.remove(n);
11216018×
        }
1122452×
    }
1123
1124
    /// Build every graph a history-only state has not built, making it full.
11252060×
    pub fn load_all_graphs(
11262060×
        &mut self,
11272060×
        load: impl FnOnce(&[ObjectRef]) -> Result<Vec<Vec<u8>>, ObjectError>,
11282060×
        read_change: impl Fn(&ChangeHash) -> Result<Arc<Vec<u8>>, ObjectError>,
11292060×
    ) -> Result<(), ObjectError> {
11302060×
        let nodes: Vec<NodeId> = match &self.deferred {
11312060×
            Some(d) => d.keys().copied().collect(),
11320×
            None => return Ok(()),
1133
        };
11342060×
        self.load_graphs(&nodes, load, read_change)?;
11352060×
        self.deferred = None;
11362060×
        Ok(())
11372060×
    }
1138
1139
    /// The one text a file's history is, when it is a single insert of
1140
    /// the whole file into nothing (a file added once and never edited,
1141
    /// with no checkpoint base): its rendering is that text, so it needs
1142
    /// no graph. `None` otherwise, or when the graph is built.
11435871×
    pub fn single_text(&self, node: &NodeId) -> Option<TextRef> {
11445871×
        if self.files.contains_key(node) {
114576×
            return None;
11465795×
        }
1147
        // A history that cannot be read is no single text: building its
1148
        // graph reports why.
11495795×
        let d = self.deferred.as_ref()?.get(node)?.body().ok()?;
11505751×
        let [(_, edits)] = d.edits.as_slice() else {
1151226×
            return None;
1152
        };
1153
        let [Edit::Insert {
1154
            after: AtomRef::Start,
1155
            before: AtomRef::End,
11565525×
            lines,
1157
            ..
11585525×
        }] = edits.as_slice()
1159
        else {
11600×
            return None;
1161
        };
11625525×
        match (&d.base, d.texts.as_slice()) {
11635525×
            (None, [t]) if lines.is_empty() => Some(*t),
11640×
            _ => None,
1165
        }
11665871×
    }
1167
1168
    /// A key for the graph of `node` in a history-only state that has not
1169
    /// built it: equal keys mean equal graphs, so equal renderings (a
1170
    /// rendering depends only on the graph and on the changes' own
1171
    /// immutable metadata). It names the node, its checkpoint base (a blob,
1172
    /// an inline value, or none) and the set of later changes that edited
1173
    /// the file.
1174
    /// `None` for a full state or a graph already built.
117528785×
    pub fn graph_key(&self, node: &NodeId) -> Option<[u8; 32]> {
117628785×
        let deferred = self.deferred.as_ref()?;
117728748×
        if self.files.contains_key(node) {
1178581×
            return None;
117928167×
        }
118028167×
        let mut h = blake3::Hasher::new();
118128167×
        h.update(b"oxy graph key 1\0");
1182
        // A change's edits are applied per node: the node is part of it.
118328167×
        h.update(&node.0);
118428167×
        let Some(d) = deferred.get(node) else {
1185
            // Never edited: the empty graph.
1186162×
            h.update(&[0]);
1187162×
            return Some(*h.finalize().as_bytes());
1188
        };
1189
        // A history that cannot be read has no key: the file is read.
119028005×
        if let Some(k) = d.key.get() {
119122181×
            return Some(*k);
11925824×
        }
11935824×
        let k = d.graph_key(h).ok()?;
11945824×
        Some(*d.key.get_or_init(|| k))
119528785×
    }
1196
11973678×
    pub fn len(&self) -> usize {
11983678×
        self.changes.len()
11993678×
    }
1200
1201
    /// Turn this history-only state's built graphs back into deferred
1202
    /// histories, each starting from its graph encoded as a checkpoint
1203
    /// graph and stored by `store` (it returns the blob's reference): what
1204
    /// they held stays, and memory goes back to what a history-only state
1205
    /// costs. A graph built later again decodes that blob and applies only
1206
    /// the edits that came after. Graph keys change (the base is now a
1207
    /// blob), which only means a cached rendering is not reused.
12081×
    pub fn shed_graphs<E>(
12091×
        &mut self,
12101×
        mut store: impl FnMut(&[u8]) -> Result<ObjectRef, E>,
12111×
    ) -> Result<usize, E> {
1212
        use rayon::prelude::*;
12131×
        let Some(deferred) = self.deferred.as_mut() else {
12140×
            return Ok(0);
1215
        };
12161×
        let graphs: Vec<(NodeId, Arc<FileGraph>)> = self.files.drain().collect();
12171×
        let encoded: Vec<(NodeId, Vec<u8>)> = graphs
12181×
            .par_iter()
12192×
            .map(|(n, g)| (*n, cbor::encode(&g.to_value())))
12201×
            .collect();
12211×
        drop(graphs);
12221×
        let shed = encoded.len();
12232×
        for (node, bytes) in encoded {
12242×
            let blob = store(&bytes)?;
12252×
            deferred.insert(
12262×
                node,
12272×
                Arc::new(Deferred::from_body(DeferredBody {
12282×
                    base: Some(DeferredBase::Blob(blob)),
12292×
                    ..DeferredBody::default()
12302×
                })),
1231
            );
1232
        }
12331×
        Ok(shed)
12341×
    }
1235
1236
    /// How many file graphs this state has built.
123777×
    pub fn built_count(&self) -> usize {
123877×
        self.files.len()
123977×
    }
1240
1241
    /// Atoms in this state's built graphs, live and deleted: what their
1242
    /// memory follows, since a graph keeps every line its file ever had.
124351×
    pub fn built_atoms(&self) -> usize {
124456×
        self.files.values().map(|g| g.atom_count()).sum()
124551×
    }
1246
1247
    /// Edits held in the deferred histories of files not built, and the
1248
    /// most any one file holds (for measuring, #25).
12490×
    pub fn deferred_edit_count(&self) -> (usize, usize) {
12500×
        let Some(d) = self.deferred.as_ref() else {
12510×
            return (0, 0);
1252
        };
12530×
        let mut total = 0;
12540×
        let mut most = 0;
12550×
        for v in d.values() {
12560×
            let n: usize = v
12570×
                .body()
12580×
                .map_or(0, |b| b.edits.iter().map(|(_, e)| e.len()).sum());
12590×
            total += n;
12600×
            most = most.max(n);
1261
        }
12620×
        (total, most)
12630×
    }
1264
}
1265
1266
impl Deferred {
1267
    /// See [`State::graph_key`]: `h` has taken the node already.
126822220×
    fn graph_key(&self, mut h: blake3::Hasher) -> Result<[u8; 32], ObjectError> {
126922220×
        let d = self.body()?;
127013×
        match &d.base {
127122207×
            None => h.update(&[0]),
12720×
            Some(DeferredBase::Blob(r)) => h.update(&[1]).update(&r.0),
127313×
            Some(DeferredBase::Inline(v)) => h.update(&[2]).update(&cbor::encode(v)),
1274
        };
127522220×
        let mut changes: Vec<&ChangeHash> = d.edits.iter().map(|(c, _)| c).collect();
127622220×
        changes.sort();
127722220×
        changes.dedup();
127822220×
        h.update(&(changes.len() as u64).to_le_bytes());
127923591×
        for c in changes {
128023591×
            h.update(&c.0);
128123591×
        }
128222220×
        Ok(*h.finalize().as_bytes())
128322220×
    }
1284
}
1285
1286
impl State {
12870×
    pub fn is_empty(&self) -> bool {
12880×
        self.changes.is_empty()
12890×
    }
129014270×
    pub fn contains(&self, h: &ChangeHash) -> bool {
129114270×
        self.changes.contains_key(h)
129214270×
    }
129367×
    pub fn get(&self, h: &ChangeHash) -> Option<&ChangeMeta> {
129467×
        self.changes.get(h).map(|m| &**m)
129567×
    }
12961647×
    pub fn changes(&self) -> impl Iterator<Item = (&ChangeHash, &ChangeMeta)> {
12974008×
        self.changes.iter().map(|(h, m)| (h, &**m))
12981647×
    }
129915529×
    pub fn tree(&self) -> &TreeGraph {
130015529×
        &self.tree
130115529×
    }
1302
1303
    /// The tree as every reader sees it: the tree graph's view, with the
1304
    /// same path added on several lines of history with identical content
1305
    /// shown as one entry rather than a name conflict (spec §9.2). The
1306
    /// text files among such candidates must be built in a history-only
1307
    /// state ([`State::unify_candidates`]); until they are, they are shown
1308
    /// as the conflict.
13096601×
    pub fn view(&self) -> TreeView {
13106601×
        if let Some(v) = self.unified_view.get() {
131121×
            return v.clone();
13126580×
        }
13136580×
        let raw = self.tree.view();
13146580×
        if !raw
13156580×
            .conflicts
13166580×
            .iter()
13176580×
            .any(|c| matches!(c, crate::tree::TreeConflict::Name { .. }))
1318
        {
13196175×
            return raw;
1320405×
        }
1321408×
        let (view, decided) = self.tree.unify(raw, |es| self.same_content(es));
1322405×
        if decided {
1323405×
            let _ = self.unified_view.set(view.clone());
1324405×
        }
1325405×
        view
13266601×
    }
1327
1328
    /// The text files [`State::view`] must compare and a history-only
1329
    /// state has not built: candidates of the tree's name conflicts.
13302163×
    pub fn unify_candidates(&self) -> Vec<NodeId> {
13312163×
        let Some(deferred) = &self.deferred else {
13320×
            return Vec::new();
1333
        };
13342163×
        if self.unified_view.get().is_some() {
13350×
            return Vec::new();
13362163×
        }
13372163×
        let raw = self.tree.view();
13382163×
        let mut out = Vec::new();
13392163×
        for c in &raw.conflicts {
134086×
            if let crate::tree::TreeConflict::Name { nodes, .. } = c {
1341182×
                out.extend(nodes.iter().filter(|n| deferred.contains_key(*n)));
13420×
            }
1343
        }
13442163×
        out
13452163×
    }
1346
1347
    /// Whether the entries' contents are identical, unconflicted: the same
1348
    /// rendering for text files, the same single blob for opaque ones,
1349
    /// symlinks and subrepos. `None` for a text file not built yet.
1350408×
    fn same_content(&self, entries: &[&crate::tree::TreeEntry]) -> Option<bool> {
1351816×
        let text = |e: &crate::tree::TreeEntry| e.kind == NodeKind::File && !e.mode.opaque;
1352408×
        let mut first: Option<Vec<u8>> = None;
1353408×
        let mut first_blob: Option<ObjectRef> = None;
1354816×
        for e in entries {
1355816×
            if text(e) {
1356802×
                if self
1357802×
                    .deferred
1358802×
                    .as_ref()
1359802×
                    .is_some_and(|d| d.contains_key(&e.node))
1360
                {
13610×
                    return None;
1362802×
                }
1363802×
                let (bytes, m) = self.render_file(&e.node);
1364802×
                if m.is_conflicted() {
13650×
                    return Some(false);
1366802×
                }
1367401×
                match &first {
1368401×
                    None => first = Some(bytes),
1369401×
                    Some(b) if *b == bytes => {}
1370262×
                    Some(_) => return Some(false),
1371
                }
1372
            } else {
137314×
                let [blob] = self.live_blobs(&e.node)[..] else {
13740×
                    return Some(false);
1375
                };
13767×
                match first_blob {
13777×
                    None => first_blob = Some(blob),
13787×
                    Some(b) if b == blob => {}
13790×
                    Some(_) => return Some(false),
1380
                }
1381
            }
1382
        }
1383146×
        Some(true)
1384408×
    }
1385569×
    pub fn file(&self, node: &NodeId) -> Option<&FileGraph> {
1386569×
        self.graph(node)
1387569×
    }
1388
1389
    /// Add a change whose dependencies are already present.
13902824×
    pub fn add(&mut self, change: Change) -> Result<ChangeHash, StateError> {
13912824×
        let h = change.hash();
13922824×
        self.add_hashed(h, change)
13932824×
    }
1394
1395
    /// [`State::add`] for a change whose hash `h` is already known (it was
1396
    /// just stored), so it is not encoded again.
13973526×
    pub fn add_hashed(&mut self, h: ChangeHash, change: Change) -> Result<ChangeHash, StateError> {
13983526×
        if self.changes.contains_key(&h) {
139938×
            return Ok(h);
14003488×
        }
14014918×
        for d in &change.deps {
14024918×
            if !self.changes.contains_key(d) {
14032×
                return Err(StateError::MissingDep(*d, h));
14044916×
            }
1405
        }
14063486×
        self.add_known(h, change, Vec::new());
14073486×
        Ok(h)
14083526×
    }
1409
1410
    /// Add a change in any order; call `validate` once all are in.
1411960×
    pub fn add_unchecked(&mut self, change: Change) -> ChangeHash {
1412960×
        let h = change.hash();
1413960×
        self.add_known(h, change, Vec::new());
1414960×
        h
1415960×
    }
1416
1417
    /// Add a change just stored under hash `h` to a history-only state that
1418
    /// has built some graphs: an edit of a built graph is applied whole
1419
    /// (from `full`), and one of a file not built waits without its text
1420
    /// (from `skeleton` and its `texts`, as the store reads it back), so the
1421
    /// state holds no text it has not built. A Git import adds each change
1422
    /// it makes this way, keeping memory to what changes.
142392×
    pub fn add_known_mixed(
142492×
        &mut self,
142592×
        h: ChangeHash,
142692×
        full: Change,
142792×
        skeleton: Change,
142892×
        texts: Vec<TextRef>,
142992×
    ) {
143092×
        if self.deferred.is_none() || self.changes.contains_key(&h) {
14310×
            self.add_known(h, full, Vec::new());
14320×
            return;
143392×
        }
1434
        let Change {
143592×
            version,
143692×
            change_id,
143792×
            deps,
143892×
            author,
143992×
            hlc,
144092×
            message,
144192×
            tree_ops,
144292×
            edits: full_edits,
144392×
            kind,
144492×
        } = full;
144592×
        let mut texts = texts.into_iter();
144692×
        let mut edits = Vec::with_capacity(full_edits.len());
144792×
        let mut kept = Vec::new();
144897×
        for (f, s) in full_edits.into_iter().zip(skeleton.edits) {
144997×
            let text = match &s {
145077×
                Edit::Insert { lines, .. } if lines.is_empty() => texts.next(),
145120×
                _ => None,
1452
            };
145397×
            if self.files.contains_key(&f.node()) {
145445×
                edits.push(f);
145552×
            } else {
145652×
                edits.push(s);
145752×
                kept.extend(text);
145852×
            }
1459
        }
146092×
        let c = Change {
146192×
            version,
146292×
            change_id,
146392×
            deps,
146492×
            author,
146592×
            hlc,
146692×
            message,
146792×
            tree_ops,
146892×
            edits,
146992×
            kind,
147092×
        };
147192×
        self.add_known(h, c, kept);
147292×
    }
1473
1474
    /// Add a change read from the store under hash `h`, without hashing it
1475
    /// again (for a version 2 change that means chunking all its text).
1476
    /// `texts` names, in order, the inserts left without their lines (see
1477
    /// [`crate::store::Store::get_change_lazy`]); only a history-only state
1478
    /// takes them, and only for graphs it has not built.
147912985×
    pub fn add_known(&mut self, h: ChangeHash, change: Change, texts: Vec<TextRef>) {
148012985×
        if self.changes.contains_key(&h) {
14810×
            return;
148212985×
        }
148312985×
        let meta = ChangeMeta::of(&change);
148412985×
        self.unified_view = Default::default();
148512985×
        self.apply_with(h, change, texts);
148612985×
        self.changes.insert(h, Arc::new(meta));
148712985×
        {
148812985×
            let mut m = self.closure.write().unwrap();
148912985×
            m.sets.clear();
149012985×
            m.held = 0;
149112985×
        }
149212985×
    }
1493
1494
    // ------------------------------------------------------------ checkpoints (spec §11)
1495
1496
    /// Serialize the whole state in payload version 1: change metadata,
1497
    /// tree, file graphs, replaces, all in one value (spec §11).
1498380×
    pub fn checkpoint_payload(&self) -> Vec<u8> {
1499380×
        self.require_full("a checkpoint");
1500380×
        let mut nodes: Vec<&NodeId> = self.files.keys().collect();
1501380×
        nodes.sort();
1502380×
        let files = Value::Array(
1503380×
            nodes
1504380×
                .into_iter()
15051539×
                .map(|n| Value::Array(vec![Value::Bytes(n.0.to_vec()), self.files[n].to_value()]))
1506380×
                .collect(),
1507
        );
1508380×
        cbor::encode(&self.payload_value(o::FORMAT_VERSION, files))
1509380×
    }
1510
1511
    /// Each file graph encoded on its own, by node, for a version 2 payload
1512
    /// (spec §11): the graphs of large files become blobs of their own.
151318×
    pub fn checkpoint_file_graphs(&self) -> Vec<(NodeId, Vec<u8>)> {
1514
        use rayon::prelude::*;
151518×
        self.require_full("a checkpoint");
151618×
        let mut nodes: Vec<(&NodeId, &FileGraph)> =
151753×
            self.files.iter().map(|(n, g)| (n, g.as_ref())).collect();
151818×
        nodes.sort_by_key(|(n, _)| **n);
151918×
        nodes
152018×
            .into_par_iter()
152153×
            .map(|(n, g)| (*n, cbor::encode(&g.to_value())))
152218×
            .collect()
152318×
    }
1524
1525
    /// The payload version 2 index for the graphs of
1526
    /// [`State::checkpoint_file_graphs`]: a graph encoded in fewer than
1527
    /// [`INLINE_GRAPH_LIMIT`] bytes is kept inline, a larger one is named by
1528
    /// `blob_of` (the ObjectRef of the blob holding its encoding).
152918×
    pub fn checkpoint_index(
153018×
        &self,
153118×
        graphs: &[(NodeId, Vec<u8>)],
153218×
        mut blob_of: impl FnMut(&[u8]) -> ObjectRef,
153318×
    ) -> Vec<u8> {
153418×
        let files = Value::Array(
153518×
            graphs
153618×
                .iter()
153753×
                .map(|(n, enc)| {
153853×
                    let entry = if enc.len() < INLINE_GRAPH_LIMIT {
153946×
                        cbor::decode(enc).expect("an encoded graph decodes")
1540
                    } else {
15417×
                        Value::Bytes(blob_of(enc).0.to_vec())
1542
                    };
154353×
                    Value::Array(vec![Value::Bytes(n.0.to_vec()), entry])
154453×
                })
154518×
                .collect(),
1546
        );
154718×
        cbor::encode(&self.payload_value(CHECKPOINT_INDEX_VERSION, files))
154818×
    }
1549
1550
    /// The payload map around `files`: changes, tree and replaces as in version 1.
1551398×
    fn payload_value(&self, version: u64, files: Value) -> Value {
1552398×
        let mut hashes: Vec<&ChangeHash> = self.changes.keys().collect();
1553398×
        hashes.sort();
1554398×
        let changes = Value::Array(
1555398×
            hashes
1556398×
                .into_iter()
15572176×
                .map(|h| {
15582176×
                    let m = &self.changes[h];
15592176×
                    Value::Array(vec![
15602176×
                        Value::Bytes(h.0.to_vec()),
15612176×
                        Value::Bytes(m.change_id.0.to_vec()),
15622176×
                        Value::Array(m.deps.iter().map(|d| Value::Bytes(d.0.to_vec())).collect()),
15632176×
                        Value::Text(m.message.clone()),
15642176×
                        Value::Uint(m.hlc.wall_ms),
15652176×
                        Value::Uint(m.hlc.logical as u64),
15662176×
                        Value::Uint(m.kind.tag()),
15672176×
                        Value::Bytes(m.author.0.to_vec()),
1568
                    ])
15692176×
                })
1570398×
                .collect(),
1571
        );
1572398×
        let mut rnodes: Vec<&NodeId> = self.replaces.keys().collect();
1573398×
        rnodes.sort();
1574398×
        let opt =
157510×
            |o: &Option<ObjectRef>| o.map(|r| Value::Bytes(r.0.to_vec())).unwrap_or(Value::Null);
1576398×
        let replaces = Value::Array(
1577398×
            rnodes
1578398×
                .into_iter()
1579398×
                .map(|n| {
15805×
                    Value::Array(vec![
15815×
                        Value::Bytes(n.0.to_vec()),
1582
                        Value::Array(
15835×
                            self.replaces[n]
15845×
                                .iter()
15855×
                                .map(|(c, o, nw)| {
15865×
                                    Value::Array(vec![Value::Bytes(c.0.to_vec()), opt(o), opt(nw)])
15875×
                                })
15885×
                                .collect(),
1589
                        ),
1590
                    ])
15915×
                })
1592398×
                .collect(),
1593
        );
1594398×
        Value::map([
1595398×
            (0, Value::Uint(version)),
1596398×
            (1, changes),
1597398×
            (2, self.tree.to_value()),
1598398×
            (3, files),
1599398×
            (4, replaces),
1600398×
        ])
1601398×
    }
1602
1603
    /// Decode a version 1 payload (the whole state in one value). A
1604
    /// version 2 payload needs its graph blobs: use [`State::from_checkpoint`].
160537×
    pub fn from_checkpoint_payload(bytes: &[u8]) -> Result<State, ObjectError> {
160637×
        State::from_checkpoint(bytes, |_| {
16071×
            Err(ObjectError::Schema(
16081×
                "a version 2 checkpoint needs its graph blobs".into(),
16091×
            ))
16101×
        })
161137×
    }
1612
1613
    /// Decode a checkpoint payload of either version (spec §11). For
1614
    /// version 2, `load` returns the contents of the named graph blobs, in
1615
    /// order; the graphs are then decoded in parallel.
161660×
    pub fn from_checkpoint(
161760×
        bytes: &[u8],
161860×
        load: impl FnOnce(&[ObjectRef]) -> Result<Vec<Vec<u8>>, ObjectError>,
161960×
    ) -> Result<State, ObjectError> {
1620
        use rayon::prelude::*;
162160×
        let v = cbor::decode(bytes)?;
162259×
        let m = o::map(&v, "checkpoint payload")?;
162359×
        let version = o::uint(o::field(m, 0, "version")?, "version")?;
162459×
        if version != o::FORMAT_VERSION && version != CHECKPOINT_INDEX_VERSION {
16251×
            return Err(ObjectError::Schema(format!(
16261×
                "checkpoint payload version {version} is not supported"
16271×
            )));
162858×
        }
162958×
        let mut st = State::payload_common(m)?;
163058×
        let entries = o::array(o::field(m, 3, "files")?, "files")?;
163158×
        let mut parsed: Vec<(NodeId, &Value)> = Vec::with_capacity(entries.len());
1632236×
        for f in entries {
1633236×
            let e = o::array(f, "file")?;
1634236×
            if e.len() != 2 {
16350×
                return Err(ObjectError::Schema("file: expected [node, graph]".into()));
1636236×
            }
1637236×
            let node = NodeId::from_slice(o::bytes(&e[0], "node")?)
1638236×
                .ok_or_else(|| ObjectError::Schema("node: expected 16 bytes".into()))?;
1639236×
            parsed.push((node, &e[1]));
1640
        }
1641
        // Version 2 names large graphs by blob: fetch those in one batch.
164258×
        let mut fetched: HashMap<NodeId, Vec<u8>> = HashMap::new();
164358×
        if version == CHECKPOINT_INDEX_VERSION {
164420×
            let wanted: Vec<(NodeId, ObjectRef)> = parsed
164520×
                .iter()
164690×
                .filter(|(_, e)| matches!(e, Value::Bytes(_)))
164720×
                .map(|(n, e)| Ok((*n, o::oref(e, "graph blob")?)))
164820×
                .collect::<Result<_, ObjectError>>()?;
164920×
            if !wanted.is_empty() {
16507×
                let refs: Vec<ObjectRef> = wanted.iter().map(|(_, r)| *r).collect();
16513×
                let contents = {
16527×
                    let _t = crate::timing::phase("load:   graph blobs read");
16537×
                    load(&refs)?
1654
                };
16553×
                if contents.len() != refs.len() {
16560×
                    return Err(ObjectError::Schema("graph blobs: wrong count".into()));
16573×
                }
16583×
                fetched.extend(wanted.iter().map(|(n, _)| *n).zip(contents));
165913×
            }
1660146×
        } else if parsed.iter().any(|(_, e)| matches!(e, Value::Bytes(_))) {
16610×
            return Err(ObjectError::Schema(
16620×
                "a version 1 payload holds graphs inline".into(),
16630×
            ));
166438×
        }
1665
        // File graphs are independent; decode and build them in parallel.
166654×
        let graphs = parsed
166754×
            .par_iter()
1668230×
            .map(|(node, entry)| {
1669230×
                let graph = match fetched.get(node) {
16703×
                    Some(bytes) => FileGraph::from_value(&cbor::decode(bytes)?)?,
1671227×
                    None => FileGraph::from_value(entry)?,
1672
                };
1673230×
                Ok((*node, graph))
1674230×
            })
167554×
            .collect::<Result<Vec<_>, ObjectError>>()?;
167654×
        st.files
1677230×
            .extend(graphs.into_iter().map(|(n, g)| (n, Arc::new(g))));
167854×
        Ok(st)
167960×
    }
1680
1681
    /// The graph blobs a payload names: empty for version 1.
16829×
    pub fn checkpoint_graph_refs(bytes: &[u8]) -> Result<Vec<ObjectRef>, ObjectError> {
16839×
        let v = cbor::decode(bytes)?;
16849×
        let m = o::map(&v, "checkpoint payload")?;
16859×
        let mut out = Vec::new();
168627×
        for f in o::array(o::field(m, 3, "files")?, "files")? {
168727×
            if let Some(entry @ Value::Bytes(_)) = o::array(f, "file")?.get(1) {
16886×
                out.push(o::oref(entry, "graph blob")?);
168921×
            }
1690
        }
16919×
        Ok(out)
16929×
    }
1693
1694
    /// Decode a checkpoint payload for history only: a version 2 payload's
1695
    /// graphs are noted, not read or built, and come later through
1696
    /// [`State::load_graphs`]. A version 1 payload holds them inline and is
1697
    /// decoded in full.
169842×
    pub fn from_checkpoint_history(bytes: &[u8]) -> Result<State, ObjectError> {
169942×
        let v = cbor::decode(bytes)?;
170042×
        let m = o::map(&v, "checkpoint payload")?;
170142×
        let version = o::uint(o::field(m, 0, "version")?, "version")?;
170242×
        if version != CHECKPOINT_INDEX_VERSION {
17031×
            return State::from_checkpoint_payload(bytes);
170441×
        }
170541×
        let mut st = State::payload_common(m)?;
170641×
        let mut deferred = crate::fasthash::FastMap::default();
1707139×
        for f in o::array(o::field(m, 3, "files")?, "files")? {
1708139×
            let e = o::array(f, "file")?;
1709139×
            if e.len() != 2 {
17100×
                return Err(ObjectError::Schema("file: expected [node, graph]".into()));
1711139×
            }
1712139×
            let node = NodeId::from_slice(o::bytes(&e[0], "node")?)
1713139×
                .ok_or_else(|| ObjectError::Schema("node: expected 16 bytes".into()))?;
1714139×
            let base = match &e[1] {
17154×
                Value::Bytes(_) => DeferredBase::Blob(o::oref(&e[1], "graph blob")?),
1716135×
                other => DeferredBase::Inline(other.clone()),
1717
            };
1718139×
            deferred.insert(
1719139×
                node,
1720139×
                Arc::new(Deferred::from_body(DeferredBody {
1721139×
                    base: Some(base),
1722139×
                    ..DeferredBody::default()
1723139×
                })),
1724
            );
1725
        }
172641×
        st.deferred = Some(deferred);
172741×
        Ok(st)
172842×
    }
1729
1730
    /// Changes, tree and replaces of a payload of either version.
173199×
    fn payload_common(m: &BTreeMap<u64, Value>) -> Result<State, ObjectError> {
173299×
        let mut st = State::new();
1733495×
        for c in o::array(o::field(m, 1, "changes")?, "changes")? {
1734495×
            let e = o::array(c, "change meta")?;
1735495×
            if e.len() != 8 {
17360×
                return Err(ObjectError::Schema("change meta: expected 8 fields".into()));
1737495×
            }
1738495×
            let hash = o::oref(&e[0], "hash")?;
1739495×
            let meta = ChangeMeta {
1740495×
                change_id: crate::ids::ChangeId::from_slice(o::bytes(&e[1], "change id")?)
1741495×
                    .ok_or_else(|| ObjectError::Schema("change id: expected 16 bytes".into()))?,
1742495×
                deps: o::array(&e[2], "deps")?
1743495×
                    .iter()
1744600×
                    .map(|d| o::oref(d, "dep"))
1745495×
                    .collect::<Result<Vec<_>, _>>()?,
1746495×
                message: o::text(&e[3], "message")?.to_owned(),
1747
                hlc: Hlc {
1748495×
                    wall_ms: o::uint(&e[4], "wall")?,
1749495×
                    logical: o::u32_(&e[5], "logical")?,
1750
                },
1751495×
                kind: ChangeKind::from_tag(o::uint(&e[6], "kind")?)?,
1752495×
                author: o::oref(&e[7], "author")?,
1753
            };
1754495×
            st.changes.insert(hash, Arc::new(meta));
1755
        }
175699×
        st.tree = TreeGraph::from_value(o::field(m, 2, "tree")?)?;
175799×
        let opt = |v: &Value| -> Result<Option<ObjectRef>, ObjectError> {
175824×
            if v.is_null() {
175912×
                Ok(None)
1760
            } else {
176112×
                o::oref(v, "blob").map(Some)
1762
            }
176324×
        };
176499×
        for r in o::array(o::field(m, 4, "replaces")?, "replaces")? {
176512×
            let e = o::array(r, "replace")?;
176612×
            if e.len() != 2 {
17670×
                return Err(ObjectError::Schema(
17680×
                    "replace: expected [node, records]".into(),
17690×
                ));
177012×
            }
177112×
            let node = NodeId::from_slice(o::bytes(&e[0], "node")?)
177212×
                .ok_or_else(|| ObjectError::Schema("node: expected 16 bytes".into()))?;
177312×
            let mut recs = Vec::new();
177412×
            for rec in o::array(&e[1], "records")? {
177512×
                let x = o::array(rec, "record")?;
177612×
                if x.len() != 3 {
17770×
                    return Err(ObjectError::Schema("record: expected 3 fields".into()));
177812×
                }
177912×
                recs.push((o::oref(&x[0], "change")?, opt(&x[1])?, opt(&x[2])?));
1780
            }
178112×
            st.replaces.insert(node, recs);
1782
        }
178399×
        Ok(st)
178499×
    }
1785
17863762×
    pub fn validate(&self) -> Result<(), StateError> {
17879569×
        for (h, c) in &self.changes {
17889569×
            for d in &c.deps {
17895294×
                if !self.changes.contains_key(d) {
17900×
                    return Err(StateError::MissingDep(*d, *h));
17915294×
                }
1792
            }
1793
        }
17943762×
        Ok(())
17953762×
    }
1796
179712985×
    fn apply_with(&mut self, h: ChangeHash, c: Change, texts: Vec<TextRef>) {
179839295×
        for (i, op) in c.tree_ops.iter().enumerate() {
179939295×
            self.tree.apply(h, i as u32, &resolve_tree_op(op, h));
180039295×
        }
180112985×
        if let Some(deferred) = self.deferred.as_mut() {
1802
            // History only: edits of files not built wait, grouped per file,
1803
            // with the texts of inserts read without their lines.
18047699×
            let mut texts = texts.into_iter();
18057699×
            let mut per_node: HashMap<NodeId, (Vec<Edit>, Vec<TextRef>)> = HashMap::new();
180638907×
            for e in c.edits {
180738907×
                match e {
1808636×
                    Edit::Replace { node, old, new } => {
1809636×
                        self.replaces.entry(node).or_default().push((h, old, new))
1810
                    }
181138271×
                    e => {
181238271×
                        let slot = per_node.entry(e.node()).or_default();
181336599×
                        if matches!(&e, Edit::Insert { lines, .. } if lines.is_empty()) {
181432949×
                            slot.1.extend(texts.next());
181532949×
                        }
181638271×
                        slot.0.push(e);
1817
                    }
1818
                }
1819
            }
182036211×
            for (node, (edits, node_texts)) in per_node {
182136211×
                match self.files.get_mut(&node) {
1822463×
                    Some(g) => {
1823463×
                        assert!(
1824463×
                            node_texts.is_empty(),
1825
                            "a change read without its text applied to a built graph"
1826
                        );
1827463×
                        apply_file_edits(Arc::make_mut(g), h, edits)
1828
                    }
182935748×
                    None => {
183035748×
                        let d = Arc::make_mut(deferred.entry(node).or_default());
183135748×
                        d.add(h, edits, node_texts);
183235748×
                        d.key = Default::default();
183335748×
                    }
1834
                }
1835
            }
18367699×
            return;
18375286×
        }
18385286×
        assert!(
18395286×
            texts.is_empty(),
1840
            "a change read without its text applied to a full state"
1841
        );
18425286×
        let mut counters: HashMap<NodeId, u32> = HashMap::new();
18436143×
        for e in c.edits {
18446143×
            match e {
1845
                Edit::Insert {
18464675×
                    node,
18474675×
                    after,
18484675×
                    before,
18494675×
                    lines,
18504675×
                } => {
18514675×
                    let idx = counters.entry(node).or_default();
18524675×
                    let first = *idx;
18534675×
                    *idx += lines.len() as u32;
18544675×
                    Arc::make_mut(self.files.entry(node).or_default()).insert_owned(
18554675×
                        h,
18564675×
                        first,
18574675×
                        vertex(after.resolve(h)),
18584675×
                        vertex(before.resolve(h)),
18594675×
                        lines,
18604675×
                    );
18614675×
                }
18621353×
                Edit::Delete { node, atoms } => {
18631353×
                    let g = Arc::make_mut(self.files.entry(node).or_default());
18641685×
                    for r in atoms {
18651685×
                        g.delete(h, &r.resolve(h));
18661685×
                    }
1867
                }
1868115×
                Edit::Replace { node, old, new } => {
1869115×
                    self.replaces.entry(node).or_default().push((h, old, new));
1870115×
                }
1871
            }
1872
        }
187312985×
    }
1874
1875
    /// Transitive dependencies of `h`, excluding `h` itself.
18768892×
    pub fn closure(&self, h: &ChangeHash) -> Arc<HashSet<ChangeHash>> {
18778892×
        if let Some(c) = self.closure.read().unwrap().sets.get(h) {
18787496×
            return c.clone();
18791396×
        }
18801396×
        let mut out = HashSet::new();
18811396×
        let mut stack: Vec<ChangeHash> = self
18821396×
            .changes
18831396×
            .get(h)
18841396×
            .map(|c| c.deps.clone())
18851396×
            .unwrap_or_default();
188616835×
        while let Some(d) = stack.pop() {
188715439×
            if out.insert(d) {
188810922×
                if let Some(c) = self.changes.get(&d) {
188910922×
                    stack.extend(c.deps.iter().copied());
189010922×
                }
18914517×
            }
1892
        }
18931396×
        let rc = Arc::new(out);
18941396×
        let mut memo = self.closure.write().unwrap();
18951396×
        if memo.held + rc.len() > CLOSURE_MEMO_HASHES {
18960×
            memo.sets.clear();
18970×
            memo.held = 0;
18981396×
        }
18991396×
        memo.held += rc.len();
19001396×
        memo.sets.insert(*h, rc.clone());
19011396×
        rc
19028892×
    }
1903
1904
    /// `h`'s level (see `levels`), worked out for it and its ancestors not
1905
    /// yet known, without recursion: histories are deeper than a stack. An
1906
    /// unknown change has level 0.
1907400285×
    fn level(&self, h: &ChangeHash) -> u32 {
1908
        // Nearly always known: read without excluding the other threads
1909
        // rendering beside this one (oxygen#96).
1910400285×
        if let Some(l) = self.levels.read().unwrap().get(h) {
1911397908×
            return *l;
19122377×
        }
19132377×
        let mut levels = self.levels.write().unwrap();
19142377×
        if let Some(l) = levels.get(h) {
191510×
            return *l;
19162367×
        }
19172367×
        let mut stack: Vec<(ChangeHash, bool)> = vec![(*h, false)];
19187801×
        while let Some((c, expanded)) = stack.pop() {
19195434×
            if levels.contains_key(&c) {
192038×
                continue;
19215396×
            }
19225396×
            let deps: &[ChangeHash] = self
19235396×
                .changes
19245396×
                .get(&c)
19255396×
                .map(|m| m.deps.as_slice())
19265396×
                .unwrap_or(&[]);
19275396×
            if expanded {
19282698×
                let l = deps
19292698×
                    .iter()
19303386×
                    .map(|d| levels.get(d).map_or(0, |l| l + 1))
19312698×
                    .max()
19322698×
                    .unwrap_or(0);
19332698×
                levels.insert(c, l);
1934
            } else {
19352698×
                stack.push((c, true));
19363386×
                for d in deps {
19373386×
                    if !levels.contains_key(d) {
1938369×
                        stack.push((*d, false));
19393017×
                    }
1940
                }
1941
            }
1942
        }
19432367×
        levels[h]
1944400285×
    }
1945
1946
    /// Changes that no other change depends on directly, sorted.
1947397×
    pub fn heads(&self) -> Vec<ChangeHash> {
1948397×
        let depended: crate::fasthash::FastSet<&ChangeHash> =
19493346×
            self.changes.values().flat_map(|c| c.deps.iter()).collect();
1950397×
        let mut heads: Vec<ChangeHash> = self
1951397×
            .changes
1952397×
            .keys()
19533346×
            .filter(|h| !depended.contains(h))
1954397×
            .copied()
1955397×
            .collect();
1956397×
        heads.sort();
1957397×
        heads
1958397×
    }
1959
1960
    /// Maximal elements among `hashes` (those not in the closure of another),
1961
    /// sorted. One walk down from all of them at once: whatever the walk
1962
    /// reaches is below some other element. Linear in the changes reached,
1963
    /// where comparing every pair's closures was quadratic in the number
1964
    /// of heads, which a bookmark of independent changes has many of.
19659458×
    pub fn maximal(&self, hashes: &[ChangeHash]) -> Vec<ChangeHash> {
19669458×
        let set: BTreeSet<ChangeHash> = hashes.iter().copied().collect();
19679458×
        let mut below: crate::fasthash::FastSet<ChangeHash> = Default::default();
19689458×
        let mut stack: Vec<ChangeHash> = set
19699458×
            .iter()
197016813×
            .filter_map(|h| self.changes.get(h))
197114285×
            .flat_map(|c| c.deps.iter().copied())
19729458×
            .collect();
197322040×
        while let Some(d) = stack.pop() {
197412582×
            if below.insert(d) {
19756221×
                if let Some(c) = self.changes.get(&d) {
19766221×
                    stack.extend(c.deps.iter().copied());
19776221×
                }
19786361×
            }
1979
        }
198016813×
        set.into_iter().filter(|h| !below.contains(h)).collect()
19819458×
    }
1982
19830×
    pub fn materialize_file(&self, node: &NodeId) -> Materialized {
19840×
        self.graph(node)
19850×
            .map(|g| g.materialize(self))
19860×
            .unwrap_or_default()
19870×
    }
1988
1989
    /// Live lines of a text node in materialized order.
199021×
    pub fn live_lines(&self, node: &NodeId) -> Vec<(AtomKey, Vec<u8>)> {
199121×
        let Some(g) = self.graph(node) else {
19922×
            return vec![];
1993
        };
199419×
        let m = g.materialize(self);
199519×
        m.live_atoms()
199619×
            .into_iter()
199745×
            .map(|a| (a, g.content(&a).unwrap_or_default().to_vec()))
199819×
            .collect()
199921×
    }
2000
2001
    /// Live blobs of an opaque node: every `new` that no later `Replace` retired.
2002887×
    pub fn live_blobs(&self, node: &NodeId) -> Vec<ObjectRef> {
2003887×
        let mut live: Vec<ObjectRef> = self
2004887×
            .live_replaces(node)
2005887×
            .into_iter()
2006887×
            .map(|(_, n)| n)
2007887×
            .collect();
2008887×
        live.sort();
2009887×
        live.dedup();
2010887×
        live
2011887×
    }
2012
2013
    /// The replace records whose blob is live, with the change that set it
2014
    /// (spec §9.3). A record's blob is superseded by a later replacement of
2015
    /// that blob: one whose change depends on the record's change, as every
2016
    /// replacement depends on the change that made its old blob current
2017
    /// (§8.1). Two replacements that do not know each other both stay live
2018
    /// (a replace conflict), and a file reverted to an earlier blob keeps it.
2019
    /// A replacement without that dependency, from before the rule, is read
2020
    /// as before: it supersedes every record that did not come after it.
2021927×
    pub fn live_replaces(&self, node: &NodeId) -> Vec<(ChangeHash, ObjectRef)> {
2022927×
        let Some(rs) = self.replaces.get(node) else {
20230×
            return vec![];
2024
        };
2025
        // Dependence by the bounded walk rather than whole closures: on a
2026
        // long history a closure is a hundred thousand changes, and the
2027
        // memo cannot keep them all (#107).
2028927×
        let dep =
2029147×
            |a: &ChangeHash, b: &ChangeHash| a != b && <Self as DepOracle>::depends_on(self, a, b);
2030927×
        let knows_its_old = |s: &ChangeHash, old: &ObjectRef| {
203111×
            rs.iter()
203221×
                .any(|(q, _, n)| n.as_ref() == Some(old) && q != s && dep(s, q))
203311×
        };
2034927×
        rs.iter()
20351067×
            .filter_map(|(c, _, n)| n.map(|n| (*c, n)))
20361067×
            .filter(|(c, n)| {
20371350×
                !rs.iter().any(|(s, o, _)| {
20381350×
                    s != c && *o == Some(*n) && (dep(s, c) || (!knows_its_old(s, n) && !dep(c, s)))
20391350×
                })
20401067×
            })
2041927×
            .collect()
2042927×
    }
2043
2044
    /// Short label for conflict markers: `changeid-prefix "message"`.
20451071×
    pub fn label(&self, h: &ChangeHash) -> String {
20461071×
        match self.changes.get(h) {
20471071×
            Some(c) => format!(
2048
                "{} \"{}\"",
20491071×
                c.change_id.short(8),
20501071×
                c.message.lines().next().unwrap_or("")
2051
            ),
20520×
            None => format!("#{}", h.short(12)),
2053
        }
20541071×
    }
2055
2056
    /// For resolving a conflicted text node (spec §9.6): the live atoms that
2057
    /// are not part of any conflict region in order, every live atom that is
2058
    /// part of a region, and every change that contributed to a region.
2059
    /// The changes that introduced the live blobs of an opaque node, for a
2060
    /// replace conflict (spec §9.3).
20612×
    pub fn blob_contributors(&self, node: &NodeId, blobs: &[ObjectRef]) -> Vec<ChangeHash> {
20622×
        let mut out: Vec<ChangeHash> = Vec::new();
20636×
        for (h, _, new) in self.replaces.get(node).map(|v| v.as_slice()).unwrap_or(&[]) {
20646×
            if new.is_some_and(|b| blobs.contains(&b)) && !out.contains(h) {
20654×
                out.push(*h);
20664×
            }
2067
        }
20682×
        out.sort();
20692×
        out
20702×
    }
2071
207232×
    pub fn conflict_regions(&self, node: &NodeId) -> ConflictRegions {
207332×
        let mut out = ConflictRegions::default();
207432×
        let Some(g) = self.graph(node) else {
20750×
            return out;
2076
        };
207732×
        let m = g.materialize(self);
207832×
        let mut contributors = BTreeSet::new();
207978×
        for seg in &m.segments {
208078×
            match seg {
208146×
                crate::graph::Segment::Lines(atoms) => {
208246×
                    for a in atoms {
208346×
                        out.plain
208446×
                            .push((*a, g.content(a).unwrap_or_default().to_vec()));
208546×
                    }
2086
                }
208732×
                crate::graph::Segment::Conflict(sides) => {
208864×
                    for side in sides {
208966×
                        for a in side {
209066×
                            out.region_atoms.push(*a);
209166×
                            contributors.insert(a.0);
209266×
                        }
2093
                    }
2094
                }
20950×
                crate::graph::Segment::Zombie { deleted, inserted } => {
20960×
                    for a in deleted {
20970×
                        contributors.insert(a.0);
20980×
                        contributors.extend(g.deleters(a).copied());
20990×
                    }
21000×
                    for a in inserted {
21010×
                        out.region_atoms.push(*a);
21020×
                        contributors.insert(a.0);
21030×
                    }
2104
                }
2105
            }
2106
        }
210732×
        out.contributors = contributors.into_iter().collect();
210832×
        out
210932×
    }
2110
211125841×
    pub fn render_file(&self, node: &NodeId) -> (Vec<u8>, Materialized) {
211225841×
        let Some(g) = self.graph(node) else {
2113581×
            return (Vec::new(), Materialized::default());
2114
        };
211525260×
        let m = g.materialize(self);
211625260×
        let bytes = g.render(&m, &|h| self.label(h));
211725260×
        (bytes, m)
211825841×
    }
2119
2120
    /// Every file rendered, in parallel: files are independent, and
2121
    /// rendering them is most of a snapshot's cost in a large tree.
21222595×
    pub fn materialize(&self) -> Snapshot {
2123
        use rayon::prelude::*;
21242595×
        self.require_full("materializing a state");
21252595×
        let tree = self.view();
21262595×
        let entries: Vec<(&String, &crate::tree::TreeEntry)> = tree.entries.iter().collect();
21272595×
        let rendered: Vec<(String, FileState)> = entries
21282595×
            .par_iter()
212917025×
            .map(|(path, e)| (path.to_string(), self.file_state(e)))
21302595×
            .collect();
21312595×
        let files: BTreeMap<String, FileState> = rendered.into_iter().collect();
21322595×
        Snapshot {
21332595×
            tree,
21342595×
            files,
21352595×
            unrendered: BTreeSet::new(),
21362595×
        }
21372595×
    }
2138
2139
    /// Like [`State::materialize`], leaving the text files of nodes in
2140
    /// `skip` unrendered (listed in [`Snapshot::unrendered`]); every other
2141
    /// file's graph must be loaded. For a caller that already knows those
2142
    /// files are unchanged and not conflicted. Its tree is the tree
2143
    /// graph's own view, as a Git import builds against it: same-path
2144
    /// additions are not unified ([`State::view`]), since the import's
2145
    /// merge records Git's tree for them itself.
214620×
    pub fn materialize_except(&self, skip: &HashSet<NodeId>) -> Snapshot {
214720×
        self.materialize_leaving(self.tree.view(), |_, e| skip.contains(&e.node))
214820×
    }
2149
2150
    /// As [`State::materialize_except`], rendering only the text files at
2151
    /// `render` (a few, where the caller knows the rest unchanged), with
2152
    /// the tree as [`State::view`] has it.
2153921×
    pub fn materialize_only(&self, render: &std::collections::HashSet<&str>) -> Snapshot {
21547927×
        self.materialize_leaving(self.view(), |p, _| !render.contains(p))
2155921×
    }
2156
2157
    /// As [`State::materialize_only`], rendering only the text files of
2158
    /// the nodes in `render` (a change's edited files), without a set of
2159
    /// every other file to leave out.
2160226×
    pub fn materialize_nodes(&self, render: &HashSet<NodeId>) -> Snapshot {
2161593×
        self.materialize_leaving(self.view(), |_, e| !render.contains(&e.node))
2162226×
    }
2163
21641167×
    fn materialize_leaving(
21651167×
        &self,
21661167×
        tree: TreeView,
21671167×
        leave: impl Fn(&str, &crate::tree::TreeEntry) -> bool + Sync,
21681167×
    ) -> Snapshot {
2169
        use rayon::prelude::*;
21701167×
        let entries: Vec<(&String, &crate::tree::TreeEntry)> = tree.entries.iter().collect();
21711167×
        let rendered: Vec<(String, FileState, bool)> = entries
21721167×
            .par_iter()
21738873×
            .map(|(path, e)| {
21748873×
                if e.kind == NodeKind::File && !e.mode.opaque && leave(path, e) {
21754580×
                    let f = FileState {
21764580×
                        node: e.node,
21774580×
                        kind: e.kind,
21784580×
                        mode: e.mode,
21794580×
                        content: FileContent::Text {
21804580×
                            bytes: Vec::new(),
21814580×
                            materialized: Materialized::default(),
21824580×
                        },
21834580×
                    };
21844580×
                    (path.to_string(), f, true)
2185
                } else {
21864293×
                    (path.to_string(), self.file_state(e), false)
2187
                }
21888873×
            })
21891167×
            .collect();
2190
        // In the view's path order, so the maps are built without searching.
21911167×
        let unrendered: BTreeSet<String> = rendered
21921167×
            .iter()
21931167×
            .filter(|(_, _, skipped)| *skipped)
21944580×
            .map(|(p, _, _)| p.clone())
21951167×
            .collect();
21961167×
        let files: BTreeMap<String, FileState> =
21978873×
            rendered.into_iter().map(|(p, f, _)| (p, f)).collect();
21981167×
        Snapshot {
21991167×
            tree,
22001167×
            files,
22011167×
            unrendered,
22021167×
        }
22031167×
    }
2204
2205
    /// A snapshot of only `paths` (those `view` has), `view` being this
2206
    /// state's tree view: for a caller that works from what changed. The
2207
    /// graphs of those text files must be loaded.
220890×
    pub fn materialize_paths<'p>(
220990×
        &self,
221090×
        view: crate::tree::TreeView,
221190×
        paths: impl IntoIterator<Item = &'p str>,
221290×
    ) -> Snapshot {
2213
        use rayon::prelude::*;
2214
        // On every core: a merge may render thousands (oxygen#96).
221590×
        let entries: Vec<(&str, &crate::tree::TreeEntry)> = paths
221690×
            .into_iter()
2217131×
            .filter_map(|p| Some((p, view.entries.get(p)?)))
221890×
            .collect();
221990×
        let files = entries
222090×
            .par_iter()
2221107×
            .map(|(p, e)| (p.to_string(), self.file_state(e)))
222290×
            .collect::<Vec<_>>()
222390×
            .into_iter()
222490×
            .collect();
222590×
        Snapshot {
222690×
            tree: view,
222790×
            files,
222890×
            unrendered: BTreeSet::new(),
222990×
        }
223090×
    }
2231
2232
    /// The state of one tree entry: a directory, an opaque file's live
2233
    /// blobs, or a text file rendered from its graph (which must be built).
22340×
    pub fn entry_state(&self, e: &crate::tree::TreeEntry) -> FileState {
22350×
        self.file_state(e)
22360×
    }
2237
223821425×
    fn file_state(&self, e: &crate::tree::TreeEntry) -> FileState {
2239
        {
224021425×
            let content = match e.kind {
22411402×
                NodeKind::Dir => FileContent::Dir,
2242124×
                NodeKind::Symlink | NodeKind::Subrepo => FileContent::Opaque {
2243124×
                    blobs: self.live_blobs(&e.node),
2244124×
                },
2245541×
                NodeKind::File if e.mode.opaque => FileContent::Opaque {
2246541×
                    blobs: self.live_blobs(&e.node),
2247541×
                },
2248
                NodeKind::File => {
224919358×
                    let (bytes, materialized) = self.render_file(&e.node);
225019358×
                    FileContent::Text {
225119358×
                        bytes,
225219358×
                        materialized,
225319358×
                    }
2254
                }
2255
            };
225621425×
            FileState {
225721425×
                node: e.node,
225821425×
                kind: e.kind,
225921425×
                mode: e.mode,
226021425×
                content,
226121425×
            }
2262
        }
226321425×
    }
2264
}
2265
2266
impl DepOracle for State {
2267
    /// Whether `a` depends on `b`, by a walk down from `a` that stops at
2268
    /// changes no higher than `b`: they cannot have `b` below them. Nothing
2269
    /// of the walk is kept, so asking about many pairs costs no more memory
2270
    /// than one walk (#25); a closure already memoised answers at once.
227166378×
    fn depends_on(&self, a: &ChangeHash, b: &ChangeHash) -> bool {
227266378×
        if a == b {
2273800×
            return true;
227465578×
        }
227565578×
        if let Some(c) = self.closure.read().unwrap().sets.get(a) {
22760×
            return c.contains(b);
227765578×
        }
227865578×
        if !self.changes.contains_key(b) {
2279800×
            return false;
228064778×
        }
228164778×
        let lb = self.level(b);
228264778×
        if self.level(a) <= lb {
228335933×
            return false;
228428845×
        }
228528845×
        let mut seen: crate::fasthash::FastSet<ChangeHash> = Default::default();
228628845×
        let mut stack: Vec<ChangeHash> = self
228728845×
            .changes
228828845×
            .get(a)
228928845×
            .map(|m| m.deps.clone())
229028845×
            .unwrap_or_default();
2291357770×
        while let Some(c) = stack.pop() {
2292338501×
            if c == *b {
22939576×
                return true;
2294328925×
            }
2295328925×
            if !seen.insert(c) || self.level(&c) <= lb {
2296221263×
                continue;
2297107662×
            }
2298107662×
            if let Some(m) = self.changes.get(&c) {
2299107662×
                stack.extend(m.deps.iter().copied());
2300107662×
            }
2301
        }
230219269×
        false
230366378×
    }
2304
}
2305
2306
#[cfg(test)]
2307
mod tests {
2308
    use super::*;
2309
2310
    /// A changes block of thousands comes back whole and in order, and a
2311
    /// truncated one is refused (oxygen#94).
2312
    #[test]
23131×
    fn a_long_changes_block_decodes_in_order() {
2314
        use crate::flat::Writer;
23151×
        let metas: Vec<(ChangeHash, ChangeMeta)> = (0..5000u32)
23165000×
            .map(|i| {
23175000×
                let mut h = [0u8; 32];
23185000×
                h[..4].copy_from_slice(&i.to_le_bytes());
23195000×
                let meta = ChangeMeta {
23205000×
                    change_id: crate::ids::ChangeId([i as u8; 16]),
23215000×
                    deps: (0..i % 3).map(|d| ObjectRef([d as u8; 32])).collect(),
23225000×
                    author: ObjectRef([7; 32]),
23235000×
                    hlc: Hlc {
23245000×
                        wall_ms: 1_000_000 - i as u64,
23255000×
                        logical: i,
23265000×
                    },
23275000×
                    message: "x".repeat((i % 50) as usize),
23285000×
                    kind: ChangeKind::Normal,
2329
                };
23305000×
                (ObjectRef(h), meta)
23315000×
            })
23321×
            .collect();
23331×
        let mut w = Writer::default();
23341×
        w.fixed(State::INDEX_MAGIC);
23351×
        w.count(metas.len());
23365000×
        for (h, m) in &metas {
23375000×
            w.object(h);
23385000×
            m.write_flat(&mut w);
23395000×
        }
23401×
        assert_eq!(State::read_index_changes_listed(&w.0).unwrap(), metas);
23411×
        let mut cut = w.0.clone();
23421×
        cut.pop();
23431×
        assert!(
23441×
            State::read_index_changes_listed(&cut).is_err(),
2345
            "a truncated block is refused"
2346
        );
23471×
    }
2348
    use crate::ids::ChangeId;
2349
    use crate::objects::{AtomRange, ChangeKind, Hlc, OpRef};
2350
23512023×
    fn change(deps: Vec<ChangeHash>, tree_ops: Vec<TreeOp>, edits: Vec<Edit>) -> Change {
23522023×
        let mut c = Change {
23532023×
            version: crate::objects::CHANGE_VERSION,
23542023×
            change_id: ChangeId::random(),
23552023×
            deps,
23562023×
            author: ObjectRef::ZERO,
23572023×
            hlc: Hlc {
23582023×
                wall_ms: 1,
23592023×
                logical: 0,
23602023×
            },
23612023×
            message: "m".into(),
23622023×
            tree_ops,
23632023×
            edits,
23642023×
            kind: ChangeKind::Normal,
23652023×
        };
23662023×
        c.normalize();
23672023×
        c
23682023×
    }
2369
2370
    #[test]
23711×
    fn build_edit_and_materialize() {
23721×
        let mut st = State::new();
23731×
        let f = NodeId::random();
23741×
        let c1 = change(
23751×
            vec![],
23761×
            vec![TreeOp::AddNode {
23771×
                node: f,
23781×
                kind: NodeKind::File,
23791×
                mode: Mode::TEXT,
23801×
                parent: NodeId::ROOT,
23811×
                name: "a.txt".into(),
23821×
            }],
23831×
            vec![Edit::Insert {
23841×
                node: f,
23851×
                after: AtomRef::Start,
23861×
                before: AtomRef::End,
23871×
                lines: vec![b"one\n".to_vec(), b"two\n".to_vec()].into(),
23881×
            }],
2389
        );
23901×
        let h1 = st.add(c1).unwrap();
2391
        // Depends on h1: append a line and delete "one".
23921×
        let c2 = change(
23931×
            vec![h1],
23941×
            vec![],
23951×
            vec![
23961×
                Edit::Delete {
23971×
                    node: f,
23981×
                    atoms: vec![AtomRange {
23991×
                        change: h1,
24001×
                        from: 0,
24011×
                        to: 1,
24021×
                    }],
24031×
                },
24041×
                Edit::Insert {
24051×
                    node: f,
24061×
                    after: AtomRef::Atom {
24071×
                        change: h1,
24081×
                        index: 1,
24091×
                    },
24101×
                    before: AtomRef::End,
24111×
                    lines: vec![b"three\n".to_vec()].into(),
24121×
                },
2413
            ],
2414
        );
24151×
        let h2 = st.add(c2).unwrap();
24161×
        assert_eq!(st.heads(), vec![h2]);
24171×
        assert!(st.depends_on(&h2, &h1));
24181×
        assert!(!st.depends_on(&h1, &h2));
24191×
        let snap = st.materialize();
24201×
        assert!(!snap.is_conflicted());
24211×
        let a = &snap.files["a.txt"];
24221×
        assert!(matches!(&a.content, FileContent::Text { bytes, .. } if bytes == b"two\nthree\n"));
24231×
        assert_eq!(st.live_lines(&f).len(), 2);
24241×
    }
2425
2426
    #[test]
24271×
    fn missing_dependency_is_rejected() {
24281×
        let mut st = State::new();
24291×
        let c = change(vec![ObjectRef([1; 32])], vec![], vec![]);
24301×
        assert!(matches!(st.add(c), Err(StateError::MissingDep(..))));
24311×
    }
2432
2433
    #[test]
24341×
    fn self_references_and_multi_insert_indexing() {
24351×
        let mut st = State::new();
24361×
        let f = NodeId::random();
24371×
        let me = OpRef {
24381×
            change: ObjectRef::ZERO,
24391×
            index: 0,
24401×
        };
24411×
        let c1 = change(
24421×
            vec![],
24431×
            vec![
24441×
                TreeOp::AddNode {
24451×
                    node: f,
24461×
                    kind: NodeKind::File,
24471×
                    mode: Mode::TEXT,
24481×
                    parent: NodeId::ROOT,
24491×
                    name: "x".into(),
24501×
                },
24511×
                TreeOp::SetMode {
24521×
                    node: f,
24531×
                    old: me,
24541×
                    mode: Mode {
24551×
                        exec: true,
24561×
                        opaque: false,
24571×
                    },
24581×
                },
2459
            ],
24601×
            vec![
24611×
                Edit::Insert {
24621×
                    node: f,
24631×
                    after: AtomRef::Start,
24641×
                    before: AtomRef::End,
24651×
                    lines: vec![b"a\n".to_vec(), b"c\n".to_vec()].into(),
24661×
                },
2467
                // Insert between the two lines just inserted, using self references.
24681×
                Edit::Insert {
24691×
                    node: f,
24701×
                    after: AtomRef::Atom {
24711×
                        change: ObjectRef::ZERO,
24721×
                        index: 0,
24731×
                    },
24741×
                    before: AtomRef::Atom {
24751×
                        change: ObjectRef::ZERO,
24761×
                        index: 1,
24771×
                    },
24781×
                    lines: vec![b"b\n".to_vec()].into(),
24791×
                },
2480
            ],
2481
        );
24821×
        st.add(c1).unwrap();
24831×
        let snap = st.materialize();
24841×
        let x = &snap.files["x"];
24851×
        assert_eq!(
2486
            x.mode,
2487
            Mode {
2488
                exec: true,
2489
                opaque: false
2490
            }
2491
        );
24921×
        assert!(matches!(&x.content, FileContent::Text { bytes, .. } if bytes == b"a\nb\nc\n"));
24931×
    }
2494
2495
    #[test]
24961×
    fn opaque_replace_and_conflict() {
24971×
        let mut st = State::new();
24981×
        let f = NodeId::random();
24991×
        let b1 = ObjectRef([1; 32]);
25001×
        let b2 = ObjectRef([2; 32]);
25011×
        let b3 = ObjectRef([3; 32]);
25021×
        let h1 = st
25031×
            .add(change(
25041×
                vec![],
25051×
                vec![TreeOp::AddNode {
25061×
                    node: f,
25071×
                    kind: NodeKind::File,
25081×
                    mode: Mode::OPAQUE,
25091×
                    parent: NodeId::ROOT,
25101×
                    name: "img".into(),
25111×
                }],
25121×
                vec![Edit::Replace {
25131×
                    node: f,
25141×
                    old: None,
25151×
                    new: Some(b1),
25161×
                }],
2517
            ))
25181×
            .unwrap();
25191×
        assert_eq!(st.live_blobs(&f), vec![b1]);
25201×
        st.add(change(
25211×
            vec![h1],
25221×
            vec![],
25231×
            vec![Edit::Replace {
25241×
                node: f,
25251×
                old: Some(b1),
25261×
                new: Some(b2),
25271×
            }],
2528
        ))
25291×
        .unwrap();
25301×
        st.add(change(
25311×
            vec![h1],
25321×
            vec![],
25331×
            vec![Edit::Replace {
25341×
                node: f,
25351×
                old: Some(b1),
25361×
                new: Some(b3),
25371×
            }],
2538
        ))
25391×
        .unwrap();
25401×
        let snap = st.materialize();
25411×
        assert!(snap.is_conflicted());
25421×
        assert!(
25431×
            matches!(&snap.files["img"].content, FileContent::Opaque { blobs } if blobs == &vec![b2, b3])
2544
        );
25451×
    }
2546
2547
    #[test]
25481×
    fn checkpoint_payload_round_trip_and_replay() {
25491×
        let mut st = State::new();
25501×
        let f = NodeId::random();
25511×
        let h1 = st
25521×
            .add(change(
25531×
                vec![],
25541×
                vec![TreeOp::AddNode {
25551×
                    node: f,
25561×
                    kind: NodeKind::File,
25571×
                    mode: Mode::TEXT,
25581×
                    parent: NodeId::ROOT,
25591×
                    name: "a".into(),
25601×
                }],
25611×
                vec![Edit::Insert {
25621×
                    node: f,
25631×
                    after: AtomRef::Start,
25641×
                    before: AtomRef::End,
25651×
                    lines: vec![b"1\n".to_vec(), b"2\n".to_vec()].into(),
25661×
                }],
2567
            ))
25681×
            .unwrap();
25691×
        let h2 = st
25701×
            .add(change(
25711×
                vec![h1],
25721×
                vec![],
25731×
                vec![Edit::Delete {
25741×
                    node: f,
25751×
                    atoms: vec![AtomRange {
25761×
                        change: h1,
25771×
                        from: 0,
25781×
                        to: 1,
25791×
                    }],
25801×
                }],
2581
            ))
25821×
            .unwrap();
25831×
        let payload = st.checkpoint_payload();
25841×
        let back = State::from_checkpoint_payload(&payload).unwrap();
25851×
        assert_eq!(back.len(), 2);
25861×
        assert_eq!(back.heads(), vec![h2]);
25871×
        assert_eq!(back.get(&h1).unwrap().message, "m");
25881×
        assert_eq!(back.checkpoint_payload(), payload, "deterministic");
2589
        // Replaying a further change onto the restored state equals a full build.
25901×
        let c3 = change(
25911×
            vec![h2],
25921×
            vec![],
25931×
            vec![Edit::Insert {
25941×
                node: f,
25951×
                after: AtomRef::Atom {
25961×
                    change: h1,
25971×
                    index: 1,
25981×
                },
25991×
                before: AtomRef::End,
26001×
                lines: vec![b"3\n".to_vec()].into(),
26011×
            }],
2602
        );
26031×
        let mut full = st.clone();
26041×
        full.add(c3.clone()).unwrap();
26051×
        let mut restored = back;
26061×
        restored.add(c3).unwrap();
26071×
        let (a, b) = (full.materialize(), restored.materialize());
26081×
        assert_eq!(
26091×
            a.files.keys().collect::<Vec<_>>(),
26101×
            b.files.keys().collect::<Vec<_>>()
2611
        );
26121×
        assert_eq!(a.files["a"].content, b.files["a"].content);
26131×
        assert!(
26141×
            matches!(&b.files["a"].content, FileContent::Text { bytes, .. } if bytes == b"2\n3\n")
2615
        );
26161×
        assert!(State::from_checkpoint_payload(b"nonsense").is_err());
26171×
    }
2618
2619
    /// History-only states (O10f): the same state as a full load once its
2620
    /// graphs are built, by replay and from a version 2 checkpoint with
2621
    /// later changes; unloaded graphs are never read as empty.
2622
    #[test]
26231×
    fn graph_keys_change_exactly_when_a_graph_does() {
26241×
        let (f, g) = (NodeId::random(), NodeId::random());
26251×
        let add = |node: NodeId, name: &str| TreeOp::AddNode {
26262×
            node,
26272×
            kind: NodeKind::File,
2628
            mode: Mode::TEXT,
2629
            parent: NodeId::ROOT,
26302×
            name: name.into(),
26312×
        };
26321×
        let ins = |node: NodeId, line: &[u8]| Edit::Insert {
26334×
            node,
26344×
            after: AtomRef::Start,
26354×
            before: AtomRef::End,
26364×
            lines: vec![line.to_vec()].into(),
26374×
        };
26381×
        let c1 = change(
26391×
            vec![],
26401×
            vec![add(f, "f"), add(g, "g")],
26411×
            vec![ins(f, b"f\n"), ins(g, b"g\n")],
2642
        );
26431×
        let h1 = c1.hash();
26441×
        let c2 = change(vec![h1], vec![], vec![ins(g, b"g2\n")]);
26451×
        let h2 = c2.hash();
26461×
        let c3 = change(vec![h2], vec![], vec![ins(f, b"f2\n")]);
26476×
        let load = |cs: &[&Change]| {
26486×
            let mut st = State::new_history_only();
264910×
            for c in cs {
265010×
                st.add((*c).clone()).unwrap();
265110×
            }
26526×
            st
26536×
        };
26541×
        let s1 = load(&[&c1]);
26551×
        let s2 = load(&[&c1, &c2]);
26561×
        let s3 = load(&[&c1, &c2, &c3]);
265718×
        let key = |s: &State, n: &NodeId| s.graph_key(n).unwrap();
26581×
        assert_eq!(key(&s1, &f), key(&load(&[&c1]), &f), "same state, same key");
26591×
        assert_eq!(key(&s1, &f), key(&s2, &f), "an edit elsewhere keeps it");
26601×
        assert_ne!(key(&s1, &g), key(&s2, &g), "an edit of the file changes it");
26611×
        assert_ne!(key(&s2, &f), key(&s3, &f));
26621×
        assert_ne!(key(&s1, &f), key(&s1, &g), "different files differ");
26631×
        let mut built = load(&[&c1]);
26641×
        built
26651×
            .load_graphs(&[f], |_| unreachable!(), |_| unreachable!())
26661×
            .unwrap();
26671×
        assert!(built.graph_key(&f).is_none(), "a built graph has no key");
26681×
        let mut full = State::new();
26691×
        full.add(c1.clone()).unwrap();
26701×
        assert!(full.graph_key(&f).is_none(), "nor has a full state");
2671
2672
        // Through a version 2 checkpoint of c1, keys name its graph blobs
2673
        // or inline values: stable across loads, changed by later edits.
26741×
        let graphs = full.checkpoint_file_graphs();
26751×
        let index = full.checkpoint_index(&graphs, |enc| {
26760×
            crate::hash::hash_object(crate::hash::ObjectType::Blob, enc)
26770×
        });
26788×
        let cp = |extra: &[&Change]| {
26798×
            let mut st = State::from_checkpoint_history(&index).unwrap();
26808×
            for c in extra {
26815×
                st.add((*c).clone()).unwrap();
26825×
            }
26838×
            st
26848×
        };
26851×
        assert_eq!(key(&cp(&[]), &f), key(&cp(&[]), &f));
26861×
        assert_eq!(key(&cp(&[]), &f), key(&cp(&[&c2]), &f));
26871×
        assert_ne!(key(&cp(&[]), &g), key(&cp(&[&c2]), &g));
26881×
        assert_ne!(key(&cp(&[&c2]), &f), key(&cp(&[&c2, &c3]), &f));
2689
2690
        // Partial rendering: skipped files are listed and empty, the others
2691
        // render as in a full state.
26921×
        let mut part = load(&[&c1, &c2]);
26931×
        part.load_graphs(&[g], |_| unreachable!(), |_| unreachable!())
26941×
            .unwrap();
26951×
        let snap = part.materialize_except(&[f].into_iter().collect());
26961×
        assert_eq!(snap.unrendered, ["f".to_string()].into_iter().collect());
26971×
        let mut whole = State::new();
26981×
        whole.add(c1.clone()).unwrap();
26991×
        whole.add(c2.clone()).unwrap();
27001×
        assert_eq!(snap.files["g"], whole.materialize().files["g"]);
27011×
        assert!(
27021×
            matches!(&snap.files["f"].content, FileContent::Text { bytes, .. } if bytes.is_empty())
2703
        );
27041×
    }
2705
2706
    #[test]
27071×
    fn history_only_states_build_the_same_graphs_on_demand() {
27081×
        let small = NodeId::random();
27091×
        let big = NodeId::random();
27101×
        let add = |node: NodeId, name: &str| TreeOp::AddNode {
27112×
            node,
27122×
            kind: NodeKind::File,
2713
            mode: Mode::TEXT,
2714
            parent: NodeId::ROOT,
27152×
            name: name.into(),
27162×
        };
27171×
        let lines: crate::lines::Lines = (0..300)
2718300×
            .map(|i| format!("line {i}\n").into_bytes())
27191×
            .collect();
27201×
        let c1 = change(
27211×
            vec![],
27221×
            vec![add(small, "small"), add(big, "big")],
27231×
            vec![
27241×
                Edit::Insert {
27251×
                    node: small,
27261×
                    after: AtomRef::Start,
27271×
                    before: AtomRef::End,
27281×
                    lines: vec![b"x\n".to_vec()].into(),
27291×
                },
27301×
                Edit::Insert {
27311×
                    node: big,
27321×
                    after: AtomRef::Start,
27331×
                    before: AtomRef::End,
27341×
                    lines,
27351×
                },
2736
            ],
2737
        );
27381×
        let h1 = c1.hash();
27391×
        let c2 = change(
27401×
            vec![h1],
27411×
            vec![],
27421×
            vec![
27431×
                Edit::Delete {
27441×
                    node: big,
27451×
                    atoms: vec![AtomRange {
27461×
                        change: h1,
27471×
                        from: 2,
27481×
                        to: 9,
27491×
                    }],
27501×
                },
27511×
                Edit::Insert {
27521×
                    node: big,
27531×
                    after: AtomRef::Atom {
27541×
                        change: h1,
27551×
                        index: 20,
27561×
                    },
27571×
                    before: AtomRef::Atom {
27581×
                        change: h1,
27591×
                        index: 21,
27601×
                    },
27611×
                    lines: vec![b"new a\n".to_vec(), b"new b\n".to_vec()].into(),
27621×
                },
27631×
                Edit::Insert {
27641×
                    node: small,
27651×
                    after: AtomRef::Atom {
27661×
                        change: h1,
27671×
                        index: 0,
27681×
                    },
27691×
                    before: AtomRef::End,
27701×
                    lines: vec![b"y\n".to_vec()].into(),
27711×
                },
2772
            ],
2773
        );
27741×
        let h2 = c2.hash();
27751×
        let c3 = change(
27761×
            vec![h2],
27771×
            vec![],
27781×
            vec![Edit::Delete {
27791×
                node: small,
27801×
                atoms: vec![AtomRange {
27811×
                    change: h1,
27821×
                    from: 0,
27831×
                    to: 1,
27841×
                }],
27851×
            }],
2786
        );
2787
27881×
        let mut full = State::new();
27893×
        for c in [&c1, &c2, &c3] {
27903×
            full.add(c.clone()).unwrap();
27913×
        }
2792
        // By replay: nothing built until asked, then exactly the full graphs.
27931×
        let mut hist = State::new_history_only();
27943×
        for c in [&c1, &c2, &c3] {
27953×
            hist.add(c.clone()).unwrap();
27963×
        }
27971×
        assert!(!hist.is_full() && hist.files.is_empty(), "no graph built");
27981×
        assert_eq!(hist.len(), 3);
27991×
        hist.load_graphs(
28001×
            &[big],
28010×
            |_| unreachable!("no checkpoint"),
28020×
            |_| unreachable!(),
2803
        )
28041×
        .unwrap();
28051×
        assert_eq!(
28061×
            hist.render_file(&big).0,
28071×
            full.render_file(&big).0,
2808
            "one file on its own"
2809
        );
28101×
        assert!(!hist.is_full());
28111×
        hist.load_graphs(&[small, big], |_| unreachable!(), |_| unreachable!())
28121×
            .unwrap();
28131×
        assert!(hist.is_full());
28141×
        assert_eq!(hist.checkpoint_payload(), full.checkpoint_payload());
2815
2816
        // From a version 2 checkpoint of c1+c2, then c3 applied later.
28171×
        let mut base = State::new();
28181×
        base.add(c1.clone()).unwrap();
28191×
        base.add(c2.clone()).unwrap();
28201×
        let graphs = base.checkpoint_file_graphs();
28211×
        let mut blobs: HashMap<ObjectRef, Vec<u8>> = HashMap::new();
28221×
        let index = base.checkpoint_index(&graphs, |enc| {
28231×
            let r = crate::hash::hash_object(crate::hash::ObjectType::Blob, enc);
28241×
            blobs.insert(r, enc.to_vec());
28251×
            r
28261×
        });
28271×
        let mut from_cp = State::from_checkpoint_history(&index).unwrap();
28281×
        from_cp.add(c3.clone()).unwrap();
28291×
        assert!(!from_cp.is_full());
28301×
        let mut fetched = 0;
28311×
        from_cp
28321×
            .load_graphs(
28331×
                &[small, big],
28341×
                |refs| {
28351×
                    fetched += refs.len();
28361×
                    refs.iter()
28371×
                        .map(|r| {
28381×
                            blobs
28391×
                                .get(r)
28401×
                                .cloned()
28411×
                                .ok_or_else(|| ObjectError::Schema("missing".into()))
28421×
                        })
28431×
                        .collect()
28441×
                },
28450×
                |_| unreachable!(),
2846
            )
28471×
            .unwrap();
28481×
        assert_eq!(
2849
            fetched, 1,
2850
            "the large graph's blob, in one batch; the small one was inline"
2851
        );
28521×
        assert_eq!(from_cp.checkpoint_payload(), full.checkpoint_payload());
28531×
    }
2854
2855
    #[test]
2856
    #[should_panic(expected = "was not loaded")]
28571×
    fn a_history_only_state_never_reads_an_unloaded_graph_as_empty() {
28581×
        let f = NodeId::random();
28591×
        let mut st = State::new_history_only();
28601×
        st.add(change(
28611×
            vec![],
28621×
            vec![TreeOp::AddNode {
28631×
                node: f,
28641×
                kind: NodeKind::File,
28651×
                mode: Mode::TEXT,
28661×
                parent: NodeId::ROOT,
28671×
                name: "a".into(),
28681×
            }],
28691×
            vec![Edit::Insert {
28701×
                node: f,
28711×
                after: AtomRef::Start,
28721×
                before: AtomRef::End,
28731×
                lines: vec![b"a\n".to_vec()].into(),
28741×
            }],
2875
        ))
28761×
        .unwrap();
28771×
        st.render_file(&f);
28781×
    }
2879
2880
    #[test]
2881
    #[should_panic(expected = "needs every file graph")]
28821×
    fn materializing_a_history_only_state_is_refused() {
28831×
        let f = NodeId::random();
28841×
        let mut st = State::new_history_only();
28851×
        st.add(change(
28861×
            vec![],
28871×
            vec![TreeOp::AddNode {
28881×
                node: f,
28891×
                kind: NodeKind::File,
28901×
                mode: Mode::TEXT,
28911×
                parent: NodeId::ROOT,
28921×
                name: "a".into(),
28931×
            }],
28941×
            vec![Edit::Insert {
28951×
                node: f,
28961×
                after: AtomRef::Start,
28971×
                before: AtomRef::End,
28981×
                lines: vec![b"a\n".to_vec()].into(),
28991×
            }],
2900
        ))
29011×
        .unwrap();
29021×
        st.materialize();
29031×
    }
2904
2905
    /// Version 2 payloads (spec §11): small graphs inline, large ones by
2906
    /// blob; the state read back is the same as from version 1.
2907
    #[test]
29081×
    fn checkpoint_index_round_trip_matches_version_1() {
29091×
        let mut st = State::new();
29101×
        let small = NodeId::random();
29111×
        let big = NodeId::random();
29121×
        let add = |node: NodeId, name: &str| TreeOp::AddNode {
29132×
            node,
29142×
            kind: NodeKind::File,
2915
            mode: Mode::TEXT,
2916
            parent: NodeId::ROOT,
29172×
            name: name.into(),
29182×
        };
29191×
        let big_lines: crate::lines::Lines = (0..400)
2920400×
            .map(|i| format!("line {i} of a long file\n").into_bytes())
29211×
            .collect();
29221×
        let h1 = st
29231×
            .add(change(
29241×
                vec![],
29251×
                vec![add(small, "small"), add(big, "big")],
29261×
                vec![
29271×
                    Edit::Insert {
29281×
                        node: small,
29291×
                        after: AtomRef::Start,
29301×
                        before: AtomRef::End,
29311×
                        lines: vec![b"x\n".to_vec()].into(),
29321×
                    },
29331×
                    Edit::Insert {
29341×
                        node: big,
29351×
                        after: AtomRef::Start,
29361×
                        before: AtomRef::End,
29371×
                        lines: big_lines,
29381×
                    },
2939
                ],
2940
            ))
29411×
            .unwrap();
29421×
        st.add(change(
29431×
            vec![h1],
29441×
            vec![],
29451×
            vec![Edit::Delete {
29461×
                node: big,
29471×
                atoms: vec![AtomRange {
29481×
                    change: h1,
29491×
                    from: 3,
29501×
                    to: 7,
29511×
                }],
29521×
            }],
2953
        ))
29541×
        .unwrap();
2955
29561×
        let graphs = st.checkpoint_file_graphs();
29571×
        let mut blobs: HashMap<ObjectRef, Vec<u8>> = HashMap::new();
29581×
        let index = st.checkpoint_index(&graphs, |enc| {
29591×
            let r = crate::hash::hash_object(crate::hash::ObjectType::Blob, enc);
29601×
            blobs.insert(r, enc.to_vec());
29611×
            r
29621×
        });
29631×
        assert_eq!(blobs.len(), 1, "only the large graph is a blob of its own");
29641×
        assert_eq!(
29651×
            State::checkpoint_graph_refs(&index).unwrap(),
29661×
            blobs.keys().copied().collect::<Vec<_>>()
2967
        );
29681×
        assert!(
29691×
            State::checkpoint_graph_refs(&st.checkpoint_payload())
29701×
                .unwrap()
29711×
                .is_empty(),
2972
            "version 1 names none"
2973
        );
29741×
        assert!(index.len() < st.checkpoint_payload().len());
2975
29761×
        let back = State::from_checkpoint(&index, |refs| {
29771×
            refs.iter()
29781×
                .map(|r| {
29791×
                    blobs
29801×
                        .get(r)
29811×
                        .cloned()
29821×
                        .ok_or_else(|| ObjectError::Schema("missing".into()))
29831×
                })
29841×
                .collect()
29851×
        })
29861×
        .unwrap();
29871×
        assert_eq!(
29881×
            back.checkpoint_payload(),
29891×
            st.checkpoint_payload(),
2990
            "the same state as version 1"
2991
        );
29921×
        assert_eq!(
29931×
            back.materialize().files["big"].content,
29941×
            st.materialize().files["big"].content
2995
        );
2996
        // Deterministic: the same state gives the same index.
29971×
        assert_eq!(
29981×
            back.checkpoint_index(&back.checkpoint_file_graphs(), |enc| {
29991×
                crate::hash::hash_object(crate::hash::ObjectType::Blob, enc)
30001×
            }),
3001
            index
3002
        );
3003
        // A missing graph blob is an error, not an empty file.
30041×
        assert!(
30051×
            State::from_checkpoint(&index, |_| Err(ObjectError::Schema("gone".into()))).is_err()
3006
        );
30071×
        assert!(
30081×
            State::from_checkpoint_payload(&index).is_err(),
3009
            "version 2 needs its blobs"
3010
        );
3011
        // Version 1 payloads still read, and unknown versions do not.
30121×
        assert!(State::from_checkpoint(&st.checkpoint_payload(), |_| unreachable!()).is_ok());
30131×
        let mut v3 = cbor::decode(&index).unwrap();
30141×
        if let Value::Map(m) = &mut v3 {
30151×
            m.insert(0, Value::Uint(3));
30161×
        }
30171×
        assert!(State::from_checkpoint(&cbor::encode(&v3), |_| unreachable!()).is_err());
30181×
    }
3019
3020
    #[test]
30211×
    fn maximal_elements() {
30221×
        let mut st = State::new();
30231×
        let h1 = st.add(change(vec![], vec![], vec![])).unwrap();
30241×
        let h2 = st.add(change(vec![h1], vec![], vec![])).unwrap();
30251×
        let h3 = st.add(change(vec![], vec![], vec![])).unwrap();
30261×
        let mut m = st.maximal(&[h1, h2, h3]);
30271×
        m.sort();
30281×
        let mut want = vec![h2, h3];
30291×
        want.sort();
30301×
        assert_eq!(m, want);
30311×
        assert_eq!(st.heads(), want);
30321×
    }
3033
3034
    /// The bounded walk of `depends_on` agrees with the closure on random
3035
    /// graphs, and the memo stays within its bound.
3036
    #[test]
30371×
    fn depends_on_matches_the_closure() {
3038
        use rand::{Rng, SeedableRng};
30391×
        let mut rng = rand::rngs::StdRng::seed_from_u64(11);
30401×
        for _ in 0..10 {
304110×
            let mut st = State::new();
304210×
            let mut all: Vec<ChangeHash> = Vec::new();
304310×
            for _ in 0..80 {
3044800×
                let deps: Vec<ChangeHash> =
304531600×
                    all.iter().copied().filter(|_| rng.gen_bool(0.06)).collect();
3046800×
                all.push(st.add(change(deps, vec![], vec![])).unwrap());
3047
            }
304810×
            let unknown = ObjectRef([0xee; 32]);
3049800×
            for a in &all {
3050800×
                let want = st.closure(a).clone();
3051800×
                st.closure.write().unwrap().sets.clear();
3052800×
                st.closure.write().unwrap().held = 0;
305364800×
                for b in all.iter().chain([&unknown]) {
305464800×
                    assert_eq!(
305564800×
                        st.depends_on(a, b),
305664800×
                        a == b || want.contains(b),
3057
                        "{a:?} on {b:?}"
3058
                    );
3059
                }
3060
            }
306110×
            assert!(st.closure.write().unwrap().held <= CLOSURE_MEMO_HASHES);
3062
        }
30631×
    }
3064
3065
    /// The one-walk `maximal` agrees with the definition (not in the
3066
    /// closure of another element) on random graphs, with repeated and
3067
    /// unknown hashes among the input.
3068
    #[test]
30691×
    fn maximal_matches_its_definition() {
3070
        use rand::{Rng, SeedableRng};
30711×
        let mut rng = rand::rngs::StdRng::seed_from_u64(7);
30721×
        for _ in 0..20 {
307320×
            let mut st = State::new();
307420×
            let mut all: Vec<ChangeHash> = Vec::new();
307520×
            for _ in 0..60 {
30761200×
                let deps: Vec<ChangeHash> =
307735400×
                    all.iter().copied().filter(|_| rng.gen_bool(0.05)).collect();
30781200×
                all.push(st.add(change(deps, vec![], vec![])).unwrap());
3079
            }
308020×
            let mut pick: Vec<ChangeHash> =
30811200×
                all.iter().copied().filter(|_| rng.gen_bool(0.4)).collect();
308220×
            pick.extend(pick.clone().into_iter().take(3));
308320×
            pick.push(ObjectRef([0xee; 32]));
308420×
            let set: BTreeSet<ChangeHash> = pick.iter().copied().collect();
308520×
            let want: Vec<ChangeHash> = set
308620×
                .iter()
30878207×
                .filter(|h| !set.iter().any(|o| o != *h && st.closure(o).contains(h)))
308820×
                .copied()
308920×
                .collect();
309020×
            assert_eq!(st.maximal(&pick), want);
309120×
            assert_eq!(st.maximal(&all), st.heads());
3092
        }
30931×
    }
3094
}