strop_engine/editor/
transact.rs1mod capability;
3pub use capability::{BufferEdit, DocumentEdit};
4use strop_core::id::{BufferRevision, DocumentId};
5use strop_core::{Change, EditError, Replacement};
6
7#[derive(Debug, Clone)]
9pub struct ChangeSet {
10 pub edits: Vec<Replacement>,
11 pub undo_open: bool,
12}
13
14#[derive(Debug, Clone, PartialEq, Eq, thiserror::Error)]
15pub enum ApplyError {
16 #[error("no such document")]
17 NoDocument,
18 #[error(transparent)]
19 Edit(#[from] EditError),
20}
21
22#[derive(Debug, Clone, Copy, PartialEq, Eq)]
23pub struct Committed {
24 pub revision: BufferRevision,
25}
26
27impl super::Editor {
28 pub(crate) fn apply(
29 &mut self,
30 document: DocumentId,
31 base: BufferRevision,
32 changes: ChangeSet,
33 ) -> Result<Committed, ApplyError> {
34 let buffer = &self.docs.get(document).ok_or(ApplyError::NoDocument)?.buf;
35 let prepared = buffer.prepare_replacements(base, changes.edits)?;
36 if prepared.is_empty() {
37 return Ok(Committed { revision: base });
38 }
39 if self
40 .pending
41 .prompt()
42 .is_some_and(|prompt| prompt.origin().pane.doc == document)
43 {
44 self.cancel_pending();
45 }
46 let revision = self
47 .doc_mut(document)
48 .buf
49 .apply_prepared(prepared, changes.undo_open)?;
50 Ok(Committed { revision })
51 }
52
53 pub(crate) fn replace_system(
54 &mut self,
55 document: DocumentId,
56 text: &str,
57 ) -> Result<(), EditError> {
58 self.doc_mut(document).buf.system_edit().replace_all(text)
59 }
60
61 pub(crate) fn tx_begin(&mut self) {
62 self.buf_mut().begin_undo_group();
63 }
64 pub(crate) fn tx_commit(&mut self) {
65 self.buf_mut().commit_undo_group();
66 strop_trace::record_with(strop_trace::EventKind::History, || {
67 serde_json::json!({
68 "operation":"commit","buffer":self.buf().trace_id(),"document":self.current(),
69 "revision":self.buf().revision(),"history_nodes":self.buf().history().depth(),
70 })
71 });
72 }
73
74 pub(super) fn sync_document(&mut self, id: DocumentId, map_active: bool) {
78 self.sync_document_positions(id, map_active, map_position);
79 }
80
81 fn sync_document_positions(
82 &mut self,
83 id: DocumentId,
84 map_active: bool,
85 position: impl Fn(usize, &Change) -> usize,
86 ) {
87 if self.collections.contains_key(&id)
91 && self.docs.get(id).is_some_and(|doc| {
92 doc.buf
93 .changes()
94 .iter()
95 .any(|c| c.origin == strop_core::ChangeOrigin::User)
96 })
97 {
98 self.sync_collection_write_back(id, map_active, &position);
99 return;
100 }
101 let change_count = self.docs.get(id).map_or(0, |doc| doc.buf.changes().len());
102 if change_count == 0 {
103 return;
104 }
105 for index in 0..change_count {
108 let change = self.doc(id).buf.changes()[index];
109 self.map_navigation_records(id, |offset| position(offset, &change));
110 }
111 let clean_views: std::collections::HashSet<DocumentId> = self
114 .collections
115 .iter()
116 .filter(|(cid, collection)| {
117 self.docs
118 .get(**cid)
119 .is_some_and(|doc| doc.buf.revision() == collection.revision)
120 })
121 .map(|(cid, _)| *cid)
122 .collect();
123 let collection_updates = self.prepare_collection_updates(id, &clean_views);
124 let Some(document) = self.docs.get_mut(id) else {
125 return;
126 };
127 if document.buf.changes().is_empty() {
128 return;
129 }
130 let active = self.active_pane;
131 for change in document.buf.changes() {
132 let map = |offset| position(offset, change);
133 for (owner, position) in self.marks.values_mut() {
134 if *owner == id {
135 *position = map(*position);
136 }
137 }
138 for (index, pane) in self.panes.iter_mut().enumerate() {
139 if pane.doc == id && (map_active || index != active) {
140 pane.sels.map_positions(map);
141 }
142 }
143 for collection in self.collections.values_mut() {
148 for excerpt in &mut collection.excerpts {
149 if excerpt.source == id {
150 let (s, e) = (change.edit.start_byte, change.edit.old_end_byte);
151 let new_len = change.edit.new_end_byte - change.edit.start_byte;
152 let delta = new_len as isize - (e - s) as isize;
153 let shift = |p: usize| (p as isize + delta) as usize;
154 let insertion = s == e;
155 excerpt.start = if excerpt.start <= s {
156 excerpt.start
157 } else if excerpt.start >= e {
158 shift(excerpt.start)
159 } else {
160 s
161 };
162 excerpt.end = if excerpt.end < s {
163 excerpt.end
164 } else if excerpt.end > e || (excerpt.end == e && !insertion) {
165 shift(excerpt.end)
166 } else {
167 s + new_len
170 };
171 for anchor in &mut excerpt.hit_anchors {
172 *anchor = map(*anchor);
173 }
174 excerpt.matches.retain_mut(|(start, length)| {
175 let finish = start.saturating_add(*length);
176 if s < finish && e > *start {
177 return false;
178 }
179 let mapped = map(*start);
180 *length = map(finish).saturating_sub(mapped);
181 *start = mapped;
182 true
183 });
184 }
185 }
186 }
187 }
188 self.analysis.edits(id, document.buf.changes());
189 document.buf.clear_changes();
190 self.publish_collection_updates(collection_updates);
191 }
192}
193
194fn map_position(position: usize, change: &Change) -> usize {
195 let edit = change.edit;
196 debug_assert!(edit.start_byte <= edit.old_end_byte);
197 debug_assert!(edit.new_end_byte >= edit.start_byte);
201 strop_core::editmap::map_position(
202 position,
203 edit.start_byte,
204 edit.old_end_byte,
205 edit.new_end_byte,
206 )
207}