rust-analyzer/crates/ra_assists/src/handlers/remove_mut.rs

34 lines
912 B
Rust
Raw Normal View History

2020-02-19 11:44:20 +00:00
use ra_syntax::{SyntaxKind, TextRange, T};
use crate::{AssistContext, AssistId, Assists};
2020-02-19 11:44:20 +00:00
// Assist: remove_mut
//
// Removes the `mut` keyword.
//
// ```
// impl Walrus {
// fn feed(&mut<|> self, amount: u32) {}
// }
// ```
// ->
// ```
// impl Walrus {
// fn feed(&self, amount: u32) {}
// }
// ```
pub(crate) fn remove_mut(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
2020-02-19 11:44:20 +00:00
let mut_token = ctx.find_token_at_offset(T![mut])?;
let delete_from = mut_token.text_range().start();
let delete_to = match mut_token.next_token() {
Some(it) if it.kind() == SyntaxKind::WHITESPACE => it.text_range().end(),
_ => mut_token.text_range().end(),
};
let target = mut_token.text_range();
acc.add(AssistId("remove_mut"), "Remove `mut` keyword", target, |edit| {
2020-02-19 11:44:20 +00:00
edit.set_cursor(delete_from);
2020-04-24 21:40:41 +00:00
edit.delete(TextRange::new(delete_from, delete_to));
2020-02-19 11:44:20 +00:00
})
}