From 4c8259e210bbb6c6c6e4781ce15d42a2f9c43f22 Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Sun, 23 May 2021 23:13:35 +0300
Subject: reduce duplication

---
 crates/ide_assists/src/handlers/generate_getter.rs | 173 +++++++++++++++++++--
 1 file changed, 160 insertions(+), 13 deletions(-)

(limited to 'crates/ide_assists/src/handlers/generate_getter.rs')

diff --git a/crates/ide_assists/src/handlers/generate_getter.rs b/crates/ide_assists/src/handlers/generate_getter.rs
index df7d1bb95..e01985112 100644
--- a/crates/ide_assists/src/handlers/generate_getter.rs
+++ b/crates/ide_assists/src/handlers/generate_getter.rs
@@ -29,6 +29,40 @@ use crate::{
 // }
 // ```
 pub(crate) fn generate_getter(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
+    generate_getter_impl(acc, ctx, false)
+}
+
+// Assist: generate_getter_mut
+//
+// Generate a mut getter method.
+//
+// ```
+// struct Person {
+//     nam$0e: String,
+// }
+// ```
+// ->
+// ```
+// struct Person {
+//     name: String,
+// }
+//
+// impl Person {
+//     /// Get a mutable reference to the person's name.
+//     fn name_mut(&mut self) -> &mut String {
+//         &mut self.name
+//     }
+// }
+// ```
+pub(crate) fn generate_getter_mut(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
+    generate_getter_impl(acc, ctx, true)
+}
+
+pub(crate) fn generate_getter_impl(
+    acc: &mut Assists,
+    ctx: &AssistContext,
+    mutable: bool,
+) -> Option<()> {
     let strukt = ctx.find_node_at_offset::<ast::Struct>()?;
     let field = ctx.find_node_at_offset::<ast::RecordField>()?;
 
@@ -37,22 +71,26 @@ pub(crate) fn generate_getter(acc: &mut Assists, ctx: &AssistContext) -> Option<
     let field_ty = field.ty()?;
 
     // Return early if we've found an existing fn
-    let fn_name = to_lower_snake_case(&field_name.to_string());
+    let mut fn_name = to_lower_snake_case(&field_name.to_string());
+    if mutable {
+        format_to!(fn_name, "_mut");
+    }
     let impl_def = find_struct_impl(&ctx, &ast::Adt::Struct(strukt.clone()), fn_name.as_str())?;
 
+    let (id, label) = if mutable {
+        ("generate_getter_mut", "Generate a mut getter method")
+    } else {
+        ("generate_getter", "Generate a getter method")
+    };
     let target = field.syntax().text_range();
     acc.add_group(
         &GroupLabel("Generate getter/setter".to_owned()),
-        AssistId("generate_getter", AssistKind::Generate),
-        "Generate a getter method",
+        AssistId(id, AssistKind::Generate),
+        label,
         target,
         |builder| {
             let mut buf = String::with_capacity(512);
 
-            let fn_name_spaced = fn_name.replace('_', " ");
-            let strukt_name_spaced =
-                to_lower_snake_case(&strukt_name.to_string()).replace('_', " ");
-
             if impl_def.is_some() {
                 buf.push('\n');
             }
@@ -60,16 +98,18 @@ pub(crate) fn generate_getter(acc: &mut Assists, ctx: &AssistContext) -> Option<
             let vis = strukt.visibility().map_or(String::new(), |v| format!("{} ", v));
             format_to!(
                 buf,
-                "    /// Get a reference to the {}'s {}.
-    {}fn {}(&self) -> &{} {{
-        &self.{}
+                "    /// Get a {}reference to the {}'s {}.
+    {}fn {}(&{mut_}self) -> &{mut_}{} {{
+        &{mut_}self.{}
     }}",
-                strukt_name_spaced,
-                fn_name_spaced,
+                mutable.then(|| "mutable ").unwrap_or_default(),
+                to_lower_snake_case(&strukt_name.to_string()).replace('_', " "),
+                fn_name.trim_end_matches("_mut").replace('_', " "),
                 vis,
                 fn_name,
                 field_ty,
-                fn_name,
+                field_name,
+                mut_ = mutable.then(|| "mut ").unwrap_or_default(),
             );
 
             let start_offset = impl_def
@@ -190,3 +230,110 @@ impl<T: Clone> Context<T> {
         );
     }
 }
+
+#[cfg(test)]
+mod tests_mut {
+    use crate::tests::{check_assist, check_assist_not_applicable};
+
+    use super::*;
+
+    fn check_not_applicable(ra_fixture: &str) {
+        check_assist_not_applicable(generate_getter_mut, ra_fixture)
+    }
+
+    #[test]
+    fn test_generate_getter_mut_from_field() {
+        check_assist(
+            generate_getter_mut,
+            r#"
+struct Context<T: Clone> {
+    dat$0a: T,
+}"#,
+            r#"
+struct Context<T: Clone> {
+    data: T,
+}
+
+impl<T: Clone> Context<T> {
+    /// Get a mutable reference to the context's data.
+    fn data_mut(&mut self) -> &mut T {
+        &mut self.data
+    }
+}"#,
+        );
+    }
+
+    #[test]
+    fn test_generate_getter_mut_already_implemented() {
+        check_not_applicable(
+            r#"
+struct Context<T: Clone> {
+    dat$0a: T,
+}
+
+impl<T: Clone> Context<T> {
+    fn data_mut(&mut self) -> &mut T {
+        &mut self.data
+    }
+}"#,
+        );
+    }
+
+    #[test]
+    fn test_generate_getter_mut_from_field_with_visibility_marker() {
+        check_assist(
+            generate_getter_mut,
+            r#"
+pub(crate) struct Context<T: Clone> {
+    dat$0a: T,
+}"#,
+            r#"
+pub(crate) struct Context<T: Clone> {
+    data: T,
+}
+
+impl<T: Clone> Context<T> {
+    /// Get a mutable reference to the context's data.
+    pub(crate) fn data_mut(&mut self) -> &mut T {
+        &mut self.data
+    }
+}"#,
+        );
+    }
+
+    #[test]
+    fn test_multiple_generate_getter_mut() {
+        check_assist(
+            generate_getter_mut,
+            r#"
+struct Context<T: Clone> {
+    data: T,
+    cou$0nt: usize,
+}
+
+impl<T: Clone> Context<T> {
+    /// Get a mutable reference to the context's data.
+    fn data_mut(&mut self) -> &mut T {
+        &mut self.data
+    }
+}"#,
+            r#"
+struct Context<T: Clone> {
+    data: T,
+    count: usize,
+}
+
+impl<T: Clone> Context<T> {
+    /// Get a mutable reference to the context's data.
+    fn data_mut(&mut self) -> &mut T {
+        &mut self.data
+    }
+
+    /// Get a mutable reference to the context's count.
+    fn count_mut(&mut self) -> &mut usize {
+        &mut self.count
+    }
+}"#,
+        );
+    }
+}
-- 
cgit v1.2.3


From c06599504bc4fcbb3d5024d9a8bdb14f263bad36 Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Sun, 23 May 2021 23:15:54 +0300
Subject: remove duplicate tests

---
 crates/ide_assists/src/handlers/generate_getter.rs | 129 +++++----------------
 1 file changed, 26 insertions(+), 103 deletions(-)

(limited to 'crates/ide_assists/src/handlers/generate_getter.rs')

diff --git a/crates/ide_assists/src/handlers/generate_getter.rs b/crates/ide_assists/src/handlers/generate_getter.rs
index e01985112..fbd47d761 100644
--- a/crates/ide_assists/src/handlers/generate_getter.rs
+++ b/crates/ide_assists/src/handlers/generate_getter.rs
@@ -130,10 +130,6 @@ mod tests {
 
     use super::*;
 
-    fn check_not_applicable(ra_fixture: &str) {
-        check_assist_not_applicable(generate_getter, ra_fixture)
-    }
-
     #[test]
     fn test_generate_getter_from_field() {
         check_assist(
@@ -154,121 +150,48 @@ impl<T: Clone> Context<T> {
     }
 }"#,
         );
-    }
-
-    #[test]
-    fn test_generate_getter_already_implemented() {
-        check_not_applicable(
-            r#"
-struct Context<T: Clone> {
-    dat$0a: T,
-}
-
-impl<T: Clone> Context<T> {
-    fn data(&self) -> &T {
-        &self.data
-    }
-}"#,
-        );
-    }
 
-    #[test]
-    fn test_generate_getter_from_field_with_visibility_marker() {
         check_assist(
-            generate_getter,
+            generate_getter_mut,
             r#"
-pub(crate) struct Context<T: Clone> {
+struct Context<T: Clone> {
     dat$0a: T,
 }"#,
             r#"
-pub(crate) struct Context<T: Clone> {
+struct Context<T: Clone> {
     data: T,
 }
 
 impl<T: Clone> Context<T> {
-    /// Get a reference to the context's data.
-    pub(crate) fn data(&self) -> &T {
-        &self.data
+    /// Get a mutable reference to the context's data.
+    fn data_mut(&mut self) -> &mut T {
+        &mut self.data
     }
 }"#,
         );
     }
 
     #[test]
-    fn test_multiple_generate_getter() {
-        check_assist(
+    fn test_generate_getter_already_implemented() {
+        check_assist_not_applicable(
             generate_getter,
             r#"
 struct Context<T: Clone> {
-    data: T,
-    cou$0nt: usize,
-}
-
-impl<T: Clone> Context<T> {
-    /// Get a reference to the context's data.
-    fn data(&self) -> &T {
-        &self.data
-    }
-}"#,
-            r#"
-struct Context<T: Clone> {
-    data: T,
-    count: usize,
+    dat$0a: T,
 }
 
 impl<T: Clone> Context<T> {
-    /// Get a reference to the context's data.
     fn data(&self) -> &T {
         &self.data
     }
-
-    /// Get a reference to the context's count.
-    fn count(&self) -> &usize {
-        &self.count
-    }
 }"#,
         );
-    }
-}
-
-#[cfg(test)]
-mod tests_mut {
-    use crate::tests::{check_assist, check_assist_not_applicable};
-
-    use super::*;
 
-    fn check_not_applicable(ra_fixture: &str) {
-        check_assist_not_applicable(generate_getter_mut, ra_fixture)
-    }
-
-    #[test]
-    fn test_generate_getter_mut_from_field() {
-        check_assist(
+        check_assist_not_applicable(
             generate_getter_mut,
             r#"
 struct Context<T: Clone> {
     dat$0a: T,
-}"#,
-            r#"
-struct Context<T: Clone> {
-    data: T,
-}
-
-impl<T: Clone> Context<T> {
-    /// Get a mutable reference to the context's data.
-    fn data_mut(&mut self) -> &mut T {
-        &mut self.data
-    }
-}"#,
-        );
-    }
-
-    #[test]
-    fn test_generate_getter_mut_already_implemented() {
-        check_not_applicable(
-            r#"
-struct Context<T: Clone> {
-    dat$0a: T,
 }
 
 impl<T: Clone> Context<T> {
@@ -280,9 +203,9 @@ impl<T: Clone> Context<T> {
     }
 
     #[test]
-    fn test_generate_getter_mut_from_field_with_visibility_marker() {
+    fn test_generate_getter_from_field_with_visibility_marker() {
         check_assist(
-            generate_getter_mut,
+            generate_getter,
             r#"
 pub(crate) struct Context<T: Clone> {
     dat$0a: T,
@@ -293,18 +216,18 @@ pub(crate) struct Context<T: Clone> {
 }
 
 impl<T: Clone> Context<T> {
-    /// Get a mutable reference to the context's data.
-    pub(crate) fn data_mut(&mut self) -> &mut T {
-        &mut self.data
+    /// Get a reference to the context's data.
+    pub(crate) fn data(&self) -> &T {
+        &self.data
     }
 }"#,
         );
     }
 
     #[test]
-    fn test_multiple_generate_getter_mut() {
+    fn test_multiple_generate_getter() {
         check_assist(
-            generate_getter_mut,
+            generate_getter,
             r#"
 struct Context<T: Clone> {
     data: T,
@@ -312,9 +235,9 @@ struct Context<T: Clone> {
 }
 
 impl<T: Clone> Context<T> {
-    /// Get a mutable reference to the context's data.
-    fn data_mut(&mut self) -> &mut T {
-        &mut self.data
+    /// Get a reference to the context's data.
+    fn data(&self) -> &T {
+        &self.data
     }
 }"#,
             r#"
@@ -324,14 +247,14 @@ struct Context<T: Clone> {
 }
 
 impl<T: Clone> Context<T> {
-    /// Get a mutable reference to the context's data.
-    fn data_mut(&mut self) -> &mut T {
-        &mut self.data
+    /// Get a reference to the context's data.
+    fn data(&self) -> &T {
+        &self.data
     }
 
-    /// Get a mutable reference to the context's count.
-    fn count_mut(&mut self) -> &mut usize {
-        &mut self.count
+    /// Get a reference to the context's count.
+    fn count(&self) -> &usize {
+        &self.count
     }
 }"#,
         );
-- 
cgit v1.2.3


From af54b1e248f377853957ada0270e269bedb577b4 Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Sun, 23 May 2021 23:19:00 +0300
Subject: minimize tests

---
 crates/ide_assists/src/handlers/generate_getter.rs | 98 ++++++++++++----------
 1 file changed, 54 insertions(+), 44 deletions(-)

(limited to 'crates/ide_assists/src/handlers/generate_getter.rs')

diff --git a/crates/ide_assists/src/handlers/generate_getter.rs b/crates/ide_assists/src/handlers/generate_getter.rs
index fbd47d761..c77395824 100644
--- a/crates/ide_assists/src/handlers/generate_getter.rs
+++ b/crates/ide_assists/src/handlers/generate_getter.rs
@@ -135,39 +135,43 @@ mod tests {
         check_assist(
             generate_getter,
             r#"
-struct Context<T: Clone> {
-    dat$0a: T,
-}"#,
+struct Context {
+    dat$0a: Data,
+}
+"#,
             r#"
-struct Context<T: Clone> {
-    data: T,
+struct Context {
+    data: Data,
 }
 
-impl<T: Clone> Context<T> {
+impl Context {
     /// Get a reference to the context's data.
-    fn data(&self) -> &T {
+    fn data(&self) -> &Data {
         &self.data
     }
-}"#,
+}
+"#,
         );
 
         check_assist(
             generate_getter_mut,
             r#"
-struct Context<T: Clone> {
-    dat$0a: T,
-}"#,
+struct Context {
+    dat$0a: Data,
+}
+"#,
             r#"
-struct Context<T: Clone> {
-    data: T,
+struct Context {
+    data: Data,
 }
 
-impl<T: Clone> Context<T> {
+impl Context {
     /// Get a mutable reference to the context's data.
-    fn data_mut(&mut self) -> &mut T {
+    fn data_mut(&mut self) -> &mut Data {
         &mut self.data
     }
-}"#,
+}
+"#,
         );
     }
 
@@ -176,29 +180,31 @@ impl<T: Clone> Context<T> {
         check_assist_not_applicable(
             generate_getter,
             r#"
-struct Context<T: Clone> {
-    dat$0a: T,
+struct Context {
+    dat$0a: Data,
 }
 
-impl<T: Clone> Context<T> {
-    fn data(&self) -> &T {
+impl Context {
+    fn data(&self) -> &Data {
         &self.data
     }
-}"#,
+}
+"#,
         );
 
         check_assist_not_applicable(
             generate_getter_mut,
             r#"
-struct Context<T: Clone> {
-    dat$0a: T,
+struct Context {
+    dat$0a: Data,
 }
 
-impl<T: Clone> Context<T> {
-    fn data_mut(&mut self) -> &mut T {
+impl Context {
+    fn data_mut(&mut self) -> &mut Data {
         &mut self.data
     }
-}"#,
+}
+"#,
         );
     }
 
@@ -207,20 +213,22 @@ impl<T: Clone> Context<T> {
         check_assist(
             generate_getter,
             r#"
-pub(crate) struct Context<T: Clone> {
-    dat$0a: T,
-}"#,
+pub(crate) struct Context {
+    dat$0a: Data,
+}
+"#,
             r#"
-pub(crate) struct Context<T: Clone> {
-    data: T,
+pub(crate) struct Context {
+    data: Data,
 }
 
-impl<T: Clone> Context<T> {
+impl Context {
     /// Get a reference to the context's data.
-    pub(crate) fn data(&self) -> &T {
+    pub(crate) fn data(&self) -> &Data {
         &self.data
     }
-}"#,
+}
+"#,
         );
     }
 
@@ -229,26 +237,27 @@ impl<T: Clone> Context<T> {
         check_assist(
             generate_getter,
             r#"
-struct Context<T: Clone> {
-    data: T,
+struct Context {
+    data: Data,
     cou$0nt: usize,
 }
 
-impl<T: Clone> Context<T> {
+impl Context {
     /// Get a reference to the context's data.
-    fn data(&self) -> &T {
+    fn data(&self) -> &Data {
         &self.data
     }
-}"#,
+}
+"#,
             r#"
-struct Context<T: Clone> {
-    data: T,
+struct Context {
+    data: Data,
     count: usize,
 }
 
-impl<T: Clone> Context<T> {
+impl Context {
     /// Get a reference to the context's data.
-    fn data(&self) -> &T {
+    fn data(&self) -> &Data {
         &self.data
     }
 
@@ -256,7 +265,8 @@ impl<T: Clone> Context<T> {
     fn count(&self) -> &usize {
         &self.count
     }
-}"#,
+}
+"#,
         );
     }
 }
-- 
cgit v1.2.3


From 8696c82777f9992fceadc531536bf90b64cf753d Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Sun, 23 May 2021 23:22:38 +0300
Subject: feat: generate getter assist places the cursor at the generated
 function

---
 crates/ide_assists/src/handlers/generate_getter.rs | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

(limited to 'crates/ide_assists/src/handlers/generate_getter.rs')

diff --git a/crates/ide_assists/src/handlers/generate_getter.rs b/crates/ide_assists/src/handlers/generate_getter.rs
index c77395824..9faaaf284 100644
--- a/crates/ide_assists/src/handlers/generate_getter.rs
+++ b/crates/ide_assists/src/handlers/generate_getter.rs
@@ -23,7 +23,7 @@ use crate::{
 //
 // impl Person {
 //     /// Get a reference to the person's name.
-//     fn name(&self) -> &String {
+//     fn $0name(&self) -> &String {
 //         &self.name
 //     }
 // }
@@ -49,7 +49,7 @@ pub(crate) fn generate_getter(acc: &mut Assists, ctx: &AssistContext) -> Option<
 //
 // impl Person {
 //     /// Get a mutable reference to the person's name.
-//     fn name_mut(&mut self) -> &mut String {
+//     fn $0name_mut(&mut self) -> &mut String {
 //         &mut self.name
 //     }
 // }
@@ -119,7 +119,12 @@ pub(crate) fn generate_getter_impl(
                     strukt.syntax().text_range().end()
                 });
 
-            builder.insert(start_offset, buf);
+            match ctx.config.snippet_cap {
+                Some(cap) => {
+                    builder.insert_snippet(cap, start_offset, buf.replacen("fn ", "fn $0", 1))
+                }
+                None => builder.insert(start_offset, buf),
+            }
         },
     )
 }
@@ -146,7 +151,7 @@ struct Context {
 
 impl Context {
     /// Get a reference to the context's data.
-    fn data(&self) -> &Data {
+    fn $0data(&self) -> &Data {
         &self.data
     }
 }
@@ -167,7 +172,7 @@ struct Context {
 
 impl Context {
     /// Get a mutable reference to the context's data.
-    fn data_mut(&mut self) -> &mut Data {
+    fn $0data_mut(&mut self) -> &mut Data {
         &mut self.data
     }
 }
@@ -224,7 +229,7 @@ pub(crate) struct Context {
 
 impl Context {
     /// Get a reference to the context's data.
-    pub(crate) fn data(&self) -> &Data {
+    pub(crate) fn $0data(&self) -> &Data {
         &self.data
     }
 }
@@ -262,7 +267,7 @@ impl Context {
     }
 
     /// Get a reference to the context's count.
-    fn count(&self) -> &usize {
+    fn $0count(&self) -> &usize {
         &self.count
     }
 }
-- 
cgit v1.2.3