aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_assists/src/handlers/add_function.rs
diff options
context:
space:
mode:
authorAleksey Kladov <[email protected]>2020-05-20 00:53:21 +0100
committerAleksey Kladov <[email protected]>2020-05-20 00:53:21 +0100
commita04cababaa144d7a6db7b1dd114494b33d281ab9 (patch)
tree70184acba3f9560bf9a6ed0eae889b9e53a85b6c /crates/ra_assists/src/handlers/add_function.rs
parente6fc0bdffb213f6e94c5bb4081e6d175ccbd518f (diff)
Use snippets in add_missing_members
Diffstat (limited to 'crates/ra_assists/src/handlers/add_function.rs')
-rw-r--r--crates/ra_assists/src/handlers/add_function.rs12
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};
11use rustc_hash::{FxHashMap, FxHashSet}; 11use rustc_hash::{FxHashMap, FxHashSet};
12 12
13use crate::{assist_config::SnippetCap, utils::render_snippet, AssistContext, AssistId, Assists}; 13use 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 {
81impl FunctionTemplate { 85impl 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)