make it work from macro

diff --git a/crates/ide_assists/src/handlers/add_missing_match_arms.rs b/crates/ide_assists/src/handlers/add_missing_match_arms.rs
index 8cb3e8fe42378b4763e5f3350d188106c11d1f39..0cfe63a18293a6bd18e26f718eaa0ca63d04dbf4 100644 (file)
--- a/crates/ide_assists/src/handlers/add_missing_match_arms.rs
+++ b/crates/ide_assists/src/handlers/add_missing_match_arms.rs
@@ -30,7 +30,7 @@
 // enum Action { Move { distance: u32 }, Stop }
 //
 // fn handle(action: Action) {
-//     match action {
+//     match action  {
 //         $0Action::Move { distance } => todo!(),
 //         Action::Stop => todo!(),
 //     }
@@ -41,8 +41,8 @@ pub(crate) fn add_missing_match_arms(acc: &mut Assists, ctx: &AssistContext) ->
     let match_arm_list = match_expr.match_arm_list()?;
 
     let available_range = TextRange::new(
-        match_expr.syntax().text_range().start(),
-        match_arm_list.syntax().text_range().start(),
+        ctx.sema.original_range(match_expr.syntax()).range.start(),
+        ctx.sema.original_range(match_arm_list.syntax()).range.start(),
     );
 
     let cursor_in_range = available_range.contains_range(ctx.frange.range);