mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Migrate promote_local_to_const to mutable ast

Browse Source
This commit is contained in:
DropDemBits 2023-07-09 18:49:29 -04:00
parent 6ab2788978
commit 27444eda56
No known key found for this signature in database
GPG Key ID: 7FE02A6C1EDFA075
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
crates/ide-assists/src/handlers/promote_local_to_const.rs
View File

@ -8,13 +8,10 @@
use stdx::to_upper_snake_case; use stdx::to_upper_snake_case;
use syntax::{ use syntax::{
ast::{self, make, HasName}, ast::{self, make, HasName},
AstNode, WalkEvent, ted, AstNode, WalkEvent,
}; };
use crate::{ use crate::assist_context::{AssistContext, Assists};
assist_context::{AssistContext, Assists},
utils::{render_snippet, Cursor},
};
// Assist: promote_local_to_const // Assist: promote_local_to_const
// //
@ -75,24 +72,28 @@ pub(crate) fn promote_local_to_const(acc: &mut Assists, ctx: &AssistContext<'_>)
AssistId("promote_local_to_const", AssistKind::Refactor), AssistId("promote_local_to_const", AssistKind::Refactor),
"Promote local to constant", "Promote local to constant",
target, target,
|builder| { |edit| {
let name = to_upper_snake_case(&name.to_string()); let name = to_upper_snake_case(&name.to_string());
let usages = Definition::Local(local).usages(&ctx.sema).all(); let usages = Definition::Local(local).usages(&ctx.sema).all();
if let Some(usages) = usages.references.get(&ctx.file_id()) { if let Some(usages) = usages.references.get(&ctx.file_id()) {
for usage in usages { for usage in usages {
builder.replace(usage.range, &name); let Some(usage) = usage.name.as_name_ref().cloned() else {
continue
};
let usage = edit.make_mut(usage);
ted::replace(usage.syntax(), make::name_ref(&name).clone_for_update().syntax());
} }
} }
let item = make::item_const(None, make::name(&name), make::ty(&ty), initializer); let item = make::item_const(None, make::name(&name), make::ty(&ty), initializer)
match ctx.config.snippet_cap.zip(item.name()) { .clone_for_update();
Some((cap, name)) => builder.replace_snippet( let let_stmt = edit.make_mut(let_stmt);
cap,
target, if let Some((cap, name)) = ctx.config.snippet_cap.zip(item.name()) {
render_snippet(cap, item.syntax(), Cursor::Before(name.syntax())), edit.add_tabstop_before(cap, name);
),
None => builder.replace(target, item.to_string()),
} }
ted::replace(let_stmt.syntax(), item.syntax());
}, },
) )
} }