diff options
author | Galilée 'Bill' Enguehard <[email protected]> | 2020-05-21 22:27:38 +0100 |
---|---|---|
committer | Galilée 'Bill' Enguehard <[email protected]> | 2020-05-21 22:27:38 +0100 |
commit | 7fece3bdd2450c0807f7dd742239cae95f0cc65e (patch) | |
tree | 866c4db826c959e79c63a6727bdb9f2c61e6fc4f /crates/ra_text_edit/src/lib.rs | |
parent | db926218b2082077750291f8426ddd28b284cd08 (diff) | |
parent | 59732df8d40dfadc6dcf5951265416576399712a (diff) |
Merge branch 'master' of github.com:rust-analyzer/rust-analyzer into modname_spacing
Diffstat (limited to 'crates/ra_text_edit/src/lib.rs')
-rw-r--r-- | crates/ra_text_edit/src/lib.rs | 174 |
1 files changed, 154 insertions, 20 deletions
diff --git a/crates/ra_text_edit/src/lib.rs b/crates/ra_text_edit/src/lib.rs index e656260c7..25554f583 100644 --- a/crates/ra_text_edit/src/lib.rs +++ b/crates/ra_text_edit/src/lib.rs | |||
@@ -1,36 +1,170 @@ | |||
1 | //! FIXME: write short doc here | 1 | //! Representation of a `TextEdit`. |
2 | //! | ||
3 | //! `rust-analyzer` never mutates text itself and only sends diffs to clients, | ||
4 | //! so `TextEdit` is the ultimate representation of the work done by | ||
5 | //! rust-analyzer. | ||
6 | use std::{slice, vec}; | ||
2 | 7 | ||
3 | mod text_edit; | 8 | pub use text_size::{TextRange, TextSize}; |
4 | 9 | ||
5 | use text_size::{TextRange, TextSize}; | 10 | /// `InsertDelete` -- a single "atomic" change to text |
6 | 11 | /// | |
7 | pub use crate::text_edit::{TextEdit, TextEditBuilder}; | 12 | /// Must not overlap with other `InDel`s |
8 | |||
9 | /// Must not overlap with other `AtomTextEdit`s | ||
10 | #[derive(Debug, Clone)] | 13 | #[derive(Debug, Clone)] |
11 | pub struct AtomTextEdit { | 14 | pub struct Indel { |
15 | pub insert: String, | ||
12 | /// Refers to offsets in the original text | 16 | /// Refers to offsets in the original text |
13 | pub delete: TextRange, | 17 | pub delete: TextRange, |
14 | pub insert: String, | ||
15 | } | 18 | } |
16 | 19 | ||
17 | impl AtomTextEdit { | 20 | #[derive(Default, Debug, Clone)] |
18 | pub fn replace(range: TextRange, replace_with: String) -> AtomTextEdit { | 21 | pub struct TextEdit { |
19 | AtomTextEdit { delete: range, insert: replace_with } | 22 | indels: Vec<Indel>, |
20 | } | 23 | } |
21 | 24 | ||
22 | pub fn delete(range: TextRange) -> AtomTextEdit { | 25 | #[derive(Debug, Default, Clone)] |
23 | AtomTextEdit::replace(range, String::new()) | 26 | pub struct TextEditBuilder { |
24 | } | 27 | indels: Vec<Indel>, |
28 | } | ||
25 | 29 | ||
26 | pub fn insert(offset: TextSize, text: String) -> AtomTextEdit { | 30 | impl Indel { |
27 | AtomTextEdit::replace(TextRange::empty(offset), text) | 31 | pub fn insert(offset: TextSize, text: String) -> Indel { |
32 | Indel::replace(TextRange::empty(offset), text) | ||
33 | } | ||
34 | pub fn delete(range: TextRange) -> Indel { | ||
35 | Indel::replace(range, String::new()) | ||
36 | } | ||
37 | pub fn replace(range: TextRange, replace_with: String) -> Indel { | ||
38 | Indel { delete: range, insert: replace_with } | ||
28 | } | 39 | } |
29 | 40 | ||
30 | pub fn apply(&self, mut text: String) -> String { | 41 | pub fn apply(&self, text: &mut String) { |
31 | let start: usize = self.delete.start().into(); | 42 | let start: usize = self.delete.start().into(); |
32 | let end: usize = self.delete.end().into(); | 43 | let end: usize = self.delete.end().into(); |
33 | text.replace_range(start..end, &self.insert); | 44 | text.replace_range(start..end, &self.insert); |
34 | text | ||
35 | } | 45 | } |
36 | } | 46 | } |
47 | |||
48 | impl TextEdit { | ||
49 | pub fn insert(offset: TextSize, text: String) -> TextEdit { | ||
50 | let mut builder = TextEditBuilder::default(); | ||
51 | builder.insert(offset, text); | ||
52 | builder.finish() | ||
53 | } | ||
54 | |||
55 | pub fn delete(range: TextRange) -> TextEdit { | ||
56 | let mut builder = TextEditBuilder::default(); | ||
57 | builder.delete(range); | ||
58 | builder.finish() | ||
59 | } | ||
60 | |||
61 | pub fn replace(range: TextRange, replace_with: String) -> TextEdit { | ||
62 | let mut builder = TextEditBuilder::default(); | ||
63 | builder.replace(range, replace_with); | ||
64 | builder.finish() | ||
65 | } | ||
66 | |||
67 | pub fn len(&self) -> usize { | ||
68 | self.indels.len() | ||
69 | } | ||
70 | |||
71 | pub fn is_empty(&self) -> bool { | ||
72 | self.indels.is_empty() | ||
73 | } | ||
74 | |||
75 | pub fn iter(&self) -> slice::Iter<'_, Indel> { | ||
76 | self.indels.iter() | ||
77 | } | ||
78 | |||
79 | pub fn into_iter(self) -> vec::IntoIter<Indel> { | ||
80 | self.indels.into_iter() | ||
81 | } | ||
82 | |||
83 | pub fn apply(&self, text: &mut String) { | ||
84 | match self.len() { | ||
85 | 0 => return, | ||
86 | 1 => { | ||
87 | self.indels[0].apply(text); | ||
88 | return; | ||
89 | } | ||
90 | _ => (), | ||
91 | } | ||
92 | |||
93 | let mut total_len = TextSize::of(&*text); | ||
94 | for indel in self.indels.iter() { | ||
95 | total_len += TextSize::of(&indel.insert); | ||
96 | total_len -= indel.delete.end() - indel.delete.start(); | ||
97 | } | ||
98 | let mut buf = String::with_capacity(total_len.into()); | ||
99 | let mut prev = 0; | ||
100 | for indel in self.indels.iter() { | ||
101 | let start: usize = indel.delete.start().into(); | ||
102 | let end: usize = indel.delete.end().into(); | ||
103 | if start > prev { | ||
104 | buf.push_str(&text[prev..start]); | ||
105 | } | ||
106 | buf.push_str(&indel.insert); | ||
107 | prev = end; | ||
108 | } | ||
109 | buf.push_str(&text[prev..text.len()]); | ||
110 | assert_eq!(TextSize::of(&buf), total_len); | ||
111 | |||
112 | // FIXME: figure out a way to mutate the text in-place or reuse the | ||
113 | // memory in some other way | ||
114 | *text = buf | ||
115 | } | ||
116 | |||
117 | pub fn union(&mut self, other: TextEdit) -> Result<(), TextEdit> { | ||
118 | // FIXME: can be done without allocating intermediate vector | ||
119 | let mut all = self.iter().chain(other.iter()).collect::<Vec<_>>(); | ||
120 | if !check_disjoint(&mut all) { | ||
121 | return Err(other); | ||
122 | } | ||
123 | self.indels.extend(other.indels); | ||
124 | assert!(check_disjoint(&mut self.indels)); | ||
125 | Ok(()) | ||
126 | } | ||
127 | |||
128 | pub fn apply_to_offset(&self, offset: TextSize) -> Option<TextSize> { | ||
129 | let mut res = offset; | ||
130 | for indel in self.indels.iter() { | ||
131 | if indel.delete.start() >= offset { | ||
132 | break; | ||
133 | } | ||
134 | if offset < indel.delete.end() { | ||
135 | return None; | ||
136 | } | ||
137 | res += TextSize::of(&indel.insert); | ||
138 | res -= indel.delete.len(); | ||
139 | } | ||
140 | Some(res) | ||
141 | } | ||
142 | } | ||
143 | |||
144 | impl TextEditBuilder { | ||
145 | pub fn replace(&mut self, range: TextRange, replace_with: String) { | ||
146 | self.indels.push(Indel::replace(range, replace_with)) | ||
147 | } | ||
148 | pub fn delete(&mut self, range: TextRange) { | ||
149 | self.indels.push(Indel::delete(range)) | ||
150 | } | ||
151 | pub fn insert(&mut self, offset: TextSize, text: String) { | ||
152 | self.indels.push(Indel::insert(offset, text)) | ||
153 | } | ||
154 | pub fn finish(self) -> TextEdit { | ||
155 | let mut indels = self.indels; | ||
156 | assert!(check_disjoint(&mut indels)); | ||
157 | TextEdit { indels } | ||
158 | } | ||
159 | pub fn invalidates_offset(&self, offset: TextSize) -> bool { | ||
160 | self.indels.iter().any(|indel| indel.delete.contains_inclusive(offset)) | ||
161 | } | ||
162 | } | ||
163 | |||
164 | fn check_disjoint(indels: &mut [impl std::borrow::Borrow<Indel>]) -> bool { | ||
165 | indels.sort_by_key(|indel| (indel.borrow().delete.start(), indel.borrow().delete.end())); | ||
166 | indels | ||
167 | .iter() | ||
168 | .zip(indels.iter().skip(1)) | ||
169 | .all(|(l, r)| l.borrow().delete.end() <= r.borrow().delete.start()) | ||
170 | } | ||