make postfix completion handle all references correctly

Author: Natural-selection1

crates/ide-completion/src/completions/postfix.rs

@@ -360,10 +360,18 @@ fn include_references(initial_element: &ast::Expr) -> (ast::Expr, String) {
         resulting_element.syntax().parent().and_then(ast::RefExpr::cast)
     {
         found_ref_or_deref = true;
-        let exclusive = parent_ref_element.mut_token().is_some();
+        let is_mut = parent_ref_element.mut_token().is_some();
+        let is_raw = parent_ref_element.raw_token().is_some();
         resulting_element = ast::Expr::from(parent_ref_element);
-
-        prefix.insert_str(0, if exclusive { "&mut " } else { "&" });
+        prefix.insert_str(
+            0,
+            match (is_raw, is_mut) {
+                (true, true) => "&raw mut ",
+                (true, false) => "&raw const ",
+                (false, true) => "&mut ",
+                (false, false) => "&",
+            },
+        );
     }
 
     if !found_ref_or_deref {
@@ -1005,6 +1013,20 @@ fn main() {
             r#"fn main() { Ok(&&42) }"#,
         );
 
+        check_edit_with_config(
+            CompletionConfig { snippets: vec![snippet.clone()], ..TEST_CONFIG },
+            "ok",
+            r#"fn main() { &raw mut 42.$0 }"#,
+            r#"fn main() { Ok(&raw mut 42) }"#,
+        );
+
+        check_edit_with_config(
+            CompletionConfig { snippets: vec![snippet.clone()], ..TEST_CONFIG },
+            "ok",
+            r#"fn main() { &raw const 42.$0 }"#,
+            r#"fn main() { Ok(&raw const 42) }"#,
+        );
+
         check_edit_with_config(
             CompletionConfig { snippets: vec![snippet], ..TEST_CONFIG },
             "ok",
@@ -1031,6 +1053,56 @@ fn main() {
         );
     }
 
+    #[test]
+    fn postfix_custom_snippets_completion_for_row_references() {
+        // https://github.com/rust-lang/rust-analyzer/issues/21035
+
+        let snippet = Snippet::new(
+            &[],
+            &["group".into()],
+            &["(${receiver})".into()],
+            "",
+            &[],
+            crate::SnippetScope::Expr,
+        )
+        .unwrap();
+
+        check_edit_with_config(
+            CompletionConfig { snippets: vec![snippet.clone()], ..TEST_CONFIG },
+            "group",
+            r#"fn main() { &42.g$0 }"#,
+            r#"fn main() { (&42) }"#,
+        );
+
+        check_edit_with_config(
+            CompletionConfig { snippets: vec![snippet.clone()], ..TEST_CONFIG },
+            "group",
+            r#"fn main() { &&42.$0 }"#,
+            r#"fn main() { (&&42) }"#,
+        );
+
+        check_edit_with_config(
+            CompletionConfig { snippets: vec![snippet.clone()], ..TEST_CONFIG },
+            "group",
+            r#"fn main() { &mut 42.$0 }"#,
+            r#"fn main() { (&mut 42) }"#,
+        );
+
+        check_edit_with_config(
+            CompletionConfig { snippets: vec![snippet.clone()], ..TEST_CONFIG },
+            "group",
+            r#"fn main() { &raw mut 42.$0 }"#,
+            r#"fn main() { (&raw mut 42) }"#,
+        );
+
+        check_edit_with_config(
+            CompletionConfig { snippets: vec![snippet.clone()], ..TEST_CONFIG },
+            "group",
+            r#"fn main() { &raw const 42.$0 }"#,
+            r#"fn main() { (&raw const 42) }"#,
+        );
+    }
+
     #[test]
     fn no_postfix_completions_in_if_block_that_has_an_else() {
         check(