diff options
Diffstat (limited to 'crates/ra_assists/src/handlers/add_function.rs')
-rw-r--r-- | crates/ra_assists/src/handlers/add_function.rs | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/crates/ra_assists/src/handlers/add_function.rs b/crates/ra_assists/src/handlers/add_function.rs index a0709630d..24f931a85 100644 --- a/crates/ra_assists/src/handlers/add_function.rs +++ b/crates/ra_assists/src/handlers/add_function.rs | |||
@@ -10,7 +10,11 @@ use ra_syntax::{ | |||
10 | }; | 10 | }; |
11 | use rustc_hash::{FxHashMap, FxHashSet}; | 11 | use rustc_hash::{FxHashMap, FxHashSet}; |
12 | 12 | ||
13 | use crate::{assist_config::SnippetCap, utils::render_snippet, AssistContext, AssistId, Assists}; | 13 | use crate::{ |
14 | assist_config::SnippetCap, | ||
15 | utils::{render_snippet, Cursor}, | ||
16 | AssistContext, AssistId, Assists, | ||
17 | }; | ||
14 | 18 | ||
15 | // Assist: add_function | 19 | // Assist: add_function |
16 | // | 20 | // |
@@ -81,7 +85,11 @@ struct FunctionTemplate { | |||
81 | impl FunctionTemplate { | 85 | impl FunctionTemplate { |
82 | fn to_string(&self, cap: Option<SnippetCap>) -> String { | 86 | fn to_string(&self, cap: Option<SnippetCap>) -> String { |
83 | let f = match cap { | 87 | let f = match cap { |
84 | Some(cap) => render_snippet(cap, self.fn_def.syntax(), self.placeholder_expr.syntax()), | 88 | Some(cap) => render_snippet( |
89 | cap, | ||
90 | self.fn_def.syntax(), | ||
91 | Cursor::Replace(self.placeholder_expr.syntax()), | ||
92 | ), | ||
85 | None => self.fn_def.to_string(), | 93 | None => self.fn_def.to_string(), |
86 | }; | 94 | }; |
87 | format!("{}{}{}", self.leading_ws, f, self.trailing_ws) | 95 | format!("{}{}{}", self.leading_ws, f, self.trailing_ws) |