crates/ra_assists/src/handlers/remove_mut.rs

   1 use ra_syntax::{SyntaxKind, TextRange, T};
   2
   3 use crate::{Assist, AssistCtx, AssistId};
   4
   5 // Assist: remove_mut
   6 //
   7 // Removes the `mut` keyword.
   8 //
   9 // ```
  10 // impl Walrus {
  11 //     fn feed(&mut<|> self, amount: u32) {}
  12 // }
  13 // ```
  14 // ->
  15 // ```
  16 // impl Walrus {
  17 //     fn feed(&self, amount: u32) {}
  18 // }
  19 // ```
  20 pub(crate) fn remove_mut(ctx: AssistCtx) -> Option<Assist> {
  21     let mut_token = ctx.find_token_at_offset(T![mut])?;
  22     let delete_from = mut_token.text_range().start();
  23     let delete_to = match mut_token.next_token() {
  24         Some(it) if it.kind() == SyntaxKind::WHITESPACE => it.text_range().end(),
  25         _ => mut_token.text_range().end(),
  26     };
  27
  28     let target = mut_token.text_range();
  29     ctx.add_assist(AssistId("remove_mut"), "Remove `mut` keyword", target, |edit| {
  30         edit.set_cursor(delete_from);
  31         edit.delete(TextRange::new(delete_from, delete_to));
  32     })
  33 }