|
1 | 1 | use ide_db::syntax_helpers::suggest_name; |
2 | | -use syntax::{ |
3 | | - ast::{self, AstNode, make}, |
4 | | - ted, |
5 | | -}; |
| 2 | +use syntax::ast::{self, AstNode, syntax_factory::SyntaxFactory}; |
6 | 3 |
|
7 | 4 | use crate::{AssistContext, AssistId, Assists}; |
8 | 5 |
|
@@ -60,21 +57,25 @@ pub(crate) fn replace_is_method_with_if_let_method( |
60 | 57 | message, |
61 | 58 | call_expr.syntax().text_range(), |
62 | 59 | |edit| { |
63 | | - let call_expr = edit.make_mut(call_expr); |
| 60 | + let make = SyntaxFactory::with_mappings(); |
| 61 | + let mut editor = edit.make_editor(call_expr.syntax()); |
64 | 62 |
|
65 | | - let var_pat = make::ident_pat(false, false, make::name(&var_name)); |
66 | | - let pat = make::tuple_struct_pat(make::ext::ident_path(text), [var_pat.into()]); |
67 | | - let let_expr = make::expr_let(pat.into(), receiver).clone_for_update(); |
| 63 | + let var_pat = make.ident_pat(false, false, make.name(&var_name)); |
| 64 | + let pat = make.tuple_struct_pat(make.ident_path(text), [var_pat.into()]); |
| 65 | + let let_expr = make.expr_let(pat.into(), receiver); |
68 | 66 |
|
69 | 67 | if let Some(cap) = ctx.config.snippet_cap { |
70 | 68 | if let Some(ast::Pat::TupleStructPat(pat)) = let_expr.pat() { |
71 | 69 | if let Some(first_var) = pat.fields().next() { |
72 | | - edit.add_placeholder_snippet(cap, first_var); |
| 70 | + let placeholder = edit.make_placeholder_snippet(cap); |
| 71 | + editor.add_annotation(first_var.syntax(), placeholder); |
73 | 72 | } |
74 | 73 | } |
75 | 74 | } |
76 | 75 |
|
77 | | - ted::replace(call_expr.syntax(), let_expr.syntax()); |
| 76 | + editor.replace(call_expr.syntax(), let_expr.syntax()); |
| 77 | + editor.add_mappings(make.finish_with_mappings()); |
| 78 | + edit.add_file_edits(ctx.vfs_file_id(), editor); |
78 | 79 | }, |
79 | 80 | ) |
80 | 81 | } |
|
0 commit comments