Move all FoldMap query methods to FoldMapSnapshot

Nathan Sobo and Max Brunsfeld created

Co-Authored-By: Max Brunsfeld <maxbrunsfeld@gmail.com>

Change summary

zed/src/editor.rs                      |   6 
zed/src/editor/buffer.rs               | 144 ++++++++---
zed/src/editor/buffer/anchor.rs        |  13 
zed/src/editor/buffer/rope.rs          |   6 
zed/src/editor/display_map.rs          |  37 +-
zed/src/editor/display_map/fold_map.rs | 330 +++++++++++++--------------
6 files changed, 295 insertions(+), 241 deletions(-)

Detailed changes

zed/src/editor.rs 🔗

@@ -958,6 +958,7 @@ impl Editor {
 
         let app = cx.as_ref();
         let buffer = self.buffer.read(cx);
+        let display_map = self.display_map.snapshot(cx.as_ref());
 
         let mut edits = Vec::new();
         let mut new_selection_ranges = Vec::new();
@@ -1013,7 +1014,7 @@ impl Editor {
 
                 // Move folds up.
                 old_folds.push(start..end);
-                for fold in self.display_map.folds_in_range(start..end, app) {
+                for fold in display_map.folds_in_range(start..end) {
                     let mut start = fold.start.to_point(buffer);
                     let mut end = fold.end.to_point(buffer);
                     start.row -= row_delta;
@@ -1042,6 +1043,7 @@ impl Editor {
 
         let app = cx.as_ref();
         let buffer = self.buffer.read(cx);
+        let display_map = self.display_map.snapshot(cx.as_ref());
 
         let mut edits = Vec::new();
         let mut new_selection_ranges = Vec::new();
@@ -1101,7 +1103,7 @@ impl Editor {
 
                 // Move folds down.
                 old_folds.push(start..end);
-                for fold in self.display_map.folds_in_range(start..end, app) {
+                for fold in display_map.folds_in_range(start..end) {
                     let mut start = fold.start.to_point(buffer);
                     let mut end = fold.end.to_point(buffer);
                     start.row += row_delta;

zed/src/editor/buffer.rs 🔗

@@ -600,8 +600,9 @@ impl Buffer {
 
     pub fn snapshot(&self) -> Snapshot {
         Snapshot {
-            text: self.visible_text.clone(),
+            visible_text: self.visible_text.clone(),
             fragments: self.fragments.clone(),
+            version: self.version.clone(),
             tree: self.syntax_tree(),
             language: self.language.clone(),
             query_cursor: QueryCursorHandle::new(),
@@ -1007,7 +1008,7 @@ impl Buffer {
     }
 
     pub fn len(&self) -> usize {
-        self.fragments.extent::<usize>(&None)
+        self.content().len()
     }
 
     pub fn line_len(&self, row: u32) -> u32 {
@@ -1874,39 +1875,11 @@ impl Buffer {
     }
 
     pub fn anchor_at<T: ToOffset>(&self, position: T, bias: Bias) -> Anchor {
-        let offset = position.to_offset(self);
-        let max_offset = self.len();
-        assert!(offset <= max_offset, "offset is out of range");
-        let mut cursor = self.fragments.cursor::<usize, FragmentTextSummary>();
-        cursor.seek(&offset, bias, &None);
-        Anchor {
-            offset: offset + cursor.sum_start().deleted,
-            bias,
-            version: self.version(),
-        }
-    }
-
-    fn full_offset_for_anchor(&self, anchor: &Anchor) -> usize {
-        let cx = Some(anchor.version.clone());
-        let mut cursor = self
-            .fragments
-            .cursor::<VersionedOffset, FragmentTextSummary>();
-        cursor.seek(&VersionedOffset::Offset(anchor.offset), anchor.bias, &cx);
-        let overshoot = if cursor.item().is_some() {
-            anchor.offset - cursor.seek_start().offset()
-        } else {
-            0
-        };
-        let summary = cursor.sum_start();
-        summary.visible + summary.deleted + overshoot
+        self.content().anchor_at(position, bias)
     }
 
     pub fn point_for_offset(&self, offset: usize) -> Result<Point> {
-        if offset <= self.len() {
-            Ok(self.content().text_summary_for_range(0..offset).lines)
-        } else {
-            Err(anyhow!("offset out of bounds"))
-        }
+        self.content().point_for_offset(offset)
     }
 
     pub fn clip_point(&self, point: Point, bias: Bias) -> Point {
@@ -1949,8 +1922,9 @@ impl Clone for Buffer {
 }
 
 pub struct Snapshot {
-    text: Rope,
+    visible_text: Rope,
     fragments: SumTree<Fragment>,
+    version: time::Global,
     tree: Option<Tree>,
     language: Option<Arc<Language>>,
     query_cursor: QueryCursorHandle,
@@ -1958,24 +1932,32 @@ pub struct Snapshot {
 
 impl Snapshot {
     pub fn len(&self) -> usize {
-        self.text.len()
+        self.visible_text.len()
     }
 
     pub fn text(&self) -> Rope {
-        self.text.clone()
+        self.visible_text.clone()
+    }
+
+    pub fn text_summary(&self) -> TextSummary {
+        self.visible_text.summary()
+    }
+
+    pub fn max_point(&self) -> Point {
+        self.visible_text.max_point()
     }
 
     pub fn text_for_range(&self, range: Range<usize>) -> Chunks {
-        self.text.chunks_in_range(range)
+        self.visible_text.chunks_in_range(range)
     }
 
     pub fn highlighted_text_for_range(&mut self, range: Range<usize>) -> HighlightedChunks {
-        let chunks = self.text.chunks_in_range(range.clone());
+        let chunks = self.visible_text.chunks_in_range(range.clone());
         if let Some((language, tree)) = self.language.as_ref().zip(self.tree.as_ref()) {
             let captures = self.query_cursor.set_byte_range(range.clone()).captures(
                 &language.highlight_query,
                 tree.root_node(),
-                TextProvider(&self.text),
+                TextProvider(&self.visible_text),
             );
 
             HighlightedChunks {
@@ -1997,33 +1979,55 @@ impl Snapshot {
         }
     }
 
+    pub fn text_summary_for_range(&self, range: Range<usize>) -> TextSummary {
+        self.content().text_summary_for_range(range)
+    }
+
+    pub fn point_for_offset(&self, offset: usize) -> Result<Point> {
+        self.content().point_for_offset(offset)
+    }
+
     pub fn clip_offset(&self, offset: usize, bias: Bias) -> usize {
-        self.text.clip_offset(offset, bias)
+        self.visible_text.clip_offset(offset, bias)
     }
 
     pub fn clip_point(&self, point: Point, bias: Bias) -> Point {
-        self.text.clip_point(point, bias)
+        self.visible_text.clip_point(point, bias)
     }
 
     pub fn to_offset(&self, point: Point) -> usize {
-        self.text.to_offset(point)
+        self.visible_text.to_offset(point)
     }
 
     pub fn to_point(&self, offset: usize) -> Point {
-        self.text.to_point(offset)
+        self.visible_text.to_point(offset)
+    }
+
+    pub fn anchor_before<T: ToOffset>(&self, position: T) -> Anchor {
+        self.content().anchor_at(position, Bias::Left)
+    }
+
+    pub fn anchor_after<T: ToOffset>(&self, position: T) -> Anchor {
+        self.content().anchor_at(position, Bias::Right)
+    }
+
+    fn content(&self) -> Content {
+        self.into()
     }
 }
 
 pub struct Content<'a> {
     visible_text: &'a Rope,
     fragments: &'a SumTree<Fragment>,
+    version: &'a time::Global,
 }
 
 impl<'a> From<&'a Snapshot> for Content<'a> {
     fn from(snapshot: &'a Snapshot) -> Self {
         Self {
-            visible_text: &snapshot.text,
+            visible_text: &snapshot.visible_text,
             fragments: &snapshot.fragments,
+            version: &snapshot.version,
         }
     }
 }
@@ -2033,6 +2037,7 @@ impl<'a> From<&'a Buffer> for Content<'a> {
         Self {
             visible_text: &buffer.visible_text,
             fragments: &buffer.fragments,
+            version: &buffer.version,
         }
     }
 }
@@ -2042,11 +2047,26 @@ impl<'a> From<&'a mut Buffer> for Content<'a> {
         Self {
             visible_text: &buffer.visible_text,
             fragments: &buffer.fragments,
+            version: &buffer.version,
+        }
+    }
+}
+
+impl<'a> From<&'a Content<'a>> for Content<'a> {
+    fn from(content: &'a Content) -> Self {
+        Self {
+            visible_text: &content.visible_text,
+            fragments: &content.fragments,
+            version: &content.version,
         }
     }
 }
 
 impl<'a> Content<'a> {
+    fn len(&self) -> usize {
+        self.fragments.extent::<usize>(&None)
+    }
+
     fn summary_for_anchor(&self, anchor: &Anchor) -> TextSummary {
         let cx = Some(anchor.version.clone());
         let mut cursor = self.fragments.cursor::<VersionedOffset, usize>();
@@ -2059,9 +2079,45 @@ impl<'a> Content<'a> {
         self.text_summary_for_range(0..*cursor.sum_start() + overshoot)
     }
 
-    pub fn text_summary_for_range(&self, range: Range<usize>) -> TextSummary {
+    fn text_summary_for_range(&self, range: Range<usize>) -> TextSummary {
         self.visible_text.cursor(range.start).summary(range.end)
     }
+
+    fn anchor_at<T: ToOffset>(&self, position: T, bias: Bias) -> Anchor {
+        let offset = position.to_offset(self);
+        let max_offset = self.len();
+        assert!(offset <= max_offset, "offset is out of range");
+        let mut cursor = self.fragments.cursor::<usize, FragmentTextSummary>();
+        cursor.seek(&offset, bias, &None);
+        Anchor {
+            offset: offset + cursor.sum_start().deleted,
+            bias,
+            version: self.version.clone(),
+        }
+    }
+
+    fn full_offset_for_anchor(&self, anchor: &Anchor) -> usize {
+        let cx = Some(anchor.version.clone());
+        let mut cursor = self
+            .fragments
+            .cursor::<VersionedOffset, FragmentTextSummary>();
+        cursor.seek(&VersionedOffset::Offset(anchor.offset), anchor.bias, &cx);
+        let overshoot = if cursor.item().is_some() {
+            anchor.offset - cursor.seek_start().offset()
+        } else {
+            0
+        };
+        let summary = cursor.sum_start();
+        summary.visible + summary.deleted + overshoot
+    }
+
+    fn point_for_offset(&self, offset: usize) -> Result<Point> {
+        if offset <= self.len() {
+            Ok(self.text_summary_for_range(0..offset).lines)
+        } else {
+            Err(anyhow!("offset out of bounds"))
+        }
+    }
 }
 
 struct RopeBuilder<'a> {

zed/src/editor/buffer/anchor.rs 🔗

@@ -1,4 +1,4 @@
-use super::Buffer;
+use super::{Buffer, Content};
 use crate::{time, util::Bias};
 use anyhow::Result;
 use std::{cmp::Ordering, ops::Range};
@@ -27,7 +27,9 @@ impl Anchor {
         }
     }
 
-    pub fn cmp(&self, other: &Anchor, buffer: &Buffer) -> Result<Ordering> {
+    pub fn cmp<'a>(&self, other: &Anchor, buffer: impl Into<Content<'a>>) -> Result<Ordering> {
+        let buffer = buffer.into();
+
         if self == other {
             return Ok(Ordering::Equal);
         }
@@ -61,12 +63,13 @@ impl Anchor {
 }
 
 pub trait AnchorRangeExt {
-    fn cmp(&self, b: &Range<Anchor>, buffer: &Buffer) -> Result<Ordering>;
+    fn cmp<'a>(&self, b: &Range<Anchor>, buffer: impl Into<Content<'a>>) -> Result<Ordering>;
 }
 
 impl AnchorRangeExt for Range<Anchor> {
-    fn cmp(&self, other: &Range<Anchor>, buffer: &Buffer) -> Result<Ordering> {
-        Ok(match self.start.cmp(&other.start, buffer)? {
+    fn cmp<'a>(&self, other: &Range<Anchor>, buffer: impl Into<Content<'a>>) -> Result<Ordering> {
+        let buffer = buffer.into();
+        Ok(match self.start.cmp(&other.start, &buffer)? {
             Ordering::Equal => other.end.cmp(&self.end, buffer)?,
             ord @ _ => ord,
         })

zed/src/editor/buffer/rope.rs 🔗

@@ -188,6 +188,12 @@ impl<'a> From<&'a str> for Rope {
     }
 }
 
+impl Into<String> for Rope {
+    fn into(self) -> String {
+        self.chunks().collect()
+    }
+}
+
 pub struct Cursor<'a> {
     rope: &'a Rope,
     chunks: sum_tree::Cursor<'a, Chunk, usize, ()>,

zed/src/editor/display_map.rs 🔗

@@ -31,17 +31,6 @@ impl DisplayMap {
         }
     }
 
-    pub fn folds_in_range<'a, T>(
-        &'a self,
-        range: Range<T>,
-        cx: &'a AppContext,
-    ) -> impl Iterator<Item = &'a Range<Anchor>>
-    where
-        T: ToOffset,
-    {
-        self.fold_map.folds_in_range(range, cx)
-    }
-
     pub fn fold<T: ToOffset>(
         &mut self,
         ranges: impl IntoIterator<Item = Range<T>>,
@@ -59,11 +48,11 @@ impl DisplayMap {
     }
 
     pub fn intersects_fold<T: ToOffset>(&self, offset: T, cx: &AppContext) -> bool {
-        self.fold_map.intersects_fold(offset, cx)
+        self.fold_map.snapshot(cx).intersects_fold(offset)
     }
 
     pub fn is_line_folded(&self, display_row: u32, cx: &AppContext) -> bool {
-        self.fold_map.is_line_folded(display_row, cx)
+        self.fold_map.snapshot(cx).is_line_folded(display_row)
     }
 
     pub fn text(&self, cx: &AppContext) -> String {
@@ -104,7 +93,7 @@ impl DisplayMap {
     }
 
     pub fn line_len(&self, row: u32, cx: &AppContext) -> u32 {
-        DisplayPoint::new(row, self.fold_map.line_len(row, cx))
+        DisplayPoint::new(row, self.fold_map.snapshot(cx).line_len(row))
             .expand_tabs(self, cx)
             .column()
     }
@@ -114,7 +103,7 @@ impl DisplayMap {
     }
 
     pub fn longest_row(&self, cx: &AppContext) -> u32 {
-        self.fold_map.longest_row(cx)
+        self.fold_map.snapshot(cx).longest_row()
     }
 
     pub fn anchor_before(&self, point: DisplayPoint, bias: Bias, cx: &AppContext) -> Anchor {
@@ -209,6 +198,16 @@ impl DisplayMapSnapshot {
         )
     }
 
+    pub fn folds_in_range<'a, T>(
+        &'a self,
+        range: Range<T>,
+    ) -> impl Iterator<Item = &'a Range<Anchor>>
+    where
+        T: ToOffset,
+    {
+        self.folds_snapshot.folds_in_range(range)
+    }
+
     fn expand_tabs(&self, mut point: DisplayPoint) -> DisplayPoint {
         let chars = self
             .folds_snapshot
@@ -260,12 +259,14 @@ impl DisplayPoint {
 
     pub fn to_buffer_point(self, map: &DisplayMap, bias: Bias, cx: &AppContext) -> Point {
         map.fold_map
-            .to_buffer_point(self.collapse_tabs(map, bias, cx), cx)
+            .snapshot(cx)
+            .to_buffer_point(self.collapse_tabs(map, bias, cx))
     }
 
     pub fn to_buffer_offset(self, map: &DisplayMap, bias: Bias, cx: &AppContext) -> usize {
         map.fold_map
-            .to_buffer_offset(self.collapse_tabs(&map, bias, cx), cx)
+            .snapshot(cx)
+            .to_buffer_offset(self.collapse_tabs(&map, bias, cx))
     }
 
     fn expand_tabs(self, map: &DisplayMap, cx: &AppContext) -> Self {
@@ -279,7 +280,7 @@ impl DisplayPoint {
 
 impl Point {
     pub fn to_display_point(self, map: &DisplayMap, cx: &AppContext) -> DisplayPoint {
-        let mut display_point = map.fold_map.to_display_point(self, cx);
+        let mut display_point = map.fold_map.snapshot(cx).to_display_point(self);
         let snapshot = map.fold_map.snapshot(cx);
         let chars = snapshot.chars_at(DisplayPoint::new(display_point.row(), 0));
         *display_point.column_mut() =

zed/src/editor/display_map/fold_map.rs 🔗

@@ -47,49 +47,11 @@ impl FoldMap {
     pub fn snapshot(&self, cx: &AppContext) -> FoldMapSnapshot {
         FoldMapSnapshot {
             transforms: self.sync(cx).clone(),
+            folds: self.folds.clone(),
             buffer: self.buffer.read(cx).snapshot(),
         }
     }
 
-    pub fn len(&self, cx: &AppContext) -> usize {
-        self.sync(cx).summary().display.bytes
-    }
-
-    pub fn line_len(&self, row: u32, cx: &AppContext) -> u32 {
-        let line_start = self.to_display_offset(DisplayPoint::new(row, 0), cx).0;
-        let line_end = if row >= self.max_point(cx).row() {
-            self.len(cx)
-        } else {
-            self.to_display_offset(DisplayPoint::new(row + 1, 0), cx).0 - 1
-        };
-        (line_end - line_start) as u32
-    }
-
-    pub fn max_point(&self, cx: &AppContext) -> DisplayPoint {
-        DisplayPoint(self.sync(cx).summary().display.lines)
-    }
-
-    pub fn longest_row(&self, cx: &AppContext) -> u32 {
-        self.sync(cx).summary().display.longest_row
-    }
-
-    pub fn folds_in_range<'a, T>(
-        &'a self,
-        range: Range<T>,
-        cx: &'a AppContext,
-    ) -> impl Iterator<Item = &'a Range<Anchor>>
-    where
-        T: ToOffset,
-    {
-        let buffer = self.buffer.read(cx);
-        let mut folds = self.intersecting_folds(range, cx);
-        iter::from_fn(move || {
-            let item = folds.item().map(|f| &f.0);
-            folds.next(buffer);
-            item
-        })
-    }
-
     pub fn fold<T: ToOffset>(
         &mut self,
         ranges: impl IntoIterator<Item = Range<T>>,
@@ -99,9 +61,9 @@ impl FoldMap {
 
         let mut edits = Vec::new();
         let mut folds = Vec::new();
-        let buffer = self.buffer.read(cx);
+        let buffer = self.buffer.read(cx).snapshot();
         for range in ranges.into_iter() {
-            let range = range.start.to_offset(buffer)..range.end.to_offset(buffer);
+            let range = range.start.to_offset(&buffer)..range.end.to_offset(&buffer);
             if range.start != range.end {
                 let fold = Fold(buffer.anchor_after(range.start)..buffer.anchor_before(range.end));
                 folds.push(fold);
@@ -113,7 +75,7 @@ impl FoldMap {
             }
         }
 
-        folds.sort_unstable_by(|a, b| sum_tree::SeekDimension::cmp(a, b, buffer));
+        folds.sort_unstable_by(|a, b| sum_tree::SeekDimension::cmp(a, b, &buffer));
         edits.sort_unstable_by(|a, b| {
             a.old_bytes
                 .start
@@ -125,10 +87,10 @@ impl FoldMap {
             let mut new_tree = SumTree::new();
             let mut cursor = self.folds.cursor::<_, ()>();
             for fold in folds {
-                new_tree.push_tree(cursor.slice(&fold, Bias::Right, buffer), buffer);
-                new_tree.push(fold, buffer);
+                new_tree.push_tree(cursor.slice(&fold, Bias::Right, &buffer), &buffer);
+                new_tree.push(fold, &buffer);
             }
-            new_tree.push_tree(cursor.suffix(buffer), buffer);
+            new_tree.push_tree(cursor.suffix(&buffer), &buffer);
             new_tree
         };
         self.apply_edits(edits, cx);
@@ -141,22 +103,23 @@ impl FoldMap {
     ) {
         let _ = self.sync(cx);
 
-        let buffer = self.buffer.read(cx);
+        let buffer = self.buffer.read(cx).snapshot();
+        let snapshot = self.snapshot(cx);
 
         let mut edits = Vec::new();
         let mut fold_ixs_to_delete = Vec::new();
         for range in ranges.into_iter() {
             // Remove intersecting folds and add their ranges to edits that are passed to apply_edits.
-            let mut folds_cursor = self.intersecting_folds(range, cx);
+            let mut folds_cursor = snapshot.intersecting_folds(range);
             while let Some(fold) = folds_cursor.item() {
-                let offset_range = fold.0.start.to_offset(buffer)..fold.0.end.to_offset(buffer);
+                let offset_range = fold.0.start.to_offset(&buffer)..fold.0.end.to_offset(&buffer);
                 edits.push(Edit {
                     old_bytes: offset_range.clone(),
                     new_bytes: offset_range,
                     ..Default::default()
                 });
                 fold_ixs_to_delete.push(*folds_cursor.start());
-                folds_cursor.next(buffer);
+                folds_cursor.next(&buffer);
             }
         }
 
@@ -173,91 +136,15 @@ impl FoldMap {
             let mut cursor = self.folds.cursor::<_, ()>();
             let mut folds = SumTree::new();
             for fold_ix in fold_ixs_to_delete {
-                folds.push_tree(cursor.slice(&fold_ix, Bias::Right, buffer), buffer);
-                cursor.next(buffer);
+                folds.push_tree(cursor.slice(&fold_ix, Bias::Right, &buffer), &buffer);
+                cursor.next(&buffer);
             }
-            folds.push_tree(cursor.suffix(buffer), buffer);
+            folds.push_tree(cursor.suffix(&buffer), &buffer);
             folds
         };
         self.apply_edits(edits, cx);
     }
 
-    fn intersecting_folds<'a, T>(
-        &self,
-        range: Range<T>,
-        cx: &'a AppContext,
-    ) -> FilterCursor<impl 'a + Fn(&FoldSummary) -> bool, Fold, usize>
-    where
-        T: ToOffset,
-    {
-        let buffer = self.buffer.read(cx);
-        let start = buffer.anchor_before(range.start.to_offset(buffer));
-        let end = buffer.anchor_after(range.end.to_offset(buffer));
-        self.folds.filter::<_, usize>(
-            move |summary| {
-                start.cmp(&summary.max_end, buffer).unwrap() == Ordering::Less
-                    && end.cmp(&summary.min_start, buffer).unwrap() == Ordering::Greater
-            },
-            buffer,
-        )
-    }
-
-    pub fn intersects_fold<T>(&self, offset: T, cx: &AppContext) -> bool
-    where
-        T: ToOffset,
-    {
-        let buffer = self.buffer.read(cx);
-        let offset = offset.to_offset(buffer);
-        let transforms = self.sync(cx);
-        let mut cursor = transforms.cursor::<usize, ()>();
-        cursor.seek(&offset, Bias::Right, &());
-        cursor.item().map_or(false, |t| t.display_text.is_some())
-    }
-
-    pub fn is_line_folded(&self, display_row: u32, cx: &AppContext) -> bool {
-        let transforms = self.sync(cx);
-        let mut cursor = transforms.cursor::<DisplayPoint, ()>();
-        cursor.seek(&DisplayPoint::new(display_row, 0), Bias::Right, &());
-        while let Some(transform) = cursor.item() {
-            if transform.display_text.is_some() {
-                return true;
-            }
-            if cursor.seek_end(&()).row() == display_row {
-                cursor.next(&())
-            } else {
-                break;
-            }
-        }
-        false
-    }
-
-    pub fn to_buffer_offset(&self, point: DisplayPoint, cx: &AppContext) -> usize {
-        self.snapshot(cx).to_buffer_offset(point)
-    }
-
-    pub fn to_display_offset(&self, point: DisplayPoint, cx: &AppContext) -> DisplayOffset {
-        self.snapshot(cx).to_display_offset(point)
-    }
-
-    pub fn to_buffer_point(&self, display_point: DisplayPoint, cx: &AppContext) -> Point {
-        let transforms = self.sync(cx);
-        let mut cursor = transforms.cursor::<DisplayPoint, Point>();
-        cursor.seek(&display_point, Bias::Right, &());
-        let overshoot = display_point.0 - cursor.seek_start().0;
-        *cursor.sum_start() + overshoot
-    }
-
-    pub fn to_display_point(&self, point: Point, cx: &AppContext) -> DisplayPoint {
-        let transforms = self.sync(cx);
-        let mut cursor = transforms.cursor::<Point, DisplayPoint>();
-        cursor.seek(&point, Bias::Right, &());
-        let overshoot = point - cursor.seek_start();
-        DisplayPoint(cmp::min(
-            cursor.sum_start().0 + overshoot,
-            cursor.end(&()).0,
-        ))
-    }
-
     fn sync(&self, cx: &AppContext) -> MutexGuard<SumTree<Transform>> {
         let buffer = self.buffer.read(cx);
         let mut edits = buffer.edits_since(self.last_sync.lock().clone()).peekable();
@@ -269,7 +156,7 @@ impl FoldMap {
     }
 
     fn apply_edits(&self, edits: impl IntoIterator<Item = Edit>, cx: &AppContext) {
-        let buffer = self.buffer.read(cx);
+        let buffer = self.buffer.read(cx).snapshot();
         let mut edits = edits.into_iter().peekable();
 
         let mut new_transforms = SumTree::new();
@@ -312,13 +199,17 @@ impl FoldMap {
 
             let anchor = buffer.anchor_before(edit.new_bytes.start);
             let mut folds_cursor = self.folds.cursor::<_, ()>();
-            folds_cursor.seek(&Fold(anchor..Anchor::max()), Bias::Left, buffer);
-            let mut folds = iter::from_fn(move || {
-                let item = folds_cursor
-                    .item()
-                    .map(|f| f.0.start.to_offset(buffer)..f.0.end.to_offset(buffer));
-                folds_cursor.next(buffer);
-                item
+            folds_cursor.seek(&Fold(anchor..Anchor::max()), Bias::Left, &buffer);
+
+            let mut folds = iter::from_fn({
+                let buffer = &buffer;
+                move || {
+                    let item = folds_cursor
+                        .item()
+                        .map(|f| f.0.start.to_offset(buffer)..f.0.end.to_offset(buffer));
+                    folds_cursor.next(buffer);
+                    item
+                }
             })
             .peekable();
 
@@ -418,10 +309,25 @@ impl FoldMap {
 
 pub struct FoldMapSnapshot {
     transforms: SumTree<Transform>,
+    folds: SumTree<Fold>,
     buffer: buffer::Snapshot,
 }
 
 impl FoldMapSnapshot {
+    pub fn len(&self) -> usize {
+        self.transforms.summary().display.bytes
+    }
+
+    pub fn line_len(&self, row: u32) -> u32 {
+        let line_start = self.to_display_offset(DisplayPoint::new(row, 0)).0;
+        let line_end = if row >= self.max_point().row() {
+            self.len()
+        } else {
+            self.to_display_offset(DisplayPoint::new(row + 1, 0)).0 - 1
+        };
+        (line_end - line_start) as u32
+    }
+
     pub fn buffer_rows(&self, start_row: u32) -> BufferRows {
         if start_row > self.transforms.summary().display.lines.row {
             panic!("invalid display row {}", start_row);
@@ -441,6 +347,71 @@ impl FoldMapSnapshot {
         DisplayPoint(self.transforms.summary().display.lines)
     }
 
+    pub fn longest_row(&self) -> u32 {
+        self.transforms.summary().display.longest_row
+    }
+
+    pub fn folds_in_range<'a, T>(
+        &'a self,
+        range: Range<T>,
+    ) -> impl Iterator<Item = &'a Range<Anchor>>
+    where
+        T: ToOffset,
+    {
+        let mut folds = self.intersecting_folds(range);
+        iter::from_fn(move || {
+            let item = folds.item().map(|f| &f.0);
+            folds.next(&self.buffer);
+            item
+        })
+    }
+
+    fn intersecting_folds<'a, T>(
+        &'a self,
+        range: Range<T>,
+    ) -> FilterCursor<impl 'a + Fn(&FoldSummary) -> bool, Fold, usize>
+    where
+        T: ToOffset,
+    {
+        let start = self
+            .buffer
+            .anchor_before(range.start.to_offset(&self.buffer));
+        let end = self.buffer.anchor_after(range.end.to_offset(&self.buffer));
+        self.folds.filter::<_, usize>(
+            move |summary| {
+                start.cmp(&summary.max_end, &self.buffer).unwrap() == Ordering::Less
+                    && end.cmp(&summary.min_start, &self.buffer).unwrap() == Ordering::Greater
+            },
+            &self.buffer,
+        )
+    }
+
+    pub fn intersects_fold<T>(&self, offset: T) -> bool
+    where
+        T: ToOffset,
+    {
+        let offset = offset.to_offset(&self.buffer);
+        let mut cursor = self.transforms.cursor::<usize, ()>();
+        cursor.seek(&offset, Bias::Right, &());
+        cursor.item().map_or(false, |t| t.display_text.is_some())
+    }
+
+    pub fn is_line_folded(&self, display_row: u32) -> bool {
+        let mut cursor = self.transforms.cursor::<DisplayPoint, ()>();
+        cursor.seek(&DisplayPoint::new(display_row, 0), Bias::Right, &());
+        while let Some(transform) = cursor.item() {
+            if transform.display_text.is_some() {
+                return true;
+            }
+            if cursor.seek_end(&()).row() == display_row {
+                cursor.next(&())
+            } else {
+                break;
+            }
+        }
+        false
+    }
+
     pub fn chunks_at(&self, offset: DisplayOffset) -> Chunks {
         let mut transform_cursor = self.transforms.cursor::<DisplayOffset, usize>();
         transform_cursor.seek(&offset, Bias::Right, &());
@@ -502,6 +473,23 @@ impl FoldMapSnapshot {
         self.buffer.to_offset(*cursor.sum_start() + overshoot)
     }
 
+    pub fn to_buffer_point(&self, display_point: DisplayPoint) -> Point {
+        let mut cursor = self.transforms.cursor::<DisplayPoint, Point>();
+        cursor.seek(&display_point, Bias::Right, &());
+        let overshoot = display_point.0 - cursor.seek_start().0;
+        *cursor.sum_start() + overshoot
+    }
+
+    pub fn to_display_point(&self, point: Point) -> DisplayPoint {
+        let mut cursor = self.transforms.cursor::<Point, DisplayPoint>();
+        cursor.seek(&point, Bias::Right, &());
+        let overshoot = point - cursor.seek_start();
+        DisplayPoint(cmp::min(
+            cursor.sum_start().0 + overshoot,
+            cursor.end(&()).0,
+        ))
+    }
+
     #[cfg(test)]
     pub fn clip_offset(&self, offset: DisplayOffset, bias: Bias) -> DisplayOffset {
         let mut cursor = self.transforms.cursor::<DisplayOffset, usize>();
@@ -635,9 +623,9 @@ impl Default for FoldSummary {
 }
 
 impl sum_tree::Summary for FoldSummary {
-    type Context = Buffer;
+    type Context = buffer::Snapshot;
 
-    fn add_summary(&mut self, other: &Self, buffer: &Buffer) {
+    fn add_summary(&mut self, other: &Self, buffer: &buffer::Snapshot) {
         if other.min_start.cmp(&self.min_start, buffer).unwrap() == Ordering::Less {
             self.min_start = other.min_start.clone();
         }
@@ -660,20 +648,20 @@ impl sum_tree::Summary for FoldSummary {
 }
 
 impl<'a> sum_tree::Dimension<'a, FoldSummary> for Fold {
-    fn add_summary(&mut self, summary: &'a FoldSummary, _: &Buffer) {
+    fn add_summary(&mut self, summary: &'a FoldSummary, _: &buffer::Snapshot) {
         self.0.start = summary.start.clone();
         self.0.end = summary.end.clone();
     }
 }
 
 impl<'a> sum_tree::SeekDimension<'a, FoldSummary> for Fold {
-    fn cmp(&self, other: &Self, buffer: &Buffer) -> Ordering {
+    fn cmp(&self, other: &Self, buffer: &buffer::Snapshot) -> Ordering {
         self.0.cmp(&other.0, buffer).unwrap()
     }
 }
 
 impl<'a> sum_tree::Dimension<'a, FoldSummary> for usize {
-    fn add_summary(&mut self, summary: &'a FoldSummary, _: &Buffer) {
+    fn add_summary(&mut self, summary: &'a FoldSummary, _: &buffer::Snapshot) {
         *self += summary.count;
     }
 }
@@ -986,7 +974,8 @@ mod tests {
             cx.as_ref(),
         );
         let fold_ranges = map
-            .folds_in_range(Point::new(1, 0)..Point::new(1, 3), cx.as_ref())
+            .snapshot(cx.as_ref())
+            .folds_in_range(Point::new(1, 0)..Point::new(1, 3))
             .map(|fold| fold.start.to_point(buffer)..fold.end.to_point(buffer))
             .collect::<Vec<_>>();
         assert_eq!(
@@ -1067,8 +1056,8 @@ mod tests {
                 }
                 map.check_invariants(cx.as_ref());
 
-                let buffer = map.buffer.read(cx);
-                let mut expected_text = buffer.text();
+                let buffer = map.buffer.read(cx).snapshot();
+                let mut expected_text: String = buffer.text().into();
                 let mut expected_buffer_rows = Vec::new();
                 let mut next_row = buffer.max_point().row;
                 for fold_range in map.merged_fold_ranges(cx.as_ref()).into_iter().rev() {
@@ -1084,12 +1073,13 @@ mod tests {
 
                 assert_eq!(map.text(cx.as_ref()), expected_text);
 
+                let snapshot = map.snapshot(cx.as_ref());
                 for (display_row, line) in expected_text.lines().enumerate() {
-                    let line_len = map.line_len(display_row as u32, cx.as_ref());
+                    let line_len = snapshot.line_len(display_row as u32);
                     assert_eq!(line_len, line.len() as u32);
                 }
 
-                let longest_row = map.longest_row(cx.as_ref());
+                let longest_row = snapshot.longest_row();
                 let longest_char_column = expected_text
                     .split('\n')
                     .nth(longest_row as usize)
@@ -1100,22 +1090,22 @@ mod tests {
                 let mut display_offset = DisplayOffset(0);
                 let mut char_column = 0;
                 for c in expected_text.chars() {
-                    let buffer_point = map.to_buffer_point(display_point, cx.as_ref());
-                    let buffer_offset = buffer_point.to_offset(buffer);
+                    let buffer_point = snapshot.to_buffer_point(display_point);
+                    let buffer_offset = buffer_point.to_offset(&buffer);
                     assert_eq!(
-                        map.to_display_point(buffer_point, cx.as_ref()),
+                        snapshot.to_display_point(buffer_point),
                         display_point,
                         "to_display_point({:?})",
                         buffer_point,
                     );
                     assert_eq!(
-                        map.to_buffer_offset(display_point, cx.as_ref()),
+                        snapshot.to_buffer_offset(display_point),
                         buffer_offset,
                         "to_buffer_offset({:?})",
                         display_point,
                     );
                     assert_eq!(
-                        map.to_display_offset(display_point, cx.as_ref()),
+                        snapshot.to_display_offset(display_point),
                         display_offset,
                         "to_display_offset({:?})",
                         display_point,
@@ -1141,35 +1131,30 @@ mod tests {
                     }
                 }
 
+                let snapshot = map.snapshot(cx.as_ref());
                 for _ in 0..5 {
-                    let offset = map.snapshot(cx.as_ref()).clip_offset(
-                        DisplayOffset(rng.gen_range(0..=map.len(cx.as_ref()))),
+                    let offset = snapshot.clip_offset(
+                        DisplayOffset(rng.gen_range(0..=snapshot.len())),
                         Bias::Right,
                     );
                     assert_eq!(
-                        map.snapshot(cx.as_ref())
-                            .chunks_at(offset)
-                            .collect::<String>(),
+                        snapshot.chunks_at(offset).collect::<String>(),
                         &expected_text[offset.0..],
                     );
                 }
 
                 for (idx, buffer_row) in expected_buffer_rows.iter().enumerate() {
-                    let display_row = map
-                        .to_display_point(Point::new(*buffer_row, 0), cx.as_ref())
-                        .row();
+                    let display_row = snapshot.to_display_point(Point::new(*buffer_row, 0)).row();
                     assert_eq!(
-                        map.snapshot(cx.as_ref())
-                            .buffer_rows(display_row)
-                            .collect::<Vec<_>>(),
+                        snapshot.buffer_rows(display_row).collect::<Vec<_>>(),
                         expected_buffer_rows[idx..],
                     );
                 }
 
                 for fold_range in map.merged_fold_ranges(cx.as_ref()) {
                     let display_point =
-                        map.to_display_point(fold_range.start.to_point(buffer), cx.as_ref());
-                    assert!(map.is_line_folded(display_point.row(), cx.as_ref()));
+                        snapshot.to_display_point(fold_range.start.to_point(&buffer));
+                    assert!(snapshot.is_line_folded(display_point.row()));
                 }
 
                 for _ in 0..5 {
@@ -1177,19 +1162,20 @@ mod tests {
                     let start = buffer.clip_offset(rng.gen_range(0..=end), Left);
                     let expected_folds = map
                         .folds
-                        .items(buffer)
+                        .items(&buffer)
                         .into_iter()
                         .filter(|fold| {
                             let start = buffer.anchor_before(start);
                             let end = buffer.anchor_after(end);
-                            start.cmp(&fold.0.end, buffer).unwrap() == Ordering::Less
-                                && end.cmp(&fold.0.start, buffer).unwrap() == Ordering::Greater
+                            start.cmp(&fold.0.end, &buffer).unwrap() == Ordering::Less
+                                && end.cmp(&fold.0.start, &buffer).unwrap() == Ordering::Greater
                         })
                         .map(|fold| fold.0)
                         .collect::<Vec<_>>();
 
                     assert_eq!(
-                        map.folds_in_range(start..end, cx.as_ref())
+                        snapshot
+                            .folds_in_range(start..end)
                             .cloned()
                             .collect::<Vec<_>>(),
                         expected_folds
@@ -1231,13 +1217,13 @@ mod tests {
         }
 
         fn merged_fold_ranges(&self, cx: &AppContext) -> Vec<Range<usize>> {
-            let buffer = self.buffer.read(cx);
-            let mut folds = self.folds.items(buffer);
+            let buffer = self.buffer.read(cx).snapshot();
+            let mut folds = self.folds.items(&buffer);
             // Ensure sorting doesn't change how folds get merged and displayed.
-            folds.sort_by(|a, b| a.0.cmp(&b.0, buffer).unwrap());
+            folds.sort_by(|a, b| a.0.cmp(&b.0, &buffer).unwrap());
             let mut fold_ranges = folds
                 .iter()
-                .map(|fold| fold.0.start.to_offset(buffer)..fold.0.end.to_offset(buffer))
+                .map(|fold| fold.0.start.to_offset(&buffer)..fold.0.end.to_offset(&buffer))
                 .peekable();
 
             let mut merged_ranges = Vec::new();