From ac505b21a8f3f1f03bfeed9209cfcf920c223852 Mon Sep 17 00:00:00 2001
From: Lukas Wirth <lukastw97@gmail.com>
Date: Fri, 15 Oct 2021 19:27:19 +0200
Subject: [PATCH] Fix missing_fields diagnostic fix replacing wrong text ranges

---
 .../src/handlers/missing_fields.rs            | 22 +++++++++++++------
 1 file changed, 15 insertions(+), 7 deletions(-)

diff --git a/crates/ide_diagnostics/src/handlers/missing_fields.rs b/crates/ide_diagnostics/src/handlers/missing_fields.rs
index d5b84715156..3d416591fe6 100644
--- a/crates/ide_diagnostics/src/handlers/missing_fields.rs
+++ b/crates/ide_diagnostics/src/handlers/missing_fields.rs
@@ -54,6 +54,7 @@ fn fixes(ctx: &DiagnosticsContext<'_>, d: &hir::MissingFields) -> Option<Vec<Ass
         Either::Right(_) => return None,
     };
     let old_field_list = field_list_parent.record_expr_field_list()?;
+
     let new_field_list = old_field_list.clone_for_update();
     let mut locals = FxHashMap::default();
     ctx.sema.scope(field_list_parent.syntax()).process_all_names(&mut |name, def| {
@@ -80,11 +81,19 @@ fn fixes(ctx: &DiagnosticsContext<'_>, d: &hir::MissingFields) -> Option<Vec<Ass
         new_field_list.add_field(field);
     }
 
-    let edit = {
-        let mut builder = TextEdit::builder();
+    let mut builder = TextEdit::builder();
+    if d.file.is_macro() {
+        // we can't map the diff up into the macro input unfortunately, as the macro loses all
+        // whitespace information so the diff wouldn't be applicable no matter what
+        // This has the downside that the cursor will be moved in macros by doing it without a diff
+        // but that is a trade off we can make.
+        // FIXE: this also currently discards a lot of whitespace in the input... we really need a formatter here
+        let range = ctx.sema.original_range_opt(old_field_list.syntax())?;
+        builder.replace(range.range, new_field_list.to_string());
+    } else {
         algo::diff(old_field_list.syntax(), new_field_list.syntax()).into_text_edit(&mut builder);
-        builder.finish()
-    };
+    }
+    let edit = builder.finish();
     Some(vec![fix(
         "fill_missing_fields",
         "Fill struct fields",
@@ -151,7 +160,6 @@ fn x(a: S) {
 
     #[test]
     fn range_mapping_out_of_macros() {
-        // FIXME: this is very wrong, but somewhat tricky to fix.
         check_fix(
             r#"
 fn some() {}
@@ -167,14 +175,14 @@ fn main() {
 pub struct Foo { pub a: i32, pub b: i32 }
 "#,
             r#"
-fn some(, b: todo!() ) {}
+fn some() {}
 fn items() {}
 fn here() {}
 
 macro_rules! id { ($($tt:tt)*) => { $($tt)*}; }
 
 fn main() {
-    let _x = id![Foo { a: 42 }];
+    let _x = id![Foo {a:42, b: todo!() }];
 }
 
 pub struct Foo { pub a: i32, pub b: i32 }