Merge pull request #1204 from zed-industries/accurate-is-dirty

Determine `Buffer::is_dirty` based on the rope's fingerprint
This commit is contained in:
Antonio Scandurra 2022-06-17 16:08:07 +02:00 committed by GitHub
commit ae2273b40a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
15 changed files with 220 additions and 85 deletions

20
Cargo.lock generated
View file

@ -561,6 +561,18 @@ dependencies = [
"workspace", "workspace",
] ]
[[package]]
name = "bromberg_sl2"
version = "0.6.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2ed88064f69518b7e3ea50ecfc1b61d43f19248618a377b95ae5c8b611134d4d"
dependencies = [
"digest 0.9.0",
"lazy_static",
"rayon",
"seq-macro",
]
[[package]] [[package]]
name = "bstr" name = "bstr"
version = "0.2.17" version = "0.2.17"
@ -4156,6 +4168,12 @@ dependencies = [
"pest", "pest",
] ]
[[package]]
name = "seq-macro"
version = "0.2.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5a9f47faea3cad316faa914d013d24f471cd90bfca1a0c70f05a3f42c6441e99"
[[package]] [[package]]
name = "serde" name = "serde"
version = "1.0.137" version = "1.0.137"
@ -4806,9 +4824,11 @@ version = "0.1.0"
dependencies = [ dependencies = [
"anyhow", "anyhow",
"arrayvec 0.7.2", "arrayvec 0.7.2",
"bromberg_sl2",
"clock", "clock",
"collections", "collections",
"ctor", "ctor",
"digest 0.9.0",
"env_logger", "env_logger",
"gpui", "gpui",
"lazy_static", "lazy_static",

View file

@ -5364,7 +5364,7 @@ impl TestClient {
(buffer.version(), buffer.save(cx)) (buffer.version(), buffer.save(cx))
}); });
let save = cx.background().spawn(async move { let save = cx.background().spawn(async move {
let (saved_version, _) = save let (saved_version, _, _) = save
.await .await
.map_err(|err| anyhow!("save request failed: {:?}", err))?; .map_err(|err| anyhow!("save request failed: {:?}", err))?;
assert!(saved_version.observed_all(&requested_version)); assert!(saved_version.observed_all(&requested_version));

View file

@ -568,7 +568,10 @@ impl workspace::Item for ProjectDiagnosticsEditor {
} }
fn should_update_tab_on_event(event: &Event) -> bool { fn should_update_tab_on_event(event: &Event) -> bool {
matches!(event, Event::Saved | Event::Dirtied | Event::TitleChanged) matches!(
event,
Event::Saved | Event::DirtyChanged | Event::TitleChanged
)
} }
fn set_nav_history(&mut self, nav_history: ItemNavHistory, cx: &mut ViewContext<Self>) { fn set_nav_history(&mut self, nav_history: ItemNavHistory, cx: &mut ViewContext<Self>) {

View file

@ -5,7 +5,7 @@ use crate::{
}; };
use collections::BTreeMap; use collections::BTreeMap;
use gpui::fonts::HighlightStyle; use gpui::fonts::HighlightStyle;
use language::{Chunk, Edit, Point, PointUtf16, TextSummary}; use language::{Chunk, Edit, Point, TextSummary};
use parking_lot::Mutex; use parking_lot::Mutex;
use std::{ use std::{
any::TypeId, any::TypeId,
@ -370,22 +370,10 @@ impl FoldMap {
if fold.end > fold.start { if fold.end > fold.start {
let output_text = ""; let output_text = "";
let chars = output_text.chars().count() as u32;
let lines = Point::new(0, output_text.len() as u32);
let lines_utf16 =
PointUtf16::new(0, output_text.encode_utf16().count() as u32);
new_transforms.push( new_transforms.push(
Transform { Transform {
summary: TransformSummary { summary: TransformSummary {
output: TextSummary { output: TextSummary::from(output_text),
bytes: output_text.len(),
lines,
lines_utf16,
first_line_chars: chars,
last_line_chars: chars,
longest_row: 0,
longest_row_chars: chars,
},
input: new_buffer.text_summary_for_range(fold.start..fold.end), input: new_buffer.text_summary_for_range(fold.start..fold.end),
}, },
output_text: Some(output_text), output_text: Some(output_text),

View file

@ -5508,7 +5508,7 @@ impl Editor {
cx.emit(Event::BufferEdited); cx.emit(Event::BufferEdited);
} }
language::Event::Reparsed => cx.emit(Event::Reparsed), language::Event::Reparsed => cx.emit(Event::Reparsed),
language::Event::Dirtied => cx.emit(Event::Dirtied), language::Event::DirtyChanged => cx.emit(Event::DirtyChanged),
language::Event::Saved => cx.emit(Event::Saved), language::Event::Saved => cx.emit(Event::Saved),
language::Event::FileHandleChanged => cx.emit(Event::TitleChanged), language::Event::FileHandleChanged => cx.emit(Event::TitleChanged),
language::Event::Reloaded => cx.emit(Event::TitleChanged), language::Event::Reloaded => cx.emit(Event::TitleChanged),
@ -5665,7 +5665,7 @@ pub enum Event {
Edited, Edited,
Reparsed, Reparsed,
Blurred, Blurred,
Dirtied, DirtyChanged,
Saved, Saved,
TitleChanged, TitleChanged,
SelectionsChanged { local: bool }, SelectionsChanged { local: bool },
@ -6181,7 +6181,10 @@ mod tests {
let events = events.clone(); let events = events.clone();
|cx| { |cx| {
cx.subscribe(&cx.handle(), move |_, _, event, _| { cx.subscribe(&cx.handle(), move |_, _, event, _| {
if matches!(event, Event::Edited | Event::BufferEdited | Event::Dirtied) { if matches!(
event,
Event::Edited | Event::BufferEdited | Event::DirtyChanged
) {
events.borrow_mut().push(("editor1", *event)); events.borrow_mut().push(("editor1", *event));
} }
}) })
@ -6193,7 +6196,10 @@ mod tests {
let events = events.clone(); let events = events.clone();
|cx| { |cx| {
cx.subscribe(&cx.handle(), move |_, _, event, _| { cx.subscribe(&cx.handle(), move |_, _, event, _| {
if matches!(event, Event::Edited | Event::BufferEdited | Event::Dirtied) { if matches!(
event,
Event::Edited | Event::BufferEdited | Event::DirtyChanged
) {
events.borrow_mut().push(("editor2", *event)); events.borrow_mut().push(("editor2", *event));
} }
}) })
@ -6211,8 +6217,8 @@ mod tests {
("editor1", Event::Edited), ("editor1", Event::Edited),
("editor1", Event::BufferEdited), ("editor1", Event::BufferEdited),
("editor2", Event::BufferEdited), ("editor2", Event::BufferEdited),
("editor1", Event::Dirtied), ("editor1", Event::DirtyChanged),
("editor2", Event::Dirtied) ("editor2", Event::DirtyChanged)
] ]
); );
@ -6235,6 +6241,8 @@ mod tests {
("editor1", Event::Edited), ("editor1", Event::Edited),
("editor1", Event::BufferEdited), ("editor1", Event::BufferEdited),
("editor2", Event::BufferEdited), ("editor2", Event::BufferEdited),
("editor1", Event::DirtyChanged),
("editor2", Event::DirtyChanged),
] ]
); );
@ -6246,6 +6254,8 @@ mod tests {
("editor1", Event::Edited), ("editor1", Event::Edited),
("editor1", Event::BufferEdited), ("editor1", Event::BufferEdited),
("editor2", Event::BufferEdited), ("editor2", Event::BufferEdited),
("editor1", Event::DirtyChanged),
("editor2", Event::DirtyChanged),
] ]
); );
@ -6257,6 +6267,8 @@ mod tests {
("editor2", Event::Edited), ("editor2", Event::Edited),
("editor1", Event::BufferEdited), ("editor1", Event::BufferEdited),
("editor2", Event::BufferEdited), ("editor2", Event::BufferEdited),
("editor1", Event::DirtyChanged),
("editor2", Event::DirtyChanged),
] ]
); );
@ -6268,6 +6280,8 @@ mod tests {
("editor2", Event::Edited), ("editor2", Event::Edited),
("editor1", Event::BufferEdited), ("editor1", Event::BufferEdited),
("editor2", Event::BufferEdited), ("editor2", Event::BufferEdited),
("editor1", Event::DirtyChanged),
("editor2", Event::DirtyChanged),
] ]
); );

View file

@ -440,7 +440,10 @@ impl Item for Editor {
} }
fn should_update_tab_on_event(event: &Event) -> bool { fn should_update_tab_on_event(event: &Event) -> bool {
matches!(event, Event::Saved | Event::Dirtied | Event::TitleChanged) matches!(
event,
Event::Saved | Event::DirtyChanged | Event::TitleChanged
)
} }
} }

View file

@ -1923,15 +1923,7 @@ impl MultiBufferSnapshot {
); );
if range.end > end_before_newline { if range.end > end_before_newline {
summary.add_assign(&D::from_text_summary(&TextSummary { summary.add_assign(&D::from_text_summary(&TextSummary::from("\n")));
bytes: 1,
lines: Point::new(1 as u32, 0),
lines_utf16: PointUtf16::new(1 as u32, 0),
first_line_chars: 0,
last_line_chars: 0,
longest_row: 0,
longest_row_chars: 0,
}));
} }
cursor.next(&()); cursor.next(&());

View file

@ -51,7 +51,10 @@ pub struct Buffer {
text: TextBuffer, text: TextBuffer,
file: Option<Arc<dyn File>>, file: Option<Arc<dyn File>>,
saved_version: clock::Global, saved_version: clock::Global,
saved_version_fingerprint: String,
saved_mtime: SystemTime, saved_mtime: SystemTime,
transaction_depth: usize,
was_dirty_before_starting_transaction: Option<bool>,
language: Option<Arc<Language>>, language: Option<Arc<Language>>,
autoindent_requests: Vec<Arc<AutoindentRequest>>, autoindent_requests: Vec<Arc<AutoindentRequest>>,
pending_autoindent: Option<Task<()>>, pending_autoindent: Option<Task<()>>,
@ -155,7 +158,7 @@ pub enum Operation {
pub enum Event { pub enum Event {
Operation(Operation), Operation(Operation),
Edited, Edited,
Dirtied, DirtyChanged,
Saved, Saved,
FileHandleChanged, FileHandleChanged,
Reloaded, Reloaded,
@ -192,7 +195,7 @@ pub trait File: Send + Sync {
text: Rope, text: Rope,
version: clock::Global, version: clock::Global,
cx: &mut MutableAppContext, cx: &mut MutableAppContext,
) -> Task<Result<(clock::Global, SystemTime)>>; ) -> Task<Result<(clock::Global, String, SystemTime)>>;
fn as_any(&self) -> &dyn Any; fn as_any(&self) -> &dyn Any;
@ -209,6 +212,7 @@ pub trait LocalFile: File {
&self, &self,
buffer_id: u64, buffer_id: u64,
version: &clock::Global, version: &clock::Global,
fingerprint: String,
mtime: SystemTime, mtime: SystemTime,
cx: &mut MutableAppContext, cx: &mut MutableAppContext,
); );
@ -426,6 +430,9 @@ impl Buffer {
Self { Self {
saved_mtime, saved_mtime,
saved_version: buffer.version(), saved_version: buffer.version(),
saved_version_fingerprint: buffer.as_rope().fingerprint(),
transaction_depth: 0,
was_dirty_before_starting_transaction: None,
text: buffer, text: buffer,
file, file,
syntax_tree: Mutex::new(None), syntax_tree: Mutex::new(None),
@ -476,7 +483,7 @@ impl Buffer {
pub fn save( pub fn save(
&mut self, &mut self,
cx: &mut ModelContext<Self>, cx: &mut ModelContext<Self>,
) -> Task<Result<(clock::Global, SystemTime)>> { ) -> Task<Result<(clock::Global, String, SystemTime)>> {
let file = if let Some(file) = self.file.as_ref() { let file = if let Some(file) = self.file.as_ref() {
file file
} else { } else {
@ -486,11 +493,11 @@ impl Buffer {
let version = self.version(); let version = self.version();
let save = file.save(self.remote_id(), text, version, cx.as_mut()); let save = file.save(self.remote_id(), text, version, cx.as_mut());
cx.spawn(|this, mut cx| async move { cx.spawn(|this, mut cx| async move {
let (version, mtime) = save.await?; let (version, fingerprint, mtime) = save.await?;
this.update(&mut cx, |this, cx| { this.update(&mut cx, |this, cx| {
this.did_save(version.clone(), mtime, None, cx); this.did_save(version.clone(), fingerprint.clone(), mtime, None, cx);
}); });
Ok((version, mtime)) Ok((version, fingerprint, mtime))
}) })
} }
@ -507,12 +514,14 @@ impl Buffer {
pub fn did_save( pub fn did_save(
&mut self, &mut self,
version: clock::Global, version: clock::Global,
fingerprint: String,
mtime: SystemTime, mtime: SystemTime,
new_file: Option<Arc<dyn File>>, new_file: Option<Arc<dyn File>>,
cx: &mut ModelContext<Self>, cx: &mut ModelContext<Self>,
) { ) {
self.saved_mtime = mtime;
self.saved_version = version; self.saved_version = version;
self.saved_version_fingerprint = fingerprint;
self.saved_mtime = mtime;
if let Some(new_file) = new_file { if let Some(new_file) = new_file {
self.file = Some(new_file); self.file = Some(new_file);
self.file_update_count += 1; self.file_update_count += 1;
@ -533,7 +542,12 @@ impl Buffer {
.await; .await;
this.update(&mut cx, |this, cx| { this.update(&mut cx, |this, cx| {
if let Some(transaction) = this.apply_diff(diff, cx).cloned() { if let Some(transaction) = this.apply_diff(diff, cx).cloned() {
this.did_reload(this.version(), new_mtime, cx); this.did_reload(
this.version(),
this.as_rope().fingerprint(),
new_mtime,
cx,
);
Ok(Some(transaction)) Ok(Some(transaction))
} else { } else {
Ok(None) Ok(None)
@ -548,13 +562,21 @@ impl Buffer {
pub fn did_reload( pub fn did_reload(
&mut self, &mut self,
version: clock::Global, version: clock::Global,
fingerprint: String,
mtime: SystemTime, mtime: SystemTime,
cx: &mut ModelContext<Self>, cx: &mut ModelContext<Self>,
) { ) {
self.saved_mtime = mtime;
self.saved_version = version; self.saved_version = version;
self.saved_version_fingerprint = fingerprint;
self.saved_mtime = mtime;
if let Some(file) = self.file.as_ref().and_then(|f| f.as_local()) { if let Some(file) = self.file.as_ref().and_then(|f| f.as_local()) {
file.buffer_reloaded(self.remote_id(), &self.saved_version, self.saved_mtime, cx); file.buffer_reloaded(
self.remote_id(),
&self.saved_version,
self.saved_version_fingerprint.clone(),
self.saved_mtime,
cx,
);
} }
cx.emit(Event::Reloaded); cx.emit(Event::Reloaded);
cx.notify(); cx.notify();
@ -581,7 +603,7 @@ impl Buffer {
if !old_file.is_deleted() { if !old_file.is_deleted() {
file_changed = true; file_changed = true;
if !self.is_dirty() { if !self.is_dirty() {
cx.emit(Event::Dirtied); cx.emit(Event::DirtyChanged);
} }
} }
} else { } else {
@ -972,12 +994,12 @@ impl Buffer {
} }
pub fn is_dirty(&self) -> bool { pub fn is_dirty(&self) -> bool {
!self.saved_version.observed_all(&self.version) self.saved_version_fingerprint != self.as_rope().fingerprint()
|| self.file.as_ref().map_or(false, |file| file.is_deleted()) || self.file.as_ref().map_or(false, |file| file.is_deleted())
} }
pub fn has_conflict(&self) -> bool { pub fn has_conflict(&self) -> bool {
!self.saved_version.observed_all(&self.version) self.saved_version_fingerprint != self.as_rope().fingerprint()
&& self && self
.file .file
.as_ref() .as_ref()
@ -993,6 +1015,10 @@ impl Buffer {
} }
pub fn start_transaction_at(&mut self, now: Instant) -> Option<TransactionId> { pub fn start_transaction_at(&mut self, now: Instant) -> Option<TransactionId> {
self.transaction_depth += 1;
if self.was_dirty_before_starting_transaction.is_none() {
self.was_dirty_before_starting_transaction = Some(self.is_dirty());
}
self.text.start_transaction_at(now) self.text.start_transaction_at(now)
} }
@ -1005,8 +1031,14 @@ impl Buffer {
now: Instant, now: Instant,
cx: &mut ModelContext<Self>, cx: &mut ModelContext<Self>,
) -> Option<TransactionId> { ) -> Option<TransactionId> {
assert!(self.transaction_depth > 0);
self.transaction_depth -= 1;
let was_dirty = if self.transaction_depth == 0 {
self.was_dirty_before_starting_transaction.take().unwrap()
} else {
false
};
if let Some((transaction_id, start_version)) = self.text.end_transaction_at(now) { if let Some((transaction_id, start_version)) = self.text.end_transaction_at(now) {
let was_dirty = start_version != self.saved_version;
self.did_edit(&start_version, was_dirty, cx); self.did_edit(&start_version, was_dirty, cx);
Some(transaction_id) Some(transaction_id)
} else { } else {
@ -1217,8 +1249,8 @@ impl Buffer {
self.reparse(cx); self.reparse(cx);
cx.emit(Event::Edited); cx.emit(Event::Edited);
if !was_dirty { if was_dirty != self.is_dirty() {
cx.emit(Event::Dirtied); cx.emit(Event::DirtyChanged);
} }
cx.notify(); cx.notify();
} }

View file

@ -91,7 +91,7 @@ fn test_edit_events(cx: &mut gpui::MutableAppContext) {
}) })
.detach(); .detach();
// An edit emits an edited event, followed by a dirtied event, // An edit emits an edited event, followed by a dirty changed event,
// since the buffer was previously in a clean state. // since the buffer was previously in a clean state.
buffer.edit([(2..4, "XYZ")], cx); buffer.edit([(2..4, "XYZ")], cx);
@ -112,21 +112,46 @@ fn test_edit_events(cx: &mut gpui::MutableAppContext) {
}); });
// Incorporating a set of remote ops emits a single edited event, // Incorporating a set of remote ops emits a single edited event,
// followed by a dirtied event. // followed by a dirty changed event.
buffer2.update(cx, |buffer, cx| { buffer2.update(cx, |buffer, cx| {
buffer buffer
.apply_ops(buffer1_ops.borrow_mut().drain(..), cx) .apply_ops(buffer1_ops.borrow_mut().drain(..), cx)
.unwrap(); .unwrap();
}); });
let buffer_1_events = buffer_1_events.borrow();
assert_eq!( assert_eq!(
*buffer_1_events, mem::take(&mut *buffer_1_events.borrow_mut()),
vec![Event::Edited, Event::Dirtied, Event::Edited, Event::Edited] vec![
Event::Edited,
Event::DirtyChanged,
Event::Edited,
Event::Edited,
]
);
assert_eq!(
mem::take(&mut *buffer_2_events.borrow_mut()),
vec![Event::Edited, Event::DirtyChanged]
); );
let buffer_2_events = buffer_2_events.borrow(); buffer1.update(cx, |buffer, cx| {
assert_eq!(*buffer_2_events, vec![Event::Edited, Event::Dirtied]); // Undoing the first transaction emits edited event, followed by a
// dirty changed event, since the buffer is again in a clean state.
buffer.undo(cx);
});
// Incorporating the remote ops again emits a single edited event,
// followed by a dirty changed event.
buffer2.update(cx, |buffer, cx| {
buffer
.apply_ops(buffer1_ops.borrow_mut().drain(..), cx)
.unwrap();
});
assert_eq!(
mem::take(&mut *buffer_1_events.borrow_mut()),
vec![Event::Edited, Event::DirtyChanged,]
);
assert_eq!(
mem::take(&mut *buffer_2_events.borrow_mut()),
vec![Event::Edited, Event::DirtyChanged]
);
} }
#[gpui::test] #[gpui::test]

View file

@ -4742,12 +4742,14 @@ impl Project {
}) })
.await; .await;
let (saved_version, mtime) = buffer.update(&mut cx, |buffer, cx| buffer.save(cx)).await?; let (saved_version, fingerprint, mtime) =
buffer.update(&mut cx, |buffer, cx| buffer.save(cx)).await?;
Ok(proto::BufferSaved { Ok(proto::BufferSaved {
project_id, project_id,
buffer_id, buffer_id,
version: serialize_version(&saved_version), version: serialize_version(&saved_version),
mtime: Some(mtime.into()), mtime: Some(mtime.into()),
fingerprint,
}) })
} }
@ -5306,7 +5308,7 @@ impl Project {
.and_then(|buffer| buffer.upgrade(cx)); .and_then(|buffer| buffer.upgrade(cx));
if let Some(buffer) = buffer { if let Some(buffer) = buffer {
buffer.update(cx, |buffer, cx| { buffer.update(cx, |buffer, cx| {
buffer.did_save(version, mtime, None, cx); buffer.did_save(version, envelope.payload.fingerprint, mtime, None, cx);
}); });
} }
Ok(()) Ok(())
@ -5332,7 +5334,7 @@ impl Project {
.and_then(|buffer| buffer.upgrade(cx)); .and_then(|buffer| buffer.upgrade(cx));
if let Some(buffer) = buffer { if let Some(buffer) = buffer {
buffer.update(cx, |buffer, cx| { buffer.update(cx, |buffer, cx| {
buffer.did_reload(version, mtime, cx); buffer.did_reload(version, payload.fingerprint, mtime, cx);
}); });
} }
Ok(()) Ok(())
@ -8105,10 +8107,16 @@ mod tests {
assert!(buffer.is_dirty()); assert!(buffer.is_dirty());
assert_eq!( assert_eq!(
*events.borrow(), *events.borrow(),
&[language::Event::Edited, language::Event::Dirtied] &[language::Event::Edited, language::Event::DirtyChanged]
); );
events.borrow_mut().clear(); events.borrow_mut().clear();
buffer.did_save(buffer.version(), buffer.file().unwrap().mtime(), None, cx); buffer.did_save(
buffer.version(),
buffer.as_rope().fingerprint(),
buffer.file().unwrap().mtime(),
None,
cx,
);
}); });
// after saving, the buffer is not dirty, and emits a saved event. // after saving, the buffer is not dirty, and emits a saved event.
@ -8129,20 +8137,23 @@ mod tests {
*events.borrow(), *events.borrow(),
&[ &[
language::Event::Edited, language::Event::Edited,
language::Event::Dirtied, language::Event::DirtyChanged,
language::Event::Edited, language::Event::Edited,
], ],
); );
events.borrow_mut().clear(); events.borrow_mut().clear();
// TODO - currently, after restoring the buffer to its // After restoring the buffer to its previously-saved state,
// previously-saved state, the is still considered dirty. // the buffer is not considered dirty anymore.
buffer.edit([(1..3, "")], cx); buffer.edit([(1..3, "")], cx);
assert!(buffer.text() == "ac"); assert!(buffer.text() == "ac");
assert!(buffer.is_dirty()); assert!(!buffer.is_dirty());
}); });
assert_eq!(*events.borrow(), &[language::Event::Edited]); assert_eq!(
*events.borrow(),
&[language::Event::Edited, language::Event::DirtyChanged]
);
// When a file is deleted, the buffer is considered dirty. // When a file is deleted, the buffer is considered dirty.
let events = Rc::new(RefCell::new(Vec::new())); let events = Rc::new(RefCell::new(Vec::new()));
@ -8164,7 +8175,10 @@ mod tests {
buffer2.condition(&cx, |b, _| b.is_dirty()).await; buffer2.condition(&cx, |b, _| b.is_dirty()).await;
assert_eq!( assert_eq!(
*events.borrow(), *events.borrow(),
&[language::Event::Dirtied, language::Event::FileHandleChanged] &[
language::Event::DirtyChanged,
language::Event::FileHandleChanged
]
); );
// When a file is already dirty when deleted, we don't emit a Dirtied event. // When a file is already dirty when deleted, we don't emit a Dirtied event.

View file

@ -634,6 +634,7 @@ impl LocalWorktree {
) -> Task<Result<()>> { ) -> Task<Result<()>> {
let buffer = buffer_handle.read(cx); let buffer = buffer_handle.read(cx);
let text = buffer.as_rope().clone(); let text = buffer.as_rope().clone();
let fingerprint = text.fingerprint();
let version = buffer.version(); let version = buffer.version();
let save = self.write_file(path, text, cx); let save = self.write_file(path, text, cx);
let handle = cx.handle(); let handle = cx.handle();
@ -648,7 +649,7 @@ impl LocalWorktree {
}; };
buffer_handle.update(&mut cx, |buffer, cx| { buffer_handle.update(&mut cx, |buffer, cx| {
buffer.did_save(version, file.mtime, Some(Arc::new(file)), cx); buffer.did_save(version, fingerprint, file.mtime, Some(Arc::new(file)), cx);
}); });
Ok(()) Ok(())
@ -1702,11 +1703,12 @@ impl language::File for File {
text: Rope, text: Rope,
version: clock::Global, version: clock::Global,
cx: &mut MutableAppContext, cx: &mut MutableAppContext,
) -> Task<Result<(clock::Global, SystemTime)>> { ) -> Task<Result<(clock::Global, String, SystemTime)>> {
self.worktree.update(cx, |worktree, cx| match worktree { self.worktree.update(cx, |worktree, cx| match worktree {
Worktree::Local(worktree) => { Worktree::Local(worktree) => {
let rpc = worktree.client.clone(); let rpc = worktree.client.clone();
let project_id = worktree.share.as_ref().map(|share| share.project_id); let project_id = worktree.share.as_ref().map(|share| share.project_id);
let fingerprint = text.fingerprint();
let save = worktree.write_file(self.path.clone(), text, cx); let save = worktree.write_file(self.path.clone(), text, cx);
cx.background().spawn(async move { cx.background().spawn(async move {
let entry = save.await?; let entry = save.await?;
@ -1716,9 +1718,10 @@ impl language::File for File {
buffer_id, buffer_id,
version: serialize_version(&version), version: serialize_version(&version),
mtime: Some(entry.mtime.into()), mtime: Some(entry.mtime.into()),
fingerprint: fingerprint.clone(),
})?; })?;
} }
Ok((version, entry.mtime)) Ok((version, fingerprint, entry.mtime))
}) })
} }
Worktree::Remote(worktree) => { Worktree::Remote(worktree) => {
@ -1737,7 +1740,7 @@ impl language::File for File {
.mtime .mtime
.ok_or_else(|| anyhow!("missing mtime"))? .ok_or_else(|| anyhow!("missing mtime"))?
.into(); .into();
Ok((version, mtime)) Ok((version, response.fingerprint, mtime))
}) })
} }
}) })
@ -1779,6 +1782,7 @@ impl language::LocalFile for File {
&self, &self,
buffer_id: u64, buffer_id: u64,
version: &clock::Global, version: &clock::Global,
fingerprint: String,
mtime: SystemTime, mtime: SystemTime,
cx: &mut MutableAppContext, cx: &mut MutableAppContext,
) { ) {
@ -1791,6 +1795,7 @@ impl language::LocalFile for File {
buffer_id, buffer_id,
version: serialize_version(&version), version: serialize_version(&version),
mtime: Some(mtime.into()), mtime: Some(mtime.into()),
fingerprint,
}) })
.log_err(); .log_err();
} }

View file

@ -364,6 +364,7 @@ message BufferSaved {
uint64 buffer_id = 2; uint64 buffer_id = 2;
repeated VectorClockEntry version = 3; repeated VectorClockEntry version = 3;
Timestamp mtime = 4; Timestamp mtime = 4;
string fingerprint = 5;
} }
message BufferReloaded { message BufferReloaded {
@ -371,6 +372,7 @@ message BufferReloaded {
uint64 buffer_id = 2; uint64 buffer_id = 2;
repeated VectorClockEntry version = 3; repeated VectorClockEntry version = 3;
Timestamp mtime = 4; Timestamp mtime = 4;
string fingerprint = 5;
} }
message ReloadBuffers { message ReloadBuffers {

View file

@ -6,4 +6,4 @@ pub use conn::Connection;
pub use peer::*; pub use peer::*;
mod macros; mod macros;
pub const PROTOCOL_VERSION: u32 = 24; pub const PROTOCOL_VERSION: u32 = 25;

View file

@ -16,6 +16,8 @@ collections = { path = "../collections" }
sum_tree = { path = "../sum_tree" } sum_tree = { path = "../sum_tree" }
anyhow = "1.0.38" anyhow = "1.0.38"
arrayvec = "0.7.1" arrayvec = "0.7.1"
digest = { version = "0.9", features = ["std"] }
bromberg_sl2 = "0.6"
lazy_static = "1.4" lazy_static = "1.4"
log = { version = "0.4.16", features = ["kv_unstable_serde"] } log = { version = "0.4.16", features = ["kv_unstable_serde"] }
parking_lot = "0.11" parking_lot = "0.11"

View file

@ -2,6 +2,7 @@ use crate::PointUtf16;
use super::Point; use super::Point;
use arrayvec::ArrayString; use arrayvec::ArrayString;
use bromberg_sl2::{DigestString, HashMatrix};
use smallvec::SmallVec; use smallvec::SmallVec;
use std::{cmp, fmt, io, mem, ops::Range, str}; use std::{cmp, fmt, io, mem, ops::Range, str};
use sum_tree::{Bias, Dimension, SumTree}; use sum_tree::{Bias, Dimension, SumTree};
@ -115,7 +116,7 @@ impl Rope {
} }
pub fn summary(&self) -> TextSummary { pub fn summary(&self) -> TextSummary {
self.chunks.summary().clone() self.chunks.summary().text.clone()
} }
pub fn len(&self) -> usize { pub fn len(&self) -> usize {
@ -290,6 +291,10 @@ impl Rope {
self.clip_point(Point::new(row, u32::MAX), Bias::Left) self.clip_point(Point::new(row, u32::MAX), Bias::Left)
.column .column
} }
pub fn fingerprint(&self) -> String {
self.chunks.summary().fingerprint.to_hex()
}
} }
impl<'a> From<&'a str> for Rope { impl<'a> From<&'a str> for Rope {
@ -709,10 +714,34 @@ impl Chunk {
} }
impl sum_tree::Item for Chunk { impl sum_tree::Item for Chunk {
type Summary = TextSummary; type Summary = ChunkSummary;
fn summary(&self) -> Self::Summary { fn summary(&self) -> Self::Summary {
TextSummary::from(self.0.as_str()) ChunkSummary::from(self.0.as_str())
}
}
#[derive(Clone, Debug, Default, Eq, PartialEq)]
pub struct ChunkSummary {
text: TextSummary,
fingerprint: HashMatrix,
}
impl<'a> From<&'a str> for ChunkSummary {
fn from(text: &'a str) -> Self {
Self {
text: TextSummary::from(text),
fingerprint: bromberg_sl2::hash_strict(text.as_bytes()),
}
}
}
impl sum_tree::Summary for ChunkSummary {
type Context = ();
fn add_summary(&mut self, summary: &Self, _: &()) {
self.text += &summary.text;
self.fingerprint = self.fingerprint * summary.fingerprint;
} }
} }
@ -819,7 +848,7 @@ impl std::ops::AddAssign<Self> for TextSummary {
} }
} }
pub trait TextDimension: 'static + for<'a> Dimension<'a, TextSummary> { pub trait TextDimension: 'static + for<'a> Dimension<'a, ChunkSummary> {
fn from_text_summary(summary: &TextSummary) -> Self; fn from_text_summary(summary: &TextSummary) -> Self;
fn add_assign(&mut self, other: &Self); fn add_assign(&mut self, other: &Self);
} }
@ -838,6 +867,12 @@ impl<'a, D1: TextDimension, D2: TextDimension> TextDimension for (D1, D2) {
} }
} }
impl<'a> sum_tree::Dimension<'a, ChunkSummary> for TextSummary {
fn add_summary(&mut self, summary: &'a ChunkSummary, _: &()) {
*self += &summary.text;
}
}
impl TextDimension for TextSummary { impl TextDimension for TextSummary {
fn from_text_summary(summary: &TextSummary) -> Self { fn from_text_summary(summary: &TextSummary) -> Self {
summary.clone() summary.clone()
@ -848,9 +883,9 @@ impl TextDimension for TextSummary {
} }
} }
impl<'a> sum_tree::Dimension<'a, TextSummary> for usize { impl<'a> sum_tree::Dimension<'a, ChunkSummary> for usize {
fn add_summary(&mut self, summary: &'a TextSummary, _: &()) { fn add_summary(&mut self, summary: &'a ChunkSummary, _: &()) {
*self += summary.bytes; *self += summary.text.bytes;
} }
} }
@ -864,9 +899,9 @@ impl TextDimension for usize {
} }
} }
impl<'a> sum_tree::Dimension<'a, TextSummary> for Point { impl<'a> sum_tree::Dimension<'a, ChunkSummary> for Point {
fn add_summary(&mut self, summary: &'a TextSummary, _: &()) { fn add_summary(&mut self, summary: &'a ChunkSummary, _: &()) {
*self += summary.lines; *self += summary.text.lines;
} }
} }
@ -880,9 +915,9 @@ impl TextDimension for Point {
} }
} }
impl<'a> sum_tree::Dimension<'a, TextSummary> for PointUtf16 { impl<'a> sum_tree::Dimension<'a, ChunkSummary> for PointUtf16 {
fn add_summary(&mut self, summary: &'a TextSummary, _: &()) { fn add_summary(&mut self, summary: &'a ChunkSummary, _: &()) {
*self += summary.lines_utf16; *self += summary.text.lines_utf16;
} }
} }