add apply ssr assist

Author: JoshMcguigan

Cargo.lock

@@ -19,6 +19,7 @@ syntax = { path = "../syntax", version = "0.0.0" }
 text_edit = { path = "../text_edit", version = "0.0.0" }
 profile = { path = "../profile", version = "0.0.0" }
 ide_db = { path = "../ide_db", version = "0.0.0" }
+ide_ssr = { path = "../ide_ssr", version = "0.0.0" }
 hir = { path = "../hir", version = "0.0.0" }
 test_utils = { path = "../test_utils", version = "0.0.0" }
 

crates/ide_assists/Cargo.toml

@@ -0,0 +1,117 @@
+use ide_db::base_db::FilePosition;
+use syntax::{ast, AstToken};
+
+use crate::{AssistContext, AssistId, AssistKind, Assists, GroupLabel};
+
+/// Assist: apply_ssr
+///
+/// Applies structural search and replace
+///
+/// ```
+/// fn foo() {}
+/// fn bar() {}
+///
+/// fn baz() { foo(); }
+///
+/// // foo() ==>> bar() $0
+/// ```
+///
+/// ```
+/// fn foo() {}
+/// fn bar() {}
+///
+/// fn baz() { bar(); }
+///
+/// // foo() ==>> bar()
+/// ```
+pub(crate) fn apply_ssr(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
+    let comment = ctx.find_token_at_offset::<ast::Comment>()?;
+    let target = comment.syntax().text_range();
+
+    let lookup_context = FilePosition { file_id: ctx.frange.file_id, offset: ctx.offset() };
+
+    let comment_text_without_prefix = comment.text().strip_prefix(comment.prefix()).unwrap();
+    // Attempt to parse the comment line as an SSR rule. If the parsing fails
+    // this assist is not applicable.
+    let ssr_rule = comment_text_without_prefix.parse().ok()?;
+
+    let mut match_finder = ide_ssr::MatchFinder::in_context(ctx.sema.db, lookup_context, vec![]);
+    match_finder.add_rule(ssr_rule).ok()?;
+
+    let group = GroupLabel("Apply SSR".into());
+
+    acc.add_group(
+        &group,
+        AssistId("ssr", AssistKind::RefactorRewrite),
+        "Apply SSR in workspace",
+        target,
+        |assist_builder| {
+            let edits = match_finder.edits();
+
+            for (file, edit) in edits {
+                assist_builder.edit_file(file);
+                for indel in edit {
+                    assist_builder.replace(indel.delete, indel.insert);
+                }
+            }
+        },
+    );
+
+    acc.add_group(
+        &group,
+        AssistId("ssr", AssistKind::RefactorRewrite),
+        "Apply SSR in file",
+        target,
+        |assist_builder| {
+            let edits = match_finder.edits();
+
+            for (file, edit) in edits {
+                if file != ctx.frange.file_id {
+                    continue;
+                }
+
+                assist_builder.edit_file(file);
+                for indel in edit {
+                    assist_builder.replace(indel.delete, indel.insert);
+                }
+            }
+        },
+    );
+
+    Some(())
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::tests::check_assist_by_label;
+
+    use super::*;
+
+    #[test]
+    fn in_file() {
+        check_assist_by_label(
+            apply_ssr,
+            r#"
+fn foo() {}
+fn bar() {}
+
+// foo() ==>>$0 bar()
+
+fn main() {
+    foo();
+}
+"#,
+            r#"
+fn foo() {}
+fn bar() {}
+
+// foo() ==>> bar()
+
+fn main() {
+    bar();
+}
+"#,
+            "Apply SSR in file",
+        );
+    }
+}

crates/ide_assists/src/handlers/apply_ssr.rs

@@ -112,6 +112,7 @@ mod handlers {
     mod add_missing_impl_members;
     mod add_turbo_fish;
     mod apply_demorgan;
+    mod apply_ssr;
     mod auto_import;
     mod change_visibility;
     mod convert_integer_literal;
@@ -176,6 +177,7 @@ mod handlers {
             add_lifetime_to_type::add_lifetime_to_type,
             add_turbo_fish::add_turbo_fish,
             apply_demorgan::apply_demorgan,
+            apply_ssr::apply_ssr,
             auto_import::auto_import,
             change_visibility::change_visibility,
             convert_integer_literal::convert_integer_literal,