user0/rust
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: convert add_braces to SyntaxFactory SyntaxEditor abstraction

Browse source
This commit is contained in:
David Kurilla 2024-11-06 14:45:00 +00:00
parent 20864efcdf
commit c588849044
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/tools/rust-analyzer/crates/ide-assists/src/handlers/add_braces.rs
View file

@ -1,5 +1,5 @@
use syntax::{
ast::{self, edit::AstNodeEdit, make},
ast::{self, edit::AstNodeEdit, make, syntax_factory::SyntaxFactory}, syntax_editor::SyntaxEditor,
AstNode,
};
@ -39,12 +39,19 @@ pub(crate) fn add_braces(acc: &mut Assists, ctx: &AssistContext<'_>) -> Option<(
},
expr.syntax().text_range(),
|builder| {
let make = SyntaxFactory::new();
let mut editor = builder.make_editor(&expr.syntax());
let block_expr = AstNodeEdit::indent(
&make::block_expr(None, Some(expr.clone())),
&make.block_expr(None, Some(expr.clone())),
AstNodeEdit::indent_level(&expr),
);
builder.replace(expr.syntax().text_range(), block_expr.syntax().text());
editor.replace(expr.syntax(), block_expr.syntax());
editor.add_mappings(make.finish_with_mappings());
builder.add_file_edits(ctx.file_id(), editor);
},
)
}