aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_ssr/src/replacing.rs
diff options
context:
space:
mode:
Diffstat (limited to 'crates/ra_ssr/src/replacing.rs')
-rw-r--r--crates/ra_ssr/src/replacing.rs33
1 files changed, 17 insertions, 16 deletions
diff --git a/crates/ra_ssr/src/replacing.rs b/crates/ra_ssr/src/replacing.rs
index 5dcde82a2..70ce1c185 100644
--- a/crates/ra_ssr/src/replacing.rs
+++ b/crates/ra_ssr/src/replacing.rs
@@ -10,21 +10,25 @@ use ra_text_edit::TextEdit;
10/// Returns a text edit that will replace each match in `matches` with its corresponding replacement 10/// Returns a text edit that will replace each match in `matches` with its corresponding replacement
11/// template. Placeholders in the template will have been substituted with whatever they matched to 11/// template. Placeholders in the template will have been substituted with whatever they matched to
12/// in the original code. 12/// in the original code.
13pub(crate) fn matches_to_edit(matches: &SsrMatches) -> TextEdit { 13pub(crate) fn matches_to_edit(matches: &SsrMatches, file_src: &str) -> TextEdit {
14 matches_to_edit_at_offset(matches, 0.into()) 14 matches_to_edit_at_offset(matches, file_src, 0.into())
15} 15}
16 16
17fn matches_to_edit_at_offset(matches: &SsrMatches, relative_start: TextSize) -> TextEdit { 17fn matches_to_edit_at_offset(
18 matches: &SsrMatches,
19 file_src: &str,
20 relative_start: TextSize,
21) -> TextEdit {
18 let mut edit_builder = ra_text_edit::TextEditBuilder::default(); 22 let mut edit_builder = ra_text_edit::TextEditBuilder::default();
19 for m in &matches.matches { 23 for m in &matches.matches {
20 edit_builder.replace(m.range.checked_sub(relative_start).unwrap(), render_replace(m)); 24 edit_builder
25 .replace(m.range.checked_sub(relative_start).unwrap(), render_replace(m, file_src));
21 } 26 }
22 edit_builder.finish() 27 edit_builder.finish()
23} 28}
24 29
25fn render_replace(match_info: &Match) -> String { 30fn render_replace(match_info: &Match, file_src: &str) -> String {
26 let mut out = String::new(); 31 let mut out = String::new();
27 let match_start = match_info.matched_node.text_range().start();
28 for r in &match_info.template.tokens { 32 for r in &match_info.template.tokens {
29 match r { 33 match r {
30 PatternElement::Token(t) => out.push_str(t.text.as_str()), 34 PatternElement::Token(t) => out.push_str(t.text.as_str()),
@@ -33,16 +37,13 @@ fn render_replace(match_info: &Match) -> String {
33 match_info.placeholder_values.get(&Var(p.ident.to_string())) 37 match_info.placeholder_values.get(&Var(p.ident.to_string()))
34 { 38 {
35 let range = &placeholder_value.range.range; 39 let range = &placeholder_value.range.range;
36 let mut matched_text = if let Some(node) = &placeholder_value.node { 40 let mut matched_text =
37 node.text().to_string() 41 file_src[usize::from(range.start())..usize::from(range.end())].to_owned();
38 } else { 42 let edit = matches_to_edit_at_offset(
39 let relative_range = range.checked_sub(match_start).unwrap(); 43 &placeholder_value.inner_matches,
40 match_info.matched_node.text().to_string() 44 file_src,
41 [usize::from(relative_range.start())..usize::from(relative_range.end())] 45 range.start(),
42 .to_string() 46 );
43 };
44 let edit =
45 matches_to_edit_at_offset(&placeholder_value.inner_matches, range.start());
46 edit.apply(&mut matched_text); 47 edit.apply(&mut matched_text);
47 out.push_str(&matched_text); 48 out.push_str(&matched_text);
48 } else { 49 } else {